JAVA实现URL转Base64

2022-07-04
JAVA实现URL转Base64
工具类

import cn.hutool.core.codec.Base64;
import cn.hutool.core.util.StrUtil;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.QRCodeWriter;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import lombok.experimental.UtilityClass;
import lombok.extern.slf4j.Slf4j;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.geom.RoundRectangle2D;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URL;
import java.util.HashMap;

@Slf4j
@UtilityClass
public class QRCodeUtil {
    /** 默认宽度 */
    private static final Integer WIDTH = 140;
    /** 默认高度 */
    private static final Integer HEIGHT = 140;

    /** LOGO 默认宽度 */
    private static final Integer LOGO_WIDTH = 22;
    /** LOGO 默认高度 */
    private static final Integer LOGO_HEIGHT = 22;

    /** 图片格式 */
    private static final String IMAGE_FORMAT = "png";

    private static final String CHARSET = "utf-8";
    /** 原生转码前面没有 data:image/png;base64 这些字段，返回给前端是无法被解析 */
    private static final String BASE64_IMAGE = "data:image/png;base64,%s";

    /**
     * 生成二维码，使用默认尺寸
     *
     * @param content 内容
     * @return
     */
    public String getBase64QRCode(String content) {
        return getBase64Image(content, WIDTH, HEIGHT, null, null, null);
    }

    /**
     * 生成二维码，使用默认尺寸二维码，插入默认尺寸logo
     *
     * @param content 内容
     * @param logoUrl logo地址
     * @return
     */
    public String getBase64QRCode(String content, String logoUrl) {
        return getBase64Image(content, WIDTH, HEIGHT, logoUrl, LOGO_WIDTH, LOGO_HEIGHT);
    }

    /**
     * 生成二维码
     *
     * @param content 内容
     * @param width 二维码宽度
     * @param height 二维码高度
     * @param logoUrl logo 在线地址
     * @param logoWidth logo 宽度
     * @param logoHeight logo 高度
     * @return
     */
    public String getBase64QRCode(
            String content,
            Integer width,
            Integer height,
            String logoUrl,
            Integer logoWidth,
            Integer logoHeight) {
        return getBase64Image(content, width, height, logoUrl, logoWidth, logoHeight);
    }

    /**
     * 获取二维码的base64字符串
     *
     * @param content
     * @param width
     * @param height
     * @param logoUrl
     * @param logoWidth
     * @param logoHeight
     * @return
     */
    private String getBase64Image(
            String content,
            Integer width,
            Integer height,
            String logoUrl,
            Integer logoWidth,
            Integer logoHeight) {
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        BufferedImage bufferedImage =
                crateQRCode(content, width, height, logoUrl, logoWidth, logoHeight);
        try {
            ImageIO.write(bufferedImage, IMAGE_FORMAT, os);
        } catch (IOException e) {
            log.error("[生成二维码，错误{}]", e);
        }
        // 转出即可直接使用
        return String.format(BASE64_IMAGE, Base64.encode(os.toByteArray()));
    }

    /**
     * 生成二维码
     *
     * @param content 内容
     * @param width 二维码宽度
     * @param height 二维码高度
     * @param logoUrl logo 在线地址
     * @param logoWidth logo 宽度
     * @param logoHeight logo 高度
     * @return
     */
    private BufferedImage crateQRCode(
            String content,
            Integer width,
            Integer height,
            String logoUrl,
            Integer logoWidth,
            Integer logoHeight) {
        if (StrUtil.isNotBlank(content)) {
            HashMap<EncodeHintType, Comparable> hints = new HashMap<>(4);
            // 指定字符编码为utf-8
            hints.put(EncodeHintType.CHARACTER_SET, CHARSET);
            // 指定二维码的纠错等级为中级
            hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
            // 设置图片的边距
            hints.put(EncodeHintType.MARGIN, 2);
            try {
                QRCodeWriter writer = new QRCodeWriter();
                BitMatrix bitMatrix =
                        writer.encode(content, BarcodeFormat.QR_CODE, width, height, hints);
                BufferedImage bufferedImage =
                        new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
                for (int x = 0; x < width; x++) {
                    for (int y = 0; y < height; y++) {
                        bufferedImage.setRGB(x, y, bitMatrix.get(x, y) ? 0xFF000000 : 0xFFFFFFFF);
                    }
                }
                if (StrUtil.isNotBlank(logoUrl)) {
                    insertLogo(bufferedImage, width, height, logoUrl, logoWidth, logoHeight);
                }
                return bufferedImage;
            } catch (Exception e) {
                log.error(e.getMessage());
            } finally {
            }
        }
        return null;
    }

    /**
     * 二维码插入logo
     *
     * @param source 二维码
     * @param width 二维码宽度
     * @param height 二维码高度
     * @param logoUrl logo 在线地址
     * @param logoWidth logo 宽度
     * @param logoHeight logo 高度
     * @throws Exception
     */
    private void insertLogo(
            BufferedImage source,
            Integer width,
            Integer height,
            String logoUrl,
            Integer logoWidth,
            Integer logoHeight)
            throws Exception {
        // logo 源可为 File/InputStream/URL
        Image src = ImageIO.read(new URL(logoUrl));
        // 插入LOGO
        Graphics2D graph = source.createGraphics();
        int x = (width - logoWidth) / 2;
        int y = (height - logoHeight) / 2;
        graph.drawImage(src, x, y, logoWidth, logoHeight, null);
        Shape shape = new RoundRectangle2D.Float(x, y, logoWidth, logoHeight, 6, 6);
        graph.setStroke(new BasicStroke(3f));
        graph.draw(shape);
        graph.dispose();
    }

    /**
     * 获取二维码
     *
     * @param content 内容
     * @param output 输出流
     * @throws IOException
     */
    public void getQRCode(String content, OutputStream output) throws IOException {
        BufferedImage image = crateQRCode(content, WIDTH, HEIGHT, null, null, null);
        ImageIO.write(image, IMAGE_FORMAT, output);
    }

    /**
     * 获取二维码
     *
     * @param content 内容
     * @param logoUrl logo资源
     * @param output 输出流
     * @throws Exception
     */
    public void getQRCode(String content, String logoUrl, OutputStream output) throws Exception {
        BufferedImage image = crateQRCode(content, WIDTH, HEIGHT, logoUrl, LOGO_WIDTH, LOGO_HEIGHT);
        ImageIO.write(image, IMAGE_FORMAT, output);
    }

    /**
     * 测试
     *
     * @param args
     */
    //    public static void main(String[] args) {
    //        String str =
    //
    // "https://fee.dccnet.com.cn:448/servlet/H5OnlinePaymentServlet?f=ICBCqr&t=2&p=33&x=0&z=&i=UEo5OTAwMTgwNTEwMDAwNTIwOTA=&n=5YWD5rCU5qOu5p6X&l=c21zQmlsbFNlcT1iNGVmMWRmMmExYzI1MGY5";
    //        System.out.println(getBase64QRCode(str, null));
    //    }
}
使用

1	String QRCodeBase64 = QRCodeUtil.getBase64QRCode(url);