java 打包下载

 

import java.io.BufferedInputStream;
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.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

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

import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

public class Xiazai {

// paths 跟目录下的文件地址  "1.txt;2.txt"
	public void downloads(String path, HttpServletRequest request, HttpServletResponse response) {
		if(path==null||path=="")return;
		List<File> fileList = new ArrayList<>();
		//根目录
		String rootPath =((ServletRequestAttributes)RequestContextHolder.getRequestAttributes()).getRequest().getSession().getServletContext().getRealPath("")+"upload\\";
	 	//文件地址数组
		String[] paths = path.split(";");
	 	for (String string : paths) {
	 		String s = rootPath+string;
			File f = new File(s);
			if(!f.exists())continue;
			fileList.add(f);
		}
	 	if(fileList.size()<=0)return;
		try {
			//保存的文件名为 当前日期 +随机数
			SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmssSSS");
			String date = sdf.format(new Date());
			String ran = 2+(int)(Math. random()*(102-2))+"";
			//判断文件夹是否存在
			File f = new File(rootPath);
			if (!f.exists()) f.mkdirs();
			//保存文件
			File file = new File(rootPath +date + ran+".rar");
			if (!file.exists()){
				file.createNewFile();
			}
			response.reset();
			//创建文件输出流
			FileOutputStream fous = new FileOutputStream(file);
			/**打包的方法我们会用到ZipOutputStream这样一个输出流, 所以这里我们把输出流转换一下*/
			ZipOutputStream zipOut= new ZipOutputStream(fous);
			zipFile(fileList, zipOut);
			zipOut.close();
            fous.close();
            
           download(file, request, response);
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	
	}
	
/**
 * 把接受的全部文件打成压缩包 
 * @param List<File>;  
 * @param org.apache.tools.zip.ZipOutputStream  
 */
	 public static void zipFile(List files,ZipOutputStream outputStream) {
		int size = files.size();
		for(int i = 0; i < size; i++) {
			File file = (File) files.get(i);
			zipFile(file, outputStream);
	  }
	 }
 /**  
    * 根据输入的文件与输出流对文件进行打包
    * @param File
    * @param org.apache.tools.zip.ZipOutputStream
    */
	public static void zipFile(File inputFile, ZipOutputStream ouputStream) {
		try {
			if (inputFile.exists()) {
				/**
				 * 如果是目录的话这里是不采取操作的，  * 至于目录的打包正在研究中
				 */
				if (inputFile.isFile()) {
					FileInputStream IN = new FileInputStream(inputFile);
					BufferedInputStream bins = new BufferedInputStream(IN, 512);
					// org.apache.tools.zip.ZipEntry
					ZipEntry entry = new ZipEntry(inputFile.getName());
					ouputStream.putNextEntry(entry);
					// 向压缩文件中输出数据  
					int nNumber;
					byte[] buffer = new byte[512];
					while ((nNumber = bins.read(buffer)) != -1) {
						ouputStream.write(buffer, 0, nNumber);
					}
					// 关闭创建的流对象  
					bins.close();
					IN.close();
				} else {
					try {
						File[] files = inputFile.listFiles();
						for (int i = 0; i < files.length; i++) {
							zipFile(files[i], ouputStream);
						}
					} catch (Exception e) {
						e.printStackTrace();
					}
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}

	}
	public void download(File f,HttpServletRequest request, HttpServletResponse response) {
		if (!f.exists()) return;
		try {
			String filename=f.getName();
			// 当文件名不是英文名的时候，最好使用url解码器去编码一下，
			filename = URLEncoder.encode(filename, "UTF-8");
			// 将响应的类型设置为图片
			//response.setContentType("image/jpeg");
			response.setHeader("Content-Disposition", "attachment;filename=" + filename);
			// 现在通过IO流来传送数据	
			InputStream input = new FileInputStream(f);
			// getServletContext().getResourceAsStream("/testImage.jpg");
			OutputStream output = response.getOutputStream();
			byte[] buff = new byte[1024 * 10];// 可以自己 指定缓冲区的大小
			int len = 0;
			while ((len = input.read(buff)) > -1) {
				output.write(buff, 0, len);
			}
			// 关闭输入输出流
			input.close();
			output.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		
	}
}