wip

Author: hienluongandpadcojp

aws/awss3/awss3.go

@@ -621,3 +621,23 @@ func SelectCSVHeaders(ctx context.Context, region awsconfig.Region, bucketName B
 	}
 	return headers, nil
 }
+
+func PresignPutObject(ctx context.Context, region awsconfig.Region, bucketName BucketName, key Key, opts ...s3presigned.OptionS3Presigned) (string, error) {
+	c := s3presigned.GetS3PresignedConf(opts...)
+	client, err := GetClient(ctx, region) // nolint:typecheck
+	if err != nil {
+		return "", err
+	}
+	input := &s3.PutObjectInput{
+		Bucket: bucketName.AWSString(),
+		Key:    key.AWSString(),
+	}
+	ps := s3.NewPresignClient(client)
+	resp, err := ps.PresignPutObject(ctx, input, func(o *s3.PresignOptions) {
+		o.Expires = c.PresignExpires
+	})
+	if err != nil {
+		return "", err
+	}
+	return resp.URL, nil
+}

aws/awss3/awss3_test.go

@@ -6,6 +6,7 @@ import (
 	"encoding/csv"
 	"errors"
 	"fmt"
+	"net/http"
 	"os"
 	"path/filepath"
 	"strings"
@@ -994,3 +995,53 @@ func TestSelectCSVHeaders(t *testing.T) {
 		assert.Error(t, err)
 	})
 }
+
+func TestPresignPutObject(t *testing.T) {
+	t.Parallel()
+	ctx := ctxawslocal.WithContext(
+		context.Background(),
+		ctxawslocal.WithS3Endpoint("http://127.0.0.1:29000"), // use Minio
+		ctxawslocal.WithAccessKey("DUMMYACCESSKEYEXAMPLE"),
+		ctxawslocal.WithSecretAccessKey("DUMMYSECRETKEYEXAMPLE"),
+	)
+
+	uploadTxtByPresignedPutObjectURL := func(presignedURL string) error {
+		content := []byte("Hello World")
+		req, err := http.NewRequest(http.MethodPut, presignedURL, bytes.NewReader(content))
+		if err != nil {
+			return err
+		}
+
+		req.Header.Set("Content-Type", "text/plain")
+		resp, err := http.DefaultClient.Do(req)
+		if err != nil {
+			return err
+		}
+
+		defer resp.Body.Close()
+
+		if resp.StatusCode != http.StatusOK {
+			return fmt.Errorf("failed to upload file, status code: %d", resp.StatusCode)
+		}
+
+		return nil
+	}
+
+	confirmedUploadedObject := func(ctx context.Context, key awss3.Key) error {
+		_, err := awss3.HeadObject(ctx, TestRegion, TestBucket, key)
+		return err
+	}
+
+	t.Run("Presign", func(t *testing.T) {
+		t.Parallel()
+		key := awss3.Key("test_presign_put_object_01.txt")
+		pURL, err := awss3.PresignPutObject(ctx, TestRegion, TestBucket, key)
+		assert.NoError(t, err)
+		assert.NotEmpty(t, pURL)
+
+		err = uploadTxtByPresignedPutObjectURL(pURL)
+		assert.NoError(t, err)
+		err = confirmedUploadedObject(ctx, key)
+		assert.NoError(t, err)
+	})
+}