CompressionUtil - 生成文件并打包下载

1、调用：

使用

CompressionUtil.zip2(Map<String, InputStream> in, OutputStream out) 方法

压缩为一个export-wsdl-20141023135549.zip的压缩包并自动下载

包括：

1、各服务名称文件夹 及子文件夹下3个文件 

2、校验码文件md5checksum.txt

 
md5checksum.txt文件内容：

  

注意：

1、Map<String, InputStream> in ：

key为各文件名

value为各文件内容InputStream流

 

2、FilenameUtils.getName(path)

数据库中文件路径SERVER_ABS_PATH为：

/home/esbapp/lightesb/SYSFILE/1403612785951/SB_FI_ORS_InquiryPaymentApplicationInfoSrv/MsgHeader.xsd

 

获取的结果为MsgHeader.xsd

FilenameUtils.getName(path)
 a/b/c.txt --> c.txt
 a.txt     --> a.txt
 a/b/c     --> c
 a/b/c/    --> ""

  

数据库表格式：

ESB_WSDL_FILE：

 

ESB_FILE（一个表，数据太长分开了）:

 

 视图ESB_WSDL_FILE_V省略

 

代码：

@RequestMapping(value = "esbService/exportWsdl.do", method = RequestMethod.GET)
@ResponseBody
public void exportWsdl(@RequestParam String info, HttpServletResponse response){
	//logger.debug("exportWsdl begin...");
	
	//logger.debug("parm : " + info);
	
	// 判断参数非空
	if(StringUtils.isNotEmpty(info)){
		/**
		* 构建Map<Long, String> basePath 
		* <srvId,srvName> 用于下面由srvId获取srvName，构建srvName文件夹
		* info = srvId1:srvName1,srvId2:srvName2,...
		* 无需关注
		*/
		// zip包基础路径
		Map<Long, String> basePath = new HashMap<Long, String>();
		
		// 要下载WSDL的服务id
		List<Long> srvId = new ArrayList<Long>();					
		String[] srvInfo = info.split(",");
		for(String si : srvInfo){
			String[] s = si.split(":");
			Long id = null;
			String srvName = null;
			try {
				id = Long.parseLong(s[0]);
				srvName = s[1];
			} catch (Exception e) {
				logger.error("exportWsdl parse parm error... " + si);
				continue;
			}
			srvId.add(id);
			basePath.put(id, srvName);
		}
		
		logger.debug("parse parm pass...");
		
		/**
		* =======================================
		* 压缩并自动下载
		* 重点关注部分
		* =======================================
		*/
		// 获取wsdl文件清单
		List<EsbWsdlFileV> ewf = esbServiceVDS.getWsdlFileList(srvId);
		
		//logger.debug("file list num : " + ewf.size());
		
		// detail为md5checksum.txt文件内容
		StringBuffer detail = new StringBuffer();
		detail.append("md5 checksum").append("\n");
		detail.append("\n");		
		
		Long timestamp = System.currentTimeMillis();
		Map<String, InputStream> in = new HashMap<String, InputStream>();
		for(EsbWsdlFileV f : ewf){
                        //文件真实路径
			String path = f.getServerAbsPath();
                        //生成的文件，如ESB_EOM_E12_ImportOrderUrgeInfoSrv/MsgHeader.xsd
			String zipPath = basePath.get(f.getServiceId()) + "/" + FilenameUtils.getName(path);
			try {
				in.put(zipPath, new FileInputStream(new File(path)));
				detail.append(f.getMd5CheckSum()).append("\t").append(zipPath).append("\n");
			} catch (FileNotFoundException e) {
				logger.error(path + " not found...");
			}
		}
		detail.append("\n");
		detail.append(String.format("generated by project %s", SysConfigCache.getConfigByName(Constants.VERSION))).append("\n");
		detail.append(String.format("%1$tF %1$tT", timestamp)).append("\n");
		detail.append("Copyright (c) 2004-2013 A-Company").append("\n");
		
		//logger.debug("zip files parm : " + in);
		
		try {
			in.put("md5checksum.txt", IOUtils.toInputStream(detail.toString(), "UTF-8"));
		} catch (IOException e1) {
			logger.error("can not generate file list...");
		}
		
		// httpheader设置
		response.setContentType("application/x-msdownload;");  
		response.setHeader("Content-disposition", "attachment; filename = export-wsdl-" + String.format("%1$tY%1$tm%1$td%1$tH%1$tM%1$tS", timestamp) + ".zip");
		// 将wsdl文件压缩至请求流中
		try {
			CompressionUtil.zip2(in, response.getOutputStream());
		} catch (IOException e) {
			e.printStackTrace();
			logger.error("exportWsdl zip error...");
		}
	}
	
	logger.debug("exportWsdl end...");
}

 

 

