quapp-common 0.0.13.dev6__tar.gz → 0.0.13.dev8__tar.gz

{quapp_common-0.0.13.dev6/quapp_common.egg-info → quapp_common-0.0.13.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quapp-common
-Version: 0.0.13.dev6
+Version: 0.0.13.dev8
 Summary: Quapp common library supporting Quapp Platform for Quantum Computing
 Author-email: "CITYNOW Co. Ltd. " <corp@citynow.vn>
 License: The MIT License (MIT)

{quapp_common-0.0.13.dev6 → quapp_common-0.0.13.dev8}/pyproject.toml

@@ -1,41 +1,41 @@
-[build-system]
-requires = ["setuptools>=61.0.0", "wheel"]
-build-backend = "setuptools.build_meta"
-
-[project]
-name = "quapp-common"
-version = "0.0.13.dev6"
-description = "Quapp common library supporting Quapp Platform for Quantum Computing"
-readme = "README.md"
-authors = [{ name = "CITYNOW Co. Ltd. ", email = "corp@citynow.vn" }]
-license = { file = "LICENSE" }
-classifiers = [
-    "License :: OSI Approved :: MIT License",
-    "Programming Language :: Python",
-    "Programming Language :: Python :: 3",
-]
-keywords = ["quapp", "quapp-common", "quantum"]
-dependencies = [
-    "loguru",
-    "requests",
-    "numpy",
-    "matplotlib",
-    "pylatexenc",
-    "starlette",
-    "qibo",
-    "dimod",
-    "qulacs",
-    "qiskit",
-    "pytket",
-    "pennylane"
-]
-requires-python = ">=3.7"
-
-[project.optional-dependencies]
-dev = ["black", "bumpver", "isort", "pip-tools", "pytest"]
-
-[project.urls]
-Homepage = "https://citynow.asia/"
-
-[tool.pylint.main]
-source-roots = ["."]
+[build-system]
+requires = ["setuptools>=61.0.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "quapp-common"
+version = "0.0.13.dev8"
+description = "Quapp common library supporting Quapp Platform for Quantum Computing"
+readme = "README.md"
+authors = [{ name = "CITYNOW Co. Ltd. ", email = "corp@citynow.vn" }]
+license = { file = "LICENSE" }
+classifiers = [
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+]
+keywords = ["quapp", "quapp-common", "quantum"]
+dependencies = [
+    "loguru",
+    "requests",
+    "numpy",
+    "matplotlib",
+    "pylatexenc",
+    "starlette",
+    "qibo",
+    "dimod",
+    "qulacs",
+    "qiskit",
+    "pytket",
+    "pennylane"
+]
+requires-python = ">=3.7"
+
+[project.optional-dependencies]
+dev = ["black", "bumpver", "isort", "pip-tools", "pytest"]
+
+[project.urls]
+Homepage = "https://citynow.asia/"
+
+[tool.pylint.main]
+source-roots = ["."]

quapp_common-0.0.13.dev8/quapp_common/component/backend/job_heartbeat.py

@@ -0,0 +1,83 @@
+#  Quapp Platform Project
+#  job_heartbeat.py
+#  Copyright © CITYNOW Co. Ltd. All rights reserved.
+
+import os
+import threading
+
+import requests
+
+from ...config.logging_config import logger
+
+DEFAULT_INTERVAL_SEC = 30.0
+REQUEST_TIMEOUT_SEC = 5.0
+TERMINAL_STATUSES = ('DONE', 'FAILED')
+
+_lock = threading.Lock()
+_thread = None
+_stop_event = threading.Event()
+
+
+def heartbeat_interval_sec() -> float:
+    raw = os.getenv('HEARTBEAT_INTERVAL_SEC', '')
+    try:
+        value = float(raw)
+        if value > 0:
+            return value
+    except ValueError:
+        pass
+    return DEFAULT_INTERVAL_SEC
+
+
+def beat_once() -> int:
+    from .job_manager import JobManager, META, STATUS, SCHEDULER_CALLBACK_URL
+
+    sent = 0
+    for job_id, data in JobManager.list_jobs().items():
+        if data.get(STATUS) in TERMINAL_STATUSES:
+            continue
+        url = (data.get(META) or {}).get(SCHEDULER_CALLBACK_URL)
+        if not url:
+            continue
+        try:
+            response = requests.patch(
+                    url, json={'status': 'RUNNING', 'heartbeat': True},
+                    timeout=REQUEST_TIMEOUT_SEC)
+            logger.debug(
+                    f'Heartbeat for job {job_id} returned {response.status_code}')
+            sent += 1
+        except Exception as exception:
+            logger.debug(f'Heartbeat for job {job_id} failed: {exception}')
+    return sent
+
+
+def _run() -> None:
+    while True:
+        try:
+            beat_once()
+        except Exception as exception:
+            logger.exception(f'Heartbeat loop error: {exception}')
+        if _stop_event.wait(heartbeat_interval_sec()):
+            return
+
+
+def ensure_started() -> None:
+    global _thread
+    with _lock:
+        if _thread is not None and _thread.is_alive():
+            return
+        _stop_event.clear()
+        _thread = threading.Thread(target=_run, name='job-heartbeat',
+                                   daemon=True)
+        _thread.start()
+        logger.info(
+                f'Job heartbeat started with interval {heartbeat_interval_sec()}s')
+
+
+def stop() -> None:
+    global _thread
+    with _lock:
+        _stop_event.set()
+        if _thread is not None:
+            _thread.join(timeout=1.0)
+        _thread = None

{quapp_common-0.0.13.dev6 → quapp_common-0.0.13.dev8}/quapp_common/component/backend/job_manager.py

@@ -66,6 +66,9 @@ class JobManager:
         """
         Stop any listener threads in this process and shutdown Manager.
         """
+        from .job_heartbeat import stop
+        stop()
+
         # stop listeners
         for key, (t, stop_ev) in list(cls._listener_threads.items()):
             stop_ev.set()
@@ -100,6 +103,9 @@ class JobManager:
         cls._jobs[job_id] = data  # proxy is process-safe
         cls._publish(job_id, {"event": "added", **data})
 
+        from .job_heartbeat import ensure_started
+        ensure_started()
+
     @classmethod
     def update_status(cls, job_id: str, status: str,
                       meta_update: Optional[Dict[str, Any]] = None) -> None: