X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/5f2937099a707ddb360954f5c10fe79cfa906110..daeeee5daf8488c1abd98d5a361356be873e4ac0:/projects/stargazer/plugins/other/smux/sensors.h?ds=sidebyside diff --git a/projects/stargazer/plugins/other/smux/sensors.h b/projects/stargazer/plugins/other/smux/sensors.h index f2840636..55780987 100644 --- a/projects/stargazer/plugins/other/smux/sensors.h +++ b/projects/stargazer/plugins/other/smux/sensors.h @@ -5,6 +5,9 @@ #include "stg/users.h" #include "stg/tariffs.h" +#include "stg/admins.h" +#include "stg/services.h" +#include "stg/corporations.h" #include "stg/user_property.h" #include "stg/ObjectSyntax.h" @@ -29,13 +32,13 @@ class TotalUsersSensor : public Sensor { bool GetValue(ObjectSyntax_t * objectSyntax) const { - ValueToOS(users.GetUserNum(), objectSyntax); + ValueToOS(users.Count(), objectSyntax); return true; } #ifdef DEBUG std::string ToString() const - { std::string res; x2str(users.GetUserNum(), res); return res; } + { std::string res; x2str(users.Count(), res); return res; } #endif private: @@ -165,19 +168,79 @@ class TotalTariffsSensor : public Sensor { bool GetValue(ObjectSyntax_t * objectSyntax) const { - ValueToOS(tariffs.GetTariffsNum(), objectSyntax); + ValueToOS(tariffs.Count(), objectSyntax); return true; } #ifdef DEBUG std::string ToString() const - { std::string res; x2str(tariffs.GetTariffsNum(), res); return res; } + { std::string res; x2str(tariffs.Count(), res); return res; } #endif private: const TARIFFS & tariffs; }; +class TotalAdminsSensor : public Sensor { + public: + TotalAdminsSensor(const ADMINS & a) : admins(a) {} + virtual ~TotalAdminsSensor() {} + + bool GetValue(ObjectSyntax_t * objectSyntax) const + { + ValueToOS(admins.Count(), objectSyntax); + return true; + } + +#ifdef DEBUG + std::string ToString() const + { std::string res; x2str(admins.Count(), res); return res; } +#endif + + private: + const ADMINS & admins; +}; + +class TotalServicesSensor : public Sensor { + public: + TotalServicesSensor(const SERVICES & s) : services(s) {} + virtual ~TotalServicesSensor() {} + + bool GetValue(ObjectSyntax_t * objectSyntax) const + { + ValueToOS(services.Count(), objectSyntax); + return true; + } + +#ifdef DEBUG + std::string ToString() const + { std::string res; x2str(services.Count(), res); return res; } +#endif + + private: + const SERVICES & services; +}; + +class TotalCorporationsSensor : public Sensor { + public: + TotalCorporationsSensor(const CORPORATIONS & c) : corporations(c) {} + virtual ~TotalCorporationsSensor() {} + + bool GetValue(ObjectSyntax_t * objectSyntax) const + { + ValueToOS(corporations.Count(), objectSyntax); + return true; + } + +#ifdef DEBUG + std::string ToString() const + { std::string res; x2str(corporations.Count(), res); return res; } +#endif + + private: + const CORPORATIONS & corporations; +}; + template class ConstSensor : public Sensor { public: