src/ceph/src/mgr/ServiceMap.h

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3
   4 #pragma once
   5
   6 #include <string>
   7 #include <map>
   8 #include <list>
   9 #include <sstream>
  10
  11 #include "include/utime.h"
  12 #include "include/buffer.h"
  13 #include "msg/msg_types.h"
  14
  15 namespace ceph {
  16   class Formatter;
  17 }
  18
  19 struct ServiceMap {
  20   struct Daemon {
  21     uint64_t gid = 0;
  22     entity_addr_t addr;
  23     epoch_t start_epoch = 0;   ///< epoch first registered
  24     utime_t start_stamp;       ///< timestamp daemon started/registered
  25     std::map<std::string,std::string> metadata;  ///< static metadata
  26
  27     void encode(bufferlist& bl, uint64_t features) const;
  28     void decode(bufferlist::iterator& p);
  29     void dump(Formatter *f) const;
  30     static void generate_test_instances(std::list<Daemon*>& ls);
  31   };
  32
  33   struct Service {
  34     map<std::string,Daemon> daemons;
  35     std::string summary;   ///< summary status string for 'ceph -s'
  36
  37     void encode(bufferlist& bl, uint64_t features) const;
  38     void decode(bufferlist::iterator& p);
  39     void dump(Formatter *f) const;
  40     static void generate_test_instances(std::list<Service*>& ls);
  41
  42     std::string get_summary() const {
  43       if (summary.size()) {
  44         return summary;
  45       }
  46       if (daemons.empty()) {
  47         return "no daemons active";
  48       }
  49       std::ostringstream ss;
  50       ss << daemons.size() << (daemons.size() > 1 ? " daemons" : " daemon")
  51          << " active";
  52       return ss.str();
  53     }
  54
  55     void count_metadata(const string& field,
  56                         std::map<std::string,int> *out) const {
  57       for (auto& p : daemons) {
  58         auto q = p.second.metadata.find(field);
  59         if (q == p.second.metadata.end()) {
  60           (*out)["unknown"]++;
  61         } else {
  62           (*out)[q->second]++;
  63         }
  64       }
  65     }
  66
  67   };
  68
  69   epoch_t epoch = 0;
  70   utime_t modified;
  71   map<std::string,Service> services;
  72
  73   void encode(bufferlist& bl, uint64_t features) const;
  74   void decode(bufferlist::iterator& p);
  75   void dump(Formatter *f) const;
  76   static void generate_test_instances(std::list<ServiceMap*>& ls);
  77
  78   Daemon* get_daemon(const std::string& service,
  79                      const std::string& daemon) {
  80     return &services[service].daemons[daemon];
  81   }
  82
  83   bool rm_daemon(const std::string& service,
  84                  const std::string& daemon) {
  85     auto p = services.find(service);
  86     if (p == services.end()) {
  87       return false;
  88     }
  89     auto q = p->second.daemons.find(daemon);
  90     if (q == p->second.daemons.end()) {
  91       return false;
  92     }
  93     p->second.daemons.erase(q);
  94     if (p->second.daemons.empty()) {
  95       services.erase(p);
  96     }
  97     return true;
  98   }
  99 };
 100 WRITE_CLASS_ENCODER_FEATURES(ServiceMap)
 101 WRITE_CLASS_ENCODER_FEATURES(ServiceMap::Service)
 102 WRITE_CLASS_ENCODER_FEATURES(ServiceMap::Daemon)