X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/82ede2a92a721d1c8f0e3fd109a2594f2096306e..355911106bc27f5504eb904a064367ef1b5f0aba:/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 16331b15..f751c82e 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,6 +18,7 @@ class Sensor { public: + virtual ~Sensor() {} virtual bool GetValue(ObjectSyntax_t * objectSyntax) const = 0; #ifdef DEBUG virtual std::string ToString() const = 0; @@ -29,13 +34,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,13 +170,13 @@ 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: @@ -185,13 +190,13 @@ class TotalAdminsSensor : public Sensor { bool GetValue(ObjectSyntax_t * objectSyntax) const { - ValueToOS(admins.GetAdminsNum(), objectSyntax); + ValueToOS(admins.Count(), objectSyntax); return true; } #ifdef DEBUG std::string ToString() const - { std::string res; x2str(admins.GetAdminsNum(), res); return res; } + { std::string res; x2str(admins.Count(), res); return res; } #endif private: @@ -205,13 +210,13 @@ class TotalServicesSensor : public Sensor { bool GetValue(ObjectSyntax_t * objectSyntax) const { - ValueToOS(services.GetServicesNum(), objectSyntax); + ValueToOS(services.Count(), objectSyntax); return true; } #ifdef DEBUG std::string ToString() const - { std::string res; x2str(services.GetServicesNum(), res); return res; } + { std::string res; x2str(services.Count(), res); return res; } #endif private: @@ -225,19 +230,39 @@ class TotalCorporationsSensor : public Sensor { bool GetValue(ObjectSyntax_t * objectSyntax) const { - ValueToOS(corporations.GetCorporationsNum(), objectSyntax); + ValueToOS(corporations.Count(), objectSyntax); return true; } #ifdef DEBUG std::string ToString() const - { std::string res; x2str(services.GetCorporationsNum(), res); return res; } + { 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: @@ -256,11 +281,13 @@ class ConstSensor : public Sensor { T value; }; +#ifdef DEBUG template <> inline std::string ConstSensor::ToString() const { return value; } +#endif #endif