Initial commit

[ta/config-manager.git] / cmframework / src / cmframework / server / cmactivator.py

diff --git a/cmframework/src/cmframework/server/cmactivator.py b/cmframework/src/cmframework/server/cmactivator.py
new file mode 100644 (file)
index 0000000..48c7a6c
--- /dev/null
+++ b/cmframework/src/cmframework/server/cmactivator.py
@@ -0,0 +1,56 @@
+# Copyright 2019 Nokia
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from Queue import Queue
+
+from cmframework.server import cmeventletrwlock
+from cmframework.server import cmactivatorworker
+
+
+class CMActivator(object):
+    def __init__(self, worker_count):
+        self.works = Queue()
+        self.node_works = Queue()
+        self.handlers = []
+        self.workers = []
+        self.worker_count = worker_count
+        self.lock = cmeventletrwlock.CMEventletRWLock()
+
+    def add_handler(self, handler):
+        self.handlers.append(handler)
+
+    def get_parallel_work(self):
+        return self.node_works.get()
+
+    def get_work(self):
+        return self.works.get()
+
+    def add_work(self, work):
+        work.release()
+        if not work.get_target():
+            self.works.put(work)
+        else:
+            self.node_works.put(work)
+
+    def get_handlers(self):
+        return self.handlers
+
+    def start(self):
+        worker = cmactivatorworker.CMActivatorWorker(self, 0, self.lock)
+        worker.start()
+        self.workers.append(worker)
+
+        for i in range(1, self.worker_count+1):
+            worker = cmactivatorworker.CMActivatorWorker(self, i, self.lock, parallel=True)
+            worker.start()
+            self.workers.append(worker)