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 ? "daemonss" : "daemon")
  51          << " active";
  52       return ss.str();
  53     }
  54   };
  55
  56   epoch_t epoch = 0;
  57   utime_t modified;
  58   map<std::string,Service> services;
  59
  60   void encode(bufferlist& bl, uint64_t features) const;
  61   void decode(bufferlist::iterator& p);
  62   void dump(Formatter *f) const;
  63   static void generate_test_instances(std::list<ServiceMap*>& ls);
  64
  65   Daemon* get_daemon(const std::string& service,
  66                      const std::string& daemon) {
  67     return &services[service].daemons[daemon];
  68   }
  69
  70   bool rm_daemon(const std::string& service,
  71                  const std::string& daemon) {
  72     auto p = services.find(service);
  73     if (p == services.end()) {
  74       return false;
  75     }
  76     auto q = p->second.daemons.find(daemon);
  77     if (q == p->second.daemons.end()) {
  78       return false;
  79     }
  80     p->second.daemons.erase(q);
  81     if (p->second.daemons.empty()) {
  82       services.erase(p);
  83     }
  84     return true;
  85   }
  86 };
  87 WRITE_CLASS_ENCODER_FEATURES(ServiceMap)
  88 WRITE_CLASS_ENCODER_FEATURES(ServiceMap::Service)
  89 WRITE_CLASS_ENCODER_FEATURES(ServiceMap::Daemon)