feat: 优化更新执行的预约分钟

oddfar · Feb 21, 2024 · 798bc3a · 798bc3a
1 parent 0c7de21
commit 798bc3a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/campus-modular/src/main/java/com/oddfar/campus/business/mapper/IUserMapper.java b/campus-modular/src/main/java/com/oddfar/campus/business/mapper/IUserMapper.java
@@ -4,7 +4,7 @@
 import com.oddfar.campus.common.core.BaseMapperX;
 import com.oddfar.campus.common.core.LambdaQueryWrapperX;
 import com.oddfar.campus.common.domain.PageResult;
-import org.apache.ibatis.annotations.Select;
+import org.apache.ibatis.annotations.Update;
 
 import java.util.List;
 
@@ -66,8 +66,14 @@ default List<IUser> selectReservationUserByMinute(int minute) {
         );
     }
 
-    @Select("UPDATE i_user SET `minute` = (SELECT FLOOR(RAND() * 50 + 1)) WHERE random_minute = \"0\"")
+    @Update("UPDATE i_user SET `minute` = (SELECT FLOOR(RAND() * 50 + 1)) WHERE random_minute = \"0\"")
     void updateUserMinuteBatch();
 
+    @Update("SET @row_number = 0;\n" +
+            "UPDATE i_user\n" +
+            "SET `minute` = (@row_number := @row_number + 1) % 50 + 1\n" +
+            "ORDER BY RAND();")
+    void updateUserMinuteEven();
+
     int deleteIUser(Long[] iUserId);
 }
diff --git a/campus-modular/src/main/java/com/oddfar/campus/business/service/impl/IUserServiceImpl.java b/campus-modular/src/main/java/com/oddfar/campus/business/service/impl/IUserServiceImpl.java
@@ -91,7 +91,12 @@ public int updateIUser(IUser iUser) {
     @Override
     @Async
     public void updateUserMinuteBatch() {
-        iUserMapper.updateUserMinuteBatch();
+        Long userCount = iUserMapper.selectCount();
+        if (userCount > 60) {
+            iUserMapper.updateUserMinuteEven();
+        }else {
+            iUserMapper.updateUserMinuteBatch();
+        }
     }
 
     @Override