generateId Object => Number

sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/api/rule/TableRule.java

@@ -215,7 +215,7 @@ public final class TableRule {
      * @param columnName 列名称
      * @return 生成的id
      */
-    public Object generateId(final String columnName) {
+    public Number generateId(final String columnName) {
         Number result = autoIncrementColumnMap.get(columnName).generateId();
         Preconditions.checkNotNull(result);
         return result;

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

@@ -161,7 +161,7 @@ public abstract class AbstractInsertParser implements SQLStatementParser {
             int count = 0;
             for (ShardingColumnContext each : shardingColumnContexts) {
                 if (each.isAutoIncrement()) {
-                    Number autoIncrementedValue = (Number) getShardingRule().findTableRule(sqlContext.getTables().get(0).getName()).generateId(each.getColumnName());
+                    Number autoIncrementedValue = getShardingRule().findTableRule(sqlContext.getTables().get(0).getName()).generateId(each.getColumnName());
                     if (parameters.isEmpty()) {
                         itemsToken.getItems().add(autoIncrementedValue.toString());
                         sqlExprs.add(new SQLNumberExpr(autoIncrementedValue));

sharding-jdbc-core/src/main/java/com/dangdang/ddframe/rdb/sharding/router/PreparedSQLRouter.java

@@ -56,13 +56,13 @@ public class PreparedSQLRouter {
         if (null == sqlContext) {
             sqlContext = engine.parseSQL(logicSql, parameters);
         } else {
-            List<Object> generatedIds = generateId();
+            List<Number> generatedIds = generateId();
             parameters.addAll(generatedIds);
         }
         return engine.routeSQL(sqlContext, parameters);
     }
     
-    private List<Object> generateId() {
+    private List<Number> generateId() {
         if (!(sqlContext instanceof InsertSQLContext)) {
             return Collections.emptyList();
         }
@@ -72,11 +72,11 @@ public class PreparedSQLRouter {
         }
         TableRule tableRule = tableRuleOptional.get();
         GeneratedKeyContext generatedKeyContext = ((InsertSQLContext) sqlContext).getGeneratedKeyContext();
-        List<Object> result = new ArrayList<>(generatedKeyContext.getColumns().size());
+        List<Number> result = new ArrayList<>(generatedKeyContext.getColumns().size());
         for (String each : generatedKeyContext.getColumns()) {
-            Object id = tableRule.generateId(each);
-            result.add(id);
-            generatedKeyContext.putValue(each, id);
+            Number generatedId = tableRule.generateId(each);
+            result.add(generatedId);
+            generatedKeyContext.putValue(each, generatedId);
         }
         return result;
     }