setTotal

liquidnet-bus-service/liquidnet-service-sweet/src/main/java/com/liquidnet/service/sweet/service/impl/SweetIntegralActivityDrawServiceImpl.java

@@ -187,23 +187,23 @@ public class SweetIntegralActivityDrawServiceImpl extends ServiceImpl<SweetInteg
         String currentUid = CurrentUtil.getCurrentUid();
         PagedResult<IntegralActivityDrawVo> sweetIntegralActivityDrawPagedResult = SweetNewObjectUtil.getSweetIntegralActivityDrawVoPagedResult();
         long count = 0;
+        //条件
+        Query query = Query.query(
+                Criteria.where("integralActivityId").is(integralActivityId).and("userId").is(currentUid)
+        );
+        if (type == 1) {
+            query.addCriteria(Criteria.where("prizeType").ne(1));
+        } else {
+            query.addCriteria(Criteria.where("prizeType").is(1));
+        }
         if (pageNum <= 1) {
             List<IntegralActivityDrawVo> integralActivityDrawAll = redisDataUtils.getIntegralActivityDrawList(type, currentUid, integralActivityId);
-            count = integralActivityDrawAll.size();
+            count = mongoTemplate.count(query, IntegralActivityDrawVo.class, IntegralActivityDrawVo.class.getSimpleName());
             sweetIntegralActivityDrawPagedResult.setList(integralActivityDrawAll).setTotal(count, 20).setPageSize(20).setCurrentPage(pageNum);
         } else {// page大于1 去mongo里面取
             HashMap<String, Object> info = CollectionUtil.mapStringObject();
             // 排序 分页
             Pageable pageable = PageRequest.of(pageNum - 1, 20, Sort.by(Sort.Direction.DESC, "createdAt"));
-            //条件
-            Query query = Query.query(
-                    Criteria.where("integralActivityId").is(integralActivityId).and("userId").is(currentUid)
-            );
-            if (type == 1) {
-                query.addCriteria(Criteria.where("prizeType").ne(1));
-            } else {
-                query.addCriteria(Criteria.where("prizeType").is(1));
-            }
             count = mongoTemplate.count(query, IntegralActivityDrawVo.class, IntegralActivityDrawVo.class.getSimpleName());
             query.with(pageable);
             List<IntegralActivityDrawVo> integralActivityDrawVos = mongoTemplate.find(query, IntegralActivityDrawVo.class, IntegralActivityDrawVo.class.getSimpleName());