83 lines
2.3 KiB
Go
83 lines
2.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"mime/multipart"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"model-asynch/model/entity"
|
|||
|
|
|
|||
|
|
commonHttp "gitea.com/red-future/common/http"
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|||
|
|
"github.com/gogf/gf/v2/util/guid"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 对接你们的 oss 文件服务：POST oss/file/uploadFile (multipart/form-data)
|
|||
|
|
type ossStorage struct{}
|
|||
|
|
|
|||
|
|
type uploadFileResponse struct {
|
|||
|
|
FileURL string `json:"fileURL"` // 文件 URL
|
|||
|
|
FileSize int `json:"fileSize"` // 文件大小（字节）
|
|||
|
|
FileName string `json:"fileName"` // 文件名
|
|||
|
|
FileFormat string `json:"fileFormat"` // 文件格式
|
|||
|
|
FileAddressPrefix string `json:"fileAddressPrefix"` // 文件地址前缀
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (s *ossStorage) UploadByTask(ctx context.Context, _ *entity.AsynchTask, data []byte, fileExt string, _ string) (ossURL string, err error) {
|
|||
|
|
// multipart
|
|||
|
|
body := &bytes.Buffer{}
|
|||
|
|
writer := multipart.NewWriter(body)
|
|||
|
|
|
|||
|
|
ext := fileExt
|
|||
|
|
if ext == "" {
|
|||
|
|
ext = ".bin"
|
|||
|
|
}
|
|||
|
|
if ext[0] != '.' {
|
|||
|
|
ext = "." + ext
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
filename := fmt.Sprintf("asynch_%d_%s%s", time.Now().Unix(), guid.S(), ext)
|
|||
|
|
part, err := writer.CreateFormFile("file", filename)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
if _, err := part.Write(data); err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
contentType := writer.FormDataContentType()
|
|||
|
|
if err := writer.Close(); err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
headers := forwardHeaders(ctx)
|
|||
|
|
headers["Content-Type"] = contentType
|
|||
|
|
|
|||
|
|
fullURL := "oss/file/uploadFile"
|
|||
|
|
g.Log().Infof(ctx, "[OSS] upload start url=%s filename=%s size=%d", fullURL, filename, len(data))
|
|||
|
|
|
|||
|
|
var resp uploadFileResponse
|
|||
|
|
if err := commonHttp.Post(ctx, fullURL, headers, &resp, body.Bytes()); err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
fmt.Println("打印结果 resp:", resp)
|
|||
|
|
g.Log().Infof(ctx, "[OSS] upload success url=%s size=%d format=%s", resp.FileURL, resp.FileSize, resp.FileFormat)
|
|||
|
|
return resp.FileURL, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// setTaskHeadersToCtx 把任务入库时保存的 header 信息注入 ctx，给 worker 调 OSS 用
|
|||
|
|
func setTaskHeadersToCtx(ctx context.Context, headers map[string]string) context.Context {
|
|||
|
|
if headers == nil {
|
|||
|
|
return ctx
|
|||
|
|
}
|
|||
|
|
if v := gconv.String(headers["Authorization"]); v != "" {
|
|||
|
|
ctx = context.WithValue(ctx, "token", v)
|
|||
|
|
}
|
|||
|
|
if v := gconv.String(headers["X-User-Info"]); v != "" {
|
|||
|
|
ctx = context.WithValue(ctx, "xUserInfo", v)
|
|||
|
|
}
|
|||
|
|
return ctx
|
|||
|
|
}
|