management/scheduler: replace queue with transparent list + semaphore

sbourdeauducq · sbourdeauducq · commit 08f2aa850389 · 2014-12-09T16:26:50.000+08:00
diff --git a/artiq/management/scheduler.py b/artiq/management/scheduler.py
@@ -6,7 +6,8 @@
 class Scheduler:
     def __init__(self, *args, **kwargs):
         self.worker = Worker(*args, **kwargs)
-        self.queue = asyncio.Queue()
+        self.queued = []
+        self.queue_count = asyncio.Semaphore(0)
 
     @asyncio.coroutine
     def start(self):
@@ -21,11 +22,13 @@ def stop(self):
         yield from self.worker.end_process()
 
     def run_once(self, run_params, timeout):
-        self.queue.put_nowait((run_params, timeout))
+        self.queued.append((run_params, timeout))
+        self.queue_count.release()
 
     @asyncio.coroutine
     def _schedule(self):
         while True:
-            run_params, timeout = yield from self.queue.get()
+            yield from self.queue_count.acquire()
+            run_params, timeout = self.queued.pop(0)
             result = yield from self.worker.run(run_params, timeout)
             print(result)
