mirror of https://github.com/grpc/grpc.git
Mark D. Roth
5 years ago
parent
3f523c1ff8
commit
b01072586d
28 changed files with 1812 additions and 1466 deletions
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,26 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2019 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_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CHANNEL_ARGS_H |
||||||
|
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CHANNEL_ARGS_H |
||||||
|
|
||||||
|
// Boolean channel arg indicating whether the target is an xds server.
|
||||||
|
#define GRPC_ARG_ADDRESS_IS_XDS_SERVER "grpc.address_is_xds_server" |
||||||
|
|
||||||
|
// Pointer channel arg containing a ref to the XdsClient object.
|
||||||
|
#define GRPC_ARG_XDS_CLIENT "grpc.xds_client" |
||||||
|
|
||||||
|
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CHANNEL_ARGS_H */ |
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,153 @@ |
|||||||
|
//
|
||||||
|
// Copyright 2019 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_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CLIENT_H |
||||||
|
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CLIENT_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/ext/filters/client_channel/service_config.h" |
||||||
|
#include "src/core/ext/filters/client_channel/xds/xds_api.h" |
||||||
|
#include "src/core/ext/filters/client_channel/xds/xds_client_stats.h" |
||||||
|
#include "src/core/lib/gprpp/map.h" |
||||||
|
#include "src/core/lib/gprpp/memory.h" |
||||||
|
#include "src/core/lib/gprpp/orphanable.h" |
||||||
|
#include "src/core/lib/gprpp/ref_counted.h" |
||||||
|
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||||
|
#include "src/core/lib/gprpp/set.h" |
||||||
|
#include "src/core/lib/gprpp/string_view.h" |
||||||
|
#include "src/core/lib/iomgr/combiner.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
extern TraceFlag xds_client_trace; |
||||||
|
|
||||||
|
class XdsClient : public InternallyRefCounted<XdsClient> { |
||||||
|
public: |
||||||
|
// Service config watcher interface. Implemented by callers.
|
||||||
|
class ServiceConfigWatcherInterface { |
||||||
|
public: |
||||||
|
virtual ~ServiceConfigWatcherInterface() = default; |
||||||
|
|
||||||
|
virtual void OnServiceConfigChanged( |
||||||
|
RefCountedPtr<ServiceConfig> service_config) = 0; |
||||||
|
|
||||||
|
virtual void OnError(grpc_error* error) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
// Cluster data watcher interface. Implemented by callers.
|
||||||
|
class ClusterWatcherInterface { |
||||||
|
public: |
||||||
|
virtual ~ClusterWatcherInterface() = default; |
||||||
|
|
||||||
|
virtual void OnClusterChanged(CdsUpdate cluster_data) = 0; |
||||||
|
|
||||||
|
virtual void OnError(grpc_error* error) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
// Endpoint data watcher interface. Implemented by callers.
|
||||||
|
class EndpointWatcherInterface { |
||||||
|
public: |
||||||
|
virtual ~EndpointWatcherInterface() = default; |
||||||
|
|
||||||
|
virtual void OnEndpointChanged(EdsUpdate update) = 0; |
||||||
|
|
||||||
|
virtual void OnError(grpc_error* error) = 0; |
||||||
|
}; |
||||||
|
|
||||||
|
XdsClient(grpc_combiner* combiner, grpc_pollset_set* interested_parties, |
||||||
|
const char* balancer_name, StringView server_name, |
||||||
|
UniquePtr<ServiceConfigWatcherInterface> watcher, |
||||||
|
const grpc_channel_args& channel_args); |
||||||
|
~XdsClient(); |
||||||
|
|
||||||
|
void Orphan() override; |
||||||
|
|
||||||
|
// Start and cancel cluster data watch for a cluster.
|
||||||
|
// The XdsClient takes ownership of the watcher, but the caller may
|
||||||
|
// keep a raw pointer to the watcher, which may be used only for
|
||||||
|
// cancellation. (Because the caller does not own the watcher, the
|
||||||
|
// pointer must not be used for any other purpose.)
|
||||||
|
void WatchClusterData(StringView cluster, |
||||||
|
UniquePtr<ClusterWatcherInterface> watcher); |
||||||
|
void CancelClusterDataWatch(StringView cluster, |
||||||
|
ClusterWatcherInterface* watcher); |
||||||
|
|
||||||
|
// Start and cancel endpoint data watch for a cluster.
|
||||||
|
// The XdsClient takes ownership of the watcher, but the caller may
|
||||||
|
// keep a raw pointer to the watcher, which may be used only for
|
||||||
|
// cancellation. (Because the caller does not own the watcher, the
|
||||||
|
// pointer must not be used for any other purpose.)
|
||||||
|
void WatchEndpointData(StringView cluster, |
||||||
|
UniquePtr<EndpointWatcherInterface> watcher); |
||||||
|
void CancelEndpointDataWatch(StringView cluster, |
||||||
|
EndpointWatcherInterface* watcher); |
||||||
|
|
||||||
|
// Adds and removes client stats for cluster.
|
||||||
|
void AddClientStats(StringView cluster, XdsClientStats* client_stats); |
||||||
|
void RemoveClientStats(StringView cluster, XdsClientStats* client_stats); |
||||||
|
|
||||||
|
// Resets connection backoff state.
|
||||||
|
void ResetBackoff(); |
||||||
|
|
||||||
|
// Helpers for encoding the XdsClient object in channel args.
|
||||||
|
grpc_arg MakeChannelArg() const; |
||||||
|
static RefCountedPtr<XdsClient> GetFromChannelArgs( |
||||||
|
const grpc_channel_args& args); |
||||||
|
|
||||||
|
private: |
||||||
|
class ChannelState; |
||||||
|
|
||||||
|
struct ClusterState { |
||||||
|
Map<ClusterWatcherInterface*, UniquePtr<ClusterWatcherInterface>> |
||||||
|
cluster_watchers; |
||||||
|
Map<EndpointWatcherInterface*, UniquePtr<EndpointWatcherInterface>> |
||||||
|
endpoint_watchers; |
||||||
|
Set<XdsClientStats*> client_stats; |
||||||
|
// The latest data seen from EDS.
|
||||||
|
EdsUpdate eds_update; |
||||||
|
}; |
||||||
|
|
||||||
|
// Sends an error notification to all watchers.
|
||||||
|
void NotifyOnError(grpc_error* error); |
||||||
|
|
||||||
|
// Channel arg vtable functions.
|
||||||
|
static void* ChannelArgCopy(void* p); |
||||||
|
static void ChannelArgDestroy(void* p); |
||||||
|
static int ChannelArgCmp(void* p, void* q); |
||||||
|
|
||||||
|
static const grpc_arg_pointer_vtable kXdsClientVtable; |
||||||
|
|
||||||
|
grpc_combiner* combiner_; |
||||||
|
grpc_pollset_set* interested_parties_; |
||||||
|
|
||||||
|
UniquePtr<char> server_name_; |
||||||
|
UniquePtr<ServiceConfigWatcherInterface> service_config_watcher_; |
||||||
|
|
||||||
|
// The channel for communicating with the xds server.
|
||||||
|
OrphanablePtr<ChannelState> chand_; |
||||||
|
|
||||||
|
// TODO(roth): When we need support for multiple clusters, replace
|
||||||
|
// cluster_state_ with a map keyed by cluster name.
|
||||||
|
ClusterState cluster_state_; |
||||||
|
// Map<StringView /*cluster*/, ClusterState, StringLess> clusters_;
|
||||||
|
|
||||||
|
bool shutting_down_ = false; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_XDS_XDS_CLIENT_H */ |
||||||
Loading…
Reference in new issue