mirror of https://github.com/grpc/grpc.git
Craig Tiller
8 years ago
committed by
GitHub
commit
eeb3b24f7e
9 changed files with 748 additions and 2 deletions
@ -0,0 +1,587 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
/* Microbenchmarks around CHTTP2 transport operations */ |
||||||
|
|
||||||
|
#include <grpc++/support/channel_arguments.h> |
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include <grpc/support/string_util.h> |
||||||
|
#include <string.h> |
||||||
|
#include <memory> |
||||||
|
#include <queue> |
||||||
|
#include <sstream> |
||||||
|
extern "C" { |
||||||
|
#include "src/core/ext/transport/chttp2/transport/chttp2_transport.h" |
||||||
|
#include "src/core/ext/transport/chttp2/transport/internal.h" |
||||||
|
#include "src/core/lib/iomgr/resource_quota.h" |
||||||
|
#include "src/core/lib/slice/slice_internal.h" |
||||||
|
#include "src/core/lib/transport/static_metadata.h" |
||||||
|
} |
||||||
|
#include "test/cpp/microbenchmarks/helpers.h" |
||||||
|
#include "third_party/benchmark/include/benchmark/benchmark.h" |
||||||
|
|
||||||
|
auto &force_library_initialization = Library::get(); |
||||||
|
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
// Helper classes
|
||||||
|
//
|
||||||
|
|
||||||
|
class DummyEndpoint : public grpc_endpoint { |
||||||
|
public: |
||||||
|
DummyEndpoint() { |
||||||
|
static const grpc_endpoint_vtable my_vtable = {read, |
||||||
|
write, |
||||||
|
get_workqueue, |
||||||
|
add_to_pollset, |
||||||
|
add_to_pollset_set, |
||||||
|
shutdown, |
||||||
|
destroy, |
||||||
|
get_resource_user, |
||||||
|
get_peer, |
||||||
|
get_fd}; |
||||||
|
grpc_endpoint::vtable = &my_vtable; |
||||||
|
ru_ = grpc_resource_user_create(Library::get().rq(), "dummy_endpoint"); |
||||||
|
} |
||||||
|
|
||||||
|
void PushInput(grpc_exec_ctx *exec_ctx, grpc_slice slice) { |
||||||
|
if (read_cb_ == nullptr) { |
||||||
|
GPR_ASSERT(!have_slice_); |
||||||
|
buffered_slice_ = slice; |
||||||
|
have_slice_ = true; |
||||||
|
return; |
||||||
|
} |
||||||
|
grpc_slice_buffer_add(slices_, slice); |
||||||
|
grpc_closure_sched(exec_ctx, read_cb_, GRPC_ERROR_NONE); |
||||||
|
read_cb_ = nullptr; |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
grpc_resource_user *ru_; |
||||||
|
grpc_closure *read_cb_ = nullptr; |
||||||
|
grpc_slice_buffer *slices_ = nullptr; |
||||||
|
bool have_slice_ = false; |
||||||
|
grpc_slice buffered_slice_; |
||||||
|
|
||||||
|
void QueueRead(grpc_exec_ctx *exec_ctx, grpc_slice_buffer *slices, |
||||||
|
grpc_closure *cb) { |
||||||
|
GPR_ASSERT(read_cb_ == nullptr); |
||||||
|
if (have_slice_) { |
||||||
|
have_slice_ = false; |
||||||
|
grpc_slice_buffer_add(slices, buffered_slice_); |
||||||
|
grpc_closure_sched(exec_ctx, cb, GRPC_ERROR_NONE); |
||||||
|
return; |
||||||
|
} |
||||||
|
read_cb_ = cb; |
||||||
|
slices_ = slices; |
||||||
|
} |
||||||
|
|
||||||
|
static void read(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep, |
||||||
|
grpc_slice_buffer *slices, grpc_closure *cb) { |
||||||
|
static_cast<DummyEndpoint *>(ep)->QueueRead(exec_ctx, slices, cb); |
||||||
|
} |
||||||
|
|
||||||
|
static void write(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep, |
||||||
|
grpc_slice_buffer *slices, grpc_closure *cb) { |
||||||
|
grpc_closure_sched(exec_ctx, cb, GRPC_ERROR_NONE); |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_workqueue *get_workqueue(grpc_endpoint *ep) { return NULL; } |
||||||
|
|
||||||
|
static void add_to_pollset(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep, |
||||||
|
grpc_pollset *pollset) {} |
||||||
|
|
||||||
|
static void add_to_pollset_set(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep, |
||||||
|
grpc_pollset_set *pollset) {} |
||||||
|
|
||||||
|
static void shutdown(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep, |
||||||
|
grpc_error *why) { |
||||||
|
grpc_resource_user_shutdown(exec_ctx, |
||||||
|
static_cast<DummyEndpoint *>(ep)->ru_); |
||||||
|
grpc_closure_sched(exec_ctx, static_cast<DummyEndpoint *>(ep)->read_cb_, |
||||||
|
why); |
||||||
|
} |
||||||
|
|
||||||
|
static void destroy(grpc_exec_ctx *exec_ctx, grpc_endpoint *ep) { |
||||||
|
grpc_resource_user_unref(exec_ctx, static_cast<DummyEndpoint *>(ep)->ru_); |
||||||
|
delete static_cast<DummyEndpoint *>(ep); |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_resource_user *get_resource_user(grpc_endpoint *ep) { |
||||||
|
return static_cast<DummyEndpoint *>(ep)->ru_; |
||||||
|
} |
||||||
|
static char *get_peer(grpc_endpoint *ep) { return gpr_strdup("test"); } |
||||||
|
static int get_fd(grpc_endpoint *ep) { return 0; } |
||||||
|
}; |
||||||
|
|
||||||
|
class Fixture { |
||||||
|
public: |
||||||
|
Fixture(const grpc::ChannelArguments &args, bool client) { |
||||||
|
grpc_channel_args c_args = args.c_channel_args(); |
||||||
|
ep_ = new DummyEndpoint; |
||||||
|
t_ = grpc_create_chttp2_transport(exec_ctx(), &c_args, ep_, client); |
||||||
|
grpc_chttp2_transport_start_reading(exec_ctx(), t_, NULL); |
||||||
|
FlushExecCtx(); |
||||||
|
} |
||||||
|
|
||||||
|
void FlushExecCtx() { grpc_exec_ctx_flush(&exec_ctx_); } |
||||||
|
|
||||||
|
~Fixture() { |
||||||
|
grpc_transport_destroy(&exec_ctx_, t_); |
||||||
|
grpc_exec_ctx_finish(&exec_ctx_); |
||||||
|
} |
||||||
|
|
||||||
|
grpc_chttp2_transport *chttp2_transport() { |
||||||
|
return reinterpret_cast<grpc_chttp2_transport *>(t_); |
||||||
|
} |
||||||
|
grpc_transport *transport() { return t_; } |
||||||
|
grpc_exec_ctx *exec_ctx() { return &exec_ctx_; } |
||||||
|
|
||||||
|
void PushInput(grpc_slice slice) { ep_->PushInput(exec_ctx(), slice); } |
||||||
|
|
||||||
|
private: |
||||||
|
DummyEndpoint *ep_; |
||||||
|
grpc_exec_ctx exec_ctx_ = GRPC_EXEC_CTX_INIT; |
||||||
|
grpc_transport *t_; |
||||||
|
}; |
||||||
|
|
||||||
|
static void DoNothing(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) {} |
||||||
|
|
||||||
|
class Stream { |
||||||
|
public: |
||||||
|
Stream(Fixture *f) : f_(f) { |
||||||
|
GRPC_STREAM_REF_INIT(&refcount_, 1, DoNothing, nullptr, "test_stream"); |
||||||
|
stream_size_ = grpc_transport_stream_size(f->transport()); |
||||||
|
stream_ = gpr_malloc(stream_size_); |
||||||
|
arena_ = gpr_arena_create(4096); |
||||||
|
} |
||||||
|
|
||||||
|
~Stream() { |
||||||
|
gpr_free(stream_); |
||||||
|
gpr_arena_destroy(arena_); |
||||||
|
} |
||||||
|
|
||||||
|
void Init(benchmark::State &state) { |
||||||
|
memset(stream_, 0, stream_size_); |
||||||
|
if ((state.iterations() & 0xffff) == 0) { |
||||||
|
gpr_arena_destroy(arena_); |
||||||
|
arena_ = gpr_arena_create(4096); |
||||||
|
} |
||||||
|
grpc_transport_init_stream(f_->exec_ctx(), f_->transport(), |
||||||
|
static_cast<grpc_stream *>(stream_), &refcount_, |
||||||
|
NULL, arena_); |
||||||
|
} |
||||||
|
|
||||||
|
void DestroyThen(grpc_closure *closure) { |
||||||
|
grpc_transport_destroy_stream(f_->exec_ctx(), f_->transport(), |
||||||
|
static_cast<grpc_stream *>(stream_), closure); |
||||||
|
} |
||||||
|
|
||||||
|
void Op(grpc_transport_stream_op *op) { |
||||||
|
grpc_transport_perform_stream_op(f_->exec_ctx(), f_->transport(), |
||||||
|
static_cast<grpc_stream *>(stream_), op); |
||||||
|
} |
||||||
|
|
||||||
|
grpc_chttp2_stream *chttp2_stream() { |
||||||
|
return static_cast<grpc_chttp2_stream *>(stream_); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
Fixture *f_; |
||||||
|
grpc_stream_refcount refcount_; |
||||||
|
gpr_arena *arena_; |
||||||
|
size_t stream_size_; |
||||||
|
void *stream_; |
||||||
|
}; |
||||||
|
|
||||||
|
class Closure : public grpc_closure { |
||||||
|
public: |
||||||
|
virtual ~Closure() {} |
||||||
|
}; |
||||||
|
|
||||||
|
template <class F> |
||||||
|
std::unique_ptr<Closure> MakeClosure( |
||||||
|
F f, grpc_closure_scheduler *sched = grpc_schedule_on_exec_ctx) { |
||||||
|
struct C : public Closure { |
||||||
|
C(const F &f, grpc_closure_scheduler *sched) : f_(f) { |
||||||
|
grpc_closure_init(this, Execute, this, sched); |
||||||
|
} |
||||||
|
F f_; |
||||||
|
static void Execute(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
||||||
|
static_cast<C *>(arg)->f_(exec_ctx, error); |
||||||
|
} |
||||||
|
}; |
||||||
|
return std::unique_ptr<Closure>(new C(f, sched)); |
||||||
|
} |
||||||
|
|
||||||
|
template <class F> |
||||||
|
grpc_closure *MakeOnceClosure( |
||||||
|
F f, grpc_closure_scheduler *sched = grpc_schedule_on_exec_ctx) { |
||||||
|
struct C : public grpc_closure { |
||||||
|
C(const F &f) : f_(f) {} |
||||||
|
F f_; |
||||||
|
static void Execute(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
||||||
|
static_cast<C *>(arg)->f_(exec_ctx, error); |
||||||
|
delete static_cast<C *>(arg); |
||||||
|
} |
||||||
|
}; |
||||||
|
auto *c = new C{f}; |
||||||
|
return grpc_closure_init(c, C::Execute, c, sched); |
||||||
|
} |
||||||
|
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
// Benchmarks
|
||||||
|
//
|
||||||
|
|
||||||
|
static void BM_StreamCreateDestroy(benchmark::State &state) { |
||||||
|
TrackCounters track_counters; |
||||||
|
Fixture f(grpc::ChannelArguments(), true); |
||||||
|
Stream s(&f); |
||||||
|
std::unique_ptr<Closure> next = |
||||||
|
MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (!state.KeepRunning()) return; |
||||||
|
s.Init(state); |
||||||
|
s.DestroyThen(next.get()); |
||||||
|
}); |
||||||
|
grpc_closure_run(f.exec_ctx(), next.get(), GRPC_ERROR_NONE); |
||||||
|
f.FlushExecCtx(); |
||||||
|
track_counters.Finish(state); |
||||||
|
} |
||||||
|
BENCHMARK(BM_StreamCreateDestroy); |
||||||
|
|
||||||
|
class RepresentativeClientInitialMetadata { |
||||||
|
public: |
||||||
|
static std::vector<grpc_mdelem> GetElems(grpc_exec_ctx *exec_ctx) { |
||||||
|
return { |
||||||
|
GRPC_MDELEM_SCHEME_HTTP, GRPC_MDELEM_METHOD_POST, |
||||||
|
grpc_mdelem_from_slices( |
||||||
|
exec_ctx, GRPC_MDSTR_PATH, |
||||||
|
grpc_slice_intern(grpc_slice_from_static_string("/foo/bar"))), |
||||||
|
grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_AUTHORITY, |
||||||
|
grpc_slice_intern(grpc_slice_from_static_string( |
||||||
|
"foo.test.google.fr:1234"))), |
||||||
|
GRPC_MDELEM_GRPC_ACCEPT_ENCODING_IDENTITY_COMMA_DEFLATE_COMMA_GZIP, |
||||||
|
GRPC_MDELEM_TE_TRAILERS, |
||||||
|
GRPC_MDELEM_CONTENT_TYPE_APPLICATION_SLASH_GRPC, |
||||||
|
grpc_mdelem_from_slices( |
||||||
|
exec_ctx, GRPC_MDSTR_USER_AGENT, |
||||||
|
grpc_slice_intern(grpc_slice_from_static_string( |
||||||
|
"grpc-c/3.0.0-dev (linux; chttp2; green)")))}; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
template <class Metadata> |
||||||
|
static void BM_StreamCreateSendInitialMetadataDestroy(benchmark::State &state) { |
||||||
|
TrackCounters track_counters; |
||||||
|
Fixture f(grpc::ChannelArguments(), true); |
||||||
|
Stream s(&f); |
||||||
|
grpc_transport_stream_op op; |
||||||
|
std::unique_ptr<Closure> start; |
||||||
|
std::unique_ptr<Closure> done; |
||||||
|
|
||||||
|
grpc_metadata_batch b; |
||||||
|
grpc_metadata_batch_init(&b); |
||||||
|
b.deadline = gpr_inf_future(GPR_CLOCK_MONOTONIC); |
||||||
|
std::vector<grpc_mdelem> elems = Metadata::GetElems(f.exec_ctx()); |
||||||
|
std::vector<grpc_linked_mdelem> storage(elems.size()); |
||||||
|
for (size_t i = 0; i < elems.size(); i++) { |
||||||
|
GPR_ASSERT(GRPC_LOG_IF_ERROR( |
||||||
|
"addmd", |
||||||
|
grpc_metadata_batch_add_tail(f.exec_ctx(), &b, &storage[i], elems[i]))); |
||||||
|
} |
||||||
|
|
||||||
|
f.FlushExecCtx(); |
||||||
|
start = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (!state.KeepRunning()) return; |
||||||
|
s.Init(state); |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.on_complete = done.get(); |
||||||
|
op.send_initial_metadata = &b; |
||||||
|
s.Op(&op); |
||||||
|
}); |
||||||
|
done = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.cancel_error = GRPC_ERROR_CANCELLED; |
||||||
|
s.Op(&op); |
||||||
|
s.DestroyThen(start.get()); |
||||||
|
}); |
||||||
|
grpc_closure_sched(f.exec_ctx(), start.get(), GRPC_ERROR_NONE); |
||||||
|
f.FlushExecCtx(); |
||||||
|
grpc_metadata_batch_destroy(f.exec_ctx(), &b); |
||||||
|
track_counters.Finish(state); |
||||||
|
} |
||||||
|
BENCHMARK_TEMPLATE(BM_StreamCreateSendInitialMetadataDestroy, |
||||||
|
RepresentativeClientInitialMetadata); |
||||||
|
|
||||||
|
static void BM_TransportEmptyOp(benchmark::State &state) { |
||||||
|
TrackCounters track_counters; |
||||||
|
Fixture f(grpc::ChannelArguments(), true); |
||||||
|
Stream s(&f); |
||||||
|
s.Init(state); |
||||||
|
grpc_transport_stream_op op; |
||||||
|
std::unique_ptr<Closure> c = |
||||||
|
MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (!state.KeepRunning()) return; |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.on_complete = c.get(); |
||||||
|
s.Op(&op); |
||||||
|
}); |
||||||
|
grpc_closure_sched(f.exec_ctx(), c.get(), GRPC_ERROR_NONE); |
||||||
|
f.FlushExecCtx(); |
||||||
|
s.DestroyThen( |
||||||
|
MakeOnceClosure([](grpc_exec_ctx *exec_ctx, grpc_error *error) {})); |
||||||
|
f.FlushExecCtx(); |
||||||
|
track_counters.Finish(state); |
||||||
|
} |
||||||
|
BENCHMARK(BM_TransportEmptyOp); |
||||||
|
|
||||||
|
static void BM_TransportStreamSend(benchmark::State &state) { |
||||||
|
TrackCounters track_counters; |
||||||
|
Fixture f(grpc::ChannelArguments(), true); |
||||||
|
Stream s(&f); |
||||||
|
s.Init(state); |
||||||
|
grpc_transport_stream_op op; |
||||||
|
grpc_slice_buffer_stream send_stream; |
||||||
|
grpc_slice_buffer send_buffer; |
||||||
|
grpc_slice_buffer_init(&send_buffer); |
||||||
|
grpc_slice_buffer_add(&send_buffer, gpr_slice_malloc(state.range(0))); |
||||||
|
memset(GRPC_SLICE_START_PTR(send_buffer.slices[0]), 0, |
||||||
|
GRPC_SLICE_LENGTH(send_buffer.slices[0])); |
||||||
|
|
||||||
|
grpc_metadata_batch b; |
||||||
|
grpc_metadata_batch_init(&b); |
||||||
|
b.deadline = gpr_inf_future(GPR_CLOCK_MONOTONIC); |
||||||
|
std::vector<grpc_mdelem> elems = |
||||||
|
RepresentativeClientInitialMetadata::GetElems(f.exec_ctx()); |
||||||
|
std::vector<grpc_linked_mdelem> storage(elems.size()); |
||||||
|
for (size_t i = 0; i < elems.size(); i++) { |
||||||
|
GPR_ASSERT(GRPC_LOG_IF_ERROR( |
||||||
|
"addmd", |
||||||
|
grpc_metadata_batch_add_tail(f.exec_ctx(), &b, &storage[i], elems[i]))); |
||||||
|
} |
||||||
|
|
||||||
|
std::unique_ptr<Closure> c = |
||||||
|
MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (!state.KeepRunning()) return; |
||||||
|
// force outgoing window to be yuge
|
||||||
|
s.chttp2_stream()->outgoing_window_delta = 1024 * 1024 * 1024; |
||||||
|
f.chttp2_transport()->outgoing_window = 1024 * 1024 * 1024; |
||||||
|
grpc_slice_buffer_stream_init(&send_stream, &send_buffer, 0); |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.on_complete = c.get(); |
||||||
|
op.send_message = &send_stream.base; |
||||||
|
s.Op(&op); |
||||||
|
}); |
||||||
|
|
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.send_initial_metadata = &b; |
||||||
|
op.on_complete = c.get(); |
||||||
|
s.Op(&op); |
||||||
|
|
||||||
|
f.FlushExecCtx(); |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.cancel_error = GRPC_ERROR_CANCELLED; |
||||||
|
s.Op(&op); |
||||||
|
s.DestroyThen( |
||||||
|
MakeOnceClosure([](grpc_exec_ctx *exec_ctx, grpc_error *error) {})); |
||||||
|
f.FlushExecCtx(); |
||||||
|
track_counters.Finish(state); |
||||||
|
grpc_metadata_batch_destroy(f.exec_ctx(), &b); |
||||||
|
grpc_slice_buffer_destroy(&send_buffer); |
||||||
|
} |
||||||
|
BENCHMARK(BM_TransportStreamSend)->Range(0, 128 * 1024 * 1024); |
||||||
|
|
||||||
|
#define SLICE_FROM_BUFFER(s) grpc_slice_from_static_buffer(s, sizeof(s) - 1) |
||||||
|
|
||||||
|
static grpc_slice CreateIncomingDataSlice(size_t length, size_t frame_size) { |
||||||
|
std::queue<char> unframed; |
||||||
|
|
||||||
|
unframed.push(static_cast<uint8_t>(0)); |
||||||
|
unframed.push(static_cast<uint8_t>(length >> 24)); |
||||||
|
unframed.push(static_cast<uint8_t>(length >> 16)); |
||||||
|
unframed.push(static_cast<uint8_t>(length >> 8)); |
||||||
|
unframed.push(static_cast<uint8_t>(length)); |
||||||
|
for (size_t i = 0; i < length; i++) { |
||||||
|
unframed.push('a'); |
||||||
|
} |
||||||
|
|
||||||
|
std::vector<char> framed; |
||||||
|
while (unframed.size() > frame_size) { |
||||||
|
// frame size
|
||||||
|
framed.push_back(static_cast<uint8_t>(frame_size >> 16)); |
||||||
|
framed.push_back(static_cast<uint8_t>(frame_size >> 8)); |
||||||
|
framed.push_back(static_cast<uint8_t>(frame_size)); |
||||||
|
// data frame
|
||||||
|
framed.push_back(0); |
||||||
|
// no flags
|
||||||
|
framed.push_back(0); |
||||||
|
// stream id
|
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(1); |
||||||
|
// frame data
|
||||||
|
for (size_t i = 0; i < frame_size; i++) { |
||||||
|
framed.push_back(unframed.front()); |
||||||
|
unframed.pop(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// frame size
|
||||||
|
framed.push_back(static_cast<uint8_t>(unframed.size() >> 16)); |
||||||
|
framed.push_back(static_cast<uint8_t>(unframed.size() >> 8)); |
||||||
|
framed.push_back(static_cast<uint8_t>(unframed.size())); |
||||||
|
// data frame
|
||||||
|
framed.push_back(0); |
||||||
|
// no flags
|
||||||
|
framed.push_back(0); |
||||||
|
// stream id
|
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(0); |
||||||
|
framed.push_back(1); |
||||||
|
while (!unframed.empty()) { |
||||||
|
framed.push_back(unframed.front()); |
||||||
|
unframed.pop(); |
||||||
|
} |
||||||
|
|
||||||
|
return grpc_slice_from_copied_buffer(framed.data(), framed.size()); |
||||||
|
} |
||||||
|
|
||||||
|
static void BM_TransportStreamRecv(benchmark::State &state) { |
||||||
|
TrackCounters track_counters; |
||||||
|
Fixture f(grpc::ChannelArguments(), true); |
||||||
|
Stream s(&f); |
||||||
|
s.Init(state); |
||||||
|
grpc_transport_stream_op op; |
||||||
|
grpc_byte_stream *recv_stream; |
||||||
|
grpc_slice incoming_data = CreateIncomingDataSlice(state.range(0), 16384); |
||||||
|
|
||||||
|
grpc_metadata_batch b; |
||||||
|
grpc_metadata_batch_init(&b); |
||||||
|
grpc_metadata_batch b_recv; |
||||||
|
grpc_metadata_batch_init(&b_recv); |
||||||
|
b.deadline = gpr_inf_future(GPR_CLOCK_MONOTONIC); |
||||||
|
std::vector<grpc_mdelem> elems = |
||||||
|
RepresentativeClientInitialMetadata::GetElems(f.exec_ctx()); |
||||||
|
std::vector<grpc_linked_mdelem> storage(elems.size()); |
||||||
|
for (size_t i = 0; i < elems.size(); i++) { |
||||||
|
GPR_ASSERT(GRPC_LOG_IF_ERROR( |
||||||
|
"addmd", |
||||||
|
grpc_metadata_batch_add_tail(f.exec_ctx(), &b, &storage[i], elems[i]))); |
||||||
|
} |
||||||
|
|
||||||
|
std::unique_ptr<Closure> do_nothing = |
||||||
|
MakeClosure([](grpc_exec_ctx *exec_ctx, grpc_error *error) {}); |
||||||
|
|
||||||
|
uint32_t received; |
||||||
|
|
||||||
|
std::unique_ptr<Closure> drain_start; |
||||||
|
std::unique_ptr<Closure> drain; |
||||||
|
std::unique_ptr<Closure> drain_continue; |
||||||
|
grpc_slice recv_slice; |
||||||
|
|
||||||
|
std::unique_ptr<Closure> c = |
||||||
|
MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (!state.KeepRunning()) return; |
||||||
|
// force outgoing window to be yuge
|
||||||
|
s.chttp2_stream()->incoming_window_delta = 1024 * 1024 * 1024; |
||||||
|
f.chttp2_transport()->incoming_window = 1024 * 1024 * 1024; |
||||||
|
received = 0; |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.on_complete = do_nothing.get(); |
||||||
|
op.recv_message = &recv_stream; |
||||||
|
op.recv_message_ready = drain_start.get(); |
||||||
|
s.Op(&op); |
||||||
|
f.PushInput(grpc_slice_ref(incoming_data)); |
||||||
|
}); |
||||||
|
|
||||||
|
drain_start = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
if (recv_stream == NULL) { |
||||||
|
GPR_ASSERT(!state.KeepRunning()); |
||||||
|
return; |
||||||
|
} |
||||||
|
grpc_closure_run(exec_ctx, drain.get(), GRPC_ERROR_NONE); |
||||||
|
}); |
||||||
|
|
||||||
|
drain = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
do { |
||||||
|
if (received == recv_stream->length) { |
||||||
|
grpc_byte_stream_destroy(exec_ctx, recv_stream); |
||||||
|
grpc_closure_sched(exec_ctx, c.get(), GRPC_ERROR_NONE); |
||||||
|
return; |
||||||
|
} |
||||||
|
} while (grpc_byte_stream_next(exec_ctx, recv_stream, &recv_slice, |
||||||
|
recv_stream->length - received, |
||||||
|
drain_continue.get())); |
||||||
|
}); |
||||||
|
|
||||||
|
drain_continue = MakeClosure([&](grpc_exec_ctx *exec_ctx, grpc_error *error) { |
||||||
|
received += GRPC_SLICE_LENGTH(recv_slice); |
||||||
|
grpc_slice_unref_internal(exec_ctx, recv_slice); |
||||||
|
grpc_closure_run(exec_ctx, drain.get(), GRPC_ERROR_NONE); |
||||||
|
}); |
||||||
|
|
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.send_initial_metadata = &b; |
||||||
|
op.recv_initial_metadata = &b_recv; |
||||||
|
op.on_complete = c.get(); |
||||||
|
s.Op(&op); |
||||||
|
f.PushInput(SLICE_FROM_BUFFER( |
||||||
|
"\x00\x00\x00\x04\x00\x00\x00\x00\x00" |
||||||
|
// Generated using:
|
||||||
|
// tools/codegen/core/gen_header_frame.py <
|
||||||
|
// test/cpp/microbenchmarks/representative_server_initial_metadata.headers
|
||||||
|
"\x00\x00X\x01\x04\x00\x00\x00\x01" |
||||||
|
"\x10\x07:status\x03" |
||||||
|
"200" |
||||||
|
"\x10\x0c" |
||||||
|
"content-type\x10" |
||||||
|
"application/grpc" |
||||||
|
"\x10\x14grpc-accept-encoding\x15identity,deflate,gzip")); |
||||||
|
|
||||||
|
f.FlushExecCtx(); |
||||||
|
memset(&op, 0, sizeof(op)); |
||||||
|
op.cancel_error = GRPC_ERROR_CANCELLED; |
||||||
|
s.Op(&op); |
||||||
|
s.DestroyThen( |
||||||
|
MakeOnceClosure([](grpc_exec_ctx *exec_ctx, grpc_error *error) {})); |
||||||
|
f.FlushExecCtx(); |
||||||
|
track_counters.Finish(state); |
||||||
|
grpc_metadata_batch_destroy(f.exec_ctx(), &b); |
||||||
|
grpc_metadata_batch_destroy(f.exec_ctx(), &b_recv); |
||||||
|
grpc_slice_unref(incoming_data); |
||||||
|
} |
||||||
|
BENCHMARK(BM_TransportStreamRecv)->Range(0, 128 * 1024 * 1024); |
||||||
|
|
||||||
|
BENCHMARK_MAIN(); |
||||||
Loading…
Reference in new issue