mirror of https://github.com/grpc/grpc.git
Sree Kuchibhotla
9 years ago
commit
8dbe2cb071
253 changed files with 6543 additions and 2602 deletions
@ -0,0 +1,153 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <iostream> |
||||||
|
#include <memory> |
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include <grpc++/grpc++.h> |
||||||
|
#include <thread> |
||||||
|
|
||||||
|
#include "helloworld.grpc.pb.h" |
||||||
|
|
||||||
|
using grpc::Channel; |
||||||
|
using grpc::ClientAsyncResponseReader; |
||||||
|
using grpc::ClientContext; |
||||||
|
using grpc::CompletionQueue; |
||||||
|
using grpc::Status; |
||||||
|
using helloworld::HelloRequest; |
||||||
|
using helloworld::HelloReply; |
||||||
|
using helloworld::Greeter; |
||||||
|
|
||||||
|
class GreeterClient { |
||||||
|
public: |
||||||
|
explicit GreeterClient(std::shared_ptr<Channel> channel) |
||||||
|
: stub_(Greeter::NewStub(channel)) {} |
||||||
|
|
||||||
|
// Assembles the client's payload and sends it to the server.
|
||||||
|
void SayHello(const std::string& user) { |
||||||
|
// Data we are sending to the server.
|
||||||
|
HelloRequest request; |
||||||
|
request.set_name(user); |
||||||
|
|
||||||
|
// Call object to store rpc data
|
||||||
|
AsyncClientCall* call = new AsyncClientCall; |
||||||
|
|
||||||
|
// stub_->AsyncSayHello() performs the RPC call, returning an instance to
|
||||||
|
// store in "call". Because we are using the asynchronous API, we need to
|
||||||
|
// hold on to the "call" instance in order to get updates on the ongoing RPC.
|
||||||
|
call->response_reader = stub_->AsyncSayHello(&call->context, request, &cq_); |
||||||
|
|
||||||
|
|
||||||
|
// Request that, upon completion of the RPC, "reply" be updated with the
|
||||||
|
// server's response; "status" with the indication of whether the operation
|
||||||
|
// was successful. Tag the request with the memory address of the call object.
|
||||||
|
call->response_reader->Finish(&call->reply, &call->status, (void*)call); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
// Loop while listening for completed responses.
|
||||||
|
// Prints out the response from the server.
|
||||||
|
void AsyncCompleteRpc() { |
||||||
|
void* got_tag; |
||||||
|
bool ok = false; |
||||||
|
|
||||||
|
// Block until the next result is available in the completion queue "cq".
|
||||||
|
while (cq_.Next(&got_tag, &ok)) { |
||||||
|
// The tag in this example is the memory location of the call object
|
||||||
|
AsyncClientCall* call = static_cast<AsyncClientCall*>(got_tag); |
||||||
|
|
||||||
|
// Verify that the request was completed successfully. Note that "ok"
|
||||||
|
// corresponds solely to the request for updates introduced by Finish().
|
||||||
|
GPR_ASSERT(ok); |
||||||
|
|
||||||
|
if (call->status.ok()) |
||||||
|
std::cout << "Greeter received: " << call->reply.message() << std::endl; |
||||||
|
else |
||||||
|
std::cout << "RPC failed" << std::endl; |
||||||
|
|
||||||
|
// Once we're complete, deallocate the call object.
|
||||||
|
delete call; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
|
||||||
|
// struct for keeping state and data information
|
||||||
|
struct AsyncClientCall { |
||||||
|
// Container for the data we expect from the server.
|
||||||
|
HelloReply reply; |
||||||
|
|
||||||
|
// Context for the client. It could be used to convey extra information to
|
||||||
|
// the server and/or tweak certain RPC behaviors.
|
||||||
|
ClientContext context; |
||||||
|
|
||||||
|
// Storage for the status of the RPC upon completion.
|
||||||
|
Status status; |
||||||
|
|
||||||
|
|
||||||
|
std::unique_ptr<ClientAsyncResponseReader<HelloReply>> response_reader; |
||||||
|
}; |
||||||
|
|
||||||
|
// Out of the passed in Channel comes the stub, stored here, our view of the
|
||||||
|
// server's exposed services.
|
||||||
|
std::unique_ptr<Greeter::Stub> stub_; |
||||||
|
|
||||||
|
// The producer-consumer queue we use to communicate asynchronously with the
|
||||||
|
// gRPC runtime.
|
||||||
|
CompletionQueue cq_; |
||||||
|
}; |
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
|
||||||
|
|
||||||
|
// Instantiate the client. It requires a channel, out of which the actual RPCs
|
||||||
|
// are created. This channel models a connection to an endpoint (in this case,
|
||||||
|
// localhost at port 50051). We indicate that the channel isn't authenticated
|
||||||
|
// (use of InsecureChannelCredentials()).
|
||||||
|
GreeterClient greeter(grpc::CreateChannel( |
||||||
|
"localhost:50051", grpc::InsecureChannelCredentials())); |
||||||
|
|
||||||
|
// Spawn reader thread that loops indefinitely
|
||||||
|
std::thread thread_ = std::thread(&GreeterClient::AsyncCompleteRpc, &greeter); |
||||||
|
|
||||||
|
for (int i = 0; i < 100; i++) { |
||||||
|
std::string user("world " + std::to_string(i)); |
||||||
|
greeter.SayHello(user); // The actual RPC call!
|
||||||
|
} |
||||||
|
|
||||||
|
std::cout << "Press control-c to quit" << std::endl << std::endl; |
||||||
|
thread_.join(); //blocks forever
|
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
@ -0,0 +1,57 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef GRPC_IMPL_CODEGEN_BYTE_BUFFER_READER_H |
||||||
|
#define GRPC_IMPL_CODEGEN_BYTE_BUFFER_READER_H |
||||||
|
|
||||||
|
#include <grpc/impl/codegen/byte_buffer.h> |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
|
||||||
|
struct grpc_byte_buffer_reader { |
||||||
|
grpc_byte_buffer *buffer_in; |
||||||
|
grpc_byte_buffer *buffer_out; |
||||||
|
/* Different current objects correspond to different types of byte buffers */ |
||||||
|
union { |
||||||
|
/* Index into a slice buffer's array of slices */ |
||||||
|
unsigned index; |
||||||
|
} current; |
||||||
|
}; |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
} |
||||||
|
#endif |
||||||
|
|
||||||
|
#endif /* GRPC_IMPL_CODEGEN_BYTE_BUFFER_READER_H */ |
||||||
@ -1,182 +0,0 @@ |
|||||||
/*
|
|
||||||
* |
|
||||||
* Copyright 2016, 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. |
|
||||||
* |
|
||||||
*/ |
|
||||||
/* Automatically generated nanopb header */ |
|
||||||
/* Generated by nanopb-0.3.5-dev */ |
|
||||||
|
|
||||||
#ifndef PB_LOAD_BALANCER_PB_H_INCLUDED |
|
||||||
#define PB_LOAD_BALANCER_PB_H_INCLUDED |
|
||||||
#include "third_party/nanopb/pb.h" |
|
||||||
#if PB_PROTO_HEADER_VERSION != 30 |
|
||||||
#error Regenerate this file with the current version of nanopb generator. |
|
||||||
#endif |
|
||||||
|
|
||||||
#ifdef __cplusplus |
|
||||||
extern "C" { |
|
||||||
#endif |
|
||||||
|
|
||||||
/* Struct definitions */ |
|
||||||
typedef struct _grpc_lb_v0_ClientStats { |
|
||||||
bool has_total_requests; |
|
||||||
int64_t total_requests; |
|
||||||
bool has_client_rpc_errors; |
|
||||||
int64_t client_rpc_errors; |
|
||||||
bool has_dropped_requests; |
|
||||||
int64_t dropped_requests; |
|
||||||
} grpc_lb_v0_ClientStats; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_Duration { |
|
||||||
bool has_seconds; |
|
||||||
int64_t seconds; |
|
||||||
bool has_nanos; |
|
||||||
int32_t nanos; |
|
||||||
} grpc_lb_v0_Duration; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_InitialLoadBalanceRequest { |
|
||||||
bool has_name; |
|
||||||
char name[128]; |
|
||||||
} grpc_lb_v0_InitialLoadBalanceRequest; |
|
||||||
|
|
||||||
typedef PB_BYTES_ARRAY_T(64) grpc_lb_v0_Server_load_balance_token_t; |
|
||||||
typedef struct _grpc_lb_v0_Server { |
|
||||||
bool has_ip_address; |
|
||||||
char ip_address[46]; |
|
||||||
bool has_port; |
|
||||||
int32_t port; |
|
||||||
bool has_load_balance_token; |
|
||||||
grpc_lb_v0_Server_load_balance_token_t load_balance_token; |
|
||||||
bool has_drop_request; |
|
||||||
bool drop_request; |
|
||||||
} grpc_lb_v0_Server; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_InitialLoadBalanceResponse { |
|
||||||
bool has_client_config; |
|
||||||
char client_config[64]; |
|
||||||
bool has_load_balancer_delegate; |
|
||||||
char load_balancer_delegate[64]; |
|
||||||
bool has_client_stats_report_interval; |
|
||||||
grpc_lb_v0_Duration client_stats_report_interval; |
|
||||||
} grpc_lb_v0_InitialLoadBalanceResponse; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_LoadBalanceRequest { |
|
||||||
bool has_initial_request; |
|
||||||
grpc_lb_v0_InitialLoadBalanceRequest initial_request; |
|
||||||
bool has_client_stats; |
|
||||||
grpc_lb_v0_ClientStats client_stats; |
|
||||||
} grpc_lb_v0_LoadBalanceRequest; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_ServerList { |
|
||||||
pb_callback_t servers; |
|
||||||
bool has_expiration_interval; |
|
||||||
grpc_lb_v0_Duration expiration_interval; |
|
||||||
} grpc_lb_v0_ServerList; |
|
||||||
|
|
||||||
typedef struct _grpc_lb_v0_LoadBalanceResponse { |
|
||||||
bool has_initial_response; |
|
||||||
grpc_lb_v0_InitialLoadBalanceResponse initial_response; |
|
||||||
bool has_server_list; |
|
||||||
grpc_lb_v0_ServerList server_list; |
|
||||||
} grpc_lb_v0_LoadBalanceResponse; |
|
||||||
|
|
||||||
/* Default values for struct fields */ |
|
||||||
|
|
||||||
/* Initializer values for message structs */ |
|
||||||
#define grpc_lb_v0_Duration_init_default {false, 0, false, 0} |
|
||||||
#define grpc_lb_v0_LoadBalanceRequest_init_default {false, grpc_lb_v0_InitialLoadBalanceRequest_init_default, false, grpc_lb_v0_ClientStats_init_default} |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceRequest_init_default {false, ""} |
|
||||||
#define grpc_lb_v0_ClientStats_init_default {false, 0, false, 0, false, 0} |
|
||||||
#define grpc_lb_v0_LoadBalanceResponse_init_default {false, grpc_lb_v0_InitialLoadBalanceResponse_init_default, false, grpc_lb_v0_ServerList_init_default} |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_init_default {false, "", false, "", false, grpc_lb_v0_Duration_init_default} |
|
||||||
#define grpc_lb_v0_ServerList_init_default {{{NULL}, NULL}, false, grpc_lb_v0_Duration_init_default} |
|
||||||
#define grpc_lb_v0_Server_init_default {false, "", false, 0, false, {0, {0}}, false, 0} |
|
||||||
#define grpc_lb_v0_Duration_init_zero {false, 0, false, 0} |
|
||||||
#define grpc_lb_v0_LoadBalanceRequest_init_zero {false, grpc_lb_v0_InitialLoadBalanceRequest_init_zero, false, grpc_lb_v0_ClientStats_init_zero} |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceRequest_init_zero {false, ""} |
|
||||||
#define grpc_lb_v0_ClientStats_init_zero {false, 0, false, 0, false, 0} |
|
||||||
#define grpc_lb_v0_LoadBalanceResponse_init_zero {false, grpc_lb_v0_InitialLoadBalanceResponse_init_zero, false, grpc_lb_v0_ServerList_init_zero} |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_init_zero {false, "", false, "", false, grpc_lb_v0_Duration_init_zero} |
|
||||||
#define grpc_lb_v0_ServerList_init_zero {{{NULL}, NULL}, false, grpc_lb_v0_Duration_init_zero} |
|
||||||
#define grpc_lb_v0_Server_init_zero {false, "", false, 0, false, {0, {0}}, false, 0} |
|
||||||
|
|
||||||
/* Field tags (for use in manual encoding/decoding) */ |
|
||||||
#define grpc_lb_v0_ClientStats_total_requests_tag 1 |
|
||||||
#define grpc_lb_v0_ClientStats_client_rpc_errors_tag 2 |
|
||||||
#define grpc_lb_v0_ClientStats_dropped_requests_tag 3 |
|
||||||
#define grpc_lb_v0_Duration_seconds_tag 1 |
|
||||||
#define grpc_lb_v0_Duration_nanos_tag 2 |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceRequest_name_tag 1 |
|
||||||
#define grpc_lb_v0_Server_ip_address_tag 1 |
|
||||||
#define grpc_lb_v0_Server_port_tag 2 |
|
||||||
#define grpc_lb_v0_Server_load_balance_token_tag 3 |
|
||||||
#define grpc_lb_v0_Server_drop_request_tag 4 |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_client_config_tag 1 |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_load_balancer_delegate_tag 2 |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_client_stats_report_interval_tag 3 |
|
||||||
#define grpc_lb_v0_LoadBalanceRequest_initial_request_tag 1 |
|
||||||
#define grpc_lb_v0_LoadBalanceRequest_client_stats_tag 2 |
|
||||||
#define grpc_lb_v0_ServerList_servers_tag 1 |
|
||||||
#define grpc_lb_v0_ServerList_expiration_interval_tag 3 |
|
||||||
#define grpc_lb_v0_LoadBalanceResponse_initial_response_tag 1 |
|
||||||
#define grpc_lb_v0_LoadBalanceResponse_server_list_tag 2 |
|
||||||
|
|
||||||
/* Struct field encoding specification for nanopb */ |
|
||||||
extern const pb_field_t grpc_lb_v0_Duration_fields[3]; |
|
||||||
extern const pb_field_t grpc_lb_v0_LoadBalanceRequest_fields[3]; |
|
||||||
extern const pb_field_t grpc_lb_v0_InitialLoadBalanceRequest_fields[2]; |
|
||||||
extern const pb_field_t grpc_lb_v0_ClientStats_fields[4]; |
|
||||||
extern const pb_field_t grpc_lb_v0_LoadBalanceResponse_fields[3]; |
|
||||||
extern const pb_field_t grpc_lb_v0_InitialLoadBalanceResponse_fields[4]; |
|
||||||
extern const pb_field_t grpc_lb_v0_ServerList_fields[3]; |
|
||||||
extern const pb_field_t grpc_lb_v0_Server_fields[5]; |
|
||||||
|
|
||||||
/* Maximum encoded size of messages (where known) */ |
|
||||||
#define grpc_lb_v0_Duration_size 22 |
|
||||||
#define grpc_lb_v0_LoadBalanceRequest_size 169 |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceRequest_size 131 |
|
||||||
#define grpc_lb_v0_ClientStats_size 33 |
|
||||||
#define grpc_lb_v0_LoadBalanceResponse_size (165 + grpc_lb_v0_ServerList_size) |
|
||||||
#define grpc_lb_v0_InitialLoadBalanceResponse_size 156 |
|
||||||
#define grpc_lb_v0_Server_size 127 |
|
||||||
|
|
||||||
/* Message IDs (where set with "msgid" option) */ |
|
||||||
#ifdef PB_MSGID |
|
||||||
|
|
||||||
#define LOAD_BALANCER_MESSAGES \ |
|
||||||
|
|
||||||
|
|
||||||
#endif |
|
||||||
|
|
||||||
#ifdef __cplusplus |
|
||||||
} /* extern "C" */ |
|
||||||
#endif |
|
||||||
|
|
||||||
#endif |
|
||||||
@ -0,0 +1,178 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
/* Automatically generated nanopb header */ |
||||||
|
/* Generated by nanopb-0.3.5-dev */ |
||||||
|
|
||||||
|
#ifndef PB_LOAD_BALANCER_PB_H_INCLUDED |
||||||
|
#define PB_LOAD_BALANCER_PB_H_INCLUDED |
||||||
|
#include "third_party/nanopb/pb.h" |
||||||
|
#if PB_PROTO_HEADER_VERSION != 30 |
||||||
|
#error Regenerate this file with the current version of nanopb generator. |
||||||
|
#endif |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
|
||||||
|
/* Struct definitions */ |
||||||
|
typedef struct _grpc_lb_v1_ClientStats { |
||||||
|
bool has_total_requests; |
||||||
|
int64_t total_requests; |
||||||
|
bool has_client_rpc_errors; |
||||||
|
int64_t client_rpc_errors; |
||||||
|
bool has_dropped_requests; |
||||||
|
int64_t dropped_requests; |
||||||
|
} grpc_lb_v1_ClientStats; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_Duration { |
||||||
|
bool has_seconds; |
||||||
|
int64_t seconds; |
||||||
|
bool has_nanos; |
||||||
|
int32_t nanos; |
||||||
|
} grpc_lb_v1_Duration; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_InitialLoadBalanceRequest { |
||||||
|
bool has_name; |
||||||
|
char name[128]; |
||||||
|
} grpc_lb_v1_InitialLoadBalanceRequest; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_Server { |
||||||
|
bool has_ip_address; |
||||||
|
char ip_address[46]; |
||||||
|
bool has_port; |
||||||
|
int32_t port; |
||||||
|
bool has_load_balance_token; |
||||||
|
char load_balance_token[64]; |
||||||
|
bool has_drop_request; |
||||||
|
bool drop_request; |
||||||
|
} grpc_lb_v1_Server; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_InitialLoadBalanceResponse { |
||||||
|
bool has_load_balancer_delegate; |
||||||
|
char load_balancer_delegate[64]; |
||||||
|
bool has_client_stats_report_interval; |
||||||
|
grpc_lb_v1_Duration client_stats_report_interval; |
||||||
|
} grpc_lb_v1_InitialLoadBalanceResponse; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_LoadBalanceRequest { |
||||||
|
bool has_initial_request; |
||||||
|
grpc_lb_v1_InitialLoadBalanceRequest initial_request; |
||||||
|
bool has_client_stats; |
||||||
|
grpc_lb_v1_ClientStats client_stats; |
||||||
|
} grpc_lb_v1_LoadBalanceRequest; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_ServerList { |
||||||
|
pb_callback_t servers; |
||||||
|
bool has_expiration_interval; |
||||||
|
grpc_lb_v1_Duration expiration_interval; |
||||||
|
} grpc_lb_v1_ServerList; |
||||||
|
|
||||||
|
typedef struct _grpc_lb_v1_LoadBalanceResponse { |
||||||
|
bool has_initial_response; |
||||||
|
grpc_lb_v1_InitialLoadBalanceResponse initial_response; |
||||||
|
bool has_server_list; |
||||||
|
grpc_lb_v1_ServerList server_list; |
||||||
|
} grpc_lb_v1_LoadBalanceResponse; |
||||||
|
|
||||||
|
/* Default values for struct fields */ |
||||||
|
|
||||||
|
/* Initializer values for message structs */ |
||||||
|
#define grpc_lb_v1_Duration_init_default {false, 0, false, 0} |
||||||
|
#define grpc_lb_v1_LoadBalanceRequest_init_default {false, grpc_lb_v1_InitialLoadBalanceRequest_init_default, false, grpc_lb_v1_ClientStats_init_default} |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceRequest_init_default {false, ""} |
||||||
|
#define grpc_lb_v1_ClientStats_init_default {false, 0, false, 0, false, 0} |
||||||
|
#define grpc_lb_v1_LoadBalanceResponse_init_default {false, grpc_lb_v1_InitialLoadBalanceResponse_init_default, false, grpc_lb_v1_ServerList_init_default} |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceResponse_init_default {false, "", false, grpc_lb_v1_Duration_init_default} |
||||||
|
#define grpc_lb_v1_ServerList_init_default {{{NULL}, NULL}, false, grpc_lb_v1_Duration_init_default} |
||||||
|
#define grpc_lb_v1_Server_init_default {false, "", false, 0, false, "", false, 0} |
||||||
|
#define grpc_lb_v1_Duration_init_zero {false, 0, false, 0} |
||||||
|
#define grpc_lb_v1_LoadBalanceRequest_init_zero {false, grpc_lb_v1_InitialLoadBalanceRequest_init_zero, false, grpc_lb_v1_ClientStats_init_zero} |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceRequest_init_zero {false, ""} |
||||||
|
#define grpc_lb_v1_ClientStats_init_zero {false, 0, false, 0, false, 0} |
||||||
|
#define grpc_lb_v1_LoadBalanceResponse_init_zero {false, grpc_lb_v1_InitialLoadBalanceResponse_init_zero, false, grpc_lb_v1_ServerList_init_zero} |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceResponse_init_zero {false, "", false, grpc_lb_v1_Duration_init_zero} |
||||||
|
#define grpc_lb_v1_ServerList_init_zero {{{NULL}, NULL}, false, grpc_lb_v1_Duration_init_zero} |
||||||
|
#define grpc_lb_v1_Server_init_zero {false, "", false, 0, false, "", false, 0} |
||||||
|
|
||||||
|
/* Field tags (for use in manual encoding/decoding) */ |
||||||
|
#define grpc_lb_v1_ClientStats_total_requests_tag 1 |
||||||
|
#define grpc_lb_v1_ClientStats_client_rpc_errors_tag 2 |
||||||
|
#define grpc_lb_v1_ClientStats_dropped_requests_tag 3 |
||||||
|
#define grpc_lb_v1_Duration_seconds_tag 1 |
||||||
|
#define grpc_lb_v1_Duration_nanos_tag 2 |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceRequest_name_tag 1 |
||||||
|
#define grpc_lb_v1_Server_ip_address_tag 1 |
||||||
|
#define grpc_lb_v1_Server_port_tag 2 |
||||||
|
#define grpc_lb_v1_Server_load_balance_token_tag 3 |
||||||
|
#define grpc_lb_v1_Server_drop_request_tag 4 |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceResponse_load_balancer_delegate_tag 2 |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceResponse_client_stats_report_interval_tag 3 |
||||||
|
#define grpc_lb_v1_LoadBalanceRequest_initial_request_tag 1 |
||||||
|
#define grpc_lb_v1_LoadBalanceRequest_client_stats_tag 2 |
||||||
|
#define grpc_lb_v1_ServerList_servers_tag 1 |
||||||
|
#define grpc_lb_v1_ServerList_expiration_interval_tag 3 |
||||||
|
#define grpc_lb_v1_LoadBalanceResponse_initial_response_tag 1 |
||||||
|
#define grpc_lb_v1_LoadBalanceResponse_server_list_tag 2 |
||||||
|
|
||||||
|
/* Struct field encoding specification for nanopb */ |
||||||
|
extern const pb_field_t grpc_lb_v1_Duration_fields[3]; |
||||||
|
extern const pb_field_t grpc_lb_v1_LoadBalanceRequest_fields[3]; |
||||||
|
extern const pb_field_t grpc_lb_v1_InitialLoadBalanceRequest_fields[2]; |
||||||
|
extern const pb_field_t grpc_lb_v1_ClientStats_fields[4]; |
||||||
|
extern const pb_field_t grpc_lb_v1_LoadBalanceResponse_fields[3]; |
||||||
|
extern const pb_field_t grpc_lb_v1_InitialLoadBalanceResponse_fields[3]; |
||||||
|
extern const pb_field_t grpc_lb_v1_ServerList_fields[3]; |
||||||
|
extern const pb_field_t grpc_lb_v1_Server_fields[5]; |
||||||
|
|
||||||
|
/* Maximum encoded size of messages (where known) */ |
||||||
|
#define grpc_lb_v1_Duration_size 22 |
||||||
|
#define grpc_lb_v1_LoadBalanceRequest_size 169 |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceRequest_size 131 |
||||||
|
#define grpc_lb_v1_ClientStats_size 33 |
||||||
|
#define grpc_lb_v1_LoadBalanceResponse_size (98 + grpc_lb_v1_ServerList_size) |
||||||
|
#define grpc_lb_v1_InitialLoadBalanceResponse_size 90 |
||||||
|
#define grpc_lb_v1_Server_size 127 |
||||||
|
|
||||||
|
/* Message IDs (where set with "msgid" option) */ |
||||||
|
#ifdef PB_MSGID |
||||||
|
|
||||||
|
#define LOAD_BALANCER_MESSAGES \ |
||||||
|
|
||||||
|
|
||||||
|
#endif |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
} /* extern "C" */ |
||||||
|
#endif |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,191 @@ |
|||||||
|
#region Copyright notice and license |
||||||
|
|
||||||
|
// 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. |
||||||
|
|
||||||
|
#endregion |
||||||
|
|
||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Runtime.InteropServices; |
||||||
|
using System.Threading.Tasks; |
||||||
|
|
||||||
|
using Grpc.Core.Internal; |
||||||
|
using NUnit.Framework; |
||||||
|
|
||||||
|
namespace Grpc.Core.Internal.Tests |
||||||
|
{ |
||||||
|
/// <summary> |
||||||
|
/// Uses fake native call to test interaction of <c>AsyncCallServer</c> wrapping code with C core in different situations. |
||||||
|
/// </summary> |
||||||
|
public class AsyncCallServerTest |
||||||
|
{ |
||||||
|
Server server; |
||||||
|
FakeNativeCall fakeCall; |
||||||
|
AsyncCallServer<string, string> asyncCallServer; |
||||||
|
|
||||||
|
[SetUp] |
||||||
|
public void Init() |
||||||
|
{ |
||||||
|
var environment = GrpcEnvironment.AddRef(); |
||||||
|
|
||||||
|
// Create a fake server just so we have an instance to refer to. |
||||||
|
// The server won't actually be used at all. |
||||||
|
server = new Server() |
||||||
|
{ |
||||||
|
Ports = { { "localhost", 0, ServerCredentials.Insecure } } |
||||||
|
}; |
||||||
|
server.Start(); |
||||||
|
|
||||||
|
fakeCall = new FakeNativeCall(); |
||||||
|
asyncCallServer = new AsyncCallServer<string, string>( |
||||||
|
Marshallers.StringMarshaller.Serializer, Marshallers.StringMarshaller.Deserializer, |
||||||
|
environment, |
||||||
|
server); |
||||||
|
asyncCallServer.InitializeForTesting(fakeCall); |
||||||
|
} |
||||||
|
|
||||||
|
[TearDown] |
||||||
|
public void Cleanup() |
||||||
|
{ |
||||||
|
server.ShutdownAsync().Wait(); |
||||||
|
GrpcEnvironment.Release(); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void CancelNotificationAfterStartDisposes() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void CancelNotificationAfterStartDisposesAfterPendingReadFinishes() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var requestStream = new ServerRequestStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
var moveNextTask = requestStream.MoveNext(); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
fakeCall.ReceivedMessageHandler(true, null); |
||||||
|
Assert.IsFalse(moveNextTask.Result); |
||||||
|
|
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void ReadAfterCancelNotificationCanSucceed() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var requestStream = new ServerRequestStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
|
||||||
|
// Check that starting a read after cancel notification has been processed is legal. |
||||||
|
var moveNextTask = requestStream.MoveNext(); |
||||||
|
Assert.IsFalse(moveNextTask.Result); |
||||||
|
|
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void ReadCompletionFailureClosesRequestStream() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var requestStream = new ServerRequestStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
// if a read completion's success==false, the request stream will silently finish |
||||||
|
// and we rely on C core cancelling the call. |
||||||
|
var moveNextTask = requestStream.MoveNext(); |
||||||
|
fakeCall.ReceivedMessageHandler(false, null); |
||||||
|
Assert.IsFalse(moveNextTask.Result); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void WriteAfterCancelNotificationFails() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var requestStream = new ServerRequestStream<string, string>(asyncCallServer); |
||||||
|
var responseStream = new ServerResponseStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
|
||||||
|
// TODO(jtattermusch): should we throw a different exception type instead? |
||||||
|
Assert.Throws(typeof(InvalidOperationException), () => responseStream.WriteAsync("request1")); |
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void WriteCompletionFailureThrows() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var responseStream = new ServerResponseStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
var writeTask = responseStream.WriteAsync("request1"); |
||||||
|
fakeCall.SendCompletionHandler(false); |
||||||
|
// TODO(jtattermusch): should we throw a different exception type instead? |
||||||
|
Assert.ThrowsAsync(typeof(InvalidOperationException), async () => await writeTask); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
[Test] |
||||||
|
public void WriteAndWriteStatusCanRunConcurrently() |
||||||
|
{ |
||||||
|
var finishedTask = asyncCallServer.ServerSideCallAsync(); |
||||||
|
var responseStream = new ServerResponseStream<string, string>(asyncCallServer); |
||||||
|
|
||||||
|
var writeTask = responseStream.WriteAsync("request1"); |
||||||
|
var writeStatusTask = asyncCallServer.SendStatusFromServerAsync(Status.DefaultSuccess, new Metadata(), null); |
||||||
|
|
||||||
|
fakeCall.SendCompletionHandler(true); |
||||||
|
fakeCall.SendStatusFromServerHandler(true); |
||||||
|
|
||||||
|
Assert.DoesNotThrowAsync(async () => await writeTask); |
||||||
|
Assert.DoesNotThrowAsync(async () => await writeStatusTask); |
||||||
|
|
||||||
|
fakeCall.ReceivedCloseOnServerHandler(true, cancelled: true); |
||||||
|
|
||||||
|
AssertFinished(asyncCallServer, fakeCall, finishedTask); |
||||||
|
} |
||||||
|
|
||||||
|
static void AssertFinished(AsyncCallServer<string, string> asyncCallServer, FakeNativeCall fakeCall, Task finishedTask) |
||||||
|
{ |
||||||
|
Assert.IsTrue(fakeCall.IsDisposed); |
||||||
|
Assert.IsTrue(finishedTask.IsCompleted); |
||||||
|
Assert.DoesNotThrow(() => finishedTask.Wait()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,184 @@ |
|||||||
|
#region Copyright notice and license |
||||||
|
|
||||||
|
// 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. |
||||||
|
|
||||||
|
#endregion |
||||||
|
|
||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Runtime.InteropServices; |
||||||
|
using System.Threading.Tasks; |
||||||
|
|
||||||
|
using Grpc.Core.Internal; |
||||||
|
using NUnit.Framework; |
||||||
|
|
||||||
|
namespace Grpc.Core.Internal.Tests |
||||||
|
{ |
||||||
|
/// <summary> |
||||||
|
/// For testing purposes. |
||||||
|
/// </summary> |
||||||
|
internal class FakeNativeCall : INativeCall |
||||||
|
{ |
||||||
|
public UnaryResponseClientHandler UnaryResponseClientHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public ReceivedStatusOnClientHandler ReceivedStatusOnClientHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public ReceivedMessageHandler ReceivedMessageHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public ReceivedResponseHeadersHandler ReceivedResponseHeadersHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public SendCompletionHandler SendCompletionHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public SendCompletionHandler SendStatusFromServerHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public ReceivedCloseOnServerHandler ReceivedCloseOnServerHandler |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public bool IsCancelled |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public bool IsDisposed |
||||||
|
{ |
||||||
|
get; |
||||||
|
set; |
||||||
|
} |
||||||
|
|
||||||
|
public void Cancel() |
||||||
|
{ |
||||||
|
IsCancelled = true; |
||||||
|
} |
||||||
|
|
||||||
|
public void CancelWithStatus(Status status) |
||||||
|
{ |
||||||
|
IsCancelled = true; |
||||||
|
} |
||||||
|
|
||||||
|
public string GetPeer() |
||||||
|
{ |
||||||
|
return "PEER"; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartUnary(UnaryResponseClientHandler callback, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags) |
||||||
|
{ |
||||||
|
UnaryResponseClientHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartUnary(BatchContextSafeHandle ctx, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags) |
||||||
|
{ |
||||||
|
throw new NotImplementedException(); |
||||||
|
} |
||||||
|
|
||||||
|
public void StartClientStreaming(UnaryResponseClientHandler callback, MetadataArraySafeHandle metadataArray) |
||||||
|
{ |
||||||
|
UnaryResponseClientHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartServerStreaming(ReceivedStatusOnClientHandler callback, byte[] payload, MetadataArraySafeHandle metadataArray, WriteFlags writeFlags) |
||||||
|
{ |
||||||
|
ReceivedStatusOnClientHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartDuplexStreaming(ReceivedStatusOnClientHandler callback, MetadataArraySafeHandle metadataArray) |
||||||
|
{ |
||||||
|
ReceivedStatusOnClientHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartReceiveMessage(ReceivedMessageHandler callback) |
||||||
|
{ |
||||||
|
ReceivedMessageHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartReceiveInitialMetadata(ReceivedResponseHeadersHandler callback) |
||||||
|
{ |
||||||
|
ReceivedResponseHeadersHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartSendInitialMetadata(SendCompletionHandler callback, MetadataArraySafeHandle metadataArray) |
||||||
|
{ |
||||||
|
SendCompletionHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartSendMessage(SendCompletionHandler callback, byte[] payload, WriteFlags writeFlags, bool sendEmptyInitialMetadata) |
||||||
|
{ |
||||||
|
SendCompletionHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartSendCloseFromClient(SendCompletionHandler callback) |
||||||
|
{ |
||||||
|
SendCompletionHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartSendStatusFromServer(SendCompletionHandler callback, Status status, MetadataArraySafeHandle metadataArray, bool sendEmptyInitialMetadata, |
||||||
|
byte[] optionalPayload, WriteFlags writeFlags) |
||||||
|
{ |
||||||
|
SendStatusFromServerHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void StartServerSide(ReceivedCloseOnServerHandler callback) |
||||||
|
{ |
||||||
|
ReceivedCloseOnServerHandler = callback; |
||||||
|
} |
||||||
|
|
||||||
|
public void Dispose() |
||||||
|
{ |
||||||
|
IsDisposed = true; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,49 @@ |
|||||||
|
<?php |
||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2016, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
$args = getopt('', ['metrics_server_address:', 'total_only::']); |
||||||
|
$parts = explode(':', $args['metrics_server_address']); |
||||||
|
$server_host = $parts[0]; |
||||||
|
$server_port = (count($parts) == 2) ? $parts[1] : ''; |
||||||
|
|
||||||
|
$socket = socket_create(AF_INET, SOCK_STREAM, 0); |
||||||
|
if (@!socket_connect($socket, $server_host, $server_port)) { |
||||||
|
echo "Cannot connect to merics server...\n"; |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
socket_write($socket, 'qps'); |
||||||
|
while ($out = socket_read($socket, 1024)) { |
||||||
|
echo "$out\n"; |
||||||
|
} |
||||||
|
socket_close($socket); |
||||||
@ -0,0 +1,116 @@ |
|||||||
|
<?php |
||||||
|
/* |
||||||
|
* |
||||||
|
* Copyright 2016, 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_once('interop_client.php'); |
||||||
|
|
||||||
|
function stress_main($args) { |
||||||
|
mt_srand(); |
||||||
|
set_time_limit(0); |
||||||
|
|
||||||
|
// open socket to listen as metrics server |
||||||
|
$socket = socket_create(AF_INET, SOCK_STREAM, 0); |
||||||
|
socket_set_option($socket, SOL_SOCKET, SO_REUSEADDR, 1); |
||||||
|
if (@!socket_bind($socket, 'localhost', $args['metrics_port'])) { |
||||||
|
echo "Cannot create socket for metrics server...\n"; |
||||||
|
exit(1); |
||||||
|
} |
||||||
|
socket_listen($socket); |
||||||
|
socket_set_nonblock($socket); |
||||||
|
|
||||||
|
$start_time = microtime(true); |
||||||
|
$count = 0; |
||||||
|
$deadline = $args['test_duration_secs'] ? |
||||||
|
($start_time + $args['test_duration_secs']) : false; |
||||||
|
$num_test_cases = count($args['test_cases']); |
||||||
|
$stub = false; |
||||||
|
|
||||||
|
while (true) { |
||||||
|
$current_time = microtime(true); |
||||||
|
if ($deadline && $current_time > $deadline) { |
||||||
|
break; |
||||||
|
} |
||||||
|
if ($client_connection = socket_accept($socket)) { |
||||||
|
// there is an incoming request, respond with qps metrics |
||||||
|
$input = socket_read($client_connection, 1024); |
||||||
|
$qps = round($count / ($current_time - $start_time)); |
||||||
|
socket_write($client_connection, "qps: $qps"); |
||||||
|
socket_close($client_connection); |
||||||
|
} else { |
||||||
|
// do actual work, run one interop test case |
||||||
|
$args['test_case'] = |
||||||
|
$args['test_cases'][mt_rand(0, $num_test_cases - 1)]; |
||||||
|
$stub = @interop_main($args, $stub); |
||||||
|
$count++; |
||||||
|
} |
||||||
|
} |
||||||
|
socket_close($socket); |
||||||
|
echo "Number of interop tests run in $args[test_duration_secs] seconds: $count.\n"; |
||||||
|
} |
||||||
|
|
||||||
|
// process command line arguments |
||||||
|
$raw_args = getopt('', |
||||||
|
['server_addresses::', |
||||||
|
'test_cases:', |
||||||
|
'metrics_port::', |
||||||
|
'test_duration_secs::', |
||||||
|
'num_channels_per_server::', |
||||||
|
'num_stubs_per_channel::']); |
||||||
|
|
||||||
|
$args = []; |
||||||
|
|
||||||
|
if (empty($raw_args['server_addresses'])) { |
||||||
|
$args['server_host'] = 'localhost'; |
||||||
|
$args['server_port'] = '8080'; |
||||||
|
} else { |
||||||
|
$parts = explode(':', $raw_args['server_addresses']); |
||||||
|
$args['server_host'] = $parts[0]; |
||||||
|
$args['server_port'] = (count($parts) == 2) ? $parts[1] : ''; |
||||||
|
} |
||||||
|
|
||||||
|
$args['metrics_port'] = empty($raw_args['metrics_port']) ? |
||||||
|
'8081' : $args['metrics_port']; |
||||||
|
|
||||||
|
$args['test_duration_secs'] = empty($raw_args['test_duration_secs']) || |
||||||
|
$raw_args['test_duration_secs'] == -1 ? |
||||||
|
false : $raw_args['test_duration_secs']; |
||||||
|
|
||||||
|
$test_cases = []; |
||||||
|
$test_case_strs = explode(',', $raw_args['test_cases']); |
||||||
|
foreach ($test_case_strs as $test_case_str) { |
||||||
|
$parts = explode(':', $test_case_str); |
||||||
|
$test_cases = array_merge($test_cases, array_fill(0, $parts[1], $parts[0])); |
||||||
|
} |
||||||
|
$args['test_cases'] = $test_cases; |
||||||
|
|
||||||
|
stress_main($args); |
||||||
@ -1,6 +0,0 @@ |
|||||||
grpc.lb.v0.InitialLoadBalanceRequest.name max_size:128 |
|
||||||
grpc.lb.v0.InitialLoadBalanceResponse.client_config max_size:64 |
|
||||||
grpc.lb.v0.InitialLoadBalanceResponse.load_balancer_delegate max_size:64 |
|
||||||
grpc.lb.v0.Server.ip_address max_size:46 |
|
||||||
grpc.lb.v0.Server.load_balance_token max_size:64 |
|
||||||
load_balancer.proto no_unions:true |
|
||||||
@ -0,0 +1,6 @@ |
|||||||
|
grpc.lb.v1.InitialLoadBalanceRequest.name max_size:128 |
||||||
|
grpc.lb.v1.InitialLoadBalanceResponse.client_config max_size:64 |
||||||
|
grpc.lb.v1.InitialLoadBalanceResponse.load_balancer_delegate max_size:64 |
||||||
|
grpc.lb.v1.Server.ip_address max_size:46 |
||||||
|
grpc.lb.v1.Server.load_balance_token max_size:64 |
||||||
|
load_balancer.proto no_unions:true |
||||||
@ -1,114 +0,0 @@ |
|||||||
# 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. |
|
||||||
|
|
||||||
import os |
|
||||||
import platform |
|
||||||
import shutil |
|
||||||
import sys |
|
||||||
import sysconfig |
|
||||||
|
|
||||||
import setuptools |
|
||||||
|
|
||||||
import commands |
|
||||||
import grpc_version |
|
||||||
|
|
||||||
try: |
|
||||||
from urllib2 import urlopen |
|
||||||
except ImportError: |
|
||||||
from urllib.request import urlopen |
|
||||||
|
|
||||||
PYTHON_STEM = os.path.dirname(os.path.abspath(__file__)) |
|
||||||
BINARIES_REPOSITORY = os.environ.get( |
|
||||||
'GRPC_PYTHON_BINARIES_REPOSITORY', |
|
||||||
'https://storage.googleapis.com/grpc-precompiled-binaries/python') |
|
||||||
USE_PRECOMPILED_BINARIES = bool(int(os.environ.get( |
|
||||||
'GRPC_PYTHON_USE_PRECOMPILED_BINARIES', '1'))) |
|
||||||
|
|
||||||
def _tagged_ext_name(base): |
|
||||||
uname = platform.uname() |
|
||||||
tags = ( |
|
||||||
grpc_version.VERSION, |
|
||||||
'py{}'.format(sysconfig.get_python_version()), |
|
||||||
uname[0], |
|
||||||
uname[4], |
|
||||||
) |
|
||||||
ucs = 'ucs{}'.format(sysconfig.get_config_var('Py_UNICODE_SIZE')) |
|
||||||
return '{base}-{tags}-{ucs}'.format( |
|
||||||
base=base, tags='-'.join(tags), ucs=ucs) |
|
||||||
|
|
||||||
|
|
||||||
class BuildTaggedExt(setuptools.Command): |
|
||||||
|
|
||||||
description = 'build the gRPC tagged extensions' |
|
||||||
user_options = [] |
|
||||||
|
|
||||||
def initialize_options(self): |
|
||||||
# distutils requires this override. |
|
||||||
pass |
|
||||||
|
|
||||||
def finalize_options(self): |
|
||||||
# distutils requires this override. |
|
||||||
pass |
|
||||||
|
|
||||||
def run(self): |
|
||||||
if 'linux' in sys.platform: |
|
||||||
self.run_command('build_ext') |
|
||||||
try: |
|
||||||
os.makedirs('dist/') |
|
||||||
except OSError: |
|
||||||
pass |
|
||||||
shutil.copyfile( |
|
||||||
os.path.join(PYTHON_STEM, 'grpc/_cython/cygrpc.so'), |
|
||||||
'dist/{}.so'.format(_tagged_ext_name('cygrpc'))) |
|
||||||
else: |
|
||||||
sys.stderr.write('nothing to do for build_tagged_ext\n') |
|
||||||
|
|
||||||
|
|
||||||
def update_setup_arguments(setup_arguments): |
|
||||||
if not USE_PRECOMPILED_BINARIES: |
|
||||||
sys.stderr.write('not using precompiled extension') |
|
||||||
return |
|
||||||
url = '{}/{}.so'.format(BINARIES_REPOSITORY, _tagged_ext_name('cygrpc')) |
|
||||||
target_path = os.path.join(PYTHON_STEM, 'grpc/_cython/cygrpc.so') |
|
||||||
try: |
|
||||||
extension = urlopen(url).read() |
|
||||||
except: |
|
||||||
sys.stderr.write( |
|
||||||
'could not download precompiled extension: {}\n'.format(url)) |
|
||||||
return |
|
||||||
try: |
|
||||||
with open(target_path, 'w') as target: |
|
||||||
target.write(extension) |
|
||||||
setup_arguments['ext_modules'] = [] |
|
||||||
except: |
|
||||||
sys.stderr.write( |
|
||||||
'could not write precompiled extension to directory: {} -> {}\n' |
|
||||||
.format(url, target_path)) |
|
||||||
return |
|
||||||
setup_arguments['package_data']['grpc._cython'].append('cygrpc.so') |
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue