feat: S3 업로드, 조회 로직 구현 (#158)

* feat: S3 업로드, 조회 로직 구현

* fix: 매핑 url 오탈자 수정

Author: Ji-minhyeok

build.gradle

@@ -33,6 +33,7 @@ dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-thymeleaf'
     implementation 'org.springdoc:springdoc-openapi-starter-webmvc-ui:2.8.9'
     implementation 'io.awspring.cloud:spring-cloud-starter-aws:2.4.4'
+    implementation 'io.awspring.cloud:spring-cloud-starter-aws-secrets-manager-config:2.4.4'
     implementation 'com.opencsv:opencsv:5.11.2'
     compileOnly 'org.projectlombok:lombok'
     runtimeOnly 'com.mysql:mysql-connector-j'

src/main/java/com/threestar/trainus/domain/file/controller/S3Controller.java

@@ -0,0 +1,32 @@
+package com.threestar.trainus.domain.file.controller;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.HttpStatusCode;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+import com.threestar.trainus.domain.file.dto.GetS3UrlDto;
+import com.threestar.trainus.domain.file.service.S3Service;
+import com.threestar.trainus.global.annotation.LoginUser;
+
+@RestController
+@RequiredArgsConstructor
+@RequestMapping("/api/v1/s3")
+public class S3Controller {
+	private final S3Service s3Service;
+
+	@GetMapping(value = "/posturl")
+	public ResponseEntity<GetS3UrlDto> getPostS3Url(@LoginUser Long userId, String filename) {
+		GetS3UrlDto getS3UrlDto = s3Service.getPostS3Url("image/"+userId, filename);
+		return new ResponseEntity<>(getS3UrlDto, HttpStatusCode.valueOf(200));
+	}
+
+	@GetMapping(value = "/geturl")
+	public ResponseEntity<GetS3UrlDto> getGetS3Url(@LoginUser Long userId, @RequestParam String key) {
+		GetS3UrlDto getS3UrlDto = s3Service.getGetS3Url(userId, key);
+		return new ResponseEntity<>(getS3UrlDto, HttpStatusCode.valueOf(200));
+	}
+}

src/main/java/com/threestar/trainus/domain/file/dto/GetS3UrlDto.java

@@ -0,0 +1,22 @@
+package com.threestar.trainus.domain.file.dto;
+
+import lombok.Builder;
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@Getter
+@Setter
+@NoArgsConstructor
+@EqualsAndHashCode
+public class GetS3UrlDto {
+	private String preSignedUrl;
+	private String key;
+
+	@Builder
+	public GetS3UrlDto(String preSignedUrl, String key) {
+		this.preSignedUrl = preSignedUrl;
+		this.key = key;
+	}
+}

src/main/java/com/threestar/trainus/domain/file/service/S3Service.java

@@ -0,0 +1,110 @@
+package com.threestar.trainus.domain.file.service;
+
+import java.net.URL;
+import java.util.Date;
+import java.util.Map;
+import java.util.UUID;
+
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import com.amazonaws.HttpMethod;
+import com.amazonaws.services.s3.AmazonS3;
+import com.amazonaws.services.s3.Headers;
+import com.amazonaws.services.s3.model.CannedAccessControlList;
+import com.amazonaws.services.s3.model.GeneratePresignedUrlRequest;
+import com.threestar.trainus.domain.file.dto.GetS3UrlDto;
+
+import lombok.RequiredArgsConstructor;
+
+@Service
+@RequiredArgsConstructor
+public class S3Service {
+	@Value("${cloud.aws.s3.bucket}")
+	private String bucket;
+
+	private final AmazonS3 amazonS3;
+
+	public GetS3UrlDto getPostS3Url(String prefix, String fileName) {
+		if (!prefix.isEmpty()) {
+			fileName = createPath(prefix, fileName);
+		}
+
+		GeneratePresignedUrlRequest generatePresignedUrlRequest = getGeneratePresignedUrlRequest(bucket, fileName);
+		URL url = amazonS3.generatePresignedUrl(generatePresignedUrlRequest);
+
+		return GetS3UrlDto.builder()
+			.preSignedUrl(url.toString())
+			.key(fileName)
+			.build();
+	}
+
+	// 파일 업로드용 Presigned URL 생성
+	private GeneratePresignedUrlRequest getGeneratePresignedUrlRequest(String bucket, String fileName) {
+		GeneratePresignedUrlRequest generatePresignedUrlRequest = new GeneratePresignedUrlRequest(bucket, fileName)
+			.withMethod(HttpMethod.PUT)
+			.withExpiration(getPresignedUrlExpiration());
+
+		generatePresignedUrlRequest.addRequestParameter(
+			Headers.S3_CANNED_ACL,
+			CannedAccessControlList.PublicRead.toString()
+		);
+
+		return generatePresignedUrlRequest;
+	}
+
+	//Presigned URL 유효기간 설정
+	private Date getPresignedUrlExpiration() {
+		Date expiration = new Date();
+		long expTimeMillis = expiration.getTime();
+		expTimeMillis += 1000 * 60 * 10;
+		expiration.setTime(expTimeMillis);
+
+		return expiration;
+	}
+
+	//UUID를 사용하여 파일 고유 ID 생성
+	private String createFileId() {
+		return UUID.randomUUID().toString();
+	}
+
+	//파일의 전체 경로 생성
+	private String createPath(String prefix, String fileName) {
+		String fileId = createFileId();
+		return String.format("%s/%s", prefix, fileId + "-" + fileName);
+	}
+
+	// get 용 URL 생성
+	private GeneratePresignedUrlRequest getGetGeneratePresignedUrlRequest(String key, Date expiration) {
+		return new GeneratePresignedUrlRequest(bucket, key)
+			.withMethod(HttpMethod.GET)
+			.withExpiration(expiration);
+	}
+
+	@Transactional(readOnly = true)
+	public GetS3UrlDto getGetS3Url(Long memberId, String key) {
+		// url 유효기간 설정하기(1시간)
+		Date expiration = getExpiration();
+
+		// presigned url 생성하기
+		GeneratePresignedUrlRequest generatePresignedUrlRequest =
+			getGetGeneratePresignedUrlRequest(key, expiration);
+
+		URL url = amazonS3.generatePresignedUrl(generatePresignedUrlRequest);
+
+		// return
+		return GetS3UrlDto.builder()
+			.preSignedUrl(url.toExternalForm())
+			.key(key)
+			.build();
+	}
+
+	private static Date getExpiration() {
+		Date expiration = new Date();
+		long expTimeMillis = expiration.getTime();
+		expTimeMillis += 1000 * 60 * 10; // 10분으로 설정하기
+		expiration.setTime(expTimeMillis);
+		return expiration;
+	}
+}