github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-05 17:30:30 +00:00
Code Issues Projects Releases Wiki Activity

perf: simplify db using

Browse Source
This commit is contained in:
ibuler
2025-05-14 15:05:12 +08:00
committed by w940853815
parent e9ff988d8c
commit f826f43495
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
apps/common/db/utils.py
View File

@@ -56,7 +56,14 @@ def close_old_connections():
@contextmanager
def safe_db_connection(auto_close=False):
def safe_db_connection(auto_close=True):
close_old_connections()
yield
close_old_connections()
@contextmanager
def safe_atomic_db_connection(auto_close=False):
in_atomic_block = connection.in_atomic_block # 当前是否处于事务中
autocommit = transaction.get_autocommit() # 是否启用了自动提交
created = False
@@ -70,7 +77,6 @@ def safe_db_connection(auto_close=False):
finally:
# 如果不是事务中（API 请求中可能需要提交事务），则关闭连接
if auto_close or (created and not in_atomic_block and autocommit):
print("close connection in safe_db_connection")
close_old_connections()