2、工具类：

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Map;

import org.apache.tools.tar.TarEntry;
import org.apache.tools.tar.TarOutputStream;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;

public class CompressionUtil {

	private static final int BUFFEREDSIZE = 1024;

	/**
	 * 将文件解压缩到指定目录
	 * 
	 * @param srcFile
	 * @param descDir
	 * @throws IOException
	 */
	public static void unzip(File srcFile, String descDir) throws IOException {
		// 创建解压目录
		new File(descDir).mkdirs();

		ZipFile zipFile = new ZipFile(srcFile);
		String strPath, gbkPath, strtemp;
		File tempFile = new File(descDir);
		strPath = tempFile.getAbsolutePath();
		@SuppressWarnings("rawtypes")
		java.util.Enumeration e = zipFile.getEntries();
		while (e.hasMoreElements()) {
			org.apache.tools.zip.ZipEntry zipEnt = (ZipEntry) e.nextElement();
			gbkPath = zipEnt.getName();
			if (zipEnt.isDirectory()) {
				strtemp = strPath + File.separator + gbkPath;
				File dir = new File(strtemp);
				dir.mkdirs();
				continue;
			} else {
				// 读写文件
				InputStream is = zipFile.getInputStream(zipEnt);
				BufferedInputStream bis = new BufferedInputStream(is);
				gbkPath = zipEnt.getName();
				strtemp = strPath + File.separator + gbkPath;

				// 建目录
				String strsubdir = gbkPath;
				for (int i = 0; i < strsubdir.length(); i++) {
					if (strsubdir.substring(i, i + 1).equalsIgnoreCase("/")) {
						String temp = strPath + File.separator
								+ strsubdir.substring(0, i);
						File subdir = new File(temp);
						if (!subdir.exists())
							subdir.mkdir();
					}
				}
				FileOutputStream fos = new FileOutputStream(strtemp);
				BufferedOutputStream bos = new BufferedOutputStream(fos);
				int c;
				while ((c = bis.read()) != -1) {
					bos.write((byte) c);
				}
				bos.close();
				fos.close();
			}
		}
	}

	/**
	 * 解压指定路径的文件至指定目录
	 * 
	 * @param srcFilepath
	 * @param descDir
	 * @throws IOException
	 */
	public static void unzip(String srcFilepath, String descDir)
			throws IOException {
		File srcFile = new File(srcFilepath);
		unzip(srcFile, descDir);
	}

	/**
	 * 解压文件到当前目录
	 * 
	 * @param srcFile
	 * @throws IOException
	 */
	public static void unzip(File srcFile) throws IOException {
		unzip(srcFile, srcFile.getParent());
	}

	/**
	 * 解压指定路径的文件至当前目录
	 * 
	 * @param srcFilepath
	 * @throws IOException
	 */
	public static void unzip(String srcFilepath) throws IOException {
		File srcFile = new File(srcFilepath);
		unzip(srcFile, srcFile.getParent());
	}

	/**
	 * 压缩zip格式的压缩文件
	 * 
	 * @param inputFile
	 *            需压缩文件
	 * @param out
	 *            输出压缩文件
	 * @param base
	 *            ZipEntry name
	 * @throws IOException
	 */
	private static void zip(File inputFile, ZipOutputStream out, String base)
			throws IOException {
		if (inputFile.isDirectory()) {
			File[] inputFiles = inputFile.listFiles();
			out.putNextEntry(new ZipEntry(base + "/"));
			base = base.length() == 0 ? "" : base + "/";
			for (int i = 0; i < inputFiles.length; i++) {
				zip(inputFiles[i], out, base + inputFiles[i].getName());
			}
		} else {
			if (base.length() > 0) {
				out.putNextEntry(new ZipEntry(base));
			} else {
				out.putNextEntry(new ZipEntry(inputFile.getName()));
			}
			FileInputStream in = new FileInputStream(inputFile);
			try {
				int c;
				byte[] by = new byte[BUFFEREDSIZE];
				while ((c = in.read(by)) != -1) {
					out.write(by, 0, c);
				}
			} catch (IOException e) {
				throw e;
			} finally {
				in.close();
			}
		}
	}

