添加过滤埋点

This commit is contained in:
Jerry Yan 2025-04-13 14:11:53 +08:00
parent 94e1f66288
commit a478902f98
2 changed files with 8 additions and 3 deletions

View File

@ -36,7 +36,7 @@ func HandleTask(ctx context.Context, device config.DeviceMapping, task dto.Task)
log.Printf("获取文件列表失败, DeviceNo: %s, 错误: %v\n", device.DeviceNo, err) log.Printf("获取文件列表失败, DeviceNo: %s, 错误: %v\n", device.DeviceNo, err)
return nil, err return nil, err
} }
files := util.FilterAndSortFiles(fileList, task.StartTime, task.EndTime) files := util.FilterAndSortFiles(subCtx, fileList, task.StartTime, task.EndTime)
if len(files) == 0 { if len(files) == 0 {
span.SetStatus(codes.Error, "没有找到文件") span.SetStatus(codes.Error, "没有找到文件")
return nil, fmt.Errorf("没有找到文件") return nil, fmt.Errorf("没有找到文件")

View File

@ -2,11 +2,16 @@ package util
import ( import (
"ZhenTuLocalPassiveAdapter/dto" "ZhenTuLocalPassiveAdapter/dto"
"context"
"go.opentelemetry.io/otel/attribute"
"sort" "sort"
"time" "time"
) )
func FilterAndSortFiles(fileList []dto.File, beginDt, endDt time.Time) []dto.File { func FilterAndSortFiles(ctx context.Context, fileList []dto.File, beginDt, endDt time.Time) []dto.File {
_, span := tracer.Start(ctx, "FilterAndSortFiles")
defer span.End()
span.SetAttributes(attribute.Int("files.count", len(fileList)))
var filteredFiles []dto.File var filteredFiles []dto.File
for _, file := range fileList { for _, file := range fileList {
@ -37,6 +42,6 @@ func FilterAndSortFiles(fileList []dto.File, beginDt, endDt time.Time) []dto.Fil
sort.Slice(filteredFiles, func(i, j int) bool { sort.Slice(filteredFiles, func(i, j int) bool {
return filteredFiles[i].StartTime.Unix() <= filteredFiles[j].StartTime.Unix() return filteredFiles[i].StartTime.Unix() <= filteredFiles[j].StartTime.Unix()
}) })
span.SetAttributes(attribute.String("files.filtered", ToJson(filteredFiles)))
return filteredFiles return filteredFiles
} }