初始化提交
This commit is contained in:
21
app/controllers/manager/memory_manager.py
Normal file
21
app/controllers/manager/memory_manager.py
Normal file
@@ -0,0 +1,21 @@
|
||||
from queue import Queue
|
||||
from typing import Dict
|
||||
|
||||
from app.controllers.manager.base_manager import TaskManager
|
||||
|
||||
|
||||
class InMemoryTaskManager(TaskManager):
|
||||
def create_queue(self):
|
||||
return Queue(maxsize=self.max_queued_tasks)
|
||||
|
||||
def enqueue(self, task: Dict):
|
||||
self.queue.put(task)
|
||||
|
||||
def dequeue(self):
|
||||
return self.queue.get()
|
||||
|
||||
def is_queue_empty(self):
|
||||
return self.queue.empty()
|
||||
|
||||
def queue_size(self):
|
||||
return self.queue.qsize()
|
||||
Reference in New Issue
Block a user