add SQLStatementParserFactory

sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/parsing/SQLParsingEngine.java

@@ -21,18 +21,8 @@ import com.dangdang.ddframe.rdb.sharding.api.rule.ShardingRule;
 import com.dangdang.ddframe.rdb.sharding.constant.DatabaseType;
 import com.dangdang.ddframe.rdb.sharding.parsing.lexer.LexerEngine;
 import com.dangdang.ddframe.rdb.sharding.parsing.lexer.LexerEngineFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.lexer.token.DefaultKeyword;
-import com.dangdang.ddframe.rdb.sharding.parsing.lexer.token.TokenType;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.exception.SQLParsingUnsupportedException;
 import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.SQLStatement;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.alter.AlterParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.create.CreateParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.drop.DropParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.truncate.TruncateParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.delete.DeleteParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.insert.InsertParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.update.UpdateParserFactory;
-import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dql.select.SelectParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.SQLStatementParserFactory;
 import lombok.RequiredArgsConstructor;
 
 /**
@@ -57,29 +47,6 @@ public final class SQLParsingEngine {
     public SQLStatement parse() {
         LexerEngine lexerEngine = LexerEngineFactory.newInstance(dbType, sql);
         lexerEngine.nextToken();
-        TokenType tokenType = lexerEngine.getCurrentToken().getType();
-        if (!(tokenType instanceof DefaultKeyword)) {
-            throw new SQLParsingUnsupportedException(tokenType);
-        }
-        switch ((DefaultKeyword) tokenType) {
-            case SELECT:
-                return SelectParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case INSERT:
-                return InsertParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case UPDATE:
-                return UpdateParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case DELETE:
-                return DeleteParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case CREATE:
-                return CreateParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case ALTER:
-                return AlterParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case DROP:
-                return DropParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            case TRUNCATE:
-                return TruncateParserFactory.newInstance(dbType, shardingRule, lexerEngine).parse();
-            default:
-                throw new SQLParsingUnsupportedException(lexerEngine.getCurrentToken().getType());
-        }
+        return SQLStatementParserFactory.newInstance(dbType, lexerEngine.getCurrentToken().getType(), shardingRule, lexerEngine).parse();
     }
 }

sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/parsing/parser/statement/SQLStatementParserFactory.java

+/*
+ * Copyright 1999-2015 dangdang.com.
+ * <p>
+ * 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.
+ * </p>
+ */
+
+package com.dangdang.ddframe.rdb.sharding.parsing.parser.statement;
+
+import com.dangdang.ddframe.rdb.sharding.api.rule.ShardingRule;
+import com.dangdang.ddframe.rdb.sharding.constant.DatabaseType;
+import com.dangdang.ddframe.rdb.sharding.parsing.lexer.LexerEngine;
+import com.dangdang.ddframe.rdb.sharding.parsing.lexer.token.DefaultKeyword;
+import com.dangdang.ddframe.rdb.sharding.parsing.lexer.token.TokenType;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.exception.SQLParsingUnsupportedException;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.alter.AlterParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.create.CreateParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.drop.DropParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.ddl.truncate.TruncateParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.delete.DeleteParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.insert.InsertParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dml.update.UpdateParserFactory;
+import com.dangdang.ddframe.rdb.sharding.parsing.parser.statement.dql.select.SelectParserFactory;
+import lombok.AccessLevel;
+import lombok.NoArgsConstructor;
+
+/**
+ * SQL语句解析器工厂.
+ *
+ * @author zhangliang
+ */
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+public final class SQLStatementParserFactory {
+    
+    /**
+     * 创建SQL语句解析器.
+     *
+     * @param dbType 数据库类型
+     * @param tokenType 词法标记类型
+     * @param shardingRule 分库分表规则配置
+     * @param lexerEngine 词法解析器引擎
+     * @return SQL语句解析器
+     */
+    public static SQLStatementParser newInstance(final DatabaseType dbType, final TokenType tokenType, final ShardingRule shardingRule, final LexerEngine lexerEngine) {
+        if (!(tokenType instanceof DefaultKeyword)) {
+            throw new SQLParsingUnsupportedException(tokenType);
+        }
+        switch ((DefaultKeyword) tokenType)  {
+            case SELECT:
+                return SelectParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case INSERT:
+                return InsertParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case UPDATE:
+                return UpdateParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case DELETE:
+                return DeleteParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case CREATE:
+                return CreateParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case ALTER:
+                return AlterParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case DROP:
+                return DropParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            case TRUNCATE:
+                return TruncateParserFactory.newInstance(dbType, shardingRule, lexerEngine);
+            default:
+                throw new SQLParsingUnsupportedException(lexerEngine.getCurrentToken().getType());
+        }
+    }
+}