consumder_thread.wait()

ekarlso · ekarlso · commit 2f53dc2f673b · 2013-07-04T14:46:17.000+02:00
Change-Id: Ia4275c9e911e36c1ea3fd409c43787f4ec1e6ab5
diff --git a/billingstack/biller/service.py b/billingstack/biller/service.py
@@ -46,6 +46,10 @@ def start(self):
         self.storage_conn = get_connection('biller')
         super(Service, self).start()
 
+    def wait(self):
+        super(Service, self).wait()
+        self.conn.consumer_thread.wait()
+
     def create_invoice_state(self, ctxt, values):
         return self.storage_conn.create_invoice_state(ctxt, values)
 
diff --git a/billingstack/central/service.py b/billingstack/central/service.py
@@ -44,6 +44,10 @@ def start(self):
         self.storage_conn = get_connection('central')
         super(Service, self).start()
 
+    def wait(self):
+        super(Service, self).wait()
+        self.conn.consumer_thread.wait()
+
     def __getattr__(self, name):
         """
         Proxy onto the storage api if there is no local method present..
diff --git a/billingstack/collector/service.py b/billingstack/collector/service.py
@@ -48,6 +48,10 @@ def __init__(self, *args, **kwargs):
         # Get a storage connection
         self.central_api = CentralAPI()
 
+    def wait(self):
+        super(Service, self).wait()
+        self.conn.consumer_thread.wait()
+
     def get_pg_provider(self, ctxt, pg_info):
         """
         Work out a PGC config either from pg_info or via ctxt fetching it
diff --git a/billingstack/rater/service.py b/billingstack/rater/service.py
@@ -50,6 +50,10 @@ def start(self):
         self.storage_conn = get_connection('rater')
         super(Service, self).start()
 
+    def wait(self):
+        super(Service, self).wait()
+        self.conn.consumer_thread.wait()
+
     def create_usage(self, ctxt, values):
         return self.storage_conn.create_usage(ctxt, values)
 
