forked from taozhi8833998/node-sql-parser
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathexpr.js
More file actions
131 lines (118 loc) · 4.4 KB
/
expr.js
File metadata and controls
131 lines (118 loc) · 4.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
import { literalToSQL, toUpper, connector, dataTypeToSQL, hasVal } from './util'
import { alterExprToSQL } from './alter'
import { aggrToSQL } from './aggregation'
import { assignToSQL } from './assign'
import { binaryToSQL } from './binary'
import { caseToSQL } from './case'
import { columnDefinitionToSQL, columnRefToSQL, fullTextSearchToSQL } from './column'
import { anyValueFuncToSQL, castToSQL, extractFunToSQL, flattenFunToSQL, funcToSQL, jsonObjectArgToSQL, lambdaToSQL, tablefuncFunToSQL } from './func'
import { intervalToSQL } from './interval'
import { jsonExprToSQL, jsonVisitorExprToSQL } from './json'
import { selectToSQL } from './select'
import { showToSQL } from './show'
import { arrayStructExprToSQL } from './array-struct'
import { tablesToSQL, unnestToSQL } from './tables'
import { unionToSQL } from './union'
import { namedWindowExprListToSQL, windowFuncToSQL } from './window'
const exprToSQLConvertFn = {
alter : alterExprToSQL,
aggr_func : aggrToSQL,
any_value : anyValueFuncToSQL,
window_func : windowFuncToSQL,
'array' : arrayStructExprToSQL,
assign : assignToSQL,
binary_expr : binaryToSQL,
case : caseToSQL,
cast : castToSQL,
column_ref : columnRefToSQL,
column_definition : columnDefinitionToSQL,
datatype : dataTypeToSQL,
extract : extractFunToSQL,
flatten : flattenFunToSQL,
fulltext_search : fullTextSearchToSQL,
function : funcToSQL,
lambda : lambdaToSQL,
insert : unionToSQL,
interval : intervalToSQL,
json : jsonExprToSQL,
json_object_arg : jsonObjectArgToSQL,
json_visitor : jsonVisitorExprToSQL,
show : showToSQL,
struct : arrayStructExprToSQL,
tablefunc : tablefuncFunToSQL,
tables : tablesToSQL,
unnest : unnestToSQL,
'window' : namedWindowExprListToSQL,
}
function varToSQL(expr) {
const { prefix = '@', name, members, keyword, quoted, suffix } = expr
const val = []
if (keyword) val.push(keyword)
const varName = members && members.length > 0 ? `${name}.${members.join('.')}` : name
let result = `${prefix || ''}${varName}`
if (suffix) result += suffix
val.push(result)
return [quoted, val.join(' '), quoted].filter(hasVal).join('')
}
exprToSQLConvertFn.var = varToSQL
function exprToSQL(exprOrigin) {
if (!exprOrigin) return
const expr = exprOrigin
if (exprOrigin.ast) {
const { ast } = expr
Reflect.deleteProperty(expr, ast)
for (const key of Object.keys(ast)) {
expr[key] = ast[key]
}
}
return exprToSQLConvertFn[expr.type] ? exprToSQLConvertFn[expr.type](expr) : literalToSQL(expr)
}
function unaryToSQL(unarExpr) {
const { operator, parentheses, expr } = unarExpr
const space = (operator === '-' || operator === '+' || operator === '~' || operator === '!') ? '' : ' '
const str = `${operator}${space}${exprToSQL(expr)}`
return parentheses ? `(${str})` : str
}
function getExprListSQL(exprList) {
if (!exprList) return []
return exprList.map(exprToSQL)
}
exprToSQLConvertFn.expr_list = expr => {
const str = getExprListSQL(expr.value)
return expr.parentheses ? `(${str.join(', ')})` : str
}
exprToSQLConvertFn.select = expr => {
const str = typeof expr._next === 'object' ? unionToSQL(expr) : selectToSQL(expr)
return expr.parentheses ? `(${str})` : str
}
exprToSQLConvertFn.unary_expr = unaryToSQL
function mapObjectToSQL(mapExpr) {
const { keyword, expr } = mapExpr
const exprStr = expr.map(exprItem => [literalToSQL(exprItem.key), literalToSQL(exprItem.value)].join(', ')).join(', ')
return [toUpper(keyword), `[${exprStr}]`].join('')
}
exprToSQLConvertFn.map_object = mapObjectToSQL
function orderOrPartitionByToSQL(expr, prefix) {
if (!Array.isArray(expr)) return ''
let expressions = []
const upperPrefix = toUpper(prefix)
switch (upperPrefix) {
case 'ORDER BY':
expressions = expr.map(info => [exprToSQL(info.expr), info.type || 'ASC', toUpper(info.nulls)].filter(hasVal).join(' '))
break
case 'PARTITION BY':
expressions = expr.map(info => exprToSQL(info.expr))
break
default:
expressions = expr.map(info => exprToSQL(info.expr))
break
}
return connector(upperPrefix, expressions.join(', '))
}
export {
exprToSQLConvertFn,
exprToSQL,
getExprListSQL,
varToSQL,
orderOrPartitionByToSQL,
}