perf: 优化订阅处理,形成框架

This commit is contained in:
ibuler
2021-11-26 11:12:53 +08:00
committed by Jiangjie.Bai
parent de006bc664
commit c85249be36
8 changed files with 82 additions and 111 deletions

View File

@@ -1,6 +1,9 @@
from common.utils import get_logger
from contextlib import contextmanager
from django.db import connections
from common.utils import get_logger
logger = get_logger(__file__)
@@ -44,3 +47,10 @@ def get_objects(model, pks):
def close_old_connections():
for conn in connections.all():
conn.close_if_unusable_or_obsolete()
@contextmanager
def safe_db_connection():
close_old_connections()
yield
close_old_connections()