fuint-application/application-web/src/main/java/com/fuint/application/web/rest/CaptchaApiController.java

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

package com.fuint.application.web.rest;

import com.fuint.application.BaseController;
import com.fuint.application.ResponseObject;
import com.fuint.application.util.Base64Util;
import com.fuint.captcha.service.CaptchaService;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import javax.imageio.ImageIO;
import java.io.ByteArrayOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

/**
 * 图形验证码控制类
 * Created by FSQ
 * Contact wx fsq_better
 * Site https://www.fuint.cn
 */
@RestController
@RequestMapping("/rest/captcha")
public class CaptchaApiController extends BaseController {

    @Resource
    private CaptchaService captchaService;

    @RequestMapping("/getCode")
    @CrossOrigin
    public ResponseObject getCode(HttpServletRequest request, HttpServletResponse response) {
        String captcha = "";
        try {
            HttpSession session = request.getSession();
            BufferedImage image = captchaService.getCode(session);
            // 输出流
            ByteArrayOutputStream stream = new ByteArrayOutputStream();
            ImageIO.write(image, "png", stream);
            captcha = new String(Base64Util.baseEncode(stream.toByteArray()), "UTF-8");
        } catch (IOException e) {
            e.printStackTrace();
        }

        response.setDateHeader("Expires", 0);
        response.setHeader("Pragma", "No-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.addHeader("Cache-Control", "no-cache");

        Map<String, Object> outParams = new HashMap<String, Object>();
        outParams.put("captcha", "data:image/jpg;base64,"+captcha);

        return getSuccessResult(outParams);
    }

    @RequestMapping("/checkCode")
    @CrossOrigin
    public ResponseObject checkCode(@RequestParam String code, HttpServletRequest request) {