Blizzard
28b95a6468
后端 - 每日任务惰性生成：某天为空则从最近一天顺延复制（done 重置），首天用模板 - 任务增删改 API（POST /pets/:id/tasks、PUT/DELETE /tasks/:id），改动自动顺延 - 30 天计划到期自动归档并按当前阶段生成新一轮 - 内置养护模板重写为兽医常识向：狗每阶段含遛狗/牵引/狂犬，猫含猫砂/梳毛/饮水 - 文件 URL 改为按 object_name + 当前配置动态拼接，换 IP/域名不再有旧地址 小程序 - 首页今日任务加「管理」入口：增删改任务弹层 - 记录时间轴显示照片缩略图（可全屏预览）+ 拍照成功提示 - 社区发布条精简为单个「发布图文」按钮，去掉头像/输入框误触 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
91 lines
2.4 KiB
Go
91 lines
2.4 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 s.PublicURL(objectName), nil
|
|
}
|
|
|
|
// PublicURL 由对象名拼出可访问 URL（用当前配置的 public_base_url，避免存旧地址）
|
|
func (s *Storage) PublicURL(objectName string) string {
|
|
return fmt.Sprintf("%s/%s/%s", s.publicBaseURL, s.bucket, objectName)
|
|
}
|