xiaoxue_admin
/
dbx
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(sql): preserve leading tenant routing hints 

Closes #5332
This commit is contained in:
t8y2 2026-08-04 23:17:35 +08:00
parent bc60339806
commit 4aa36cb893
No known key found for this signature in database
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/desktop/src/lib/__tests__/sql/sqlStatementRanges.spec.ts vendored
View File

@ -1095,6 +1095,15 @@ describe("buildExecutionCandidates", () => {
expect(candidates[0].sql).toBe(hintedSql);
});
it("preserves leading tenant routing hints in current statement candidates", () => {
const hintedSql = "/*@global:true*/\nSELECT * FROM tenant_table";
const sql = `SELECT 1;\n${hintedSql};`;
const candidates = buildExecutionCandidates(sql, indexOf(sql, "tenant_table"), "mysql");
expect(candidates[0].sql).toBe(hintedSql);
expect(splitSqlStatementRanges("/*@global:true*/", "mysql")).toEqual([]);
});
it("uses the cursor statement for the first candidate when there is no selection", () => {
const sql = "SELECT *\nFROM users\nWHERE active = 1";
const candidates = buildExecutionCandidates(sql, indexOf(sql, "users"));

3
apps/desktop/src/lib/sql/sqlStatementRanges.ts
View File

@ -444,7 +444,8 @@ export function splitSqlStatementRanges(sql: string, databaseType?: DatabaseType
}
// Block comments consume until the closing */.
if (ch === "/" && next === "*") {
if (statementStart === -1 && pendingHintStart === -1 && sql[i + 2] === "+") pendingHintStart = i;
const hintMarker = sql[i + 2];
if (statementStart === -1 && pendingHintStart === -1 && (hintMarker === "+" || hintMarker === "@")) pendingHintStart = i;
const close = sql.indexOf("*/", i + 2);
i = close === -1 ? len : close + 2;
continue;