samples/grpc_middleware_service/src/middlewares/server/meta_filter.hpp

#pragma once
 
#include <userver/utest/using_namespace_userver.hpp>
 
#include <userver/ugrpc/server/middlewares/base.hpp>
 
namespace samples::grpc::auth::server {
 
struct MiddlewareConfig;
 
class MetaFilter final : public ugrpc::server::MiddlewareBase {
public:
    MetaFilter(MiddlewareConfig&& config);
 
    void OnCallStart(ugrpc::server::MiddlewareCallContext& context) const override;
 
private:
    const std::vector<std::string> headers_;
};

struct MiddlewareConfig final {
    std::vector<std::string> headers{};
};
 
class MetaFilterComponent final : public ugrpc::server::MiddlewareFactoryComponentBase {
public:
    static constexpr std::string_view kName = "grpc-server-meta-filter";
 
    MetaFilterComponent(const components::ComponentConfig& config, const components::ComponentContext& context);
 
    static yaml_config::Schema GetStaticConfigSchema();
 
    // Needed to pass static config options to the middleware.
    yaml_config::Schema GetMiddlewareConfigSchema() const override;
 
    std::shared_ptr<const MiddlewareBase> CreateMiddleware(
        const ugrpc::server::ServiceInfo&,
        const yaml_config::YamlConfig& middleware_config
    ) const override;
};
 
}  // namespace samples::grpc::auth::server