X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/76e975a593f0194edcbfbfc952eb391b2b8605cf..05072df6840a3e6622d235cd48e0ced723e05b26:/projects/stargazer/plugins/other/smux/sensors.h diff --git a/projects/stargazer/plugins/other/smux/sensors.h b/projects/stargazer/plugins/other/smux/sensors.h index 37bd2020..3787611f 100644 --- a/projects/stargazer/plugins/other/smux/sensors.h +++ b/projects/stargazer/plugins/other/smux/sensors.h @@ -5,6 +5,10 @@ #include "stg/users.h" #include "stg/tariffs.h" +#include "stg/admins.h" +#include "stg/services.h" +#include "stg/corporations.h" +#include "stg/traffcounter.h" #include "stg/user_property.h" #include "stg/ObjectSyntax.h" @@ -14,16 +18,15 @@ class Sensor { public: + virtual ~Sensor() {} virtual bool GetValue(ObjectSyntax_t * objectSyntax) const = 0; +#ifdef DEBUG + virtual std::string ToString() const = 0; +#endif }; typedef std::map Sensors; -class TableSensor { - public: - virtual bool appendTable(Sensors & sensors); -}; - class TotalUsersSensor : public Sensor { public: TotalUsersSensor(const USERS & u) : users(u) {} @@ -31,10 +34,15 @@ 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.Count(), res); return res; } +#endif + private: const USERS & users; }; @@ -42,9 +50,12 @@ class TotalUsersSensor : public Sensor { class UsersSensor : public Sensor { public: UsersSensor(USERS & u) : users(u) {} - virtual ~UsersSensor() {}; + virtual ~UsersSensor() {} bool GetValue(ObjectSyntax_t * objectSyntax) const; +#ifdef DEBUG + std::string ToString() const; +#endif private: USERS & users; @@ -149,7 +160,16 @@ class TariffChangeUsersSensor : public UsersSensor { private: bool UserPredicate(USER_PTR userPtr) const - { return userPtr->GetProperty().nextTariff.ConstData().empty(); } + { return !userPtr->GetProperty().nextTariff.ConstData().empty(); } +}; + +class ActiveUsersSensor : public UsersSensor { + public: + ActiveUsersSensor(USERS & u) : UsersSensor(u) {} + virtual ~ActiveUsersSensor() {} + + private: + bool UserPredicate(USER_PTR userPtr) const; }; class TotalTariffsSensor : public Sensor { @@ -159,14 +179,99 @@ 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.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; +}; + +class TotalRulesSensor : public Sensor { + public: + TotalRulesSensor(const TRAFFCOUNTER & t) : traffcounter(t) {} + virtual ~TotalRulesSensor() {} + + bool GetValue(ObjectSyntax_t * objectSyntax) const + { + ValueToOS(traffcounter.RulesCount(), objectSyntax); + return true; + } + +#ifdef DEBUG + std::string ToString() const + { std::string res; x2str(traffcounter.RulesCount(), res); return res; } +#endif + + private: + const TRAFFCOUNTER & traffcounter; +}; + template class ConstSensor : public Sensor { public: @@ -176,8 +281,22 @@ class ConstSensor : public Sensor { bool GetValue(ObjectSyntax * objectSyntax) const { return ValueToOS(value, objectSyntax); } +#ifdef DEBUG + std::string ToString() const + { std::string res; x2str(value, res); return res; } +#endif + private: T value; }; +#ifdef DEBUG +template <> +inline +std::string ConstSensor::ToString() const +{ +return value; +} +#endif + #endif