]> git.stg.codes - stg.git/blobdiff - projects/stargazer/plugins/other/rscript/ur_functor.h
Ticket 52. Definition of the tariff pointer added to the
[stg.git] / projects / stargazer / plugins / other / rscript / ur_functor.h
index c5752e870a7811c60629391b86499d86443b8677..9ecf2bc5887f30fd3f899eedffb717a693613155 100644 (file)
 #ifndef __UR_FUNCTOR_H__
 #define __UR_FUNCTOR_H__
 
-#include <functional>
-#include <algorithm>
-#include <utility>
+#include "rscript.h"
 
 #include "stg/os_int.h"
 #include "stg/common.h"
 
-#include "rscript.h"
+#include <functional>
+#include <algorithm>
+#include <utility>
 
 namespace RS
 {
@@ -36,12 +36,12 @@ namespace RS
 class UpdateRouter : public std::unary_function<std::pair<const uint32_t, RS::USER>, void>
 {
 public:
-    UpdateRouter(REMOTE_SCRIPT & t)
-        : obj(t) {};
+    explicit UpdateRouter(REMOTE_SCRIPT & t)
+        : obj(t) {}
 
     void operator() (std::pair<const uint32_t, USER> & val)
         {
-        std::vector<uint32_t> newRouters = obj.IP2Routers(val.first);
+        std::vector<uint32_t> newRouters = obj.IP2Routers(val.second.ip);
         std::vector<uint32_t>::const_iterator oldIt(val.second.routers.begin());
         std::vector<uint32_t>::const_iterator newIt(newRouters.begin());
         val.second.shortPacketsCount = 0;
@@ -52,29 +52,28 @@ public:
                 {
                 if (newIt != newRouters.end())
                     {
-                    obj.SendDirect(val.first, val.second, *newIt); // Connect on new router
+                    obj.SendDirect(val.second, *newIt); // Connect on new router
                     ++newIt;
                     }
                 }
             else if (newIt == newRouters.end())
                 {
-                obj.SendDirect(val.first, val.second, *oldIt, true); // Disconnect on old router
+                obj.SendDirect(val.second, *oldIt, true); // Disconnect on old router
                 ++oldIt;
-                } 
+                }
             else if (*oldIt < *newIt)
                 {
-                obj.SendDirect(val.first, val.second, *oldIt, true); // Disconnect on old router
+                obj.SendDirect(val.second, *oldIt, true); // Disconnect on old router
                 ++oldIt;
                 }
             else if (*oldIt > *newIt)
                 {
-                obj.SendDirect(val.first, val.second, *newIt); // Connect on new router
+                obj.SendDirect(val.second, *newIt); // Connect on new router
                 ++newIt;
                 }
             else
                 {
-                if (oldIt != val.second.routers.end())
-                    ++oldIt;
+                ++oldIt;
                 if (newIt != newRouters.end())
                     ++newIt;
                 }