|
|
|
@ -52,12 +52,12 @@ static bool is_census_enabled(const grpc_channel_args *a) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
static bool maybe_add_census_filter(grpc_channel_stack_builder *builder, |
|
|
|
|
void *arg_must_be_null) { |
|
|
|
|
void *arg) { |
|
|
|
|
const grpc_channel_args *args = |
|
|
|
|
grpc_channel_stack_builder_get_channel_arguments(builder); |
|
|
|
|
if (is_census_enabled(args)) { |
|
|
|
|
return grpc_channel_stack_builder_prepend_filter( |
|
|
|
|
builder, &grpc_client_census_filter, NULL, NULL); |
|
|
|
|
builder, (const grpc_channel_filter *)arg, NULL, NULL); |
|
|
|
|
} |
|
|
|
|
return true; |
|
|
|
|
} |
|
|
|
@ -72,9 +72,11 @@ void census_grpc_plugin_init(void) { |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
grpc_channel_init_register_stage(GRPC_CLIENT_CHANNEL, INT_MAX, |
|
|
|
|
maybe_add_census_filter, NULL); |
|
|
|
|
maybe_add_census_filter, |
|
|
|
|
(void *)&grpc_client_census_filter); |
|
|
|
|
grpc_channel_init_register_stage(GRPC_SERVER_CHANNEL, INT_MAX, |
|
|
|
|
maybe_add_census_filter, NULL); |
|
|
|
|
maybe_add_census_filter, |
|
|
|
|
(void *)&grpc_server_census_filter); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void census_grpc_plugin_shutdown(void) { census_shutdown(); } |
|
|
|
|