qa-prevention-gwj/src/main/java/com/zcloud/util/HiddenExcelImgView.java

package com.zcloud.util;

import java.awt.image.BufferedImage;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;

import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.fasterxml.jackson.databind.introspect.AnnotationMap;
import net.coobird.thumbnailator.Thumbnails;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFClientAnchor;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFPatriarch;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.HorizontalAlignment;
import org.apache.poi.ss.usermodel.VerticalAlignment;
import org.apache.poi.ss.usermodel.Workbook;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.servlet.view.document.AbstractXlsView;

import com.zcloud.entity.PageData;

/**
 * 说明：导出到EXCEL
 * 作者：luoxiaobao
 * 官网：www.qdkjchina.com
 */
public class HiddenExcelImgView extends AbstractXlsView{

	@Value("${http.file.url}")
	private String fileUrl;

	@Override
	protected void buildExcelDocument(Map<String, Object> model,
									  Workbook workbook, HttpServletRequest request,
									  HttpServletResponse response) throws Exception {
		// TODO Auto-generated method stub
		String proPath = PathUtil.getProjectpath();
		String beji = fileUrl;
		//beji =beji.replaceAll("\\\\", "/");

		Date date = new Date();
		String filename = DateUtil.date2Str(date, "yyyyMMddHHmmss");
		HSSFSheet sheet;
		response.setContentType("application/octet-stream");
		response.setHeader("Content-Disposition", "attachment;filename="+filename+".xls");

		HSSFWorkbook book = (HSSFWorkbook) workbook;
		sheet = book.createSheet("sheet1");

		List<String> titles = (List<String>) model.get("titles");
		int len = titles.size();
		HSSFCellStyle headerStyle = book.createCellStyle(); //标题样式
		headerStyle.setAlignment(HorizontalAlignment.CENTER);
		headerStyle.setVerticalAlignment(VerticalAlignment.CENTER);
		HSSFFont headerFont = book.createFont();			//标题字体
		headerFont.setBold(true);
		headerFont.setFontHeightInPoints((short)11);
		headerStyle.setFont(headerFont);
		short height=25*20;
		HSSFRow row = sheet.createRow(0);
		if(len>250) {
			len = 250;
		}
		for(int i=0; i<len; i++){ //设置标题
			String title = titles.get(i);
			row.setRowStyle(headerStyle);
			if(title.length()>250) {
				row.createCell(i).setCellValue(title.substring(0, 250));
			}else {
				row.createCell(i).setCellValue(title);
			}
		}
		sheet.getRow(0).setHeight(height);

		HSSFCellStyle contentStyle = book.createCellStyle(); //内容样式
		HSSFPatriarch patriarch = sheet.createDrawingPatriarch();

		contentStyle.setAlignment(HorizontalAlignment.CENTER);
		List<PageData> varList = (List<PageData>) model.get("varList");


		int varCount = varList.size();
		for(int i=0; i<varCount; i++){
			PageData vpd = varList.get(i);
			HSSFRow rows = sheet.createRow(i+1);
			for(int j=0;j<len;j++){
				String valKey = "var"+(j+1);
				String imgKey = "img" +(j+1);
				if(vpd.containsKey(valKey)) {
					String varstr = vpd.getString(valKey) != null ? vpd.getString("var"+(j+1)) : "";
					rows.setRowStyle(contentStyle);
					rows.createCell(j).setCellValue(varstr);
				}
				if(vpd.containsKey(imgKey)) {
					String varstr = vpd.getString(imgKey) != null ? vpd.getString("img"+(j+1)) : "";
					if(Tools.isEmpty(varstr)) {
						continue;
					}
					try {
						URL url = new URL(beji + varstr);
						HttpURLConnection conn = (HttpURLConnection)url.openConnection();
						conn.setRequestMethod("GET");
						conn.setConnectTimeout(5 * 1000);
						InputStream inStream = conn.getInputStream();
						byte[] data = readInputStream(inStream);
						byte[] compressPicForScale = compressPicForScale(data, 100);
						HSSFClientAnchor anchor1 = new HSSFClientAnchor(0, 0, 0, 0,(short) (j), (i+1), (short) (j+1), (i+2));
						patriarch.createPicture(anchor1, book.addPicture(compressPicForScale, HSSFWorkbook.PICTURE_TYPE_JPEG));
					}catch (Exception e) {
						System.out.println("未找到图片");
						e.printStackTrace();
					}
//					BufferedImage bufferImg = null;//图片一
//					ByteArrayOutputStream byteArrayOut = new ByteArrayOutputStream();//
//					File file=new File(beji+varstr);
//					if(file.exists()) {//https://wwag.qhdsafety.com/file//uploadFiles/yhtp/07bd077457ec4da7bd6ec2219984079c/20220126/545463a3effe4a3d9d50dc701c69cc01.png
//						bufferImg = ImageIO.read(file);
//						ImageIO.write(bufferImg, "jpg", byteArrayOut);
//						 HSSFClientAnchor anchor1 = new HSSFClientAnchor(0, 0, 0, 0,(short) (j), (i+1), (short) (j+1), (i+2));
//						   patriarch.createPicture(anchor1, book.addPicture(byteArrayOut
//			                       .toByteArray(), HSSFWorkbook.PICTURE_TYPE_JPEG));
//					}

				}

			}

		}

	}

