22 lines
692 B
Python
22 lines
692 B
Python
# shared_utils.py
|
|
import time
|
|
import logging
|
|
from threading import Lock
|
|
|
|
# 全局变量
|
|
task_registry = {}
|
|
file_registry = {}
|
|
file_lock = Lock()
|
|
task_lock = Lock()
|
|
|
|
def add_task_log(task_id, message, task_registry, task_lock):
|
|
"""添加任务日志"""
|
|
with task_lock:
|
|
if task_id in task_registry:
|
|
timestamp = time.strftime("%Y-%m-%d %H:%M:%S")
|
|
log_entry = f"[{timestamp}] {message}"
|
|
task_registry[task_id]['logs'].append(log_entry)
|
|
|
|
# 保持日志数量不超过1000条
|
|
if len(task_registry[task_id]['logs']) > 1000:
|
|
task_registry[task_id]['logs'] = task_registry[task_id]['logs'][-1000:] |