package cn.iocoder.yudao.module.wxcp.framework.qrcode;
|
|
import com.google.zxing.BarcodeFormat;
|
import com.google.zxing.EncodeHintType;
|
import com.google.zxing.client.j2se.MatrixToImageWriter;
|
import com.google.zxing.common.BitMatrix;
|
import com.google.zxing.qrcode.QRCodeWriter;
|
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
|
|
import java.io.ByteArrayOutputStream;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.wxcp.enums.ErrorCodeConstants.QRCODE_GENERATE_FAILED;
|
|
/**
|
* 企业微信模块 - 二维码生成工具
|
*
|
* 用于把活动报名页地址生成二维码图片，供企业微信推送或后台下载
|
*
|
* @author 超级管理员
|
*/
|
public class WxCpQrCodeUtil {
|
|
/**
|
* 默认二维码边长（像素）
|
*/
|
public static final int DEFAULT_SIZE = 400;
|
|
private WxCpQrCodeUtil() {
|
}
|
|
/**
|
* 生成二维码 PNG 字节流
|
*
|
* @param content 二维码内容（一般是一个 URL）
|
* @param size 边长像素，传 0 或负数则使用默认值
|
* @return PNG 图片字节
|
*/
|
public static byte[] generatePng(String content, int size) {
|
int actualSize = size > 0 ? size : DEFAULT_SIZE;
|
try {
|
Map<EncodeHintType, Object> hints = new HashMap<>(4);
|
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
|
// 容错级别 M：约 15%，兼顾扫码成功率与码点密度，适合印在物料上
|
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
|
hints.put(EncodeHintType.MARGIN, 1);
|
|
BitMatrix matrix = new QRCodeWriter().encode(content, BarcodeFormat.QR_CODE,
|
actualSize, actualSize, hints);
|
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
MatrixToImageWriter.writeToStream(matrix, "PNG", out);
|
return out.toByteArray();
|
} catch (Exception e) {
|
throw exception(QRCODE_GENERATE_FAILED);
|
}
|
}
|
|
}
|
