Throttle the number of connections

1 files changed, 14 insertions, 1 deletions

diff --git a/wallace/__init__.py b/wallace/__init__.py
index e018553..2db963e 100644
--- a/wallace/__init__.py
+++ b/wallace/__init__.py
@@ -63,6 +63,9 @@ def worker_process(*args, **kw):
 
 class WallaceDaemon(object):
     def __init__(self):
+        self.current_connections = 0
+        self.max_connections = 24
+
         daemon_group = conf.add_cli_parser_option_group(_("Daemon Options"))
 
         daemon_group.add_option(
@@ -215,21 +218,29 @@ class WallaceDaemon(object):
                     if stage.lower() == "defer":
                         continue
 
+                    self.current_connections += 1
                     self.pool.apply_async(pickup_message, (filepath, (self.modules), {'module': module, 'stage': stage}))
+                    self.current_connections -= 1
 
                     continue
 
+                self.current_connections += 1
                 self.pool.apply_async(pickup_message, (filepath, (self.modules)))
+                self.current_connections -= 1
 
         try:
             while 1:
+                while self.current_connections >= self.max_connections:
+                    time.sleep(0.5)
+
                 pair = s.accept()
                 log.info(_("Accepted connection"))
                 if not pair == None:
+                    self.current_connections += 1
                     connection, address = pair
-                    #print "Accepted connection from %r" % (address)
                     channel = SMTPChannel(self, connection, address)
                     asyncore.loop()
+
         except Exception, errmsg:
             traceback.print_exc()
             s.shutdown(1)
@@ -258,6 +269,8 @@ class WallaceDaemon(object):
 
         self.pool.apply_async(pickup_message, (filename, (self.modules)))
 
+        self.current_connections -= 1
+
         return
 
     def reload_config(self, *args, **kw):