mirror of https://github.com/grpc/grpc.git
client channel: create a header file for internal interfaces (#32423)
This is just a bit of a code reorganization, no functional changes.pull/32451/head
Mark D. Roth
2 years ago
committed by
GitHub
parent
50e6af14ff
commit
3faeb3af80
17 changed files with 157 additions and 144 deletions
@ -0,0 +1,114 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2015 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.
|
||||||
|
//
|
||||||
|
|
||||||
|
#ifndef GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_CLIENT_CHANNEL_INTERNAL_H |
||||||
|
#define GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_CLIENT_CHANNEL_INTERNAL_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <utility> |
||||||
|
|
||||||
|
#include "absl/strings/string_view.h" |
||||||
|
|
||||||
|
#include "src/core/ext/filters/client_channel/config_selector.h" |
||||||
|
#include "src/core/lib/channel/context.h" |
||||||
|
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||||
|
#include "src/core/lib/gprpp/unique_type_name.h" |
||||||
|
#include "src/core/lib/load_balancing/lb_policy.h" |
||||||
|
#include "src/core/lib/service_config/service_config.h" |
||||||
|
#include "src/core/lib/service_config/service_config_call_data.h" |
||||||
|
#include "src/core/lib/service_config/service_config_parser.h" |
||||||
|
|
||||||
|
//
|
||||||
|
// This file contains internal interfaces used to allow various plugins
|
||||||
|
// (filters, LB policies, etc) to access internal data provided by the
|
||||||
|
// ClientChannel that is not normally accessible via external APIs.
|
||||||
|
//
|
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
// Internal type for LB call state interface. Provides an interface for
|
||||||
|
// LB policies to access internal call attributes.
|
||||||
|
class ClientChannelLbCallState : public LoadBalancingPolicy::CallState { |
||||||
|
public: |
||||||
|
virtual absl::string_view GetCallAttribute(UniqueTypeName type) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
// Internal type for ServiceConfigCallData. Provides access to the
|
||||||
|
// CallDispatchController.
|
||||||
|
class ClientChannelServiceConfigCallData : public ServiceConfigCallData { |
||||||
|
public: |
||||||
|
ClientChannelServiceConfigCallData( |
||||||
|
RefCountedPtr<ServiceConfig> service_config, |
||||||
|
const ServiceConfigParser::ParsedConfigVector* method_configs, |
||||||
|
ServiceConfigCallData::CallAttributes call_attributes, |
||||||
|
ConfigSelector::CallDispatchController* call_dispatch_controller, |
||||||
|
grpc_call_context_element* call_context) |
||||||
|
: ServiceConfigCallData(std::move(service_config), method_configs, |
||||||
|
std::move(call_attributes)), |
||||||
|
call_dispatch_controller_(call_dispatch_controller) { |
||||||
|
call_context[GRPC_CONTEXT_SERVICE_CONFIG_CALL_DATA].value = this; |
||||||
|
call_context[GRPC_CONTEXT_SERVICE_CONFIG_CALL_DATA].destroy = Destroy; |
||||||
|
} |
||||||
|
|
||||||
|
ConfigSelector::CallDispatchController* call_dispatch_controller() { |
||||||
|
return &call_dispatch_controller_; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
// A wrapper for the CallDispatchController returned by the ConfigSelector.
|
||||||
|
// Handles the case where the ConfigSelector doees not return any
|
||||||
|
// CallDispatchController.
|
||||||
|
// Also ensures that we call Commit() at most once, which allows the
|
||||||
|
// client channel code to call Commit() when the call is complete in case
|
||||||
|
// it wasn't called earlier, without needing to know whether or not it was.
|
||||||
|
class CallDispatchControllerWrapper |
||||||
|
: public ConfigSelector::CallDispatchController { |
||||||
|
public: |
||||||
|
explicit CallDispatchControllerWrapper( |
||||||
|
ConfigSelector::CallDispatchController* call_dispatch_controller) |
||||||
|
: call_dispatch_controller_(call_dispatch_controller) {} |
||||||
|
|
||||||
|
bool ShouldRetry() override { |
||||||
|
if (call_dispatch_controller_ != nullptr) { |
||||||
|
return call_dispatch_controller_->ShouldRetry(); |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
void Commit() override { |
||||||
|
if (call_dispatch_controller_ != nullptr && !commit_called_) { |
||||||
|
call_dispatch_controller_->Commit(); |
||||||
|
commit_called_ = true; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
ConfigSelector::CallDispatchController* call_dispatch_controller_; |
||||||
|
bool commit_called_ = false; |
||||||
|
}; |
||||||
|
|
||||||
|
static void Destroy(void* ptr) { |
||||||
|
auto* self = static_cast<ClientChannelServiceConfigCallData*>(ptr); |
||||||
|
self->~ClientChannelServiceConfigCallData(); |
||||||
|
} |
||||||
|
|
||||||
|
CallDispatchControllerWrapper call_dispatch_controller_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_CLIENT_CHANNEL_INTERNAL_H
|
||||||
@ -1,39 +0,0 @@ |
|||||||
//
|
|
||||||
// Copyright 2015 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.
|
|
||||||
//
|
|
||||||
|
|
||||||
#ifndef GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_LB_CALL_STATE_INTERNAL_H |
|
||||||
#define GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_LB_CALL_STATE_INTERNAL_H |
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include "absl/strings/string_view.h" |
|
||||||
|
|
||||||
#include "src/core/lib/gprpp/unique_type_name.h" |
|
||||||
#include "src/core/lib/load_balancing/lb_policy.h" |
|
||||||
|
|
||||||
namespace grpc_core { |
|
||||||
|
|
||||||
//
|
|
||||||
// LbCallStateInternal
|
|
||||||
//
|
|
||||||
class LbCallStateInternal : public LoadBalancingPolicy::CallState { |
|
||||||
public: |
|
||||||
virtual absl::string_view GetCallAttribute(UniqueTypeName type) = 0; |
|
||||||
}; |
|
||||||
|
|
||||||
} // namespace grpc_core
|
|
||||||
|
|
||||||
#endif // GRPC_SRC_CORE_EXT_FILTERS_CLIENT_CHANNEL_LB_CALL_STATE_INTERNAL_H
|
|
||||||
Loading…
Reference in new issue