80 lines
1.5 KiB
Go
80 lines
1.5 KiB
Go
package handle
|
|
|
|
import (
|
|
"allapp-go/internal/errors"
|
|
"allapp-go/internal/httpx"
|
|
"allapp-go/pkg/s3store"
|
|
"fmt"
|
|
"mime/multipart"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
"github.com/gofiber/fiber/v3"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func S3ListBuckets(c fiber.Ctx) error {
|
|
result, err := s3store.S3Client.ListBuckets(c.Context(), &s3.ListBucketsInput{})
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
buckets := make([]string, 0, len(result.Buckets))
|
|
for _, b := range result.Buckets {
|
|
if b.Name != nil {
|
|
buckets = append(buckets, *b.Name)
|
|
}
|
|
}
|
|
|
|
return httpx.OK(c, buckets)
|
|
}
|
|
|
|
func S3UploadFile(c fiber.Ctx) error {
|
|
fileHeader, err := c.FormFile("file")
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
file, err := fileHeader.Open()
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
defer func(file multipart.File) {
|
|
_ = file.Close()
|
|
}(file)
|
|
|
|
// 日期目录
|
|
now := time.Now()
|
|
folderPath := fmt.Sprintf("%d/%02d/%02d",
|
|
now.Year(),
|
|
now.Month(),
|
|
now.Day(),
|
|
)
|
|
|
|
// UUID 文件名（保留后缀）
|
|
ext := filepath.Ext(fileHeader.Filename)
|
|
newName := fmt.Sprintf("%s%s", uuid.NewString(), ext)
|
|
|
|
// S3 object key
|
|
key := fmt.Sprintf("%s/%s", folderPath, newName)
|
|
|
|
// ContentType（防止前端乱传）
|
|
contentType := fileHeader.Header.Get("Content-Type")
|
|
|
|
// 调用存储层
|
|
err = s3store.UploadToRustFS(
|
|
c.Context(),
|
|
key,
|
|
file,
|
|
fileHeader.Size,
|
|
contentType,
|
|
)
|
|
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
return httpx.OK(c, nil)
|
|
}
|