java对tar、zip包的相关操作

package com.mash5.web.services.util;

import java.io.BufferedInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.List;

import java.util.zip.ZipEntry;

import java.util.zip.ZipFile;

import java.util.zip.ZipInputStream;

import java.util.zip.ZipOutputStream;

import org.bson.types.ObjectId;

import com.ice.tar.TarEntry;

import com.ice.tar.TarInputStream;

import com.mongodb.DB;

import com.mongodb.Mongo;

import com.mongodb.gridfs.GridFS;

import com.mongodb.gridfs.GridFSDBFile;

import com.mongodb.gridfs.GridFSInputFile;

public class ParseTar_Zip_Jar {

/**
* 解压tar包
* @param tarPath
* @param basePath
* @return
*/
public static File getFileListByTarPath(String tarPath, String basePath) {
if(tarPath == null) return null;
try {
OutputStream out =  null;
TarEntry entry = null;
TarInputStream tar = new TarInputStream(new FileInputStream(new File(tarPath)));
File returnFile = null;
int i = 0;
while((entry = tar.getNextEntry()) != null) {
if(i == 0) {
File outFile = new File(basePath + entry.getName());
outFile.mkdir();
returnFile = outFile;
}
i ++;
if(entry.isDirectory()) {
continue;
}
File outFile = new File(basePath + entry.getName());
new File(outFile.getParent()).mkdirs();
out = new FileOutputStream(outFile);
int data = -1;
while((data = tar.read()) != -1) {
out.write(data);
}
out.close();
}
tar.close();
return returnFile;
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return null;
}

/**
* 解压Zip包
* @param zipPath
* @param basePath
* @return
*/
public static File getFileListByZipPath(String zipPath, String basePath) {
if(zipPath == null || "".equals(zipPath)) return null;
try {
ZipInputStream in = new ZipInputStream(new FileInputStream(new File(zipPath)));
ZipEntry entry = null;
OutputStream out = null;
File returnFile = null;
int i = 0;
while((entry = in.getNextEntry()) != null) {
if(i == 0) {
File outFile = new File(basePath + entry.getName());
returnFile = outFile;
}
i ++;
if(entry.isDirectory()) {
continue;
}
File outFile = new File(basePath + entry.getName());
//创建他的父文件夹
new File(outFile.getParent()).mkdirs();
out = new FileOutputStream(outFile);
int data = -1;
while((data = in.read()) != -1) {
out.write(data);
}
out.close();
}
in.close();
return returnFile;
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return null;
}

/**
* 解压Zip包
* @param zipPath
* @param basePath
* @return
*/
public static List<GridFSDBFile> getFilesByFSZip(GridFSDBFile file, GridFS myFS) {
if(file == null || "".equals(file)) return null;
List<GridFSDBFile> files = new ArrayList<GridFSDBFile>();

try {
ZipInputStream zis = new ZipInputStream(file.getInputStream());
ZipEntry entry = null;
while(( entry = zis.getNextEntry()) != null) {
if(entry.isDirectory()) continue;
GridFSInputFile inputFile  = myFS.createFile(zis);
inputFile.setFilename(entry.getName());
inputFile.setContentType(file.getMD5());
inputFile.save();
System.out.println(entry.getName()+","+inputFile.getId().toString());
ObjectId fileId = new ObjectId(inputFile.getId().toString());
GridFSDBFile fileObj = myFS.find(fileId);
files.add(fileObj);
}

zis.close();
return files;
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return null;
}

public static void zipFS(List<GridFSDBFile> fileList,String zipFileName) throws Exception{
ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFileName));
for(GridFSDBFile file : fileList){
zipFS(zos, file);
}
zos.close();
}

/**
* @param zipFileName
*            生成的压缩文件夹路径
* @param filePath
*            源文件夹
* @throws FileNotFoundException
* @throws IOException
*/

public static void zip(String zipFileName, List<String> filePath) throws FileNotFoundException, IOException {
ZipOutputStream zos = null;
zos = new ZipOutputStream(new FileOutputStream(zipFileName));
boolean has=false;
for(String str : filePath) {
File f=new File(str);
if(f.exists()){
has=true;
zip(zos, f, "");
}
}
if(!has)zos.putNextEntry(new ZipEntry(""));
zos.close();
}

public static void zip(String zipFileName){
ZipOutputStream zos = null;
try {
zos = new ZipOutputStream(new FileOutputStream(zipFileName));
zos.putNextEntry(new ZipEntry(""));
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}

private static void zipFS (ZipOutputStream zos, GridFSDBFile file) throws Exception {
InputStream is = file.getInputStream();
BufferedInputStream bis =  new BufferedInputStream(is);
zos.putNextEntry(new ZipEntry(file.getFilename()));
int length = 0;
byte[] buffer = new byte[1024 * 1024];
while ((length = bis.read(buffer)) != -1) {
zos.write(buffer, 0, length);
zos.flush();
}
is.close();
bis.close();
}

/**
*
* @param zos
* @param f
*            源文件夹
* @param base
* @throws FileNotFoundException
* @throws IOException
*/
private static void zip(ZipOutputStream zos, File f, String base) throws FileNotFoundException, IOException {
BufferedInputStream bis = null;
FileInputStream fis = null;
if (f.isDirectory()) {
File[] files = f.listFiles();
if(base.equals("")){
base=f.getName()+"/";
} else {
base=base+"/";
}
zos.putNextEntry(new ZipEntry(base));
for (int i = 0; i < files.length; i++) {
zip(zos, files[i], base + files[i].getName());
}
} else {
if(base.equals("")){
base=f.getName();
}
zos.putNextEntry(new ZipEntry(base));
fis = new FileInputStream(f);
bis = new BufferedInputStream(fis);
byte[] b = new byte[1024 * 1024];
int len = 0;
while ((len = bis.read(b)) != -1) {
zos.write(b, 0, len);
zos.flush();
}
fis.close();
bis.close();
}
}

public static void main(String[] args) throws Exception {
Mongo db = new Mongo("192.168.3.171", 30000);
DB mydb = db.getDB("mash5");

GridFS myFS = new GridFS(mydb);

//打包文件
/*
List<GridFSDBFile> fileList = new ArrayList<GridFSDBFile>();

ObjectId fileId = new ObjectId("54bf1389fffe27eced2f5e7f");
GridFSDBFile file = myFS.find(fileId);
fileList.add(file);

ObjectId fileId1 = new ObjectId("54bdbe2ba43e5ada8d875439");
GridFSDBFile file1 = myFS.find(fileId1);
fileList.add(file1);

ObjectId fileId2 = new ObjectId("54bdb6b1a43ee44e955eec45");
GridFSDBFile file2 = myFS.find(fileId2);
fileList.add(file2);

zipFS(fileList, "D:/0203-zh.rar");
*/

//解压包

ObjectId fileId = new ObjectId("54d1dd6bfe1cae7da80f6b68");
GridFSDBFile file = myFS.find(fileId);
List<GridFSDBFile> fileList = ParseTar_Zip_Jar.getFilesByFSZip(file, myFS);
for(GridFSDBFile fileObj : fileList){
InputStream is = fileObj.getInputStream();
String fileName = fileObj.getFilename();
int lastIndex = fileName.lastIndexOf("/");
fileName = fileName.substring(lastIndex+1, fileName.length());
FileOutputStream out = new FileOutputStream(new File("D:/test/"+fileName));
//设置缓存;
int length = 0;
byte[] buffer = new byte[1024000];
while ((length = is.read(buffer)) != -1) {
out.write(buffer, 0, length);
}
// 关闭对应的流
out.flush();
is.close();
out.close();

}

//解压tar包成功

// List fileList = ParseTar_Zip_Jar.getFileListByTarPath(“e:/WEB-INF.tar”, “e:/a/”);

// for(File f : fileList) {

// System.out.println(f.getAbsolutePath());

// }

//解压zip包成功
//ParseTar_Zip_Jar.getFileListByZipPath("f:/my.zip", "f:/my/");

// for(File f : fileList) {

// System.out.println(f.getAbsolutePath());

// }

// File f = new File(“e:/WEB-INF.tar”);

// System.out.println(f.length());

// List filePathList = new ArrayList();

// filePathList.add(“F:\a”);

// filePathList.add(“F:\b”);

// ParseTar_Zip_Jar.getTarByFileList(filePathList);

// System.out.println(new Date().toLocaleString());

//List<String> filePathList = new ArrayList<String>();

// filePathList.add(“f:/a/a/a.txt”);

// filePathList.add(“f:/c.txt”);

// try {

// ParseTar_Zip_Jar.zip(“F:/my.zip”, filePathList);

// } catch (FileNotFoundException e) {

// e.printStackTrace();

// } catch (IOException e) {

// e.printStackTrace();

// }

}

}