package com.gkhy.hazmat.admin.controller.system;
|
|
import cn.hutool.core.codec.Base64;
|
import cn.hutool.core.lang.UUID;
|
import com.gkhy.hazmat.common.api.CommonResult;
|
import com.gkhy.hazmat.common.constant.CacheConstant;
|
import com.gkhy.hazmat.common.utils.RedisUtils;
|
import com.google.code.kaptcha.Producer;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.FastByteArrayOutputStream;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.annotation.Resource;
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
import java.util.HashMap;
|
import java.util.Map;
|
import java.util.concurrent.TimeUnit;
|
|
|
@Api(tags = "验证码前端控制器")
|
@RestController
|
@RequestMapping("/system/captcha")
|
public class CaptchaController
|
{
|
@Resource(name = "captchaProducer")
|
private Producer captchaProducer;
|
|
@Resource(name = "captchaProducerMath")
|
private Producer captchaProducerMath;
|
|
|
@Autowired
|
private RedisUtils redisUtils;
|
|
/**
|
* 生成验证码
|
*/
|
@ApiOperation(value = "生成验证码")
|
@GetMapping("/captchaImage")
|
public CommonResult captchaImage(){
|
Map<String,Object> map=new HashMap<>();
|
|
// 保存验证码信息
|
String uuid = UUID.randomUUID().toString();
|
String verifyKey = CacheConstant.CAPTCHA_CODE_KEY + uuid;
|
|
String capStr = null, code = null;
|
BufferedImage image = null;
|
|
// 生成验证码
|
String captchaType = "math";
|
if ("math".equals(captchaType))
|
{
|
String capText = captchaProducerMath.createText();
|
capStr = capText.substring(0, capText.lastIndexOf("@"));
|
code = capText.substring(capText.lastIndexOf("@") + 1);
|
image = captchaProducerMath.createImage(capStr);
|
}
|
else if ("char".equals(captchaType))
|
{
|
capStr = code = captchaProducer.createText();
|
image = captchaProducer.createImage(capStr);
|
}
|
|
redisUtils.set(verifyKey, code, 2, TimeUnit.MINUTES);
|
// 转换流信息写出
|
FastByteArrayOutputStream os = new FastByteArrayOutputStream();
|
try
|
{
|
ImageIO.write(image, "jpg", os);
|
}
|
catch (IOException e)
|
{
|
return CommonResult.failed(e.getMessage());
|
}
|
|
map.put("uuid",uuid);
|
map.put("image", Base64.encode(os.toByteArray()));
|
return CommonResult.success(map);
|
}
|
}
|
