123

ruoyi-cai/src/main/java/com/ruoyi/cai/service/SmsVerifyService.java

@@ -15,4 +15,6 @@ public interface SmsVerifyService extends IService<SmsVerify> {
     boolean check(CodeEnum codeEnum, String mobile, String code);
 
     void put(CodeEnum codeEnum, String mobile);
+
+    void clearSmsVerify(int days);
 }

ruoyi-cai/src/main/java/com/ruoyi/cai/service/UserChatRecordService.java

@@ -14,4 +14,6 @@ import com.ruoyi.cai.dto.app.dto.ImMessageDTO;
 public interface UserChatRecordService extends IService<UserChatRecord> {
 
     UserChatRecord saveRecord(User fromUser, User toUser,Long traceId, ImMessageDTO message);
+
+    void clearChatRecord(Integer hours);
 }

ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/SmsVerifyServiceImpl.java

@@ -17,12 +17,15 @@ import com.ruoyi.cai.service.UserInfoService;
 import com.ruoyi.cai.service.UserService;
 import com.ruoyi.common.exception.ServiceException;
 import com.ruoyi.common.utils.ServletUtils;
+import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
 import java.time.temporal.ChronoUnit;
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 短信验证码Service业务层处理
@@ -116,4 +119,21 @@ public class SmsVerifyServiceImpl extends ServiceImpl<SmsVerifyMapper,SmsVerify>
 //        String code = codeManager.put(codeEnum, mobile);
     }
 
+    @Override
+    public void clearSmsVerify(int days) {
+        int index = 0;
+        LocalDateTime deleteTimeFlag = LocalDateTime.now().plusDays(-days);
+        while (true){
+            List<SmsVerify> list = this.list(Wrappers.lambdaUpdate(SmsVerify.class)
+                    .lt(SmsVerify::getCreateTime, deleteTimeFlag)
+                    .last("limit 100"));
+            if(index > 500 || CollectionUtils.isEmpty(list)){
+                break;
+            }
+            index++;
+            List<Long> ids = list.stream().map(SmsVerify::getId).collect(Collectors.toList());
+            this.removeBatchByIds(ids);
+        }
+    }
+
 }

ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserChatRecordServiceImpl.java

@@ -1,5 +1,6 @@
 package com.ruoyi.cai.service.impl;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.cai.domain.User;
 import com.ruoyi.cai.domain.UserChatRecord;
@@ -7,9 +8,12 @@ import com.ruoyi.cai.dto.app.dto.ImMessageDTO;
 import com.ruoyi.cai.mapper.UserChatRecordMapper;
 import com.ruoyi.cai.service.UserChatRecordService;
 import lombok.RequiredArgsConstructor;
+import org.apache.commons.collections4.CollectionUtils;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
+import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 聊天记录Service业务层处理
@@ -35,4 +39,21 @@ public class UserChatRecordServiceImpl extends ServiceImpl<UserChatRecordMapper,
         this.save(userChatRecord);
         return userChatRecord;
     }
+
+    @Override
+    public void clearChatRecord(Integer hours) {
+        int index = 0;
+        LocalDateTime deleteTimeFlag = LocalDateTime.now().plusHours(-hours);
+        while (true){
+            List<UserChatRecord> list = this.list(Wrappers.lambdaUpdate(UserChatRecord.class)
+                    .lt(UserChatRecord::getCreateTime, deleteTimeFlag)
+                    .last("limit 100"));
+            if(index > 500 || CollectionUtils.isEmpty(list)){
+                break;
+            }
+            index++;
+            List<Long> ids = list.stream().map(UserChatRecord::getId).collect(Collectors.toList());
+            this.removeBatchByIds(ids);
+        }
+    }
 }

ruoyi-cai/src/main/java/com/ruoyi/cai/service/impl/UserServiceImpl.java

@@ -124,12 +124,13 @@ public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements Us
         vo.setAvatar(user.getAvatar());
         vo.setPrice(0L);
         vo.setServiceCount(0L);
+        vo.setOrderSwitch(1);
         if(anchor != null){
             vo.setPrice(anchor.getPrice());
             if(anchor.getOrderSwitch() == 2){ // 不隐藏接单数
                 vo.setServiceCount(anchor.getServiceCount());
             }
-
+            vo.setOrderSwitch(anchor.getOrderSwitch());
         }
         vo.setCity(user.getCity());
         vo.setCityId(user.getCityId());