From bd97fda26e0432ae5634f6756b94f4169350c0f0 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期二, 03 九月 2024 21:55:21 +0800
Subject: [PATCH] 视频点位在线率mongo重构

---
 ycl-server/src/main/java/com/ycl/utils/MongoUtil.java |   58 +++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 51 insertions(+), 7 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/utils/MongoUtil.java b/ycl-server/src/main/java/com/ycl/utils/MongoUtil.java
index 4f88595..0089c62 100644
--- a/ycl-server/src/main/java/com/ycl/utils/MongoUtil.java
+++ b/ycl-server/src/main/java/com/ycl/utils/MongoUtil.java
@@ -1,11 +1,17 @@
 package com.ycl.utils;
 
+import com.ycl.platform.domain.query.DataCenterQuery;
+import org.springframework.data.domain.Sort;
+import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.data.mongodb.core.query.TextCriteria;
+import org.springframework.util.CollectionUtils;
 import org.springframework.util.StringUtils;
 
 import java.lang.reflect.Field;
+import java.util.ArrayList;
 import java.util.Date;
+import java.util.List;
 import java.util.Objects;
 
 /**
@@ -17,16 +23,54 @@
 public class MongoUtil {
 
     /**
-     * 鏋勯�犲叏鏂囩储寮曠殑鏉′欢
+     * 鏋勯�犳暟鎹腑蹇冩悳绱㈢殑鏉′欢
      *
-     * @param query
-     * @param keyword
+     * @param params 鏁版嵁涓績璇锋眰鍙傛暟
+     * @param gbFiled 鍥芥爣鐮佸瓧娈垫槸鍝竴涓�
+     * @param timeFiled 鏃堕棿瀛楁鏄摢涓�涓�
+     * @param gbList 鍥芥爣鐮�
      */
-    public static void fullText(Query query, String keyword) {
-        if (StringUtils.hasText(keyword)) {
-            TextCriteria criteria = TextCriteria.forDefaultLanguage().matching(keyword);
-            query.addCriteria(criteria);
+    public static Query getQuery(DataCenterQuery params, String gbFiled, String timeFiled, List<String> gbList) {
+        // 鍏抽敭瀛�
+        TextCriteria text = null;
+        if (StringUtils.hasText(params.getKeyword())) {
+            text = TextCriteria.forDefaultLanguage().matching(params.getKeyword());
         }
+
+        Criteria in = null;
+        // in
+        if (! CollectionUtils.isEmpty(gbList)) {
+            in = Criteria.where(gbFiled).in(gbList);
+        }
+
+        // 鏃堕棿
+        Criteria time = Criteria.where(timeFiled).gte(params.getStartTime()).lte(params.getEndTime());
+
+        // 灏嗘煡璇㈡潯浠禷nd璧锋潵
+        Criteria and = new Criteria();
+        List<Criteria> qList = new ArrayList<>(3);
+        if (Objects.nonNull(text)) {
+            qList.add(Criteria.byExample(text));
+        }
+        if (Objects.nonNull(in)) {
+            qList.add(in);
+        }
+        qList.add(time);
+        and.andOperator(qList);
+        // 杩斿洖鏌ヨ
+        return Query.query(and);
+    }
+
+    /**
+     * 鏋勯�犳暟鎹腑蹇冩悳绱㈢殑鏉′欢
+     *
+     * @param query 鏌ヨ鏉′欢
+     * @param timeFiled 鏃堕棿瀛楁鏄摢涓�涓�
+     * @param params 璇锋眰鍙傛暟
+     */
+    public static void setPage(Query query,DataCenterQuery params, String timeFiled) {
+        // 鎺掑簭鍒嗛〉
+        query.with(Sort.by(Sort.Order.desc(timeFiled))).skip(params.getSkipNum()).limit(Math.toIntExact(params.getPageSize()));
     }
 
 }

--
Gitblit v1.8.0