java實現伺服器檔案打包zip並下載的樣本(邊打包邊下載)_java

使用該方法,可以即時打包檔案,一邊打包一邊傳輸,不使用任何的緩衝,讓使用者零等待!

複製代碼 代碼如下:

/**
 *
 * mySocket 用戶端 Socket
 * @param file 待打包的檔案夾或檔案
 * @param fileName 打包下載的檔案名稱
 * @throws IOException
 */

private void down(File file, String fileName) throws IOException {
 OutputStream outputStream = mySocket.getOutputStream();
 StringBuffer sb = new StringBuffer("HTTP/1.1 200 OK\r\n");
 sb.append("Server: java/1.1\r\n");
 sb.append("Content-Type:application/octet-stream;charset=UTF-8\r\n");
 //sb.append("User-Agent: Mozilla/4.0 (compatible;MSIE6.0;Windows NT 5.0)\r\n");
 sb.append("Content-Disposition: attachment; filename=" + fileName
   + "\r\n");
 sb.append("Transfer-Encoding: chunked\r\n");
 sb.append("Connection: Keep-Alive\r\n\r\n");
 outputStream.write(sb.toString().getBytes());
 outputStream.flush();
 ZipCompressor zipCompressor = new ZipCompressor(new MyOutputStream(
   outputStream));
 zipCompressor.compress(file);
 System.out.println("zip end");  
 System.out.println("write '0\\r\\n\\r\\n'");
 outputStream.write("0\r\n\r\n".getBytes());//Transfer-Encoding: chunked傳輸結束標記
 outputStream.flush();
 outputStream.close();
 System.out.println("download stop");
 try {
  mySocket.close();
 } catch (Throwable t) {
 }
}

複製代碼 代碼如下:

package cn.liangjintang.webserver.zipFile;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipCompressor {
 static final int BUFFER = 8192;
 private OutputStream outputStream;
 public ZipCompressor(MyOutputStream outputStream) {
  this.outputStream=outputStream;
 }
 public void compress(File file) {
  if (!file.exists())
   throw new RuntimeException(file.getAbsolutePath() + "不存在！");
  try {
   CheckedOutputStream cos = new CheckedOutputStream(outputStream,
     new CRC32());
   ZipOutputStream out = new ZipOutputStream(cos);
   String basedir = "";
   compress(file, out, basedir);
   out.close();//必須關閉,這樣才會寫入zip的結束資訊,否則zip檔案不完整.若想繼續寫入,可重寫outputStream.close()方法
  } catch (Exception e) {
   throw new RuntimeException(e);
  }
 }

 private void compress(File file, ZipOutputStream out, String basedir) {
  //判斷是目錄還是檔案
  if (file.isDirectory()) {
   System.out.println("壓縮：" + basedir + file.getName());
   this.compressDirectory(file, out, basedir);
  } else {
   System.out.println("壓縮：" + basedir + file.getName());
   this.compressFile(file, out, basedir);
  }
 }

 // 壓縮一個目錄
 private void compressDirectory(File dir, ZipOutputStream out, String basedir) {
  if (!dir.exists())
   return;

  File[] files = dir.listFiles();
  for (int i = 0; i < files.length; i++) {
   /** 遞迴 */
   compress(files[i], out, basedir + dir.getName() + "/");
  }
 }

 //壓縮一個檔案
 private void compressFile(File file, ZipOutputStream out, String basedir) {
  if (!file.exists()) {
   return;
  }
  try {
   BufferedInputStream bis = new BufferedInputStream(
     new FileInputStream(file));
   ZipEntry entry = new ZipEntry(basedir + file.getName());
   out.putNextEntry(entry);
   int count;
   byte data[] = new byte[BUFFER];
   while ((count = bis.read(data, 0, BUFFER)) != -1) {
    out.write(data, 0, count);
   }
   bis.close();
  } catch (Exception e) {
   throw new RuntimeException(e);
  }
 }
}

複製代碼 代碼如下:

package cn.liangjintang.webserver.zipFile;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

public class MyOutputStream extends FilterOutputStream {
 public MyOutputStream(OutputStream out) {
  super(out);
 }

 final byte[] oneBytes = "1\r\n".getBytes();
 final byte[] rnBytes = "\r\n".getBytes();

 public void write(int b) throws IOException {
  out.write(oneBytes);//位元組數1+CRLF
  out.write(b);//資料實體
  out.write(rnBytes);//CRLF
 }

 public void write(byte[] b) throws IOException {
  out.write(Integer.toHexString(b.length).getBytes());//位元組數,十六進位
  out.write(rnBytes);//CRLF
  out.write(b);//資料實體
  out.write(rnBytes);//CRLF
 }

 public void write(byte[] b, int off, int len) throws IOException {
  out.write(Integer.toHexString(len - off).getBytes());//位元組數,十六進位
  out.write(rnBytes);//CRLF
  out.write(b, off, len);//資料實體
  out.write(rnBytes);//CRLF
 }

 /**
  * 重寫該方法,否則OutputStream會被關閉,其他的資料<br/>
  * (如Transfer-Encoding: chunked傳輸結束標記)就不能再繼續寫入了
  */
 public void close() throws IOException {
 }
}