fix(mysql): strip CLI \G suffix before executing queries
This commit is contained in:
5 files changed
+82
-8
No files matched your search
@@ -59,7 +59,7 @@ import { appendLargeValueCells, canUseTableDataLargeValuePreview, remapLargeValu
|
||||
import { simpleDataGridOrderByReferencesMissingColumn, sortDataGridRowIndexes, type DataGridSortDirection } from "@/lib/dataGrid/dataGridSort";
|
||||
import { normalizeResultPageSize } from "@/lib/dataGrid/paginationPageSize";
|
||||
import { agentProtocolQueryResultMaxRows, capQueryResultTotal, effectiveQueryResultMaxRows, limitQueryPagination, queryResultLimitReached } from "@/lib/dataGrid/queryResultRowLimit";
|
||||
import { elasticsearchRestRequestRanges, executableStatementRanges, splitSqlStatementRanges, sqlStatementParameterOptionsForCompatibility } from "@/lib/sql/sqlStatementRanges";
|
||||
import { elasticsearchRestRequestRanges, executableStatementRanges, splitSqlStatementRanges, sqlStatementParameterOptionsForCompatibility, stripMysqlClientDisplayCommand } from "@/lib/sql/sqlStatementRanges";
|
||||
import type { SqlParameterOptions } from "@/lib/sql/sqlParameters";
|
||||
import { replaceSqlServerLeadingUseQuery, sqlServerLeadingUseScript, sqlServerUseDatabaseFromStatement } from "@/lib/sql/sqlCompletionLookupTarget";
|
||||
import { classifySqlRisk } from "@/lib/sql/sqlRisk";
|
||||
@@ -6241,6 +6241,9 @@ export const useQueryStore = defineStore("query", () => {
|
||||
mongoCommands = splitMongoCommandRanges(sql);
|
||||
}
|
||||
const effectiveDbType = effectiveDatabaseTypeForConnection(conn);
|
||||
if (effectiveDbType === "mysql") {
|
||||
sqlToExecute = stripMysqlClientDisplayCommand(sqlToExecute);
|
||||
}
|
||||
if (tab.autoCommit === false && !supportsTransaction(conn?.db_type)) {
|
||||
tab.autoCommit = true;
|
||||
}
|
||||
@@ -8291,12 +8294,13 @@ export const useQueryStore = defineStore("query", () => {
|
||||
await connStore.ensureConnected(location.connectionId);
|
||||
const conn = connStore.getConfig(location.connectionId);
|
||||
const effectiveDbType = effectiveDatabaseTypeForConnection(conn);
|
||||
const executableSql = effectiveDbType === "mysql" ? stripMysqlClientDisplayCommand(sql) : sql;
|
||||
const executionDatabase = dataTabExecutionDatabase(conn, location.database, location.catalog);
|
||||
// main 引入全局查询超时:queryTimeoutSecsForConnection 现需传入全局默认值;
|
||||
// settingsStore 取 defineStore 顶层声明的实例(本函数无局部覆盖)。
|
||||
const queryTimeoutSecs = queryTimeoutSecsForConnection(conn, settingsStore.editorSettings.globalQueryTimeoutSecs);
|
||||
const useAgentCursor = usesAgentCursorForQuery(conn?.db_type, conn?.driver_profile);
|
||||
const queryBaseSql = queryResultBaseSql(tab);
|
||||
const queryBaseSql = effectiveDbType === "mysql" ? stripMysqlClientDisplayCommand(queryResultBaseSql(tab)) : queryResultBaseSql(tab);
|
||||
const exportSettings = useSettingsStore().editorSettings;
|
||||
const exportRowLimit = exportSettings.exportRowLimitEnabled ? exportSettings.exportRowLimit : Number.POSITIVE_INFINITY;
|
||||
|
||||
@@ -8375,7 +8379,7 @@ export const useQueryStore = defineStore("query", () => {
|
||||
const remaining = exportRowLimit - rows.length;
|
||||
const effectivePageLimit = Math.min(pageLimit, remaining);
|
||||
const plan = await api.prepareQueryPaginationExecutionPlan({
|
||||
sql,
|
||||
sql: executableSql,
|
||||
queryBaseSql,
|
||||
databaseType: effectiveDbType,
|
||||
pagination: { limit: effectivePageLimit, offset, sessionId },
|
||||
@@ -8437,7 +8441,8 @@ export const useQueryStore = defineStore("query", () => {
|
||||
if (!effectiveDbType) return undefined;
|
||||
if (effectiveDbType === "mongodb") return undefined;
|
||||
const useAgentCursor = usesAgentCursorForQuery(conn?.db_type, conn?.driver_profile);
|
||||
const queryBaseSql = queryResultBaseSql(tab);
|
||||
const executableSql = effectiveDbType === "mysql" ? stripMysqlClientDisplayCommand(sql) : sql;
|
||||
const queryBaseSql = effectiveDbType === "mysql" ? stripMysqlClientDisplayCommand(queryResultBaseSql(tab)) : queryResultBaseSql(tab);
|
||||
const resultStatementIndex = tab.result.statement_index;
|
||||
const batchSql = tab.resultBaseSql ?? tab.lastExecutedSql ?? tab.sql;
|
||||
const batchStatements = effectiveDbType === "postgres" && tab.result.truncated === true && Number.isInteger(resultStatementIndex) && resultStatementIndex! > 0 ? splitSqlStatementRanges(batchSql, effectiveDbType) : [];
|
||||
@@ -8452,7 +8457,7 @@ export const useQueryStore = defineStore("query", () => {
|
||||
database: dataTabExecutionDatabase(conn, location.database, location.catalog),
|
||||
schema: location.schema,
|
||||
catalog: location.catalog,
|
||||
sql,
|
||||
sql: executableSql,
|
||||
queryBaseSql,
|
||||
setupSql,
|
||||
databaseType: effectiveDbType,
|
||||
@@ -8488,13 +8493,14 @@ export const useQueryStore = defineStore("query", () => {
|
||||
if (!effectiveDbType) return;
|
||||
|
||||
const exportId = uuid();
|
||||
const executableSql = effectiveDbType === "mysql" ? stripMysqlClientDisplayCommand(sql) : sql;
|
||||
const request: api.QueryResultExportRequest = {
|
||||
exportId,
|
||||
connectionId: tab.connectionId,
|
||||
database: tab.database,
|
||||
schema: tab.schema,
|
||||
sql,
|
||||
queryBaseSql: sql,
|
||||
sql: executableSql,
|
||||
queryBaseSql: executableSql,
|
||||
databaseType: effectiveDbType,
|
||||
useAgentCursor: usesAgentCursorForQuery(conn?.db_type, conn?.driver_profile),
|
||||
filePath,
|
||||
|
||||
Reference in new issue
Block a user