120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"git.hpds.cc/Component/logging"
|
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
|
"hpds-iot-web/config"
|
|
"hpds-iot-web/internal/proto"
|
|
"hpds-iot-web/model"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
"xorm.io/xorm"
|
|
|
|
"github.com/minio/minio-go/v7"
|
|
)
|
|
|
|
type FileService interface {
|
|
UploadFile(ctx context.Context, req proto.UploadFileRequest) (rsp *proto.BaseResponse, err error)
|
|
UploadFileToMinIo(ctx context.Context, srcFile *multipart.FileHeader, scene string, creator int64) (data *model.FileManager, err error)
|
|
}
|
|
|
|
func NewFileService(cfg *config.WebConfig, engine *xorm.Engine, logger *logging.Logger) FileService {
|
|
return &repo{
|
|
AppConfig: cfg,
|
|
engine: engine,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (rp *repo) UploadFile(ctx context.Context, req proto.UploadFileRequest) (rsp *proto.BaseResponse, err error) {
|
|
rsp = new(proto.BaseResponse)
|
|
select {
|
|
case <-ctx.Done():
|
|
err = fmt.Errorf("超时/取消")
|
|
rsp.Code = http.StatusInternalServerError
|
|
rsp.Status = http.StatusText(http.StatusInternalServerError)
|
|
rsp.Message = "超时/取消"
|
|
rsp.Err = ctx.Err()
|
|
return rsp, ctx.Err()
|
|
default:
|
|
list := make([]*model.FileManager, len(req.Files))
|
|
fileUrl := make([]string, len(req.Files))
|
|
for k, _ := range req.Files {
|
|
fileItem, err := rp.UploadFileToMinIo(ctx, req.Files[k], req.Scene, req.Creator)
|
|
if err != nil {
|
|
goto ReturnPoint
|
|
}
|
|
list[k] = fileItem
|
|
fileUrl[k] = fileItem.AccessUrl
|
|
}
|
|
_, err = rp.engine.Insert(list)
|
|
res := proto.UploadResponse{
|
|
Url: strings.Join(fileUrl, ","),
|
|
}
|
|
rsp.Code = http.StatusOK
|
|
rsp.Status = http.StatusText(http.StatusOK)
|
|
rsp.Message = "成功"
|
|
rsp.Data = res
|
|
rsp.Err = err
|
|
return rsp, err
|
|
}
|
|
ReturnPoint:
|
|
if err != nil {
|
|
rsp.Code = http.StatusInternalServerError
|
|
rsp.Status = http.StatusText(http.StatusInternalServerError)
|
|
rsp.Err = err
|
|
rsp.Message = "失败"
|
|
}
|
|
return rsp, err
|
|
}
|
|
|
|
func (rp *repo) UploadFileToMinIo(ctx context.Context, srcFile *multipart.FileHeader, scene string, creator int64) (data *model.FileManager, err error) {
|
|
file, err := srcFile.Open()
|
|
defer file.Close()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fileName := srcFile.Filename
|
|
|
|
opt := &minio.Options{
|
|
Creds: credentials.NewStaticV4(rp.AppConfig.Minio.AccessKeyId, rp.AppConfig.Minio.SecretAccessKey, ""),
|
|
Secure: false,
|
|
}
|
|
minioClient, err := minio.New(rp.AppConfig.Minio.Endpoint, opt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
//bucketName := fmt.Sprintf("jky-data/%s/%s", scene, time.Now().Format("20060102"))
|
|
objPath := fmt.Sprintf("%s/%s/%s", scene, time.Now().Format("20060102"), fileName)
|
|
info, err := minioClient.PutObject(ctx, "jky-data", objPath, file, srcFile.Size, minio.PutObjectOptions{ContentType: "application/octet-stream"})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fmt.Println("info =====> ", info)
|
|
accessUrl := fmt.Sprintf("%s://%s/jky-data/%s", rp.AppConfig.Minio.Protocol, rp.AppConfig.Minio.Endpoint, objPath)
|
|
|
|
md5hash := md5.New()
|
|
if _, err := io.Copy(md5hash, file); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data = new(model.FileManager)
|
|
data.FileName = fileName
|
|
data.Scene = scene
|
|
data.AccessUrl = accessUrl
|
|
data.FileSize = srcFile.Size
|
|
data.Creator = creator
|
|
data.FileMd5 = hex.EncodeToString(md5hash.Sum(nil))
|
|
data.CreateAt = time.Now().Unix()
|
|
data.UpdateAt = time.Now().Unix()
|
|
return data, nil
|
|
}
|