312 lines
10 KiB
JavaScript
312 lines
10 KiB
JavaScript
// import { GenericSQL, SqlParserVisitor } from 'dt-sql-parser'
|
||
import Meta, { connection, condition, columnType, cloneMeta } from '@/components/advancedSearch/meta/meta'
|
||
import { riskLevelMapping } from '@/utils/constants'
|
||
import _ from 'lodash'
|
||
// 补全语句，用于解析
|
||
const sqlPrev = 'select a from b where '
|
||
export default class SqlParser/* extends SqlParserVisitor */ {
|
||
constructor (init, columnList) {
|
||
// super()
|
||
this.tempMeta = null
|
||
this.originalSql = ''
|
||
this.metaList = []
|
||
this.columnList = columnList
|
||
// 原始数据
|
||
if (_.isArray(init)) {
|
||
this.metaList = init
|
||
} else {
|
||
this.originalSql = init
|
||
}
|
||
// 工具类实例
|
||
this.dtSqlParser = new GenericSQL()
|
||
}
|
||
|
||
// 返回的数组是错误信息，为空即校验通过
|
||
validate () {
|
||
return this.dtSqlParser.validate(sqlPrev + this.originalSql)
|
||
}
|
||
|
||
// 规范化原始语句
|
||
formatSql () {
|
||
// 先使用originalSql走一遍parse，获取metaList，再将metaList转为sql
|
||
const tree = this.dtSqlParser.parse(sqlPrev + this.originalSql)
|
||
this.visit(tree)
|
||
if (this.metaList.length === 0 && this.tempMeta.column.name) {
|
||
this.tempMeta.column.type = columnType.fullText
|
||
this.tempMeta.column.label = this.tempMeta.column.name
|
||
this.metaList.push(cloneMeta(this.tempMeta))
|
||
}
|
||
this.tempMeta = null
|
||
return {
|
||
metaList: this.metaList,
|
||
formatSql: this.parseMetaToSql(this.metaList)
|
||
}
|
||
}
|
||
|
||
formatMetaList () {
|
||
const tempMetaList = cloneMeta(this.metaList)
|
||
this.metaList = []
|
||
tempMetaList.forEach(meta => {
|
||
if (meta.meta === condition) {
|
||
if (meta.column.type === columnType.fullText) {
|
||
const m = cloneMeta(meta)
|
||
m.column.name = meta.column.name
|
||
m.column.label = m.column.name
|
||
this.metaList.push(m)
|
||
} else {
|
||
if (_.isArray(meta.value.value)) {
|
||
const tempArr = meta.value.value.map(v => {
|
||
return meta.column.type === columnType.string ? stringInQuot(v.trim()) : v.trim()
|
||
})
|
||
meta.value.value = `(${tempArr.join(',')})`
|
||
meta.value.label = meta.value.value
|
||
} else {
|
||
const isArray = ['IN', 'NOT IN'].indexOf(meta.operator.value) > -1
|
||
if (isArray) {
|
||
const leftBracketIndex = meta.value.value.indexOf('(')
|
||
const rightBracketIndex = meta.value.value.lastIndexOf(')')
|
||
let temp = meta.value.value.substring(0, rightBracketIndex)
|
||
temp = meta.value.value.substring(leftBracketIndex + 1, temp.length)
|
||
let tempArr = temp.split(',')
|
||
tempArr = tempArr.map(t => meta.column.type === columnType.string ? stringInQuot(t.trim()) : t.trim())
|
||
meta.value.value = `(${tempArr.join(',')})`
|
||
} else {
|
||
meta.value.value = meta.column.type === columnType.string ? stringInQuot(meta.value.value.trim()) : meta.value.value.trim()
|
||
}
|
||
meta.value.label = meta.value.value
|
||
}
|
||
this.metaList.push(meta)
|
||
}
|
||
} else {
|
||
this.metaList.push(meta)
|
||
}
|
||
})
|
||
return {
|
||
metaList: this.metaList,
|
||
formatSql: this.parseMetaToSql(this.metaList)
|
||
}
|
||
}
|
||
|
||
parseMetaToSql (metaList, isFullText) {
|
||
let sql = ''
|
||
metaList.forEach(meta => {
|
||
if (isFullText) {
|
||
if (meta.meta === condition) {
|
||
if (meta.column.type !== columnType.fullText) {
|
||
sql += `${meta.column.name}${handleOperatorSpace(meta.operator.value)}${meta.value.value} `
|
||
} else {
|
||
sql += "QUERY('"
|
||
this.columnList.forEach(column => {
|
||
sql += `${column.name}:${meta.column.name.replace(/'/g, '')} `
|
||
})
|
||
sql += "') "
|
||
}
|
||
}
|
||
} else {
|
||
if (meta.meta === condition) {
|
||
sql += (meta.column.name)
|
||
if (meta.column.type !== columnType.fullText) {
|
||
sql += `${handleOperatorSpace(meta.operator.value)}${meta.value.value} `
|
||
} else {
|
||
sql += ' '
|
||
}
|
||
}
|
||
}
|
||
if (meta.meta === connection) {
|
||
sql += `${meta.value} `
|
||
}
|
||
})
|
||
return sql
|
||
}
|
||
|
||
buildMeta (type, value) {
|
||
switch (type) {
|
||
case 'expression': {
|
||
// 如果tempMeta是空或已是完整的condition，新建meta
|
||
if (!this.tempMeta || this.tempMeta.isCompleteCondition()) {
|
||
this.tempMeta = new Meta(condition)
|
||
}
|
||
// 如果column.name为空，则参数值是column，否则是value
|
||
if (!this.tempMeta.column.name) {
|
||
// 在columnList中的，按columnList；不在则为全文检索
|
||
const column = this.columnList.find(column => {
|
||
return column.name === value
|
||
})
|
||
this.tempMeta.column.name = value
|
||
this.tempMeta.column.label = column ? column.label : value
|
||
this.tempMeta.column.type = column ? column.type : columnType.fullText
|
||
if (this.tempMeta.column.type === columnType.fullText) {
|
||
this.tempMeta.value.show = false
|
||
this.tempMeta.operator.show = false
|
||
this.metaList.push(cloneMeta(this.tempMeta))
|
||
}
|
||
} else {
|
||
if (_.isArray(value)) {
|
||
const tempArr = value.map(v => {
|
||
return this.tempMeta.column.type === columnType.string ? stringInQuot(v.trim()) : v.trim()
|
||
})
|
||
this.tempMeta.value.value = `(${tempArr.join(',')})`
|
||
this.tempMeta.value.label = this.tempMeta.value.value
|
||
} else {
|
||
const isArray = ['IN', 'NOT IN'].indexOf(this.tempMeta.operator.value) > -1
|
||
if (isArray) {
|
||
const leftBracketIndex = value.indexOf('(')
|
||
const rightBracketIndex = value.lastIndexOf(')')
|
||
let temp = value.substring(0, rightBracketIndex)
|
||
temp = value.substring(leftBracketIndex + 1, temp.length)
|
||
let tempArr = temp.split(',')
|
||
tempArr = tempArr.map(t => this.tempMeta.column.type === columnType.string ? stringInQuot(t.trim()) : t.trim())
|
||
this.tempMeta.value.value = `(${tempArr.join(',')})`
|
||
} else {
|
||
this.tempMeta.value.value = this.tempMeta.column.type === columnType.string ? stringInQuot(value.trim()) : value.trim()
|
||
}
|
||
this.tempMeta.value.label = this.tempMeta.value.value
|
||
}
|
||
this.tempMeta.value.show = true
|
||
this.metaList.push(cloneMeta(this.tempMeta))
|
||
}
|
||
break
|
||
}
|
||
case 'operator': {
|
||
this.tempMeta.operator.value = value
|
||
this.tempMeta.operator.label = value
|
||
this.tempMeta.operator.show = true
|
||
break
|
||
}
|
||
// (not)in和(not)like特殊处理
|
||
case 'like':
|
||
case 'in': {
|
||
const { column, operator, handleValue } = handleInOrLike(value, type)
|
||
this.buildMeta('expression', column)
|
||
this.buildMeta('operator', operator)
|
||
this.buildMeta('expression', handleValue)
|
||
break
|
||
}
|
||
case 'connection': {
|
||
// tempMeta的value为空，则上个条件是全文搜索
|
||
if (!this.tempMeta.column.type && !this.tempMeta.value.value) {
|
||
this.tempMeta.column.type = columnType.fullText
|
||
this.tempMeta.column.label = this.tempMeta.column.name
|
||
this.metaList.push(cloneMeta(this.tempMeta))
|
||
this.tempMeta = null
|
||
}
|
||
const meta = new Meta(connection)
|
||
meta.value = value
|
||
this.metaList.push(meta)
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
// 字段或值
|
||
visitExpressionAtomPredicate (ctx) {
|
||
const constant = ctx.getText().toLowerCase()
|
||
this.buildMeta('expression', constant)
|
||
}
|
||
|
||
// 操作符
|
||
visitComparisonOperator (ctx) {
|
||
const comparisonOperator = ctx.getText().toLowerCase()
|
||
this.buildMeta('operator', comparisonOperator)
|
||
}
|
||
|
||
// 连接符
|
||
visitLogicalOperator (ctx) {
|
||
const logicalOperator = ctx.getText().toUpperCase()
|
||
this.buildMeta('connection', logicalOperator)
|
||
}
|
||
|
||
// in语句
|
||
visitInPredicate (ctx) {
|
||
const inPredicate = ctx.getText().toLowerCase()
|
||
this.buildMeta('in', inPredicate)
|
||
}
|
||
|
||
// like语句
|
||
visitLikePredicate (ctx) {
|
||
const likePredicate = ctx.getText().toLowerCase()
|
||
this.buildMeta('like', likePredicate)
|
||
}
|
||
}
|
||
|
||
function handleType (value) {
|
||
if (_.isInteger(value)) {
|
||
return columnType.long
|
||
} else if (_.isNumber(value)) {
|
||
return columnType.number
|
||
} else if (_.isString(value)) {
|
||
return columnType.string
|
||
} else if (_.isArray(value)) {
|
||
const arr = value.split(',')
|
||
const hasString = arr.some(v => {
|
||
return handleType(v) === columnType.string
|
||
})
|
||
return hasString ? columnType.string : columnType.number
|
||
}/* else if (isStringInQuot(value)) {
|
||
return columnType.string
|
||
} */
|
||
return null
|
||
}
|
||
|
||
// 使用单引号包裹
|
||
export function stringInQuot (value) {
|
||
if (value.indexOf("'") > -1) {
|
||
value = value.split("'").join("''") // 如xi'an，多添加一个单引号，处理为xi''an
|
||
}
|
||
|
||
const match = `${value}`.match(/^'.+?'$/)
|
||
return match ? value : `'${value}'`
|
||
}
|
||
// IN和LIKE前后加空格
|
||
export function handleOperatorSpace (operator) {
|
||
return ['IN', 'NOT IN', 'LIKE', 'NOT LIKE'].indexOf(operator) > -1 ? ` ${operator} ` : operator
|
||
}
|
||
// in、like拆分方法；解析后内容挤一堆，如：a in ('b')解析结果是ain('b')，需要拆分
|
||
function handleInOrLike (value, type) {
|
||
let sep
|
||
if (type === 'in') {
|
||
sep = `${type}(`
|
||
} else if (type === 'like') {
|
||
sep = `${type}'`
|
||
}
|
||
const notSep = `not${sep}`
|
||
let arr
|
||
let operator
|
||
if (value.split(notSep).length === 2) {
|
||
arr = value.split(notSep)
|
||
operator = `NOT ${type.toUpperCase()}`
|
||
} else {
|
||
arr = value.split(sep)
|
||
operator = type.toUpperCase()
|
||
}
|
||
const columnName = arr[0]
|
||
|
||
let v
|
||
if (type === 'in') {
|
||
v = `(${arr[1]}`
|
||
} else if (type === 'like') {
|
||
v = `'${arr[1]}`
|
||
}
|
||
return {
|
||
column: columnName,
|
||
operator,
|
||
handleValue: v
|
||
}
|
||
}
|
||
|
||
function valueHandle (value, columnName) {
|
||
if (value === "'unknown'") {
|
||
return "''"
|
||
} else {
|
||
if (columnName) {
|
||
if (columnName === 'app_risk') {
|
||
const m = riskLevelMapping.find(mapping => {
|
||
return `'${mapping.name}'`.toLowerCase() === value.toLowerCase() || `'${mapping.value}'` === value
|
||
})
|
||
return (m && m.value) || value
|
||
}
|
||
}
|
||
return value
|
||
}
|
||
}
|