使用迭代器模式和組合模式實現瀏覽器一鍵匯出下載為zip壓縮包檔案

由於專案需要，於是又想起之前看過的設計模式，於是便有了一鍵匯出的想法。
思路簡單明瞭。一步一步看下去就好。

1.建立組合物件

public abstract class FileComponent {
    /**
     * Description:   遞迴建立資料夾，或者檔案 
     */
    public void mkFile(){
        throw new UnsupportedOperationException();
    }

    /**
     * Description:   獲取檔案輸入路徑 
     */
 

    public String getInPath(){
        throw new UnsupportedOperationException();
    }
    /**
     * Description:    獲取檔案輸出路徑
     */
    public String getOutPath(){
        throw new UnsupportedOperationException();
    }
    /**
     * Description:    對於資料夾來說是可以add其他資料夾或者檔案
     */
    public void add
 
(FileComponent fileComponent){
        throw new UnsupportedOperationException();
    }
}

此組合物件，可以是資料夾物件，也可是具體的檔案物件，再後面呼叫中，不需要了解到底是一個資料夾還是一個檔案（即組合模式的透明性）。

2.組合物件抽象類的實現

上述抽象類的實現如下：

public class ZipFileItem extends FileComponent{
    //輸入檔案的路徑
    String inPath;
    //輸出檔案的路徑
    String outPath;
    //子節點檔案資訊
 

    List<FileComponent> fileComponents = new ArrayList<FileComponent>();

    //inPath 可以為null
    public ZipFileItem(String outPath){
            this.outPath =outPath;
    }
    //壓縮檔案的源目錄路徑和壓縮好的目標位置
    public ZipFileItem(String inPath,String outPath){
        this.inPath =inPath;
        this.outPath =outPath;
    }
    public void add(FileComponent fileComponent){
        fileComponents.add(fileComponent);
    }

    public void remove(FileComponent fileComponent){
        fileComponents.remove(fileComponent);
    }
    @Override
    public String getInPath(){
        return inPath;
    }
    @Override
    public String getOutPath(){
        return outPath;
    }
    @Override
    public void mkFile(){
        FileUtils.createFile(inPath, outPath);
        Iterator<FileComponent> iterator = fileComponents.iterator();
        //如果是資料夾，那麼還可以迭代檔案及物件中的具體檔案物件
        while (iterator.hasNext()) {
            FileComponent fileComponent = iterator.next();
            fileComponent.mkFile();
        }
    }
}

3.檔案工具類

public class ConferenceFileUtils {
    /**
     * Description:    根據檔案的絕對路徑，在絕對的輸出路徑進行建立檔案
     * @param inPath    輸入路徑，如果是要根據已有的檔案來建立，那麼一定要傳
     * @param outPath   輸出路徑，如果是目錄則不用
     */
    public static void createFile(String inPath,String outPath){
        File fileIn = new File(inPath);
        File fileOut = new File(outPath);
            //如果目標檔案已存在，則忽略，如果檔案不存在 。則進行建立
            if (!fileOut.exists()) {

                int lastSeparator = outPath.lastIndexOf(File.separator);
                String lastPart = outPath.substring(lastSeparator);
                //如果不是資料夾,則建立檔案
                if (lastPart.lastIndexOf(".")!=-1) {
                    LoggerUtil.info("----------making concreteFile--------"+outPath);
                    FileInputStream in = null;
                    FileOutputStream out = null;
                    File directory = null;  
                            try {
                                directory = new File(outPath.substring(0, lastSeparator+1));
                                directory.mkdirs();
                                out=new FileOutputStream(fileOut);
                                //如果原始檔存在
                                if (fileIn.exists()) {
                                    in=new FileInputStream(fileIn);  
                                    int len; 
                                    byte[] buf=new byte[10240]; 
                                    while((len=in.read(buf))>0){  
                                        out.write(buf,0,len);  
                                    }  
                                    out.close();  
                                    in.close();  
                                    in = null;
                                }
                            }  catch (IOException e) {
                                System.err.println("creating file failed!", e);
                            }
                }
                //如果是資料夾則建立資料夾，如果父類資料夾不存在，那麼也建立
                   else {
                      System.err.println("----------making directory--------"+outPath);
                       fileOut.mkdirs();
                    }
            }

    }
    //遞迴刪除資料夾以及檔案
    public 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 static void outputFile(File file, HttpServletResponse response) throws IOException {
        OutputStream out=null;
        FileInputStream in=null;
        try {
        byte[] src = new byte[1024];
         out = response.getOutputStream();
         in = new FileInputStream(file);
        int len=0;
        while ((len = in.read(src)) > 0) {
            out.write(src, 0, len);
        }
        out.flush();
        out.close();
        in.close();
        } catch (IOException e) {
            throw new IOException(e);
        }finally{
            if(null!=out){
                FortifyUtil.commonReleasedResource(out);
            }
            if(null!=in){
                FortifyUtil.commonReleasedResource(in);
            }
        }

    }
}