	/**
	 * 压缩zip格式的压缩文件
	 * 
	 * @param inputFile
	 *            需压缩文件
	 * @param zipFilename
	 *            输出文件及详细路径
	 * @throws IOException
	 */
	public static void zip(File inputFile, String zipFilename)
			throws IOException {
		ZipOutputStream out = new ZipOutputStream(new FileOutputStream(
				zipFilename));
		try {
			zip(inputFile, out, "");
		} catch (IOException e) {
			throw e;
		} finally {
			out.close();
		}
	}

	/**
	 * 压缩zip格式的压缩文件
	 * 
	 * @param inputFilename
	 *            压缩的文件或文件夹及详细路径
	 * @param zipFilename
	 *            输出文件名称及详细路径
	 * @throws IOException
	 */
	public static void zip(String inputFilename, String zipFilename)
			throws IOException {
		zip(new File(inputFilename), zipFilename);
	}

	/**
	 * 压缩tar格式的压缩文件
	 * 
	 * @param inputFilename
	 *            压缩文件
	 * @param tarFilename
	 *            输出路径
	 * @throws IOException
	 */
	public static void tar(String inputFilename, String tarFilename)
			throws IOException {
		tar(new File(inputFilename), tarFilename);
	}

	/**
	 * 压缩tar格式的压缩文件
	 * 
	 * @param inputFile
	 *            压缩文件
	 * @param tarFilename
	 *            输出路径
	 * @throws IOException
	 */
	public static void tar(File inputFile, String tarFilename)
			throws IOException {
		TarOutputStream out = new TarOutputStream(new FileOutputStream(
				tarFilename));
		try {
			tar(inputFile, out, "");
		} catch (IOException e) {
			throw e;
		} finally {
			out.close();
		}
	}

	/**
	 * 压缩tar格式的压缩文件
	 * 
	 * @param inputFile
	 *            压缩文件
	 * @param out
	 *            输出文件
	 * @param base
	 *            结束标识
	 * @throws IOException
	 */
	private static void tar(File inputFile, TarOutputStream out, String base)
			throws IOException {
		if (inputFile.isDirectory()) {
			File[] inputFiles = inputFile.listFiles();
			out.putNextEntry(new TarEntry(base + "/"));
			base = base.length() == 0 ? "" : base + "/";
			for (int i = 0; i < inputFiles.length; i++) {
				tar(inputFiles[i], out, base + inputFiles[i].getName());
			}
		} else {
			if (base.length() > 0) {
				out.putNextEntry(new TarEntry(base));
			} else {
				out.putNextEntry(new TarEntry(inputFile.getName()));
			}
			FileInputStream in = new FileInputStream(inputFile);
			try {
				int c;
				byte[] by = new byte[BUFFEREDSIZE];
				while ((c = in.read(by)) != -1) {
					out.write(by, 0, c);
				}
			} catch (IOException e) {
				throw e;
			} finally {
				in.close();
			}
		}
	}
	
	/**
	 * @param files <k, v>
	 *        k zip包中的路径
	 *        v 待压缩的文件
	 * @param out 输出流
	 */
	public static void zip(Map<String, File> files, OutputStream out) throws IOException {
		ZipOutputStream zos = new ZipOutputStream(out);
		for(String path : files.keySet()){
			zip(files.get(path), zos, path);
		}
		zos.close();
	}
	
	/**
	 * 流压缩
	 * 
	 * @param in 输入流
	 * @param out 输出流
	 * @param name 压缩包中路径
	 * @throws IOException
	 */
	private static void zip(InputStream in, ZipOutputStream out, String name)
			throws IOException {
		out.putNextEntry(new ZipEntry(name));
		try {
			int c;
			byte[] by = new byte[BUFFEREDSIZE];
			while ((c = in.read(by)) != -1) {
				out.write(by, 0, c);
			}
		} catch (IOException e) {
			throw e;
		} finally {
			in.close();
		}
	}
	
	/**
	 * 流压缩
	 * 
	 * @param in
	 * @param out
	 * @throws IOException
	 */
	public static void zip2(Map<String, InputStream> in, OutputStream out) throws IOException {
		ZipOutputStream zos = new ZipOutputStream(out);
		for(String path : in.keySet()){
			zip(in.get(path), zos, path);
		}
		zos.close();
	}
	
	/*
	public static void main(String[] args) throws IOException {
		ZipOutputStream zos = new ZipOutputStream(new File("D:/btest/jjx.zip"));
		zip(new File("D:/atest/build.xml"), zos, "srv1/build.xml");
		zip(new File("D:/atest/jaxws-custom.xml"), zos, "srv2/jaxws-custom.xml");
	}
	*/
}

 。。