klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 17:26:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add timeout to the datasource for driver connected

Browse Source
This commit is contained in:
junjun
2025-08-20 10:03:01 +08:00
parent 19d43d431f
commit 5021c14513
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/datasource/crud/datasource.py
+1 -1
View File
@@ -67,7 +67,7 @@ def check_status(session: SessionDep, trans: Trans, ds: CoreDatasource, is_raise
with dmPython.connect(user=conf.username, password=conf.password, server=conf.host,
port=conf.port) as conn, conn.cursor() as cursor:
try:
cursor.execute('select 1').fetchall()
cursor.execute('select 1', timeout=10).fetchall()
SQLBotLogUtil.info("success")
return True
except Exception as e:
backend/apps/db/db.py
+4 -4
View File
@@ -113,7 +113,7 @@ def get_schema(ds: CoreDatasource):
if ds.type == 'dm':
with dmPython.connect(user=conf.username, password=conf.password, server=conf.host,
port=conf.port) as conn, conn.cursor() as cursor:
cursor.execute(f"""select OBJECT_NAME from dba_objects where object_type='SCH'""")
cursor.execute(f"""select OBJECT_NAME from dba_objects where object_type='SCH'""", timeout=conf.timeout)
res = cursor.fetchall()
res_list = [item[0] for item in res]
return res_list
@@ -206,7 +206,7 @@ def get_tables(ds: CoreDatasource):
from all_tab_comments
where owner='{conf.dbSchema}'
AND (table_type = 'TABLE' or table_type = 'VIEW')
""")
""", timeout=conf.timeout)
res = cursor.fetchall()
res_list = [TableSchema(*item) for item in res]
return res_list
@@ -326,7 +326,7 @@ def get_fields(ds: CoreDatasource, table_name: str = None):
c.OWNER = '{conf.dbSchema}'
"""
sql2 = f" AND c.TABLE_NAME = '{table_name}'" if table_name is not None and table_name != "" else ""
cursor.execute(sql1 + sql2)
cursor.execute(sql1 + sql2, timeout=conf.timeout)
res = cursor.fetchall()
res_list = [ColumnSchema(*item) for item in res]
return res_list
@@ -358,7 +358,7 @@ def exec_sql(ds: CoreDatasource | AssistantOutDsSchema, sql: str, origin_column=
with dmPython.connect(user=conf.username, password=conf.password, server=conf.host,
port=conf.port) as conn, conn.cursor() as cursor:
try:
cursor.execute(sql)
cursor.execute(sql, timeout=conf.timeout)
res = cursor.fetchall()
columns = [field[0] for field in cursor.description] if origin_column else [field[0].lower() for
field in