123456789101112131415161718192021222324252627282930313233 |
- import time
- from concurrent.futures import ThreadPoolExecutor, wait
- from common.redis import install_video_data
- from video_processing.video_processing import VideoProcessing
- max_workers = 10
- def video_ai_task_start():
- with ThreadPoolExecutor( max_workers=max_workers) as executor:
- while True:
- try:
- redis_task_list = ['task:video_ai_top', 'task:video_ai_recommend', 'task:video_ai_complex_mode']
- # 提交所有任务并等待完成
- futures = [executor.submit( process_video_ai, redis_task ) for redis_task in redis_task_list]
- wait( futures ) # 等待所有任务完成
- except Exception as e:
- print(f"异常信息{e}")
- time.sleep(3)
- continue
- def process_video_ai(redis_task):
- try:
- print("开始执行任务")
- video_processor = VideoProcessing()
- video_processor.get_video(redis_task)
- print("执行完成")
- time.sleep(5)
- except Exception as e:
- print("处理任务时出现异常:", e)
- time.sleep(5)
- if __name__ == '__main__':
- install_video_data()
|