File tree 1 file changed +8
-4
lines changed
ibis-server/app/routers/v2
1 file changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -48,12 +48,15 @@ async def query(
48
48
query_cache_manager : QueryCacheManager = Depends (get_query_cache_manager ),
49
49
headers : Annotated [str | None , Header ()] = None ,
50
50
) -> Response :
51
- span_name = (
52
- f"v2_query_{ data_source } _dry_run" if dry_run else f"v2_query_{ data_source } "
53
- )
51
+ span_name = f"v2_query_{ data_source } "
52
+ if dry_run :
53
+ span_name += "_dry_run"
54
+ if cache_enable :
55
+ span_name += "_cache_enable"
56
+
54
57
with tracer .start_as_current_span (
55
58
name = span_name , kind = trace .SpanKind .SERVER , context = build_context (headers )
56
- ):
59
+ ) as span :
57
60
try :
58
61
sql = pushdown_limit (dto .sql , limit )
59
62
except Exception as e :
@@ -89,6 +92,7 @@ async def query(
89
92
cache_hit = cached_result is not None
90
93
91
94
if cache_hit :
95
+ span .add_event ("cache hit" )
92
96
response = ORJSONResponse (to_json (cached_result ))
93
97
response .headers ["X-Cache-Hit" ] = str (cache_hit ).lower ()
94
98
return response
You can’t perform that action at this time.
0 commit comments