rb-commons 0.2.3__tar.gz → 0.2.5__tar.gz

{rb_commons-0.2.3 → rb_commons-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.2.3
+Version: 0.2.5
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

rb_commons-0.2.5/rb_commons/http/consul.py

@@ -0,0 +1,44 @@
+import aiocache
+import consul
+from fastapi import HTTPException, Request
+from rb_commons.configs.config import configs
+
+
+class ServiceDiscovery:
+    def __init__(self) -> None:
+        self.consul_client = consul.Consul(host=configs.consul_host, port=configs.consul_port)
+        self.cache = aiocache.Cache(aiocache.SimpleMemoryCache)
+
+    async def _get_service_instances(self, service_name: str) -> dict:
+        """Get a healthy service instance from Consul"""
+        index, services = self.consul_client.health.service(service_name, passing=True)
+
+        if not services:
+            raise HTTPException(status_code=503,
+                                detail="Service not available")
+        return services
+
+    def _select_instance(self, instances: list, host: str) -> dict:
+        """Select instance using consistent hashing"""
+        if not instances:
+            raise HTTPException(status_code=503, detail="No healthy instances")
+
+        key = host
+        hash_value = hash(key)
+        return instances[hash_value % len(instances)]
+
+    async def _get_cached_service_instances(self, service_name: str) -> list:
+        """Get service instances with caching"""
+        cache_key = f"service:{service_name}"
+        instances = await self.cache.get(cache_key)
+        if not instances:
+            instances = await self._get_service_instances(service_name)
+            if instances:
+                await self.cache.set(cache_key, instances, ttl=30)  # Cache for 30 seconds
+        return instances
+
+    def _build_instance_url(self, instance: dict) -> str:
+        """Build target URL with proper path handling"""
+        service_address = instance['Service']['Address'] or instance['Node']['Address']
+        service_port = instance['Service']['Port']
+        return f"http://{service_address}:{service_port}"

{rb_commons-0.2.3 → rb_commons-0.2.5}/rb_commons.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.2.3
+Version: 0.2.5
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

{rb_commons-0.2.3 → rb_commons-0.2.5}/rb_commons.egg-info/SOURCES.txt

@@ -14,6 +14,7 @@ rb_commons/configs/injections.py
 rb_commons/configs/rabbitmq.py
 rb_commons/http/__init__.py
 rb_commons/http/base_api.py
+rb_commons/http/consul.py
 rb_commons/http/exceptions.py
 rb_commons/orm/__init__.py
 rb_commons/orm/exceptions.py

{rb_commons-0.2.3 → rb_commons-0.2.5}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="rb-commons",
-    version="0.2.3",
+    version="0.2.5",
     author="Abdulvoris",
     author_email="erkinovabdulvoris101@gmail.com",
     description="Commons of project and simplified orm based on sqlalchemy.",