85 lines
1.7 KiB
Go
85 lines
1.7 KiB
Go
package upload
|
|
|
|
import (
|
|
"context"
|
|
"mime/multipart"
|
|
"path"
|
|
|
|
"github.com/minio/minio-go/v7"
|
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
|
)
|
|
|
|
// Minio MinIO存储
|
|
type Minio struct {
|
|
ID string
|
|
Secret string
|
|
Bucket string
|
|
Endpoint string
|
|
BasePath string
|
|
UseSSL bool
|
|
}
|
|
|
|
// NewMinio 创建MinIO
|
|
func NewMinio(id, secret, bucket, endpoint, basePath string, useSSL bool) *Minio {
|
|
return &Minio{
|
|
ID: id,
|
|
Secret: secret,
|
|
Bucket: bucket,
|
|
Endpoint: endpoint,
|
|
BasePath: basePath,
|
|
UseSSL: useSSL,
|
|
}
|
|
}
|
|
|
|
func (m *Minio) getClient() (*minio.Client, error) {
|
|
return minio.New(m.Endpoint, &minio.Options{
|
|
Creds: credentials.NewStaticV4(m.ID, m.Secret, ""),
|
|
Secure: m.UseSSL,
|
|
})
|
|
}
|
|
|
|
// UploadFile 上传文件
|
|
func (m *Minio) UploadFile(file *multipart.FileHeader) (string, string, error) {
|
|
client, err := m.getClient()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
// 生成文件名和路径
|
|
fileName := GenerateFileName(file.Filename)
|
|
filePath := GeneratePath(m.BasePath)
|
|
key := path.Join(filePath, fileName)
|
|
|
|
// 打开文件
|
|
src, err := file.Open()
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
defer src.Close()
|
|
|
|
// 上传
|
|
_, err = client.PutObject(context.Background(), m.Bucket, key, src, file.Size, minio.PutObjectOptions{
|
|
ContentType: file.Header.Get("Content-Type"),
|
|
})
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
// 返回访问URL和key
|
|
protocol := "http"
|
|
if m.UseSSL {
|
|
protocol = "https"
|
|
}
|
|
accessURL := protocol + "://" + m.Endpoint + "/" + m.Bucket + "/" + key
|
|
return accessURL, key, nil
|
|
}
|
|
|
|
// DeleteFile 删除文件
|
|
func (m *Minio) DeleteFile(key string) error {
|
|
client, err := m.getClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return client.RemoveObject(context.Background(), m.Bucket, key, minio.RemoveObjectOptions{})
|
|
}
|