mirror of https://github.com/grpc/grpc.git
Expose channel stack type to builder (#29088)
* split builder * expose channel stack type to builder * Automated change: Fix sanity tests * Update channel_stack_builder_impl.h * Update channel_init.h Co-authored-by: ctiller <ctiller@users.noreply.github.com>pull/29118/head
Craig Tiller
3 years ago
committed by
GitHub
parent
28f1453ad5
commit
0ea1eeb4e7
31 changed files with 257 additions and 99 deletions
@ -0,0 +1,102 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016 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/lib/channel/channel_stack_builder_impl.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/lib/channel/channel_stack.h" |
||||
#include "src/core/lib/gprpp/memory.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
grpc_error_handle ChannelStackBuilderImpl::Build(size_t prefix_bytes, |
||||
int initial_refs, |
||||
grpc_iomgr_cb_func destroy, |
||||
void* destroy_arg, |
||||
void** result) { |
||||
auto* stack = mutable_stack(); |
||||
|
||||
// create an array of filters
|
||||
std::vector<const grpc_channel_filter*> filters; |
||||
filters.reserve(stack->size()); |
||||
for (const auto& elem : *stack) { |
||||
filters.push_back(elem.filter); |
||||
} |
||||
|
||||
// calculate the size of the channel stack
|
||||
size_t channel_stack_size = |
||||
grpc_channel_stack_size(filters.data(), filters.size()); |
||||
|
||||
// allocate memory, with prefix_bytes followed by channel_stack_size
|
||||
*result = gpr_zalloc(prefix_bytes + channel_stack_size); |
||||
// fetch a pointer to the channel stack
|
||||
grpc_channel_stack* channel_stack = reinterpret_cast<grpc_channel_stack*>( |
||||
static_cast<char*>(*result) + prefix_bytes); |
||||
|
||||
const grpc_channel_args* final_args; |
||||
if (transport() != nullptr) { |
||||
static const grpc_arg_pointer_vtable vtable = { |
||||
// copy
|
||||
[](void* p) { return p; }, |
||||
// destroy
|
||||
[](void*) {}, |
||||
// cmp
|
||||
[](void* a, void* b) { return QsortCompare(a, b); }, |
||||
}; |
||||
grpc_arg arg = grpc_channel_arg_pointer_create( |
||||
const_cast<char*>(GRPC_ARG_TRANSPORT), transport(), &vtable); |
||||
final_args = grpc_channel_args_copy_and_add(channel_args(), &arg, 1); |
||||
} else { |
||||
final_args = channel_args(); |
||||
} |
||||
|
||||
// and initialize it
|
||||
grpc_error_handle error = grpc_channel_stack_init( |
||||
initial_refs, destroy, destroy_arg == nullptr ? *result : destroy_arg, |
||||
filters.data(), filters.size(), final_args, name(), channel_stack); |
||||
|
||||
if (final_args != channel_args()) { |
||||
grpc_channel_args_destroy(final_args); |
||||
} |
||||
|
||||
if (error != GRPC_ERROR_NONE) { |
||||
grpc_channel_stack_destroy(channel_stack); |
||||
gpr_free(*result); |
||||
*result = nullptr; |
||||
return error; |
||||
} |
||||
|
||||
// run post-initialization functions
|
||||
for (size_t i = 0; i < filters.size(); i++) { |
||||
if ((*stack)[i].post_init != nullptr) { |
||||
(*stack)[i].post_init(channel_stack, |
||||
grpc_channel_stack_element(channel_stack, i)); |
||||
} |
||||
} |
||||
|
||||
return GRPC_ERROR_NONE; |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,48 @@ |
||||
// Copyright 2016 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_LIB_CHANNEL_CHANNEL_STACK_BUILDER_IMPL_H |
||||
#define GRPC_CORE_LIB_CHANNEL_CHANNEL_STACK_BUILDER_IMPL_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include <stdbool.h> |
||||
|
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/channel/channel_stack.h" |
||||
#include "src/core/lib/channel/channel_stack_builder.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
// Build a channel stack.
|
||||
// Allows interested parties to add filters to the stack, and to query an
|
||||
// in-progress build.
|
||||
// Carries some useful context for the channel stack, such as a target string
|
||||
// and a transport.
|
||||
class ChannelStackBuilderImpl final : public ChannelStackBuilder { |
||||
public: |
||||
using ChannelStackBuilder::ChannelStackBuilder; |
||||
|
||||
// Build the channel stack.
|
||||
// After success, *result holds the new channel stack,
|
||||
// prefix_bytes are allocated before the channel stack,
|
||||
// initial_refs, destroy, destroy_arg are as per grpc_channel_stack_init
|
||||
// On failure, *result is nullptr.
|
||||
grpc_error_handle Build(size_t prefix_bytes, int initial_refs, |
||||
grpc_iomgr_cb_func destroy, void* destroy_arg, |
||||
void** result) override; |
||||
}; |
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_CORE_LIB_CHANNEL_CHANNEL_STACK_BUILDER_IMPL_H
|
||||
Loading…
Reference in new issue