diff --git a/epmet-auth/src/main/java/com/epmet/controller/LoginController.java b/epmet-auth/src/main/java/com/epmet/controller/LoginController.java index 849daa1ca5..f1fb644703 100644 --- a/epmet-auth/src/main/java/com/epmet/controller/LoginController.java +++ b/epmet-auth/src/main/java/com/epmet/controller/LoginController.java @@ -3,26 +3,56 @@ package com.epmet.controller; import com.epmet.common.token.dto.form.LoginByPassWordFormDTO; import com.epmet.common.token.dto.form.LoginByWxCodeFormDTO; import com.epmet.common.token.dto.result.UserTokenResultDTO; +import com.epmet.commons.tools.exception.ErrorCode; import com.epmet.commons.tools.utils.Result; +import com.epmet.commons.tools.validator.AssertUtils; import com.epmet.commons.tools.validator.ValidatorUtils; +import com.epmet.service.CaptchaService; import com.epmet.service.LoginService; import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.web.bind.annotation.PostMapping; -import org.springframework.web.bind.annotation.RequestBody; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; + +import javax.imageio.ImageIO; +import javax.servlet.ServletOutputStream; +import javax.servlet.http.HttpServletResponse; +import java.awt.image.BufferedImage; +import java.io.IOException; /** - * @Description 通用登陆接口 + * @Description 通用登录接口 * @Author yinzuomei * @Date 2020/3/14 13:58 */ @RestController @RequestMapping("login") public class LoginController { + @Autowired + private CaptchaService captchaService; + @Autowired private LoginService loginService; + /** + * @return void + * @param response + * @param uuid + * @Author yinzuomei + * @Description 生成验证码 + * @Date 2020/3/17 16:08 + **/ + @GetMapping("captcha") + public void captcha(HttpServletResponse response, String uuid) throws IOException { + //uuid不能为空 + AssertUtils.isBlank(uuid, ErrorCode.IDENTIFIER_NOT_NULL); + //生成图片验证码 + BufferedImage image = captchaService.create(uuid); + response.setHeader("Cache-Control", "no-store, no-cache"); + response.setContentType("image/jpeg"); + ServletOutputStream out = response.getOutputStream(); + ImageIO.write(image, "jpg", out); + out.close(); + } + /** * @param formDTO * @return com.epmet.commons.tools.utils.Result diff --git a/epmet-auth/src/main/java/com/epmet/redis/CaptchaRedis.java b/epmet-auth/src/main/java/com/epmet/redis/CaptchaRedis.java index abd3da1d06..2eea36e2a5 100644 --- a/epmet-auth/src/main/java/com/epmet/redis/CaptchaRedis.java +++ b/epmet-auth/src/main/java/com/epmet/redis/CaptchaRedis.java @@ -10,6 +10,8 @@ package com.epmet.redis; import com.epmet.commons.tools.redis.RedisKeys; import com.epmet.commons.tools.redis.RedisUtils; +import org.apache.logging.log4j.LogManager; +import org.apache.logging.log4j.Logger; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Component; @@ -21,6 +23,7 @@ import org.springframework.stereotype.Component; */ @Component public class CaptchaRedis { + private Logger logger = LogManager.getLogger(CaptchaRedis.class); /** * 验证码5分钟过期 */ @@ -30,13 +33,14 @@ public class CaptchaRedis { public void set(String uuid, String captcha){ String key = RedisKeys.getLoginCaptchaKey(uuid); + logger.info("保存验证码key=["+key+"]"); redisUtils.set(key, captcha, EXPIRE); } public String get(String uuid){ String key = RedisKeys.getLoginCaptchaKey(uuid); String captcha = (String)redisUtils.get(key); - + logger.info("获取验证码key=["+key+"]captcha=["+captcha+"]"); //删除验证码 if(captcha != null){ redisUtils.delete(key); diff --git a/epmet-auth/src/main/java/com/epmet/service/impl/CaptchaServiceImpl.java b/epmet-auth/src/main/java/com/epmet/service/impl/CaptchaServiceImpl.java index d9a0c39d18..672fe52c68 100644 --- a/epmet-auth/src/main/java/com/epmet/service/impl/CaptchaServiceImpl.java +++ b/epmet-auth/src/main/java/com/epmet/service/impl/CaptchaServiceImpl.java @@ -11,6 +11,8 @@ package com.epmet.service.impl; import com.google.code.kaptcha.Producer; import com.epmet.redis.CaptchaRedis; import com.epmet.service.CaptchaService; +import org.apache.logging.log4j.LogManager; +import org.apache.logging.log4j.Logger; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; @@ -24,6 +26,7 @@ import java.awt.image.BufferedImage; */ @Service public class CaptchaServiceImpl implements CaptchaService { + private Logger logger = LogManager.getLogger(CaptchaServiceImpl.class); @Autowired private Producer producer; @Autowired @@ -33,7 +36,7 @@ public class CaptchaServiceImpl implements CaptchaService { public BufferedImage create(String uuid) { //生成验证码 String captcha = producer.createText(); - + logger.info("uuid:"+uuid+",生成的验证码:"+captcha); //保存验证码 captchaRedis.set(uuid, captcha);