File tree 4 files changed +6
-2
lines changed
4 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -37,6 +37,7 @@ const STREAM_QUERY_CHANNEL_LEN: usize = 20;
37
37
38
38
impl < Q : QueryExecutor + ' static > Proxy < Q > {
39
39
pub async fn handle_sql_query ( & self , ctx : Context , req : SqlQueryRequest ) -> SqlQueryResponse {
40
+ GRPC_HANDLER_COUNTER_VEC . query . inc ( ) ;
40
41
self . hotspot_recorder . inc_sql_query_reqs ( & req) . await ;
41
42
match self . handle_sql_query_internal ( ctx, req) . await {
42
43
Err ( e) => {
@@ -80,6 +81,7 @@ impl<Q: QueryExecutor + 'static> Proxy<Q> {
80
81
ctx : Context ,
81
82
req : SqlQueryRequest ,
82
83
) -> BoxStream < ' static , SqlQueryResponse > {
84
+ GRPC_HANDLER_COUNTER_VEC . stream_query . inc ( ) ;
83
85
self . hotspot_recorder . inc_sql_query_reqs ( & req) . await ;
84
86
match self . clone ( ) . handle_stream_query_internal ( ctx, req) . await {
85
87
Err ( e) => stream:: once ( async {
Original file line number Diff line number Diff line change @@ -12,8 +12,10 @@ make_auto_flush_static_metric! {
12
12
write_failed,
13
13
query_succeeded,
14
14
query_failed,
15
+ query,
15
16
stream_query_succeeded,
16
17
stream_query_failed,
18
+ stream_query,
17
19
write_succeeded_row,
18
20
write_failed_row,
19
21
query_succeeded_row,
Original file line number Diff line number Diff line change @@ -38,8 +38,7 @@ make_auto_flush_static_metric! {
38
38
pub label_enum RemoteEngineGrpcTypeKind {
39
39
write_succeeded,
40
40
write_failed,
41
- query_succeeded,
42
- query_failed,
41
+ stream_query,
43
42
stream_query_succeeded,
44
43
stream_query_failed,
45
44
write_succeeded_row,
Original file line number Diff line number Diff line change @@ -235,6 +235,7 @@ impl<Q: QueryExecutor + 'static> RemoteEngineService for RemoteEngineServiceImpl
235
235
& self ,
236
236
request : Request < ReadRequest > ,
237
237
) -> std:: result:: Result < Response < Self :: ReadStream > , Status > {
238
+ REMOTE_ENGINE_GRPC_HANDLER_COUNTER_VEC . stream_query . inc ( ) ;
238
239
match self . stream_read_internal ( request) . await {
239
240
Ok ( stream) => {
240
241
let new_stream: Self :: ReadStream = Box :: pin ( stream. map ( |res| match res {
You can’t perform that action at this time.
0 commit comments