ceph/src/tools/rbd_mirror/main.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3
   4 #include "common/ceph_argparse.h"
   5 #include "common/config.h"
   6 #include "common/debug.h"
   7 #include "common/errno.h"
   8 #include "common/perf_counters.h"
   9 #include "global/global_init.h"
  10 #include "global/signal_handler.h"
  11 #include "Mirror.h"
  12 #include "Types.h"
  13
  14 #include <vector>
  15
  16 rbd::mirror::Mirror *mirror = nullptr;
  17 PerfCounters *g_perf_counters = nullptr;
  18
  19 void usage() {
  20   std::cout << "usage: rbd-mirror [options...]" << std::endl;
  21   std::cout << "options:\n";
  22   std::cout << "  -m monaddress[:port]      connect to specified monitor\n";
  23   std::cout << "  --keyring=<path>          path to keyring for local cluster\n";
  24   std::cout << "  --log-file=<logfile>       file to log debug output\n";
  25   std::cout << "  --debug-rbd-mirror=<log-level>/<memory-level>  set rbd-mirror debug level\n";
  26   generic_server_usage();
  27 }
  28
  29 static void handle_signal(int signum)
  30 {
  31   if (mirror)
  32     mirror->handle_signal(signum);
  33 }
  34
  35 int main(int argc, const char **argv)
  36 {
  37   std::vector<const char*> args;
  38   argv_to_vec(argc, argv, args);
  39   if (args.empty()) {
  40     cerr << argv[0] << ": -h or --help for usage" << std::endl;
  41     exit(1);
  42   }
  43   if (ceph_argparse_need_usage(args)) {
  44     usage();
  45     exit(0);
  46   }
  47
  48   auto cct = global_init(nullptr, args, CEPH_ENTITY_TYPE_CLIENT,
  49                          CODE_ENVIRONMENT_DAEMON,
  50                          CINIT_FLAG_UNPRIVILEGED_DAEMON_DEFAULTS);
  51
  52   if (g_conf()->daemonize) {
  53     global_init_daemonize(g_ceph_context);
  54   }
  55
  56   common_init_finish(g_ceph_context);
  57
  58   init_async_signal_handler();
  59   register_async_signal_handler(SIGHUP, sighup_handler);
  60   register_async_signal_handler_oneshot(SIGINT, handle_signal);
  61   register_async_signal_handler_oneshot(SIGTERM, handle_signal);
  62
  63   std::vector<const char*> cmd_args;
  64   argv_to_vec(argc, argv, cmd_args);
  65
  66   // disable unnecessary librbd cache
  67   g_ceph_context->_conf.set_val_or_die("rbd_cache", "false");
  68
  69   auto prio =
  70       g_ceph_context->_conf.get_val<int64_t>("rbd_mirror_perf_stats_prio");
  71   PerfCountersBuilder plb(g_ceph_context, "rbd_mirror",
  72                           rbd::mirror::l_rbd_mirror_first,
  73                           rbd::mirror::l_rbd_mirror_last);
  74   plb.add_u64_counter(rbd::mirror::l_rbd_mirror_replay, "replay", "Replays",
  75                       "r", prio);
  76   plb.add_u64_counter(rbd::mirror::l_rbd_mirror_replay_bytes, "replay_bytes",
  77                       "Replayed data", "rb", prio, unit_t(UNIT_BYTES));
  78   plb.add_time_avg(rbd::mirror::l_rbd_mirror_replay_latency, "replay_latency",
  79                    "Replay latency", "rl", prio);
  80   g_perf_counters = plb.create_perf_counters();
  81   g_ceph_context->get_perfcounters_collection()->add(g_perf_counters);
  82
  83   mirror = new rbd::mirror::Mirror(g_ceph_context, cmd_args);
  84   int r = mirror->init();
  85   if (r < 0) {
  86     std::cerr << "failed to initialize: " << cpp_strerror(r) << std::endl;
  87     goto cleanup;
  88   }
  89
  90   mirror->run();
  91
  92  cleanup:
  93   unregister_async_signal_handler(SIGHUP, sighup_handler);
  94   unregister_async_signal_handler(SIGINT, handle_signal);
  95   unregister_async_signal_handler(SIGTERM, handle_signal);
  96   shutdown_async_signal_handler();
  97
  98   g_ceph_context->get_perfcounters_collection()->remove(g_perf_counters);
  99
 100   delete mirror;
 101   delete g_perf_counters;
 102
 103   return r < 0 ? EXIT_SUCCESS : EXIT_FAILURE;
 104 }