package com.monkeylessey.controller.system;
|
|
import com.google.code.kaptcha.Producer;
|
import com.monkeylessey.constant.RedisKeyExpireConstants;
|
import com.monkeylessey.constant.RedisKeyPrefixConstants;
|
import com.monkeylessey.sys.domain.vo.CaptchaVO;
|
import com.monkeylessey.response.Result;
|
import com.monkeylessey.framework.utils.RedisUtil;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.util.Base64Utils;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.ByteArrayOutputStream;
|
import java.io.IOException;
|
import java.util.UUID;
|
|
/**
|
* @author 29443
|
* @version 1.0
|
* @date 2022/4/21
|
*/
|
@RestController
|
@RequiredArgsConstructor
|
@Api(value = "登录相关", tags = "登录相关")
|
public class SysLoginController {
|
|
private final RedisUtil redisUtil;
|
private final Producer captcha;
|
|
|
@GetMapping("/captcha")
|
@ApiOperation(value = "获取验证码", notes = "获取验证码")
|
public Result getCaptcha() throws IOException {
|
// 生成验证码id
|
String captchaId = UUID.randomUUID().toString().substring(0, 10);
|
String key = RedisKeyPrefixConstants.CAPTCHA + captchaId;
|
|
// 生成验证码
|
String code = captcha.createText();
|
BufferedImage image = captcha.createImage(code);
|
|
// 创建字节数组缓冲
|
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
|
|
// 将生成的验证码BufferedImage转为byte数组
|
ImageIO.write(image, "jpg", byteArrayOutputStream);
|
byte[] bytes = byteArrayOutputStream.toByteArray();
|
|
// 将byte数组使用Base64编码
|
String base64 = Base64Utils.encodeToString(bytes);
|
|
// 将生成的验证码放入redis，默认有效2分钟
|
redisUtil.saveForValueWithExpire(key, code, RedisKeyExpireConstants.CAPTCHA_EXPIRE_TIME, RedisKeyExpireConstants.CAPTCHA_TIME_UNIT);
|
return Result.ok("获取验证码成功").data(new CaptchaVO(captchaId, "data:image/jpg;base64," + base64));
|
}
|
}
|
