mirror of https://github.com/grpc/grpc.git
Mark D. Roth
6 years ago
committed by
GitHub
parent
6f675abe85
commit
c5f344deaf
58 changed files with 846 additions and 1044 deletions
@ -1,163 +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. |
|
||||||
* |
|
||||||
*/ |
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include <string.h> |
|
||||||
|
|
||||||
#include <grpc/support/alloc.h> |
|
||||||
#include <grpc/support/string_util.h> |
|
||||||
|
|
||||||
#include "src/core/lib/channel/channel_args.h" |
|
||||||
|
|
||||||
#include "src/core/ext/filters/client_channel/lb_policy_factory.h" |
|
||||||
#include "src/core/ext/filters/client_channel/parse_address.h" |
|
||||||
|
|
||||||
grpc_lb_addresses* grpc_lb_addresses_create( |
|
||||||
size_t num_addresses, const grpc_lb_user_data_vtable* user_data_vtable) { |
|
||||||
grpc_lb_addresses* addresses = |
|
||||||
static_cast<grpc_lb_addresses*>(gpr_zalloc(sizeof(grpc_lb_addresses))); |
|
||||||
addresses->num_addresses = num_addresses; |
|
||||||
addresses->user_data_vtable = user_data_vtable; |
|
||||||
const size_t addresses_size = sizeof(grpc_lb_address) * num_addresses; |
|
||||||
addresses->addresses = |
|
||||||
static_cast<grpc_lb_address*>(gpr_zalloc(addresses_size)); |
|
||||||
return addresses; |
|
||||||
} |
|
||||||
|
|
||||||
grpc_lb_addresses* grpc_lb_addresses_copy(const grpc_lb_addresses* addresses) { |
|
||||||
grpc_lb_addresses* new_addresses = grpc_lb_addresses_create( |
|
||||||
addresses->num_addresses, addresses->user_data_vtable); |
|
||||||
memcpy(new_addresses->addresses, addresses->addresses, |
|
||||||
sizeof(grpc_lb_address) * addresses->num_addresses); |
|
||||||
for (size_t i = 0; i < addresses->num_addresses; ++i) { |
|
||||||
if (new_addresses->addresses[i].balancer_name != nullptr) { |
|
||||||
new_addresses->addresses[i].balancer_name = |
|
||||||
gpr_strdup(new_addresses->addresses[i].balancer_name); |
|
||||||
} |
|
||||||
if (new_addresses->addresses[i].user_data != nullptr) { |
|
||||||
new_addresses->addresses[i].user_data = addresses->user_data_vtable->copy( |
|
||||||
new_addresses->addresses[i].user_data); |
|
||||||
} |
|
||||||
} |
|
||||||
return new_addresses; |
|
||||||
} |
|
||||||
|
|
||||||
void grpc_lb_addresses_set_address(grpc_lb_addresses* addresses, size_t index, |
|
||||||
const void* address, size_t address_len, |
|
||||||
bool is_balancer, const char* balancer_name, |
|
||||||
void* user_data) { |
|
||||||
GPR_ASSERT(index < addresses->num_addresses); |
|
||||||
if (user_data != nullptr) GPR_ASSERT(addresses->user_data_vtable != nullptr); |
|
||||||
grpc_lb_address* target = &addresses->addresses[index]; |
|
||||||
memcpy(target->address.addr, address, address_len); |
|
||||||
target->address.len = static_cast<socklen_t>(address_len); |
|
||||||
target->is_balancer = is_balancer; |
|
||||||
target->balancer_name = gpr_strdup(balancer_name); |
|
||||||
target->user_data = user_data; |
|
||||||
} |
|
||||||
|
|
||||||
bool grpc_lb_addresses_set_address_from_uri(grpc_lb_addresses* addresses, |
|
||||||
size_t index, const grpc_uri* uri, |
|
||||||
bool is_balancer, |
|
||||||
const char* balancer_name, |
|
||||||
void* user_data) { |
|
||||||
grpc_resolved_address address; |
|
||||||
if (!grpc_parse_uri(uri, &address)) return false; |
|
||||||
grpc_lb_addresses_set_address(addresses, index, address.addr, address.len, |
|
||||||
is_balancer, balancer_name, user_data); |
|
||||||
return true; |
|
||||||
} |
|
||||||
|
|
||||||
int grpc_lb_addresses_cmp(const grpc_lb_addresses* addresses1, |
|
||||||
const grpc_lb_addresses* addresses2) { |
|
||||||
if (addresses1->num_addresses > addresses2->num_addresses) return 1; |
|
||||||
if (addresses1->num_addresses < addresses2->num_addresses) return -1; |
|
||||||
if (addresses1->user_data_vtable > addresses2->user_data_vtable) return 1; |
|
||||||
if (addresses1->user_data_vtable < addresses2->user_data_vtable) return -1; |
|
||||||
for (size_t i = 0; i < addresses1->num_addresses; ++i) { |
|
||||||
const grpc_lb_address* target1 = &addresses1->addresses[i]; |
|
||||||
const grpc_lb_address* target2 = &addresses2->addresses[i]; |
|
||||||
if (target1->address.len > target2->address.len) return 1; |
|
||||||
if (target1->address.len < target2->address.len) return -1; |
|
||||||
int retval = memcmp(target1->address.addr, target2->address.addr, |
|
||||||
target1->address.len); |
|
||||||
if (retval != 0) return retval; |
|
||||||
if (target1->is_balancer > target2->is_balancer) return 1; |
|
||||||
if (target1->is_balancer < target2->is_balancer) return -1; |
|
||||||
const char* balancer_name1 = |
|
||||||
target1->balancer_name != nullptr ? target1->balancer_name : ""; |
|
||||||
const char* balancer_name2 = |
|
||||||
target2->balancer_name != nullptr ? target2->balancer_name : ""; |
|
||||||
retval = strcmp(balancer_name1, balancer_name2); |
|
||||||
if (retval != 0) return retval; |
|
||||||
if (addresses1->user_data_vtable != nullptr) { |
|
||||||
retval = addresses1->user_data_vtable->cmp(target1->user_data, |
|
||||||
target2->user_data); |
|
||||||
if (retval != 0) return retval; |
|
||||||
} |
|
||||||
} |
|
||||||
return 0; |
|
||||||
} |
|
||||||
|
|
||||||
void grpc_lb_addresses_destroy(grpc_lb_addresses* addresses) { |
|
||||||
for (size_t i = 0; i < addresses->num_addresses; ++i) { |
|
||||||
gpr_free(addresses->addresses[i].balancer_name); |
|
||||||
if (addresses->addresses[i].user_data != nullptr) { |
|
||||||
addresses->user_data_vtable->destroy(addresses->addresses[i].user_data); |
|
||||||
} |
|
||||||
} |
|
||||||
gpr_free(addresses->addresses); |
|
||||||
gpr_free(addresses); |
|
||||||
} |
|
||||||
|
|
||||||
static void* lb_addresses_copy(void* addresses) { |
|
||||||
return grpc_lb_addresses_copy(static_cast<grpc_lb_addresses*>(addresses)); |
|
||||||
} |
|
||||||
static void lb_addresses_destroy(void* addresses) { |
|
||||||
grpc_lb_addresses_destroy(static_cast<grpc_lb_addresses*>(addresses)); |
|
||||||
} |
|
||||||
static int lb_addresses_cmp(void* addresses1, void* addresses2) { |
|
||||||
return grpc_lb_addresses_cmp(static_cast<grpc_lb_addresses*>(addresses1), |
|
||||||
static_cast<grpc_lb_addresses*>(addresses2)); |
|
||||||
} |
|
||||||
static const grpc_arg_pointer_vtable lb_addresses_arg_vtable = { |
|
||||||
lb_addresses_copy, lb_addresses_destroy, lb_addresses_cmp}; |
|
||||||
|
|
||||||
grpc_arg grpc_lb_addresses_create_channel_arg( |
|
||||||
const grpc_lb_addresses* addresses) { |
|
||||||
return grpc_channel_arg_pointer_create( |
|
||||||
(char*)GRPC_ARG_LB_ADDRESSES, (void*)addresses, &lb_addresses_arg_vtable); |
|
||||||
} |
|
||||||
|
|
||||||
grpc_lb_addresses* grpc_lb_addresses_find_channel_arg( |
|
||||||
const grpc_channel_args* channel_args) { |
|
||||||
const grpc_arg* lb_addresses_arg = |
|
||||||
grpc_channel_args_find(channel_args, GRPC_ARG_LB_ADDRESSES); |
|
||||||
if (lb_addresses_arg == nullptr || lb_addresses_arg->type != GRPC_ARG_POINTER) |
|
||||||
return nullptr; |
|
||||||
return static_cast<grpc_lb_addresses*>(lb_addresses_arg->value.pointer.p); |
|
||||||
} |
|
||||||
|
|
||||||
bool grpc_lb_addresses_contains_balancer_address( |
|
||||||
const grpc_lb_addresses& addresses) { |
|
||||||
for (size_t i = 0; i < addresses.num_addresses; ++i) { |
|
||||||
if (addresses.addresses[i].is_balancer) return true; |
|
||||||
} |
|
||||||
return false; |
|
||||||
} |
|
||||||
@ -0,0 +1,103 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/ext/filters/client_channel/server_address.h" |
||||||
|
|
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
//
|
||||||
|
// ServerAddress
|
||||||
|
//
|
||||||
|
|
||||||
|
ServerAddress::ServerAddress(const grpc_resolved_address& address, |
||||||
|
grpc_channel_args* args) |
||||||
|
: address_(address), args_(args) {} |
||||||
|
|
||||||
|
ServerAddress::ServerAddress(const void* address, size_t address_len, |
||||||
|
grpc_channel_args* args) |
||||||
|
: args_(args) { |
||||||
|
memcpy(address_.addr, address, address_len); |
||||||
|
address_.len = static_cast<socklen_t>(address_len); |
||||||
|
} |
||||||
|
|
||||||
|
int ServerAddress::Cmp(const ServerAddress& other) const { |
||||||
|
if (address_.len > other.address_.len) return 1; |
||||||
|
if (address_.len < other.address_.len) return -1; |
||||||
|
int retval = memcmp(address_.addr, other.address_.addr, address_.len); |
||||||
|
if (retval != 0) return retval; |
||||||
|
return grpc_channel_args_compare(args_, other.args_); |
||||||
|
} |
||||||
|
|
||||||
|
bool ServerAddress::IsBalancer() const { |
||||||
|
return grpc_channel_arg_get_bool( |
||||||
|
grpc_channel_args_find(args_, GRPC_ARG_ADDRESS_IS_BALANCER), false); |
||||||
|
} |
||||||
|
|
||||||
|
//
|
||||||
|
// ServerAddressList
|
||||||
|
//
|
||||||
|
|
||||||
|
namespace { |
||||||
|
|
||||||
|
void* ServerAddressListCopy(void* addresses) { |
||||||
|
ServerAddressList* a = static_cast<ServerAddressList*>(addresses); |
||||||
|
return New<ServerAddressList>(*a); |
||||||
|
} |
||||||
|
|
||||||
|
void ServerAddressListDestroy(void* addresses) { |
||||||
|
ServerAddressList* a = static_cast<ServerAddressList*>(addresses); |
||||||
|
Delete(a); |
||||||
|
} |
||||||
|
|
||||||
|
int ServerAddressListCompare(void* addresses1, void* addresses2) { |
||||||
|
ServerAddressList* a1 = static_cast<ServerAddressList*>(addresses1); |
||||||
|
ServerAddressList* a2 = static_cast<ServerAddressList*>(addresses2); |
||||||
|
if (a1->size() > a2->size()) return 1; |
||||||
|
if (a1->size() < a2->size()) return -1; |
||||||
|
for (size_t i = 0; i < a1->size(); ++i) { |
||||||
|
int retval = (*a1)[i].Cmp((*a2)[i]); |
||||||
|
if (retval != 0) return retval; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
const grpc_arg_pointer_vtable server_addresses_arg_vtable = { |
||||||
|
ServerAddressListCopy, ServerAddressListDestroy, ServerAddressListCompare}; |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
grpc_arg CreateServerAddressListChannelArg(const ServerAddressList* addresses) { |
||||||
|
return grpc_channel_arg_pointer_create( |
||||||
|
const_cast<char*>(GRPC_ARG_SERVER_ADDRESS_LIST), |
||||||
|
const_cast<ServerAddressList*>(addresses), &server_addresses_arg_vtable); |
||||||
|
} |
||||||
|
|
||||||
|
ServerAddressList* FindServerAddressListChannelArg( |
||||||
|
const grpc_channel_args* channel_args) { |
||||||
|
const grpc_arg* lb_addresses_arg = |
||||||
|
grpc_channel_args_find(channel_args, GRPC_ARG_SERVER_ADDRESS_LIST); |
||||||
|
if (lb_addresses_arg == nullptr || lb_addresses_arg->type != GRPC_ARG_POINTER) |
||||||
|
return nullptr; |
||||||
|
return static_cast<ServerAddressList*>(lb_addresses_arg->value.pointer.p); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
@ -0,0 +1,108 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVER_ADDRESS_H |
||||||
|
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVER_ADDRESS_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/lib/channel/channel_args.h" |
||||||
|
#include "src/core/lib/gprpp/inlined_vector.h" |
||||||
|
#include "src/core/lib/iomgr/resolve_address.h" |
||||||
|
#include "src/core/lib/uri/uri_parser.h" |
||||||
|
|
||||||
|
// Channel arg key for ServerAddressList.
|
||||||
|
#define GRPC_ARG_SERVER_ADDRESS_LIST "grpc.server_address_list" |
||||||
|
|
||||||
|
// Channel arg key for a bool indicating whether an address is a grpclb
|
||||||
|
// load balancer (as opposed to a backend).
|
||||||
|
#define GRPC_ARG_ADDRESS_IS_BALANCER "grpc.address_is_balancer" |
||||||
|
|
||||||
|
// Channel arg key for a string indicating an address's balancer name.
|
||||||
|
#define GRPC_ARG_ADDRESS_BALANCER_NAME "grpc.address_balancer_name" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
//
|
||||||
|
// ServerAddress
|
||||||
|
//
|
||||||
|
|
||||||
|
// A server address is a grpc_resolved_address with an associated set of
|
||||||
|
// channel args. Any args present here will be merged into the channel
|
||||||
|
// args when a subchannel is created for this address.
|
||||||
|
class ServerAddress { |
||||||
|
public: |
||||||
|
// Takes ownership of args.
|
||||||
|
ServerAddress(const grpc_resolved_address& address, grpc_channel_args* args); |
||||||
|
ServerAddress(const void* address, size_t address_len, |
||||||
|
grpc_channel_args* args); |
||||||
|
|
||||||
|
~ServerAddress() { grpc_channel_args_destroy(args_); } |
||||||
|
|
||||||
|
// Copyable.
|
||||||
|
ServerAddress(const ServerAddress& other) |
||||||
|
: address_(other.address_), args_(grpc_channel_args_copy(other.args_)) {} |
||||||
|
ServerAddress& operator=(const ServerAddress& other) { |
||||||
|
address_ = other.address_; |
||||||
|
grpc_channel_args_destroy(args_); |
||||||
|
args_ = grpc_channel_args_copy(other.args_); |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
// Movable.
|
||||||
|
ServerAddress(ServerAddress&& other) |
||||||
|
: address_(other.address_), args_(other.args_) { |
||||||
|
other.args_ = nullptr; |
||||||
|
} |
||||||
|
ServerAddress& operator=(ServerAddress&& other) { |
||||||
|
address_ = other.address_; |
||||||
|
args_ = other.args_; |
||||||
|
other.args_ = nullptr; |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
bool operator==(const ServerAddress& other) const { return Cmp(other) == 0; } |
||||||
|
|
||||||
|
int Cmp(const ServerAddress& other) const; |
||||||
|
|
||||||
|
const grpc_resolved_address& address() const { return address_; } |
||||||
|
const grpc_channel_args* args() const { return args_; } |
||||||
|
|
||||||
|
bool IsBalancer() const; |
||||||
|
|
||||||
|
private: |
||||||
|
grpc_resolved_address address_; |
||||||
|
grpc_channel_args* args_; |
||||||
|
}; |
||||||
|
|
||||||
|
//
|
||||||
|
// ServerAddressList
|
||||||
|
//
|
||||||
|
|
||||||
|
typedef InlinedVector<ServerAddress, 1> ServerAddressList; |
||||||
|
|
||||||
|
// Returns a channel arg containing \a addresses.
|
||||||
|
grpc_arg CreateServerAddressListChannelArg(const ServerAddressList* addresses); |
||||||
|
|
||||||
|
// Returns the ServerListAddress instance in channel_args or NULL.
|
||||||
|
ServerAddressList* FindServerAddressListChannelArg( |
||||||
|
const grpc_channel_args* channel_args); |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_SERVER_ADDRESS_H */ |
||||||
Loading…
Reference in new issue