AQ test

Author: GoodManWEN

cx_Oracle_async/AQ.py

@@ -20,7 +20,7 @@ async def deqOne(self , *args , **kwargs):
         async with self._deqlock:
             return await self._loop.run_in_executor(self._thread_pool , self._queue.deqOne , *args , **kwargs)
 
-    def deqMany(self , maxMessages):
+    def deqMany(self , maxMessages = -1):
         return DeqManyWrapper(self._loop , self._thread_pool , self._queue , self._deqlock , maxMessages)
 
     @property
@@ -39,9 +39,12 @@ def __init__(self , loop , thread_pool , queue , deqlock ,maxMessages):
         self._thread_pool = thread_pool
         self._queue = queue
         self._count = 0
-        self._max = maxMessages
+        self._max = maxMessages if maxMessages != -1 else (1 << 16 - 1) 
         self._deqlock = deqlock
 
+    def __await__(self):
+        return self._loop.run_in_executor(self._thread_pool , self._queue.deqMany , self._max).__await__()
+
     def __aiter__(self):
         return self
 

cx_Oracle_async/context.py

@@ -4,8 +4,8 @@ def __init__(self , coro):
         self._coro = coro
         self._obj = None
 
-    def __next__(self):
-        return self.send(None)
+    # def __next__(self):
+    #     return self.send(None)
 
     def __iter__(self):
         return self._coro.__await__()