|
|
|
@ -112,11 +112,13 @@ void resource_quota_server(grpc_end2end_test_config config) { |
|
|
|
|
grpc_resource_quota_create("test_server"); |
|
|
|
|
grpc_resource_quota_resize(resource_quota, 5 * 1024 * 1024); |
|
|
|
|
|
|
|
|
|
#define NUM_CALLS 100 |
|
|
|
|
#define NUM_CALLS 10 |
|
|
|
|
#define CLIENT_BASE_TAG 1000 |
|
|
|
|
#define SERVER_START_BASE_TAG 2000 |
|
|
|
|
#define SERVER_RECV_BASE_TAG 3000 |
|
|
|
|
#define SERVER_END_BASE_TAG 4000 |
|
|
|
|
#define NUM_ROUNDS 1 |
|
|
|
|
#define MAX_READING 4 |
|
|
|
|
|
|
|
|
|
grpc_arg arg; |
|
|
|
|
arg.key = GRPC_ARG_RESOURCE_QUOTA; |
|
|
|
@ -132,132 +134,60 @@ void resource_quota_server(grpc_end2end_test_config config) { |
|
|
|
|
* multiple round trips to deliver to the peer, and their exact contents of |
|
|
|
|
* will be verified on completion. */ |
|
|
|
|
gpr_slice request_payload_slice = generate_random_slice(); |
|
|
|
|
|
|
|
|
|
grpc_call *client_calls[NUM_CALLS]; |
|
|
|
|
grpc_call *server_calls[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array initial_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array trailing_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array request_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_call_details call_details[NUM_CALLS]; |
|
|
|
|
grpc_status_code status[NUM_CALLS]; |
|
|
|
|
char *details[NUM_CALLS]; |
|
|
|
|
size_t details_capacity[NUM_CALLS]; |
|
|
|
|
grpc_byte_buffer *request_payload_recv[NUM_CALLS]; |
|
|
|
|
int was_cancelled[NUM_CALLS]; |
|
|
|
|
grpc_call_error error; |
|
|
|
|
int pending_client_calls = 0; |
|
|
|
|
int pending_server_start_calls = 0; |
|
|
|
|
int pending_server_recv_calls = 0; |
|
|
|
|
int pending_server_end_calls = 0; |
|
|
|
|
int cancelled_calls_on_client = 0; |
|
|
|
|
int cancelled_calls_on_server = 0; |
|
|
|
|
|
|
|
|
|
grpc_byte_buffer *request_payload = |
|
|
|
|
grpc_raw_byte_buffer_create(&request_payload_slice, 1); |
|
|
|
|
|
|
|
|
|
grpc_op ops[6]; |
|
|
|
|
grpc_op *op; |
|
|
|
|
|
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
grpc_metadata_array_init(&initial_metadata_recv[i]); |
|
|
|
|
grpc_metadata_array_init(&trailing_metadata_recv[i]); |
|
|
|
|
grpc_metadata_array_init(&request_metadata_recv[i]); |
|
|
|
|
grpc_call_details_init(&call_details[i]); |
|
|
|
|
details[i] = NULL; |
|
|
|
|
details_capacity[i] = 0; |
|
|
|
|
request_payload_recv[i] = NULL; |
|
|
|
|
was_cancelled[i] = 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
error = grpc_server_request_call( |
|
|
|
|
f.server, &server_calls[i], &call_details[i], &request_metadata_recv[i], |
|
|
|
|
f.cq, f.cq, tag(SERVER_START_BASE_TAG + i)); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
pending_server_start_calls++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
client_calls[i] = grpc_channel_create_call( |
|
|
|
|
f.client, NULL, GRPC_PROPAGATE_DEFAULTS, f.cq, "/foo", |
|
|
|
|
"foo.test.google.fr", n_seconds_time(60), NULL); |
|
|
|
|
|
|
|
|
|
memset(ops, 0, sizeof(ops)); |
|
|
|
|
op = ops; |
|
|
|
|
op->op = GRPC_OP_SEND_INITIAL_METADATA; |
|
|
|
|
op->data.send_initial_metadata.count = 0; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_SEND_MESSAGE; |
|
|
|
|
op->data.send_message = request_payload; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_RECV_INITIAL_METADATA; |
|
|
|
|
op->data.recv_initial_metadata = &initial_metadata_recv[i]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
|
|
|
op->data.recv_status_on_client.trailing_metadata = |
|
|
|
|
&trailing_metadata_recv[i]; |
|
|
|
|
op->data.recv_status_on_client.status = &status[i]; |
|
|
|
|
op->data.recv_status_on_client.status_details = &details[i]; |
|
|
|
|
op->data.recv_status_on_client.status_details_capacity = |
|
|
|
|
&details_capacity[i]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = grpc_call_start_batch(client_calls[i], ops, (size_t)(op - ops), |
|
|
|
|
tag(CLIENT_BASE_TAG + i), NULL); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
pending_client_calls++; |
|
|
|
|
} |
|
|
|
|
for (int r = 0; r < NUM_ROUNDS; r++) { |
|
|
|
|
grpc_call *client_calls[NUM_CALLS]; |
|
|
|
|
grpc_call *server_calls[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array initial_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array trailing_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_metadata_array request_metadata_recv[NUM_CALLS]; |
|
|
|
|
grpc_call_details call_details[NUM_CALLS]; |
|
|
|
|
grpc_status_code status[NUM_CALLS]; |
|
|
|
|
char *details[NUM_CALLS]; |
|
|
|
|
size_t details_capacity[NUM_CALLS]; |
|
|
|
|
grpc_byte_buffer *request_payload_recv[NUM_CALLS]; |
|
|
|
|
int was_cancelled[NUM_CALLS]; |
|
|
|
|
grpc_call_error error; |
|
|
|
|
int pending_client_calls = 0; |
|
|
|
|
int pending_server_start_calls = 0; |
|
|
|
|
int pending_server_recv_calls = 0; |
|
|
|
|
int pending_server_end_calls = 0; |
|
|
|
|
int cancelled_calls_on_client = 0; |
|
|
|
|
int cancelled_calls_on_server = 0; |
|
|
|
|
int num_ready_for_reading_on_server = 0; |
|
|
|
|
int num_currently_reading_on_server = 0; |
|
|
|
|
int ready_for_reading[NUM_CALLS]; |
|
|
|
|
|
|
|
|
|
grpc_op ops[6]; |
|
|
|
|
grpc_op *op; |
|
|
|
|
|
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
grpc_metadata_array_init(&initial_metadata_recv[i]); |
|
|
|
|
grpc_metadata_array_init(&trailing_metadata_recv[i]); |
|
|
|
|
grpc_metadata_array_init(&request_metadata_recv[i]); |
|
|
|
|
grpc_call_details_init(&call_details[i]); |
|
|
|
|
details[i] = NULL; |
|
|
|
|
details_capacity[i] = 0; |
|
|
|
|
request_payload_recv[i] = NULL; |
|
|
|
|
was_cancelled[i] = 0; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
while (pending_client_calls + pending_server_recv_calls + |
|
|
|
|
pending_server_end_calls > |
|
|
|
|
0) { |
|
|
|
|
grpc_event ev = grpc_completion_queue_next(f.cq, n_seconds_time(10), NULL); |
|
|
|
|
GPR_ASSERT(ev.type == GRPC_OP_COMPLETE); |
|
|
|
|
|
|
|
|
|
int ev_tag = (int)(intptr_t)ev.tag; |
|
|
|
|
if (ev_tag < CLIENT_BASE_TAG) { |
|
|
|
|
abort(); /* illegal tag */ |
|
|
|
|
} else if (ev_tag < SERVER_START_BASE_TAG) { |
|
|
|
|
/* client call finished */ |
|
|
|
|
int call_id = ev_tag - CLIENT_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
switch (status[call_id]) { |
|
|
|
|
case GRPC_STATUS_RESOURCE_EXHAUSTED: |
|
|
|
|
cancelled_calls_on_client++; |
|
|
|
|
break; |
|
|
|
|
case GRPC_STATUS_OK: |
|
|
|
|
break; |
|
|
|
|
default: |
|
|
|
|
gpr_log(GPR_ERROR, "Unexpected status code: %d", status[call_id]); |
|
|
|
|
abort(); |
|
|
|
|
} |
|
|
|
|
GPR_ASSERT(pending_client_calls > 0); |
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
error = |
|
|
|
|
grpc_server_request_call(f.server, &server_calls[i], &call_details[i], |
|
|
|
|
&request_metadata_recv[i], f.cq, f.cq, |
|
|
|
|
tag(SERVER_START_BASE_TAG + i)); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
grpc_metadata_array_destroy(&initial_metadata_recv[call_id]); |
|
|
|
|
grpc_metadata_array_destroy(&trailing_metadata_recv[call_id]); |
|
|
|
|
grpc_call_destroy(client_calls[call_id]); |
|
|
|
|
gpr_free(details[call_id]); |
|
|
|
|
pending_server_start_calls++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
pending_client_calls--; |
|
|
|
|
} else if (ev_tag < SERVER_RECV_BASE_TAG) { |
|
|
|
|
/* new incoming call to the server */ |
|
|
|
|
int call_id = ev_tag - SERVER_START_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
for (int i = 0; i < NUM_CALLS; i++) { |
|
|
|
|
client_calls[i] = grpc_channel_create_call( |
|
|
|
|
f.client, NULL, GRPC_PROPAGATE_DEFAULTS, f.cq, "/foo", |
|
|
|
|
"foo.test.google.fr", n_seconds_time(60), NULL); |
|
|
|
|
|
|
|
|
|
memset(ops, 0, sizeof(ops)); |
|
|
|
|
op = ops; |
|
|
|
@ -266,81 +196,174 @@ void resource_quota_server(grpc_end2end_test_config config) { |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_RECV_MESSAGE; |
|
|
|
|
op->data.recv_message = &request_payload_recv[call_id]; |
|
|
|
|
op->op = GRPC_OP_SEND_MESSAGE; |
|
|
|
|
op->data.send_message = request_payload; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = |
|
|
|
|
grpc_call_start_batch(server_calls[call_id], ops, (size_t)(op - ops), |
|
|
|
|
tag(SERVER_RECV_BASE_TAG + call_id), NULL); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
GPR_ASSERT(pending_server_start_calls > 0); |
|
|
|
|
pending_server_start_calls--; |
|
|
|
|
pending_server_recv_calls++; |
|
|
|
|
|
|
|
|
|
grpc_call_details_destroy(&call_details[call_id]); |
|
|
|
|
grpc_metadata_array_destroy(&request_metadata_recv[call_id]); |
|
|
|
|
} else if (ev_tag < SERVER_END_BASE_TAG) { |
|
|
|
|
/* finished read on the server */ |
|
|
|
|
int call_id = ev_tag - SERVER_RECV_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
|
|
|
|
|
if (ev.success) { |
|
|
|
|
if (request_payload_recv[call_id] != NULL) { |
|
|
|
|
grpc_byte_buffer_destroy(request_payload_recv[call_id]); |
|
|
|
|
request_payload_recv[call_id] = NULL; |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
GPR_ASSERT(request_payload_recv[call_id] == NULL); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
memset(ops, 0, sizeof(ops)); |
|
|
|
|
op = ops; |
|
|
|
|
op->op = GRPC_OP_RECV_CLOSE_ON_SERVER; |
|
|
|
|
op->data.recv_close_on_server.cancelled = &was_cancelled[call_id]; |
|
|
|
|
op->op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_SEND_STATUS_FROM_SERVER; |
|
|
|
|
op->data.send_status_from_server.trailing_metadata_count = 0; |
|
|
|
|
op->data.send_status_from_server.status = GRPC_STATUS_OK; |
|
|
|
|
op->data.send_status_from_server.status_details = "xyz"; |
|
|
|
|
op->op = GRPC_OP_RECV_INITIAL_METADATA; |
|
|
|
|
op->data.recv_initial_metadata = &initial_metadata_recv[i]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = |
|
|
|
|
grpc_call_start_batch(server_calls[call_id], ops, (size_t)(op - ops), |
|
|
|
|
tag(SERVER_END_BASE_TAG + call_id), NULL); |
|
|
|
|
op->op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
|
|
|
op->data.recv_status_on_client.trailing_metadata = |
|
|
|
|
&trailing_metadata_recv[i]; |
|
|
|
|
op->data.recv_status_on_client.status = &status[i]; |
|
|
|
|
op->data.recv_status_on_client.status_details = &details[i]; |
|
|
|
|
op->data.recv_status_on_client.status_details_capacity = |
|
|
|
|
&details_capacity[i]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = grpc_call_start_batch(client_calls[i], ops, (size_t)(op - ops), |
|
|
|
|
tag(CLIENT_BASE_TAG + i), NULL); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
GPR_ASSERT(pending_server_recv_calls > 0); |
|
|
|
|
pending_server_recv_calls--; |
|
|
|
|
pending_server_end_calls++; |
|
|
|
|
} else { |
|
|
|
|
int call_id = ev_tag - SERVER_END_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
pending_client_calls++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
if (was_cancelled[call_id]) { |
|
|
|
|
cancelled_calls_on_server++; |
|
|
|
|
while (pending_client_calls + pending_server_recv_calls + |
|
|
|
|
pending_server_end_calls > |
|
|
|
|
0) { |
|
|
|
|
gpr_log(GPR_DEBUG, "cur=%d ready=%d", num_currently_reading_on_server, |
|
|
|
|
num_ready_for_reading_on_server); |
|
|
|
|
while (num_currently_reading_on_server < MAX_READING && |
|
|
|
|
num_ready_for_reading_on_server > 0) { |
|
|
|
|
int call_id = ready_for_reading[--num_ready_for_reading_on_server]; |
|
|
|
|
|
|
|
|
|
gpr_log(GPR_DEBUG, "start reading %d", call_id); |
|
|
|
|
|
|
|
|
|
memset(ops, 0, sizeof(ops)); |
|
|
|
|
op = ops; |
|
|
|
|
op->op = GRPC_OP_SEND_INITIAL_METADATA; |
|
|
|
|
op->data.send_initial_metadata.count = 0; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_RECV_MESSAGE; |
|
|
|
|
op->data.recv_message = &request_payload_recv[call_id]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = grpc_call_start_batch( |
|
|
|
|
server_calls[call_id], ops, (size_t)(op - ops), |
|
|
|
|
tag(SERVER_RECV_BASE_TAG + call_id), NULL); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
num_currently_reading_on_server++; |
|
|
|
|
|
|
|
|
|
gpr_log(GPR_DEBUG, "> cur=%d ready=%d", num_currently_reading_on_server, |
|
|
|
|
num_ready_for_reading_on_server); |
|
|
|
|
} |
|
|
|
|
GPR_ASSERT(pending_server_end_calls > 0); |
|
|
|
|
pending_server_end_calls--; |
|
|
|
|
|
|
|
|
|
grpc_call_destroy(server_calls[call_id]); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
grpc_event ev = |
|
|
|
|
grpc_completion_queue_next(f.cq, n_seconds_time(10), NULL); |
|
|
|
|
GPR_ASSERT(ev.type == GRPC_OP_COMPLETE); |
|
|
|
|
|
|
|
|
|
int ev_tag = (int)(intptr_t)ev.tag; |
|
|
|
|
if (ev_tag < CLIENT_BASE_TAG) { |
|
|
|
|
abort(); /* illegal tag */ |
|
|
|
|
} else if (ev_tag < SERVER_START_BASE_TAG) { |
|
|
|
|
/* client call finished */ |
|
|
|
|
int call_id = ev_tag - CLIENT_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
switch (status[call_id]) { |
|
|
|
|
case GRPC_STATUS_RESOURCE_EXHAUSTED: |
|
|
|
|
cancelled_calls_on_client++; |
|
|
|
|
break; |
|
|
|
|
case GRPC_STATUS_OK: |
|
|
|
|
break; |
|
|
|
|
default: |
|
|
|
|
gpr_log(GPR_ERROR, "Unexpected status code: %d", status[call_id]); |
|
|
|
|
abort(); |
|
|
|
|
} |
|
|
|
|
GPR_ASSERT(pending_client_calls > 0); |
|
|
|
|
|
|
|
|
|
grpc_metadata_array_destroy(&initial_metadata_recv[call_id]); |
|
|
|
|
grpc_metadata_array_destroy(&trailing_metadata_recv[call_id]); |
|
|
|
|
grpc_call_destroy(client_calls[call_id]); |
|
|
|
|
gpr_free(details[call_id]); |
|
|
|
|
|
|
|
|
|
pending_client_calls--; |
|
|
|
|
} else if (ev_tag < SERVER_RECV_BASE_TAG) { |
|
|
|
|
/* new incoming call to the server */ |
|
|
|
|
int call_id = ev_tag - SERVER_START_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
|
|
|
|
|
ready_for_reading[num_ready_for_reading_on_server++] = call_id; |
|
|
|
|
gpr_log(GPR_DEBUG, "queue read %d", call_id); |
|
|
|
|
|
|
|
|
|
GPR_ASSERT(pending_server_start_calls > 0); |
|
|
|
|
pending_server_start_calls--; |
|
|
|
|
pending_server_recv_calls++; |
|
|
|
|
|
|
|
|
|
grpc_call_details_destroy(&call_details[call_id]); |
|
|
|
|
grpc_metadata_array_destroy(&request_metadata_recv[call_id]); |
|
|
|
|
} else if (ev_tag < SERVER_END_BASE_TAG) { |
|
|
|
|
/* finished read on the server */ |
|
|
|
|
int call_id = ev_tag - SERVER_RECV_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
|
|
|
|
|
if (ev.success) { |
|
|
|
|
if (request_payload_recv[call_id] != NULL) { |
|
|
|
|
grpc_byte_buffer_destroy(request_payload_recv[call_id]); |
|
|
|
|
request_payload_recv[call_id] = NULL; |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
GPR_ASSERT(request_payload_recv[call_id] == NULL); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
gpr_log( |
|
|
|
|
GPR_INFO, |
|
|
|
|
"Done. %d total calls: %d cancelled at server, %d cancelled at client.", |
|
|
|
|
NUM_CALLS, cancelled_calls_on_server, cancelled_calls_on_client); |
|
|
|
|
memset(ops, 0, sizeof(ops)); |
|
|
|
|
op = ops; |
|
|
|
|
op->op = GRPC_OP_RECV_CLOSE_ON_SERVER; |
|
|
|
|
op->data.recv_close_on_server.cancelled = &was_cancelled[call_id]; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
op->op = GRPC_OP_SEND_STATUS_FROM_SERVER; |
|
|
|
|
op->data.send_status_from_server.trailing_metadata_count = 0; |
|
|
|
|
op->data.send_status_from_server.status = GRPC_STATUS_OK; |
|
|
|
|
op->data.send_status_from_server.status_details = "xyz"; |
|
|
|
|
op->flags = 0; |
|
|
|
|
op->reserved = NULL; |
|
|
|
|
op++; |
|
|
|
|
error = grpc_call_start_batch(server_calls[call_id], ops, |
|
|
|
|
(size_t)(op - ops), |
|
|
|
|
tag(SERVER_END_BASE_TAG + call_id), NULL); |
|
|
|
|
GPR_ASSERT(GRPC_CALL_OK == error); |
|
|
|
|
|
|
|
|
|
GPR_ASSERT(pending_server_recv_calls > 0); |
|
|
|
|
pending_server_recv_calls--; |
|
|
|
|
pending_server_end_calls++; |
|
|
|
|
num_currently_reading_on_server--; |
|
|
|
|
} else { |
|
|
|
|
int call_id = ev_tag - SERVER_END_BASE_TAG; |
|
|
|
|
GPR_ASSERT(call_id >= 0); |
|
|
|
|
GPR_ASSERT(call_id < NUM_CALLS); |
|
|
|
|
|
|
|
|
|
if (was_cancelled[call_id]) { |
|
|
|
|
cancelled_calls_on_server++; |
|
|
|
|
} |
|
|
|
|
GPR_ASSERT(pending_server_end_calls > 0); |
|
|
|
|
pending_server_end_calls--; |
|
|
|
|
|
|
|
|
|
GPR_ASSERT(cancelled_calls_on_client >= cancelled_calls_on_server); |
|
|
|
|
GPR_ASSERT(cancelled_calls_on_server >= 0.9 * cancelled_calls_on_client); |
|
|
|
|
grpc_call_destroy(server_calls[call_id]); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
gpr_log( |
|
|
|
|
GPR_INFO, |
|
|
|
|
"Done. %d total calls: %d cancelled at server, %d cancelled at client.", |
|
|
|
|
NUM_CALLS, cancelled_calls_on_server, cancelled_calls_on_client); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
grpc_byte_buffer_destroy(request_payload); |
|
|
|
|
gpr_slice_unref(request_payload_slice); |
|
|
|
|