improve sql parser bigquery support

alibaba · Dec 19, 2024 · 9faeba9 · 9faeba9
1 parent 5a1a5f0
commit 9faeba9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLMergeStatement.java b/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLMergeStatement.java
@@ -117,6 +117,18 @@ public void addItem(SQLUpdateSetItem item) {
             this.items.add(item);
         }
 
+        public SQLUpdateSetItem findItemByColumn(SQLExpr column) {
+            if (column == null) {
+                return null;
+            }
+            for (SQLUpdateSetItem item : items) {
+                if (item.getColumn().equals(column)) {
+                    return item;
+                }
+            }
+            return null;
+        }
+
         @Override
         public void accept0(SQLASTVisitor v) {
             if (v.visit(this)) {

diff --git a/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLUpdateSetItem.java b/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLUpdateSetItem.java
@@ -30,6 +30,11 @@ public class SQLUpdateSetItem extends SQLObjectImpl implements SQLReplaceable {
     public SQLUpdateSetItem() {
     }
 
+    public SQLUpdateSetItem(SQLExpr column, SQLExpr value) {
+        this.setColumn(column);
+        this.setValue(value);
+    }
+
     public SQLExpr getColumn() {
         return column;
     }