Set IServiceCollection as return type for AddWorkFlow

davidrneves · Jul 21, 2018 · 41403dc · 41403dc
1 parent 9a7ae01
commit 41403dc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/WorkflowCore/ServiceCollectionExtensions.cs b/src/WorkflowCore/ServiceCollectionExtensions.cs
@@ -16,7 +16,7 @@ namespace Microsoft.Extensions.DependencyInjection
 {
     public static class ServiceCollectionExtensions
     {
-        public static void AddWorkflow(this IServiceCollection services, Action<WorkflowOptions> setupAction = null)
+        public static IServiceCollection AddWorkflow(this IServiceCollection services, Action<WorkflowOptions> setupAction = null)
         {
             if (services.Any(x => x.ServiceType == typeof(WorkflowOptions)))
                 throw new InvalidOperationException("Workflow services already registered");
@@ -47,6 +47,8 @@ public static void AddWorkflow(this IServiceCollection services, Action<Workflow
             services.AddTransient<IDefinitionLoader, DefinitionLoader>();
 
             services.AddTransient<Foreach>();
+
+            return services;
         }
     }
 }