/**
 * 
 */
package iotest.serversocket;
 
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;
 
/**
 *

@author Brandon B. Lin
 * 
 */
public class GZipThread extends Thread {
 
    private List<File> jobQueue;
    private static int filesCompressed = 0;
 
    public GZipThread(List<File> jobQueue) {
        this.jobQueue = jobQueue;
    }
 
    private static synchronized void incrementFilesCompressed() {
        filesCompressed++;
    }
 
    

@Override
    public void run() {
        while (filesCompressed != GZipAllFiles
                .getNumbersOfFilesToBeCompressed()) { // if there is any job, even not in jobqueue
            
            File input = getAJobFromQueue();
            if(input == null) {
                return;
            } else {        
                incrementFilesCompressed();
                compressedAnFile(input);
            }
        }
    }
    
    /**
     * Get a job from queue if any , null if all job done
     */
    private File getAJobFromQueue() {
        
        File result = null;
        synchronized (jobQueue) {
            while (jobQueue.isEmpty()) {
                if (filesCompressed == GZipAllFiles
                        .getNumbersOfFilesToBeCompressed()) {
                    System.out.println(Thread.currentThread() + " ending!");
                    return result;
                }
                
                try {
                    jobQueue.wait();
                } catch (InterruptedException exception) {
                    exception.printStackTrace();
                }
            }
            result =  jobQueue.remove(jobQueue.size() - 1);
        }
        return result;
    }
 
    /**
     * compress an file
     */
    private void compressedAnFile(File fileToBeCompressed) {
        if (!fileToBeCompressed.getName().endsWith(".gz")) { // 不壓縮已經壓縮的檔案
            try {
                InputStream in = new BufferedInputStream(new FileInputStream(
                        fileToBeCompressed));
                File output = new File(fileToBeCompressed.getParent(),
                        fileToBeCompressed.getName() + ".gz");
                if (!output.exists()) { // 不重複壓縮
                    OutputStream out = new BufferedOutputStream(
                            new GZIPOutputStream(new FileOutputStream(output)));
                    copyIntoOut(in, out);
                }
            } catch (IOException exception) {
                exception.printStackTrace();
            }
        }
    }
 
    /**
     * copy data from in to out
     */
    private void copyIntoOut(InputStream in, OutputStream out)
            throws IOException {
        int readByte;
        while ((readByte = in.read()) != -1) {
            out.write(readByte);
        }
        out.flush();
        out.close();
        in.close();
    }
 
}

/**
 * 
 */
package iotest.serversocket;
 
import java.io.File;
import java.util.Vector;
 
/**
 * @author Brandon B. Lin
 * 
 */
public class GZipAllFiles {
 
    public final static int THREAD_COUNT = 4;
    private static int filesToBeCompressed = -1;
    private static GZipThread[] threads = new GZipThread[THREAD_COUNT];
    private static Vector<File> jobQueue = new Vector<>();
 
    public static void main(String[] args) {
        createThreadPool();
        String directoryToBeCompressed = "F:\\java\\UML\\";
        addFilesToJobQueue(directoryToBeCompressed);
        notifyAllThreadNoMoreJobs();
    }
    
    /**
     * create THREAD_COUNT threads in thread pool
     */
    private static void createThreadPool() {
        for (int i = 0; i < threads.length; i++) {
            threads[i] = new GZipThread(jobQueue);
            threads[i].start();
        }
    }
    
    /**
     * if parameter is an file, add the file to jobQueue,
     * if parameter is an directory, add first level's uncompressed files to jobqueue
     */
    private static void addFilesToJobQueue(String directory) {
        File file = new File(directory);
        int totalFiles = 0;
        
        if (file.exists()) {
            if (file.isDirectory()) {
                totalFiles += addDirectoryToJobQueue(file);
            } else {
                addAnFileToJobQueue(file);
                totalFiles++;
            }
        }
        
        filesToBeCompressed = totalFiles; // 必須一次性增加
    }
    
    /**
     * 新增目錄的第一級文件到作業佇列
     */
    private static int addDirectoryToJobQueue(File directory) {
        File[] files = directory.listFiles();
        int numberOfFiles = 0;
        for (int j = 0; j < files.length; j++) {
            if (!files[j].isDirectory()) {
                addAnFileToJobQueue(files[j]);
                numberOfFiles++;
            }
        }
        return numberOfFiles;
    }
    
    /**
     * add an file (not directory) to jobqueuqe
     */
    private static void addAnFileToJobQueue(File fileToAdd) {
        synchronized (jobQueue) {
            jobQueue.add(0, fileToAdd);
            jobQueue.notifyAll();
        }
    }
    
    /**
     * Notify all threads in thread pool that no more job will be added.
     */
    private static void notifyAllThreadNoMoreJobs() {
        for (int i = 0; i < threads.length; i++) {
            threads[i].interrupt();
        }
    }
 
    /**
     * How many files need to be compressed
     */ 
    public static int getNumbersOfFilesToBeCompressed() {
        return filesToBeCompressed;
    }
 
}

歡迎工作一到五年的Java工程師朋友們加入Java技術交流：659270626
群內提供免費的Java架構學習資料（裡面有高可用、高併發、高效能及分散式、Jvm效能調優、Spring原始碼，MyBatis，Netty,Redis,Kafka,Mysql,Zookeeper,Tomcat,Docker,Dubbo,Nginx等多個知識點的架構資料）合理利用自己每一分每一秒的時間來學習提升自己，不要再用"沒有時間“來掩飾自己思想上的懶惰！趁年輕，使勁拼，給未來的