|
|
|
@ -438,7 +438,8 @@ class MetaSecurityService: |
|
|
|
|
|
|
|
try: |
|
|
|
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 == "原始结果": |
|
|
|
count_sql = cls.build_count_sql(sql_query,db_type) |
|
|
|
@ -474,6 +475,7 @@ class MetaSecurityService: |
|
|
|
if select: |
|
|
|
select.set("order", None) |
|
|
|
select.set("limit", None) |
|
|
|
select.set("offset", None) |
|
|
|
|
|
|
|
cleaned_sql = parsed.sql(dialect=dialect) |
|
|
|
return f"SELECT COUNT(*) AS cnt FROM ({cleaned_sql}) t" |
|
|
|
@ -495,7 +497,7 @@ class MetaSecurityService: |
|
|
|
elif db_type.lower() == "mysql": |
|
|
|
# MySQL: 直接查询表字段(MySQL 没有 schema 的概念) |
|
|
|
query = f""" |
|
|
|
SELECT COLUMN_NAME |
|
|
|
SELECT COLUMN_NAME AS column_name |
|
|
|
FROM INFORMATION_SCHEMA.COLUMNS |
|
|
|
WHERE TABLE_NAME = '{table}' |
|
|
|
""" |
|
|
|
|