improve message clean logic (#11487)

This commit is contained in:
Jyong 2024-12-09 16:12:30 +08:00 committed by GitHub
parent 537068cfde
commit 00ac7edeb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -36,14 +36,16 @@ def clean_messages():
db.session.query(Message) db.session.query(Message)
.filter(Message.created_at < plan_sandbox_clean_message_day) .filter(Message.created_at < plan_sandbox_clean_message_day)
.order_by(Message.created_at.desc()) .order_by(Message.created_at.desc())
.paginate(page=page, per_page=100) .limit(100)
.all()
) )
except NotFound: except NotFound:
break break
if messages.items is None or len(messages.items) == 0: if not messages:
break break
for message in messages.items: for message in messages:
plan_sandbox_clean_message_day = message.created_at
app = App.query.filter_by(id=message.app_id).first() app = App.query.filter_by(id=message.app_id).first()
features_cache_key = f"features:{app.tenant_id}" features_cache_key = f"features:{app.tenant_id}"
plan_cache = redis_client.get(features_cache_key) plan_cache = redis_client.get(features_cache_key)