From 30a883a74f1f281524325e82c836f0f9e80add06 Mon Sep 17 00:00:00 2001 From: wenshao Date: Sun, 4 Feb 2018 16:14:52 +0800 Subject: [PATCH] improved sql parser for db2. --- .../java/com/alibaba/druid/sql/dialect/db2/ast/DB2Object.java | 1 + .../druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/alibaba/druid/sql/dialect/db2/ast/DB2Object.java b/src/main/java/com/alibaba/druid/sql/dialect/db2/ast/DB2Object.java index c6c8dcf055..2fcaa0e337 100644 --- a/src/main/java/com/alibaba/druid/sql/dialect/db2/ast/DB2Object.java +++ b/src/main/java/com/alibaba/druid/sql/dialect/db2/ast/DB2Object.java @@ -26,5 +26,6 @@ public interface DB2Object extends SQLObject { interface Constants { long CURRENT_DATE = FnvHash.fnv1a_64_lower("CURRENT DATE"); long CURRENT_TIME = FnvHash.fnv1a_64_lower("CURRENT TIME"); + long CURRENT_SCHEMA = FnvHash.fnv1a_64_lower("CURRENT SCHEMA"); } } diff --git a/src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java b/src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java index 266f16e52f..eebe92be30 100644 --- a/src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java +++ b/src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java @@ -60,6 +60,8 @@ public class DB2SchemaStatVisitor extends SchemaStatVisitor implements DB2ASTVis } protected boolean isPseudoColumn(long hash64) { - return hash64 == DB2Object.Constants.CURRENT_DATE || hash64 == DB2Object.Constants.CURRENT_TIME; + return hash64 == DB2Object.Constants.CURRENT_DATE + || hash64 == DB2Object.Constants.CURRENT_TIME + || hash64 == DB2Object.Constants.CURRENT_SCHEMA; } }