Update Consumer.py
Browse files- Consumer.py +5 -2
Consumer.py
CHANGED
|
@@ -152,10 +152,13 @@ async def main():
|
|
| 152 |
print(f"Received task {uid}")
|
| 153 |
|
| 154 |
# Create async task
|
|
|
|
|
|
|
|
|
|
| 155 |
task = asyncio.create_task(process_task(uid, code))
|
| 156 |
active_tasks[uid] = task
|
| 157 |
-
if
|
| 158 |
-
|
| 159 |
|
| 160 |
await asyncio.sleep(POLL_INTERVAL)
|
| 161 |
|
|
|
|
| 152 |
print(f"Received task {uid}")
|
| 153 |
|
| 154 |
# Create async task
|
| 155 |
+
taskold = None
|
| 156 |
+
if uid in active_tasks:
|
| 157 |
+
taskold = active_tasks[uid]
|
| 158 |
task = asyncio.create_task(process_task(uid, code))
|
| 159 |
active_tasks[uid] = task
|
| 160 |
+
if taskold and taskold.is_running():
|
| 161 |
+
taskold.cancel()
|
| 162 |
|
| 163 |
await asyncio.sleep(POLL_INTERVAL)
|
| 164 |
|