接口访问次数限制

src/main/java/com/sqx/common/utils/ApiAccessLimitUtil.java

@@ -1,13 +1,9 @@
 package com.sqx.common.utils;
 
+import cn.hutool.core.date.DateUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
-
-import java.time.DayOfWeek;
-import java.time.LocalDate;
-import java.time.ZoneOffset;
-import java.time.temporal.TemporalAdjusters;
-import java.util.Objects;
+import java.util.Date;
 
 
 /**
@@ -117,7 +113,7 @@ public class ApiAccessLimitUtil {
             redisUtils.set(redisKey, 1, expireAt);
             return true;
         }
-        if ((int) countObj < count) {
+        if (Integer.parseInt(countObj.toString()) < count) {
             // 访问次数未达上限，次数加1
             redisUtils.incr(redisKey);
             return true;
@@ -131,17 +127,19 @@ public class ApiAccessLimitUtil {
     }
 
     private static long calculateExpireAt(String timePeriod) {
-        LocalDate now = LocalDate.now();
-        LocalDate expireDate = null;
+        Date now = DateUtil.beginOfDay(DateUtil.date());
+        Date expireDate = null;
         if ("day".equals(timePeriod)) {
-            expireDate = now.plusDays(1).atStartOfDay().toLocalDate();
+            expireDate =  DateUtil.endOfDay(now);
         } else if ("week".equals(timePeriod)) {
-            expireDate = now.plusWeeks(0).with(TemporalAdjusters.nextOrSame(java.time.DayOfWeek.SUNDAY));
+            expireDate = DateUtil.endOfWeek(now);
         } else if ("month".equals(timePeriod)) {
-            expireDate = now.plusMonths(1).withDayOfMonth(1).minusDays(1);
+            expireDate = DateUtil.endOfMonth(now);
         } else if ("year".equals(timePeriod)) {
-            expireDate = now.plusYears(1).withDayOfYear(1).minusDays(1);
+            expireDate = DateUtil.endOfYear(now);
         }
-        return Objects.requireNonNull(expireDate).atTime(23, 59, 59).toEpochSecond(ZoneOffset.UTC);
+        long endTimeStamp = DateUtil.endOfDay(expireDate).getTime() / 1000L;
+        long currentTimeStamp = DateUtil.currentSeconds();
+        return endTimeStamp - currentTimeStamp;
     }
 }

src/main/java/com/sqx/modules/app/controller/app/AppController.java

@@ -2,6 +2,7 @@ package com.sqx.modules.app.controller.app;
 
 
 import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.sqx.common.annotation.Debounce;
 import com.sqx.common.utils.ApiAccessLimitUtil;
 import com.sqx.common.utils.Result;
@@ -10,6 +11,7 @@ import com.sqx.modules.app.annotation.LoginUser;
 import com.sqx.modules.app.entity.UserEntity;
 import com.sqx.modules.app.service.AppService;
 import com.sqx.modules.app.service.UserService;
+import com.sqx.modules.message.entity.MessageInfo;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.apache.commons.codec.digest.DigestUtils;
@@ -20,7 +22,6 @@ import javax.servlet.http.HttpServletRequest;
 
 /**
  * APP登录授权
- *
  */
 @RestController
 @RequestMapping("/app/user")
@@ -74,8 +75,16 @@ public class AppController {
     @ApiOperation("用户修改个人信息")
     @ResponseBody
     public Result updateUserImageUrl(@RequestAttribute("userId") Long userId, String zhiFuBao, String zhiFuBaoName) {
-        if(!ApiAccessLimitUtil.isAccessAllowed(userId.toString(), "updateZFB", 2, "month")){
-            return Result.error("每月仅支持修改两次,请联系管理员");
+        int count = userService.count(new QueryWrapper<UserEntity>()
+                .ne( "user_id", userId)
+                .eq("zhi_fu_bao_name", zhiFuBaoName)
+                .eq("zhi_fu_bao", zhiFuBao));
+
+        if (count > 0) {
+            return Result.error("一个支付宝账号仅可绑定一个支付宝用户");
+        }
+        if (!ApiAccessLimitUtil.isAccessAllowed(userId.toString(), "updateZFB", 3, "month")) {
+            return Result.error("每月仅支持修改三次,请联系管理员");
         }
         UserEntity userEntity = new UserEntity();
         userEntity.setZhiFuBao(zhiFuBao);
@@ -86,7 +95,6 @@ public class AppController {
     }
 
 
-
     @Login
     @RequestMapping(value = "/updateUsers", method = RequestMethod.POST)
     @ApiOperation("用户修改个人信息")

src/main/resources/application.yml

@@ -34,7 +34,7 @@ spring:
     database: 0
     host: localhost
     port: 6379
-    password:     # 密码（默认为空）
+    password: 111111    # 密码（默认为空）
     timeout: 6000ms  # 连接超时时长（毫秒）
     jedis:
       pool: