q792602257/VptPassiveAdapter
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

telemetry

Browse Source
This commit is contained in:
Jerry Yan
2025-04-13 11:24:20 +08:00
parent a6c5ba5957
commit 3d989c2f47
14 changed files with 443 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
api/oss_upload.go
View File

@ -3,38 +3,55 @@ package api
import (
"ZhenTuLocalPassiveAdapter/dto"
"bytes"
"context"
"fmt"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"io"
"log"
"net/http"
"os"
)
func UploadTaskFile(task dto.Task, file dto.FileObject) error {
url, err := QueryUploadUrlForTask(task.TaskID)
func UploadTaskFile(ctx context.Context, task dto.Task, file dto.FileObject) error {
url, err := QueryUploadUrlForTask(ctx, task.TaskID)
if err != nil {
return err
}
if err := OssUpload(url, file.URL); err != nil {
log.Printf("开始上传文件, URL：【%s】\n", url)
if err := OssUpload(ctx, url, file.URL); err != nil {
return err
}
return nil
}
func OssUpload(url, filePath string) error {
func OssUpload(ctx context.Context, url, filePath string) error {
_, span := tracer.Start(ctx, "OssUpload")
defer span.End()
span.SetAttributes(attribute.String("file.path", filePath))
// 使用 http put 请求上传文件
file, err := os.Open(filePath)
defer os.Remove(filePath)
defer file.Close()
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "文件打开失败")
return err
}
fileBytes, err := io.ReadAll(file)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "文件读取失败")
return err
}
span.SetAttributes(attribute.Int("file.size", len(fileBytes)))
span.SetAttributes(attribute.String("http.url", url))
span.SetAttributes(attribute.String("http.method", "PUT"))
req, err := http.NewRequest("PUT", url, bytes.NewBuffer(fileBytes))
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
return err
}
req.Header.Set("Content-Type", "video/mp4")
@ -42,11 +59,17 @@ func OssUpload(url, filePath string) error {
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
return err
}
defer resp.Body.Close()
span.SetAttributes(attribute.String("http.status", resp.Status))
span.SetAttributes(attribute.Int("http.status_code", resp.StatusCode))
if resp.StatusCode != http.StatusOK {
span.SetStatus(codes.Error, "上传失败")
return fmt.Errorf("upload failed with status code %d", resp.StatusCode)
}
span.SetStatus(codes.Ok, "上传成功")
return nil
}