sharedkernel 2.5.5__tar.gz → 2.5.7__tar.gz

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sharedkernel
-Version: 2.5.5
+Version: 2.5.7
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -21,6 +21,7 @@ Requires-Dist: beautifulsoup4
 Requires-Dist: deepdiff
 Requires-Dist: kombu
 Requires-Dist: redis
+Requires-Dist: pymongo
 Dynamic: author
 Dynamic: description
 Dynamic: description-content-type
@@ -31,6 +32,10 @@ Dynamic: summary
 this is a shared kernel package
 
 # Change Log
+### Version 2.5.7
+- Add Database Health Check
+### Version 2.5.6
+- fix bug
 ### Version 2.5.5
 - fix bug
 ### Version 2.5.4

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/README.md

@@ -2,6 +2,10 @@
 this is a shared kernel package
 
 # Change Log
+### Version 2.5.7
+- Add Database Health Check
+### Version 2.5.6
+- fix bug
 ### Version 2.5.5
 - fix bug
 ### Version 2.5.4

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/setup.py

@@ -37,10 +37,11 @@ setup(
         "beautifulsoup4",
         "deepdiff",
         "kombu",
-        "redis"
+        "redis",
+        "pymongo"
     ],
     # *strongly* suggested for sharing
-    version="2.5.5",
+    version="2.5.7",
     description="sharekernel is a shared package between all python projects",
     long_description=long_description,
     long_description_content_type="text/markdown",

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel/config.py

@@ -10,6 +10,7 @@ LOG_EXCHANGE = os.getenv("LOG_EXCHANGE")
 LOG_REQUESTS_QUEUE = os.getenv("LOG_REQUESTS_QUEUE")
 SERVICE_NAME = os.getenv("SERVICE_NAME")
 IP_HEADER_NAME = os.getenv("IP_HEADER_NAME")
+DB_HEALTH_CHECK_INTERVAL = int(os.getenv("DB_HEALTH_CHECK_INTERVAL", 15))
 
 # IP Session ************************************************************************
 IP_SESSION_AUTH_ENABLE = True if os.getenv("IP_SESSION_AUTH_ENABLE", "False").lower() == "true" else False

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel/database/distributed_cache.py

@@ -8,7 +8,7 @@ from redis.cluster import RedisCluster, ClusterNode
 from sharedkernel.exception.exception import BusinessException
 
 from sharedkernel import config
-from core.enums import ErrorCode, RedisMode
+from sharedkernel.enum import ErrorCode, RedisMode
 
 T = TypeVar("T", bound=BaseModel)
 

sharedkernel-2.5.7/sharedkernel/database/mongo_health_checker.py

@@ -0,0 +1,67 @@
+import time
+import threading
+from typing import Callable, Optional
+from pymongo.errors import PyMongoError
+
+from sharedkernel import config
+
+class MongoHealthChecker:
+    def __init__(
+        self,
+        client,
+        interval: int = config.DB_HEALTH_CHECK_INTERVAL,
+        on_connect: Optional[Callable[[], None]] = None,
+        on_disconnect: Optional[Callable[[Exception], None]] = None,
+    ):
+
+        self.client = client
+        self.interval = interval
+        self.on_connect = on_connect
+        self.on_disconnect = on_disconnect
+
+        self._is_connected = None
+        self._thread = None
+        self._stop_event = threading.Event()
+
+    def _default_on_connect(self):
+        print(f"Mongo connected")
+
+    def _default_on_disconnect(self, error):
+        print(f"Mongo DISCONNECTED: {error}")
+
+    def _run(self):
+        db = self.client.admin
+
+        while not self._stop_event.is_set():
+            try:
+                db.command("ping")
+
+                if self._is_connected is not True:
+                    (self.on_connect or self._default_on_connect)()
+                    self._is_connected = True
+
+            except PyMongoError as e:
+                if self._is_connected is not False:
+                    (self.on_disconnect or self._default_on_disconnect)()
+                    self._is_connected = False
+
+            time.sleep(self.interval)
+
+    def start(self):
+        if self._thread and self._thread.is_alive():
+            return
+
+        self._stop_event.clear()
+
+        self._thread = threading.Thread(
+            target=self._run,
+            daemon=True
+        )
+        self._thread.start()
+
+    def stop(self):
+        if not self._thread:
+            return
+
+        self._stop_event.set()
+        self._thread.join()

sharedkernel-2.5.7/sharedkernel/enum/__init__.py

@@ -0,0 +1,2 @@
+from .error_code import ErrorCode
+from .redis_mode_enum import RedisMode

sharedkernel-2.5.7/sharedkernel/enum/redis_mode_enum.py

@@ -0,0 +1,8 @@
+
+from enum import Enum
+
+class RedisMode(str, Enum):
+    SENTINEL = "sentinel"
+    CLUSTER = "cluster"
+    STANDALONE = "standalone"
+    

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel/logger/logger_service.py

@@ -1,9 +1,11 @@
 from fastapi import HTTPException, WebSocket
 from kombu import Connection, Producer, Exchange
