From 49c2e1d43c7a5b6232db24634059d531700a6a66 Mon Sep 17 00:00:00 2001
From: Franklin-F <dewujie64@gmail.com>
Date: Tue, 20 May 2025 21:43:00 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=20l0=E3=80=81l1=20?=
 =?UTF-8?q?=E5=92=8C=20l2=20=E9=98=9F=E5=88=97=E7=9A=84=E6=8E=A8=E9=80=81?=
 =?UTF-8?q?=E6=96=B9=E6=B3=95=EF=BC=8C=E4=BC=98=E5=8C=96=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=A4=84=E7=90=86=E6=B5=81=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DB.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/DB.py b/DB.py
index ce339e1..c8b62ee 100644
--- a/DB.py
+++ b/DB.py
@@ -129,6 +129,30 @@ class DBVidcon:
                 continue
         return out
 
+    @redis_retry(max_retries=3)
+    def push_l0(self, raws):
+        """向 l0（加急）队列写入数据"""
+        if isinstance(raws, str):
+            raws = [raws]
+        self.redis.lpush(self.l0_list_key, *raws)
+        print(f"[写入l0] 已推入 {len(raws)} 条")
+
+    @redis_retry(max_retries=3)
+    def push_l1(self, payloads):
+        """向 l1（普通）队列写入数据"""
+        if isinstance(payloads, str):
+            payloads = [payloads]
+        self.redis.rpush(self.l1_list_key, *payloads)
+        print(f"[写入l1] 已推入 {len(payloads)} 条")
+
+    @redis_retry(max_retries=3)
+    def push_l2(self, raws):
+        """向 l2（低优先）队列写入数据"""
+        if isinstance(raws, str):
+            raws = [raws]
+        self.redis.lpush(self.l2_list_key, *raws)
+        print(f"[写入l2] 已推入 {len(raws)} 条")
+
     @mysql_retry()
     def get_proxy_agent_dict(self) -> dict:
         sql = "SELECT rn, parameter FROM proxy_agent"
@@ -175,8 +199,6 @@ class DBVidcon:
 
     @redis_retry(max_retries=3)
     def rollback_l1(self, payloads):
-        if not payloads:
-            return
         if isinstance(payloads, str):
             payloads = [payloads]
         self.redis.rpush(self.l1_list_key, *payloads)