mirror of https://github.com/grpc/grpc.git
Sree Kuchibhotla
8 years ago
commit
2e0aeccd71
9 changed files with 329 additions and 4 deletions
@ -0,0 +1,142 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2017, 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 <benchmark/benchmark.h> |
||||
#include <string.h> |
||||
#include <atomic> |
||||
|
||||
#include <grpc/grpc.h> |
||||
#include "test/cpp/microbenchmarks/helpers.h" |
||||
|
||||
extern "C" { |
||||
#include "src/core/lib/iomgr/ev_posix.h" |
||||
#include "src/core/lib/iomgr/port.h" |
||||
#include "src/core/lib/surface/completion_queue.h" |
||||
} |
||||
|
||||
struct grpc_pollset { |
||||
gpr_mu mu; |
||||
}; |
||||
|
||||
namespace grpc { |
||||
namespace testing { |
||||
|
||||
static void* make_tag(int i) { return (void*)(intptr_t)i; } |
||||
static grpc_completion_queue* g_cq; |
||||
static grpc_event_engine_vtable g_vtable; |
||||
|
||||
static __thread int g_thread_idx; |
||||
static __thread grpc_cq_completion g_cq_completion; |
||||
|
||||
static void pollset_shutdown(grpc_exec_ctx* exec_ctx, grpc_pollset* ps, |
||||
grpc_closure* closure) { |
||||
grpc_closure_sched(exec_ctx, closure, GRPC_ERROR_NONE); |
||||
} |
||||
|
||||
static void pollset_init(grpc_pollset* ps, gpr_mu** mu) { |
||||
gpr_mu_init(&ps->mu); |
||||
*mu = &ps->mu; |
||||
} |
||||
|
||||
static void pollset_destroy(grpc_pollset* ps) { gpr_mu_destroy(&ps->mu); } |
||||
|
||||
static grpc_error* pollset_kick(grpc_pollset* p, grpc_pollset_worker* worker) { |
||||
return GRPC_ERROR_NONE; |
||||
} |
||||
|
||||
/* Callback when the tag is dequeued from the completion queue. Does nothing */ |
||||
static void cq_done_cb(grpc_exec_ctx* exec_ctx, void* done_arg, |
||||
grpc_cq_completion* cq_completion) {} |
||||
|
||||
/* Queues a completion tag. ZERO polling overhead */ |
||||
static grpc_error* pollset_work(grpc_exec_ctx* exec_ctx, grpc_pollset* ps, |
||||
grpc_pollset_worker** worker, gpr_timespec now, |
||||
gpr_timespec deadline) { |
||||
gpr_mu_unlock(&ps->mu); |
||||
grpc_cq_end_op(exec_ctx, g_cq, make_tag(g_thread_idx), GRPC_ERROR_NONE, |
||||
cq_done_cb, NULL, &g_cq_completion); |
||||
grpc_exec_ctx_flush(exec_ctx); |
||||
gpr_mu_lock(&ps->mu); |
||||
return GRPC_ERROR_NONE; |
||||
} |
||||
|
||||
static void init_engine_vtable() { |
||||
memset(&g_vtable, 0, sizeof(g_vtable)); |
||||
|
||||
g_vtable.pollset_size = sizeof(grpc_pollset); |
||||
g_vtable.pollset_init = pollset_init; |
||||
g_vtable.pollset_shutdown = pollset_shutdown; |
||||
g_vtable.pollset_destroy = pollset_destroy; |
||||
g_vtable.pollset_work = pollset_work; |
||||
g_vtable.pollset_kick = pollset_kick; |
||||
} |
||||
|
||||
static void setup() { |
||||
grpc_init(); |
||||
init_engine_vtable(); |
||||
grpc_set_event_engine_test_only(&g_vtable); |
||||
|
||||
g_cq = grpc_completion_queue_create(NULL); |
||||
} |
||||
|
||||
static void BM_Cq_Throughput(benchmark::State& state) { |
||||
TrackCounters track_counters; |
||||
gpr_timespec deadline = gpr_inf_past(GPR_CLOCK_MONOTONIC); |
||||
|
||||
if (state.thread_index == 0) { |
||||
setup(); |
||||
} |
||||
|
||||
while (state.KeepRunning()) { |
||||
g_thread_idx = state.thread_index; |
||||
void* dummy_tag = make_tag(g_thread_idx); |
||||
grpc_cq_begin_op(g_cq, dummy_tag); |
||||
grpc_completion_queue_next(g_cq, deadline, NULL); |
||||
} |
||||
|
||||
state.SetItemsProcessed(state.iterations()); |
||||
|
||||
if (state.thread_index == 0) { |
||||
grpc_completion_queue_shutdown(g_cq); |
||||
grpc_completion_queue_destroy(g_cq); |
||||
} |
||||
|
||||
track_counters.Finish(state); |
||||
} |
||||
|
||||
BENCHMARK(BM_Cq_Throughput)->ThreadRange(1, 16)->UseRealTime(); |
||||
|
||||
} // namespace testing
|
||||
} // namespace grpc
|
||||
|
||||
BENCHMARK_MAIN(); |
||||
Loading…
Reference in new issue