mirror of https://github.com/grpc/grpc.git
ncteisen
7 years ago
parent
54e8f37e53
commit
dd34f3caf6
14 changed files with 198 additions and 276 deletions
@ -1,219 +0,0 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015 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 "src/core/lib/iomgr/closure.h" |
||||
|
||||
#include <assert.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
|
||||
#include "src/core/lib/profiling/timers.h" |
||||
|
||||
#ifndef NDEBUG |
||||
grpc_tracer_flag grpc_trace_closure = GRPC_TRACER_INITIALIZER(false, "closure"); |
||||
#endif |
||||
|
||||
#ifndef NDEBUG |
||||
grpc_closure* grpc_closure_init(const char* file, int line, |
||||
grpc_closure* closure, grpc_iomgr_cb_func cb, |
||||
void* cb_arg, |
||||
grpc_closure_scheduler* scheduler) { |
||||
#else |
||||
grpc_closure* grpc_closure_init(grpc_closure* closure, grpc_iomgr_cb_func cb, |
||||
void* cb_arg, |
||||
grpc_closure_scheduler* scheduler) { |
||||
#endif |
||||
closure->cb = cb; |
||||
closure->cb_arg = cb_arg; |
||||
closure->scheduler = scheduler; |
||||
#ifndef NDEBUG |
||||
closure->scheduled = false; |
||||
closure->file_initiated = NULL; |
||||
closure->line_initiated = 0; |
||||
closure->run = false; |
||||
closure->file_created = file; |
||||
closure->line_created = line; |
||||
#endif |
||||
return closure; |
||||
} |
||||
|
||||
void grpc_closure_list_init(grpc_closure_list* closure_list) { |
||||
closure_list->head = closure_list->tail = NULL; |
||||
} |
||||
|
||||
bool grpc_closure_list_append(grpc_closure_list* closure_list, |
||||
grpc_closure* closure, grpc_error* error) { |
||||
if (closure == NULL) { |
||||
GRPC_ERROR_UNREF(error); |
||||
return false; |
||||
} |
||||
closure->error_data.error = error; |
||||
closure->next_data.next = NULL; |
||||
bool was_empty = (closure_list->head == NULL); |
||||
if (was_empty) { |
||||
closure_list->head = closure; |
||||
} else { |
||||
closure_list->tail->next_data.next = closure; |
||||
} |
||||
closure_list->tail = closure; |
||||
return was_empty; |
||||
} |
||||
|
||||
void grpc_closure_list_fail_all(grpc_closure_list* list, |
||||
grpc_error* forced_failure) { |
||||
for (grpc_closure* c = list->head; c != NULL; c = c->next_data.next) { |
||||
if (c->error_data.error == GRPC_ERROR_NONE) { |
||||
c->error_data.error = GRPC_ERROR_REF(forced_failure); |
||||
} |
||||
} |
||||
GRPC_ERROR_UNREF(forced_failure); |
||||
} |
||||
|
||||
bool grpc_closure_list_empty(grpc_closure_list closure_list) { |
||||
return closure_list.head == NULL; |
||||
} |
||||
|
||||
void grpc_closure_list_move(grpc_closure_list* src, grpc_closure_list* dst) { |
||||
if (src->head == NULL) { |
||||
return; |
||||
} |
||||
if (dst->head == NULL) { |
||||
*dst = *src; |
||||
} else { |
||||
dst->tail->next_data.next = src->head; |
||||
dst->tail = src->tail; |
||||
} |
||||
src->head = src->tail = NULL; |
||||
} |
||||
|
||||
typedef struct { |
||||
grpc_iomgr_cb_func cb; |
||||
void* cb_arg; |
||||
grpc_closure wrapper; |
||||
} wrapped_closure; |
||||
|
||||
static void closure_wrapper(grpc_exec_ctx* exec_ctx, void* arg, |
||||
grpc_error* error) { |
||||
wrapped_closure* wc = (wrapped_closure*)arg; |
||||
grpc_iomgr_cb_func cb = wc->cb; |
||||
void* cb_arg = wc->cb_arg; |
||||
gpr_free(wc); |
||||
cb(exec_ctx, cb_arg, error); |
||||
} |
||||
|
||||
#ifndef NDEBUG |
||||
grpc_closure* grpc_closure_create(const char* file, int line, |
||||
grpc_iomgr_cb_func cb, void* cb_arg, |
||||
grpc_closure_scheduler* scheduler) { |
||||
#else |
||||
grpc_closure* grpc_closure_create(grpc_iomgr_cb_func cb, void* cb_arg, |
||||
grpc_closure_scheduler* scheduler) { |
||||
#endif |
||||
wrapped_closure* wc = (wrapped_closure*)gpr_malloc(sizeof(*wc)); |
||||
wc->cb = cb; |
||||
wc->cb_arg = cb_arg; |
||||
#ifndef NDEBUG |
||||
grpc_closure_init(file, line, &wc->wrapper, closure_wrapper, wc, scheduler); |
||||
#else |
||||
grpc_closure_init(&wc->wrapper, closure_wrapper, wc, scheduler); |
||||
#endif |
||||
return &wc->wrapper; |
||||
} |
||||
|
||||
#ifndef NDEBUG |
||||
void grpc_closure_run(const char* file, int line, grpc_exec_ctx* exec_ctx, |
||||
grpc_closure* c, grpc_error* error) { |
||||
#else |
||||
void grpc_closure_run(grpc_exec_ctx* exec_ctx, grpc_closure* c, |
||||
grpc_error* error) { |
||||
#endif |
||||
GPR_TIMER_BEGIN("grpc_closure_run", 0); |
||||
if (c != NULL) { |
||||
#ifndef NDEBUG |
||||
c->file_initiated = file; |
||||
c->line_initiated = line; |
||||
c->run = true; |
||||
#endif |
||||
assert(c->cb); |
||||
c->scheduler->vtable->run(exec_ctx, c, error); |
||||
} else { |
||||
GRPC_ERROR_UNREF(error); |
||||
} |
||||
GPR_TIMER_END("grpc_closure_run", 0); |
||||
} |
||||
|
||||
#ifndef NDEBUG |
||||
void grpc_closure_sched(const char* file, int line, grpc_exec_ctx* exec_ctx, |
||||
grpc_closure* c, grpc_error* error) { |
||||
#else |
||||
void grpc_closure_sched(grpc_exec_ctx* exec_ctx, grpc_closure* c, |
||||
grpc_error* error) { |
||||
#endif |
||||
GPR_TIMER_BEGIN("grpc_closure_sched", 0); |
||||
if (c != NULL) { |
||||
#ifndef NDEBUG |
||||
if (c->scheduled) { |
||||
gpr_log(GPR_ERROR, |
||||
"Closure already scheduled. (closure: %p, created: [%s:%d], " |
||||
"previously scheduled at: [%s: %d] run?: %s", |
||||
c, c->file_created, c->line_created, c->file_initiated, |
||||
c->line_initiated, c->run ? "true" : "false"); |
||||
abort(); |
||||
} |
||||
c->scheduled = true; |
||||
c->file_initiated = file; |
||||
c->line_initiated = line; |
||||
c->run = false; |
||||
#endif |
||||
assert(c->cb); |
||||
c->scheduler->vtable->sched(exec_ctx, c, error); |
||||
} else { |
||||
GRPC_ERROR_UNREF(error); |
||||
} |
||||
GPR_TIMER_END("grpc_closure_sched", 0); |
||||
} |
||||
|
||||
#ifndef NDEBUG |
||||
void grpc_closure_list_sched(const char* file, int line, |
||||
grpc_exec_ctx* exec_ctx, grpc_closure_list* list) { |
||||
#else |
||||
void grpc_closure_list_sched(grpc_exec_ctx* exec_ctx, grpc_closure_list* list) { |
||||
#endif |
||||
grpc_closure* c = list->head; |
||||
while (c != NULL) { |
||||
grpc_closure* next = c->next_data.next; |
||||
#ifndef NDEBUG |
||||
if (c->scheduled) { |
||||
gpr_log(GPR_ERROR, |
||||
"Closure already scheduled. (closure: %p, created: [%s:%d], " |
||||
"previously scheduled at: [%s: %d] run?: %s", |
||||
c, c->file_created, c->line_created, c->file_initiated, |
||||
c->line_initiated, c->run ? "true" : "false"); |
||||
abort(); |
||||
} |
||||
c->scheduled = true; |
||||
c->file_initiated = file; |
||||
c->line_initiated = line; |
||||
c->run = false; |
||||
#endif |
||||
assert(c->cb); |
||||
c->scheduler->vtable->sched(exec_ctx, c, c->error_data.error); |
||||
c = next; |
||||
} |
||||
list->head = list->tail = NULL; |
||||
} |
||||
Loading…
Reference in new issue