nnnn

ruoyi-admin/src/main/java/com/ruoyi/job/IpJob.java

@@ -0,0 +1,41 @@
+package com.ruoyi.job;
+
+import com.ruoyi.cai.enums.SystemConfigEnum;
+import com.ruoyi.cai.enums.rank.RankDataTypeEnum;
+import com.ruoyi.cai.manager.SystemConfigManager;
+import com.ruoyi.cai.mapper.IpBlackMapper;
+import com.ruoyi.cai.service.AnchorImCountDayService;
+import com.ruoyi.cai.service.FastPayTotalService;
+import com.ruoyi.cai.service.UnionTotalService;
+import com.ruoyi.job.op.BusOp;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.Resource;
+import java.time.DayOfWeek;
+import java.time.LocalDate;
+import java.time.LocalDateTime;
+
+@Component
+@Slf4j
+public class IpJob {
+    @Resource
+    private IpBlackMapper ipBlackMapper;
+    @Autowired
+    private SystemConfigManager systemConfigManager;
+    // 凌晨0点3分执行一次
+    @Scheduled(cron = "0/5 * * * * ? ")
+    public void run() {
+        boolean openIpAuto = systemConfigManager.getSystemConfigOfBool(SystemConfigEnum.OPEN_IP_AUTO);
+        if(!openIpAuto){
+            return;
+        }
+        Integer ipNumber = systemConfigManager.getSystemConfigOfInt(SystemConfigEnum.OPEN_IP_NUMBER);
+        int refreshed = ipBlackMapper.refreshIp(ipNumber, LocalDate.now());
+        if(refreshed > 0){
+            log.info("封禁IP成功！个数:{}",refreshed);
+        }
+    }
+}

ruoyi-admin/src/main/java/com/ruoyi/web/controller/cai/app/AuthAppController.java

@@ -57,7 +57,7 @@ public class AuthAppController {
     @Operation(summary = "注册")
     @Log(title = "注册", businessType = BusinessType.OTHER, isSaveDb = true)
     public R<LoginVo> register(@Validated @RequestBody CaiRegisterUser caiUser){
-        boolean openOldRegisterCode = systemConfigManager.getSystemConfigOfBool(SystemConfigEnum.OPEN_OLD_REGISTER_CODE);
+        boolean openOldRegisterCode = systemConfigManager.getSystemConfigOfBool(SystemConfigEnum.OPEN_REGISTER);
         if(!openOldRegisterCode){
             return R.fail(600,"未开放注册");
         }
@@ -138,7 +138,17 @@ public class AuthAppController {
         if(!mobile){
             return R.fail(600,"请输入正确的手机格式");
         }
-        smsVerifyService.put(CodeEnum.RESET_PASSWORD,code.getMobile());
+        boolean openOldRegisterCode = systemConfigManager.getSystemConfigOfBool(SystemConfigEnum.OPEN_RESET_PASSWORD);
+        if(!openOldRegisterCode){
+            return R.fail(600,"9000009");
+        }
+        ipBlackService.checkIpThrowException(ServletUtils.getClientIP());
+        try {
+            smsVerifyService.put(CodeEnum.RESET_PASSWORD,code.getMobile());
+        }catch (Exception e){
+            ipRecordService.saveLoginIp(ServletUtils.getClientIP());
+            throw e;
+        }
         return R.ok("发送成功");
     }
 
@@ -154,6 +164,7 @@ public class AuthAppController {
     @Log(title = "登陆", businessType = BusinessType.OTHER, isSaveDb = false)
     public R<LoginVo> login(@Validated @RequestBody LoginCaiUser loginBody){
         LoginVo vo = new LoginVo();
+        ipBlackService.checkIpThrowException(ServletUtils.getClientIP());
         try {
             String token = caiLoginManager.login(loginBody.getUsername(), loginBody.getPassword());
             vo.setToken(token);