Skip to content

스프링단에서 업로드 파일 크기 제한, 게시판 업로드 사진 자동 리사이징 기능 #102

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
leesewon00 merged 1 commit into from
Oct 6, 2024
Merged

스프링단에서 업로드 파일 크기 제한, 게시판 업로드 사진 자동 리사이징 기능 #102

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
51 changes: 51 additions & 0 deletions src/main/java/com/example/solidconnection/s3/FileUploadService.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
package com.example.solidconnection.s3;

import com.amazonaws.AmazonServiceException;
import com.amazonaws.SdkClientException;
import com.amazonaws.services.s3.AmazonS3Client;
import com.amazonaws.services.s3.model.CannedAccessControlList;
import com.amazonaws.services.s3.model.ObjectMetadata;
import com.amazonaws.services.s3.model.PutObjectRequest;
import com.example.solidconnection.custom.exception.CustomException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

import static com.example.solidconnection.custom.exception.ErrorCode.S3_CLIENT_EXCEPTION;
import static com.example.solidconnection.custom.exception.ErrorCode.S3_SERVICE_EXCEPTION;

@Component
@EnableAsync
@Slf4j
public class FileUploadService {
private final AmazonS3Client amazonS3;

public FileUploadService(AmazonS3Client amazonS3) {
this.amazonS3 = amazonS3;
}

@Async
public void uploadFile(String bucket, String fileName, MultipartFile multipartFile) {
// 메타데이터 생성
String contentType = multipartFile.getContentType();
ObjectMetadata metadata = new ObjectMetadata();
metadata.setContentType(contentType);
metadata.setContentLength(multipartFile.getSize());

try {
amazonS3.putObject(new PutObjectRequest(bucket, fileName, multipartFile.getInputStream(), metadata)
.withCannedAcl(CannedAccessControlList.PublicRead));
log.info("이미지 업로드 정상적 완료 thread: {}", Thread.currentThread().getName());
} catch (AmazonServiceException e) {
log.error("이미지 업로드 중 s3 서비스 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_SERVICE_EXCEPTION);
} catch (SdkClientException | IOException e) {
log.error("이미지 업로드 중 s3 클라이언트 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_CLIENT_EXCEPTION);
}
}
}
64 changes: 17 additions & 47 deletions src/main/java/com/example/solidconnection/s3/S3Service.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,7 @@
import com.amazonaws.AmazonServiceException;
import com.amazonaws.SdkClientException;
import com.amazonaws.services.s3.AmazonS3Client;
import com.amazonaws.services.s3.model.CannedAccessControlList;
import com.amazonaws.services.s3.model.DeleteObjectRequest;
import com.amazonaws.services.s3.model.ObjectMetadata;
import com.amazonaws.services.s3.model.PutObjectRequest;
import com.example.solidconnection.custom.exception.CustomException;
import com.example.solidconnection.siteuser.domain.SiteUser;
import com.example.solidconnection.siteuser.repository.SiteUserRepository;
Expand All @@ -15,10 +12,10 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;

import static com.example.solidconnection.custom.exception.ErrorCode.FILE_NOT_EXIST;
Expand All @@ -34,8 +31,11 @@ public class S3Service {
private static final Logger log = LoggerFactory.getLogger(S3Service.class);
private final AmazonS3Client amazonS3;
private final SiteUserRepository siteUserRepository;
private final FileUploadService fileUploadService;
private final ThreadPoolTaskExecutor asyncExecutor;
@Value("${cloud.aws.s3.bucket}")
private String bucket;
private final long MAX_FILE_SIZE_MB = 1024 * 1024 * 3;

/*
* 파일을 S3에 업로드한다.
Expand All @@ -44,65 +44,35 @@ public class S3Service {
* - 파일에 대한 메타 데이터를 생성한다.
* - 임의의 랜덤한 문자열로 파일 이름을 생성한다.
* - S3에 파일을 업로드한다.
* - 3mb 이상의 파일은 /origin/ 경로로 업로드하여 lambda 함수로 리사이징 진행한다.
* - 3mb 미만의 파일은 바로 업로드한다.
* */
public UploadedFileUrlResponse uploadFile(MultipartFile multipartFile, ImgType imageFile) {
// 파일 검증
validateImgFile(multipartFile);

// 메타데이터 생성
String contentType = multipartFile.getContentType();
ObjectMetadata metadata = new ObjectMetadata();
metadata.setContentType(contentType);
metadata.setContentLength(multipartFile.getSize());

// 파일 이름 생성
UUID randomUUID = UUID.randomUUID();
String fileName = imageFile.getType() + "/" + randomUUID;

try {
amazonS3.putObject(new PutObjectRequest(bucket, fileName, multipartFile.getInputStream(), metadata)
.withCannedAcl(CannedAccessControlList.PublicRead));
} catch (AmazonServiceException e) {
log.error("이미지 업로드 중 s3 서비스 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_SERVICE_EXCEPTION);
} catch (SdkClientException | IOException e) {
log.error("이미지 업로드 중 s3 클라이언트 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_CLIENT_EXCEPTION);
// 파일업로드 비동기로 진행
if (multipartFile.getSize() >= MAX_FILE_SIZE_MB) {
asyncExecutor.submit(() -> {
fileUploadService.uploadFile(bucket, "origin/" + fileName, multipartFile);
});
} else {
asyncExecutor.submit(() -> {
fileUploadService.uploadFile(bucket, fileName, multipartFile);
});
}
return new UploadedFileUrlResponse(fileName);
}

public List<UploadedFileUrlResponse> uploadFiles(List<MultipartFile> multipartFile, ImgType imageFile) {

List<UploadedFileUrlResponse> uploadedFileUrlResponseList = new ArrayList<>();

for (MultipartFile file : multipartFile) {
// 파일 검증
validateImgFile(file);

// 메타데이터 생성
String contentType = file.getContentType();
ObjectMetadata metadata = new ObjectMetadata();
metadata.setContentType(contentType);
metadata.setContentLength(file.getSize());

// 파일 이름 생성
UUID randomUUID = UUID.randomUUID();
String fileName = imageFile.getType() + "/" + randomUUID;

try {
amazonS3.putObject(new PutObjectRequest(bucket, fileName, file.getInputStream(), metadata)
.withCannedAcl(CannedAccessControlList.PublicRead));
} catch (AmazonServiceException e) {
log.error("이미지 업로드 중 s3 서비스 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_SERVICE_EXCEPTION);
} catch (SdkClientException | IOException e) {
log.error("이미지 업로드 중 s3 클라이언트 예외 발생 : {}", e.getMessage());
throw new CustomException(S3_CLIENT_EXCEPTION);
}
uploadedFileUrlResponseList.add(new UploadedFileUrlResponse(fileName));
UploadedFileUrlResponse uploadedFileUrlResponse = uploadFile(file, imageFile);
uploadedFileUrlResponseList.add(uploadedFileUrlResponse);
}

return uploadedFileUrlResponseList;
}

Expand Down