Merge pull request danielgerlag#160 from Kahbazi/master

Set IServiceCollection as return type for AddWorkFlow
davidrneves · Jul 22, 2018 · 259bcf1 · 259bcf1
2 parents 9a7ae01 + 41403dc
commit 259bcf1
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;
         }
     }
 }