[Java] 자바 HttpUrlConnection 을 이용하여 파일 다운받기

개요

---

솔루션 엔지니어로서 일을 하다보면 종종 외부 라이브러리의 도움 없이 HttpUrlConnection 을 이용하여 HTTP 통신 테스트를 하는 경우가 자주 있다. 필자는 이를 쌩자바 라고 부르는데, 이번 포스트에서는 쌩자바를 이용한 HTTP 파일 다운로드 📂 하는 방법을 정리해보려고 한다.

본문

---

기본적인 흐름은 다음과 같이 크게 2가지로 나눌 수 있다.

1. HttpUrlConnection 연결
2. Stream 을 통해 다운로드

1. HttpUrlConnection 연결

• URL 객체 생성
• URL 객체로부터 HttpUrlConnection 객체를 받아옴

import java.net.HttpURLConnection;
import java.net.URL;

public class FileDownloadTest {

    public static void main(String[] args) {
    	String spec = "https://file-examples-com.github.io/uploads/2017/02/file-sample_100kB.doc";
        String outputDir = "D:/sample/output/download";

        try{
            URL url = new URL(spec);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            int responseCode = conn.getResponseCode();
            System.out.println("responseCode " + responseCode);
       	} catch (Exception e){
            System.out.println("An error occurred while trying to download a file.");
            e.printStackTrace();
        }
}

console : responseCode 200

통신에 성공했으면 남은 작업은 Stream 을 통해 파일을 가져오자.

2. Stream 을 통해 다운로드

• Content-Disposition or URL 에서 파일명 가져오기
• InputStream ➡️ FileOutputStream 을 통해 파일 다운로드

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

public class FileDownloadTest {

    public static void main(String[] args) {
        String spec = "https://file-examples-com.github.io/uploads/2017/02/file-sample_100kB.doc";
        String outputDir = "D:/sample/output/download";
        InputStream is = null;
        FileOutputStream os = null;
        try{
            URL url = new URL(spec);
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            int responseCode = conn.getResponseCode();

            System.out.println("responseCode " + responseCode);

            // Status 가 200 일 때
            if (responseCode == HttpURLConnection.HTTP_OK) {
                String fileName = "";
                String disposition = conn.getHeaderField("Content-Disposition");
                String contentType = conn.getContentType();
                
                // 일반적으로 Content-Disposition 헤더에 있지만 
                // 없을 경우 url 에서 추출해 내면 된다.
                if (disposition != null) {
                    String target = "filename=";
                    int index = disposition.indexOf(target);
                    if (index != -1) {
                        fileName = disposition.substring(index + target.length() + 1);
                    }
                } else {
                    fileName = spec.substring(spec.lastIndexOf("/") + 1);
                }

                System.out.println("Content-Type = " + contentType);
                System.out.println("Content-Disposition = " + disposition);
                System.out.println("fileName = " + fileName);

                is = conn.getInputStream();
                os = new FileOutputStream(new File(outputDir, fileName));

                final int BUFFER_SIZE = 4096;
                int bytesRead;
                byte[] buffer = new byte[BUFFER_SIZE];
                while ((bytesRead = is.read(buffer)) != -1) {
                    os.write(buffer, 0, bytesRead);
                }
                os.close();
                is.close();
                System.out.println("File downloaded");
            } else {
                System.out.println("No file to download. Server replied HTTP code: " + responseCode);
            }
            conn.disconnect();
        } catch (Exception e){
            System.out.println("An error occurred while trying to download a file.");
            e.printStackTrace();
            try {
                if (is != null){
                    is.close();
                }
                if (os != null){
                    os.close();
                }
            } catch (IOException e1){
                e1.printStackTrace();
            }
        }
    }
}

console :

responseCode 200
Content-Type = application/msword
Content-Disposition = null
fileName = file-sample_100kB.doc
File downloaded

Process finished with exit code 0

정리

---

이와같이 크게 2가지 작업만 해주면 쌩자바를 이용해 쉽게 파일을 다운받을 수 있다.
이번 포스트는 기본적으로 HttpUrlConnection 객체를 통해 오픈된 파일을 다운받았지만, 나중에 기회가 된다면 setRequestProperty() 등의 메서드를 이용해 토큰값이 있어야 접근 가능한 url 등에 대해서도 정리해보겠다.