improved sql parser.

2023-03-04 19:17:16 +00:00 · 2017-06-05 09:59:34 +08:00
parent 21ae9b5722
commit bea1d70af2
28 changed files with 419 additions and 233 deletions
@@ -810,5 +810,54 @@ public class SQLUtils {

        return false;
    }
+
+    public static SQLCaseExpr decodeToCase(SQLMethodInvokeExpr x) {
+        if (x == null) {
+            return null;
+        }
+
+        if (!"decode".equalsIgnoreCase(x.getMethodName())) {
+            throw new IllegalArgumentException(x.getMethodName());
+        }
+
+        List<SQLExpr> parameters = x.getParameters();
+        SQLCaseExpr caseExpr = new SQLCaseExpr();
+
+        caseExpr.setValueExpr(parameters.get(0));
+
+        for (int i = 1; i + 1 < parameters.size(); i += 2) {
+            SQLCaseExpr.Item item = new SQLCaseExpr.Item();
+            SQLExpr conditionExpr = parameters.get(i);
+
+            item.setConditionExpr(conditionExpr);
+
+            SQLExpr valueExpr = parameters.get(i + 1);
+
+            if (valueExpr instanceof SQLMethodInvokeExpr) {
+                SQLMethodInvokeExpr methodInvokeExpr = (SQLMethodInvokeExpr) valueExpr;
+                if ("decode".equalsIgnoreCase(methodInvokeExpr.getMethodName())) {
+                    valueExpr = decodeToCase(methodInvokeExpr);
+                }
+            }
+
+            item.setValueExpr(valueExpr);
+            caseExpr.addItem(item);
+        }
+
+        if (parameters.size() % 2 == 0) {
+            SQLExpr defaultExpr = parameters.get(parameters.size() - 1);
+
+            if (defaultExpr instanceof SQLMethodInvokeExpr) {
+                SQLMethodInvokeExpr methodInvokeExpr = (SQLMethodInvokeExpr) defaultExpr;
+                if ("decode".equalsIgnoreCase(methodInvokeExpr.getMethodName())) {
+                    defaultExpr = decodeToCase(methodInvokeExpr);
+                }
+            }
+
+            caseExpr.setElseExpr(defaultExpr);
+        }
+
+        return caseExpr;
+    }
 }