java 기능 구현 | 디렉토리 전체 / 폴더 전체 압축하기 / 압축 파일 다운로드 구현

파일과 폴더가 들어있는 폴더 DIR 구조

[zipDirDownLoad]

대상 폴더를 압축해 zip 파일로 생성(compressDir, compressFile)후, 생성한 zip 파일을 response에 담아서 client에 보낸다.

package com.study.controller;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.io.File;

   /**
    * @param targetDirPath 압축할 폴더 경로
    * @param outputPath 출력파일 경로
    * @param outputFileName 출력파일명
    * @description 폴더 압축 메소드
    */
    @SuppressWarnings("resource")
    protected void zipDirDownLoad(HttpServletResponse response, String targetDirPath, String resultPath, String resultFileName) throws Exception {
        // 파일명에 .zip이 없는 경우, .zip 을 붙여준다.
        int pos = resultFileName.lastIndexOf(".") == -1 ? resultFileName.length() : resultFileName.lastIndexOf(".");
        if (!resultZipName.substring(pos).equalsIgnoreCase(".zip")) {
            resultFileName += ".zip";
            }
        
        // 압축 대상 파일 존재 여부체크
        File targetDir = new File(targetDirPath);
        if (!targetDir.exists()) {
            response.getWriter().println("<script>alert('File Not Found');history.back();</script>");
            logger.error("TargetDir does not exist.");
            }

        // 파일출력 스트림
        FileOutputStream fos = null;
        // 압축파일출력 스트림
        ZipOutputStream zos = null;
        try {
            fos = new FileOutputStream(new File(resultPath + resultFileName));  // 파일 객체(new File(resultPath + resultFileName))에 쓸 파일출력 스트림
            zos = new ZipOutputStream(fos); // zip output stream
            
            // 디렉토리 검색를 통한 하위 파일과 폴더 검색 및 압축
            compressDir(targetDir, targetDir.getPath(), zos);
            
            } finally {
                if (zos != null) zos.close();
                if (fos != null) fos.close();
                }
        
        // 압축 파일을 response로 보내기
        File resultZipFile = new File(resultPath + resultFileName);
        FileInputStream fis = new FileInputStream(resultZipFile);
        
        // 파일명 인코딩 설정
        String userAgent = request.getHeader("User-Agent");
        if(userAgent.contains("Edge") || userAgent.contains("MSIE") || userAgent.contains("Trident")) {
            resultZipName = URLEncoder.encode(resultZipName, "UTF-8").replace("\\+", "%20");
        } else if(userAgent.contains("Chrome") || userAgent.contains("Opera") || userAgent.contains("Firefox")) {
            resultZipName = new String(resultZipName.getBytes("UTF-8"), "ISO-8859-1");
        }
        
        byte[] byteFile = new byte[(int) resultZipFile.length()];
        fis.read(byteFile);    // 주어진 배열 byteFile만큼의 데이터를 읽어서 byteFile에 저장하고 읽은 바이트 수를 반환
        
        response.setContentType("application/zip");
        response.setContentLength(byteFile.length);
        
        response.setHeader("Content-Disposition",  "attachment; filename=\""+resultFileName+"\"");
        response.getOutputStream().write(byteFile); // 주어진 배열 byteFile에 저장된 모든 내용을 출력소스에 쓴다.
        
        if(fis != null) fis.close();
        if(resultZipFile.exists()) resultZipFile.delete();
    }

 

[ compressDir ]

디렉토리를 파일 인자로 받아서 내용 파일들의 디렉토리여부에 따라 분기 처리

    /**
     * @param file 현재 파일
     * @param resultRootPath 루트 경로
     * @param zos  압축 스트림
     * @description 디렉토리 탐색 및 분기
     */
    private void compressDir(File file, String resultRootPath, ZipOutputStream zos) throws Exception {
        // 인자로 주어진 파일이 디렉토리인지 파일인지에 따라 분기
        if (file.isDirectory()) {
            // 디렉토리일 경우 재귀
            File[] files = file.listFiles();
            for (File f : files) {
                compressDir(f, resultRootPath, zos);
                }
            file.delete();
            } else {    // 파일일 경우 압축을 한다.
                compressFile(file, resultRootPath, zos);
                }
	}

 

[ compressFile ]

분기처리 중 디렉토리가 아닌 파일일 경우 실질적으로 압축 수행

    /**
     * @param file 현재 파일
     * @param resultRootPath 루트 경로
     * @param zos  압축 스트림
     * @description 파일 압축 메서드
     */
    private void compressFile(File file, String resultRootPath, ZipOutputStream zos) throws Exception {
        FileInputStream fis = null;
        try {
            String zipFileName = file.getPath().replace(resultRootPath + "\\", "");
            // 파일을 읽어들임
            fis = new FileInputStream(file);
            // Zip엔트리 생성
            ZipEntry zipentry = new ZipEntry(zipFileName);  // zipFileName을 이름으로 가지는 zipEntry 생성
            // 스트림에 밀어넣기(자동 오픈)
            zos.putNextEntry(zipentry);         // zip entry를 쓰고, 엔트리 데이터 시작에 stream을 위치시킴
            int length = (int) file.length();   // 파일의 길이
            byte[] buffer = new byte[length];   // 파일의 길이만큼의 버퍼
            fis.read(buffer, 0, length);        // 최대 length개의 byte를 읽어서, 배열 buffer의 지정된 위치(0)부터 저장하고 읽은 바이트 수 반환
            zos.write(buffer, 0, length);       // 주어진 배열 buffer에 저장된 내용 중에서 0번째 부터 length만큼만을 읽어서 출력소스에 쓴다.
            zos.closeEntry();                   // 현재 zip entry를 닫고 다음 엔트리 작성을 위해 위치함 
            } finally {
                if (fis != null) fis.close();
                if (file != null && file.exists())file.delete();
                }
	}