[Golang] Dùng Lambda Để Lưu và Lấy Image Từ AWS S3

Mục Tiêu

• Lưu ảnh lên Amazon Simple Storage Service
• Lấy ảnh từ Amazon Simple Storage Service, giới hạn TTL

Chuẩn Bị

• Bạn có đọc thêm series để nắm căn bản: https://viblo.asia/s/golang-lambda-serverless-vElaB8eD5kw

API Lưu Image

• Image được input dưới dạng base64

Cấu Hình ENV

Trong file serverless.yaml, add 2 env

provider: environment: S3_BUCKET: "images-dev-serverlessdeploymentbucket-1g3trol0zorpp" IMAGE_BLOG: "image-blog/"

• Sau khi deploy thành công, thì truy cập aws-console search S3 service để check bucket của project => copy và bỏ vào env
• Env IMAGE_BLOG là image sẽ được lưu sub-folder trong bucket

List function Code

Tạo AWS session

func CreateSession(req AwsReq) *session.Session { sess := session.Must(session.NewSessionWithOptions(session.Options{ SharedConfigState: session.SharedConfigEnable, })) return sess
}

• Để có thể tạo được session, yêu cầu cấu hình credential trước. Checkout Cấu Hình AWS Credential để cấu hình credential

Func kiểm tra size của image

// fn upload image to s3, request iamge is base64
func CheckSizeBase64(baseImage *string) error { sizeByte := base64.StdEncoding.DecodedLen(len(*baseImage)) sizeRe := 5000000 if sizeByte > sizeRe { err := fmt.Errorf("size image large, require size %d and size input %d", sizeRe, sizeByte) return err } return nil
}

• Image từ client dưới dạng base64 => check size bằng cách kiểm tra số lượng kí tự của base64
• Func check số lượng kí tự vượt quá 5 triệu thì sẽ báo lỗi

Func Lưu image đến s3

func StoreS3(ctx context.Context, reqDTO StoreS3Req) (*s3.PutObjectOutput, StoreS3Reponse, error) { var ( dataRes = StoreS3Reponse{} err error putOut = &s3.PutObjectOutput{} ) if reqDTO.BaseImage == nil { return putOut, dataRes, fmt.Errorf("base64 of image is require") } if len(*reqDTO.BaseImage) == 0 { return putOut, dataRes, fmt.Errorf("byte image decode empty") } splitBase := strings.Split(*reqDTO.BaseImage, "base64,") var newBase string if len(splitBase) > 1 { newBase = splitBase[1] } else { newBase = splitBase[0] } byteImage, err := base64.StdEncoding.DecodeString(newBase) if err != nil { fmt.Println("image decode string err", err) return putOut, dataRes, err } mimeType := http.DetectContentType(byteImage) fmt.Println("mime type:", mimeType) if !strings.Contains(mimeType, "jpeg") && !strings.Contains(mimeType, "png") { return putOut, dataRes, fmt.Errorf("mime type error") } byteReader := bytes.NewReader(byteImage) dataRes.PathImage = reqDTO.PathImage session := CreateSession(AwsReq{ CustomEndpoint: reqDTO.CustomEndpoint, }) putOut, err = s3.New(session).PutObject(&s3.PutObjectInput{ Body: byteReader, Bucket: aws.String(GetS3Bucket()), Key: aws.String(dataRes.PathImage), ContentType: &mimeType, }) return putOut, dataRes, err
}

• Line code mimeType := http.DetectContentType(byteImage) kiểm tra type của image, chỉ handle 2 dạng type ['png','jpeg']

Một số func phụ khác:

• Get bucket name từ environment:

func GetS3Bucket() string { bucket := os.Getenv("S3_BUCKET") if bucket != "" { return bucket } log.Default().Println("GetS3Bucket default value:", "pkg.BucketDefault") return "pkg.BucketDefault"
}

=> lấy bucket-name thừ environment, nếu env rỗng thì sử dụng bucket-name mặc định.

Test API

• Request data:

{ "requestId":"", "requestTime":"", "data": { "file_name": "ducnp", "name":"ducnp5-test1", "phone":"0335288888", "partner_id":"myst", "user_id":"ducnp", "image":"****am6firEVHTsD//2Q==****" }
}

• Response data:. Truy cập S3 thì sẽ thấy ảnh đã lưu thành công.

ERROR

• Có thể xuất hiện lỗi này: ==> truy cập vào cloud-watch, để check error. Thường là do chưa cấp quyền cho Role access vào S3.

API Get Image

• Tương tự như lưu ảnh, API get image thì chỉ khác 1 chỗ đó là hàm PresignUrl

List function Code

PresignUrl

func PresignUrl(reqAws AwsReq, presign PresignUrlReq) (string, error) { var ( url string err error ) clientS3 := CreateS3Client(reqAws) req, _ := clientS3.GetObjectRequest(&s3.GetObjectInput{ Bucket: aws.String(presign.Bucket), Key: aws.String(presign.KeyPath), }) url, err = req.Presign(presign.UrlTtl) if err != nil { return url, fmt.Errorf("presign url err %s", err) } return url, err
}

• Trong func có thuộc tính TTL là giới hạn thời gian access đến image.

Test API

• Request:

{ "requestId":"", "requestTime":"", "data": { "pathImage": "image-blog/0335288888landucnp5-test1" }
}

• Response:

{ "data": { "url": "https://images-dev-serverle****blog/0335288888landucnp5-test1?X-Amz-***mz-Credential=ASIAQ52H7T***" }, "responseCode": "00", "responseId": "8b2f9168-4ad4-4f6c-b0b6-8b78f3d7c99d", "responseMessage": "successfully", "responseTime": "2023-02-05T01:28:17Z"
}

==> copy url và mở bằng browser để xem.

• Image hết hạn:

Contact

• facebook: https://www.facebook.com/phucducdev/
• gmail: ducnp09081998@gmail.com or phucducktpm@gmail.com
• linkedin: https://www.linkedin.com/in/phucducktpm/
• hashnode: https://hashnode.com/@OpenDev
• telegram: https://t.me/OpenDevGolang