From 39ba4860ed8bf168c22562efaf1a0586bf30eba8 Mon Sep 17 00:00:00 2001 From: taloric Date: Wed, 11 Sep 2024 00:51:16 +0800 Subject: [PATCH] feat: set max_iteration only query trace_id data when is 0 --- app/app/application/l7_flow_tracing.py | 8 ++++++++ app/app/models/models.py | 2 +- 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/app/app/application/l7_flow_tracing.py b/app/app/application/l7_flow_tracing.py index c25aa84..419e77c 100644 --- a/app/app/application/l7_flow_tracing.py +++ b/app/app/application/l7_flow_tracing.py @@ -282,6 +282,12 @@ async def query_and_trace_flowmetas( if trace_id: allowed_trace_ids.add(trace_id) new_trace_ids_in_prev_iteration.add(trace_id) + + # max_iterations set to 0 means only query data with trace_id + only_query_trace_id = False + if max_iteration == 0: + max_iteration = 1 + only_query_trace_id = True # 进行迭代查询，上限为 config.spec.max_iteration for i in range(max_iteration): @@ -358,6 +364,8 @@ async def query_and_trace_flowmetas( if only_query_app_spans: # no more iterations needed break + if only_query_trace_id: # no more iterations needed + break # 2. Query by tcp_seq / syscall_trace_id / x_request_id new_filters = [] diff --git a/app/app/models/models.py b/app/app/models/models.py index 0cb3d6a..46afa7b 100644 --- a/app/app/models/models.py +++ b/app/app/models/models.py @@ -21,7 +21,7 @@ class FlowLogL7Tracing(Model): debug = BooleanType(serialized_name="DEBUG", required=False) max_iteration = IntType(serialized_name="MAX_ITERATION", required=False, - min_value=1, + min_value=0, default=config.max_iteration) network_delay_us = IntType(serialized_name="NETWORK_DELAY_US", required=False,