mybatis mongo order 查询

Author: xuejike

.gitignore

@@ -0,0 +1,99 @@
+# Created by .ignore support plugin (hsz.mobi)
+### JetBrains template
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm and Rider
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# Generated files
+.idea/**/contentModel.xml
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/artifacts
+# .idea/compiler.xml
+# .idea/jarRepositories.xml
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+# *.iml
+# *.ipr
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+
+# Android studio 3.1+ serialized cache file
+.idea/caches/build_file_checksums.ser
+
+### Java template
+# Compiled class file
+*.class
+
+# Log file
+*.log
+
+# BlueJ files
+*.ctxt
+
+# Mobile Tools for Java (J2ME)
+.mtj.tmp/
+
+# Package Files #
+*.jar
+*.war
+*.nar
+*.ear
+*.zip
+*.tar.gz
+*.rar
+
+# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
+hs_err_pid*
+

lambda-query-core/src/main/java/com/github/xuejike/query/core/JkLambdaQuery.java

@@ -3,13 +3,19 @@
 import com.github.xuejike.query.core.base.BaseNestedWhereQuery;
 import com.github.xuejike.query.core.criteria.DaoCriteria;
 import com.github.xuejike.query.core.criteria.InjectionBaseQuery;
+import com.github.xuejike.query.core.criteria.OrderCriteria;
+import com.github.xuejike.query.core.enums.OrderType;
+import com.github.xuejike.query.core.po.FieldInfo;
 import com.github.xuejike.query.core.tool.lambda.FieldFunction;
+import com.github.xuejike.query.core.tool.lambda.LambdaTool;
+
+import java.util.Arrays;
 
 /**
  * @author xuejike
  * @date 2020/12/18
  */
