22 lines
519 B
Python
22 lines
519 B
Python
from queue import Queue
|
|
|
|
|
|
class WaitingQueue:
|
|
def __init__(self):
|
|
self.queue = Queue(maxsize=500)
|
|
|
|
def enqueue(self,uid, video_path,audio_path):
|
|
data = {
|
|
"uid": uid,
|
|
"video_path": video_path,
|
|
"audio_path": audio_path
|
|
}
|
|
self.queue.put(data,timeout=10)
|
|
|
|
def dequeue(self):
|
|
data = self.queue.get(timeout=10)
|
|
return data["uid"], data["video_path"], data["audio_path"]
|
|
|
|
def get_size(self):
|
|
return self.queue.qsize()
|