data-fast support page query

Author: xlorne

springboot-example/src/main/java/com/codingapi/springboot/example/infrastructure/dto/DemoDTO.java

@@ -0,0 +1,15 @@
+package com.codingapi.springboot.example.infrastructure.dto;
+
+import com.codingapi.springboot.framework.dto.request.PageRequest;
+import lombok.Getter;
+import lombok.Setter;
+
+public class DemoDTO {
+
+    @Setter
+    @Getter
+    public static class DemoQuery extends PageRequest {
+
+        private String name;
+    }
+}

springboot-example/src/main/java/com/codingapi/springboot/example/infrastructure/query/FastDemoApi.java

@@ -1,17 +1,19 @@
 package com.codingapi.springboot.example.infrastructure.query;
 
+import com.codingapi.springboot.example.infrastructure.dto.DemoDTO;
 import com.codingapi.springboot.example.infrastructure.entity.DemoEntity;
 import com.codingapi.springboot.fast.annotation.FastController;
 import com.codingapi.springboot.fast.annotation.FastMapping;
+import com.codingapi.springboot.framework.dto.response.MultiResponse;
 import org.springframework.web.bind.annotation.RequestMethod;
-import org.springframework.web.bind.annotation.RequestParam;
-
-import java.util.List;
 
 @FastController
 public interface FastDemoApi {
 
-    @FastMapping(method = RequestMethod.GET,mapping = "/open/fast/demo/findAll",hql = "select d from DemoEntity d where name = ?1")
-    List<DemoEntity> findAll(@RequestParam("name") String name);
+    @FastMapping(method = RequestMethod.GET,
+            mapping = "/open/fast/demo/findAll",
+            hql = "select d from DemoEntity d where name = :name",
+            countHql = "select count(d) from DemoEntity d where name = :name")
+    MultiResponse<DemoEntity> findAll(DemoDTO.DemoQuery query);
 
 }

springboot-example/src/main/java/com/codingapi/springboot/example/ui/controller/OpenController.java

@@ -1,6 +1,10 @@
 package com.codingapi.springboot.example.ui.controller;
 
 import com.codingapi.springboot.example.application.executor.DemoExecutor;
+import com.codingapi.springboot.example.infrastructure.entity.DemoEntity;
+import com.codingapi.springboot.example.infrastructure.jap.repository.DemoEntityRepository;
+import com.codingapi.springboot.framework.dto.request.PageRequest;
+import com.codingapi.springboot.framework.dto.response.MultiResponse;
 import com.codingapi.springboot.framework.dto.response.Response;
 import lombok.AllArgsConstructor;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -15,9 +19,16 @@ public class OpenController {
 
     private final DemoExecutor executor;
 
+    private final DemoEntityRepository demoEntityRepository;
+
     @GetMapping("/save")
     public Response save(@RequestParam("name") String name){
         executor.create(name);
         return Response.buildSuccess();
     }
+
+    @GetMapping("/list")
+    public MultiResponse<DemoEntity> list(PageRequest pageRequest){
+        return MultiResponse.of(demoEntityRepository.findAll(pageRequest));
+    }
 }

springboot-example/src/main/resources/application.properties

@@ -1,4 +1,4 @@
-server.port=8085
+server.port=8080
 spring.datasource.driver-class-name=org.h2.Driver
 spring.datasource.url=jdbc:h2:file:./demo.db
 spring.jpa.database-platform=org.hibernate.dialect.H2Dialect

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/annotation/FastMapping.java

@@ -16,6 +16,9 @@
 
     String hql() default "";
 
+
+    String countHql() default "";
+
     RequestMethod method() default RequestMethod.GET;
 
 

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/exception/FastMappingErrorException.java

