@ReturnGeneratedKeys增加DAO返回类型判断

Author: Qieqie

paoding-rose-jade/src/main/java/net/paoding/rose/jade/annotation/ReturnGeneratedKeys.java

@@ -49,7 +49,7 @@
 
     Class<? extends DynamicReturnGeneratedKeys>value() default Yes.class;
 
-    static class Yes implements DynamicReturnGeneratedKeys {
+    static class Yes extends DynamicReturnGeneratedKeys {
 
         @Override
         public boolean shouldReturnGerneratedKeys(StatementRuntime runtime) {

paoding-rose-jade/src/main/java/net/paoding/rose/jade/statement/DynamicReturnGeneratedKeys.java

@@ -15,17 +15,34 @@
  */
 package net.paoding.rose.jade.statement;
 
+import org.springframework.dao.InvalidDataAccessApiUsageException;
+
 import net.paoding.rose.jade.annotation.ReturnGeneratedKeys;
 
 /**
  * 
  * @see ReturnGeneratedKeys
  */
-public interface DynamicReturnGeneratedKeys {
+public abstract class DynamicReturnGeneratedKeys {
 
     /**
      * 是否要启动 return generated keys机制
      * @param runtime
      */
-    public boolean shouldReturnGerneratedKeys(StatementRuntime runtime);
+    public abstract boolean shouldReturnGerneratedKeys(StatementRuntime runtime);
+
+    /**
+     * 检查DAO返回的类型是否合格
+     * 
+     * @param returnType DAO方法的返回类型（如果方法声明的返回类型是泛型，框架会根据上下文信息解析为运行时实际应该返回的真正类型)
+     * 
+     * @throws InvalidDataAccessApiUsageException DAO方法的返回类型不合格
+     */
+    public void checkMethodReturnType(Class<?> returnType, StatementMetaData metaData) {
+        if (returnType != void.class && !Number.class.isAssignableFrom(returnType)) {
+            throw new InvalidDataAccessApiUsageException(
+                "error return type, only support int/long/double/float/void type for method with @ReturnGeneratedKeys:"
+                                                         + metaData.getMethod());
+        }
+    }
 }

paoding-rose-jade/src/main/java/net/paoding/rose/jade/statement/JdbcStatement.java

@@ -94,6 +94,7 @@ public JdbcStatement(StatementMetaData statementMetaData, SQLType sqlType,
             if (types.length > 0 && List.class.isAssignableFrom(types[0])) {
                 this.batchUpdate = true;
                 if (metaData.getMethod().getAnnotation(ReturnGeneratedKeys.class) != null) {
+                    // 批量处理的直接不支持@ReturnGeneratedKeys注解
                     throw new InvalidDataAccessApiUsageException(
                         "batch update method cannot return generated keys: " + method);
                 }
@@ -106,11 +107,8 @@ public JdbcStatement(StatementMetaData statementMetaData, SQLType sqlType,
             } else {
                 this.batchUpdate = false;
                 if (metaData.getMethod().getAnnotation(ReturnGeneratedKeys.class) != null) {
-                    if (returnType != void.class && !Number.class.isAssignableFrom(returnType)) {
-                        throw new InvalidDataAccessApiUsageException(
-                            "error return type, only support numberic/void type for method with @ReturnGeneratedKeys:"
-                                                                     + method);
-                    }
+                    metaData.getReturnGeneratedKeys().checkMethodReturnType(metaData.getReturnType(),
+                        metaData);
                 } else if (returnType != void.class && returnType != Boolean.class
                            && returnType != Integer.class) {
                     throw new InvalidDataAccessApiUsageException(