mirror of https://github.com/grpc/grpc.git
Mark D. Roth
6 years ago
parent
c4c4b9152f
commit
40f22bfc94
10 changed files with 312 additions and 99 deletions
@ -0,0 +1,25 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2018 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include "test/core/util/forwarding_load_balancing_policy.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
TraceFlag grpc_trace_forwarding_lb(false, "forwarding_lb"); |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,129 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2018 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <string> |
||||
|
||||
#include "src/core/ext/filters/client_channel/lb_policy.h" |
||||
#include "src/core/ext/filters/client_channel/lb_policy_registry.h" |
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/channel/channelz.h" |
||||
#include "src/core/lib/debug/trace.h" |
||||
#include "src/core/lib/gprpp/orphanable.h" |
||||
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||
#include "src/core/lib/iomgr/closure.h" |
||||
#include "src/core/lib/iomgr/combiner.h" |
||||
#include "src/core/lib/iomgr/error.h" |
||||
#include "src/core/lib/iomgr/pollset_set.h" |
||||
#include "src/core/lib/json/json.h" |
||||
#include "src/core/lib/transport/connectivity_state.h" |
||||
|
||||
#ifndef GRPC_TEST_CORE_UTIL_FORWARDING_LOAD_BALANCING_POLICY_H |
||||
#define GRPC_TEST_CORE_UTIL_FORWARDING_LOAD_BALANCING_POLICY_H |
||||
|
||||
namespace grpc_core { |
||||
|
||||
extern TraceFlag grpc_trace_forwarding_lb; |
||||
|
||||
// A minimal forwarding class to avoid implementing a standalone test LB.
|
||||
class ForwardingLoadBalancingPolicy : public LoadBalancingPolicy { |
||||
public: |
||||
ForwardingLoadBalancingPolicy(const Args& args, |
||||
const std::string& delegate_policy_name) |
||||
: LoadBalancingPolicy(args) { |
||||
delegate_ = |
||||
LoadBalancingPolicyRegistry::CreateLoadBalancingPolicy( |
||||
delegate_policy_name.c_str(), args); |
||||
grpc_pollset_set_add_pollset_set(delegate_->interested_parties(), |
||||
interested_parties()); |
||||
// Give re-resolution closure to delegate.
|
||||
GRPC_CLOSURE_INIT(&on_delegate_request_reresolution_, |
||||
OnDelegateRequestReresolutionLocked, this, |
||||
grpc_combiner_scheduler(combiner())); |
||||
Ref().release(); // held by callback.
|
||||
delegate_->SetReresolutionClosureLocked(&on_delegate_request_reresolution_); |
||||
} |
||||
|
||||
const char* name() const override { return delegate_->name(); } |
||||
|
||||
void UpdateLocked(const grpc_channel_args& args, |
||||
grpc_json* lb_config) override { |
||||
delegate_->UpdateLocked(args, lb_config); |
||||
} |
||||
|
||||
bool PickLocked(PickState* pick, grpc_error** error) override { |
||||
return delegate_->PickLocked(pick, error); |
||||
} |
||||
|
||||
void CancelPickLocked(PickState* pick, grpc_error* error) override { |
||||
delegate_->CancelPickLocked(pick, error); |
||||
} |
||||
|
||||
void CancelMatchingPicksLocked(uint32_t initial_metadata_flags_mask, |
||||
uint32_t initial_metadata_flags_eq, |
||||
grpc_error* error) override { |
||||
delegate_->CancelMatchingPicksLocked(initial_metadata_flags_mask, |
||||
initial_metadata_flags_eq, error); |
||||
} |
||||
|
||||
void NotifyOnStateChangeLocked(grpc_connectivity_state* state, |
||||
grpc_closure* closure) override { |
||||
delegate_->NotifyOnStateChangeLocked(state, closure); |
||||
} |
||||
|
||||
grpc_connectivity_state CheckConnectivityLocked( |
||||
grpc_error** connectivity_error) override { |
||||
return delegate_->CheckConnectivityLocked(connectivity_error); |
||||
} |
||||
|
||||
void HandOffPendingPicksLocked(LoadBalancingPolicy* new_policy) override { |
||||
delegate_->HandOffPendingPicksLocked(new_policy); |
||||
} |
||||
|
||||
void ExitIdleLocked() override { delegate_->ExitIdleLocked(); } |
||||
|
||||
void ResetBackoffLocked() override { delegate_->ResetBackoffLocked(); } |
||||
|
||||
void FillChildRefsForChannelz( |
||||
channelz::ChildRefsList* child_subchannels, |
||||
channelz::ChildRefsList* child_channels) override { |
||||
delegate_->FillChildRefsForChannelz(child_subchannels, child_channels); |
||||
} |
||||
|
||||
private: |
||||
void ShutdownLocked() override { delegate_.reset(); } |
||||
|
||||
static void OnDelegateRequestReresolutionLocked(void* arg, |
||||
grpc_error* error) { |
||||
ForwardingLoadBalancingPolicy* self = |
||||
static_cast<ForwardingLoadBalancingPolicy*>(arg); |
||||
if (error != GRPC_ERROR_NONE || self->delegate_ == nullptr) { |
||||
self->Unref(); |
||||
return; |
||||
} |
||||
self->TryReresolutionLocked(&grpc_trace_forwarding_lb, GRPC_ERROR_NONE); |
||||
self->delegate_->SetReresolutionClosureLocked( |
||||
&self->on_delegate_request_reresolution_); |
||||
} |
||||
|
||||
OrphanablePtr<LoadBalancingPolicy> delegate_; |
||||
grpc_closure on_delegate_request_reresolution_; |
||||
}; |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_TEST_CORE_UTIL_FORWARDING_LOAD_BALANCING_POLICY_H
|
||||
Loading…
Reference in new issue