|
50 | 50 |
|
51 | 51 | public class InsertSqlParser implements IParser { |
52 | 52 |
|
| 53 | + // 用来标识当前解析节点的上一层节点是否为 insert 节点 |
| 54 | + private static Boolean parentIsInsert = false; |
| 55 | + |
53 | 56 | @Override |
54 | 57 | public boolean verify(String sql) { |
55 | 58 | return StringUtils.isNotBlank(sql) && sql.trim().toLowerCase().startsWith("insert"); |
@@ -79,24 +82,19 @@ private static void parseNode(SqlNode sqlNode, SqlParseResult sqlParseResult){ |
79 | 82 | SqlNode sqlTarget = ((SqlInsert)sqlNode).getTargetTable(); |
80 | 83 | SqlNode sqlSource = ((SqlInsert)sqlNode).getSource(); |
81 | 84 | sqlParseResult.addTargetTable(sqlTarget.toString()); |
| 85 | + parentIsInsert = true; |
82 | 86 | parseNode(sqlSource, sqlParseResult); |
83 | 87 | break; |
84 | 88 | case SELECT: |
85 | 89 | SqlSelect sqlSelect = (SqlSelect) sqlNode; |
86 | | - SqlNodeList selectList = sqlSelect.getSelectList(); |
87 | | - SqlNodeList sqlNodes = new SqlNodeList(selectList.getParserPosition()); |
88 | | - for (int index = 0; index < selectList.size(); index++) { |
89 | | - if (selectList.get(index).getKind().equals(SqlKind.AS)) { |
90 | | - sqlNodes.add(selectList.get(index)); |
91 | | - continue; |
92 | | - } |
93 | | - sqlNodes.add(transformToSqlBasicCall(selectList.get(index))); |
| 90 | + if (parentIsInsert) { |
| 91 | + rebuildSelectNode(sqlSelect.getSelectList(), sqlSelect); |
94 | 92 | } |
95 | | - sqlSelect.setSelectList(sqlNodes); |
96 | 93 | SqlNode sqlFrom = ((SqlSelect) sqlNode).getFrom(); |
97 | 94 | if (sqlFrom.getKind() == IDENTIFIER) { |
98 | 95 | sqlParseResult.addSourceTable(sqlFrom.toString()); |
99 | 96 | } else { |
| 97 | + parentIsInsert = false; |
100 | 98 | parseNode(sqlFrom, sqlParseResult); |
101 | 99 | } |
102 | 100 | break; |
@@ -153,8 +151,30 @@ private static void parseNode(SqlNode sqlNode, SqlParseResult sqlParseResult){ |
153 | 151 | } |
154 | 152 | } |
155 | 153 |
|
156 | | - // 将 sqlNode 转换为 SqlBasicCall |
157 | | - public static SqlBasicCall transformToSqlBasicCall(SqlNode sqlNode) { |
| 154 | + /** |
| 155 | + * 将第一层 select 中的 sqlNode 转化为 AsNode,解决字段名冲突问题 |
| 156 | + * @param selectList select Node 的 select 字段 |
| 157 | + * @param sqlSelect 第一层解析出来的 selectNode |
| 158 | + */ |
| 159 | + private static void rebuildSelectNode(SqlNodeList selectList, SqlSelect sqlSelect) { |
| 160 | + SqlNodeList sqlNodes = new SqlNodeList(selectList.getParserPosition()); |
| 161 | + |
| 162 | + for (int index = 0; index < selectList.size(); index++) { |
| 163 | + if (selectList.get(index).getKind().equals(SqlKind.AS)) { |
| 164 | + sqlNodes.add(selectList.get(index)); |
| 165 | + continue; |
| 166 | + } |
| 167 | + sqlNodes.add(transformToAsNode(selectList.get(index))); |
| 168 | + } |
| 169 | + sqlSelect.setSelectList(sqlNodes); |
| 170 | + } |
| 171 | + |
| 172 | + /** |
| 173 | + * 将 sqlNode 转化为 AsNode |
| 174 | + * @param sqlNode 需要转化的 sqlNode |
| 175 | + * @return 重新构造的 AsNode |
| 176 | + */ |
| 177 | + public static SqlBasicCall transformToAsNode(SqlNode sqlNode) { |
158 | 178 | String asName = ""; |
159 | 179 | SqlParserPos pos = new SqlParserPos(sqlNode.getParserPosition().getLineNum(), |
160 | 180 | sqlNode.getParserPosition().getEndColumnNum()); |
|
0 commit comments