Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2017 gRPC authors. |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
| 19 | #include "src/core/ext/filters/client_channel/lb_policy/grpclb/client_load_reporting_filter.h" |
| 20 | |
| 21 | #include <grpc/support/atm.h> |
| 22 | #include <grpc/support/log.h> |
| 23 | |
| 24 | #include "src/core/ext/filters/client_channel/lb_policy/grpclb/grpclb_client_stats.h" |
| 25 | #include "src/core/lib/iomgr/error.h" |
| 26 | #include "src/core/lib/profiling/timers.h" |
| 27 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 28 | static grpc_error* init_channel_elem(grpc_exec_ctx* exec_ctx, |
| 29 | grpc_channel_element* elem, |
| 30 | grpc_channel_element_args* args) { |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 31 | return GRPC_ERROR_NONE; |
| 32 | } |
| 33 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 34 | static void destroy_channel_elem(grpc_exec_ctx* exec_ctx, |
| 35 | grpc_channel_element* elem) {} |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 36 | |
| 37 | typedef struct { |
| 38 | // Stats object to update. |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 39 | grpc_grpclb_client_stats* client_stats; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 40 | // State for intercepting send_initial_metadata. |
| 41 | grpc_closure on_complete_for_send; |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 42 | grpc_closure* original_on_complete_for_send; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 43 | bool send_initial_metadata_succeeded; |
| 44 | // State for intercepting recv_initial_metadata. |
| 45 | grpc_closure recv_initial_metadata_ready; |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 46 | grpc_closure* original_recv_initial_metadata_ready; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 47 | bool recv_initial_metadata_succeeded; |
| 48 | } call_data; |
| 49 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 50 | static void on_complete_for_send(grpc_exec_ctx* exec_ctx, void* arg, |
| 51 | grpc_error* error) { |
| 52 | call_data* calld = (call_data*)arg; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 53 | if (error == GRPC_ERROR_NONE) { |
| 54 | calld->send_initial_metadata_succeeded = true; |
| 55 | } |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 56 | GRPC_CLOSURE_RUN(exec_ctx, calld->original_on_complete_for_send, |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 57 | GRPC_ERROR_REF(error)); |
| 58 | } |
| 59 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 60 | static void recv_initial_metadata_ready(grpc_exec_ctx* exec_ctx, void* arg, |
| 61 | grpc_error* error) { |
| 62 | call_data* calld = (call_data*)arg; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 63 | if (error == GRPC_ERROR_NONE) { |
| 64 | calld->recv_initial_metadata_succeeded = true; |
| 65 | } |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 66 | GRPC_CLOSURE_RUN(exec_ctx, calld->original_recv_initial_metadata_ready, |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 67 | GRPC_ERROR_REF(error)); |
| 68 | } |
| 69 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 70 | static grpc_error* init_call_elem(grpc_exec_ctx* exec_ctx, |
| 71 | grpc_call_element* elem, |
| 72 | const grpc_call_element_args* args) { |
| 73 | call_data* calld = (call_data*)elem->call_data; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 74 | // Get stats object from context and take a ref. |
Craig Tiller | 4782d92 | 2017-11-10 09:53:21 -0800 | [diff] [blame^] | 75 | GPR_ASSERT(args->context != nullptr); |
| 76 | GPR_ASSERT(args->context[GRPC_GRPCLB_CLIENT_STATS].value != nullptr); |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 77 | calld->client_stats = grpc_grpclb_client_stats_ref( |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 78 | (grpc_grpclb_client_stats*)args->context[GRPC_GRPCLB_CLIENT_STATS].value); |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 79 | // Record call started. |
| 80 | grpc_grpclb_client_stats_add_call_started(calld->client_stats); |
| 81 | return GRPC_ERROR_NONE; |
| 82 | } |
| 83 | |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 84 | static void destroy_call_elem(grpc_exec_ctx* exec_ctx, grpc_call_element* elem, |
| 85 | const grpc_call_final_info* final_info, |
| 86 | grpc_closure* ignored) { |
| 87 | call_data* calld = (call_data*)elem->call_data; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 88 | // Record call finished, optionally setting client_failed_to_send and |
| 89 | // received. |
| 90 | grpc_grpclb_client_stats_add_call_finished( |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 91 | !calld->send_initial_metadata_succeeded /* client_failed_to_send */, |
| 92 | calld->recv_initial_metadata_succeeded /* known_received */, |
| 93 | calld->client_stats); |
| 94 | // All done, so unref the stats object. |
| 95 | grpc_grpclb_client_stats_unref(calld->client_stats); |
| 96 | } |
| 97 | |
| 98 | static void start_transport_stream_op_batch( |
Craig Tiller | baa14a9 | 2017-11-03 09:09:36 -0700 | [diff] [blame] | 99 | grpc_exec_ctx* exec_ctx, grpc_call_element* elem, |
| 100 | grpc_transport_stream_op_batch* batch) { |
| 101 | call_data* calld = (call_data*)elem->call_data; |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 102 | GPR_TIMER_BEGIN("clr_start_transport_stream_op_batch", 0); |
| 103 | // Intercept send_initial_metadata. |
| 104 | if (batch->send_initial_metadata) { |
| 105 | calld->original_on_complete_for_send = batch->on_complete; |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 106 | GRPC_CLOSURE_INIT(&calld->on_complete_for_send, on_complete_for_send, calld, |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 107 | grpc_schedule_on_exec_ctx); |
| 108 | batch->on_complete = &calld->on_complete_for_send; |
| 109 | } |
| 110 | // Intercept recv_initial_metadata. |
| 111 | if (batch->recv_initial_metadata) { |
| 112 | calld->original_recv_initial_metadata_ready = |
| 113 | batch->payload->recv_initial_metadata.recv_initial_metadata_ready; |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 114 | GRPC_CLOSURE_INIT(&calld->recv_initial_metadata_ready, |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 115 | recv_initial_metadata_ready, calld, |
| 116 | grpc_schedule_on_exec_ctx); |
| 117 | batch->payload->recv_initial_metadata.recv_initial_metadata_ready = |
| 118 | &calld->recv_initial_metadata_ready; |
| 119 | } |
| 120 | // Chain to next filter. |
| 121 | grpc_call_next_op(exec_ctx, elem, batch); |
| 122 | GPR_TIMER_END("clr_start_transport_stream_op_batch", 0); |
| 123 | } |
| 124 | |
| 125 | const grpc_channel_filter grpc_client_load_reporting_filter = { |
| 126 | start_transport_stream_op_batch, |
| 127 | grpc_channel_next_op, |
| 128 | sizeof(call_data), |
| 129 | init_call_elem, |
| 130 | grpc_call_stack_ignore_set_pollset_or_pollset_set, |
| 131 | destroy_call_elem, |
| 132 | 0, // sizeof(channel_data) |
| 133 | init_channel_elem, |
| 134 | destroy_channel_elem, |
Mark D. Roth | 09e458c | 2017-05-02 08:13:26 -0700 | [diff] [blame] | 135 | grpc_channel_next_get_info, |
| 136 | "client_load_reporting"}; |