diff --git a/libdd-trace-protobuf/build.rs b/libdd-trace-protobuf/build.rs index ae0eef29ae..14cfe8389b 100644 --- a/libdd-trace-protobuf/build.rs +++ b/libdd-trace-protobuf/build.rs @@ -207,6 +207,14 @@ fn generate_protobuf() { "ClientGroupedStats.DB_type", "#[serde(rename = \"DBType\")]", ); + config.field_attribute( + "ClientGroupedStats.GRPC_status_code", + "#[serde(rename = \"GRPCStatusCode\")]", + ); + config.field_attribute( + "ClientGroupedStats.service_source", + "#[serde(rename = \"srv_src\")]", + ); // idx module type attributes config.type_attribute("pb.idx.AnyValue", "#[derive(Deserialize, Serialize)]"); diff --git a/libdd-trace-protobuf/src/pb.rs b/libdd-trace-protobuf/src/pb.rs index c242c1b131..5e28ba08ad 100644 --- a/libdd-trace-protobuf/src/pb.rs +++ b/libdd-trace-protobuf/src/pb.rs @@ -635,6 +635,7 @@ pub struct ClientGroupedStats { pub is_trace_root: i32, #[prost(string, tag = "18")] #[serde(default)] + #[serde(rename = "GRPCStatusCode")] pub grpc_status_code: ::prost::alloc::string::String, /// HTTP method of the request #[prost(string, tag = "19")] @@ -649,6 +650,7 @@ pub struct ClientGroupedStats { /// @inject_tag: msg:"srv_src" #[prost(string, tag = "21")] #[serde(default)] + #[serde(rename = "srv_src")] pub service_source: ::prost::alloc::string::String, /// used to identify service override origin /// span_derived_primary_tags are user-configured tags that are extracted from spans and used for stats aggregation diff --git a/libdd-trace-protobuf/src/pb_test.rs b/libdd-trace-protobuf/src/pb_test.rs index 4d0672c4a5..5d9cb68b84 100644 --- a/libdd-trace-protobuf/src/pb_test.rs +++ b/libdd-trace-protobuf/src/pb_test.rs @@ -72,7 +72,7 @@ mod tests { ], "HTTPMethod": "GET", "HTTPEndpoint": "/test", - "GrpcStatusCode": "0" + "GRPCStatusCode": "0" } ] } diff --git a/libdd-trace-utils/src/stats_utils.rs b/libdd-trace-utils/src/stats_utils.rs index fe559a2692..4e17dcf181 100644 --- a/libdd-trace-utils/src/stats_utils.rs +++ b/libdd-trace-utils/src/stats_utils.rs @@ -155,7 +155,7 @@ mod mini_agent_tests { 0, 0 ], - "GrpcStatusCode": "0", + "GRPCStatusCode": "0", "HTTPMethod": "GET", "HTTPEndpoint": "/test" }