package com.gkhy.assess.system.service.impl;
|
|
import cn.hutool.core.codec.Base64;
|
import com.gkhy.assess.common.constant.CacheConstant;
|
import com.gkhy.assess.common.exception.ApiException;
|
import com.gkhy.assess.common.utils.RedisUtils;
|
import com.gkhy.assess.system.domain.vo.CaptchaVO;
|
import com.gkhy.assess.system.service.CaptchaService;
|
import com.google.code.kaptcha.Producer;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.FastByteArrayOutputStream;
|
|
import javax.annotation.Resource;
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.IOException;
|
import java.util.UUID;
|
import java.util.concurrent.TimeUnit;
|
|
@Slf4j
|
@Service
|
public class CaptchaServiceImpl implements CaptchaService {
|
@Resource(name = "captchaProducer")
|
private Producer captchaProducer;
|
|
@Resource(name = "captchaProducerMath")
|
private Producer captchaProducerMath;
|
@Autowired
|
private RedisUtils redisUtils;
|
|
@Override
|
public CaptchaVO captchaImage() {
|
String uuid= UUID.randomUUID().toString();
|
String verifyKey= CacheConstant.CAPTCHA_CODE_KEY+uuid;
|
String capText=captchaProducerMath.createText();
|
String capStr=capText.substring(0,capText.lastIndexOf("@"));
|
String code=capText.substring(capText.lastIndexOf("@")+1);
|
BufferedImage image=captchaProducerMath.createImage(capStr);
|
redisUtils.set(verifyKey,code,2, TimeUnit.MINUTES);
|
// 转换流信息写出
|
FastByteArrayOutputStream os = new FastByteArrayOutputStream();
|
try
|
{
|
ImageIO.write(image, "jpg", os);
|
}
|
catch (IOException e)
|
{
|
log.error(e.getMessage());
|
throw new ApiException("生成验证码失败");
|
}
|
CaptchaVO captchaVO=new CaptchaVO();
|
captchaVO.setUuid(uuid);
|
captchaVO.setImage(Base64.encode(os.toByteArray()));
|
return captchaVO;
|
}
|
|
|
}
|
