samples/grpc_middleware_service/src/middlewares/server/middleware.cpp

#include "middleware.hpp"
 
#include <middlewares/auth.hpp>
 
#include <userver/utest/using_namespace_userver.hpp>
 
#include <userver/components/component.hpp>
 
namespace sample::grpc::auth::server {
 
Middleware::Middleware() = default;
 
void Middleware::Handle(ugrpc::server::MiddlewareCallContext& context) const {
    const auto& metadata = context.GetCall().GetContext().client_metadata();
 
    auto it = metadata.find(kKey);
 
    if (it == metadata.cend() || it->second != kCredentials) {
        auto& rpc = context.GetCall();
        rpc.FinishWithError(::grpc::Status{::grpc::StatusCode::PERMISSION_DENIED, "Invalid credentials"});
        LOG_ERROR() << "Invalid credentials";
        return;
    }
 
    context.Next();
}
 
}  // namespace sample::grpc::auth::server