Blizzard
609f7d06cf
- pets-fe: 微信原生小程序（首页/计划/记录/报告/社区/引导）， 服务端驱动、无假数据；弹层改用 scroll-view，打开时隐藏自定义 tabBar - pets-be: Gin + GORM(MySQL, sundynix_ 前缀) + MinIO，统一响应/分页， 微信 code2session 登录，provider-neutral AI(DeepSeek)，go:embed React 后台 - 修复：分段选择类型不匹配（字符串 vs 数字）导致选不中 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/minio/minio-go/v7"
|
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
|
|
|
"github.com/sundynix/pets-be/internal/config"
|
|
)
|
|
|
|
// Storage MinIO 对象存储封装
|
|
type Storage struct {
|
|
client *minio.Client
|
|
bucket string
|
|
publicBaseURL string
|
|
}
|
|
|
|
// New 初始化 MinIO 客户端，并确保 bucket 存在且可公开下载
|
|
func New(cfg config.MinIOConfig) (*Storage, error) {
|
|
client, err := minio.New(cfg.Endpoint, &minio.Options{
|
|
Creds: credentials.NewStaticV4(cfg.AccessKey, cfg.SecretKey, ""),
|
|
Secure: cfg.UseSSL,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("minio new: %w", err)
|
|
}
|
|
|
|
s := &Storage{
|
|
client: client,
|
|
bucket: cfg.Bucket,
|
|
publicBaseURL: strings.TrimRight(cfg.PublicBaseURL, "/"),
|
|
}
|
|
if err := s.ensureBucket(); err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
}
|
|
|
|
func (s *Storage) ensureBucket() error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
exists, err := s.client.BucketExists(ctx, s.bucket)
|
|
if err != nil {
|
|
return fmt.Errorf("bucket exists: %w", err)
|
|
}
|
|
if !exists {
|
|
if err := s.client.MakeBucket(ctx, s.bucket, minio.MakeBucketOptions{}); err != nil {
|
|
return fmt.Errorf("make bucket: %w", err)
|
|
}
|
|
}
|
|
// dev：设置公开只读策略，便于直接以 URL 访问对象
|
|
policy := fmt.Sprintf(`{
|
|
"Version": "2012-10-17",
|
|
"Statement": [{
|
|
"Effect": "Allow",
|
|
"Principal": {"AWS": ["*"]},
|
|
"Action": ["s3:GetObject"],
|
|
"Resource": ["arn:aws:s3:::%s/*"]
|
|
}]
|
|
}`, s.bucket)
|
|
if err := s.client.SetBucketPolicy(ctx, s.bucket, policy); err != nil {
|
|
return fmt.Errorf("set bucket policy: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Upload 上传对象，返回可公开访问的 URL
|
|
func (s *Storage) Upload(objectName string, reader io.Reader, size int64, contentType string) (string, error) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
_, err := s.client.PutObject(ctx, s.bucket, objectName, reader, size, minio.PutObjectOptions{
|
|
ContentType: contentType,
|
|
})
|
|
if err != nil {
|
|
return "", fmt.Errorf("put object: %w", err)
|
|
}
|
|
return fmt.Sprintf("%s/%s/%s", s.publicBaseURL, s.bucket, objectName), nil
|
|
}
|