Fix DbContext Registration for Activities

MassTransit · May 19, 2020 · ca91e77 · ca91e77
1 parent a8bcbf1
commit ca91e77
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/src/SampleBatch.Components/Activities/CancelOrder/CancelOrderActivity.cs b/src/SampleBatch.Components/Activities/CancelOrder/CancelOrderActivity.cs
@@ -10,10 +10,12 @@
     public class CancelOrderActivity :
         IExecuteActivity<CancelOrderArguments>
     {
+        private readonly SampleBatchDbContext _dbContext;
         readonly ILogger _logger;
 
-        public CancelOrderActivity(ILoggerFactory loggerFactory)
+        public CancelOrderActivity(SampleBatchDbContext dbContext, ILoggerFactory loggerFactory)
         {
+            _dbContext = dbContext;
             _logger = loggerFactory.CreateLogger<CancelOrderActivity>();
         }
 

diff --git a/src/SampleBatch.Components/Activities/SuspendOrder/SuspendOrderActivity.cs b/src/SampleBatch.Components/Activities/SuspendOrder/SuspendOrderActivity.cs
@@ -10,10 +10,12 @@
     public class SuspendOrderActivity :
         IExecuteActivity<SuspendOrderArguments>
     {
+        private readonly SampleBatchDbContext _dbContext;
         readonly ILogger _logger;
 
-        public SuspendOrderActivity(ILoggerFactory loggerFactory)
+        public SuspendOrderActivity(SampleBatchDbContext dbContext, ILoggerFactory loggerFactory)
         {
+            _dbContext = dbContext;
             _logger = loggerFactory.CreateLogger<SuspendOrderActivity>();
         }
 

diff --git a/src/SampleBatch.Service/Program.cs b/src/SampleBatch.Service/Program.cs
@@ -84,7 +84,7 @@ static async Task Main(string[] args)
                         cfg.AddBus(ConfigureBus);
                     });
 
-                    services.AddDbContext<DbContext, SampleBatchDbContext>(x => x.UseSqlServer(hostContext.Configuration.GetConnectionString("sample-batch")));
+                    services.AddDbContext<SampleBatchDbContext>(x => x.UseSqlServer(hostContext.Configuration.GetConnectionString("sample-batch")));
 
                     services.AddHostedService<MassTransitConsoleHostedService>();