-public class JkLambdaQuery<T> extends BaseNestedWhereQuery<T, FieldFunction<T,?>, JkLambdaQuery<T>> implements DaoCriteria<T> {
+public class JkLambdaQuery<T> extends BaseNestedWhereQuery<T, FieldFunction<T,?>, JkLambdaQuery<T>> implements DaoCriteria<T> , OrderCriteria<FieldFunction<T,?>, JkLambdaQuery<T>> {
     private DaoCriteria<T> daoCriteria;
 
     public JkLambdaQuery(DaoCriteria<T> daoCriteria) {
@@ -24,4 +30,22 @@ public DaoCriteria<T> getDao() {
         return daoCriteria;
     }
 
+
+    @Override
+    public JkLambdaQuery<T> orderAsc(FieldFunction<T, ?>... fields) {
+        Arrays.stream(fields).map(LambdaTool::getFieldInfo).forEach(it->{
+            orderMap.put(it, OrderType.asc);
+        });
+
+        return returnObj;
+    }
+
+    @Override
+    public JkLambdaQuery<T> orderDesc(FieldFunction<T, ?>... fields) {
+        Arrays.stream(fields).map(LambdaTool::getFieldInfo).forEach(it->{
+            orderMap.put(it, OrderType.desc);
+        });
+
+        return returnObj;
+    }
 }

lambda-query-core/src/main/java/com/github/xuejike/query/core/base/BaseQuery.java

@@ -3,6 +3,7 @@
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.log.Log;
 import com.github.xuejike.query.core.criteria.*;
+import com.github.xuejike.query.core.enums.OrderType;
 import com.github.xuejike.query.core.enums.StringMatchMode;
 import com.github.xuejike.query.core.enums.WhereOperation;
 import com.github.xuejike.query.core.exception.LambdaQueryException;
@@ -26,6 +27,7 @@ public class BaseWhereQuery<T,F,C> implements  WhereCriteria<C,F>,SelectCriteria
     protected List<BaseWhereQuery<T,F,?>> orList = new LinkedList<>();
     protected List<FieldInfo> selectList = new LinkedList<>();
     protected List<FieldInfo> excludeList = new LinkedList<>();
+    protected Map<FieldInfo, OrderType> orderMap = new ConcurrentHashMap<>();
     protected C returnObj = (C)this;
 
     public BaseWhereQuery() {
@@ -220,7 +222,7 @@ public C between(boolean condition, F field, Object start, Object end) {
 
 
     public boolean isEmpty(){
-        return CollUtil.isEmpty(whereMap) && CollUtil.isEmpty(orList);
+        return CollUtil.isEmpty(whereMap) && CollUtil.isEmpty(orList) && CollUtil.isEmpty(selectList) && CollUtil.isEmpty(excludeList) && CollUtil.isEmpty(orderMap);
     }
     public boolean isNotEmpty(){
         return !isEmpty();
@@ -284,6 +286,9 @@ public QueryInfo buildQueryInfo(){
         return queryInfo;
     }
 
+    public Map<FieldInfo, OrderType> getOrderMap() {
+        return orderMap;
+    }
 
     @Override
     public C select(F... fields) {

lambda-query-core/src/main/java/com/github/xuejike/query/core/base/BaseWhereQuery.java

@@ -46,6 +46,16 @@ default T findById(Serializable id){
     default boolean updateById(T entity){
         return getDao().updateById(entity);
     }
+
+    /**
+     * 插入
+     * @param entity
+     * @return
+     */
+    default T insert(T entity){
+        return getDao().insert(entity);
+    }
+
     default Long updateFindAll(){
         return getDao().updateFindAll();
     }

lambda-query-core/src/main/java/com/github/xuejike/query/core/criteria/DaoCriteria.java

@@ -1,6 +1,6 @@
 package com.github.xuejike.query.core.criteria;
 
-public interface OrderCriteria<T,F> {
-    T orderAsc(F ...fields);
-    T orderDesc(F ...fields);
+public interface OrderCriteria<F,R> {
+    R orderAsc(F ...fields);
+    R orderDesc(F ...fields);
 }

lambda-query-core/src/main/java/com/github/xuejike/query/core/criteria/OrderCriteria.java

@@ -0,0 +1,9 @@
+package com.github.xuejike.query.core.enums;
+
+/**
+ * @author xuejike
+ * @date 2020/12/30
+ */
+public enum OrderType {
+    desc,asc
+}

lambda-query-core/src/main/java/com/github/xuejike/query/core/enums/OrderType.java

@@ -52,4 +52,5 @@ private void buildInfo(FieldFunction<?, ?> fieldFunction) {
         this.type = info.type;
     }
 
+
 }

lambda-query-core/src/main/java/com/github/xuejike/query/core/po/FieldInfo.java

@@ -1,5 +1,4 @@
 import com.alibaba.fastjson.JSON;
-import com.github.xuejike.query.core.base.BaseQuery;
 import com.github.xuejike.query.core.po.QueryInfo;
 import com.github.xuejike.query.core.tool.lambda.CascadeField;
 
@@ -12,22 +11,6 @@ public class Test {
     public void t(){
 
 
-        BaseQuery<TestBean> baseQuery = new BaseQuery<>(TestBean.class);
-        QueryInfo queryInfo = baseQuery.eq(cascadeField().sub(TestBean::getSub).
-                sub(TestBean.SubNameCls::getSubName)
-                .subFieldName("xxx1")
-                .subFieldName("xxx2"), "xx")
-                .eq(TestBean::getSub, "xxx")
-                .or()
-                .eq(TestBean::getName, "666")
-                .eq(TestBean::getSubList,"666666")
-                .eq(TestBean::getSubArray,"666666")
-                .gt(TestBean::getName,"n1").lt(TestBean::getName,"n2")
-                .buildQueryInfo();
-        System.out.println(JSON.toJSONString(queryInfo));
-        System.out.println("666");
-        cascadeField().subList(TestBean::getSubList).sub(TestBean.SubNameCls::getSub);
-
 
     }
 

lambda-query-core/src/test/java/Test.java

@@ -5,9 +5,11 @@
 import com.github.xuejike.query.core.base.BaseWhereQuery;
 import com.github.xuejike.query.core.criteria.DaoCriteria;
 import com.github.xuejike.query.core.criteria.IJPage;
+import com.github.xuejike.query.core.enums.OrderType;
 import com.github.xuejike.query.core.po.FieldInfo;
 import com.mongodb.client.result.DeleteResult;
 import org.bson.Document;
+import org.springframework.data.domain.Sort;
 import org.springframework.data.mongodb.core.MongoTemplate;
 import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Field;
@@ -16,6 +18,9 @@
 
 import java.io.Serializable;
 import java.util.List;
+import java.util.Map;
+import java.util.function.IntFunction;
+import java.util.stream.Collectors;
 
 /**
  * @author xuejike
@@ -49,6 +54,17 @@ private Query buildQuery(){
                 fields.exclude(MongoQueryBuilder.buildField(info));
             }
         }
+        if (CollUtil.isNotEmpty(baseWhereQuery.getOrderMap())){
+            Map<FieldInfo, OrderType> orderMap = baseWhereQuery.getOrderMap();
+            List<Sort.Order> orderList = orderMap.entrySet().stream().map(it -> {
+                if (it.getValue() == OrderType.desc) {
+                    return Sort.Order.desc(MongoQueryBuilder.buildField(it.getKey()));
+                } else {
+                    return Sort.Order.asc(MongoQueryBuilder.buildField(it.getKey()));
+                }
+            }).collect(Collectors.toList());
+            builder.with(Sort.by(orderList));
+        }
         return builder;
 
     }
@@ -100,6 +116,12 @@ public boolean updateById(T entity) {
         return true;
     }
 
+    @Override
+    public T insert(T entity) {
+        T insert = mongoTemplate.insert(entity);
+        return insert;
+    }
+
     @Override
     public Long updateFindAll() {
         Query query = buildQuery();