com.kqinfo.universal.imagecode.core.ImageCodeHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kq-universal-imagecode-starter Show documentation

The newest version!

package com.kqinfo.universal.imagecode.core;

import com.kqinfo.universal.imagecode.properties.ImageCodeProperties;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.lang.NonNull;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Objects;
import java.util.concurrent.TimeUnit;

/**
 * 图形验证码工具
 *
 * @author Zijian Liao
 * @since 1.5.0
 */
@Slf4j
public class ImageCodeHandler {

    @Resource
    private StringRedisTemplate stringRedisTemplate;
    @Resource
    private ImageCodeProperties imageCodeProperties;

    public void sendCodeImage(String key, OutputStream outputStream){
        ImageCode imageCode = ImageCodeUtil.createImage();
        stringRedisTemplate.opsForValue().set(key, imageCode.getCode(), imageCodeProperties.getCodeExpireSeconds(), TimeUnit.SECONDS);
        ImageCodeUtil.sendImage(imageCode.getBufferedImage(), outputStream);
    }

    public void sendCodeImage(String key) throws IOException {
        HttpServletResponse response = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getResponse();
        ImageCode imageCode = ImageCodeUtil.createImage();
        stringRedisTemplate.opsForValue().set(key, imageCode.getCode(), imageCodeProperties.getCodeExpireSeconds(), TimeUnit.SECONDS);
        ImageCodeUtil.sendImage(imageCode.getBufferedImage(), response.getOutputStream());
    }

    public boolean checkCode(@NonNull String key, @NonNull String code){
        String cacheCode = stringRedisTemplate.opsForValue().get(key);
        stringRedisTemplate.delete(key);
        if(cacheCode != null){
            return code.toLowerCase().equals(cacheCode);
        }
        return false;
    }
}