From da8d80b934ea002ac39b41cbebc035f653e86e7c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E9=93=81?= Date: Sat, 5 Oct 2019 08:18:31 +0800 Subject: [PATCH] backport sql parser improved from fastsql --- .../druid/sql/ast/SQLDataTypeRefExpr.java | 53 +++++++++++++++++++ .../druid/sql/ast/SQLDbTypedObject.java | 6 +++ 2 files changed, 59 insertions(+) create mode 100644 src/main/java/com/alibaba/druid/sql/ast/SQLDataTypeRefExpr.java create mode 100644 src/main/java/com/alibaba/druid/sql/ast/SQLDbTypedObject.java diff --git a/src/main/java/com/alibaba/druid/sql/ast/SQLDataTypeRefExpr.java b/src/main/java/com/alibaba/druid/sql/ast/SQLDataTypeRefExpr.java new file mode 100644 index 0000000000..2201e8a2f2 --- /dev/null +++ b/src/main/java/com/alibaba/druid/sql/ast/SQLDataTypeRefExpr.java @@ -0,0 +1,53 @@ +package com.alibaba.druid.sql.ast; + +import com.alibaba.druid.sql.visitor.SQLASTVisitor; + +public class SQLDataTypeRefExpr extends SQLExprImpl { + private SQLDataType dataType; + + public SQLDataTypeRefExpr(SQLDataType dataType) { + this.dataType = dataType; + } + + public SQLDataType getDataType() { + return dataType; + } + + public void setDataType(SQLDataType x) { + if (x != null) { + x.setParent(this); + } + this.dataType = x; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + + SQLDataTypeRefExpr that = (SQLDataTypeRefExpr) o; + + return dataType != null ? dataType.equals(that.dataType) : that.dataType == null; + } + + @Override + public int hashCode() { + return dataType != null ? dataType.hashCode() : 0; + } + + @Override + protected void accept0(SQLASTVisitor v) { + if (v.visit(this)) { + acceptChild(v, dataType); + } + v.endVisit(this); + } + + @Override + public SQLExpr clone() { + return new SQLDataTypeRefExpr( + dataType == null + ? null + : dataType.clone()); + } +} diff --git a/src/main/java/com/alibaba/druid/sql/ast/SQLDbTypedObject.java b/src/main/java/com/alibaba/druid/sql/ast/SQLDbTypedObject.java new file mode 100644 index 0000000000..3625a7b396 --- /dev/null +++ b/src/main/java/com/alibaba/druid/sql/ast/SQLDbTypedObject.java @@ -0,0 +1,6 @@ +package com.alibaba.druid.sql.ast; + +public interface SQLDbTypedObject extends SQLObject { + String getDbType(); +} +