86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package gateway
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"strings"
|
|
|
|
commonHttp "gitea.redpowerfuture.com/red-future/common/http"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// Upload 以 multipart 方式上传文件字节到 OSS，返回可访问 URL。
|
|
// 原签名基于 workflow/model/dto 的 UploadFileBytesReq/Res，抽离时简化为直接传文件名与字节，便于业务侧解耦 workflow。
|
|
func Upload(ctx context.Context, fileName string, fileBytes []byte) (string, error) {
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
|
|
part, err := writer.CreateFormFile("file", fileName)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if _, err = part.Write(fileBytes); err != nil {
|
|
return "", err
|
|
}
|
|
if err = writer.Close(); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
headers := make(map[string]string)
|
|
if r := g.RequestFromCtx(ctx); r != nil {
|
|
for k, v := range r.Request.Header {
|
|
if len(v) > 0 {
|
|
headers[k] = v[0]
|
|
}
|
|
}
|
|
}
|
|
// 后台恢复续跑无 HTTP 请求时（ctx 携带合成 user），补充 X-User-Info 供 OSS GetUserInfo 识别用户与桶名；
|
|
// 与 gateway/model.go requestHeaders 的恢复兜底保持一致，否则 OSS GetBucketName 落到空 token 解析报错
|
|
if headers["X-User-Info"] == "" {
|
|
if u := ctx.Value("user"); u != nil {
|
|
headers["X-User-Info"] = gconv.String(u)
|
|
}
|
|
}
|
|
headers["Content-Type"] = writer.FormDataContentType()
|
|
|
|
res := &uploadFileRes{}
|
|
if err = commonHttp.Post(ctx, "oss/file/uploadFile", headers, res, body.Bytes()); err != nil {
|
|
return "", err
|
|
}
|
|
return res.FileURL, nil
|
|
}
|
|
|
|
// uploadFileRes OSS 上传响应的最小字段（原 workflow/model/dto.UploadFileBytesRes 的 URL 部分）。
|
|
type uploadFileRes struct {
|
|
FileURL string `json:"fileURL"`
|
|
}
|
|
|
|
// GetFileBytesFromURL 下载远程文件内容（把 filePrefix 前缀替换为 minioPrefix 后经 GoFrame 客户端下载）
|
|
func GetFileBytesFromURL(ctx context.Context, fileUrl string) ([]byte, error) {
|
|
newS := strings.ReplaceAll(fileUrl, g.Cfg().MustGet(ctx, "filePrefix").String(), g.Cfg().MustGet(ctx, "minioPrefix").String())
|
|
// 使用 GoFrame 客户端（自带超时、追踪、日志等能力）
|
|
resp, err := g.Client().Get(ctx, newS)
|
|
if err != nil {
|
|
return nil, gerror.Wrapf(err, "failed to request url: %s", newS)
|
|
}
|
|
defer resp.Close()
|
|
|
|
// 校验状态码
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, gerror.Newf("request failed with status code: %d, url: %s", resp.StatusCode, newS)
|
|
}
|
|
|
|
// 读取全部内容
|
|
allBytes, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, gerror.Wrapf(err, "failed to read response body, url: %s", fileUrl)
|
|
}
|
|
|
|
return allBytes, nil
|
|
}
|