package com.pollex.pam.web.rest;
|
|
import java.awt.image.BufferedImage;
|
import java.io.ByteArrayOutputStream;
|
import java.io.IOException;
|
|
import javax.imageio.ImageIO;
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import javax.servlet.http.HttpSession;
|
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import com.pollex.pam.business.service.util.VerifyCodeUtil;
|
|
@RestController
|
@RequestMapping("/api/login/validate")
|
public class ConsultantLoginValidateResource {
|
|
@GetMapping("/get_img_code")
|
public void getVerifyCodeImg(HttpServletResponse response, HttpServletRequest request) {
|
try {
|
String code = VerifyCodeUtil.createcode();
|
BufferedImage image = VerifyCodeUtil.createimage(code);
|
// Return to the client in a stream
|
response.setContentType("image/jpeg");
|
// response.setContentType("image/bmp");
|
ByteArrayOutputStream bt = new ByteArrayOutputStream();
|
// Convert pictures to byte stream
|
ImageIO.write(image, "jpeg", bt);
|
ServletOutputStream outputStream = response.getOutputStream();
|
outputStream.write(bt.toByteArray());
|
HttpSession session = request.getSession();
|
session.setAttribute("img_code", code);
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
|
}
|
|
@GetMapping("/verify_img_code/{imgCode}")
|
public boolean verifyImgCode(HttpServletResponse response, HttpServletRequest request,
|
@PathVariable String imgCode) {
|
HttpSession session = request.getSession();
|
String sessionImpCode = (String) session.getAttribute("img_code");
|
if (!StringUtils.hasText(sessionImpCode)
|
|| !StringUtils.hasText(imgCode)) {
|
return false;
|
}
|
|
if (imgCode.equals(sessionImpCode)) {
|
return true;
|
}
|
|
return false;
|
}
|
}
|