@@ -0,0 +1,8 @@
+package com.codingapi.springboot.fast.exception;
+
+public class FastMappingErrorException extends Exception{
+
+    public FastMappingErrorException(String message) {
+        super(message);
+    }
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/executor/JpaExecutor.java

@@ -1,22 +1,36 @@
 package com.codingapi.springboot.fast.executor;
 
+import com.codingapi.springboot.framework.dto.response.MultiResponse;
+import com.codingapi.springboot.framework.dto.response.SingleResponse;
 import lombok.AllArgsConstructor;
+import org.springframework.data.domain.Page;
 
 import javax.persistence.EntityManager;
-import javax.persistence.Query;
+import java.util.Collection;
 
 @AllArgsConstructor
 public class JpaExecutor {
 
     private final EntityManager entityManager;
 
-    public Object execute(String jpaSql,Object[] args){
-        Query query =  entityManager.createQuery(jpaSql);
-        if(args!=null) {
-            for (int i = 0; i < args.length; i++) {
-                query.setParameter(i+1,args[i]);
+    public Object execute(String hql,String countHql,Object[] args,Class<?> returnType){
+        JpaQuery query = new JpaQuery(hql,countHql,args,entityManager);
+
+        if(returnType.equals(SingleResponse.class)){
+            return SingleResponse.of(query.getSingleResult());
+        }
+
+        if(returnType.equals(MultiResponse.class)){
+            Object returnData =  query.getResultList();
+            if(Page.class.isAssignableFrom(returnData.getClass())) {
+                return MultiResponse.of((Page)returnData);
+            }
+
+            if(Collection.class.isAssignableFrom(returnData.getClass())) {
+                return MultiResponse.of((Collection) returnData);
             }
         }
+
         return query.getResultList();
     }
 }

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/executor/JpaQuery.java

@@ -0,0 +1,99 @@
+package com.codingapi.springboot.fast.executor;
+
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.data.domain.PageImpl;
+import org.springframework.data.domain.Pageable;
+import org.springframework.util.ReflectionUtils;
+import org.springframework.util.StringUtils;
+
+import javax.persistence.EntityManager;
+import javax.persistence.Parameter;
+import javax.persistence.Query;
+import java.lang.reflect.Field;
+import java.util.Set;
+
+@Slf4j
+public class JpaQuery {
+    private final Object[] args;
+    private final Query query;
+    private final String hql;
+    private final String countHql;
+
+    private final EntityManager entityManager;
+
+    public JpaQuery(String hql,String countHql,Object[] args, EntityManager entityManager) {
+        this.hql = hql;
+        this.countHql = countHql;
+        this.args = args;
+        this.entityManager = entityManager;
+        this.query = entityManager.createQuery(hql);
+        this.setParameter(query);
+    }
+
+    @SneakyThrows
+    private void setParameter(Query query){
+        if(args!=null&&args.length>0) {
+            Set<Parameter<?>> parameters = query.getParameters();
+            for(Parameter<?> parameter:parameters){
+                Integer position = parameter.getPosition();
+                if(position!=null){
+                    query.setParameter(position, args[position - 1]);
+                }
+                if(StringUtils.hasText(parameter.getName())){
+                    String name = parameter.getName();
+                    Object obj = args[0];
+                    Field field =  ReflectionUtils.findField(obj.getClass(),name);
+                    if(field!=null) {
+                        field.setAccessible(true);
+                        query.setParameter(name, field.get(obj));
+                    }
+                }
+            }
+
+        }
+    }
+
+    private boolean isPageable() {
+        if(args!=null&& args.length>0){
+            Object lastObj = args[args.length-1];
+            return lastObj instanceof Pageable;
+        }
+        return false;
+    }
+
+    private Pageable getPageable(){
+        if(args!=null&& args.length>0){
+            Object lastObj = args[args.length-1];
+            if(lastObj instanceof Pageable){
+                return (Pageable) lastObj;
+            }
+        }
+        return null;
+    }
+
+    public Object getResultList() {
+        if(isPageable()&&StringUtils.hasText(countHql)){
+            Pageable pageable = getPageable();
+            query.setFirstResult((int) pageable.getOffset());
+            query.setMaxResults(pageable.getPageSize());
+            long total = getCount();
+            return new PageImpl<>(query.getResultList(),pageable,total);
+        }
+        return query.getResultList();
+    }
+
+
+    private long getCount(){
+        Query countQuery = entityManager.createQuery(countHql);
+        setParameter(countQuery);
+        return (Long) countQuery.getSingleResult();
+    }
+
+
+    public Object getSingleResult(){
+        return query.getSingleResult();
+    }
+
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/executor/MvcMethodProxy.java

@@ -21,6 +21,7 @@ public Object invoke(Object proxy, Method method, Object[] args)
             return hashCode();
         }
         FastMapping fastMapping = method.getAnnotation(FastMapping.class);
-        return jpaExecutor.execute(fastMapping.hql(),args);
+        Class<?> returnType =  method.getReturnType();
+        return jpaExecutor.execute(fastMapping.hql(),fastMapping.countHql(),args,returnType);
     }
 }

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/registrar/MvcMappingRegistrar.java

@@ -1,12 +1,15 @@
 package com.codingapi.springboot.fast.registrar;
 
 import com.codingapi.springboot.fast.annotation.FastMapping;
+import com.codingapi.springboot.fast.exception.FastMappingErrorException;
 import com.codingapi.springboot.fast.executor.JpaExecutor;
 import com.codingapi.springboot.fast.executor.MvcMethodProxy;
 import com.codingapi.springboot.fast.mapping.MvcEndpointMapping;
 import lombok.AllArgsConstructor;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.data.domain.Pageable;
+import org.springframework.util.StringUtils;
 
 import java.lang.reflect.Method;
 import java.lang.reflect.Proxy;
@@ -26,7 +29,7 @@ public void registerMvcMapping() {
            Method[] methods = clazz.getDeclaredMethods();
            for(Method method:methods){
                FastMapping fastMapping =  method.getAnnotation(FastMapping.class);
-               if(fastMapping!=null) {
+               if(fastMapping!=null&&isVerify(fastMapping,method)) {
                    MvcMethodProxy handler = new MvcMethodProxy(jpaExecutor);
                    Object methodProxy =  Proxy.newProxyInstance(clazz.getClassLoader(), new Class[]{clazz}, handler);
                    mvcEndpointMapping.addMapping(fastMapping.mapping(), fastMapping.method(), methodProxy, method);
@@ -35,4 +38,16 @@ public void registerMvcMapping() {
        }
     }
 
+    private boolean isVerify(FastMapping fastMapping,Method method) throws FastMappingErrorException {
+        Class<?>[] parameterTypes = method.getParameterTypes();
+        for(Class<?> parameter:parameterTypes){
+            if(Pageable.class.isAssignableFrom(parameter)){
+                if(!StringUtils.hasText(fastMapping.countHql())){
+                    throw new FastMappingErrorException(String.format("fast mapping %s missing countHql .",fastMapping.mapping()));
+                }
+            }
+        }
+        return true;
+    }
+
 }