sixGuoDjango/django_定时器/团购任务.py

import os

import django

from sixGuoDjango.settings import BASE_DIR

os.environ.setdefault(key="DJANGO_SETTINGS_MODULE", value=f"{BASE_DIR.name}.settings")
django.setup()

# timer_manager.py
import threading
import secrets
from datetime import datetime, timedelta
from zoneinfo import ZoneInfo

from apscheduler.jobstores.base import JobLookupError
from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.jobstores.memory import MemoryJobStore
from apscheduler.executors.pool import ThreadPoolExecutor

from django_定时器.幸运拼团数据库 import lucky拼团
from wechat_数据库.雪花算法.雪花算法工具 import generator


class TwoHourWindowManager:
    """
    团购任务
    """
    _instance = None
    _lock = threading.Lock()

    def __new__(cls):
        if cls._instance is None:
            with cls._lock:
                if cls._instance is None:
                    cls._instance = super().__new__(cls)
                    cls._instance._initialized = False
        return cls._instance

    def __init__(self):
        if self._initialized:
            return
        self._initialized = True
        self.generate_str_id = None
        # 共享状态
        self.items = []
        self.current_cleanup_job_id: str | None = None
        self.state_lock = threading.RLock()  # 保护 items 和 job_id
        # 在这里,需要对数据库的数据进行处理
        self.init初始化数据库()
        # 初始化调度器（单例）
        jobstores = {'default': MemoryJobStore()}
        executors = {'default': ThreadPoolExecutor(max_workers=2)}
        self.scheduler = BackgroundScheduler(
            jobstores=jobstores, executors=executors, timezone='Asia/Shanghai',  # 替换为你的 settings.TIME_ZONE
            )
        self.scheduler.start()
        # 启动初始清理任务（2小时后）
        self._schedule_cleanup()

    def 开团重置(self):
        """APScheduler 调用的清理函数"""
        with self.state_lock:
            print("开团重置任务触发了啦")

            print(f"[{datetime.now()}] ⏰⏰⏰⏰⏰⏰⏰⏰⏰⏰⏰ 定时器到期，清空列表（当前 {len(self.items)} 项）")
            self.items.clear()
            self.current_cleanup_job_id = None  # 任务已执行，ID 失效
            self.init初始化数据库()
            self._schedule_cleanup()  # 开启新的任务
            print("开启新的任务")

    def init初始化数据库(self):
        print("初始化数据库执行中")
        generate_str = generator.generate_str()
        创建对象 = lucky拼团.objects.create(idKey=generate_str, lucky拼团详细=[])
        self.generate_str_id = 创建对象.idKey

    def _schedule_cleanup(self):
        """安排一个新的 2 小时后清理任务"""
        with self.state_lock:
            # 取消旧任务（如果存在且未执行）
            if self.current_cleanup_job_id is not None:
                print("存在旧的任务的吗?")
                try:
                    self.scheduler.remove_job(self.current_cleanup_job_id)
                    print("存在旧的任务的吗?")
                    pass
                except JobLookupError:
                    pass  # 可能已被执行或不存在
                    print("打印报错", JobLookupError)

            # 添加新任务
            job = self.scheduler.add_job(
                func=self.开团重置,
                trigger='date',
                run_date=datetime.now(tz=ZoneInfo("Asia/Shanghai")) + timedelta(seconds=7200),
                id=f"cleanup_{secrets.token_hex(8)}",
                name="Auto Cleanup",
                )
            self.current_cleanup_job_id = job.id
            print(f"[{datetime.now()}] 🕒 新建清理任务，2小时后执行，ID: {job.id}")

    def add_item(self, item):
        """外部调用：安全地添加元素"""
        with self.state_lock:
            self.items.append(item)
            print(f"[{datetime.now()}] ➕ 添加: {item}, 当前列表: {self.items}")
            if len(self.items) >= 4:
                print(f"[{datetime.now()}] 🧹 列表已满，立即清空并重置2小时计时器")
                self.items.clear()
                self._schedule_cleanup()  # 重置计时器

    def get_items(self):
        """
        :return:
        :rtype:
        """
        with self.state_lock:
            return self.items.copy()

    def shutdown(self):
        """
        """
        self.scheduler.shutdown()


if __name__ == '__main__':
    pass
    # main.py 或 views.py
    # from timer_manager import TwoHourWindowManager

    # 获取单例实例
    window = TwoHourWindowManager()

    # 添加元素（可在多线程/Web请求中调用）
    window.add_item("user_123")
    window.add_item("order_456")

    # 查看当前状态
    print("当前列表:", window.get_items())