]>
Commit | Line | Data |
---|---|---|
224ce89b WB |
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; | |
c07f9fc5 | 50 | ss << daemons.size() << (daemons.size() > 1 ? " daemons" : " daemon") |
224ce89b WB |
51 | << " active"; |
52 | return ss.str(); | |
53 | } | |
c07f9fc5 FG |
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 | ||
224ce89b WB |
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) |