107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/minio/minio-go"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type CmiiMinioOperator struct {
|
|
LocalEndpoint string
|
|
PublicEndpoint string
|
|
|
|
AccessKeyID string
|
|
SecretAccessKey string
|
|
|
|
LocalClient *minio.Client
|
|
PublicClient *minio.Client
|
|
DemoClient *minio.Client
|
|
|
|
OctopusBucketName string
|
|
}
|
|
|
|
const (
|
|
DefaultLocalEndpoint = "10.250.0.100:9000"
|
|
DefaultPublicEndpoint = "42.192.52.227:9000"
|
|
DemoEndpoint = "oss.demo.uavcmlc.com:18000"
|
|
DefaultAccessKeyID = "cmii"
|
|
DefaultSecretAccessKey = "B#923fC7mk"
|
|
DefaultOctopusBucketName = "octopus"
|
|
)
|
|
|
|
func (op CmiiMinioOperator) buildCmiiMinioOperator() {
|
|
|
|
var err error
|
|
if op.LocalClient == nil {
|
|
// 初始化Minio客户端对象。
|
|
op.LocalClient, err = minio.New(DefaultLocalEndpoint, DefaultAccessKeyID, DefaultSecretAccessKey, false)
|
|
if err != nil {
|
|
log.ErrorF("[buildCmiiMinioOperator] - build for LocalClient error !=> %s", err.Error())
|
|
}
|
|
}
|
|
|
|
if op.PublicClient == nil {
|
|
// 初始化Minio客户端对象。
|
|
op.PublicClient, err = minio.New(DefaultPublicEndpoint, DefaultAccessKeyID, DefaultSecretAccessKey, false)
|
|
if err != nil {
|
|
log.ErrorF("[buildCmiiMinioOperator] - build for PublicClient error ! => %s", err.Error())
|
|
}
|
|
}
|
|
|
|
if op.DemoClient == nil {
|
|
// 初始化Minio客户端对象。
|
|
op.DemoClient, err = minio.New(DemoEndpoint, DefaultAccessKeyID, DefaultSecretAccessKey, false)
|
|
if err != nil {
|
|
log.ErrorF("[buildCmiiMinioOperator] - build for DemoClient error ! => %s", err.Error())
|
|
}
|
|
}
|
|
|
|
log.DebugF("[buildCmiiMinioOperator] - build client success !")
|
|
}
|
|
|
|
func (op CmiiMinioOperator) UploadToLocalOctopus(filePath, fileName string) bool {
|
|
op.buildCmiiMinioOperator()
|
|
return op.uploadToOctopus(op.LocalClient, filePath, fileName)
|
|
}
|
|
|
|
func (op CmiiMinioOperator) UploadToPublicOctopus(filePath, fileName string) bool {
|
|
|
|
op.buildCmiiMinioOperator()
|
|
return op.uploadToOctopus(op.PublicClient, filePath, fileName)
|
|
}
|
|
|
|
func (op CmiiMinioOperator) UploadToDemo(bucketName, filePath, fileName string) bool {
|
|
|
|
op.buildCmiiMinioOperator()
|
|
return op.uploadToOss(op.DemoClient, bucketName, filePath, fileName)
|
|
}
|
|
|
|
func (op CmiiMinioOperator) uploadToOctopus(client *minio.Client, filePath, fileName string) bool {
|
|
|
|
return op.uploadToOss(client, DefaultOctopusBucketName, filePath, fileName)
|
|
}
|
|
|
|
func (op CmiiMinioOperator) uploadToOss(client *minio.Client, bucketName, filePath, fileName string) bool {
|
|
|
|
if !strings.HasSuffix(filePath, "/") {
|
|
filePath += "/"
|
|
}
|
|
|
|
// 打开要上传的文件
|
|
file, err := os.Open(filePath + fileName)
|
|
if err != nil {
|
|
log.ErrorF("[uploadToOss] - file %s open error! %s", filePath+fileName, err.Error())
|
|
}
|
|
defer file.Close()
|
|
|
|
// 使用PutObject上传文件
|
|
n, err := client.PutObject(bucketName, fileName, file, -1, minio.PutObjectOptions{})
|
|
if err != nil {
|
|
log.ErrorF("[uploadToOss] - upload %s error %s", filePath+fileName, err.Error())
|
|
return false
|
|
}
|
|
|
|
log.InfoF("[uploadToOss] - uploaded %s of size %d", filePath+fileName, n)
|
|
return true
|
|
}
|