1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586 |
- /*
- Copyright (c) 2020 sogou, Inc.
- 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 <signal.h>
- #include "echo_pb.srpc.h"
- #include "workflow/WFFacilities.h"
- #include "srpc/rpc_types.h"
- using namespace srpc;
- static WFFacilities::WaitGroup wait_group(1);
- template<class CLIENT>
- class ExampleProxyServiceImpl : public Example::Service
- {
- public:
- ExampleProxyServiceImpl(CLIENT *client)
- {
- this->client = client;
- }
- private:
- CLIENT *client;
- public:
- void Echo(EchoRequest *request, EchoResponse *response, RPCContext *context) override
- {
- printf("Proxy Server Echo() get and transfer request:\n%s\n",
- request->DebugString().c_str());
- auto *task = this->client->create_Echo_task([response](EchoResponse *resp,
- RPCContext *ctx) {
- printf("Proxy Server Echo() get and transfer response:\n%s\n",
- resp->DebugString().c_str());
- if (ctx->success())
- *response = std::move(*resp);
- });
- task->serialize_input(request);
- context->get_series()->push_back(task);
- }
- };
- static void sig_handler(int signo)
- {
- wait_group.done();
- }
- int main()
- {
- GOOGLE_PROTOBUF_VERIFY_VERSION;
- signal(SIGINT, sig_handler);
- signal(SIGTERM, sig_handler);
- SRPCServer server;
- Example::SRPCClient client("127.0.0.1", 1412);
- ExampleProxyServiceImpl<Example::SRPCClient> impl(&client);
- server.add_service(&impl);
- if (server.start(61412) == 0)
- {
- printf("Proxy server serving on 61412 and redirect to backend server 127.0.0.1:1412\n"
- "Try start ./srpc_pb_server as backend and send requests to 61412.\n\n");
- wait_group.wait();
- server.stop();
- }
- else
- perror("server start");
- google::protobuf::ShutdownProtobufLibrary();
- return 0;
- }
|