empayre/fleet
14
0
Fork 0
mirror of https://github.com/empayre/fleet.git synced 2024-11-06 08:55:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve accuracy of query platform compatibility check when WITH expressions used (#3731)

Browse Source
This commit is contained in:
gillespi314 2022-01-17 20:01:29 -06:00 committed by GitHub
parent 371c533bfc
commit dea23356de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
changes/issue-3590-query-compatibilty Normal file
View File

@ -0,0 +1 @@
* Improve accuracy of query platform compatibility check when `WITH` expressions used

25
frontend/utilities/sql_tools.js
View File

@ -44,17 +44,22 @@ export const listCompatiblePlatforms = (tablesList) => {
return compatiblePlatforms.length ? compatiblePlatforms : ["None"];
};
export const parseSqlTables = (sqlString) => {
const tablesList = [];
export const parseSqlTables = (sqlString, inludeCteTables = false) => {
let results = [];
// Tables defined via common table expression will be excluded from results by default
const cteTables = [];
const _callback = (node) => {
if (node) {
if (node.variant === "recursive") {
throw new Error(
"Invalid usage: `recursive` is not supported by `parseSqlTables`"
);
if (
(node.variant === "common" || node.variant === "recursive") &&
node.format === "table" &&
node.type === "expression"
) {
cteTables.push(node.target?.name);
} else if (node.variant === "table") {
tablesList.push(node.name);
results.push(node.name);
}
}
};
@ -63,7 +68,11 @@ export const parseSqlTables = (sqlString) => {
const sqlTree = sqliteParser(sqlString);
_visit(sqlTree, _callback);
return tablesList.length ? tablesList : ["No tables in query AST"];
if (cteTables.length) {
results = results.filter((t) => !cteTables.includes(t));
}
return results.length ? results : ["No tables in query AST"];
} catch (err) {
// console.log(`Invalid query syntax: ${err.message}\n\n${sqlString}`);