| Index: third_party/grpc/src/cpp/common/channel_arguments.cc
|
| diff --git a/third_party/grpc/src/cpp/common/channel_arguments.cc b/third_party/grpc/src/cpp/common/channel_arguments.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..d7faa5e173d6a63e4711fa3ff02ff82521ecb20e
|
| --- /dev/null
|
| +++ b/third_party/grpc/src/cpp/common/channel_arguments.cc
|
| @@ -0,0 +1,158 @@
|
| +/*
|
| + *
|
| + * Copyright 2015-2016, Google Inc.
|
| + * All rights reserved.
|
| + *
|
| + * Redistribution and use in source and binary forms, with or without
|
| + * modification, are permitted provided that the following conditions are
|
| + * met:
|
| + *
|
| + * * Redistributions of source code must retain the above copyright
|
| + * notice, this list of conditions and the following disclaimer.
|
| + * * Redistributions in binary form must reproduce the above
|
| + * copyright notice, this list of conditions and the following disclaimer
|
| + * in the documentation and/or other materials provided with the
|
| + * distribution.
|
| + * * Neither the name of Google Inc. nor the names of its
|
| + * contributors may be used to endorse or promote products derived from
|
| + * this software without specific prior written permission.
|
| + *
|
| + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| + *
|
| + */
|
| +#include <grpc++/support/channel_arguments.h>
|
| +
|
| +#include <sstream>
|
| +
|
| +#include <grpc/impl/codegen/grpc_types.h>
|
| +#include <grpc/support/log.h>
|
| +#include "src/core/channel/channel_args.h"
|
| +
|
| +namespace grpc {
|
| +
|
| +ChannelArguments::ChannelArguments() {
|
| + std::ostringstream user_agent_prefix;
|
| + user_agent_prefix << "grpc-c++/" << grpc_version_string();
|
| + // This will be ignored if used on the server side.
|
| + SetString(GRPC_ARG_PRIMARY_USER_AGENT_STRING, user_agent_prefix.str());
|
| +}
|
| +
|
| +ChannelArguments::ChannelArguments(const ChannelArguments& other)
|
| + : strings_(other.strings_) {
|
| + args_.reserve(other.args_.size());
|
| + auto list_it_dst = strings_.begin();
|
| + auto list_it_src = other.strings_.begin();
|
| + for (auto a = other.args_.begin(); a != other.args_.end(); ++a) {
|
| + grpc_arg ap;
|
| + ap.type = a->type;
|
| + GPR_ASSERT(list_it_src->c_str() == a->key);
|
| + ap.key = const_cast<char*>(list_it_dst->c_str());
|
| + ++list_it_src;
|
| + ++list_it_dst;
|
| + switch (a->type) {
|
| + case GRPC_ARG_INTEGER:
|
| + ap.value.integer = a->value.integer;
|
| + break;
|
| + case GRPC_ARG_STRING:
|
| + GPR_ASSERT(list_it_src->c_str() == a->value.string);
|
| + ap.value.string = const_cast<char*>(list_it_dst->c_str());
|
| + ++list_it_src;
|
| + ++list_it_dst;
|
| + break;
|
| + case GRPC_ARG_POINTER:
|
| + ap.value.pointer = a->value.pointer;
|
| + ap.value.pointer.p = a->value.pointer.vtable->copy(ap.value.pointer.p);
|
| + break;
|
| + }
|
| + args_.push_back(ap);
|
| + }
|
| +}
|
| +
|
| +void ChannelArguments::Swap(ChannelArguments& other) {
|
| + args_.swap(other.args_);
|
| + strings_.swap(other.strings_);
|
| +}
|
| +
|
| +void ChannelArguments::SetCompressionAlgorithm(
|
| + grpc_compression_algorithm algorithm) {
|
| + SetInt(GRPC_COMPRESSION_ALGORITHM_ARG, algorithm);
|
| +}
|
| +
|
| +// Note: a second call to this will add in front the result of the first call.
|
| +// An example is calling this on a copy of ChannelArguments which already has a
|
| +// prefix. The user can build up a prefix string by calling this multiple times,
|
| +// each with more significant identifier.
|
| +void ChannelArguments::SetUserAgentPrefix(
|
| + const grpc::string& user_agent_prefix) {
|
| + if (user_agent_prefix.empty()) {
|
| + return;
|
| + }
|
| + bool replaced = false;
|
| + for (auto it = args_.begin(); it != args_.end(); ++it) {
|
| + const grpc_arg& arg = *it;
|
| + if (arg.type == GRPC_ARG_STRING &&
|
| + grpc::string(arg.key) == GRPC_ARG_PRIMARY_USER_AGENT_STRING) {
|
| + strings_.push_back(user_agent_prefix + " " + arg.value.string);
|
| + it->value.string = const_cast<char*>(strings_.back().c_str());
|
| + replaced = true;
|
| + break;
|
| + }
|
| + }
|
| + if (!replaced) {
|
| + SetString(GRPC_ARG_PRIMARY_USER_AGENT_STRING, user_agent_prefix);
|
| + }
|
| +}
|
| +
|
| +void ChannelArguments::SetInt(const grpc::string& key, int value) {
|
| + grpc_arg arg;
|
| + arg.type = GRPC_ARG_INTEGER;
|
| + strings_.push_back(key);
|
| + arg.key = const_cast<char*>(strings_.back().c_str());
|
| + arg.value.integer = value;
|
| +
|
| + args_.push_back(arg);
|
| +}
|
| +
|
| +void ChannelArguments::SetPointer(const grpc::string& key, void* value) {
|
| + static const grpc_arg_pointer_vtable vtable = {
|
| + &PointerVtableMembers::Copy, &PointerVtableMembers::Destroy,
|
| + &PointerVtableMembers::Compare};
|
| + grpc_arg arg;
|
| + arg.type = GRPC_ARG_POINTER;
|
| + strings_.push_back(key);
|
| + arg.key = const_cast<char*>(strings_.back().c_str());
|
| + arg.value.pointer.p = value;
|
| + arg.value.pointer.vtable = &vtable;
|
| + args_.push_back(arg);
|
| +}
|
| +
|
| +void ChannelArguments::SetString(const grpc::string& key,
|
| + const grpc::string& value) {
|
| + grpc_arg arg;
|
| + arg.type = GRPC_ARG_STRING;
|
| + strings_.push_back(key);
|
| + arg.key = const_cast<char*>(strings_.back().c_str());
|
| + strings_.push_back(value);
|
| + arg.value.string = const_cast<char*>(strings_.back().c_str());
|
| +
|
| + args_.push_back(arg);
|
| +}
|
| +
|
| +void ChannelArguments::SetChannelArgs(grpc_channel_args* channel_args) const {
|
| + channel_args->num_args = args_.size();
|
| + if (channel_args->num_args > 0) {
|
| + channel_args->args = const_cast<grpc_arg*>(&args_[0]);
|
| + }
|
| +}
|
| +
|
| +} // namespace grpc
|
|
|