feat: allow middlewares to be registered at the beginning

Farfetch · Oct 15, 2020 · 2a2de8a · 2a2de8a
1 parent 1fff317
commit 2a2de8a
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/src/KafkaFlow.Abstractions/Configuration/IConsumerMiddlewareConfigurationBuilder.cs b/src/KafkaFlow.Abstractions/Configuration/IConsumerMiddlewareConfigurationBuilder.cs
@@ -19,12 +19,31 @@ public interface IConsumerMiddlewareConfigurationBuilder
         IConsumerMiddlewareConfigurationBuilder Add<T>(Factory<T> factory)
             where T : class, IMessageMiddleware;
 
+        /// <summary>
+        /// Registers a middleware at the beginning of the middleware list
+        /// The middleware will run before other middlewares that already have been registered
+        /// </summary>
+        /// <param name="factory">A factory to create the instance</param>
+        /// <typeparam name="T">A class that implements the <see cref="IMessageMiddleware"/></typeparam>
+        /// <returns></returns>
+        IConsumerMiddlewareConfigurationBuilder AddAtBeginning<T>(Factory<T> factory)
+            where T : class, IMessageMiddleware;
+
         /// <summary>
         /// Registers a middleware
         /// </summary>
         /// <typeparam name="T">A class that implements the <see cref="IMessageMiddleware"/></typeparam>
         /// <returns></returns>
         IConsumerMiddlewareConfigurationBuilder Add<T>()
             where T : class, IMessageMiddleware;
+
+        /// <summary>
+        /// Registers a middleware at the beginning of the middleware list 
+        /// The middleware will run before other middlewares that already have been registered
+        /// </summary>
+        /// <typeparam name="T">A class that implements the <see cref="IMessageMiddleware"/></typeparam>
+        /// <returns></returns>
+        IConsumerMiddlewareConfigurationBuilder AddAtBeginning<T>()
+            where T : class, IMessageMiddleware;
     }
 }
diff --git a/src/KafkaFlow/Configuration/ConsumerMiddlewareConfigurationBuilder.cs b/src/KafkaFlow/Configuration/ConsumerMiddlewareConfigurationBuilder.cs
@@ -21,13 +21,31 @@ public IConsumerMiddlewareConfigurationBuilder Add<T>(Factory<T> factory) where
             return this;
         }
 
+        public IConsumerMiddlewareConfigurationBuilder AddAtBeginning<T>(Factory<T> factory) where T : class, IMessageMiddleware
+        {
+            this.middlewaresFactories.Insert(0, factory);
+            return this;
+        }
+
         public IConsumerMiddlewareConfigurationBuilder Add<T>() where T : class, IMessageMiddleware
         {
-            this.DependencyConfigurator.AddTransient<T>();
+            this.RegisterType<T>();
             this.middlewaresFactories.Add(resolver => resolver.Resolve<T>());
             return this;
         }
 
+        public IConsumerMiddlewareConfigurationBuilder AddAtBeginning<T>() where T : class, IMessageMiddleware
+        {
+            this.RegisterType<T>();
+            this.middlewaresFactories.Insert(0, resolver => resolver.Resolve<T>());
+            return this;
+        }        
+
         public MiddlewareConfiguration Build() => new MiddlewareConfiguration(this.middlewaresFactories);
+
+        private void RegisterType<T>() where T : class, IMessageMiddleware
+        {
+            this.DependencyConfigurator.AddTransient<T>();
+        }
     }
 }