David Garcia Quintas
7 years ago
parent
d043fa06bc
commit
fa460dcf1d
17 changed files with 121 additions and 24 deletions
@ -0,0 +1,42 @@ |
||||
/*
|
||||
* |
||||
* 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/authority.h" |
||||
|
||||
grpc_channel_args* grpc_add_default_authority_if_not_present( |
||||
const grpc_channel_args* args) { |
||||
const bool has_default_authority = |
||||
grpc_channel_args_find(args, GRPC_ARG_DEFAULT_AUTHORITY) != nullptr; |
||||
grpc_arg new_args[1]; |
||||
size_t num_new_args = 0; |
||||
grpc_core::UniquePtr<char> default_authority; |
||||
if (!has_default_authority) { |
||||
const grpc_arg* server_uri_arg = |
||||
grpc_channel_args_find(args, GRPC_ARG_SERVER_URI); |
||||
const char* server_uri_str = grpc_channel_arg_get_string(server_uri_arg); |
||||
GPR_ASSERT(server_uri_str != nullptr); |
||||
default_authority = |
||||
grpc_core::ResolverRegistry::GetDefaultAuthority(server_uri_str); |
||||
GPR_ASSERT(default_authority != nullptr); |
||||
new_args[num_new_args++] = grpc_channel_arg_string_create( |
||||
const_cast<char*>(GRPC_ARG_DEFAULT_AUTHORITY), default_authority.get()); |
||||
} |
||||
return grpc_channel_args_copy_and_add(args, new_args, num_new_args); |
||||
} |
||||
@ -0,0 +1,36 @@ |
||||
/*
|
||||
* |
||||
* 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_AUTHORITY_H |
||||
#define GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_AUTHORITY_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include <grpc/grpc.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/client_channel.h" |
||||
#include "src/core/ext/filters/client_channel/resolver_registry.h" |
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/gprpp/memory.h" |
||||
|
||||
/// Returns a copy of \a args with the default authority channel arg set if it
|
||||
/// wasn't already present.
|
||||
grpc_channel_args* grpc_add_default_authority_if_not_present( |
||||
const grpc_channel_args* args); |
||||
|
||||
#endif /* GRPC_CORE_EXT_FILTERS_CLIENT_CHANNEL_AUTHORITY_H */ |
||||
Loading…
Reference in new issue