feat grpc: add per-service middleware config overriding

Relates: <HIDDEN_URL>
commit_hash:0c3ff8492fc854cbdacea653291cce9861af2177

Author: ArkadyRudenko

.mapping.json

@@ -1978,9 +1978,11 @@
   "grpc/include/userver/ugrpc/server/impl/service_worker_impl.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/impl/service_worker_impl.hpp",
   "grpc/include/userver/ugrpc/server/metadata_utils.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/metadata_utils.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/baggage/component.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/baggage/component.hpp",
+  "grpc/include/userver/ugrpc/server/middlewares/baggage/middleware.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/baggage/middleware.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/base.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/base.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/congestion_control/component.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/congestion_control/component.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/deadline_propagation/component.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/deadline_propagation/component.hpp",
+  "grpc/include/userver/ugrpc/server/middlewares/deadline_propagation/middleware.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/deadline_propagation/middleware.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/field_mask/component.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/field_mask/component.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/fwd.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/fwd.hpp",
   "grpc/include/userver/ugrpc/server/middlewares/groups.hpp":"taxi/uservices/userver/grpc/include/userver/ugrpc/server/middlewares/groups.hpp",
@@ -2092,16 +2094,12 @@
   "grpc/src/ugrpc/server/impl/service_worker_impl.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/impl/service_worker_impl.cpp",
   "grpc/src/ugrpc/server/impl/topology_sort.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/impl/topology_sort.cpp",
   "grpc/src/ugrpc/server/impl/topology_sort.hpp":"taxi/uservices/userver/grpc/src/ugrpc/server/impl/topology_sort.hpp",
-  "grpc/src/ugrpc/server/middlewares/baggage/component.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/baggage/component.cpp",
   "grpc/src/ugrpc/server/middlewares/baggage/middleware.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/baggage/middleware.cpp",
-  "grpc/src/ugrpc/server/middlewares/baggage/middleware.hpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/baggage/middleware.hpp",
   "grpc/src/ugrpc/server/middlewares/base.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/base.cpp",
   "grpc/src/ugrpc/server/middlewares/congestion_control/component.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/congestion_control/component.cpp",
   "grpc/src/ugrpc/server/middlewares/congestion_control/middleware.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/congestion_control/middleware.cpp",
   "grpc/src/ugrpc/server/middlewares/congestion_control/middleware.hpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/congestion_control/middleware.hpp",
-  "grpc/src/ugrpc/server/middlewares/deadline_propagation/component.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/deadline_propagation/component.cpp",
   "grpc/src/ugrpc/server/middlewares/deadline_propagation/middleware.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/deadline_propagation/middleware.cpp",
-  "grpc/src/ugrpc/server/middlewares/deadline_propagation/middleware.hpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/deadline_propagation/middleware.hpp",
   "grpc/src/ugrpc/server/middlewares/field_mask/component.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/field_mask/component.cpp",
   "grpc/src/ugrpc/server/middlewares/field_mask/middleware.cpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/field_mask/middleware.cpp",
   "grpc/src/ugrpc/server/middlewares/field_mask/middleware.hpp":"taxi/uservices/userver/grpc/src/ugrpc/server/middlewares/field_mask/middleware.hpp",

grpc/functional_tests/basic_server/static_config.yaml

@@ -37,6 +37,7 @@ components_manager:
             disable-all-pipeline-middlewares: true
             middlewares:
                 grpc-server-headers-propagator:
+                    enabled: true
 
         grpc-server-middlewares-pipeline:
 

grpc/functional_tests/middleware_server/src/my_middleware.cpp

@@ -18,6 +18,9 @@ void MyMiddleware::CallResponseHook(const ugrpc::server::MiddlewareCallContext&,
 
 void MyMiddleware::Handle(ugrpc::server::MiddlewareCallContext& context) const { context.Next(); }
 
-std::shared_ptr<ugrpc::server::MiddlewareBase> MyMiddlewareComponent::GetMiddleware() { return middleware_; }
+std::shared_ptr<ugrpc::server::MiddlewareBase>
+MyMiddlewareComponent::CreateMiddleware(const ugrpc::server::ServiceInfo&, const yaml_config::YamlConfig&) const {
+    return middleware_;
+}
 
 }  // namespace functional_tests

grpc/functional_tests/middleware_server/src/my_middleware.hpp

@@ -20,14 +20,17 @@ class MyMiddleware final : public ugrpc::server::MiddlewareBase {
     void Handle(ugrpc::server::MiddlewareCallContext& context) const override;
 };
 
