|
2 | 2 | import uuid |
3 | 3 | import pandas as pd |
4 | 4 | from log import logger |
5 | | -from typing import List, Dict, Set, Callable |
| 5 | +from typing import List, Dict, Set, Callable, Tuple |
6 | 6 |
|
7 | | -from ast import Tuple |
8 | 7 | from pandas import DataFrame |
9 | 8 | from collections import defaultdict |
10 | 9 | from data.querier_client import Querier |
|
223 | 222 | TRACING_SRC_SYSCALL = "syscall" |
224 | 223 | TRACING_SRC_TCP_SEQ = "tcp_seq" |
225 | 224 | TRACING_SRC_X_REQ_ID = "x_request_id" |
| 225 | +# 默认不开启 dns 追踪 |
226 | 226 | TRACING_SRC_DNS = "dns" |
227 | 227 |
|
228 | 228 | DEFAULT_TRACING_SOURCE = [ |
229 | 229 | TRACING_SRC_TRACE_ID, TRACING_SRC_SYSCALL, TRACING_SRC_TCP_SEQ, |
230 | | - TRACING_SRC_X_REQ_ID, TRACING_SRC_DNS |
| 230 | + TRACING_SRC_X_REQ_ID |
231 | 231 | ] |
232 | 232 |
|
233 | 233 |
|
@@ -282,7 +282,7 @@ async def query_and_trace_flowmetas( |
282 | 282 | max_iteration: int = config.max_iteration, |
283 | 283 | network_delay_us: int = config.network_delay_us, |
284 | 284 | host_clock_offset_us: int = config.host_clock_offset_us, |
285 | | - app_spans_from_api: list = []) -> Tuple(list, list): |
| 285 | + app_spans_from_api: list = []) -> Tuple[Set, list]: |
286 | 286 | """多次迭代,查询可追踪到的所有 l7_flow_log 的摘要 |
287 | 287 | 参数说明: |
288 | 288 | time_filter: 查询的时间范围过滤条件,SQL表达式 |
@@ -312,7 +312,7 @@ async def query_and_trace_flowmetas( |
312 | 312 | dataframe_flowmetas = await self.query_flowmetas("1=1", base_filter) |
313 | 313 | if type(dataframe_flowmetas) != DataFrame or dataframe_flowmetas.empty: |
314 | 314 | # when app_spans_from_api got values from api, return it |
315 | | - return [], app_spans_from_api |
| 315 | + return set(), app_spans_from_api |
316 | 316 | l7_flow_ids = set(dataframe_flowmetas['_id']) # set(flow._id) |
317 | 317 |
|
318 | 318 | # 用于下一轮迭代,记录元信息 |
@@ -430,6 +430,7 @@ async def query_and_trace_flowmetas( |
430 | 430 | # 写入 trace_id_index 时,遇到空 trace_id 有可能会复用 index,导致重复 |
431 | 431 | # 于是,这里可能导致误查询,需要额外过滤一下 len(new_trace_id_arr)=0(trace_id='') 的情况 |
432 | 432 | new_trace_id_flow_delete_index.append(index) |
| 433 | + |
433 | 434 | if new_trace_id_flow_delete_index: |
434 | 435 | new_trace_id_flows = new_trace_id_flows.drop( |
435 | 436 | new_trace_id_flow_delete_index).reset_index( |
|
0 commit comments