Fixed compilation with new TRAFF_TYPE.

[stg.git] / projects / stargazer / plugins / other / rscript / ur_functor.h

diff --git a/projects/stargazer/plugins/other/rscript/ur_functor.h b/projects/stargazer/plugins/other/rscript/ur_functor.h
index b3a5cdaca3889ca338da3985ff46905b58829b76..95382400f6403e945b3e05768ac40d718b3aece1 100644 (file)
--- a/projects/stargazer/plugins/other/rscript/ur_functor.h
+++ b/projects/stargazer/plugins/other/rscript/ur_functor.h
@@ -18,12 +18,6 @@
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 

-/*
- $Revision: 1.3 $
- $Date: 2010/03/04 12:07:03 $
- $Author: faust $
-*/
-

 #ifndef __UR_FUNCTOR_H__
 #define __UR_FUNCTOR_H__
 
 #ifndef __UR_FUNCTOR_H__
 #define __UR_FUNCTOR_H__
 


@@ -36,15 +30,18 @@
 
 #include "rscript.h"
 
 
 #include "rscript.h"
 

-class UpdateRouter : public std::unary_function<std::pair<const uint32_t, RS_USER>, void>
+namespace RS
+{
+
+class UpdateRouter : public std::unary_function<std::pair<const uint32_t, RS::USER>, void>

 {
 public:
     UpdateRouter(REMOTE_SCRIPT & t)
 {
 public:
     UpdateRouter(REMOTE_SCRIPT & t)

-        : obj(t) {};
+        : obj(t) {}

 
 

-    void operator() (std::pair<const uint32_t, RS_USER> & val)
+    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;
         std::vector<uint32_t>::const_iterator oldIt(val.second.routers.begin());
         std::vector<uint32_t>::const_iterator newIt(newRouters.begin());
         val.second.shortPacketsCount = 0;


@@ -55,26 +52,23 @@ public:
                 {
                 if (newIt != newRouters.end())
                     {
                 {
                 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())
                 {
                     ++newIt;
                     }
                 }
             else if (newIt == newRouters.end())
                 {

-                //if (oldIt != newRouters.end())
-                    //{ // Already checked it
-                    obj.SendDirect(val.first, val.second, *oldIt, true); // Disconnect on old router
-                    ++oldIt;
-                    //}
+                obj.SendDirect(val.second, *oldIt, true); // Disconnect on old router
+                ++oldIt;

                 } 
             else if (*oldIt < *newIt)
                 {
                 } 
             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)
                 {
                 ++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
                 ++newIt;
                 }
             else


@@ -86,16 +80,11 @@ public:
                 }
             }
         val.second.routers = newRouters;
                 }
             }
         val.second.routers = newRouters;

-        /*if (val.second.souters != newRouters)
-            {
-            obj.Send(val.first, val.second, true); // Disconnect on old router
-            val.second.routerIP = obj.IP2Router(val.first); // Change router
-            val.second.shortPacketsCount = 0; // Reset packets count (to prevent alive send)
-            obj.Send(val.first, val.second); // Connect on new router
-            }*/

         }
 private:
     REMOTE_SCRIPT & obj;
 };
 
         }
 private:
     REMOTE_SCRIPT & obj;
 };
 

+} // namespace RS
+

 #endif
 #endif