dennii/dbgate
1
0
Fork 0
mirror of https://github.com/DeNNiiInc/dbgate.git synced 2026-04-20 20:35:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

split query rich result refactor

Browse Source
This commit is contained in:
Jan Prochazka
2021-12-12 11:48:20 +01:00
parent c1e7314df1
commit 6930a2543c
3 changed files with 84 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
packages/query-splitter/src/splitQuery.ts
View File

@@ -30,14 +30,16 @@ export interface SplitLineContext extends SplitStreamContext {
// semicolonKeyTokenRegex: RegExp;
}
export interface SplitPositionDefinition {
position: number;
line: number;
column: number;
}
export interface SplitResultItemRich {
text: string;
startPosition: number;
endPosition: number;
startLine: number;
startColumn: number;
endLine: number;
endColumn: number;
start: SplitPositionDefinition;
end: SplitPositionDefinition;
}
export type SplitResultItem = string | SplitResultItemRich;
@@ -212,13 +214,17 @@ function pushQuery(context: SplitLineContext) {
trimPositions(sql, {
text: trimmed,
startPosition: context.commandStartPosition,
startLine: context.commandStartLine,
startColumn: context.commandStartColumn,
start: {
position: context.commandStartPosition,
line: context.commandStartLine,
column: context.commandStartColumn,
},
endPosition: context.streamPosition,
endLine: context.line,
endColumn: context.column,
end: {
position: context.streamPosition,
line: context.line,
column: context.column,
},
})
);
} else {
@@ -229,15 +235,20 @@ function pushQuery(context: SplitLineContext) {
function trimPositions(full: string, positions: SplitResultItemRich): SplitResultItemRich {
const startIndex = full.indexOf(positions.text);
const res = { ...positions };
const res = {
...positions,
start: {
...positions.start,
},
};
for (let i = 0; i < startIndex; i += 1) {
if (full[i] == '\n') {
res.startPosition += 1;
res.startLine += 1;
res.startColumn = 0;
res.start.position += 1;
res.start.line += 1;
res.start.column = 0;
} else {
res.startPosition += 1;
res.startColumn += 1;
res.start.position += 1;
res.start.column += 1;
}
}
@@ -321,12 +332,18 @@ export function finishSplitStream(context: SplitStreamContext) {
context.pushOutput(
trimPositions(context.commandPart, {
text: trimmed,
startPosition: context.commandStartPosition,
startLine: context.commandStartLine,
startColumn: context.commandStartColumn,
endPosition: context.streamPosition,
endLine: context.line,
endColumn: context.column,
start: {
position: context.commandStartPosition,
line: context.commandStartLine,
column: context.commandStartColumn,
},
end: {
position: context.streamPosition,
line: context.line,
column: context.column,
},
})
);
} else {

64
packages/query-splitter/src/splitter.test.ts
View File

@@ -100,24 +100,32 @@ test('count lines', () => {
expect.objectContaining({
text: 'SELECT * FROM `table1`',
startPosition: 0,
startLine: 0,
startColumn: 0,
start: expect.objectContaining({
position: 0,
line: 0,
column: 0,
}),
endPosition: 22,
endLine: 0,
endColumn: 22,
end: expect.objectContaining({
position: 22,
line: 0,
column: 22,
}),
}),
expect.objectContaining({
text: 'SELECT * FROM `table2`',
startPosition: 24,
startLine: 1,
startColumn: 0,
start: expect.objectContaining({
position: 24,
line: 1,
column: 0,
}),
endPosition: 46,
endLine: 1,
endColumn: 22,
end: expect.objectContaining({
position: 46,
line: 1,
column: 22,
}),
}),
])
);
@@ -133,24 +141,32 @@ test('count lines with flush', () => {
expect.objectContaining({
text: 'SELECT * FROM `table1`',
startPosition: 0,
startLine: 0,
startColumn: 0,
start: expect.objectContaining({
position: 0,
line: 0,
column: 0,
}),
endPosition: 22,
endLine: 0,
endColumn: 22,
end: expect.objectContaining({
position: 22,
line: 0,
column: 22,
}),
}),
expect.objectContaining({
text: 'SELECT * FROM `table2`',
startPosition: 24,
startLine: 1,
startColumn: 0,
start: expect.objectContaining({
position: 24,
line: 1,
column: 0,
}),
endPosition: 46,
endLine: 1,
endColumn: 22,
end: expect.objectContaining({
position: 46,
line: 1,
column: 22,
}),
}),
])
);