스프링 프레임워크(Spring Framework)에서 파일 다운로드 방법

후바스탱크 2014. 10. 8. 21:37

스프링 프레임워크(Spring Framework)에서 jsp가 아닌 컨트롤러를 호출하여 파일을 다운로드 하려면 다음과 같은 단계를 거쳐야 한다.

1. 서블릿 컨텐스트(dispatcher-servlet) 에 다음과 같이 뷰 리졸버를 추가 해 준다. 쉽게 생각해서 파일 다운로드 처리를 할 view 페이지 라고 생각하면 된다.

2. AbstractView 를 상속받아 뷰로 사용 될 클래스를 만들어 준다.

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.util.FileCopyUtils;
import org.springframework.web.servlet.view.AbstractView;

public class FileDownloadView extends AbstractView {

public FileDownloadView() {
		// 객체가 생성될 때 Content Type을 다음과 같이 변경 
		setContentType("application/download; charset=utf-8");
	}

@Override
	protected void renderMergedOutputModel(
			Map<String, Object> model, 
			HttpServletRequest request, 
			HttpServletResponse response
			) throws Exception {
		
		Map<String, Object> fileInfo = (Map<String, Object>) model.get("downloadFile"); // 넘겨받은 모델(파일 정보)
		
		String fileUploadPath = (String) fileInfo.get("fileUploadPath");	// 파일 업로드 경로
		String fileLogicName = (String) fileInfo.get("fileLogicName");		// 파일 논리명(화면에 표시될 파일 이름)
		String filePhysicName = (String) fileInfo.get("filePhysicName");	// 파일 물리명(실제 저장된 파일 이름)
		
		File file = new File(fileUploadPath, filePhysicName);
		
		response.setContentType(getContentType());
		response.setContentLength((int) file.length());
		String userAgent = request.getHeader("User-Agent");
		boolean ie = userAgent.indexOf("MSIE") > -1;
		String fileName = null;

if(ie) {
			fileName = URLEncoder.encode(fileLogicName, "utf-8");
		} else {
			fileName = new String(fileLogicName.getBytes("utf-8"), "iso-8859-1");
		}

response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + "\";");
		response.setHeader("Content-Transfer-Encoding", "binary");
		OutputStream out = response.getOutputStream();

FileInputStream fis = null;

try {
			fis = new FileInputStream(file);
			FileCopyUtils.copy(fis, out);
		} finally {
			if(fis != null)
				try {
					fis.close();
				} catch(IOException ex) {
				}
		}
		out.flush();
	}
}

3. 컨트롤러에서 처리 할 부분을 만들어 준다. 다음은 첨부파일 일련번호를 파라메터로 받아서 첨부파일 정보를 추출 해 낸 후 아까만든 뷰 클래스에 던져주는 예 이다.

※ 보충 설명을 하자면 뷰 리졸버로 사용될 fileDownloadView 을 만들었으니 컨트롤러 단에서 fileDownloadView 에

downloadFile 라는 이름으로 모델을 던저준 것이다. fileDownloadView 에서는 그 정보를 받아서 다운로드 처리를 해 주는 것이다.