package com.gkhy.exam.admin.system; import cn.hutool.core.codec.Base64; import cn.hutool.core.lang.UUID; import com.gkhy.exam.common.api.CommonResult; import com.gkhy.exam.common.constant.CacheConstant; import com.gkhy.exam.common.utils.RedisUtils; import com.gkhy.exam.system.service.SysConfigService; 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 SysConfigService configService; @Autowired private RedisUtils redisUtils; /** * 生成验证码 */ @ApiOperation(value = "生成验证码") @GetMapping("/captchaImage") public CommonResult captchaImage(){ Map map=new HashMap<>(); boolean captchaEnabled=configService.selectCaptchaEnabled(); map.put("captchaEnabled", captchaEnabled); if(!captchaEnabled){ return CommonResult.success(captchaEnabled); } // 保存验证码信息 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); } }