add SortRepository & DynamicRepository

Author: xlorne

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/dynamic/DynamicConfiguration.java

@@ -0,0 +1,21 @@
+package com.codingapi.springboot.fast.dynamic;
+
+import jakarta.persistence.EntityManager;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+
+@Configuration
+public class DynamicConfiguration {
+
+    @Bean
+    public DynamicQuery dynamicQuery(EntityManager entityManager){
+        return new DynamicQuery(entityManager);
+    }
+
+    @Bean
+    public DynamicQueryContextRegister dynamicQueryContextRegister(DynamicQuery dynamicQuery){
+        return new DynamicQueryContextRegister(dynamicQuery);
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/dynamic/DynamicQuery.java

@@ -0,0 +1,50 @@
+package com.codingapi.springboot.fast.dynamic;
+
+import jakarta.persistence.EntityManager;
+import jakarta.persistence.TypedQuery;
+import lombok.AllArgsConstructor;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageImpl;
+import org.springframework.data.domain.PageRequest;
+
+import java.util.List;
+
+@AllArgsConstructor
+public class DynamicQuery {
+
+    private final EntityManager entityManager;
+
+    public List<?> listQuery(Class<?> clazz, String sql, Object... params) {
+        TypedQuery<?> query = entityManager.createQuery(sql, clazz);
+        if (params != null) {
+            for (int i = 0; i < params.length; i++) {
+                query.setParameter(i + 1, params[i]);
+            }
+        }
+        return query.getResultList();
+    }
+
+    public Page<?> pageQuery(Class<?> clazz, String sql, String countSql, PageRequest pageRequest, Object... params) {
+        TypedQuery<?> query = entityManager.createQuery(sql, clazz);
+        if (params != null) {
+            for (int i = 0; i < params.length; i++) {
+                query.setParameter(i + 1, params[i]);
+            }
+        }
+        query.setFirstResult(pageRequest.getPageNumber() * pageRequest.getPageSize());
+        query.setMaxResults(pageRequest.getPageSize());
+        return new PageImpl<>(query.getResultList(), pageRequest, countQuery(countSql, params));
+    }
+
+
+    private long countQuery(String sql, Object... params) {
+        TypedQuery<Long> query = entityManager.createQuery(sql, Long.class);
+        if (params != null) {
+            for (int i = 0; i < params.length; i++) {
+                query.setParameter(i + 1, params[i]);
+            }
+        }
+        return query.getSingleResult();
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/dynamic/DynamicQueryContext.java

@@ -0,0 +1,23 @@
+package com.codingapi.springboot.fast.dynamic;
+
+
+import lombok.Getter;
+
+public class DynamicQueryContext {
+
+    @Getter
+    private static final DynamicQueryContext instance = new DynamicQueryContext();
+
+    private DynamicQueryContext() {
+
+    }
+
+    @Getter
+    private DynamicQuery dynamicQuery;
+
+    void setDynamicQuery(DynamicQuery dynamicQuery) {
+        this.dynamicQuery = dynamicQuery;
+    }
+
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/dynamic/DynamicQueryContextRegister.java

@@ -0,0 +1,16 @@
+package com.codingapi.springboot.fast.dynamic;
+
+import lombok.AllArgsConstructor;
+import org.springframework.beans.factory.InitializingBean;
+
+@AllArgsConstructor
+public class DynamicQueryContextRegister implements InitializingBean {
+
+    private DynamicQuery dynamicQuery;
+
+    @Override
+    public void afterPropertiesSet() throws Exception {
+        DynamicQueryContext.getInstance().setDynamicQuery(dynamicQuery);
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/dynamic/DynamicRepository.java

@@ -0,0 +1,39 @@
+package com.codingapi.springboot.fast.dynamic;
+
+import org.springframework.core.ResolvableType;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.repository.NoRepositoryBean;
+
+import java.util.List;
+
+@NoRepositoryBean
+public interface DynamicRepository<T, ID> extends JpaRepository<T, ID> {
+
+    default Class<?> getEntityClass() {
+        ResolvableType resolvableType = ResolvableType.forClass(this.getClass()).as(DynamicRepository.class);
+        return resolvableType.getGeneric(new int[]{0}).resolve();
+    }
+
+    default List<T> dynamicListQuery(String sql, Object... params) {
+        return (List<T>) DynamicQueryContext.getInstance().getDynamicQuery().listQuery(getEntityClass(), sql, params);
+    }
+
+    default <V> List<V> dynamicListQuery(Class<V> clazz, String sql, Object... params) {
+        return (List<V>) DynamicQueryContext.getInstance().getDynamicQuery().listQuery(clazz, sql, params);
+    }
+
+    default Page<T> dynamicPageQuery(String sql, String countSql, PageRequest request, Object... params) {
+        return (Page<T>) DynamicQueryContext.getInstance().getDynamicQuery().pageQuery(getEntityClass(), sql, countSql, request, params);
+    }
+
+    default Page<T> dynamicPageQuery(String sql, PageRequest request, Object... params) {
+        return (Page<T>) DynamicQueryContext.getInstance().getDynamicQuery().pageQuery(getEntityClass(), sql, "select count(1) " + sql, request, params);
+    }
+
+    default <V> Page<V> dynamicPageQuery(Class<V> clazz, String sql, String countSql, PageRequest request, Object... params) {
+        return (Page<V>) DynamicQueryContext.getInstance().getDynamicQuery().pageQuery(clazz, sql, countSql, request, params);
+    }
+
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/query/FastRepository.java

@@ -1,5 +1,6 @@
 package com.codingapi.springboot.fast.query;
 
+import com.codingapi.springboot.fast.dynamic.DynamicRepository;
 import com.codingapi.springboot.framework.dto.request.PageRequest;
 import jakarta.persistence.criteria.Order;
 import jakarta.persistence.criteria.Predicate;
@@ -13,7 +14,7 @@
 import java.util.List;
 
 @NoRepositoryBean
-public interface FastRepository<T, ID> extends JpaRepository<T, ID>, JpaSpecificationExecutor<T> {
+public interface FastRepository<T, ID> extends JpaRepository<T, ID>, JpaSpecificationExecutor<T>, DynamicRepository<T,ID> {
 
     default Page<T> findAll(PageRequest request) {
         if (request.hasFilter()) {

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/sort/ISort.java

@@ -0,0 +1,8 @@
+package com.codingapi.springboot.fast.sort;
+
+public interface ISort {
+
+    Integer getSort();
+
+    void setSort(Integer sort);
+}

springboot-starter-data-fast/src/main/java/com/codingapi/springboot/fast/sort/SortRepository.java

@@ -0,0 +1,21 @@
+package com.codingapi.springboot.fast.sort;
+
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.repository.NoRepositoryBean;
+
+import java.util.List;
+
+@NoRepositoryBean
+public interface SortRepository<T extends ISort, I> extends JpaRepository<T, I> {
+
+
+    default void sort(PageRequest request, List<Integer> ids) {
+        for (int i = 0; i < ids.size(); i++) {
+            ISort entity = getById((I) ids.get(i));
+            entity.setSort(i + ((request.getPageNumber() - 1) * request.getPageSize()));
+            save((T) entity);
+        }
+    }
+
+}

springboot-starter-data-fast/src/main/resources/META-INF/spring.factories

@@ -1,3 +1,4 @@
 org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
 com.codingapi.springboot.fast.DataFastConfiguration,\
-com.codingapi.springboot.fast.registrar.DataFastBeanDefinitionRegistrar
+com.codingapi.springboot.fast.registrar.DataFastBeanDefinitionRegistrar,\
+com.codingapi.springboot.fast.dynamic.DynamicConfiguration

springboot-starter-data-fast/src/main/resources/META-INF/spring/org.springframework.boot.autoconfigure.AutoConfiguration.imports

@@ -1,2 +1,3 @@
 com.codingapi.springboot.fast.DataFastConfiguration
-com.codingapi.springboot.fast.registrar.DataFastBeanDefinitionRegistrar
+com.codingapi.springboot.fast.registrar.DataFastBeanDefinitionRegistrar
+com.codingapi.springboot.fast.dynamic.DynamicConfiguration