Add support for custom scope configuration

This allows configuring `ILifetimeScope` based on data available in the `JobActivatorContext`, such as job parameters.
HangfireIO · Sep 24, 2024 · 0646e23 · 0646e23
1 parent ac65834
commit 0646e23
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/src/Hangfire.Autofac/AutofacJobActivator.cs b/src/Hangfire.Autofac/AutofacJobActivator.cs
@@ -16,6 +16,9 @@ public class AutofacJobActivator : JobActivator
         public static readonly object LifetimeScopeTag = "BackgroundJobScope";
 
         private readonly ILifetimeScope _lifetimeScope;
+#if !NET45
+        private readonly Action<ContainerBuilder, JobActivatorContext> _scopeConfigurationAction;
+#endif
         private readonly bool _useTaggedLifetimeScope;
 
         /// <summary>
@@ -32,6 +35,18 @@ public AutofacJobActivator([NotNull] ILifetimeScope lifetimeScope, bool useTagge
             _useTaggedLifetimeScope = useTaggedLifetimeScope;
         }
 
+#if !NET45
+        public AutofacJobActivator(
+            [NotNull] ILifetimeScope lifetimeScope,
+            [CanBeNull] Action<ContainerBuilder, JobActivatorContext> scopeConfigurationAction,
+            bool useTaggedLifetimeScope = true)
+        {
+            _lifetimeScope = lifetimeScope ?? throw new ArgumentNullException(nameof(lifetimeScope));
+            _scopeConfigurationAction = scopeConfigurationAction;
+            _useTaggedLifetimeScope = useTaggedLifetimeScope;
+        }
+#endif
+
         /// <inheritdoc />
         public override object ActivateJob(Type jobType)
         {
@@ -48,6 +63,18 @@ public override JobActivatorScope BeginScope()
 #else
         public override JobActivatorScope BeginScope(JobActivatorContext context)
         {
+            if (_scopeConfigurationAction != null)
+            {
+                Action<ContainerBuilder> configurationAction = builder =>
+                {
+                    _scopeConfigurationAction(builder, context);
+                };
+
+                return new AutofacScope(_useTaggedLifetimeScope
+                    ? _lifetimeScope.BeginLifetimeScope(LifetimeScopeTag, configurationAction)
+                    : _lifetimeScope.BeginLifetimeScope(configurationAction));
+            }
+
             return new AutofacScope(_useTaggedLifetimeScope
                 ? _lifetimeScope.BeginLifetimeScope(LifetimeScopeTag)
                 : _lifetimeScope.BeginLifetimeScope());

diff --git a/src/Hangfire.Autofac/GlobalConfigurationExtensions.cs b/src/Hangfire.Autofac/GlobalConfigurationExtensions.cs
@@ -16,5 +16,19 @@ public static IGlobalConfiguration<AutofacJobActivator> UseAutofacActivator(
 
             return configuration.UseActivator(new AutofacJobActivator(lifetimeScope, useTaggedLifetimeScope));
         }
+
+#if !NET45
+        public static IGlobalConfiguration<AutofacJobActivator> UseAutofacActivator(
+            [NotNull] this IGlobalConfiguration configuration, 
+            [NotNull] ILifetimeScope lifetimeScope,
+            [CanBeNull] Action<ContainerBuilder, JobActivatorContext> scopeConfigurationAction,
+            bool useTaggedLifetimeScope = true)
+        {
+            if (configuration == null) throw new ArgumentNullException(nameof(configuration));
+            if (lifetimeScope == null) throw new ArgumentNullException(nameof(lifetimeScope));
+
+            return configuration.UseActivator(new AutofacJobActivator(lifetimeScope, scopeConfigurationAction, useTaggedLifetimeScope));
+        }
+#endif
     }
 }