使用谷歌生成二维码的工具类

- 日理万妓 2024-04-01 19:45 63阅读 0赞

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 com.photon.Common.Exception.CustomerException;
    
    import javax.imageio.ImageIO;
    import javax.servlet.ServletOutputStream;
    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;
    
    /**
     * <p>QRCodeUtil</p>
     *
     * @author 李修睿
     * @version 1.0.0
     * @date 2020/7/8
     */
    
    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 static String getBase64QRCode(String content) {
            return getBase64Image(content, WIDTH, HEIGHT, null, null, null);
        }
    
        /**
         * 生成二维码，使用默认尺寸二维码，插入默认尺寸logo
         *
         * @param content 内容
         * @param logoUrl logo地址
         * @return
         */
        public static 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 static String getBase64QRCode(String content, Integer width, Integer height, String logoUrl, Integer logoWidth, Integer logoHeight) {
            return getBase64Image(content, width, height, logoUrl, logoWidth, logoHeight);
        }
    
        private static 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) {
                throw new CustomerException("[生成二维码，错误{}]", 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 static BufferedImage crateQRCode(String content, Integer width, Integer height, String logoUrl, Integer logoWidth, Integer logoHeight) {
            if (StrUtil.isNotBlank(content)) {
                ServletOutputStream stream = null;
                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) {
                    e.printStackTrace();
                } finally {
                    if (stream != null) {
                        try {
                            stream.flush();
                            stream.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }
            }
            return null;
        }
    
        /**
         * 二维码插入logo
         *
         * @param source     二维码
         * @param width      二维码宽度
         * @param height     二维码高度
         * @param logoUrl    logo 在线地址
         * @param logoWidth  logo 宽度
         * @param logoHeight logo 高度
         * @throws Exception
         */
        private static 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 static 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 static 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);
        }
    
    
        public static void main(String[] args) {
            System.out.println(getBase64QRCode("haha"));
        }
    }

<dependency>
                <groupId>com.google.zxing</groupId>
                <artifactId>core</artifactId>
                <version>3.3.3</version>
            </dependency>
            <dependency>
                <groupId>com.google.zxing</groupId>
                <artifactId>javase</artifactId>
                <version>3.3.3</version>
            </dependency>