spring.turbo.module.misc.captcha.hutool.AbstractHutoolCaptchaService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-turbo-module-misc Show documentation

There is a newer version: 3.3.3

Show newest version

package spring.turbo.module.misc.captcha.hutool;

import cn.hutool.captcha.ICaptcha;
import org.springframework.lang.NonNull;
import spring.turbo.module.misc.captcha.Captcha;
import spring.turbo.module.misc.captcha.CaptchaService;
import spring.turbo.util.io.CloseUtils;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.UncheckedIOException;

/**
 * @author 应卓
 * @since 1.0.1
 */
public abstract class AbstractHutoolCaptchaService implements CaptchaService {

    @Override
    public final Captcha create() {
        final ICaptcha c = createCaptcha();
        return new Captcha(c.getCode(), toBufferedImage(c));
    }

    private BufferedImage toBufferedImage(ICaptcha c) {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        ByteArrayInputStream in = null;

        try {
            c.write(out);
            in = new ByteArrayInputStream(out.toByteArray());
            return ImageIO.read(in);
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        } finally {
            CloseUtils.closeQuietly(in);
            CloseUtils.closeQuietly(out);
        }
    }

    @NonNull
    protected abstract ICaptcha createCaptcha();

}