aboutsummaryrefslogtreecommitdiffstats
path: root/configutil/src/apps/configstatus/main.cpp
blob: 6cc1b8f12408f8c67382b6b4ee215c7f706640b5 (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
// 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-config-status");
#include <iostream>
#include <lib/configstatus.h>

class Application : public FastOS_Application
{
    ConfigStatus::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("c:s:vC:", optArg, optInd)) != -1) {
        switch (c) {
        case 'v':
            _flags.verbose = true;
            break;
        case 'C':
            _cfgId = optArg;
            break;
        case 'c':
            _specString = optArg;
            break;
        case 'h':
            usage();
            exit(0);
        default:
            usage();
            exit(1);
        }
    }
    if (_specString.empty()) {
        _specString = getSources();
    }
    return optInd;
}


void
Application::usage(void)
{
    std::cerr <<
        "vespa-config-status version 1.0"                                  << std::endl <<
        "Usage: " << _argv[0] << " [options] "                             << std::endl <<
        "options: [-v] for verbose"                                        << std::endl <<
        "         [-c host] or [-c host:port] to specify config server"    << std::endl <<
        std::endl;
}

int
Application::Main(void)
{
    parseOpts();

    config::ServerSpec spec(_specString);
    config::ConfigUri uri = config::ConfigUri::createFromSpec(_cfgId, spec);
    ConfigStatus status(_flags, uri);

    return status.action();
}

vespalib::string
Application::getSources(void)
{
    vespalib::string specs;
    for (std::string v : vespa::Defaults::vespaConfigSourcesRpcAddrs()) {
        if (! specs.empty()) specs += ",";
        specs += v;
    }
    return specs;
}

int
main(int argc, char **argv)
{
    Application app;
    return app.Entry(argc, argv);
}