samples/grpc_service/src/greeter_client.hpp

#pragma once
 
#include <string_view>
 
#include <userver/utest/using_namespace_userver.hpp>

#include <userver/components/component_base.hpp>
#include <userver/components/component_fwd.hpp>
 
#include <userver/ugrpc/client/fwd.hpp>
#include <userver/ugrpc/client/simple_client_component.hpp>
#include <userver/ugrpc/impl/static_service_metadata.hpp>
 
#include <samples/greeter_client.usrv.pb.hpp>
 
namespace samples {

// A user-defined wrapper around api::GreeterServiceClient that handles
// the metadata and deadline bureaucracy and provides a simplified interface.
//
// Alternatively, you can use ugrpc::client::SimpleClientComponent directly.
//
// Note that we have both service and client to that service in the same
// microservice. Ignore that, it's just for the sake of example.
class GreeterClient final {
public:
    explicit GreeterClient(api::GreeterServiceClient& raw_client);
 
    std::string SayHello(std::string name) const;
 
    std::vector<std::string> SayHelloResponseStream(std::string name) const;
 
    std::string SayHelloRequestStream(const std::vector<std::string_view>& names) const;
 
    std::vector<std::string> SayHelloStreams(const std::vector<std::string_view>& names) const;
 
private:
    static ugrpc::client::CallOptions MakeCallOptions();
 
    api::GreeterServiceClient& raw_client_;
};
 
using Client = GreeterClient;

extern const USERVER_NAMESPACE::dynamic_config::Key<ugrpc::client::ClientQos> kGreeterClientQos;
 
class GreeterClientComponent final : public ugrpc::client::SimpleClientComponent<api::GreeterServiceClient> {
public:
    static constexpr std::string_view kName = "greeter-client";
 
    using Base = ugrpc::client::SimpleClientComponent<api::GreeterServiceClient>;
 
    GreeterClientComponent(const ::components::ComponentConfig& config, const ::components::ComponentContext& context)
        : Base(config, context, kGreeterClientQos), client_wrapper_(GetClient()) {}
 
    using Base::GetClient;
 
    GreeterClient& GetClientWrapper() noexcept { return client_wrapper_; }
 
private:
    GreeterClient client_wrapper_;
};
 
}  // namespace samples