package com.sim.newlook;
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.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
import org.junit.Test;
import com.aliyun.oss.model.ListObjectsRequest;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.sim.newlook.data.domain.Team;
public class ZipFileTest {
static final int BUFFER = 8192;
@Test
public void test() {
try {
zipFile("D:\task001", "001", "D:\test");
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
private static boolean deleteDir(File dir) {
if (dir.isDirectory()) {
String[] children = dir.list();
for (int i=0; i<children.length; i++) {
boolean success = deleteDir(new File(dir, children[i]));
if (!success) {
return false;
}
}
}
// 目录此时为空，可以删除
return dir.delete();
}
public void zipFile(String taskId, String teamId, String tempPath) throws IOException {
Node root = getTeamNode(teamId);
loopfile(taskId, root, tempPath);
ZipOutputStream out = null;
File file = new File(tempPath + File.separator + root.getName());
try {
FileOutputStream fileOutputStream = new FileOutputStream(new File(tempPath+File.separator+"test.zip"));
CheckedOutputStream cos = new CheckedOutputStream(fileOutputStream,
new CRC32());
out = new ZipOutputStream(cos);
String basedir = "";
compress(file, out, basedir);
out.close();
} catch (Exception e) {
throw new RuntimeException(e);
}
deleteDir(file);
}
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);
}
}
private void loopfile(String taskId, Node root, String path) throws IOException {
String folderName = root.getName();
String prefix = root.getId();
List<Node> nodes = root.getChilds();
System.out.println(path + File.separator + folderName);
File tempFile = new File(path + File.separator + folderName);
if (!tempFile.exists()) {
tempFile.mkdirs();
}
if (nodes == null) {
List<ObjectSummary> objectSummaries = listObjects(new ListObjectsRequest(taskId).withPrefix(prefix));
for (ObjectSummary objectSummary : objectSummaries) {
InputStream inputStream = getObject(objectSummary.bucket, objectSummary.key);
FileOutputStream fos = new FileOutputStream(new File(tempFile.getPath() + File.separator + objectSummary.key));
byte[] b = new byte[BUFFER];
while ((inputStream.read(b)) != -1) {
fos.write(b);
}
inputStream.close();
fos.close();
}
} else {
for (Node node : nodes) {
loopfile(taskId, node, tempFile.getPath());
}
}
}
private List<ObjectSummary> listObjects(ListObjectsRequest listObjectRequest) {
File file = new File(listObjectRequest.getBucketName());
List<ObjectSummary> objectSummaries = new ArrayList<ZipFileTest.ObjectSummary>();
if (file.exists()) {
File[] files = file.listFiles();
if (files != null) {
for (File f : files) {
if (f.getName().startsWith(listObjectRequest.getPrefix())) {
objectSummaries.add(new ObjectSummary(listObjectRequest.getBucketName(), f.getName()));
}
}
}
}
return objectSummaries;
}
public InputStream getObject(String bucket, String key) throws FileNotFoundException {
return new FileInputStream(new File(bucket + File.separator + key));
}
public Node getTeamNode(String teamId) {
List<Team> teams = getChildTeams(teamId);
Node root = null;
Map<String, Node> nodeMap = new HashMap<String, Node>();
Map<String, List<Node>> siblings = new HashMap<String, List<Node>>();
for (Team department : teams) {
String parentId = department.getParentTeam() == null ? null : department.getParentTeam().getId();
Node node2Create = new Node(department.getId(), department.getName(), parentId);
nodeMap.put(department.getId(), node2Create);
if (node2Create.getParentid() == null) {
root = node2Create;
}
}
// Find parent for each node if any
for (Node node : nodeMap.values()) {
if (node.getParentid() != null) {
node.setParent(nodeMap.get(node.getParentid()));
if (siblings.get(node.getParentid()) == null) {
siblings.put(node.getParentid(), new ArrayList<Node>());
}
siblings.get(node.getParentid()).add(node);
}
}
// Find children for each node if any
for (Entry<String, List<Node>> entry : siblings.entrySet()) {
nodeMap.get(entry.getKey()).setChilds(entry.getValue());
}
System.out.println(root);
return root;
}
public List<Team> getChildTeams(String teamId) {
List<Team> teams = new ArrayList<>();
// create team for test
Team team1 = new Team();
team1.setId("001");
team1.setName("team01");
Team team2 = new Team();
team2.setId("002");
team2.setName("team02");
team2.setParentTeam(team1);
Team team3 = new Team();
team3.setId("003");
team3.setName("team03");
team3.setParentTeam(team1);
Team team4 = new Team();
team4.setId("004");
team4.setName("team04");
team4.setParentTeam(team2);
Team team5 = new Team();
team5.setId("005");
team5.setName("team05");
team5.setParentTeam(team2);
Team team6 = new Team();
team6.setId("006");
team6.setName("team06");
team6.setParentTeam(team3);
Team team7 = new Team();
team7.setId("007");
team7.setName("team07");
team7.setParentTeam(team3);
teams.add(team7);
teams.add(team6);
teams.add(team5);
teams.add(team4);
teams.add(team3);
teams.add(team2);
teams.add(team1);
return teams;
}
class ObjectSummary {
private String bucket;
private String key;
public ObjectSummary(String bucket, String key) {
super();
this.bucket = bucket;
this.key = key;
}
}
class Node {
@JsonIgnore
private Node parent;
public String id;
public String name;
public String parentid;
private List<Node> childs;
public Node() {
super();
}
public Node(String id, String name, String parentid) {
super();
this.id = id;
this.parentid = parentid;
this.name = name;
}
public Node getParent() {
return parent;
}
public void setParent(Node parent) {
this.parent = parent;
}
public List<Node> getChilds() {
return childs;
}
public void setChilds(List<Node> childs) {
this.childs = childs;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public String getParentid() {
return parentid;
}
public void setParentid(String parentid) {
this.parentid = parentid;
}
@Override
public String toString() {
return "Node {id=" + id + ", name=" + name + ", childs="
+ childs + "}";
}
}
}
处理后压缩文件的文件目录结构如下
team01
-team02
-team04
-team05
-team03
-team06
-team07