java多线程文件压缩

java多线程压缩文件

（1）压缩线程类

package gzip;


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.List;
import java.util.zip.GZIPOutputStream;


public class GZipThread extends Thread {
private List pool;
private static int filesCompressed=0;
public GZipThread(List pool){
this.pool=pool;
}
private static synchronized void incrementFilesCompressed(){
filesCompressed++;
}
@Override
public void run() {
// TODO Auto-generated method stub
File input=null;
synchronized (pool) {
while(pool.isEmpty()){
if(this.filesCompressed==GZipAllFiles.getNumberOfFilesToBeCompressed()){
System.out.println("Thread ending");
return;
}
try{
pool.wait();
}catch(InterruptedException e){
e.printStackTrace();
}
}

}
input=(File)pool.remove(pool.size()-1);
incrementFilesCompressed();
if(!input.getName().endsWith(".gz")){
try{
InputStream in=new FileInputStream(input);
in= new BufferedInputStream(in);
File output=new File(input.getParent(),input.getName()+".gz");
if(!output.exists()){
OutputStream out=new FileOutputStream(output);
out=new GZIPOutputStream(out);
out=new BufferedOutputStream(out);
int b;
while((b=in.read())!=-1) out.write(b);
out.flush();   //必须进行刷新工作，不然buffer中的字节无法完全清空
out.close();
in.close();
}
}catch(IOException e){
e.printStackTrace();
}
}
}
}

（2）调用压缩线程类的主类（main参数传入压缩路径）

package gzip;


import java.io.File;
import java.util.Vector;


public class GZipAllFiles {
private final static int THREAD_COUNT=4;
private static int filesToBeCompressed=-1;


public static int getNumberOfFilesToBeCompressed(){
return filesToBeCompressed;
}
/**
* @param args
* 
*/
public static void main(String[] args) {
// TODO Auto-generated method stub
Vector pool=new Vector();
GZipThread[] threads=new GZipThread[THREAD_COUNT];
for(int i=0;i<THREAD_COUNT;i++){
threads[i]=new GZipThread(pool);
threads[i].start();
}
int totalFiles=0;
for(int i=0;i<args.length;i++){
File f=new File(args[i]);
if(f.exists()){
if(f.isDirectory()){
File[] files=f.listFiles();
for(int j=0;j<files.length;j++){
if(!files[j].isDirectory()){
totalFiles++;
synchronized (pool) {
pool.add(0,files[j]);
pool.notifyAll();
}
}
}
}
else{
totalFiles++;
synchronized (pool) {
pool.add(0.f);
pool.notifyAll();
}
}
}
}
filesToBeCompressed=totalFiles;
for(int i=0;i<threads.length;i++){
threads[i].interrupt();
}
}


}