package com.ruoyi.framework.util;
|
|
import com.google.zxing.BarcodeFormat;
|
import com.google.zxing.EncodeHintType;
|
import com.google.zxing.MultiFormatWriter;
|
import com.google.zxing.WriterException;
|
import com.google.zxing.common.BitMatrix;
|
import com.ruoyi.framework.exception.ErrorException;
|
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.File;
|
import java.io.IOException;
|
import java.time.LocalDateTime;
|
import java.time.format.DateTimeFormatter;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* 配置图像写入器
|
*
|
* @author z1292
|
*
|
*/
|
public class MatrixToImageWriter {
|
private final int BLACK = 0xFF000000;
|
private final int WHITE = 0xFFFFFFFF;
|
|
private BufferedImage toBufferedImage(BitMatrix matrix) {
|
int width = matrix.getWidth();
|
int height = matrix.getHeight();
|
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
|
for (int x = 0; x < width; x++) {
|
for (int y = 0; y < height; y++) {
|
image.setRGB(x, y, matrix.get(x, y) ? BLACK : WHITE);
|
}
|
}
|
return image;
|
}
|
|
private void writeToFile(BitMatrix matrix, String format, File file) throws IOException {
|
BufferedImage image = toBufferedImage(matrix);
|
if (!ImageIO.write(image, format, file)) {
|
throw new ErrorException("Could not write an image of format " + format + " to " + file);
|
}
|
}
|
|
public String code(String content, String path) {
|
try {
|
String codeName = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yy_MM_dd&HH_mm_ss"));// 二维码的图片名
|
String imageType = "jpg";// 图片类型
|
MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
|
Map<EncodeHintType, Object> hints = new HashMap<>();
|
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
|
hints.put(EncodeHintType.MARGIN, 0);
|
BitMatrix bitMatrix = multiFormatWriter.encode(content, BarcodeFormat.QR_CODE, 400, 400, hints);
|
File file1 = new File(path, codeName + "." + imageType);
|
writeToFile(bitMatrix, imageType, file1);
|
return file1.getPath();
|
} catch (WriterException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
throw new ErrorException("二维码生成失败");
|
}
|
|
}
|
