oushu/lava-oushudb-dt-sql-parser
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: unify the parser generator to antlr4

Browse Source
This commit is contained in:
xiaowei
2020-08-28 13:29:11 +08:00
parent 534e42ee63
commit c069f9f714
245 changed files with 414915 additions and 155437 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
src/utils/index.ts
View File

@ -1,14 +1,13 @@
function replaceStrFormIndexArr(str, replaceStr, indexArr) {
let arr = [];
let result = "";
let result = '';
let index = 0;
if (!indexArr || indexArr.length < 1) {
return str;
}
for (let i = 0; i < indexArr.length; i++) {
let indexItem = indexArr[i];
let begin = indexItem.begin;
const indexItem = indexArr[i];
const begin = indexItem.begin;
result = result + str.substring(index, begin) + replaceStr;
index = indexItem.end + 1;
@ -20,7 +19,7 @@ function replaceStrFormIndexArr(str, replaceStr, indexArr) {
return result;
}
function splitSql (sql: string) {
function splitSql(sql: string) {
let haveEnd = true;
if (!sql.endsWith(';')) {
sql += ';';
@ -31,7 +30,7 @@ function splitSql (sql: string) {
queue: string;
sqls: number[];
}
function pushSql (parser: splitParser, sql: string) {
function pushSql(parser: splitParser, sql: string) {
if (!haveEnd && parser.index == sql.length - 1) {
parser.sqls.push(parser.index - 1);
parser.queue = '';
@ -41,11 +40,11 @@ function splitSql (sql: string) {
}
}
// 处理引号
function quoteToken (parser: splitParser, sql: string): string {
let queue = parser.queue;
let endsWith = queue[queue.length - 1];
function quoteToken(parser: splitParser, sql: string): string {
const queue = parser.queue;
const endsWith = queue[queue.length - 1];
if (endsWith == '\'' || endsWith == '"') {
let nextToken = sql.indexOf(endsWith, parser.index + 1);
const nextToken = sql.indexOf(endsWith, parser.index + 1);
if (nextToken != -1) {
parser.index = nextToken;
parser.queue = '';
@ -57,10 +56,10 @@ function splitSql (sql: string) {
}
}
// 处理单行注释
function singleLineCommentToken (parser: splitParser, sql: string): string {
function singleLineCommentToken(parser: splitParser, sql: string): string {
let queue = parser.queue;
if (queue.endsWith('--')) {
let nextToken = sql.indexOf('\n', parser.index + 1);
const nextToken = sql.indexOf('\n', parser.index + 1);
if (nextToken != -1) {
parser.index = nextToken;
queue = '';
@ -72,10 +71,12 @@ function splitSql (sql: string) {
}
}
// 处理多行注释
function multipleLineCommentToken (parser: splitParser, sql: string): string {
let queue = parser.queue;
function multipleLineCommentToken(
parser: splitParser, sql: string,
): string {
const queue = parser.queue;
if (queue.endsWith('/*')) {
let nextToken = sql.indexOf('*/', parser.index + 1);
const nextToken = sql.indexOf('*/', parser.index + 1);
if (nextToken != -1) {
parser.index = nextToken + 1;
parser.queue = '';
@ -87,34 +88,38 @@ function splitSql (sql: string) {
return null;
}
}
function splitToken (parser: splitParser, sql: string): string {
let queue = parser.queue;
function splitToken(parser: splitParser, sql: string): string {
const queue = parser.queue;
if (queue.endsWith(';')) {
pushSql(parser, sql)
pushSql(parser, sql);
} else {
return null;
}
}
let parser: splitParser = {
const parser: splitParser = {
index: 0,
queue: '',
sqls: []
sqls: [],
};
for (parser.index = 0; parser.index < sql.length; parser.index++) {
let char = sql[parser.index];
const char = sql[parser.index];
parser.queue += char;
let tokenFuncs = [quoteToken, singleLineCommentToken, multipleLineCommentToken, splitToken];
const tokenFuncs = [
quoteToken,
singleLineCommentToken,
multipleLineCommentToken,
splitToken,
];
for (let i = 0; i < tokenFuncs.length; i++) {
tokenFuncs[i](parser, sql);
}
if (parser.index == sql.length - 1 && parser.queue) {
pushSql(parser, sql)
pushSql(parser, sql);
}
}
return parser.sqls;
}
export {
replaceStrFormIndexArr,
splitSql
}
splitSql,
};