harukaandClaude Opus 4.8
0eb6e21aaa
为异步生图任务增加 S3 兼容对象存储支持，任务结果不再把大图内联存进 Redis： - 新增可插拔接口 service.ImageStorage（Save -> url），适配别的厂商只需实现它 - S3 实现 S3ImageStorage（AWS S3 / R2 / 阿里云 OSS / MinIO），与备份共用 S3 客户端构造 - 新增 image_storage 配置（config.yaml + IMAGE_STORAGE_* 环境变量），默认关闭 - enabled 同时作为总开关：关闭或未配置对象存储时，异步生图接口返回 404 且不写 Redis，从根上避免几 MB 的 b64_json 结果撑爆 Redis - 完成时把图片上传对象存储并把结果改写为短链接（公开直链或 presigned）， 上传失败则任务标记为失败 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01SM1tf3CFVRzC7guuhBXvMd
54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/aws"
|
|
v4 "github.com/aws/aws-sdk-go-v2/aws/signer/v4"
|
|
awsconfig "github.com/aws/aws-sdk-go-v2/config"
|
|
"github.com/aws/aws-sdk-go-v2/credentials"
|
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
|
)
|
|
|
|
// s3ClientParams 描述构造 S3 兼容客户端所需的参数。
|
|
type s3ClientParams struct {
|
|
Endpoint string
|
|
Region string
|
|
AccessKeyID string
|
|
SecretAccessKey string
|
|
ForcePathStyle bool
|
|
}
|
|
|
|
// newS3Client 构造一个 S3 兼容客户端，兼容 AWS S3 / Cloudflare R2 / 阿里云 OSS / MinIO。
|
|
//
|
|
// 通过 SwapComputePayloadSHA256ForUnsignedPayloadMiddleware + RequestChecksumCalculationWhenRequired
|
|
// 规避阿里云 OSS 不兼容 s3manager 分片签名的问题（backup 与 image storage 共用此构造）。
|
|
func newS3Client(ctx context.Context, p s3ClientParams) (*s3.Client, error) {
|
|
region := p.Region
|
|
if region == "" {
|
|
region = "auto" // Cloudflare R2 默认 region
|
|
}
|
|
|
|
awsCfg, err := awsconfig.LoadDefaultConfig(ctx,
|
|
awsconfig.WithRegion(region),
|
|
awsconfig.WithCredentialsProvider(
|
|
credentials.NewStaticCredentialsProvider(p.AccessKeyID, p.SecretAccessKey, ""),
|
|
),
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("load aws config: %w", err)
|
|
}
|
|
|
|
return s3.NewFromConfig(awsCfg, func(o *s3.Options) {
|
|
if p.Endpoint != "" {
|
|
o.BaseEndpoint = &p.Endpoint
|
|
}
|
|
if p.ForcePathStyle {
|
|
o.UsePathStyle = true
|
|
}
|
|
o.APIOptions = append(o.APIOptions, v4.SwapComputePayloadSHA256ForUnsignedPayloadMiddleware)
|
|
o.RequestChecksumCalculation = aws.RequestChecksumCalculationWhenRequired
|
|
}), nil
|
|
}
|