본문 바로가기

Study Information Technology

Spring Boot의 파일 업로드 및 다운로드 지원

728x90
반응형

Spring Boot의 파일 업로드 및 다운로드 지원

Overview

Spring Boot는 파일 업로드와 다운로드를 효율적으로 처리할 수 있는 강력한 기능을 제공합니다. 이를 통해 웹 애플리케이션에서 파일 관리가 용이해지며, 개발자들은 복잡한 설정 없이도 필요한 기능을 손쉽게 구현할 수 있습니다. 이번 글에서는 Spring Boot에서 파일 업로드 및 다운로드를 구현하는 방법을 자세히 설명하겠습니다.

1. 파일 업로드 구현하기

파일 업로드를 구현하기 위해서는 MultipartFile 클래스를 사용합니다. 다음은 기본적인 파일 업로드 예제입니다.

예제: 파일 업로드 컨트롤러

import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.http.ResponseEntity;

@RestController
@RequestMapping("/api/files")
public class FileUploadController {

@PostMapping("/upload")
public ResponseEntity<String> uploadFile(@RequestParam("file") MultipartFile file) {
if (file.isEmpty()) {
return ResponseEntity.badRequest().body("파일이 비어있습니다.");
}

// 파일 저장 로직 (예: 로컬 파일 시스템)
String filePath = "/uploads/" + file.getOriginalFilename();
try {
file.transferTo(new File(filePath));
} catch (IOException e) {
return ResponseEntity.status(500).body("파일 저장 중 오류 발생: " + e.getMessage());
}

return ResponseEntity.ok("파일 업로드 성공: " + filePath);
}
}

코드 설명:

  • @RestController@RequestMapping 어노테이션을 사용하여 REST API를 정의합니다.
  • uploadFile 메소드는 파일을 업로드하는 엔드포인트로, @RequestParam을 통해 MultipartFile 객체를 받습니다.
  • 파일이 비어 있는 경우 적절한 에러 메시지를 반환하고, 파일이 성공적으로 저장되면 업로드 성공 메시지를 반환합니다.

2. 파일 다운로드 구현하기

파일 다운로드를 구현하기 위해서는 클라이언트가 요청한 파일을 찾아서 응답으로 반환해야 합니다. 다음은 파일 다운로드의 기본 예제입니다.

예제: 파일 다운로드 컨트롤러

import org.springframework.web.bind.annotation.*;
import org.springframework.core.io.Resource;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.ResponseEntity;
import org.springframework.http.HttpHeaders;

@RestController
@RequestMapping("/api/files")
public class FileDownloadController {

@GetMapping("/download/{filename}")
public ResponseEntity<Resource> downloadFile(@PathVariable String filename) {
FileSystemResource resource = new FileSystemResource("/uploads/" + filename);

if (!resource.exists()) {
return ResponseEntity.notFound().build();
}

return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + resource.getFilename() + "\"")
.body(resource);
}
}

코드 설명:

  • @GetMapping을 사용하여 파일 다운로드 엔드포인트를 정의합니다.
  • 요청된 파일이 존재하지 않는 경우 404 Not Found를 반환하고, 파일이 존재하면 Content-Disposition 헤더를 설정하여 브라우저가 파일을 다운로드하도록 합니다.

3. 파일 업로드 및 다운로드 설정

파일 업로드와 다운로드 기능을 제대로 사용하기 위해서는 application.properties 파일에서 몇 가지 설정을 추가해야 합니다.

설정 예시:

# 파일 업로드 크기 제한
spring.servlet.multipart.max-file-size=10MB
spring.servlet.multipart.max-request-size=10MB

# 파일 저장 경로
file.upload-dir=/uploads

이 설정은 최대 파일 크기를 10MB로 제한하고, 파일이 저장될 경로를 설정합니다. 이러한 설정을 통해 파일 업로드 시 용량 초과로 인한 오류를 방지할 수 있습니다.

4. 에러 처리

파일 업로드 및 다운로드 과정에서 발생할 수 있는 일반적인 에러와 그 해결 방법을 알아보겠습니다.

에러: 파일 크기 초과

  • 에러 메시지: org.springframework.web.multipart.MaxUploadSizeExceededException
  • 해결 방법: application.properties에서 max-file-sizemax-request-size를 적절히 조정합니다.

에러: 파일이 존재하지 않음

  • 에러 메시지: 404 Not Found
  • 해결 방법: 요청된 파일이 올바른 경로에 있는지 확인합니다. 파일 이름이 정확한지, 경로 설정이 올바른지 점검합니다.

5. 보안 고려사항

파일 업로드 및 다운로드 기능을 구현할 때 보안도 매우 중요합니다. 다음은 보안을 강화하기 위한 몇 가지 방법입니다.

  • 파일 유형 검증: 업로드된 파일의 MIME 타입을 확인하여 허용된 파일 유형만 처리하도록 합니다.
  • 파일 이름 정제: 사용자가 입력한 파일 이름에 대해 공격적인 문자(예: ../)를 제거하여 디렉토리 탐색 공격을 방지합니다.
  • 업로드 경로 제한: 파일이 저장될 경로를 설정할 때 외부 접근이 불가능한 경로를 사용하여 보안을 강화합니다.

6. 참고 문서

이와 같은 방식으로 Spring Boot를 사용하여 파일 업로드 및 다운로드 기능을 구현하면, 개발자는 손쉽게 파일 관리를 할 수 있으며, 최종 사용자에게도 원활한 경험을 제공할 수 있습니다.

반응형