X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/fdee6fdb88e79929c952fea956fa57e5780804cf..519fd4f22bc83ba85a8377a7ada53fac4e2e2b39:/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 dbfb807b..35de996c 100644 --- a/projects/stargazer/plugins/other/smux/sensors.h +++ b/projects/stargazer/plugins/other/smux/sensors.h @@ -5,46 +5,57 @@ #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 "asn1/ObjectSyntax.h" +#include "stg/ObjectSyntax.h" #include "value2os.h" #include "types.h" 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) {} + explicit TotalUsersSensor(const USERS & u) : users(u) {} virtual ~TotalUsersSensor() {} 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; }; class UsersSensor : public Sensor { public: - UsersSensor(USERS & u) : users(u) {} - virtual ~UsersSensor() {}; + explicit UsersSensor(USERS & u) : users(u) {} + virtual ~UsersSensor() {} bool GetValue(ObjectSyntax_t * objectSyntax) const; +#ifdef DEBUG + std::string ToString() const; +#endif private: USERS & users; @@ -54,7 +65,7 @@ class UsersSensor : public Sensor { class ConnectedUsersSensor : public UsersSensor { public: - ConnectedUsersSensor(USERS & u) : UsersSensor(u) {} + explicit ConnectedUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~ConnectedUsersSensor() {} private: @@ -64,7 +75,7 @@ class ConnectedUsersSensor : public UsersSensor { class AuthorizedUsersSensor : public UsersSensor { public: - AuthorizedUsersSensor(USERS & u) : UsersSensor(u) {} + explicit AuthorizedUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~AuthorizedUsersSensor() {} private: @@ -74,7 +85,7 @@ class AuthorizedUsersSensor : public UsersSensor { class AlwaysOnlineUsersSensor : public UsersSensor { public: - AlwaysOnlineUsersSensor(USERS & u) : UsersSensor(u) {} + explicit AlwaysOnlineUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~AlwaysOnlineUsersSensor() {} private: @@ -84,7 +95,7 @@ class AlwaysOnlineUsersSensor : public UsersSensor { class NoCashUsersSensor : public UsersSensor { public: - NoCashUsersSensor(USERS & u) : UsersSensor(u) {} + explicit NoCashUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~NoCashUsersSensor() {} private: @@ -94,7 +105,7 @@ class NoCashUsersSensor : public UsersSensor { class DisabledDetailStatsUsersSensor : public UsersSensor { public: - DisabledDetailStatsUsersSensor(USERS & u) : UsersSensor(u) {} + explicit DisabledDetailStatsUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~DisabledDetailStatsUsersSensor() {} private: @@ -104,7 +115,7 @@ class DisabledDetailStatsUsersSensor : public UsersSensor { class DisabledUsersSensor : public UsersSensor { public: - DisabledUsersSensor(USERS & u) : UsersSensor(u) {} + explicit DisabledUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~DisabledUsersSensor() {} private: @@ -114,7 +125,7 @@ class DisabledUsersSensor : public UsersSensor { class PassiveUsersSensor : public UsersSensor { public: - PassiveUsersSensor(USERS & u) : UsersSensor(u) {} + explicit PassiveUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~PassiveUsersSensor() {} private: @@ -124,7 +135,7 @@ class PassiveUsersSensor : public UsersSensor { class CreditUsersSensor : public UsersSensor { public: - CreditUsersSensor(USERS & u) : UsersSensor(u) {} + explicit CreditUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~CreditUsersSensor() {} private: @@ -134,7 +145,7 @@ class CreditUsersSensor : public UsersSensor { class FreeMbUsersSensor : public UsersSensor { public: - FreeMbUsersSensor(USERS & u) : UsersSensor(u) {} + explicit FreeMbUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~FreeMbUsersSensor() {} private: @@ -144,40 +155,148 @@ class FreeMbUsersSensor : public UsersSensor { class TariffChangeUsersSensor : public UsersSensor { public: - TariffChangeUsersSensor(USERS & u) : UsersSensor(u) {} + explicit TariffChangeUsersSensor(USERS & u) : UsersSensor(u) {} virtual ~TariffChangeUsersSensor() {} private: bool UserPredicate(USER_PTR userPtr) const - { return userPtr->GetProperty().nextTariff.ConstData().empty(); } + { return !userPtr->GetProperty().nextTariff.ConstData().empty(); } +}; + +class ActiveUsersSensor : public UsersSensor { + public: + explicit ActiveUsersSensor(USERS & u) : UsersSensor(u) {} + virtual ~ActiveUsersSensor() {} + + private: + bool UserPredicate(USER_PTR userPtr) const; }; class TotalTariffsSensor : public Sensor { public: - TotalTariffsSensor(const TARIFFS & t) : tariffs(t) {} + explicit TotalTariffsSensor(const TARIFFS & t) : tariffs(t) {} virtual ~TotalTariffsSensor() {} 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: + explicit 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: + explicit 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: + explicit 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: + explicit 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: - ConstSensor(const T & v) : value(v) {} + explicit ConstSensor(const T & v) : value(v) {} virtual ~ConstSensor() {} 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