Modify threadpool task queue

tencentyun · Jan 10, 2018 · ca2d335 · ca2d335
1 parent f438fe4
commit ca2d335
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qcloud_cos/cos_threadpool.py b/qcloud_cos/cos_threadpool.py
@@ -43,9 +43,9 @@ def get_result(self):
 
 class SimpleThreadPool:
 
-    def __init__(self, num_threads=5):
+    def __init__(self, num_threads=5, num_queue=0):
         self._num_threads = num_threads
-        self._queue = Queue()
+        self._queue = Queue(num_queue)
         self._lock = Lock()
         self._active = False
         self._workers = list()