93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package logic
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"net/url"
|
||
"path/filepath"
|
||
"time"
|
||
|
||
"godemo/file/file"
|
||
"godemo/file/internal/svc"
|
||
|
||
"github.com/google/uuid"
|
||
"github.com/minio/minio-go/v7"
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
"google.golang.org/grpc/codes"
|
||
"google.golang.org/grpc/status"
|
||
)
|
||
|
||
type UploadLogic struct {
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
logx.Logger
|
||
}
|
||
|
||
func NewUploadLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadLogic {
|
||
return &UploadLogic{
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
Logger: logx.WithContext(ctx),
|
||
}
|
||
}
|
||
|
||
// 上传文件（图片/头像/壁纸等）
|
||
func (l *UploadLogic) Upload(in *file.UploadRequest) (*file.UploadResponse, error) {
|
||
bucketName := "default"
|
||
if in.Bucket != "" {
|
||
bucketName = in.Bucket
|
||
}
|
||
|
||
// 确保 bucket 存在（自动创建）
|
||
exists, err := l.svcCtx.MinioClient.BucketExists(l.ctx, bucketName)
|
||
if err != nil {
|
||
return nil, status.Error(codes.Internal, err.Error())
|
||
}
|
||
if !exists {
|
||
err := l.svcCtx.MinioClient.MakeBucket(l.ctx, bucketName, minio.MakeBucketOptions{})
|
||
if err != nil {
|
||
return nil, status.Error(codes.Internal, err.Error())
|
||
}
|
||
}
|
||
|
||
// 生成唯一文件名
|
||
ext := getFileExtension(in.Filename)
|
||
newFileName := uuid.New().String() + ext
|
||
|
||
if in.Folder != "" {
|
||
newFileName = in.Folder + "/" + newFileName
|
||
}
|
||
|
||
// 上传对象
|
||
_, err = l.svcCtx.MinioClient.PutObject(l.ctx, bucketName, newFileName,
|
||
bytes.NewReader(in.Content),
|
||
int64(len(in.Content)),
|
||
minio.PutObjectOptions{
|
||
ContentType: in.ContentType,
|
||
})
|
||
if err != nil {
|
||
return nil, status.Error(codes.Internal, err.Error())
|
||
}
|
||
|
||
// 生成预签名 URL（有效期1小时）
|
||
reqParams := make(url.Values)
|
||
presignedUrl, err := l.svcCtx.MinioClient.PresignedGetObject(l.ctx, bucketName, newFileName, time.Hour, reqParams)
|
||
if err != nil {
|
||
return nil, status.Error(codes.Internal, err.Error())
|
||
}
|
||
|
||
return &file.UploadResponse{
|
||
FileId: newFileName,
|
||
Url: presignedUrl.String(),
|
||
}, nil
|
||
}
|
||
|
||
// 获取文件扩展名（默认 .jpg）
|
||
func getFileExtension(name string) string {
|
||
ext := filepath.Ext(name)
|
||
if ext == "" {
|
||
return ".jpg"
|
||
}
|
||
return ext
|
||
}
|