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

perf: Optimize the speed of data querying

Browse Source
This commit is contained in:
junjun
2025-09-24 21:13:17 +08:00
committed by Junjun
parent 0e09a0b480
commit 74586ee630
2 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/datasource/crud/datasource.py
+13 -11
View File
@@ -384,17 +384,19 @@ def get_table_schema(session: SessionDep, current_user: CurrentUser, ds: CoreDat
else:
schema_table += f", {table_comment}\n[\n"
field_list = []
for field in obj.fields:
field_comment = ''
if field.custom_comment:
field_comment = field.custom_comment.strip()
if field_comment == '':
field_list.append(f"({field.field_name}:{field.field_type})")
else:
field_list.append(f"({field.field_name}:{field.field_type}, {field_comment})")
schema_table += ",\n".join(field_list)
schema_table += '\n]\n'
if obj.fields:
field_list = []
for field in obj.fields:
field_comment = ''
if field.custom_comment:
field_comment = field.custom_comment.strip()
if field_comment == '':
field_list.append(f"({field.field_name}:{field.field_type})")
else:
field_list.append(f"({field.field_name}:{field.field_type}, {field_comment})")
schema_table += ",\n".join(field_list)
schema_table += '\n]\n'
t_obj = {"id": obj.table.id, "schema_table": schema_table}
tables.append(t_obj)
all_tables.append(t_obj)
backend/apps/datasource/crud/permission.py
+2
View File
@@ -37,6 +37,7 @@ def get_row_permission_filters(session: SessionDep, current_user: CurrentUser, d
if p_list is not None and u_list is not None and permission.id in p_list and (
current_user.id in u_list or f'{current_user.id}' in u_list):
flag = True
break
if flag:
res.append(transRecord2DTO(session, permission))
where_str = transFilterTree(session, res, ds)
@@ -64,6 +65,7 @@ def get_column_permission_fields(session: SessionDep, current_user: CurrentUser,
if p_list is not None and u_list is not None and permission.id in p_list and (
current_user.id in u_list or f'{current_user.id}' in u_list):
flag = True
break
if flag:
permission_list = json.loads(permission.permissions)
fields = filter_list(fields, permission_list)