初始化项目
This commit is contained in:
@@ -0,0 +1,62 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/minio/minio-go/v7"
|
||||
"github.com/minio/minio-go/v7/pkg/credentials"
|
||||
)
|
||||
|
||||
const (
|
||||
minioEndpoint = "ai.ronsunny.cn:9000"
|
||||
minioAccessKey = "minioadmin"
|
||||
minioSecretKey = "minioadmin"
|
||||
useSSL = true
|
||||
)
|
||||
|
||||
var client *minio.Client
|
||||
|
||||
func Init() error {
|
||||
if client != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
c, err := minio.New(minioEndpoint, &minio.Options{
|
||||
Creds: credentials.NewStaticV4(minioAccessKey, minioSecretKey, ""),
|
||||
Secure: useSSL,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
client = c
|
||||
return nil
|
||||
}
|
||||
func UploadFile(
|
||||
ctx context.Context,
|
||||
bucketName string,
|
||||
objectName string,
|
||||
filePath string,
|
||||
contentType string,
|
||||
) (int64, error) {
|
||||
|
||||
if client == nil {
|
||||
return 0, errors.New("minio client not initialized")
|
||||
}
|
||||
|
||||
info, err := client.FPutObject(
|
||||
ctx,
|
||||
bucketName,
|
||||
objectName,
|
||||
filePath,
|
||||
minio.PutObjectOptions{
|
||||
ContentType: contentType,
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return info.Size, nil
|
||||
}
|
||||
Reference in New Issue
Block a user