+from pymongo import monitoring
 from pydantic import BaseModel
 from typing import Optional
 import datetime
 import uuid
+import os
 
 from sharedkernel.logger.log_info import log_info
 from sharedkernel.enum import ErrorCode
@@ -11,6 +13,7 @@ from sharedkernel.objects.user_info import current_user_info
 from sharedkernel.exception.exception import BusinessException
 from sharedkernel.logger.log_dto import LogDTO
 from sharedkernel.logger.log_enums import LogEventTypeEnum, LogLevelEnum
+from sharedkernel.database.mongo_health_checker import MongoHealthChecker
 from sharedkernel import config
 
 
@@ -19,11 +22,13 @@ rabbitmq_url = (
     f"@{config.RABBITMQ_HOST}/{config.RABBITMQ_VHOST}"
 )
 
+
 class LoggerService:
     def __init__(self):
         self._connection: Optional[Connection] = None
         self._producer: Optional[Producer] = None
         self._exchange = Exchange(config.LOG_EXCHANGE, type="topic")
+        self.checker: Optional[MongoHealthChecker] = None
 
 
     def _ensure_connection(self):
@@ -51,10 +56,10 @@ class LoggerService:
 
     def build(
         self,
-        method_name: str,
-        event_type: LogEventTypeEnum,
-        endpoint_name: str = None,
-        is_success: bool = True,
+        method_name: Optional[str] = None,
+        event_type: Optional[LogEventTypeEnum] = None,
+        endpoint_name: Optional[str] = None,
+        is_success: Optional[bool] = True,
         failure_reason: Optional[str] = None,
     ) -> LogDTO:
 
@@ -147,3 +152,56 @@ class LoggerService:
         )
 
         log_info.set(log_obj)
+
+
+    def set_log_to_context(self, correlation_id: str = None):
+
+        if not config.LOG_ENABLE:
+            return
+
+        log_obj = LogDTO(CorrelationId=correlation_id)
+        log_info.set(log_obj)
+
+
+    def _on_mongo_connect(self):
+        self.set_log_to_context(correlation_id=uuid.uuid4().hex)
+
+        log = self.build(
+            method_name="mongo_health_check",
+            event_type=LogEventTypeEnum.START,
+            endpoint_name="اتصال به دیتابیس",
+            is_success=True,
+        )
+        self.emit(log)
+
+
+    def _on_mongo_disconnect(self):
+        log = self.build(
+            method_name="mongo_health_check",
+            event_type=LogEventTypeEnum.END,
+            endpoint_name="قطع اتصال به دیتابیس",
+            is_success=False,
+        )
+        self.emit(log)
+
+
+    def start_db_logger(self, db_client):
+        if not config.LOG_ENABLE:
+            return
+
+        self.checker = MongoHealthChecker(
+            db_client,
+            on_connect=self._on_mongo_connect,
+            on_disconnect=self._on_mongo_disconnect,
+        )
+
+        self.checker.start()
+
+
+    def stop_db_logger(self):
+        if not config.LOG_ENABLE:
+            return
+
+        if self.checker:
+            self.checker.stop()
+            self.checker = None

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sharedkernel
-Version: 2.5.5
+Version: 2.5.7
 Summary: sharekernel is a shared package between all python projects
 Author: Smilinno
 Description-Content-Type: text/markdown
@@ -21,6 +21,7 @@ Requires-Dist: beautifulsoup4
 Requires-Dist: deepdiff
 Requires-Dist: kombu
 Requires-Dist: redis
+Requires-Dist: pymongo
 Dynamic: author
 Dynamic: description
 Dynamic: description-content-type
@@ -31,6 +32,10 @@ Dynamic: summary
 this is a shared kernel package
 
 # Change Log
+### Version 2.5.7
+- Add Database Health Check
+### Version 2.5.6
+- fix bug
 ### Version 2.5.5
 - fix bug
 ### Version 2.5.4

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel.egg-info/SOURCES.txt

@@ -22,9 +22,11 @@ sharedkernel/database/audit_model.py
 sharedkernel/database/distributed_cache.py
 sharedkernel/database/mongo_generic_audit_repository.py
 sharedkernel/database/mongo_generic_repository.py
+sharedkernel/database/mongo_health_checker.py
 sharedkernel/database/pagination_response_dto.py
 sharedkernel/enum/__init__.py
 sharedkernel/enum/error_code.py
+sharedkernel/enum/redis_mode_enum.py
 sharedkernel/enum/sort_order.py
 sharedkernel/exception/__init__.py
 sharedkernel/exception/exception.py

{sharedkernel-2.5.5 → sharedkernel-2.5.7}/sharedkernel.egg-info/requires.txt

@@ -15,3 +15,4 @@ beautifulsoup4
 deepdiff
 kombu
 redis
+pymongo

sharedkernel-2.5.5/sharedkernel/enum/__init__.py

@@ -1 +0,0 @@
-from .error_code import ErrorCode