Craig Tiller | af69180 | 2015-06-23 14:57:07 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Craig Tiller | 6169d5f | 2016-03-31 07:46:18 -0700 | [diff] [blame] | 3 | * Copyright 2015, Google Inc. |
Craig Tiller | af69180 | 2015-06-23 14:57:07 -0700 | [diff] [blame] | 4 | * All rights reserved. |
| 5 | * |
| 6 | * Redistribution and use in source and binary forms, with or without |
| 7 | * modification, are permitted provided that the following conditions are |
| 8 | * met: |
| 9 | * |
| 10 | * * Redistributions of source code must retain the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer. |
| 12 | * * Redistributions in binary form must reproduce the above |
| 13 | * copyright notice, this list of conditions and the following disclaimer |
| 14 | * in the documentation and/or other materials provided with the |
| 15 | * distribution. |
| 16 | * * Neither the name of Google Inc. nor the names of its |
| 17 | * contributors may be used to endorse or promote products derived from |
| 18 | * this software without specific prior written permission. |
| 19 | * |
| 20 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 21 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 22 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 23 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 24 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 25 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 26 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 27 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 28 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 29 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 30 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 31 | * |
| 32 | */ |
| 33 | |
Mark D. Roth | 2137cd8 | 2016-09-14 09:04:00 -0700 | [diff] [blame] | 34 | #include "src/core/ext/client_channel/lb_policy.h" |
Craig Tiller | 7c0b4d7 | 2015-06-25 14:44:44 -0700 | [diff] [blame] | 35 | |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 36 | #define WEAK_REF_BITS 16 |
| 37 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 38 | void grpc_lb_policy_init(grpc_lb_policy *policy, |
| 39 | const grpc_lb_policy_vtable *vtable) { |
Craig Tiller | 4ab82d2 | 2015-06-29 09:40:33 -0700 | [diff] [blame] | 40 | policy->vtable = vtable; |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 41 | gpr_atm_no_barrier_store(&policy->ref_pair, 1 << WEAK_REF_BITS); |
Craig Tiller | 69b093b | 2016-02-25 19:04:07 -0800 | [diff] [blame] | 42 | policy->interested_parties = grpc_pollset_set_create(); |
Craig Tiller | d7b68e7 | 2015-06-28 11:41:09 -0700 | [diff] [blame] | 43 | } |
Craig Tiller | 7c0b4d7 | 2015-06-25 14:44:44 -0700 | [diff] [blame] | 44 | |
Craig Tiller | d7b68e7 | 2015-06-28 11:41:09 -0700 | [diff] [blame] | 45 | #ifdef GRPC_LB_POLICY_REFCOUNT_DEBUG |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 46 | #define REF_FUNC_EXTRA_ARGS , const char *file, int line, const char *reason |
| 47 | #define REF_MUTATE_EXTRA_ARGS REF_FUNC_EXTRA_ARGS, const char *purpose |
| 48 | #define REF_FUNC_PASS_ARGS(new_reason) , file, line, new_reason |
| 49 | #define REF_MUTATE_PASS_ARGS(purpose) , file, line, reason, purpose |
Craig Tiller | d7b68e7 | 2015-06-28 11:41:09 -0700 | [diff] [blame] | 50 | #else |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 51 | #define REF_FUNC_EXTRA_ARGS |
| 52 | #define REF_MUTATE_EXTRA_ARGS |
| 53 | #define REF_FUNC_PASS_ARGS(new_reason) |
| 54 | #define REF_MUTATE_PASS_ARGS(x) |
Craig Tiller | d7b68e7 | 2015-06-28 11:41:09 -0700 | [diff] [blame] | 55 | #endif |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 56 | |
Craig Tiller | 1d881fb | 2015-12-01 07:39:04 -0800 | [diff] [blame] | 57 | static gpr_atm ref_mutate(grpc_lb_policy *c, gpr_atm delta, |
| 58 | int barrier REF_MUTATE_EXTRA_ARGS) { |
| 59 | gpr_atm old_val = barrier ? gpr_atm_full_fetch_add(&c->ref_pair, delta) |
| 60 | : gpr_atm_no_barrier_fetch_add(&c->ref_pair, delta); |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 61 | #ifdef GRPC_LB_POLICY_REFCOUNT_DEBUG |
Craig Tiller | 1d881fb | 2015-12-01 07:39:04 -0800 | [diff] [blame] | 62 | gpr_log(file, line, GPR_LOG_SEVERITY_DEBUG, |
David Garcia Quintas | 2a6c683 | 2016-06-15 18:11:08 -0700 | [diff] [blame] | 63 | "LB_POLICY: 0x%" PRIxPTR " %12s 0x%" PRIxPTR " -> 0x%" PRIxPTR |
| 64 | " [%s]", |
| 65 | (intptr_t)c, purpose, old_val, old_val + delta, reason); |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 66 | #endif |
| 67 | return old_val; |
Craig Tiller | d7b68e7 | 2015-06-28 11:41:09 -0700 | [diff] [blame] | 68 | } |
| 69 | |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 70 | void grpc_lb_policy_ref(grpc_lb_policy *policy REF_FUNC_EXTRA_ARGS) { |
| 71 | ref_mutate(policy, 1 << WEAK_REF_BITS, 0 REF_MUTATE_PASS_ARGS("STRONG_REF")); |
| 72 | } |
| 73 | |
Craig Tiller | 1d881fb | 2015-12-01 07:39:04 -0800 | [diff] [blame] | 74 | void grpc_lb_policy_unref(grpc_exec_ctx *exec_ctx, |
| 75 | grpc_lb_policy *policy REF_FUNC_EXTRA_ARGS) { |
| 76 | gpr_atm old_val = |
| 77 | ref_mutate(policy, (gpr_atm)1 - (gpr_atm)(1 << WEAK_REF_BITS), |
| 78 | 1 REF_MUTATE_PASS_ARGS("STRONG_UNREF")); |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 79 | gpr_atm mask = ~(gpr_atm)((1 << WEAK_REF_BITS) - 1); |
| 80 | gpr_atm check = 1 << WEAK_REF_BITS; |
| 81 | if ((old_val & mask) == check) { |
| 82 | policy->vtable->shutdown(exec_ctx, policy); |
| 83 | } |
Craig Tiller | 1d881fb | 2015-12-01 07:39:04 -0800 | [diff] [blame] | 84 | grpc_lb_policy_weak_unref(exec_ctx, |
| 85 | policy REF_FUNC_PASS_ARGS("strong-unref")); |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 86 | } |
| 87 | |
| 88 | void grpc_lb_policy_weak_ref(grpc_lb_policy *policy REF_FUNC_EXTRA_ARGS) { |
| 89 | ref_mutate(policy, 1, 0 REF_MUTATE_PASS_ARGS("WEAK_REF")); |
| 90 | } |
| 91 | |
Craig Tiller | 1d881fb | 2015-12-01 07:39:04 -0800 | [diff] [blame] | 92 | void grpc_lb_policy_weak_unref(grpc_exec_ctx *exec_ctx, |
| 93 | grpc_lb_policy *policy REF_FUNC_EXTRA_ARGS) { |
| 94 | gpr_atm old_val = |
| 95 | ref_mutate(policy, -(gpr_atm)1, 1 REF_MUTATE_PASS_ARGS("WEAK_UNREF")); |
Craig Tiller | 4861304 | 2015-11-29 14:45:11 -0800 | [diff] [blame] | 96 | if (old_val == 1) { |
Craig Tiller | 9e5ac1b | 2017-02-14 22:25:50 -0800 | [diff] [blame] | 97 | grpc_pollset_set_destroy(exec_ctx, policy->interested_parties); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 98 | policy->vtable->destroy(exec_ctx, policy); |
| 99 | } |
Craig Tiller | 7c0b4d7 | 2015-06-25 14:44:44 -0700 | [diff] [blame] | 100 | } |
| 101 | |
Craig Tiller | 577c9b2 | 2015-11-02 14:11:15 -0800 | [diff] [blame] | 102 | int grpc_lb_policy_pick(grpc_exec_ctx *exec_ctx, grpc_lb_policy *policy, |
David Garcia Quintas | 8aace51 | 2016-08-15 14:55:12 -0700 | [diff] [blame] | 103 | const grpc_lb_policy_pick_args *pick_args, |
David Garcia Quintas | 331b9c0 | 2016-09-12 18:37:05 -0700 | [diff] [blame] | 104 | grpc_connected_subchannel **target, void **user_data, |
Craig Tiller | ab33b48 | 2015-11-21 08:11:04 -0800 | [diff] [blame] | 105 | grpc_closure *on_complete) { |
David Garcia Quintas | 331b9c0 | 2016-09-12 18:37:05 -0700 | [diff] [blame] | 106 | return policy->vtable->pick(exec_ctx, policy, pick_args, target, user_data, |
| 107 | on_complete); |
Craig Tiller | 577c9b2 | 2015-11-02 14:11:15 -0800 | [diff] [blame] | 108 | } |
| 109 | |
| 110 | void grpc_lb_policy_cancel_pick(grpc_exec_ctx *exec_ctx, grpc_lb_policy *policy, |
Mark D. Roth | 5f84400 | 2016-09-08 08:20:53 -0700 | [diff] [blame] | 111 | grpc_connected_subchannel **target, |
| 112 | grpc_error *error) { |
| 113 | policy->vtable->cancel_pick(exec_ctx, policy, target, error); |
Craig Tiller | 7c0b4d7 | 2015-06-25 14:44:44 -0700 | [diff] [blame] | 114 | } |
Craig Tiller | ca3e9d3 | 2015-06-27 18:37:27 -0700 | [diff] [blame] | 115 | |
Craig Tiller | 8c0d96f | 2016-03-11 14:27:52 -0800 | [diff] [blame] | 116 | void grpc_lb_policy_cancel_picks(grpc_exec_ctx *exec_ctx, |
| 117 | grpc_lb_policy *policy, |
| 118 | uint32_t initial_metadata_flags_mask, |
Mark D. Roth | e65ff11 | 2016-09-09 13:48:38 -0700 | [diff] [blame] | 119 | uint32_t initial_metadata_flags_eq, |
| 120 | grpc_error *error) { |
Craig Tiller | 8c0d96f | 2016-03-11 14:27:52 -0800 | [diff] [blame] | 121 | policy->vtable->cancel_picks(exec_ctx, policy, initial_metadata_flags_mask, |
Mark D. Roth | e65ff11 | 2016-09-09 13:48:38 -0700 | [diff] [blame] | 122 | initial_metadata_flags_eq, error); |
Craig Tiller | 8c0d96f | 2016-03-11 14:27:52 -0800 | [diff] [blame] | 123 | } |
| 124 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 125 | void grpc_lb_policy_exit_idle(grpc_exec_ctx *exec_ctx, grpc_lb_policy *policy) { |
| 126 | policy->vtable->exit_idle(exec_ctx, policy); |
Craig Tiller | 48cb07c | 2015-07-15 16:16:15 -0700 | [diff] [blame] | 127 | } |
Craig Tiller | 1ada6ad | 2015-07-16 16:19:14 -0700 | [diff] [blame] | 128 | |
Craig Tiller | e2c6237 | 2015-12-07 16:11:03 -0800 | [diff] [blame] | 129 | void grpc_lb_policy_ping_one(grpc_exec_ctx *exec_ctx, grpc_lb_policy *policy, |
| 130 | grpc_closure *closure) { |
Craig Tiller | 28bf891 | 2015-12-07 16:07:04 -0800 | [diff] [blame] | 131 | policy->vtable->ping_one(exec_ctx, policy, closure); |
| 132 | } |
| 133 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 134 | void grpc_lb_policy_notify_on_state_change(grpc_exec_ctx *exec_ctx, |
| 135 | grpc_lb_policy *policy, |
| 136 | grpc_connectivity_state *state, |
| 137 | grpc_closure *closure) { |
| 138 | policy->vtable->notify_on_state_change(exec_ctx, policy, state, closure); |
Craig Tiller | 1ada6ad | 2015-07-16 16:19:14 -0700 | [diff] [blame] | 139 | } |
| 140 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 141 | grpc_connectivity_state grpc_lb_policy_check_connectivity( |
Craig Tiller | 804ff71 | 2016-05-05 16:25:40 -0700 | [diff] [blame] | 142 | grpc_exec_ctx *exec_ctx, grpc_lb_policy *policy, |
| 143 | grpc_error **connectivity_error) { |
| 144 | return policy->vtable->check_connectivity(exec_ctx, policy, |
| 145 | connectivity_error); |
Craig Tiller | 1ada6ad | 2015-07-16 16:19:14 -0700 | [diff] [blame] | 146 | } |