整理不易，点点小编，一键三连走一波~~~ 

笔记来自《小滴课堂-新版Redis6学习笔记》

第2集 案例实战之企业级-验证码存储Redis和工具类介绍

简介：验证码存储Redis逻辑编码实战

• CommonUtil工具类

    /**

     * 获取ip

     * @param request

     * @return

     */

    public static String getIpAddr(HttpServletRequest request) {

        String ipAddress = null;

        try {

            ipAddress = request.getHeader("x-forwarded-for");

            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getHeader("Proxy-Client-IP");

            }

            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getHeader("WL-Proxy-Client-IP");

            }

            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {

                ipAddress = request.getRemoteAddr();

                if (ipAddress.equals("127.0.0.1")) {

                    // 根据网卡取本机配置的IP

                    InetAddress inet = null;

                    try {

                        inet = InetAddress.getLocalHost();

                    } catch (UnknownHostException e) {

                        e.printStackTrace();

                    }

                    ipAddress = inet.getHostAddress();

                }

            }

            // 对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照','分割

            if (ipAddress != null && ipAddress.length() > 15) {

                // "***.***.***.***".length()

                // = 15

                if (ipAddress.indexOf(",") > 0) {

                    ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));

                }

            }

        } catch (Exception e) {

            ipAddress="";

        }

        return ipAddress;

    }

​

​

​

    public static String MD5(String data)  {

        try {

            java.security.MessageDigest md = MessageDigest.getInstance("MD5");

            byte[] array = md.digest(data.getBytes("UTF-8"));

            StringBuilder sb = new StringBuilder();

            for (byte item : array) {

                sb.append(Integer.toHexString((item & 0xFF) | 0x100).substring(1, 3));

            }

            return sb.toString().toUpperCase();

        } catch (Exception exception) {

        }

        return null;

​

    }

• 接口开发

    @Autowired

    private StringRedisTemplate redisTemplate;

​

    @Autowired

    private Producer captchaProducer;

​

    /**

     *临时使用10分钟有效，方便测试

     */

    private static final long CAPTCHA_CODE_EXPIRED = 60 * 1000 * 10;

​

    /**

     * 获取图形验证码

     * @param request

     * @param response

     */

    @GetMapping("captcha")

    public void getCaptcha(HttpServletRequest request, HttpServletResponse response){

​

        String captchaText = captchaProducer.createText();

​

        //存储

        redisTemplate.opsForValue().set(getCaptchaKey(request),captchaText,CAPTCHA_CODE_EXPIRED,TimeUnit.MILLISECONDS);

​

        BufferedImage bufferedImage = captchaProducer.createImage(captchaText);

        ServletOutputStream outputStream = null;

        try {

            outputStream = response.getOutputStream();

            ImageIO.write(bufferedImage,"jpg",outputStream);

            outputStream.flush();

            outputStream.close();

        } catch (IOException e) {

        }

​

    }

​

    /**

     * 获取缓存的key

     * @param request

     * @return

     */

    private String getCaptchaKey(HttpServletRequest request){

​

        String ip = CommonUtils.getIpAddr(request);

        String userAgent = request.getHeader("User-Agent");

​

        String key = "user-service:captcha:"+CommonUtils.MD5(ip+userAgent);

​

        return key;

​

    }

 

愿景："让编程不再难学，让技术与生活更加有趣"

更多架构课程请访问 xdclass.net

想要了解更多？关注公众号领取更多干货！！！

前端、后端、测试等方面，各种干货等你来拿