add db2 sql parser support

src/main/java/com/alibaba/druid/sql/dialect/db2/ast/stmt/DB2SelectQueryBlock.java

@@ -0,0 +1,25 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.ast.stmt;
+
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+
+public class DB2SelectQueryBlock extends SQLSelectQueryBlock {
+
+    private static final long serialVersionUID = 1L;
+
+ 
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/parser/DB2ExprParser.java

@@ -0,0 +1,20 @@
+package com.alibaba.druid.sql.dialect.db2.parser;
+
+import com.alibaba.druid.sql.parser.Lexer;
+import com.alibaba.druid.sql.parser.SQLExprParser;
+
+public class DB2ExprParser extends SQLExprParser {
+
+    public final static String[] AGGREGATE_FUNCTIONS = { "AVG", "COUNT", "MAX", "MIN", "STDDEV", "SUM", "ROW_NUMBER" };
+
+    public DB2ExprParser(String sql){
+        this(new DB2Lexer(sql));
+        this.lexer.nextToken();
+    }
+
+    public DB2ExprParser(Lexer lexer){
+        super(lexer);
+        this.aggregateFunctions = AGGREGATE_FUNCTIONS;
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/parser/DB2Lexer.java

@@ -0,0 +1,27 @@
+package com.alibaba.druid.sql.dialect.db2.parser;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import com.alibaba.druid.sql.parser.Keywords;
+import com.alibaba.druid.sql.parser.Lexer;
+import com.alibaba.druid.sql.parser.Token;
+
+
+public class DB2Lexer extends Lexer {
+
+    public final static Keywords DEFAULT_DB2_KEYWORDS;
+
+    static {
+        Map<String, Token> map = new HashMap<String, Token>();
+
+        map.putAll(Keywords.DEFAULT_KEYWORDS.getKeywords());
+        
+        DEFAULT_DB2_KEYWORDS = new Keywords(map);
+    }
+
+    public DB2Lexer(String input){
+        super(input);
+        super.keywods = DEFAULT_DB2_KEYWORDS;
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/parser/DB2SelectParser.java

@@ -0,0 +1,80 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.parser;
+
+import com.alibaba.druid.sql.ast.SQLSetQuantifier;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQuery;
+import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
+import com.alibaba.druid.sql.dialect.db2.ast.stmt.DB2SelectQueryBlock;
+import com.alibaba.druid.sql.parser.SQLExprParser;
+import com.alibaba.druid.sql.parser.SQLSelectParser;
+import com.alibaba.druid.sql.parser.Token;
+
+public class DB2SelectParser extends SQLSelectParser {
+
+    public DB2SelectParser(SQLExprParser exprParser){
+        super(exprParser);
+    }
+
+    public DB2SelectParser(String sql){
+        this(new DB2ExprParser(sql));
+    }
+
+    protected SQLExprParser createExprParser() {
+        return new DB2ExprParser(lexer);
+    }
+
+    @Override
+    public SQLSelectQuery query() {
+        if (lexer.token() == Token.LPAREN) {
+            lexer.nextToken();
+
+            SQLSelectQuery select = query();
+            accept(Token.RPAREN);
+
+            return queryRest(select);
+        }
+
+        accept(Token.SELECT);
+        
+        if (lexer.token() == Token.COMMENT) {
+            lexer.nextToken();
+        }
+
+        DB2SelectQueryBlock queryBlock = new DB2SelectQueryBlock();
+
+        if (lexer.token() == Token.DISTINCT) {
+            queryBlock.setDistionOption(SQLSetQuantifier.DISTINCT);
+            lexer.nextToken();
+        } else if (lexer.token() == Token.UNIQUE) {
+            queryBlock.setDistionOption(SQLSetQuantifier.UNIQUE);
+            lexer.nextToken();
+        } else if (lexer.token() == Token.ALL) {
+            queryBlock.setDistionOption(SQLSetQuantifier.ALL);
+            lexer.nextToken();
+        }
+
+        parseSelectList(queryBlock);
+
+        parseFrom(queryBlock);
+
+        parseWhere(queryBlock);
+
+        parseGroupBy(queryBlock);
+
+        return queryRest(queryBlock);
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/parser/DB2StatementParser.java

@@ -0,0 +1,19 @@
+package com.alibaba.druid.sql.dialect.db2.parser;
+
+import com.alibaba.druid.sql.parser.Lexer;
+import com.alibaba.druid.sql.parser.SQLStatementParser;
+
+
+public class DB2StatementParser extends SQLStatementParser {
+    public DB2StatementParser(String sql) {
+        super (new DB2ExprParser(sql));
+    }
+
+    public DB2StatementParser(Lexer lexer){
+        super(new DB2ExprParser(lexer));
+    }
+    
+    public DB2SelectParser createSQLSelectParser() {
+        return new DB2SelectParser(this.exprParser);
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2ASTVisitor.java

@@ -0,0 +1,8 @@
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import com.alibaba.druid.sql.visitor.SQLASTVisitor;
+
+
+public interface DB2ASTVisitor extends SQLASTVisitor {
+
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2ASTVisitorAdapter.java

@@ -0,0 +1,7 @@
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import com.alibaba.druid.sql.visitor.SQLASTVisitorAdapter;
+
+public class DB2ASTVisitorAdapter extends SQLASTVisitorAdapter implements DB2ASTVisitor {
+
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2EvalVisitor.java

@@ -0,0 +1,142 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCaseExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
+import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNullExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
+import com.alibaba.druid.sql.ast.expr.SQLQueryExpr;
+import com.alibaba.druid.sql.ast.expr.SQLUnaryExpr;
+import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitor;
+import com.alibaba.druid.sql.visitor.SQLEvalVisitorUtils;
+import com.alibaba.druid.sql.visitor.functions.Function;
+
+public class DB2EvalVisitor extends DB2ASTVisitorAdapter implements SQLEvalVisitor {
+
+    private Map<String, Function> functions        = new HashMap<String, Function>();
+    private List<Object>          parameters       = new ArrayList<Object>();
+
+    private int                   variantIndex     = -1;
+
+    private boolean               markVariantIndex = true;
+
+    public DB2EvalVisitor(){
+        this(new ArrayList<Object>(1));
+    }
+
+    public DB2EvalVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    public void setParameters(List<Object> parameters) {
+        this.parameters = parameters;
+    }
+
+    public boolean visit(SQLCharExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public int incrementAndGetVariantIndex() {
+        return ++variantIndex;
+    }
+
+    public int getVariantIndex() {
+        return variantIndex;
+    }
+
+    public boolean visit(SQLVariantRefExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLUnaryExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLIntegerExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLNumberExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLCaseExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLInListExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLNullExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLMethodInvokeExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    @Override
+    public boolean visit(SQLQueryExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+
+    public boolean isMarkVariantIndex() {
+        return markVariantIndex;
+    }
+
+    public void setMarkVariantIndex(boolean markVariantIndex) {
+        this.markVariantIndex = markVariantIndex;
+    }
+
+    @Override
+    public Function getFunction(String funcName) {
+        return functions.get(funcName);
+    }
+
+    @Override
+    public void registerFunction(String funcName, Function function) {
+        functions.put(funcName, function);
+    }
+    
+    public boolean visit(SQLIdentifierExpr x) {
+        return SQLEvalVisitorUtils.visit(this, x);
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2ExportParameterVisitor.java

@@ -0,0 +1,87 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import com.alibaba.druid.sql.ast.SQLOrderBy;
+import com.alibaba.druid.sql.ast.expr.SQLBetweenExpr;
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
+import com.alibaba.druid.sql.ast.statement.SQLSelectGroupByClause;
+import com.alibaba.druid.sql.ast.statement.SQLSelectItem;
+import com.alibaba.druid.sql.visitor.ExportParameterVisitor;
+import com.alibaba.druid.sql.visitor.ExportParameterVisitorUtils;
+
+public class DB2ExportParameterVisitor extends DB2ASTVisitorAdapter implements ExportParameterVisitor {
+
+    private final List<Object> parameters;
+
+    public DB2ExportParameterVisitor(){
+        this(new ArrayList<Object>());
+    }
+
+    public DB2ExportParameterVisitor(List<Object> parameters){
+        this.parameters = parameters;
+    }
+
+    public List<Object> getParameters() {
+        return parameters;
+    }
+
+    @Override
+    public boolean visit(SQLSelectItem x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLOrderBy x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLSelectGroupByClause x) {
+        return false;
+    }
+
+    @Override
+    public boolean visit(SQLMethodInvokeExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getParameters());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLInListExpr x) {
+        ExportParameterVisitorUtils.exportParamterAndAccept(this.parameters, x.getTargetList());
+
+        return true;
+    }
+
+    @Override
+    public boolean visit(SQLBetweenExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        ExportParameterVisitorUtils.exportParameter(this.parameters, x);
+        return true;
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2OutputVisitor.java

@@ -0,0 +1,10 @@
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
+
+
+public class DB2OutputVisitor extends SQLASTOutputVisitor implements DB2ASTVisitor {
+    public DB2OutputVisitor(Appendable appender){
+        super(appender);
+    }
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2ParameterizedOutputVisitor.java

@@ -0,0 +1,88 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
+import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
+import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNCharExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNullExpr;
+import com.alibaba.druid.sql.ast.expr.SQLNumberExpr;
+import com.alibaba.druid.sql.visitor.ParameterizedOutputVisitorUtils;
+
+public class DB2ParameterizedOutputVisitor extends DB2OutputVisitor {
+
+    public DB2ParameterizedOutputVisitor(){
+        this(new StringBuilder());
+    }
+
+    public DB2ParameterizedOutputVisitor(Appendable appender){
+        super(appender);
+    }
+
+    public boolean visit(SQLInListExpr x) {
+        return ParameterizedOutputVisitorUtils.visit(this, x);
+    }
+
+    public boolean visit(SQLBinaryOpExpr x) {
+        x = ParameterizedOutputVisitorUtils.merge(x);
+
+        return super.visit(x);
+    }
+
+    public boolean visit(SQLNullExpr x) {
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLIntegerExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLNumberExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLCharExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+    public boolean visit(SQLNCharExpr x) {
+        if (Boolean.TRUE.equals(x.getAttribute(ParameterizedOutputVisitorUtils.ATTR_PARAMS_SKIP))) {
+            return super.visit(x);
+        }
+
+        print('?');
+        return false;
+    }
+
+}

src/main/java/com/alibaba/druid/sql/dialect/db2/visitor/DB2SchemaStatVisitor.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright 1999-2011 Alibaba Group Holding Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.alibaba.druid.sql.dialect.db2.visitor;
+
+import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
+import com.alibaba.druid.util.JdbcUtils;
+
+public class DB2SchemaStatVisitor extends SchemaStatVisitor implements DB2ASTVisitor {
+
+    @Override
+    public String getDbType() {
+        return JdbcUtils.DB2;
+    }
+
+}