cafs-cache-cdn-client 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

cafs_cache_cdn_client/cafs/client.py

@@ -1,5 +1,4 @@
 import asyncio
-from asyncio import Queue, QueueShutDown
 from collections.abc import (
     AsyncIterator,
     Callable,
@@ -282,7 +281,8 @@ class ConnectionPool:
 
     _lock: asyncio.Lock
     _connections: set[CAFSConnection]
-    _connection_queue: Queue[CAFSConnection]
+    _connection_queue: asyncio.Queue[CAFSConnection]
+    _close_event: asyncio.Event
 
     def __init__(
         self,
@@ -297,8 +297,9 @@ class ConnectionPool:
         self.connection_per_server = connection_per_server
 
         self._connections = set()
-        self._connection_queue = Queue()
+        self._connection_queue = asyncio.Queue()
         self._lock = asyncio.Lock()
+        self._close_event = asyncio.Event()
 
     async def get_connection_count(self) -> int:
         async with self._lock:
@@ -315,10 +316,21 @@ class ConnectionPool:
                 await self._connection_queue.put(conn)
 
     async def _get_connection(self) -> CAFSConnection:
-        try:
-            return await self._connection_queue.get()
-        except QueueShutDown as err:
-            raise EmptyConnectionPoolError() from err
+        if self._close_event.is_set():
+            raise EmptyConnectionPoolError()
+        get_task = asyncio.create_task(self._connection_queue.get())
+        close_task = asyncio.create_task(self._close_event.wait())
+        _, pending = await asyncio.wait(
+            [get_task, close_task], return_when=asyncio.FIRST_COMPLETED
+        )
+
+        for task in pending:
+            task.cancel()
+
+        if get_task in pending:
+            raise EmptyConnectionPoolError()
+
+        return get_task.result()
 
     async def _release_connection(self, conn: CAFSConnection) -> None:
         await self._connection_queue.put(conn)
@@ -328,14 +340,15 @@ class ConnectionPool:
         async with self._lock:
             self._connections.remove(conn)
             if not self._connections:
-                self._connection_queue.shutdown(immediate=True)
+                self._close_event.set()
 
     async def close(self) -> None:
         async with self._lock:
-            self._connection_queue.shutdown(immediate=True)
+            self._close_event.set()
             for conn in self._connections:
                 if conn.is_connected:
                     await conn.disconnect()
+            self._connections.clear()
 
     @asynccontextmanager
     async def connection(self) -> AsyncIterator[CAFSConnection]:
@@ -428,7 +441,6 @@ class CAFSClient:
         stop_event = asyncio.Event()
         workers = [asyncio.create_task(worker(stop_event)) for _ in range(len(blobs))]
         errors = await asyncio.gather(*workers, return_exceptions=True)
-        files_queue.shutdown(immediate=True)
 
         for err in errors:
             if isinstance(err, Exception):
@@ -508,7 +520,6 @@ class CAFSClient:
             asyncio.create_task(worker(stop_event)) for _ in range(max_concurrent)
         ]
         errors = await asyncio.gather(*workers, return_exceptions=True)
-        files_queue.shutdown(immediate=True)
 
         for err in errors:
             if isinstance(err, Exception):

cafs_cache_cdn_client/file_utils.py

@@ -40,9 +40,10 @@ class LocalFile:
 
 def walk(directory: Path) -> list[LocalFile]:
     results = []
-    for root, _, files in directory.walk():
+    for root, _, files in os.walk(str(directory)):
+        root_ = Path(root)
         for file in files:
-            file_ = root / file
+            file_ = root_ / file
             file_stat = file_.stat()
             results.append(
                 LocalFile(

cafs_cache_cdn_client/repo/client.py

@@ -82,7 +82,7 @@ class RepoClient:
 
     async def tag_ref(self, repo: str, ref: str, tag: str) -> None:
         await self._request(
-            f'/repository/{repo}/{ref}/tag', method=HTTPMethod.PUT, data={'tag': tag}
+            f'/repository/{repo}/{ref}/tag', method=HTTPMethod.POST, data={'tag': tag}
         )
 
     async def post_ref_info(self, repo: str, ref: str, data: dt.RefInfoBody) -> None:

{cafs_cache_cdn_client-1.0.5.dist-info → cafs_cache_cdn_client-1.0.7.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.3
 Name: cafs-cache-cdn-client
-Version: 1.0.5
+Version: 1.0.7
 Summary: Async Cache CDN client implementation
 Keywords: cafs,cache
 Author: Konstantin Belov
 Author-email: k.belov@gaijin.team
 Requires-Python: >=3.11,<4.0
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Development Status :: 5 - Production/Stable
+Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13

{cafs_cache_cdn_client-1.0.5.dist-info → cafs_cache_cdn_client-1.0.7.dist-info}/RECORD

@@ -5,14 +5,14 @@ cafs_cache_cdn_client/cafs/blob/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm
 cafs_cache_cdn_client/cafs/blob/hash_.py,sha256=tI4qenyHnEEtKnNUOq4a25L1GnlKjoieyPGQDoz14rA,736
 cafs_cache_cdn_client/cafs/blob/package.py,sha256=pOMF5Dw42APHqIlRfcs1YPscqjcbO32Z35aEG0SqoHo,5865
 cafs_cache_cdn_client/cafs/blob/utils.py,sha256=tqAOvPWIL-sOwXwPiH9Kbsqn5VekJtlf6UO8V9TTtwM,1021
-cafs_cache_cdn_client/cafs/client.py,sha256=TJxwc_PFFmSeCxfTjybh--NrCDzuzuxh3Si42Lwm4Tc,18232
+cafs_cache_cdn_client/cafs/client.py,sha256=ebaFivgSQAzS0verVwuzRGZ98pwSEoaiTaUlmL8VB3o,18536
 cafs_cache_cdn_client/cafs/exceptions.py,sha256=I4E3lFrQ_ysR8gDIh3E0Gz4OMihfLD6vLj_KwKzxnTU,654
 cafs_cache_cdn_client/cafs/types.py,sha256=q7zqDQ8yKRn-HRSsXDuzawmikAEcyoYQ-_p2WmuDFy4,324
 cafs_cache_cdn_client/client.py,sha256=d7Dw2U9h8dPYRDPu0mbnXc62SlLy4ruvQk2Alj3cVk8,4584
-cafs_cache_cdn_client/file_utils.py,sha256=kDaak0n3emLIn0DXB02yWFkUhYxuLyTmH93WPszKEUQ,2642
+cafs_cache_cdn_client/file_utils.py,sha256=uX5gAc3CUGXpOktD3m_xrz31d4PRnKkq2RkBMqZSe0o,2677
 cafs_cache_cdn_client/repo/__init__.py,sha256=JzCxKznnLV7WElJuznG_jL2s9tYPecJL5uJRnPZsW5M,58
-cafs_cache_cdn_client/repo/client.py,sha256=FEiLefNmvJKwA2gz6dj2gcvt3uKRzLqBARGy4dvXP98,3545
+cafs_cache_cdn_client/repo/client.py,sha256=cxZ6PLwqKG0tIbAXYacpd_eB_TE31-u506q2JyjOenY,3546
 cafs_cache_cdn_client/repo/datatypes.py,sha256=aTlzFoEibfAOQTqcsg-RmFjnwenS_rBCEMKtN6glOhY,531
-cafs_cache_cdn_client-1.0.5.dist-info/METADATA,sha256=m-dhAfuTdQz97XW280baoDd2vZDxEQW0dtJF-eLqbFI,3679
-cafs_cache_cdn_client-1.0.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-cafs_cache_cdn_client-1.0.5.dist-info/RECORD,,
+cafs_cache_cdn_client-1.0.7.dist-info/METADATA,sha256=8Bgg1yCjjO5LCpRiX0IaJ6G7RuMhzmGlJp_xu0mtZLA,3666
+cafs_cache_cdn_client-1.0.7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+cafs_cache_cdn_client-1.0.7.dist-info/RECORD,,