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
|
#include "common/ceph_argparse.h"
#include "common/config.h"
#include "common/debug.h"
#include "global/global_init.h"
#include "global/global_context.h"
#include "global/signal_handler.h"
#include "exporter/DaemonMetricCollector.h"
#include "exporter/web_server.h"
#include <boost/thread/thread.hpp>
#include <iostream>
#include <map>
#include <string>
#include <atomic>
#include <chrono>
#include <thread>
#define dout_context g_ceph_context
#define dout_subsys ceph_subsys_ceph_exporter
DaemonMetricCollector &collector = collector_instance();
static void handle_signal(int signum)
{
ceph_assert(signum == SIGINT || signum == SIGTERM);
derr << "*** Got signal " << sig_str(signum) << " ***" << dendl;
// Finish the DaemonMetricCollector
collector.shutdown();
}
static void usage() {
std::cout << "usage: ceph-exporter [options]\n"
<< "options:\n"
" --sock-dir: The path to ceph daemons socket files dir\n"
" --addrs: Host ip address where exporter is deployed\n"
" --port: Port to deploy exporter on. Default is 9926\n"
" --cert-file: Path to the certificate file to use https\n"
" --key-file: Path to the certificate key file to use https\n"
" --prio-limit: Only perf counters greater than or equal to prio-limit are fetched. Default: 5\n"
" --stats-period: Time to wait before sending requests again to exporter server (seconds). Default: 5s"
<< std::endl;
generic_server_usage();
}
int main(int argc, char **argv) {
auto args = argv_to_vec(argc, argv);
if (args.empty()) {
std::cerr << argv[0] << ": -h or --help for usage" << std::endl;
exit(1);
}
if (ceph_argparse_need_usage(args)) {
usage();
exit(0);
}
auto cct = global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT,
CODE_ENVIRONMENT_DAEMON, 0);
std::string val;
for (auto i = args.begin(); i != args.end();) {
if (ceph_argparse_double_dash(args, i)) {
break;
} else if (ceph_argparse_witharg(args, i, &val, "--sock-dir", (char *)NULL)) {
cct->_conf.set_val("exporter_sock_dir", val);
} else if (ceph_argparse_witharg(args, i, &val, "--addrs", (char *)NULL)) {
cct->_conf.set_val("exporter_addr", val);
} else if (ceph_argparse_witharg(args, i, &val, "--port", (char *)NULL)) {
cct->_conf.set_val("exporter_http_port", val);
} else if (ceph_argparse_witharg(args, i, &val, "--cert-file", (char *)NULL)) {
cct->_conf.set_val("exporter_cert_file", val);
} else if (ceph_argparse_witharg(args, i, &val, "--key-file", (char *)NULL)) {
cct->_conf.set_val("exporter_key_file", val);
} else if (ceph_argparse_witharg(args, i, &val, "--prio-limit", (char *)NULL)) {
cct->_conf.set_val("exporter_prio_limit", val);
} else if (ceph_argparse_witharg(args, i, &val, "--stats-period", (char *)NULL)) {
cct->_conf.set_val("exporter_stats_period", val);
} else {
++i;
}
}
common_init_finish(g_ceph_context);
// Register signal handlers
init_async_signal_handler();
register_async_signal_handler(SIGHUP, sighup_handler);
register_async_signal_handler_oneshot(SIGINT, handle_signal);
register_async_signal_handler_oneshot(SIGTERM, handle_signal);
// Start the web server thread
boost::thread server_thread(web_server_thread_entrypoint);
// Start the DaemonMetricCollector
collector.main();
// Interrupted. Time to terminate
unregister_async_signal_handler(SIGHUP, sighup_handler);
unregister_async_signal_handler(SIGINT, handle_signal);
unregister_async_signal_handler(SIGTERM, handle_signal);
shutdown_async_signal_handler();
// Stop the web server thread by interrupting it
stop_web_server();
server_thread.interrupt(); // Interrupt the web server thread
server_thread.join();
dout(1) << "Ceph exporter stopped" << dendl;
return 0;
}
|
