diff options
-rw-r--r-- | test/core/fling/client.c | 16 | ||||
-rw-r--r-- | test/core/fling/server.c | 128 |
2 files changed, 68 insertions, 76 deletions
diff --git a/test/core/fling/client.c b/test/core/fling/client.c index 2cd08f8197..76a0b2eed9 100644 --- a/test/core/fling/client.c +++ b/test/core/fling/client.c @@ -83,7 +83,7 @@ static void init_ping_pong_request(void) { static void step_ping_pong_request(void) { call = grpc_channel_create_call(channel, cq, "/Reflector/reflectUnary", - "localhost", gpr_inf_future); + "localhost", gpr_inf_future); GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, ops, 6, (void *)1)); grpc_event_finish(grpc_completion_queue_next(cq, gpr_inf_future)); grpc_call_destroy(call); @@ -92,11 +92,11 @@ static void step_ping_pong_request(void) { static void init_ping_pong_stream(void) { call = grpc_channel_create_call(channel, cq, "/Reflector/reflectStream", - "localhost", gpr_inf_future); + "localhost", gpr_inf_future); stream_init_op.op = GRPC_OP_SEND_INITIAL_METADATA; stream_init_op.data.send_initial_metadata.count = 0; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &stream_init_op, 1, - (void *)1)); + GPR_ASSERT(GRPC_CALL_OK == + grpc_call_start_batch(call, &stream_init_op, 1, (void *)1)); grpc_event_finish(grpc_completion_queue_next(cq, gpr_inf_future)); grpc_metadata_array_init(&initial_metadata_recv); @@ -108,8 +108,8 @@ static void init_ping_pong_stream(void) { } static void step_ping_pong_stream(void) { - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, stream_step_ops, 2, - (void *)1)); + GPR_ASSERT(GRPC_CALL_OK == + grpc_call_start_batch(call, stream_step_ops, 2, (void *)1)); grpc_event_finish(grpc_completion_queue_next(cq, gpr_inf_future)); } @@ -126,7 +126,8 @@ typedef struct { static const scenario scenarios[] = { {"ping-pong-request", init_ping_pong_request, step_ping_pong_request}, - {"ping-pong-stream", init_ping_pong_stream, step_ping_pong_stream}, }; + {"ping-pong-stream", init_ping_pong_stream, step_ping_pong_stream}, +}; int main(int argc, char **argv) { gpr_slice slice = gpr_slice_from_copied_string("x"); @@ -176,7 +177,6 @@ int main(int argc, char **argv) { the_buffer = grpc_byte_buffer_create(&slice, payload_size); histogram = gpr_histogram_create(0.01, 60e9); - sc.init(); for (i = 0; i < 1000; i++) { diff --git a/test/core/fling/server.c b/test/core/fling/server.c index 2ac5500ee5..b1f3ca2588 100644 --- a/test/core/fling/server.c +++ b/test/core/fling/server.c @@ -60,7 +60,6 @@ static grpc_byte_buffer *payload_buffer = NULL; /* Used to drain the terminal read in unary calls. */ static grpc_byte_buffer *terminal_buffer = NULL; - static grpc_op read_op; static grpc_op metadata_send_op; static grpc_op write_op; @@ -80,7 +79,7 @@ static void request_call(void) { grpc_metadata_array_init(&request_metadata_recv); grpc_call_details_init(&call_details); grpc_server_request_call(server, &call, &call_details, &request_metadata_recv, - cq, tag(101)); + cq, tag(101)); } static void handle_unary_method(void) { @@ -101,24 +100,22 @@ static void handle_unary_method(void) { unary_ops[4].op = GRPC_OP_RECV_CLOSE_ON_SERVER; unary_ops[4].data.recv_close_on_server.cancelled = &was_cancelled; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, unary_ops, 5, - tag(6))); + GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, unary_ops, 5, tag(6))); } static void send_initial_metadata(void) { grpc_metadata_array_init(&initial_metadata_send); metadata_send_op.op = GRPC_OP_SEND_INITIAL_METADATA; metadata_send_op.data.send_initial_metadata.count = 0; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &metadata_send_op, 1, - tag(3))); + GPR_ASSERT(GRPC_CALL_OK == + grpc_call_start_batch(call, &metadata_send_op, 1, tag(3))); } static void start_read_op(int t) { /* Starting read at server */ read_op.op = GRPC_OP_RECV_MESSAGE; read_op.data.recv_message = &payload_buffer; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &read_op, 1, - tag(t))); + GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &read_op, 1, tag(t))); } static void start_write_op(void) { @@ -128,11 +125,9 @@ static void start_write_op(void) { gpr_log(GPR_INFO, "NULL payload buffer !!!"); } write_op.data.send_message = payload_buffer; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &write_op, 1, - tag(2))); + GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, &write_op, 1, tag(2))); } - static void start_send_status(void) { status_op[0].op = GRPC_OP_SEND_STATUS_FROM_SERVER; status_op[0].data.send_status_from_server.status = GRPC_STATUS_OK; @@ -141,8 +136,7 @@ static void start_send_status(void) { status_op[1].op = GRPC_OP_RECV_CLOSE_ON_SERVER; status_op[1].data.recv_close_on_server.cancelled = &was_cancelled; - GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, status_op, 2, - tag(4))); + GPR_ASSERT(GRPC_CALL_OK == grpc_call_start_batch(call, status_op, 2, tag(4))); } static void sigint_handler(int x) { got_sigint = 1; } @@ -214,66 +208,64 @@ int main(int argc, char **argv) { s = ev->tag; switch (ev->type) { case GRPC_OP_COMPLETE: - switch ((gpr_intptr) s) { - case 101: - if(call != NULL) { - if (0 == strcmp(call_details.method, - "/Reflector/reflectStream")) { - /* Received streaming call. Send metadata here. */ - start_read_op(1); - send_initial_metadata(); - } else { - /* Received unary call. Can do all ops in one batch. */ - start_read_op(5); - } - } - else { - GPR_ASSERT(shutdown_started); - } - /* request_call(); - */ - break; - case 1: - if (payload_buffer != NULL) { - /* Received payload from client. */ - start_write_op(); - } - else { - /* Received end of stream from client. */ - start_send_status(); - } - break; - case 2: - /* Write completed at server */ - start_read_op(1); - break; - case 3: - /* Metadata send completed at server */ - break; - case 4: - /* Send status and close completed at server */ - grpc_call_destroy(call); - request_call(); - break; - case 5: - /* Finished payload read for unary. Start all reamaining - * unary ops in a batch. - */ - handle_unary_method(); - break; - case 6: - /* Finished unary call. */ - grpc_call_destroy(call); - request_call(); - break; - } - break; + switch ((gpr_intptr)s) { + case 101: + if (call != NULL) { + if (0 == + strcmp(call_details.method, "/Reflector/reflectStream")) { + /* Received streaming call. Send metadata here. */ + start_read_op(1); + send_initial_metadata(); + } else { + /* Received unary call. Can do all ops in one batch. */ + start_read_op(5); + } + } else { + GPR_ASSERT(shutdown_started); + } + /* request_call(); + */ + break; + case 1: + if (payload_buffer != NULL) { + /* Received payload from client. */ + start_write_op(); + } else { + /* Received end of stream from client. */ + start_send_status(); + } + break; + case 2: + /* Write completed at server */ + start_read_op(1); + break; + case 3: + /* Metadata send completed at server */ + break; + case 4: + /* Send status and close completed at server */ + grpc_call_destroy(call); + request_call(); + break; + case 5: + /* Finished payload read for unary. Start all reamaining + * unary ops in a batch. + */ + handle_unary_method(); + break; + case 6: + /* Finished unary call. */ + grpc_call_destroy(call); + request_call(); + break; + } + break; case GRPC_SERVER_RPC_NEW: case GRPC_WRITE_ACCEPTED: case GRPC_READ: case GRPC_FINISH_ACCEPTED: case GRPC_FINISHED: - gpr_log(GPR_ERROR, "Unexpected event type."); + gpr_log(GPR_ERROR, "Unexpected event type."); GPR_ASSERT(0); break; case GRPC_QUEUE_SHUTDOWN: |