저번 포스팅 [NestJS] Using AWS S3 Presigned URL에서 이어집니다.

위 포스팅에서 Presigned URL 개념에 대해 다루니 참고하시길 바랍니다.

0. Overview

저번 포스팅에서 Presigned URL을 통해 S3에 업로드할 수 있도록 해보았다. 대충 요약하자면 Presigned URL을 사용하지 않았을 경우 아래와 같은 구조를 구현해야 했다.

그런데 이러면 서버를 거쳐 파일을 업로드하기 때문에 비효율적일 수 있다. 그래서 Presigned URL을 사용하게 되면 임시적으로 업로드 권한이 담긴 URL을 발급받고, 클라이언트는 직접 그 URL에 업로드하는 방식이다.

그런데 문제가 있다. 버킷에서 직접적으로 파일의 크기 등을 체크하여 거부하거나 하는 기능은 제공하지 않는다. 더군다나 서버를 거치지 않고 직접 업로드하기 때문에 큰 문제가 될 수 있다.

출처: AWS re:Post - S3 PUT file size limit

요약하자면 버킷 정책에서 파일 크기 제한같은건 못하니 아래와 같은 방법을 사용하라는 말이다.

람다를 연결해서 크기 체크해라.
클라이언트에서 업로드하기 전에 서버에서 체크해라. (Presigned URL이라 못함)
멀티파트 방식(POST 방식)을 쓰고 거기에 제한을 둬라.

두번째는 애초에 불가능하고, 먼저 첫번째 방법을 사용하면 아래와 같이 사용할 수 있을 것이다.

S3에 업로드하게 되면 연결된 람다(또는 Lambda@Edge)가 파일의 사이즈를 체크해보고, 제한을 넘는다 싶으면 거부하거나 파일(객체)를 삭제하는 방식이다.

그런데 이 방식은 일단 S3에 업로드하고 뭘 한다는 점과, 람다 실행 시간도 발생한다는 문제가 있다. 애초에 업로드되기 전에 자체적으로 체크하면 되는데 말이다.

그래서 3번 방식인 POST 방식의 Presigned URL을 사용하는 방법이 있다.

원래 S3 Presigned URL를 통해 파일을 업로드하려면 HTTP PUT 방식을 써서 업로드를 하게 된다.

하지만 위 사진처럼 HTTP POST 방식과 멀티파트 데이터를 통해 업로드를 하게 되면 사이즈 제한이나 Content-Type 헤더 내용 제한과 같은 조건을 추가할 수 있게 된다. (그 조건들은 여기에서 확인할 수 있다.)

1. Using in NestJS

먼저 HTTP PUT 방식을 사용하던 Presigned URL 코드에선 @aws-sdk/s3-request-presigner 라이브러리를 설치하여 사용하였으나, POST 방식에선 @aws-sdk/s3-presigned-post 라이브러리를 설치해서 사용해야 한다.

먼저 S3 클라이언트 선언은 똑같은데, NestJS 모듈에 아래와 같이 작성해두었다. 이전 글에서 다뤘으니 참고하도록 하자.

// upload.module.ts
 
import { Module } from '@nestjs/common'
import { UploadController } from './upload.controller'
import { UploadService } from './upload.service'
import { ConfigService } from '@nestjs/config'
import { S3Client } from '@aws-sdk/client-s3'
 
@Module({
    controllers: [UploadController],
    providers: [
        {
            provide: 'S3_CLIENT',
            inject: [ConfigService],
            useFactory: (configService: ConfigService) => {
                return new S3Client({
                    region: configService.get<string>('AWS_REGION') ?? 'ap-northeast-2',
                    credentials: {
                        accessKeyId: configService.get('AWS_ACCESS_KEY_ID') ?? '',
                        secretAccessKey: configService.get('AWS_SECRET_ACCESS_KEY') ?? '',
                    },
                })
            },
        },
        UploadService,
    ],
    exports: [UploadService],
})
export class UploadModule {}

다음으로 서비스 코드를 살펴보자.

// upload.service.ts
 
import { Inject, Injectable } from '@nestjs/common'
import { S3Client } from '@aws-sdk/client-s3'
import { createPresignedPost } from '@aws-sdk/s3-presigned-post'
import { ConfigService } from '@nestjs/config'
import { GeneratePresignedUrlRequestDto, PresignedUrlResponseDto } from './dto' // DTO 코드 생략
import { v4 as uuidv4 } from 'uuid'
 
@Injectable()
export class UploadService {
    constructor(
        @Inject('S3_CLIENT')
        private readonly s3: S3Client,
        private readonly configService: ConfigService,
    ) {}
 
    async generatePresignedUrl(dto: GeneratePresignedUrlRequestDto): Promise<PresignedUrlResponseDto> {
        const key = `uploads/${uuidv4()}/${dto.filename}`
 
        const { url, fields } = await createPresignedPost(this.s3, {
            Bucket: this.configService.get<string>('AWS_S3_BUCKET_NAME') ?? '',
            Key: key,
            Fields: {
                'Content-Type': dto.contentType,
            },
            Conditions: [
                ['content-length-range', 0, 3 * 1024 * 1024], // Max 3MB
                ['starts-with', '$Content-Type', 'image/'],
            ],
            Expires: 600, // 10 minutes
        })
 
        return { url, key, fields }
    }
}