third

pkg/common/db/obj/minio.go

@@ -9,6 +9,7 @@ import (
 	"github.com/minio/minio-go/v7"
 	"github.com/minio/minio-go/v7/pkg/credentials"
 	"github.com/minio/minio-go/v7/pkg/s3utils"
+	"io"
 	"net/http"
 	"net/url"
 	"time"
@@ -196,6 +197,29 @@ func (m *minioImpl) IsNotFound(err error) bool {
 	}
 }
 
+func (m *minioImpl) PutObject(ctx context.Context, info *BucketObject, reader io.Reader, size int64) (*ObjectInfo, error) {
+	update, err := m.client.PutObject(ctx, info.Bucket, info.Name, reader, size, minio.PutObjectOptions{})
+	if err != nil {
+		return nil, err
+	}
+	return &ObjectInfo{
+		Size: update.Size,
+		Hash: update.ETag,
+	}, nil
+}
+
+func (m *minioImpl) GetObject(ctx context.Context, info *BucketObject) (SizeReader, error) {
+	object, err := m.client.GetObject(ctx, info.Bucket, info.Name, minio.GetObjectOptions{})
+	if err != nil {
+		return nil, err
+	}
+	stat, err := object.Stat()
+	if err != nil {
+		return nil, err
+	}
+	return NewSizeReader(object, stat.Size), nil
+}
+
 func (m *minioImpl) CheckName(name string) error {
 	return s3utils.CheckValidObjectName(name)
 }

pkg/common/db/obj/obj.go

@@ -2,6 +2,7 @@ package obj
 
 import (
 	"context"
+	"io"
 	"net/http"
 	"time"
 )
@@ -29,6 +30,30 @@ type ObjectInfo struct {
 	Hash string
 }
 
+type SizeReader interface {
+	io.ReadCloser
+	Size() int64
+}
+
+func NewSizeReader(r io.ReadCloser, size int64) SizeReader {
+	if r == nil {
+		return nil
+	}
+	return &sizeReader{
+		size:       size,
+		ReadCloser: r,
+	}
+}
+
+type sizeReader struct {
+	size int64
+	io.ReadCloser
+}
+
+func (r *sizeReader) Size() int64 {
+	return r.size
+}
+
 type Interface interface {
 	// Name 存储名字
 	Name() string
@@ -58,4 +83,8 @@ type Interface interface {
 	IsNotFound(err error) bool
 	// CheckName 检查名字是否可用
 	CheckName(name string) error
+	// PutObject 上传文件
+	PutObject(ctx context.Context, info *BucketObject, reader io.Reader, size int64) (*ObjectInfo, error)
+	// GetObject 下载文件
+	GetObject(ctx context.Context, info *BucketObject) (SizeReader, error)
 }