Merge pull request #9 from RHDZMOTA/DEV

2022-09-14 RELEASE-PRD (1)

Author: RHDZMOTA

src/rhdzmota/VERSION

@@ -1 +1 @@
-2.0.0
+2.1.0

src/rhdzmota/celery.py

@@ -1,12 +1,34 @@
 from celery import Celery
 
+from .heartbeat.daemon import Daemon
 from .settings import (
-    RHDZMOTA_CELERY_BROKER_HOST
+    RHDZMOTA_CELERY_BROKER_HOST,
+    RHDZMOTA_DAEMON_NAME
 )
 
 
 # Celery application instance
 app = Celery(
     "tasks",
     broker=f"redis://{RHDZMOTA_CELERY_BROKER_HOST}",
+    backend=f"redis://{RHDZMOTA_CELERY_BROKER_HOST}",
 )
+
+app.conf.update(
+    task_routes={
+        "rhdzmota.celery.daemon_heartbeat_publisher": {
+            "queue": "queue_heartbeat"
+        }
+    }
+)
+
+
+@app.task
+def daemon_heartbeat_publisher(**kwargs):
+    print(kwargs)
+
+
+# Daemon instance
+if RHDZMOTA_DAEMON_NAME:
+    daemon = Daemon(name=RHDZMOTA_DAEMON_NAME, publisher=daemon_heartbeat_publisher, interval=3)
+    daemon.broadcast()

src/rhdzmota/celery_worker_hello.py

@@ -0,0 +1,7 @@
+
+from ..celery import app
+
+
+@app.task
+def worker(name: str) -> str:
+    return f"Hello, {name or 'world'}!"

src/rhdzmota/celery_workers/__init__.py

@@ -16,7 +16,7 @@ def hello(self, world: Optional[str] = None, sleep: int = 1, delegate: bool = Fa
         world = world or "world"
         time.sleep(sleep)
         try:
-            from .celery_worker_hello import worker
+            from .celery_workers.hello import worker
 
             if delegate:
                 return worker.delay(name=world)

src/rhdzmota/celery_workers/hello.py

@@ -130,6 +130,11 @@ def singleton(cls, overwrite_config_dictionary: Optional[Dict] = None) -> 'Logge
     default="127.0.0.1"
 )
 
+# Daemon
+RHDZMOTA_DAEMON_NAME = get_environ_variable(
+    name="RHDZMOTA_DAEMON_NAME",
+)
+
 # JWT
 
 JWT_ENCRYPTION_ALGORITHM: str = get_environ_variable(

src/rhdzmota/cli.py

@@ -0,0 +1 @@
+celery -A rhdzmota.celery worker --loglevel=INFO --concurrency=1 -O fair -P prefork -Q queue_heartbeat

src/rhdzmota/settings.py

@@ -1 +1 @@
-celery -A rhdzmota.celery_worker_hello worker --loglevel=INFO --concurrency=2 -O fair -P prefork
+celery -A rhdzmota.celery_workers.hello worker --loglevel=INFO --concurrency=2 -O fair -P prefork