summaryrefslogtreecommitdiffstats
path: root/configutil/src/apps/modelinspect/main.cpp
blob: f5a1e28e21862a4a671dc6fce5a3d67d57968ca1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <vespa/fastos/fastos.h>
#include <vespa/defaults.h>
#include <vespa/log/log.h>
LOG_SETUP("vespa-model-inspect");
#include <iostream>
#include <lib/modelinspect.h>
#include <vespa/vespalib/text/stringtokenizer.h>
#include <vespa/vespalib/stllike/asciistream.h>

class Application : public FastOS_Application
{
    ModelInspect::Flags _flags;
    vespalib::string _cfgId;
    vespalib::string _specString;
    int parseOpts();
    vespalib::string getSources();
public:
    void usage(void);
    int Main(void);

    Application() : _flags(), _cfgId("admin/model"), _specString("") {}
};

int
Application::parseOpts()
{
    char c = '?';
    const char *optArg = NULL;
    int optInd = 0;
    while ((c = GetOpt("hvut:c:C:", optArg, optInd)) != -1) {
        switch (c) {
        case 'v':
            _flags.verbose = true;
            break;
        case 'u':
            _flags.makeuri = true;
            break;
        case 't':
            _flags.tagFilter.push_back(optArg);
            _flags.tagfilt = true;
            break;
        case 'C':
            _cfgId = optArg;
            break;
        case 'c':
            _specString = optArg;
            break;
        case 'h':
            return _argc;
        default:
            usage();
            exit(1);
        }
    }
    if (_specString.empty()) {
        _specString = getSources();
    }
    return optInd;
}

vespalib::string
Application::getSources(void)
{
    vespalib::string cmd = vespa::Defaults::vespaHome();
    cmd.append("libexec/vespa/vespa-config.pl -configsources");
    FILE* fp = popen(cmd.c_str(), "r");
    if (fp == 0) {
        std::cerr << "Failed to run " << cmd << " ("
                  << errno << "): " << strerror(errno) << "\n";
        return "";
    }
    vespalib::asciistream specs;
    char data[500];
    while (fgets(data, 500, fp) != 0) {
        specs << &data[0] << "\n";
    }
    pclose(fp);
    return specs.str();
}

void
Application::usage(void)
{
    std::cerr <<
        "vespa-model-inspect version 2.0"                                  << std::endl <<
        "Usage: " << _argv[0] << " [options] <command> <options>"          << std::endl <<
        "options: [-u] for URLs, [-v] for verbose"                         << std::endl <<
        "         [-c host] or [-c host:port] to specify server"           << std::endl <<
        "         [-t tag] to filter on a port tag"                        << std::endl <<
        "Where command is:"                                                << std::endl <<
        "    hosts - show all hosts"                                       << std::endl <<
        "    services - show all services"                                 << std::endl <<
        "    clusters - show all cluster names"                            << std::endl <<
        "    configids - show all config IDs"                              << std::endl <<
        "    filter:ports - list ports matching filter options"            << std::endl <<
        "    host <hostname> - show services on a given host"              << std::endl <<
        "    service [cluster:]<servicetype>"                                           <<
        " - show all instances of a given servicetype"                     << std::endl <<
        "    cluster <clustername>"                                                     <<
        " - show all services associated with the cluster"                 << std::endl <<
        "    configid <configid>"                                                       <<
        " - show service using configid"                                   << std::endl <<
        "    get-index-of <servicetype> <host>"                                         <<
        " - show all indexes for instances of the servicetype on the host" << std::endl <<
        std::endl;
}

int
Application::Main(void)
{
    int cnt = parseOpts();
    if (_argc == cnt) {
        usage();
        return 0;
    }

    config::ServerSpec spec(_specString);
    config::ConfigUri uri = config::ConfigUri::createFromSpec(_cfgId, spec);
    ModelInspect model(_flags, uri, std::cout);
    return model.action(_argc - cnt, &_argv[cnt]);
}

int
main(int argc, char** argv)
{
    vespa::Defaults::bootstrap(argv[0]);
    Application app;
    return app.Entry(argc, argv);
}