From f48655de19e4f02abf980f12899b891c79505de9 Mon Sep 17 00:00:00 2001
From: wood <wood@czl.net>
Date: Tue, 3 Sep 2024 19:02:29 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=B3=E9=94=AE=E8=AF=8D=E6=A3=80=E6=B5=8B?=
 =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E4=B8=BA=E6=AF=8F=E7=A7=92=E6=9C=80=E5=A4=9A?=
 =?UTF-8?q?10=E6=9D=A1=E6=B6=88=E6=81=AF,=20=E9=98=B2=E6=AD=A2CPU=E5=8D=A0?=
 =?UTF-8?q?=E7=94=A8=E8=BF=87=E9=AB=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/guard.py | 34 ++++++++++++++++++++++++++++++----
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/src/guard.py b/src/guard.py
index 6c734f1..4aacaca 100644
--- a/src/guard.py
+++ b/src/guard.py
@@ -4,6 +4,7 @@ import logging
 import asyncio
 import time
 from telethon import TelegramClient, events
+from collections import deque
 
 # 环境变量
 BOT_TOKEN = os.environ.get('BOT_TOKEN')
@@ -26,6 +27,33 @@ def save_keywords(keywords):
 
 KEYWORDS = load_keywords()
 
+class RateLimiter:
+    def __init__(self, max_calls, period):
+        self.max_calls = max_calls
+        self.period = period
+        self.calls = deque()
+
+    async def __aenter__(self):
+        now = time.time()
+        while self.calls and now - self.calls[0] >= self.period:
+            self.calls.popleft()
+        if len(self.calls) >= self.max_calls:
+            await asyncio.sleep(self.period - (now - self.calls[0]))
+        self.calls.append(time.time())
+        return self
+
+    async def __aexit__(self, *args):
+        pass
+
+
+rate_limiter = RateLimiter(max_calls=10, period=1)  # 每秒最多处理10条消息
+
+async def process_message(event):
+    if not event.is_private and any(keyword in event.message.text.lower() for keyword in KEYWORDS):
+        if event.sender_id != ADMIN_ID:
+            await event.delete()
+            await event.respond("已撤回该消息。注:已发送的推广链接不要多次发送,置顶已有项目的推广链接也会自动撤回。")
+
 async def start_bot():
     client = TelegramClient('bot', api_id=6, api_hash='eb06d4abfb49dc3eeb1aeb98ae0f581e')
     await client.start(bot_token=BOT_TOKEN)
@@ -55,10 +83,8 @@ async def start_bot():
                 await event.respond(f"当前关键词和语句列表：\n" + "\n".join(KEYWORDS))
             return
 
-        if not event.is_private and any(keyword in event.message.text.lower() for keyword in KEYWORDS):
-            if event.sender_id != ADMIN_ID:
-                await event.delete()
-                await event.respond("已撤回该消息。注:已发送的推广链接不要多次发送,置顶已有项目的推广链接也会自动撤回。")
+        async with rate_limiter:
+            await process_message(event)
 
     logger.info("TeleGuard is running...")
     await client.run_until_disconnected()