Some refactoring.

[stg.git] / projects / stargazer / plugins / other / smux / sensors.cpp

diff --git a/projects/stargazer/plugins/other/smux/sensors.cpp b/projects/stargazer/plugins/other/smux/sensors.cpp
index 6018566c72d358efeac54bda5246d797bb6eb06b..f52a413b7aeff6543e3a4ccf32da5e67012cc0cf 100644 (file)
--- a/projects/stargazer/plugins/other/smux/sensors.cpp
+++ b/projects/stargazer/plugins/other/smux/sensors.cpp
@@ -1,3 +1,5 @@
+#include <cassert>
+

 #include "stg/INTEGER.h"
 
 #include "stg/user.h"
 #include "stg/INTEGER.h"
 
 #include "stg/user.h"


@@ -7,8 +9,7 @@
 bool UsersSensor::GetValue(ObjectSyntax_t * objectSyntax) const
 {
 int handle = users.OpenSearch();
 bool UsersSensor::GetValue(ObjectSyntax_t * objectSyntax) const
 {
 int handle = users.OpenSearch();

-if (!handle)
-    return false;
+assert(handle && "USERS::OpenSearch is always correct");

 
 USER_PTR user;
 size_t count = 0;
 
 USER_PTR user;
 size_t count = 0;


@@ -28,8 +29,7 @@ return true;
 std::string UsersSensor::ToString() const
 {
 int handle = users.OpenSearch();
 std::string UsersSensor::ToString() const
 {
 int handle = users.OpenSearch();

-if (!handle)
-    return "";
+assert(handle && "USERS::OpenSearch is always correct");

 
 USER_PTR user;
 size_t count = 0;
 
 USER_PTR user;
 size_t count = 0;


@@ -46,3 +46,16 @@ x2str(count, res);
 return res;
 }
 #endif
 return res;
 }
 #endif

+
+bool ActiveUsersSensor::UserPredicate(USER_PTR userPtr) const
+{
+if (!userPtr->GetConnected())
+    return false;
+for (size_t i = 0; i < DIR_NUM; ++i)
+    {
+    if (userPtr->GetSessionUpload()[i] > 0 ||
+        userPtr->GetSessionDownload()[i] > 0)
+        return true;
+    }
+return false;
+}