feat: 搜索框（部分内容）
This commit is contained in:
chenjinsong
257
src/utils/sql-parser.js
Normal file
257
src/utils/sql-parser.js
Normal file
@@ -0,0 +1,257 @@
|
||||
import { GenericSQL, SqlParserVisitor } from 'dt-sql-parser'
|
||||
import Meta, { connection, condition, columnType } from '@/components/advancedSearch/meta/meta'
|
||||
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.name = stringInQuot(this.tempMeta.column.name)
|
||||
this.tempMeta.column.label = this.tempMeta.column.name
|
||||
this.metaList.push(_.cloneDeep(this.tempMeta))
|
||||
}
|
||||
this.tempMeta = null
|
||||
return {
|
||||
metaList: this.metaList,
|
||||
formatSql: this.parseMetaToSql(this.metaList)
|
||||
}
|
||||
}
|
||||
|
||||
formatMetaList () {
|
||||
const tempMetaList = _.cloneDeep(this.metaList)
|
||||
this.metaList = []
|
||||
tempMetaList.forEach(meta => {
|
||||
if (meta.meta === condition) {
|
||||
if (meta.column.type === columnType.fullText) {
|
||||
this.metaList.push({ ...meta, name: stringInQuot(meta.name), label: stringInQuot(meta.name) })
|
||||
} else if (meta.column.type === columnType.string) {
|
||||
if (_.isArray(meta.value.value)) {
|
||||
meta.value.value = meta.value.value.map(v => {
|
||||
return stringInQuot(v)
|
||||
})
|
||||
meta.value.label = `(${meta.value.value.join(',')})`
|
||||
} else {
|
||||
meta.value.value = stringInQuot(meta.value.value)
|
||||
meta.value.label = stringInQuot(meta.value.value)
|
||||
}
|
||||
this.metaList.push(meta)
|
||||
} else {
|
||||
if (_.isArray(meta.value.value)) {
|
||||
meta.value.label = `(${meta.value.value.join(',')})`
|
||||
} else {
|
||||
meta.value.label = meta.value.value
|
||||
}
|
||||
this.metaList.push(meta)
|
||||
}
|
||||
}
|
||||
})
|
||||
return {
|
||||
metaList: this.metaList,
|
||||
formatSql: this.parseMetaToSql(this.metaList)
|
||||
}
|
||||
}
|
||||
|
||||
parseMetaToSql (metaList) {
|
||||
let sql = ''
|
||||
metaList.forEach(meta => {
|
||||
if (meta.meta === condition) {
|
||||
sql += meta.column.name
|
||||
if (meta.column.type !== columnType.fullText) {
|
||||
sql += `${meta.operator.value}${meta.value.value} `
|
||||
}
|
||||
}
|
||||
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；不在则先设为空串，由之后value值来自动判断
|
||||
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 : ''
|
||||
} else {
|
||||
// 若column的type为空，则根据value自动判断赋值
|
||||
if (!this.tempMeta.column.type) {
|
||||
this.tempMeta.column.type = handleType(value)
|
||||
}
|
||||
if (this.tempMeta.column.type === columnType.string) {
|
||||
if (_.isArray(value)) {
|
||||
this.tempMeta.value.value = value.map(v => {
|
||||
return stringInQuot(v)
|
||||
})
|
||||
this.tempMeta.value.label = `(${this.tempMeta.value.value.join(',')})`
|
||||
} else {
|
||||
this.tempMeta.value.value = stringInQuot(value)
|
||||
this.tempMeta.value.label = stringInQuot(value)
|
||||
}
|
||||
} else {
|
||||
this.tempMeta.value.value = value
|
||||
if (_.isArray(value)) {
|
||||
this.tempMeta.value.label = `(${this.tempMeta.value.value.join(',')})`
|
||||
} else {
|
||||
this.tempMeta.value.label = value
|
||||
}
|
||||
}
|
||||
this.tempMeta.value.show = true
|
||||
this.metaList.push(_.cloneDeep(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)
|
||||
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.name = stringInQuot(this.tempMeta.column.name)
|
||||
this.tempMeta.column.label = this.tempMeta.column.name
|
||||
this.metaList.push(_.cloneDeep(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) {
|
||||
const match = value.match(/^'.+?'$/)
|
||||
return match ? value : `'${value}'`
|
||||
}
|
||||
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 === 3) {
|
||||
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[2].substring(0, arr[2].length - 2).split(',')
|
||||
} else if (type === 'like') {
|
||||
v = `'${arr[2]}'`
|
||||
}
|
||||
return {
|
||||
column: columnName,
|
||||
operator,
|
||||
handleValue: v
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user