/* * * Copyright 2015 gRPC authors. * * 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. * */ #ifndef GRPCPP_IMPL_CODEGEN_ASYNC_GENERIC_SERVICE_H #define GRPCPP_IMPL_CODEGEN_ASYNC_GENERIC_SERVICE_H // IWYU pragma: private, include <grpcpp/generic/async_generic_service.h> #include <grpc/impl/codegen/port_platform.h> #include <grpcpp/impl/codegen/async_stream.h> #include <grpcpp/impl/codegen/byte_buffer.h> #include <grpcpp/impl/codegen/server_callback.h> #include <grpcpp/impl/codegen/server_callback_handlers.h> struct grpc_server; namespace grpc { typedef ServerAsyncReaderWriter<ByteBuffer, ByteBuffer> GenericServerAsyncReaderWriter; typedef ServerAsyncResponseWriter<ByteBuffer> GenericServerAsyncResponseWriter; typedef ServerAsyncReader<ByteBuffer, ByteBuffer> GenericServerAsyncReader; typedef ServerAsyncWriter<ByteBuffer> GenericServerAsyncWriter; class GenericServerContext final : public ServerContext { public: const std::string& method() const { return method_; } const std::string& host() const { return host_; } private: friend class ServerInterface; std::string method_; std::string host_; }; // A generic service at the server side accepts all RPC methods and hosts. It is // typically used in proxies. The generic service can be registered to a server // which also has other services. // Sample usage: // ServerBuilder builder; // auto cq = builder.AddCompletionQueue(); // AsyncGenericService generic_service; // builder.RegisterAsyncGenericService(&generic_service); // auto server = builder.BuildAndStart(); // // // request a new call // GenericServerContext context; // GenericServerAsyncReaderWriter stream; // generic_service.RequestCall(&context, &stream, cq.get(), cq.get(), tag); // // When tag is retrieved from cq->Next(), context.method() can be used to look // at the method and the RPC can be handled accordingly. class AsyncGenericService final { public: AsyncGenericService() : server_(nullptr) {} void RequestCall(GenericServerContext* ctx, GenericServerAsyncReaderWriter* reader_writer, grpc::CompletionQueue* call_cq, grpc::ServerCompletionQueue* notification_cq, void* tag); private: friend class grpc::Server; grpc::Server* server_; }; /// \a ServerGenericBidiReactor is the reactor class for bidi streaming RPCs /// invoked on a CallbackGenericService. It is just a ServerBidi reactor with /// ByteBuffer arguments. using ServerGenericBidiReactor = ServerBidiReactor<ByteBuffer, ByteBuffer>; class GenericCallbackServerContext final : public grpc::CallbackServerContext { public: const std::string& method() const { return method_; } const std::string& host() const { return host_; } private: friend class grpc::Server; std::string method_; std::string host_; }; /// \a CallbackGenericService is the base class for generic services implemented /// using the callback API and registered through the ServerBuilder using /// RegisterCallbackGenericService. class CallbackGenericService { public: CallbackGenericService() {} virtual ~CallbackGenericService() {} /// The "method handler" for the generic API. This function should be /// overridden to provide a ServerGenericBidiReactor that implements the /// application-level interface for this RPC. Unimplemented by default. virtual ServerGenericBidiReactor* CreateReactor( GenericCallbackServerContext* /*ctx*/) { class Reactor : public ServerGenericBidiReactor { public: Reactor() { this->Finish(Status(StatusCode::UNIMPLEMENTED, "")); } void OnDone() override { delete this; } }; return new Reactor; } private: friend class grpc::Server; internal::CallbackBidiHandler<ByteBuffer, ByteBuffer>* Handler() { return new internal::CallbackBidiHandler<ByteBuffer, ByteBuffer>( [this](grpc::CallbackServerContext* ctx) { return CreateReactor(static_cast<GenericCallbackServerContext*>(ctx)); }); } grpc::Server* server_{nullptr}; }; } // namespace grpc #endif // GRPCPP_IMPL_CODEGEN_ASYNC_GENERIC_SERVICE_H