lava-oushudb-dt-sql-parser/lib/filter.js

const replaceStrFormIndexArr = require("../utils").replaceStrFormIndexArr;

/**
 * 过滤--注释
 * @param {String} sql 
 */
function filterComments(sql) {
    let tmpArr = [];
    const comments = [];

    for (let i = 0; i < sql.length; i++) {
        let char = sql[i];

        //读取字符
        if (char == "'" || char == "\"" || char == "-" || char == "\n" || char == "/" || char == "*") {
            //推入数组
            tmpArr.push({
                index: i,
                char: char
            });
        }
        //校验数组是否有匹配语法
        if (tmpArr.length < 2) {
            if (tmpArr[0] && (tmpArr[0].char == "\n" || tmpArr[0].char == "*")) {
                tmpArr = [];
            }
            continue;
        }

        let firstChar = tmpArr[0];
        let lastChar = tmpArr[tmpArr.length - 1];

        if (firstChar.char == "'" || firstChar.char == "\"") {
            //引号匹配，则清空
            if (lastChar.char == firstChar.char) {
                tmpArr = [];
                continue;
            }
        } else if (firstChar.char == "/") {
            if (tmpArr[1].char != "*") {
                tmpArr = [];
            } else if (lastChar.char == "/") {
                if (tmpArr.length > 3 && tmpArr[tmpArr.length - 2].char == "*") {
                    comments.push({
                        begin: firstChar.index,
                        end: lastChar.index
                    })
                    tmpArr = [];
                }   
            }
            continue;
        } else if (firstChar.char == "-") {
            //假如第一个是横线，则开始校验注释规则

            //判断是否为两个注释符号，不是，则清空
            if (tmpArr[1].char != "-") {
                tmpArr = [];
                continue;
            }
            //为注释作用域，遇到换行符，则结束注释
            else if (lastChar.char == "\n") {
                comments.push({
                    begin: firstChar.index,
                    end: lastChar.index
                })
                tmpArr = [];
                continue;
            }
            //解析结束
            else if (i == sql.length - 1) {
                comments.push({
                    begin: firstChar.index,
                    end: i
                })
                continue;
            }
        } else {
            tmpArr = [];
        }
    }

    sql = replaceStrFormIndexArr(sql, '', comments)

    return sql;
}

/**
 * 清除注释和前后空格
 * @param {String} sql 
 */
function cleanSql(sql) {
    return filterComments(sql).trim();
}

/**
 * 分割sql
 * @param {String} sqlText 
 */
function splitSql(sqlText) {
    if (!sqlText) {
        return sqlText;
    }
    sqlText = sqlText.trim();
    if (!endsWith(sqlText, ';')) {
        sqlText += ';';
    }

    let results = [];
    let index = 0;
    let tmpChar = null;
    for (let i = 0; i < sqlText.length; i++) {
        let char = sqlText[i];

        if (char == "'" || char == '"') {
            if (tmpChar == char) {
                tmpChar = null;
            } else {
                tmpChar = char;
            }
        } else if (char == ';') {
            if (tmpChar == null) {
                results.push(sqlText.substring(index, i));
                index = i + 1;
            }
        }
    }

    return results;
}

exports.filterComments = filterComments;
exports.cleanSql = cleanSql;
exports.splitSql = splitSql;