	private static byte[] readInputStream(InputStream inStream) throws Exception{
		ByteArrayOutputStream outStream = new ByteArrayOutputStream();
		//创建一个Buffer字符串
		byte[] buffer = new byte[1024];
		//每次读取的字符串长度，如果为-1，代表全部读取完毕
		int len = 0;
		//使用一个输入流从buffer里把数据读取出来
		while( (len=inStream.read(buffer)) != -1 ){
			//用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
			outStream.write(buffer, 0, len);
		}
		//关闭输入流
		inStream.close();
		//把outStream里的数据写入内存
		return outStream.toByteArray();
	}
	private static final Integer ZERO = 0;
	private static final Integer ONE_ZERO_TWO_FOUR = 1024;
	private static final Integer NINE_ZERO_ZERO = 900;
	private static final Integer THREE_TWO_SEVEN_FIVE = 3275;
	private static final Integer TWO_ZERO_FOUR_SEVEN = 2047;
	private static final Double ZERO_EIGHT_FIVE = 0.85;
	private static final Double ZERO_SIX = 0.6;
	private static final Double ZERO_FOUR_FOUR = 0.44;
	private static final Double ZERO_FOUR = 0.4;
	public static byte[] compressPicForScale(byte[] imageBytes, long desFileSize) {
		if (imageBytes == null || imageBytes.length <= ZERO || imageBytes.length < desFileSize * ONE_ZERO_TWO_FOUR) {
			return imageBytes;
		}
		double accuracy = 0.5d;
		try {
			while (imageBytes.length > desFileSize * ONE_ZERO_TWO_FOUR) {
				ByteArrayInputStream inputStream = new ByteArrayInputStream(imageBytes);
				ByteArrayOutputStream outputStream = new ByteArrayOutputStream(imageBytes.length);
				Thumbnails.of(inputStream)
						.scale(accuracy)
						.outputQuality(accuracy)
						.toOutputStream(outputStream);
				imageBytes = outputStream.toByteArray();
			}
			System.out.println("图片原大小={}kb | 压缩后大小={}kb");
//			logger.info("图片原大小={}kb | 压缩后大小={}kb",
//					srcSize / ONE_ZERO_TWO_FOUR, imageBytes.length / ONE_ZERO_TWO_FOUR);
		} catch (Exception e) {
//			logger.error("【图片压缩】msg=图片压缩失败!", e);
		}
		return imageBytes;
	}

}