Browse Source

mysql相关适配

master
siyaqi 4 weeks ago
parent
commit
b28aa16c50
  1. 6
      vue-fastapi-backend/module_admin/service/metasecurity_service.py

6
vue-fastapi-backend/module_admin/service/metasecurity_service.py

@ -438,7 +438,8 @@ class MetaSecurityService:
try: try:
async with async_session() as session: async with async_session() as session:
await session.execute(text("SET statement_timeout = 30000")) if (db_type or "").upper() == "POSTGRESQL":
await session.execute(text("SET statement_timeout = 30000"))
# ⭐ 原始数量 # ⭐ 原始数量
if sql_type == "原始结果": if sql_type == "原始结果":
count_sql = cls.build_count_sql(sql_query,db_type) count_sql = cls.build_count_sql(sql_query,db_type)
@ -474,6 +475,7 @@ class MetaSecurityService:
if select: if select:
select.set("order", None) select.set("order", None)
select.set("limit", None) select.set("limit", None)
select.set("offset", None)
cleaned_sql = parsed.sql(dialect=dialect) cleaned_sql = parsed.sql(dialect=dialect)
return f"SELECT COUNT(*) AS cnt FROM ({cleaned_sql}) t" return f"SELECT COUNT(*) AS cnt FROM ({cleaned_sql}) t"
@ -495,7 +497,7 @@ class MetaSecurityService:
elif db_type.lower() == "mysql": elif db_type.lower() == "mysql":
# MySQL: 直接查询表字段(MySQL 没有 schema 的概念) # MySQL: 直接查询表字段(MySQL 没有 schema 的概念)
query = f""" query = f"""
SELECT COLUMN_NAME SELECT COLUMN_NAME AS column_name
FROM INFORMATION_SCHEMA.COLUMNS FROM INFORMATION_SCHEMA.COLUMNS
WHERE TABLE_NAME = '{table}' WHERE TABLE_NAME = '{table}'
""" """

Loading…
Cancel
Save