The configure method takes test_only flag so we can avoid

[metaproxy-moved-to-github.git] / src / filter_load_balance.cpp

diff --git a/src/filter_load_balance.cpp b/src/filter_load_balance.cpp
index 3783ee1..c91791b 100644 (file)
--- a/src/filter_load_balance.cpp
+++ b/src/filter_load_balance.cpp
@@ -1,7 +1,22 @@
-/* $Id: filter_load_balance.cpp,v 1.4 2007-01-04 13:03:17 marc Exp $
-   Copyright (c) 2005-2006, Index Data.
+/* $Id: filter_load_balance.cpp,v 1.10 2008-02-20 15:07:52 adam Exp $
+   Copyright (c) 2005-2007, Index Data.
 
-   See the LICENSE file for details
+This file is part of Metaproxy.
+
+Metaproxy is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 2, or (at your option) any later
+version.
+
+Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with Metaproxy; see the file LICENSE.  If not, write to the
+Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
+02111-1307, USA.
  */
 
 #include "config.hpp"
@@ -11,12 +26,18 @@
 #include "filter_load_balance.hpp"
 #include "util.hpp"
 
+
 #include <boost/thread/mutex.hpp>
 #include <boost/date_time/posix_time/posix_time.hpp>
 
 #include <yaz/zgdu.h>
 
-#include <iostream>
+// remove max macro if already defined (defined later in <limits>)
+#ifdef max
+#undef max
+#endif
+
+//#include <iostream>
 #include <list>
 #include <map>
 #include <limits>
@@ -54,11 +75,11 @@ namespace metaproxy_1 {
                 unsigned int deads;
                 unsigned int cost() {
                     unsigned int c = sessions + packages + deads;
-                    std::cout << "stats  c:" << c 
-                              << " s:" << sessions 
-                              << " p:" << packages 
-                              << " d:" << deads 
-                              <<"\n";
+                    //std::cout << "stats  c:" << c 
+                    //          << " s:" << sessions 
+                    //          << " p:" << packages 
+                    //          << " d:" << deads 
+                    //          <<"\n";
                     return c;
                 }
             };
@@ -81,7 +102,7 @@ yf::LoadBalance::~LoadBalance()
 {  // must have a destructor because of boost::scoped_ptr
 }
 
-void yf::LoadBalance::configure(const xmlNode *xmlnode)
+void yf::LoadBalance::configure(const xmlNode *xmlnode, bool test_only)
 {
     m_p->configure(xmlnode);
 }
@@ -168,7 +189,7 @@ void yf::LoadBalance::Impl::process(mp::Package &package)
                 
                 // copying new target into init package
                 mp::util::set_vhost_otherinfo(&(org_init->otherInfo), 
-                                              odr_en, target); 
+                                              odr_en, target, 1);
                 package.request() = gdu_req;
         }
             
@@ -247,8 +268,8 @@ void yf::LoadBalance::Impl::add_dead(unsigned long session_id){
         if (itarg != m_target_stat.end()
             && itarg->second.deads < std::numeric_limits<unsigned int>::max()){
             itarg->second.deads += 1;
-            std::cout << "add_dead " << session_id << " " << target 
-                      << " d:" << itarg->second.deads << "\n";
+            // std:.cout << "add_dead " << session_id << " " << target 
+            //          << " d:" << itarg->second.deads << "\n";
         }
     }
 };
@@ -268,8 +289,8 @@ void yf::LoadBalance::Impl::add_package(unsigned long session_id){
             && itarg->second.packages 
                < std::numeric_limits<unsigned int>::max()){
             itarg->second.packages += 1;
-            std::cout << "add_package " << session_id << " " << target 
-                      << " p:" << itarg->second.packages << "\n";
+            // std:.cout << "add_package " << session_id << " " << target 
+            //          << " p:" << itarg->second.packages << "\n";
         }
     }
 };
@@ -283,8 +304,8 @@ void yf::LoadBalance::Impl::remove_package(unsigned long session_id){
         if (itarg != m_target_stat.end()
             && itarg->second.packages > 0){
             itarg->second.packages -= 1;
-            std::cout << "remove_package " << session_id << " " << target 
-                      << " p:" << itarg->second.packages << "\n";
+            // std:.cout << "remove_package " << session_id << " " << target 
+            //          << " p:" << itarg->second.packages << "\n";
         }
     }
 };
@@ -308,14 +329,14 @@ void yf::LoadBalance::Impl::add_session(unsigned long session_id,
         stat.packages = 0;  // no idea why the defaut constructor TargetStat()
         stat.deads = 0;     // is not initializig this correctly to zero ??
        m_target_stat.insert(std::make_pair(target, stat));
-        std::cout << "add_session " << session_id << " " << target 
-                  << " s:1\n";
+        // std:.cout << "add_session " << session_id << " " << target 
+        //          << " s:1\n";
     } 
     else if (itarg->second.sessions < std::numeric_limits<unsigned int>::max())
     {
         itarg->second.sessions += 1;
-        std::cout << "add_session " << session_id << " " << target 
-                  << " s:" << itarg->second.sessions << "\n";
+        // std:.cout << "add_session " << session_id << " " << target 
+        //          << " s:" << itarg->second.sessions << "\n";
     }
 };
 
@@ -343,8 +364,8 @@ void yf::LoadBalance::Impl::remove_session(unsigned long session_id){
     if (itarg->second.sessions > 0)
         itarg->second.sessions -= 1;
 
-    std::cout << "remove_session " << session_id << " " << target 
-              << " s:" << itarg->second.sessions << "\n";
+    // std:.cout << "remove_session " << session_id << " " << target 
+    //          << " s:" << itarg->second.sessions << "\n";
     
     // clearing empty sessions and targets
     if (itarg->second.sessions == 0 && itarg->second.deads == 0 ){