Go-zero 文件上传

最新推荐文章于 2024-04-17 06:19:20 发布

tony_qucheng

最新推荐文章于 2024-04-17 06:19:20 发布

阅读量423

点赞数 4

文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/weixin_45410801/article/details/137586300

版权

直接上代码了哈

API定义

@server(
	prefix: /api/file/v1
	group: file
)
service file {
	@doc "查看app列表"
	@handler uploadFile
	post /uploadFile (UploadFileReq) returns (UploadFileResp)
}```

### handler

```go
// UploadFileHandler 上传文件
func UploadFileHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// if err := r.ParseMultipartForm(32 << 20); err != nil {
		// 	httpx.ErrorCtx(r.Context(), w, err)
		// 	return
		// }
		// 解析多部分表单，并处理错误
		fileData, handler, err := r.FormFile("file")
		if err != nil {
			httpx.Error(w, err)
			return
		}
		// 关闭文件数据
		defer fileData.Close()
		// 获取文件名
		filename := handler.Filename
		// 实例化文件上传逻辑
		l := file.NewUploadFileLogic(r.Context(), svcCtx)
		// 上传文件，并获取响应和错误
		resp, err := l.UploadFile(&fileData, filename)
		// 响应错误
		syserr.ResponseErr(resp, err, r.Context(), w)
	}
}

service

package file

import (
	"context"
	"mime/multipart"

	"AggregatePayment/common/file"
	"AggregatePayment/common/utils"
	"AggregatePayment/file/cmd/api/internal/svc"
	"AggregatePayment/file/cmd/api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type UploadFileLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewUploadFileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadFileLogic {
	return &UploadFileLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UploadFileLogic) UploadFile(data *multipart.File, fileName string) (resp *types.UploadFileResp, err error) {
	resp = new(types.UploadFileResp)
	filePath := "businessData/" + utils.CreateUUID() + "." + fileName
	ossConfig := l.svcCtx.Config.OssConfig
	l.Logger.Info(ossConfig)
	err = file.UploadFile(*data, filePath, file.OssConfig{
		AccessKeyId:     ossConfig.AccessKeyId,
		AccessKeySecret: ossConfig.AccessKeySecret,
		BucketName:      ossConfig.BucketName,
		Endpoint:        ossConfig.Endpoint,
		BucketDoMan:     ossConfig.BucketDoMan,
	})
	if err != nil {
		l.Logger.Errorf("上传文件失败, err:%s", err.Error())
		return
	}
	path := l.svcCtx.Config.OssConfig.BucketDoMan + "/" + filePath
	resp.Data = path
	return
}```