94 lines
3.5 KiB
Java
94 lines
3.5 KiB
Java
package xin.merlin.myblog_server.controller;
|
|
|
|
|
|
import com.github.benmanes.caffeine.cache.Cache;
|
|
import com.github.benmanes.caffeine.cache.Caffeine;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
import xin.merlin.myblog_server.entity.Code;
|
|
import xin.merlin.myblog_server.entity.User;
|
|
import xin.merlin.myblog_server.service.CacheService;
|
|
import xin.merlin.myblog_server.service.impl.MailService;
|
|
import xin.merlin.myblog_server.utils.RandomCode;
|
|
import xin.merlin.myblog_server.utils.RequestBack;
|
|
import xin.merlin.myblog_server.utils.enums.ResultCode;
|
|
|
|
import java.util.Map;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
@RestController
|
|
@RequestMapping("/code")
|
|
public class MailController {
|
|
@Autowired
|
|
private CacheService cacheService;
|
|
|
|
// 冷却缓存：限制邮箱请求频率
|
|
private static final Cache<String, Boolean> emailCooldown = Caffeine.newBuilder()
|
|
.expireAfterWrite(60, TimeUnit.SECONDS) // 冷却 60 秒
|
|
.build();
|
|
|
|
// 验证码验证次数
|
|
private static final Cache<String, Integer> codeFailCount = Caffeine.newBuilder()
|
|
.expireAfterWrite(5, TimeUnit.MINUTES)
|
|
.build();
|
|
|
|
@Autowired
|
|
private MailService mailService;
|
|
|
|
@PostMapping("/sendcode")
|
|
RequestBack sendcode(@RequestBody User user) {
|
|
if (user.getAccount() == null) return RequestBack.fail(ResultCode.BAD_REQUEST);
|
|
System.out.println("发送验证码到：" + user.getAccount());
|
|
String email = user.getAccount();
|
|
|
|
// 检查是否在冷却中
|
|
if (emailCooldown.getIfPresent(email) != null) {
|
|
return RequestBack.fail(ResultCode.USER_SEND_TOO_FAST);
|
|
}
|
|
String tempId;
|
|
do {
|
|
tempId = RandomCode.generateCode();
|
|
}
|
|
while (cacheService.getWaitingList().getIfPresent(tempId) != null);
|
|
|
|
try {
|
|
cacheService.getWaitingList().put(tempId, mailService.sendMail(user.getAccount()));
|
|
// 加入验证码冷却
|
|
emailCooldown.put(user.getAccount(), true);
|
|
return RequestBack.success(ResultCode.SUCCESS, Map.of("c_id", tempId));
|
|
} catch (Exception e) {
|
|
return RequestBack.fail(ResultCode.SERVER_ERROR);
|
|
}
|
|
|
|
}
|
|
|
|
@PostMapping("/verifycode")
|
|
RequestBack verifyCode(@RequestBody Code code) {
|
|
//测试验证码
|
|
String id=code.getC_id();
|
|
|
|
if(codeFailCount.getIfPresent(id) == null) codeFailCount.put(id, 1);
|
|
else if(codeFailCount.getIfPresent(id)<=5){
|
|
Integer time = codeFailCount.getIfPresent(id);
|
|
codeFailCount.put(id, ++time);
|
|
}else{
|
|
return RequestBack.fail(ResultCode.USER_SEND_TOO_OFTEN);
|
|
}
|
|
|
|
|
|
String tempCode = cacheService.getWaitingList().getIfPresent(id);
|
|
System.out.println("cacheService.getWaitingList()" + tempCode + "\nv_id:" + id + "\ncode:" + code.getCode());
|
|
if (tempCode == null) return RequestBack.success(ResultCode.USER_VERIFICATION_ERROR);
|
|
if (!tempCode.equals(code.getCode())) return RequestBack.success(ResultCode.USER_VERIFICATION_ERROR);
|
|
// cacheService.getWaitingList().invalidate(id);
|
|
codeFailCount.invalidate(id);
|
|
emailCooldown.invalidate(code.getAccount());
|
|
|
|
return RequestBack.success(ResultCode.SUCCESS);
|
|
|
|
}
|
|
}
|