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

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
}
}