/
RegexReplaceArg.go
117 lines (106 loc) · 2.41 KB
/
RegexReplaceArg.go
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
package ast
import (
"bytes"
"errors"
"fmt"
"regexp"
"strings"
)
var defaultArgRegex, _ = regexp.Compile("\\$\\{[^}]*\\}")
var sqlArgRegex, _ = regexp.Compile("\\#\\{[^}]*\\}")
//执行替换操作
func Replace(startChar string, findStrs []string, data string, typeConvert SqlArgTypeConvert, arg map[string]interface{}, engine ExpressionEngine) (string, error) {
for _, findStr := range findStrs {
var evalData interface{}
//find param arg
var argValue = arg[findStr]
if argValue != nil {
evalData = argValue
} else {
//exec lexer
var err error
evalData, err = engine.LexerAndEval(findStr, arg)
if err != nil {
return "", errors.New(engine.Name() + ":" + err.Error())
}
}
var resultStr string
if typeConvert != nil {
resultStr = typeConvert.Convert(evalData)
} else {
resultStr = fmt.Sprint(evalData)
}
data = strings.Replace(data, startChar+findStr+"}", resultStr, -1)
}
arg = nil
typeConvert = nil
return data, nil
}
//find like #{*} value *
func FindAllExpressConvertString(str string) []string {
var finds = map[string]int{}
var item []byte
var lastIndex = -1
var startIndex = -1
var strBytes = []byte(str)
for index, v := range strBytes {
if v == 35 {
lastIndex = index
}
if v == 123 && lastIndex == (index-1) {
startIndex = index + 1
}
if v == 125 && startIndex != -1 {
item = strBytes[startIndex:index]
//去掉逗号之后的部分
if bytes.Contains(item, []byte(",")) {
item = bytes.Split(item, []byte(","))[0]
}
finds[string(item)] = 1
item = nil
startIndex = -1
lastIndex = -1
}
}
item = nil
strBytes = nil
var strs = []string{}
for k, _ := range finds {
strs = append(strs, k)
}
return strs
}
//find like ${*} value *
func FindAllExpressString(str string) []string {
var finds = map[string]int{}
var item []byte
var lastIndex = -1
var startIndex = -1
var strBytes = []byte(str)
for index, v := range str {
if v == 36 {
lastIndex = index
}
if v == 123 && lastIndex == (index-1) {
startIndex = index + 1
}
if v == 125 && startIndex != -1 {
item = strBytes[startIndex:index]
//去掉逗号之后的部分
if bytes.Contains(item, []byte(",")) {
item = bytes.Split(item, []byte(","))[0]
}
finds[string(item)] = 1
item = nil
startIndex = -1
lastIndex = -1
}
}
item = nil
strBytes = nil
var strs = []string{}
for k, _ := range finds {
strs = append(strs, k)
}
return strs
}