closed #103: NamedQuery support assign operation

Author: Jarvis

src/main/java/org/springframework/data/mybatis/repository/annotation/Query.java

@@ -22,6 +22,8 @@
 
 import java.lang.annotation.*;
 
+import static org.springframework.data.mybatis.repository.annotation.Query.Operation.unknown;
+
 /**
  * Annotated to named query.
  *
@@ -50,9 +52,19 @@
 
     boolean basic() default true;
 
+    Operation operation() default unknown;
+
     class Unspecified {
     }
 
+    enum Operation {
+        insert,
+        update,
+        select,
+        delete,
+        unknown
+    }
+
 }
 
 

src/main/java/org/springframework/data/mybatis/repository/query/AbstractMybatisQuery.java

@@ -22,6 +22,7 @@
 import org.springframework.core.convert.converter.Converter;
 import org.springframework.data.mybatis.repository.annotation.Basic;
 import org.springframework.data.mybatis.repository.annotation.Query;
+import org.springframework.data.mybatis.repository.annotation.Query.Operation;
 import org.springframework.data.mybatis.repository.query.MybatisQueryExecution.*;
 import org.springframework.data.repository.query.ParametersParameterAccessor;
 import org.springframework.data.repository.query.RepositoryQuery;
@@ -30,6 +31,8 @@
 import org.springframework.util.Assert;
 import org.springframework.util.StringUtils;
 
+import static org.springframework.data.mybatis.repository.annotation.Query.Operation.unknown;
+
 /**
  * abstract mybatis query.
  *
@@ -81,6 +84,14 @@ protected Class<?> getStatementReturnType() {
         return annotation.returnType();
     }
 
+    protected Operation getOperation() {
+        Query annotation = method.getQueryAnnotation();
+        if (null == annotation) {
+            return null;
+        }
+        return annotation.operation();
+    }
+
     protected String getStatementId() {
         return getNamespace() + "." + getStatementName();
     }
@@ -125,6 +136,22 @@ public SqlSessionTemplate getSqlSessionTemplate() {
 
 
     protected MybatisQueryExecution getExecution() {
+        Operation operation = getOperation();
+        if (null != operation && operation != unknown) {
+            switch (operation) {
+                case insert:
+                    return new InsertExecution();
+                case update:
+                    return new UpdateExecution();
+                case select:
+                    break;
+                case delete:
+                    return new DeleteExecution();
+                case unknown:
+                    break;
+            }
+        }
+
 
         if (method.isStreamQuery()) {
             return new StreamExecution();

src/main/java/org/springframework/data/mybatis/repository/query/MybatisQueryExecution.java

@@ -30,7 +30,7 @@
 import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
- import java.util.Collection;
+import java.util.Collection;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -201,6 +201,50 @@ protected <X> long calculateTotal(Pageable pager, List<X> result) {
             return -1;
         }
     }
+    static class InsertExecution extends MybatisQueryExecution {
+        @Override
+        protected Object doExecute(AbstractMybatisQuery query, Object[] values) {
+            if (null == values || values.length == 0) {
+                return query.getSqlSessionTemplate().insert(query.getStatementId());
+            }
+            MybatisParameters parameters = query.getQueryMethod().getParameters();
+            Map<String, Object> parameter = new HashMap<String, Object>();
+
+            int c = 0;
+            for (MybatisParameter param : parameters.getBindableParameters()) {
+                String name = param.getName();
+                if (StringUtils.isEmpty(name)) {
+                    name = "p" + c;
+                }
+                parameter.put(name, values[param.getIndex()]);
+                c++;
+            }
+
+            return query.getSqlSessionTemplate().insert(query.getStatementId(), parameter);
+        }
+    }
+    static class UpdateExecution extends MybatisQueryExecution {
+        @Override
+        protected Object doExecute(AbstractMybatisQuery query, Object[] values) {
+            if (null == values || values.length == 0) {
+                return query.getSqlSessionTemplate().update(query.getStatementId());
+            }
+            MybatisParameters parameters = query.getQueryMethod().getParameters();
+            Map<String, Object> parameter = new HashMap<String, Object>();
+
+            int c = 0;
+            for (MybatisParameter param : parameters.getBindableParameters()) {
+                String name = param.getName();
+                if (StringUtils.isEmpty(name)) {
+                    name = "p" + c;
+                }
+                parameter.put(name, values[param.getIndex()]);
+                c++;
+            }
+
+            return query.getSqlSessionTemplate().update(query.getStatementId(), parameter);
+        }
+    }
 
     static class SingleEntityExecution extends MybatisQueryExecution {