This commit is contained in:
2025-08-04 10:49:24 +08:00
parent 84ccaa56de
commit 4b1eb11986
11 changed files with 194 additions and 72 deletions

View File

@@ -3,13 +3,14 @@ package api
import (
"ZhenTuLocalPassiveAdapter/config"
"ZhenTuLocalPassiveAdapter/dto"
"ZhenTuLocalPassiveAdapter/logger"
"bytes"
"context"
"encoding/json"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
"go.uber.org/zap"
"io"
"log"
"net/http"
)
@@ -23,14 +24,14 @@ func QueryUploadUrlForTask(ctx context.Context, taskId string) (string, error) {
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
logger.Error("创建请求失败", zap.Error(err))
return "", err
}
resp, err := GetAPIClient().Do(req)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
logger.Error("发送请求失败", zap.Error(err))
return "", err
}
span.SetAttributes(attribute.String("http.status", resp.Status))
@@ -40,7 +41,7 @@ func QueryUploadUrlForTask(ctx context.Context, taskId string) (string, error) {
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "读取响应体失败")
log.Println("Error reading response body:", err)
logger.Error("读取响应体失败", zap.Error(err))
return "", err
}
return string(body), nil
@@ -57,7 +58,7 @@ func ReportTaskFailure(ctx context.Context, taskId string) bool {
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
logger.Error("创建请求失败", zap.Error(err))
return false
}
@@ -65,7 +66,7 @@ func ReportTaskFailure(ctx context.Context, taskId string) bool {
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
logger.Error("发送请求失败", zap.Error(err))
return false
}
defer resp.Body.Close()
@@ -92,7 +93,7 @@ func ReportTaskSuccess(ctx context.Context, taskId string, file *dto.FileObject)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "序列化JSON失败")
log.Println("Error marshaling JSON:", err)
logger.Error("序列化JSON失败", zap.Error(err))
return false
}
@@ -100,7 +101,7 @@ func ReportTaskSuccess(ctx context.Context, taskId string, file *dto.FileObject)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "创建请求失败")
log.Println("Error creating request:", err)
logger.Error("创建请求失败", zap.Error(err))
return false
}
req.Header.Set("Content-Type", "application/json")
@@ -109,7 +110,7 @@ func ReportTaskSuccess(ctx context.Context, taskId string, file *dto.FileObject)
if err != nil {
span.SetAttributes(attribute.String("error", err.Error()))
span.SetStatus(codes.Error, "发送请求失败")
log.Println("Error sending request:", err)
logger.Error("发送请求失败", zap.Error(err))
return false
}
defer resp.Body.Close()