]> git.stg.codes - stg.git/blobdiff - projects/stargazer/plugins/other/smux/tables.cpp
Merge branch 'stg-2.409-radius'
[stg.git] / projects / stargazer / plugins / other / smux / tables.cpp
index 023b90b2b1df4f6f5db8d5b818368ca03d30b072..c3e3ef54c48d870fee774d16ac11e3420b8b67e2 100644 (file)
@@ -1,23 +1,37 @@
+#include <cassert>
 #include <utility>
+#include <iterator>
+#include <algorithm>
 
 #include "stg/user_property.h"
+#include "stg/tariffs.h"
+#include "stg/users.h"
 
 #include "tables.h"
 
+std::pair<std::string, size_t> TD2Info(const TARIFF_DATA & td);
+
 void TariffUsersTable::UpdateSensors(Sensors & sensors) const
 {
 std::map<std::string, size_t> data;
 
+std::list<TARIFF_DATA> tdl;
+tariffs.GetTariffsData(&tdl);
+std::transform(tdl.begin(),
+               tdl.end(),
+               std::inserter(data, data.begin()),
+               TD2Info);
+
 int handle = users.OpenSearch();
-if (!handle)
-    return;
+assert(handle && "USERS::OpenSearch is always correct");
 
 USER_PTR user;
 while (!users.SearchNext(handle, &user))
     {
+    if (user->GetDeleted())
+        continue;
     std::string tariffName(user->GetProperty().tariffName.ConstData());
-    std::map<std::string, size_t>::iterator it;
-    it = data.lower_bound(tariffName);
+    std::map<std::string, size_t>::iterator it(data.lower_bound(tariffName));
     if (it == data.end() ||
         it->first != tariffName)
         {
@@ -37,9 +51,14 @@ OID prefixOid(prefix);
 std::map<std::string, size_t>::const_iterator it(data.begin());
 while (it != data.end())
     {
-    sensors[prefixOid.copyWithSuffix(2, idx)] = new ConstSensor<std::string>(it->first);
-    sensors[prefixOid.copyWithSuffix(3, idx)] = new ConstSensor<int>(it->second);
+    sensors[prefixOid.copyWithSuffix(2, static_cast<unsigned int>(idx))] = new ConstSensor<std::string>(it->first);
+    sensors[prefixOid.copyWithSuffix(3, static_cast<unsigned int>(idx))] = new ConstSensor<unsigned long>(it->second);
     ++idx;
     ++it;
     }
 }
+
+std::pair<std::string, size_t> TD2Info(const TARIFF_DATA & td)
+{
+return std::make_pair(td.tariffConf.name, 0);
+}