| | |
|---|
| | | package com.xiaopeng.gateway.service.impl; |
|---|
| | | |
|---|
| | | import java.awt.image.BufferedImage; |
|---|
| | | import java.io.IOException; |
|---|
| | | import java.util.concurrent.TimeUnit; |
|---|
| | | import javax.annotation.Resource; |
|---|
| | | import javax.imageio.ImageIO; |
|---|
| | | |
|---|
| | | import org.springframework.beans.factory.annotation.Autowired; |
|---|
| | | import org.springframework.stereotype.Service; |
|---|
| | | import org.springframework.util.FastByteArrayOutputStream; |
|---|
| | | import com.google.code.kaptcha.Producer; |
|---|
| | | import com.xiaopeng.common.core.constant.CacheConstants; |
|---|
| | | import com.xiaopeng.common.core.constant.Constants; |
|---|
| | | import com.xiaopeng.common.core.exception.CaptchaException; |
|---|
| | | import com.xiaopeng.common.core.utils.StringUtils; |
|---|
| | | import com.xiaopeng.common.core.utils.sign.Base64; |
|---|
| | | import com.xiaopeng.common.core.utils.uuid.IdUtils; |
|---|
| | | import com.xiaopeng.common.core.web.domain.AjaxResult; |
|---|
| | | import com.xiaopeng.common.redis.service.RedisService; |
|---|
| | | import com.xiaopeng.gateway.config.properties.CaptchaProperties; |
|---|
| | | import com.xiaopeng.gateway.service.ValidateCodeService; |
|---|
| | | |
|---|
| | | /** |
|---|
| | |
|---|
| | | */ |
|---|
| | | @Service |
|---|
| | | public class ValidateCodeServiceImpl implements ValidateCodeService { |
|---|
| | | @Resource(name = "captchaProducer") |
|---|
| | | private Producer captchaProducer; |
|---|
| | | |
|---|
| | | @Resource(name = "captchaProducerMath") |
|---|
| | | private Producer captchaProducerMath; |
|---|
| | | |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private RedisService redisService; |
|---|
| | | |
|---|
| | | @Autowired |
|---|
| | | private CaptchaProperties captchaProperties; |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 生成验证码 |
|---|
| | | */ |
|---|
| | | @Override |
|---|
| | | public AjaxResult createCaptcha() throws IOException, CaptchaException { |
|---|
| | | AjaxResult ajax = AjaxResult.success(); |
|---|
| | | boolean captchaEnabled = captchaProperties.getEnabled(); |
|---|
| | | ajax.put("captchaEnabled", captchaEnabled); |
|---|
| | | if (!captchaEnabled) { |
|---|
| | | return ajax; |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 保存验证码信息 |
|---|
| | | String uuid = IdUtils.simpleUUID(); |
|---|
| | | String verifyKey = CacheConstants.CAPTCHA_CODE_KEY + uuid; |
|---|
| | | |
|---|
| | | String capStr = null, code = null; |
|---|
| | | BufferedImage image = null; |
|---|
| | | |
|---|
| | | String captchaType = captchaProperties.getType(); |
|---|
| | | // 生成验证码 |
|---|
| | | 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); |
|---|
| | | } |
|---|
| | | |
|---|
| | | redisService.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES); |
|---|
| | | // 转换流信息写出 |
|---|
| | | FastByteArrayOutputStream os = new FastByteArrayOutputStream(); |
|---|
| | | try { |
|---|
| | | ImageIO.write(image, "jpg", os); |
|---|
| | | } catch (IOException e) { |
|---|
| | | return AjaxResult.error(e.getMessage()); |
|---|
| | | } |
|---|
| | | |
|---|
| | | ajax.put("uuid", uuid); |
|---|
| | | ajax.put("img", Base64.encode(os.toByteArray())); |
|---|
| | | return ajax; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
| | | * 校验验证码 |
|---|
