6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
# File 'lib/newrelic_security/instrumentation-security/grpc/server/chain.rb', line 6
def self.instrument!
::GRPC::RpcDesc.class_eval do
include NewRelic::Security::Instrumentation::GRPC::RpcDesc
alias_method :handle_request_response_without_security, :handle_request_response
def handle_request_response(active_call, mth, inter_ctx)
grpc_server_on_enter(active_call, mth, inter_ctx, false, false) { return handle_request_response_without_security(active_call, mth, inter_ctx) }
end
alias_method :handle_client_streamer_without_security, :handle_client_streamer
def handle_client_streamer(active_call, mth, inter_ctx)
grpc_server_on_enter(active_call, mth, inter_ctx, true, false) { return handle_request_response_without_security(active_call, mth, inter_ctx) }
end
alias_method :handle_server_streamer_without_security, :handle_server_streamer
def handle_server_streamer(active_call, mth, inter_ctx)
grpc_server_on_enter(active_call, mth, inter_ctx, false, true) { return handle_request_response_without_security(active_call, mth, inter_ctx) }
end
alias_method :handle_bidi_streamer_without_security, :handle_bidi_streamer
def handle_bidi_streamer(active_call, mth, inter_ctx)
grpc_server_on_enter(active_call, mth, inter_ctx, true, true) { return handle_request_response_without_security(active_call, mth, inter_ctx) }
end
end
end
|