4.核心匯出邏輯程式碼

public class exportMaterialToZipTemplate {

    @Resource
    private EnrichFileLevelsService enrichFileLevelsService;
    //根目錄資料夾名稱  or 下載瀏覽器檔名
    private String downloadZipName;
    //根目錄地址
    private String savePath =  "d:\\tempFile";
    //根目錄路徑
    private String superRootPath;
    //根目錄物件
    private FileComponent superRoot;
    //業務引數DTO
    private ExportAllTheMaterialDTO paramDTO;
    //response
    private HttpServletResponse response;

    public exportMaterialToZipTemplate(ExportAllTheMaterialDTO paramDTO,EnrichFileLevelsService enrichFileLevelsService,HttpServletResponse response) {
        this.downloadZipName = paramDTO.getDownloadZipName();
        this.paramDTO = paramDTO;
        this.response = response;
        this.enrichFileLevelsService = enrichFileLevelsService;
        this.superRootPath =savePath+File.separator+downloadZipName;
        this.superRoot = new ZipFileItem(superRootPath); 
    }   

    //1.封裝根目錄
    private void enrichFileLevels(){
        enrichFileLevelsService.enrichFileLevels(superRoot,superRootPath,paramDTO);
    }
    //2.生成檔案目錄層級，即建立所有的檔案（包括資料夾）
    private void createAllTheFiles(){
        if (null!=superRoot) {
            superRoot.mkFile();
        }
    }
    //3.生成檔案層級後後再壓縮後下載到瀏覽器
    private void compressAndDownload() {
        File srcFile = new File(FortifyUtil.filterFileName(superRootPath));
        String targetFilePath = savePath+File.separator+srcFile.getName()+".zip";
        File targetFile = new File(FortifyUtil.filterFileName(targetFilePath));
        ZipFileUtil.zipFiles(srcFile,targetFile);
        try {
            //壓縮檔案臨時路徑
            String downFileName = downloadZipName+".zip";
            response.reset();
            // 定義輸出型別
            response.setContentType("application/octet-stream");
            response.setHeader("content-disposition", "attachment;filename="
                    + new String(downFileName.getBytes("GBK"), "ISO-8859-1")
                    + ";size=" + targetFile.length());
            OutputFileUtil.outputFile(targetFile, response);
            // 刪除臨時存放的資料夾
            if (srcFile.exists()) {
                ConferenceFileUtils.deleteDir(srcFile);
            }
            //刪除臨時的壓縮包
            if (targetFile.exists()) {
                targetFile.delete();
            }
        } catch (IOException e) {
            DevLog.error(e.getMessage());
        }
    }
    //一鍵匯出,外觀模式
    public void export() {
        enrichFileLevels();
        createAllTheFiles();
        compressAndDownload();
    }
}

5.豐富檔案層級的介面

public interface EnrichFileLevelsService {
    public void enrichFileLevels(FileComponent superRoot,String superRootPath,ExportAllTheMaterialDTO paramDTO);
}

不同的業務場景只要實現這介面，實現enrichFileLevels()方法，將實現此介面的
類例項傳到exportMaterialToZipTemplate類的構造方法，然後呼叫exportMaterialToZipTemplate類例項的export()方法即可。即
new exportMaterialToZipTemplate(dtoParams,
enrichFileLevelsService, response).export();