feat(backend): refactor advance info

Author: kidkidkid

backend/modules/observability/domain/trace/entity/loop_span/span.go

@@ -550,8 +550,11 @@ func (s *Span) ClipSpan() {
 }
 
 func (s SpanList) Stat(ctx context.Context) (inputTokens, outputTokens int64, err error) {
-	modelSpans := s.FilterModelSpans()
-	for _, v := range modelSpans {
+	filter := GetModelSpansFilter()
+	for _, v := range s {
+		if !filter.Satisfied(v) {
+			continue
+		}
 		in, out, err := v.getTokens(ctx)
 		if err != nil {
 			return -1, -1, err
@@ -572,10 +575,7 @@ func (s SpanList) FilterSpans(f *FilterFields) SpanList {
 	return ret
 }
 
-func (s SpanList) FilterModelSpans() SpanList {
-	if len(s) == 0 {
-		return s
-	}
+func GetModelSpansFilter() *FilterFields {
 	modelFilter := &FilterFields{
 		QueryAndOr: ptr.Of(QueryAndOrEnumOr),
 		FilterFields: []*FilterField{
@@ -594,7 +594,7 @@ func (s SpanList) FilterModelSpans() SpanList {
 			},
 		},
 	}
-	return s.FilterSpans(modelFilter)
+	return modelFilter
 }
 
 func (s SpanList) SortByStartTime(desc bool) {

backend/modules/observability/domain/trace/entity/loop_span/span_test.go

@@ -744,8 +744,14 @@ func TestSpanList_FilterModelSpans(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			got := tt.spans.FilterModelSpans()
-			assert.Equal(t, tt.want, len(got))
+			f := GetModelSpansFilter()
+			got := 0
+			for _, span := range tt.spans {
+				if f.Satisfied(span) {
+					got++
+				}
+			}
+			assert.Equal(t, tt.want, got)
 		})
 	}
 }

backend/modules/observability/domain/trace/service/trace_service.go

@@ -596,6 +596,7 @@ func (r *TraceServiceImpl) GetTracesAdvanceInfo(ctx context.Context, req *GetTra
 					loop_span.SpanFieldInput,
 					loop_span.SpanFieldOutput,
 				},
+				Filters: loop_span.GetModelSpansFilter(),
 			}
 			st := time.Now()
 			spans, err := r.traceRepo.GetTrace(ctx, qReq)