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
}

74
api/task_report.go
View File

@ -4,69 +4,127 @@ import (
"ZhenTuLocalPassiveAdapter/config"
"ZhenTuLocalPassiveAdapter/dto"
"bytes"
"context"
"encoding/json"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"io"
"log"
"net/http"
)
func QueryUploadUrlForTask(taskId string) (string, error) {
func QueryUploadUrlForTask(ctx context.Context, taskId string) (string, error) {
_, span := tracer.Start(ctx, "QueryUploadUrlForTask")
defer span.End()
url := config.Config.Api.BaseUrl + "/" + taskId + "/uploadUrl"
span.SetAttributes(attribute.String("http.url", url))
span.SetAttributes(attribute.String("http.method", "POST"))
req, err := http.NewRequest("POST", url, nil)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
return "", err
}
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
return "", err
}
span.SetAttributes(attribute.String("http.status", resp.Status))
span.SetAttributes(attribute.Int("http.status_code", resp.StatusCode))
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "读取响应体失败")
log.Println("Error reading response body:", err)
return "", err
}
return string(body), nil
}
func ReportTaskFailure(taskId string) {
func ReportTaskFailure(ctx context.Context, taskId string) bool {
_, span := tracer.Start(ctx, "ReportTaskFailure")
defer span.End()
url := config.Config.Api.BaseUrl + "/" + taskId + "/failure"
span.SetAttributes(attribute.String("http.url", url))
span.SetAttributes(attribute.String("http.method", "POST"))
req, err := http.NewRequest("POST", url, nil)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
return
return false
}
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
return
return false
}
defer resp.Body.Close()
span.SetAttributes(attribute.String("http.status", resp.Status))
span.SetAttributes(attribute.Int("http.status_code", resp.StatusCode))
if resp.StatusCode == 200 {
span.SetStatus(codes.Ok, "成功")
return true
} else {
span.SetStatus(codes.Error, "失败")
return false
}
}
func ReportTaskSuccess(taskId string, file *dto.FileObject) {
func ReportTaskSuccess(ctx context.Context, taskId string, file *dto.FileObject) (b bool) {
_, span := tracer.Start(ctx, "ReportTaskSuccess")
defer span.End()
url := config.Config.Api.BaseUrl + "/" + taskId + "/success"
span.SetAttributes(attribute.String("http.url", url))
span.SetAttributes(attribute.String("http.method", "POST"))
jsonData, err := json.Marshal(file)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "序列化JSON失败")
log.Println("Error marshaling JSON:", err)
return
return false
}
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
return
return false
}
req.Header.Set("Content-Type", "application/json")
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
return
return false
}
defer resp.Body.Close()
span.SetAttributes(attribute.String("http.status", resp.Status))
span.SetAttributes(attribute.Int("http.status_code", resp.StatusCode))
if resp.StatusCode == 200 {
span.SetStatus(codes.Ok, "成功")
return true
} else {
span.SetStatus(codes.Error, "失败")
return false
}
}

5
api/tracer.go Normal file
View File

@ -0,0 +1,5 @@
package api
import "go.opentelemetry.io/otel"
var tracer = otel.Tracer("api")