Function main

example/streaming-server.cpp:244–295

using ShutdownRPC = agrpc::ServerRPC<&ExampleExtService::RequestShutdown>;

int main(int argc, const char** argv)
{
    const char* port = argc >= 2 ? argv[1] : "50051";
    const auto host = std::string("0.0.0.0:") + port;

    std::unique_ptr<grpc::Server> server;

    grpc::ServerBuilder builder;
    agrpc::GrpcContext grpc_context{builder.AddCompletionQueue()};
    builder.AddListeningPort(host, grpc::InsecureServerCredentials());
    ExampleService service;
    builder.RegisterService(&service);
    ExampleExtService service_ext;
    builder.RegisterService(&service_ext);
    server = builder.BuildAndStart();
    abort_if_not(bool{server});

    example::ServerShutdown server_shutdown{*server, grpc_context};

    asio::thread_pool thread_pool{1};

    agrpc::register_awaitable_rpc_handler<ClientStreamingRPC>(grpc_context, service, &handle_client_streaming_request,
                                                              example::RethrowFirstArg{});

    agrpc::register_awaitable_rpc_handler<ServerStreamingRPC>(grpc_context, service, &handle_server_streaming_request,
                                                              example::RethrowFirstArg{});

    agrpc::register_awaitable_rpc_handler<ServerStreamingNotifyWhenDoneRPC>(
        grpc_context, service_ext, server_streaming_notify_when_done_request_handler(grpc_context),
        example::RethrowFirstArg{});

    agrpc::register_awaitable_rpc_handler<BidiStreamingRPC>(
        grpc_context, service, bidirectional_streaming_rpc_handler(thread_pool), example::RethrowFirstArg{});

    agrpc::register_awaitable_rpc_handler<SlowUnaryRPC>(grpc_context, service_ext, &handle_slow_unary_request,
                                                        example::RethrowFirstArg{});

    agrpc::register_awaitable_rpc_handler<ShutdownRPC>(
        grpc_context, service_ext,
        [&](ShutdownRPC& rpc, const ShutdownRPC::Request&) -> asio::awaitable<void>
        {
            if (co_await rpc.finish({}, grpc::Status::OK))
            {
                std::cout << "Received shutdown request from client\n";
                server_shutdown.shutdown();
            }
        },
        example::RethrowFirstArg{});

    grpc_context.run();
    std::cout << "Shutdown completed\n";
}