-class MyMiddlewareComponent final : public ugrpc::server::MiddlewareComponentBase {
+class MyMiddlewareComponent final : public ugrpc::server::MiddlewareFactoryComponentBase {
 public:
     static constexpr std::string_view kName = "my-middleware-server";
 
     MyMiddlewareComponent(const components::ComponentConfig& config, const components::ComponentContext& ctx)
-        : ugrpc::server::MiddlewareComponentBase(config, ctx), middleware_(std::make_shared<MyMiddleware>()) {}
+        : ugrpc::server::MiddlewareFactoryComponentBase(config, ctx), middleware_(std::make_shared<MyMiddleware>()) {}
 
-    std::shared_ptr<ugrpc::server::MiddlewareBase> GetMiddleware() override;
+    std::shared_ptr<ugrpc::server::MiddlewareBase> CreateMiddleware(
+        const ugrpc::server::ServiceInfo&,
+        const yaml_config::YamlConfig& middleware_config
+    ) const override;
 
 private:
     std::shared_ptr<ugrpc::server::MiddlewareBase> middleware_;

grpc/functional_tests/middleware_server/src/my_second_middleware.cpp

@@ -24,6 +24,9 @@ void MySecondMiddleware::CallResponseHook(
 
 void MySecondMiddleware::Handle(ugrpc::server::MiddlewareCallContext& context) const { context.Next(); }
 
-std::shared_ptr<ugrpc::server::MiddlewareBase> MySecondMiddlewareComponent::GetMiddleware() { return middleware_; }
+std::shared_ptr<ugrpc::server::MiddlewareBase>
+MySecondMiddlewareComponent::CreateMiddleware(const ugrpc::server::ServiceInfo&, const yaml_config::YamlConfig&) const {
+    return middleware_;
+}
 
 }  // namespace functional_tests

grpc/functional_tests/middleware_server/src/my_second_middleware.hpp

@@ -20,14 +20,18 @@ class MySecondMiddleware final : public ugrpc::server::MiddlewareBase {
     void Handle(ugrpc::server::MiddlewareCallContext& context) const override;
 };
 
-class MySecondMiddlewareComponent final : public ugrpc::server::MiddlewareComponentBase {
+class MySecondMiddlewareComponent final : public ugrpc::server::MiddlewareFactoryComponentBase {
 public:
     static constexpr std::string_view kName = "my-second-middleware-server";
 
     MySecondMiddlewareComponent(const components::ComponentConfig& config, const components::ComponentContext& ctx)
-        : ugrpc::server::MiddlewareComponentBase(config, ctx), middleware_(std::make_shared<MySecondMiddleware>()) {}
+        : ugrpc::server::MiddlewareFactoryComponentBase(config, ctx),
+          middleware_(std::make_shared<MySecondMiddleware>()) {}
 
-    std::shared_ptr<ugrpc::server::MiddlewareBase> GetMiddleware() override;
+    std::shared_ptr<ugrpc::server::MiddlewareBase> CreateMiddleware(
+        const ugrpc::server::ServiceInfo&,
+        const yaml_config::YamlConfig& middleware_config
+    ) const override;
 
 private:
     std::shared_ptr<ugrpc::server::MiddlewareBase> middleware_;

grpc/functional_tests/middleware_server/static_config.yaml

@@ -21,7 +21,7 @@ components_manager:
             port: 8091
 
         grpc-server-logging:
-            msg-size-log-limit: 10
+            msg-size-log-limit: 15
 
         congestion-control:
 
@@ -37,6 +37,7 @@ components_manager:
             disable-all-pipeline-middlewares: true
             middlewares:
                 grpc-server-logging:
+                    msg-size-log-limit: 10
                     enabled: true
                 my-middleware-server:
                     enabled: true

grpc/include/userver/ugrpc/server/middlewares/baggage/component.hpp

@@ -4,7 +4,7 @@
 /// @brief @copybrief
 /// ugrpc::server::middlewares::baggage::Component
 
-#include <userver/ugrpc/server/middlewares/base.hpp>
+#include <userver/ugrpc/server/middlewares/baggage/middleware.hpp>
 
 USERVER_NAMESPACE_BEGIN
 
@@ -25,22 +25,10 @@ namespace ugrpc::server::middlewares::baggage {
 
 // clang-format on
 
-class Component final : public MiddlewareComponentBase {
-public:
-    /// @ingroup userver_component_names
-    /// @brief The default name of ugrpc::server::middlewares::baggage::Component
-    static constexpr std::string_view kName = "grpc-server-baggage";
-
-    Component(const components::ComponentConfig& config, const components::ComponentContext& context);
-
-    std::shared_ptr<MiddlewareBase> GetMiddleware() override;
-};
+using Component = SimpleMiddlewareFactoryComponent<Middleware>;
 
 }  // namespace ugrpc::server::middlewares::baggage
 
-template <>
-inline constexpr bool components::kHasValidate<ugrpc::server::middlewares::baggage::Component> = true;
-
 template <>
 inline constexpr auto components::kConfigFileMode<ugrpc::server::middlewares::baggage::Component> =
     ConfigFileMode::kNotRequired;

grpc/src/ugrpc/server/middlewares/baggage/middleware.hpp renamed to grpc/include/userver/ugrpc/server/middlewares/baggage/middleware.hpp

@@ -8,6 +8,13 @@ namespace ugrpc::server::middlewares::baggage {
 
 class Middleware final : public MiddlewareBase {
 public:
+    /// @ingroup userver_component_names
+    /// @brief The default name of ugrpc::server::middlewares::baggage::Component
+    static constexpr std::string_view kName = "grpc-server-baggage";
+
+    /// @brief dependency of this middleware
+    static inline const auto kDependency = MiddlewareDependencyBuilder();
+
     void Handle(MiddlewareCallContext& context) const override;
 };
 

grpc/include/userver/ugrpc/server/middlewares/base.hpp

@@ -10,6 +10,9 @@
 
 #include <userver/components/component_base.hpp>
 #include <userver/utils/function_ref.hpp>
+#include <userver/utils/impl/internal_tag.hpp>
+#include <userver/yaml_config/schema.hpp>
+#include <userver/yaml_config/yaml_config.hpp>
 
 #include <userver/ugrpc/server/call.hpp>
 #include <userver/ugrpc/server/middlewares/fwd.hpp>
@@ -19,6 +22,8 @@ USERVER_NAMESPACE_BEGIN
 
 namespace ugrpc::server {
 
+struct ServiceInfo;
+
 /// @brief Context for middleware-specific data during gRPC call
 class MiddlewareCallContext final {
 public:
@@ -83,26 +88,63 @@ class MiddlewareBase {
 /// @ingroup userver_base_classes
 ///
 /// @brief Base class for middleware component
-class MiddlewareComponentBase : public components::ComponentBase {
+class MiddlewareFactoryComponentBase : public components::ComponentBase {
 public:
-    MiddlewareComponentBase(
+    MiddlewareFactoryComponentBase(
         const components::ComponentConfig&,
         const components::ComponentContext&,
         MiddlewareDependencyBuilder&& builder = MiddlewareDependencyBuilder().InGroup<groups::User>()
     );
 
     /// @brief Returns a middleware according to the component's settings
-    virtual std::shared_ptr<MiddlewareBase> GetMiddleware() = 0;
+    ///
+    /// @param info is a info of a grpc-service
+    /// @param middleware_config config for the middleware
+    virtual std::shared_ptr<MiddlewareBase>
+    CreateMiddleware(const ServiceInfo& info, const yaml_config::YamlConfig& middleware_config) const = 0;
+
+    /// @brief This method should return the schema of a middleware configuration.
+    /// Always write `return GetStaticConfigSchema();` in this method
+    virtual yaml_config::Schema GetMiddlewareConfigSchema() const { return GetStaticConfigSchema(); }
+
+    static yaml_config::Schema GetStaticConfigSchema();
 
     /// @cond
     /// Only for internal use.
-    const impl::MiddlewareDependency& GetMiddlewareDependency() const;
+    const impl::MiddlewareDependency& GetMiddlewareDependency(utils::impl::InternalTag) const;
+
+    const formats::yaml::Value& GetGlobalConfig(utils::impl::InternalTag) const;
     /// @endcond
 
 private:
-    impl::MiddlewareDependency dependency_;
+    const impl::MiddlewareDependency dependency_;
+    const formats::yaml::Value global_config_;
+};
+
+/// @ingroup userver_middlewares
+///
+/// @brief A short-cut for defining a middleware-factory
+template <typename Middleware>
+class SimpleMiddlewareFactoryComponent final : public MiddlewareFactoryComponentBase {
+public:
+    static constexpr std::string_view kName = Middleware::kName;
+
+    SimpleMiddlewareFactoryComponent(
+        const components::ComponentConfig& config,
+        const components::ComponentContext& context
+    )
+        : MiddlewareFactoryComponentBase(config, context, MiddlewareDependencyBuilder{Middleware::kDependency}) {}
+
+private:
+    std::shared_ptr<MiddlewareBase> CreateMiddleware(const ServiceInfo&, const yaml_config::YamlConfig&)
+        const override {
+        return std::make_shared<Middleware>();
+    }
 };
 
 }  // namespace ugrpc::server
 
+template <typename Middleware>
+inline constexpr bool components::kHasValidate<ugrpc::server::SimpleMiddlewareFactoryComponent<Middleware>> = true;
+
 USERVER_NAMESPACE_END