GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-07-09 06:21:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(minio): pass the size to minio puitobject

Browse Source
This commit is contained in:
krau
2025-06-17 22:10:20 +08:00
parent 473a5b9413
commit eb6f8675a4
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
storage/minio/client.go
View File

@@ -9,6 +9,7 @@ import (
"github.com/charmbracelet/log" "github.com/charmbracelet/log"
config "github.com/krau/SaveAny-Bot/config/storage" config "github.com/krau/SaveAny-Bot/config/storage"
"github.com/krau/SaveAny-Bot/pkg/enums/key"
storenum "github.com/krau/SaveAny-Bot/pkg/enums/storage" storenum "github.com/krau/SaveAny-Bot/pkg/enums/storage"
"github.com/minio/minio-go/v7" "github.com/minio/minio-go/v7"
"github.com/minio/minio-go/v7/pkg/credentials" "github.com/minio/minio-go/v7/pkg/credentials"
@@ -72,8 +73,14 @@ func (m *Minio) Save(ctx context.Context, r io.Reader, storagePath string) error
for i := 1; m.Exists(ctx, candidate); i++ { for i := 1; m.Exists(ctx, candidate); i++ {
candidate = fmt.Sprintf("%s_%d%s", base, i, ext) candidate = fmt.Sprintf("%s_%d%s", base, i, ext)
} }
size := int64(-1)
_, err := m.client.PutObject(ctx, m.config.BucketName, candidate, r, -1, minio.PutObjectOptions{}) if length := ctx.Value(key.ContextKeyContentLength); length != nil {
length, ok := length.(int64)
if ok && length > 0 {
size = length
}
}
_, err := m.client.PutObject(ctx, m.config.BucketName, candidate, r, size, minio.PutObjectOptions{})
if err != nil { if err != nil {
return fmt.Errorf("failed to upload file to minio: %w", err) return fmt.Errorf("failed to upload file to minio: %w", err)
} }