图片打包Zip压缩包文件流给前端
前言:
本文实现将图片打包成zip流返回给前端,图片来源可以是OSS存储中获取的url下载,也可以是字节生成的图片字节流,直接打包成zip文件给前端
需要代码源码可以私信我获取
1.导入包
<!-- 二维码需要 -->
<dependency>
<groupId>com.google.zxing</groupId>
<artifactId>core</artifactId>
<version>3.4.1</version>
</dependency>
<dependency>
<groupId>com.google.zxing</groupId>
<artifactId>javase</artifactId>
<version>3.4.1</version>
</dependency>
<!-- 图片转换需要 -->
<dependency>
<groupId>org.springframework</groupId>
<artifactId>spring-test</artifactId>
<version>5.3.24</version>
</dependency>
2.工具类
package com.qrcode.until;
/**
* @ClassName QRCodeUtils
* @Date 2023/3/29 9:53
* @Author legend
*/
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.Base64;
import java.util.HashMap;
import java.util.Map;
public class QRCodeUtils {
/**
* 黑色
*/
private static final int BLACK = 0xFF000000;
/**
* 白色
*/
private static final int WHITE = 0xFFFFFFFF;
/**
* 宽
*/
private static final int WIDTH = 2000;
/**
* 高
*/
private static final int HEIGHT = 2000;
/**
* 图片高度增加60
*/
private static final int PIC_HEIGHT = HEIGHT + 120;
/**
* 二维码传图片
*
* @param matrix
* @return
*/
public static BufferedImage toBufferedImage(BitMatrix matrix) {
int width = matrix.getWidth();
int height = matrix.getHeight();
BufferedImage image = new BufferedImage(width, PIC_HEIGHT, BufferedImage.TYPE_INT_RGB);
for (int x = 0; x < width; x++) {
for (int y = 0; y < PIC_HEIGHT; y++) {
image.setRGB(x, y, WHITE);
}
}
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;
}
/**
* 生成二维码
*
* @param content 扫描二维码的内容
* @param format 图片格式 jpg
* 文件
* @throws Exception
*/
@SuppressWarnings("unchecked")
public static BufferedImage generateQrCode(String content, String format) throws Exception {
MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
@SuppressWarnings("rawtypes")
Map hints = new HashMap();
// 设置UTF-8， 防止中文乱码
hints.put(EncodeHintType.CHARACTER_SET, "UTF-8");
// 设置二维码四周白色区域的大小
hints.put(EncodeHintType.MARGIN, 5);
// 设置二维码的容错性
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
// 画二维码
BitMatrix bitMatrix = multiFormatWriter.encode(content, BarcodeFormat.QR_CODE, WIDTH, HEIGHT, hints);
BufferedImage image = toBufferedImage(bitMatrix);
return image;
}
/**
* 把生成的图片写到指定路径
*
* @param qrcFile 路径
* @param qrCodeContent 二维码内容
* @param pressText 增加的文字
* @throws Exception
*/
public static void generateQrCodeByPath(File qrcFile, String qrCodeContent, String pressText) throws Exception {
BufferedImage image = generateQrCode(qrCodeContent, "jpg");
Graphics g = image.getGraphics();
((Graphics2D) g).setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
//设置字体，大小
Font font = new Font("黑体", Font.PLAIN, 150);
g.setFont(font);
g.setColor(Color.black);
FontMetrics metrics = g.getFontMetrics(font);
// 文字在图片中的坐标 这里设置在中间
int startX = (WIDTH - metrics.stringWidth(pressText)) / 2;
// int startY=HEIGHT+(PIC_HEIGHT-HEIGHT)/2; //文字在二维码上面
int startY = PIC_HEIGHT - HEIGHT; //文字在二维码下面
g.drawString(pressText, startX, startY);
g.dispose();
image.flush();
try {
ImageIO.write(image, "jpg", qrcFile);
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 把生成的图片返回到前端
*
* @param qrCodeContent 二维码内容
* @throws Exception
*/
public static BufferedImage generateQrCodeBack( String qrCodeContent) throws Exception {
BufferedImage image = generateQrCode(qrCodeContent, "jpg");
Graphics g = image.getGraphics();
((Graphics2D) g).setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
image.flush();
return image;
}
/**
* 生成二维码并使用Base64编码
*
* @param content 二维码内容
* @return 返回base64图片
* @throws Exception
*/
@SuppressWarnings("unchecked")
public static String getBase64QRCode(String content) throws Exception {
String format = "png";
MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
@SuppressWarnings("rawtypes")
Map hints = new HashMap();
// 设置二维码四周白色区域的大小
hints.put(EncodeHintType.MARGIN, 10);
// 设置二维码的容错性
hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.H);
// 画二维码
BitMatrix bitMatrix = multiFormatWriter.encode(content, BarcodeFormat.QR_CODE, WIDTH, HEIGHT, hints);
BufferedImage image = MatrixToImageWriter.toBufferedImage(bitMatrix);
ByteArrayOutputStream os = new ByteArrayOutputStream();//新建流。
ImageIO.write(image, format, os);//利用ImageIO类提供的write方法，将bi以png图片的数据模式写入流。
byte b[] = os.toByteArray();//从流中获取数据数组。
String base64String = Base64.getEncoder().encodeToString(b);
// Base64编码
return base64String;
}
}
3.生成二维码(代码片段)
for (String uuid : uuidArray) {
//使用zxing 生成二维码
bufferedImage = QRCodeUtils.generateQrCodeBack(uuid);
log.info("二维码生成成功");
byte[] bytes = imageToBytes(bufferedImage);
//MockMultipartFile 需要导入spring-test的依赖包
/*这里自定义流程 可以将图片上传到OSS 也可以直接获取生成的图片输出
* 本段代码实现的是直接将图片输出
* */
hashMap.put(uuid, bytes);
log.info("完成");
}
4.将图片打包成zip(代码片段)
for (Map.Entry<String, byte[]> entry : hashMap.entrySet()) {
// 获取到oss的图片路径,使用上传到Oss放开这两行代码
// URL url = new URL(entry.getValue());
// imageInputStream = url.openStream();
imageInputStream= new ByteArrayInputStream(entry.getValue());
//二维码图片名称
ZipEntry zipEntry = new ZipEntry(entry.getKey() + ".jpg");
zipOutputStream.putNextEntry(zipEntry);
int len;
// 定义每次读取的字节数组
byte[] buffer = new byte[1024];
log.info("开始写入zip");
while ((len = imageInputStream.read(buffer)) > 0) {
zipOutputStream.write(buffer, 0, len);
}
}
5.将图片转换成字节数组
private byte[] imageToBytes(BufferedImage bImg) {
ByteArrayOutputStream out = new ByteArrayOutputStream();
try {
ImageIO.write(bImg, "jpg", out);
} catch (IOException e) {
log.error("转成字节数组异常",e);
}
return out.toByteArray();
}