1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package org.synchronoss.cpo.parser;
22
23 import org.slf4j.*;
24
25 import java.io.StringReader;
26 import java.text.ParseException;
27 import java.util.*;
28
29
30
31
32
33 public class BoundExpressionParser implements ExpressionParser {
34
35 private static final Logger logger = LoggerFactory.getLogger(BoundExpressionParser.class);
36
37 private final static String COMPARE_CHARS = " =<>!()";
38 private final static String SEPARATOR_CHARS = " .,()\n";
39
40 private String expression;
41
42 public BoundExpressionParser() {
43 }
44
45 @Override
46 public String getExpression() {
47 return expression;
48 }
49
50 public void setExpression(String expression) {
51 this.expression = expression;
52 }
53
54
55
56
57
58
59 @Override
60 public int countArguments() {
61 return getArgumentIndexes().size();
62 }
63
64 private Collection<Integer> getArgumentIndexes() {
65 Collection<Integer> indexes = new ArrayList<>();
66
67 if (expression != null) {
68 StringReader reader = new StringReader(expression);
69
70 try {
71
72 int idx = 0;
73 int rc = -1;
74 boolean inDoubleQuotes = false;
75 boolean inSingleQuotes = false;
76
77 do {
78 rc = reader.read();
79 if (((char)rc) == '\'') {
80 inSingleQuotes = !inSingleQuotes;
81 } else if (((char)rc) == '"') {
82 inDoubleQuotes = !inDoubleQuotes;
83 } else if (!inSingleQuotes && !inDoubleQuotes && ((char)rc) == '?') {
84 indexes.add(idx);
85 }
86 idx++;
87 } while (rc != -1);
88 } catch (Exception e) {
89 logger.error("error counting bind markers");
90 }
91 }
92 return indexes;
93 }
94
95
96
97
98
99
100
101 @Override
102 public List<String> parse() throws ParseException {
103
104 if (expression == null)
105 throw new ParseException("The expression is null", -1);
106
107 if (logger.isDebugEnabled())
108 logger.debug("Expression: " + expression);
109
110
111 if (expression.length() < 1)
112 return null;
113
114
115 if (!expression.contains("?")) {
116 return null;
117 }
118
119
120 expression = expression.toUpperCase();
121
122 List<String> colList = new ArrayList<>();
123
124 if (expression.startsWith("INSERT")) {
125
126
127
128 int colParenStart = expression.indexOf("(");
129 if (colParenStart == -1)
130 throw new ParseException("Unable to locate starting parenthesis for the column names.", -1);
131
132 int colParenEnd = expression.indexOf(")", colParenStart);
133 if (colParenEnd == -1)
134 throw new ParseException("Unable to locate ending parenthesis for the column names.", -1);
135
136 int valParenStart = expression.indexOf("(", colParenEnd);
137 if (valParenStart == -1)
138 throw new ParseException("Unable to locate starting parenthesis for the column values.", -1);
139
140
141
142 int valParenEnd = expression.lastIndexOf(")");
143 if (valParenEnd == -1)
144 throw new ParseException("Unable to locate ending parenthesis for the column values.", -1);
145
146 String[] cols = expression.substring(colParenStart + 1, colParenEnd).split(",");
147 String[] vals = expression.substring(valParenStart + 1, valParenEnd).split(",");
148
149
150 if (cols == null || vals == null)
151 return null;
152
153 if (logger.isDebugEnabled()) {
154 logger.debug("Found cols: " + cols.length);
155 logger.debug("Found vals: " + vals.length);
156 }
157
158 if (cols.length != vals.length)
159 throw new ParseException("You seem to have " + cols.length + " columns, and " + vals.length + " values.\n\nThose numbers should be equal.", -1);
160
161
162 for (int i = 0; i < vals.length; i++) {
163 String val = vals[i];
164 if (val.trim().equals("?")) {
165
166 colList.add(cols[i].trim());
167 } else if (val.contains("?")) {
168
169 ExpressionParser qp = new BoundExpressionParser();
170 qp.setExpression(val);
171 colList.addAll(qp.parse());
172 }
173 }
174 } else {
175
176
177
178 int startIdx = 0;
179 Collection<Integer> indexes = getArgumentIndexes();
180 for (int qIdx : indexes) {
181 String chunk = expression.substring(startIdx, qIdx);
182
183 if (logger.isDebugEnabled())
184 logger.debug("Chunk [" + chunk + "]");
185
186 int idx = chunk.length() - 1;
187 int fieldStartIdx = -1;
188 int fieldEndIdx = -1;
189
190 boolean found = false;
191 boolean inFunction = false;
192 while (!found && (idx >= 0)) {
193 char c = chunk.charAt(idx);
194
195 if (fieldEndIdx == -1) {
196
197 if (!inFunction && c == '(') {
198 inFunction = true;
199 } else if (inFunction && COMPARE_CHARS.indexOf(c) != -1) {
200 inFunction = false;
201 }
202
203
204 if (!inFunction && COMPARE_CHARS.indexOf(c) == -1) {
205
206 fieldEndIdx = idx;
207 }
208 } else {
209
210 if (SEPARATOR_CHARS.indexOf(c) >= 0) {
211 fieldStartIdx = idx + 1;
212 found = true;
213 }
214 }
215 idx--;
216 }
217 if (found) {
218 String col = chunk.substring(fieldStartIdx, fieldEndIdx + 1);
219 colList.add(col);
220 }
221
222
223 startIdx = qIdx + 1;
224 }
225 }
226
227 return colList;
228 }
229 }