Updating jobs as cancelable

Author: dolunay

demos/MainDemo/src/Syrna.QuartzAdmin.MainDemo.Jobs/AutoJob.cs

@@ -17,7 +17,7 @@ private static async Task CanFireIt()
             await Task.CompletedTask;
         }
 
-        public async Task Execute(IJobExecutionContext context)
+        private async Task ExecuteJob(IJobExecutionContext context)
         {
             context.CancellationToken.ThrowIfCancellationRequested();
             Console.WriteLine($"Hello from AutoJob {DateTime.Now}");
@@ -31,5 +31,19 @@ public async Task Execute(IJobExecutionContext context)
 
             await Task.CompletedTask;
         }
+
+        public async Task Execute(IJobExecutionContext context)
+        {
+            var taskCompletionSource = new TaskCompletionSource();
+            context.CancellationToken.Register(() =>
+            {
+                // We received a cancellation message, cancel the TaskCompletionSource.Task
+                // ReSharper disable once InvertIf
+                taskCompletionSource.TrySetCanceled();
+            });
+            var completedTask = await Task.WhenAny(ExecuteJob(context), taskCompletionSource.Task);
+
+            await completedTask;
+        }
     }
 }

demos/MainDemo/src/Syrna.QuartzAdmin.MainDemo.Jobs/AutoJob1.cs

@@ -7,7 +7,7 @@ namespace Syrna.QuartzAdmin.MainDemo.Jobs
     [QuartzTrigger(5, 0, 0, Desciption = "Automatic job of welcome information")]
     public class AutoJob1 : IJob
     {
-        public Task Execute(IJobExecutionContext context)
+        private Task ExecuteJob(IJobExecutionContext context)
         {
             context.CancellationToken.ThrowIfCancellationRequested();
             Console.WriteLine($"Hello from Auto Job1 {DateTime.Now}");
@@ -19,5 +19,19 @@ public Task Execute(IJobExecutionContext context)
             context.Result = $"Hello from Auto Job1 {DateTime.Now}";
             return Task.CompletedTask;
         }
+
+        public async Task Execute(IJobExecutionContext context)
+        {
+            var taskCompletionSource = new TaskCompletionSource();
+            context.CancellationToken.Register(() =>
+            {
+                // We received a cancellation message, cancel the TaskCompletionSource.Task
+                // ReSharper disable once InvertIf
+                taskCompletionSource.TrySetCanceled();
+            });
+            var completedTask = await Task.WhenAny(ExecuteJob(context), taskCompletionSource.Task);
+
+            await completedTask;
+        }
     }
 }

demos/MainDemo/src/Syrna.QuartzAdmin.MainDemo.Jobs/AutoJob2.cs

@@ -8,7 +8,7 @@ namespace Syrna.QuartzAdmin.MainDemo.Jobs
     [QuartzTrigger(2,0, "this is an long job test", "_longjobauto")]
     public class AutoJob2(ILogger<HelloJob> logger) : IJob
     {
-        public async Task Execute(IJobExecutionContext context)
+        private async Task ExecuteJob(IJobExecutionContext context)
         {
             context.CancellationToken.ThrowIfCancellationRequested();
             Console.WriteLine($"Hello from AutoJob {DateTime.Now}");
@@ -36,5 +36,18 @@ public async Task Execute(IJobExecutionContext context)
 
             await Task.CompletedTask;
         }
+        public async Task Execute(IJobExecutionContext context)
+        {
+            var taskCompletionSource = new TaskCompletionSource();
+            context.CancellationToken.Register(() =>
+            {
+                // We received a cancellation message, cancel the TaskCompletionSource.Task
+                // ReSharper disable once InvertIf
+                taskCompletionSource.TrySetCanceled();
+            });
+            var completedTask = await Task.WhenAny(ExecuteJob(context), taskCompletionSource.Task);
+
+            await completedTask;
+        }
     }
 }

demos/MainDemo/src/Syrna.QuartzAdmin.MainDemo.Jobs/HttpJob.cs

@@ -12,7 +12,7 @@ public class HttpJob(IHttpClientFactory httpClientFactory,
 		ILogger<HttpJob> logger,
 		IDataMapValueResolver dmvResolver) : IJob
     {
-		public async Task Execute(IJobExecutionContext context)
+        private async Task ExecuteJob(IJobExecutionContext context)
         {
             context.CancellationToken.ThrowIfCancellationRequested();
             try
@@ -127,6 +127,19 @@ public async Task Execute(IJobExecutionContext context)
                 throw new JobExecutionException("Failed to execute http job", ex);
             }
         }
+        public async Task Execute(IJobExecutionContext context)
+        {
+            var taskCompletionSource = new TaskCompletionSource();
+            context.CancellationToken.Register(() =>
+            {
+                // We received a cancellation message, cancel the TaskCompletionSource.Task
+                // ReSharper disable once InvertIf
+                taskCompletionSource.TrySetCanceled();
+            });
+            var completedTask = await Task.WhenAny(ExecuteJob(context), taskCompletionSource.Task);
+
+            await completedTask;
+        }
     }
 }
 

demos/MainDemo/src/Syrna.QuartzAdmin.MainDemo.Jobs/LongJob.cs

@@ -12,7 +12,7 @@ public class LongJob(
     private const string PropertyMessage = "message";
     private const string PropertyDelayInMs = "delay";
 
-    public async Task Execute(IJobExecutionContext context)
+    private async Task ExecuteJob(IJobExecutionContext context)
     {
         var taskCompletionSource = new TaskCompletionSource();
         context.CancellationToken.Register(() =>
@@ -47,5 +47,19 @@ public async Task Execute(IJobExecutionContext context)
 
         await completedTask;
     }
+
+    public async Task Execute(IJobExecutionContext context)
+    {
+        var taskCompletionSource = new TaskCompletionSource();
+        context.CancellationToken.Register(() =>
+        {
+            // We received a cancellation message, cancel the TaskCompletionSource.Task
+            // ReSharper disable once InvertIf
+            taskCompletionSource.TrySetCanceled();
+        });
+        var completedTask = await Task.WhenAny(ExecuteJob(context), taskCompletionSource.Task);
+
+        await completedTask;
+    }
 }