Added notification for device code removal (#3979)

* Added notification for device code removal * Updated TestOperationalStoreNotification
IdentityServer · Mar 24, 2020 · ee3a557 · ee3a557
1 parent c004ae4
commit ee3a557
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/src/EntityFramework.Storage/src/TokenCleanup/IOperationalStoreNotification.cs b/src/EntityFramework.Storage/src/TokenCleanup/IOperationalStoreNotification.cs
@@ -19,5 +19,12 @@ public interface IOperationalStoreNotification
         /// <param name="persistedGrants"></param>
         /// <returns></returns>
         Task PersistedGrantsRemovedAsync(IEnumerable<PersistedGrant> persistedGrants);
+
+        /// <summary>
+        /// Notification for device codes being removed.
+        /// </summary>
+        /// <param name="deviceCodes"></param>
+        /// <returns></returns>
+        Task DeviceCodesRemovedAsync(IEnumerable<DeviceFlowCodes> deviceCodes);
     }
 }
diff --git a/src/EntityFramework.Storage/src/TokenCleanup/TokenCleanupService.cs b/src/EntityFramework.Storage/src/TokenCleanup/TokenCleanupService.cs
@@ -130,6 +130,11 @@ protected virtual async Task RemoveDeviceCodesAsync()
                     try
                     {
                         await _persistedGrantDbContext.SaveChangesAsync();
+
+                        if (_operationalStoreNotification != null)
+                        {
+                            await _operationalStoreNotification.DeviceCodesRemovedAsync(expiredCodes);
+                        }
                     }
                     catch (DbUpdateConcurrencyException ex)
                     {

diff --git a/src/EntityFramework/host/TestOperationalStoreNotification.cs b/src/EntityFramework/host/TestOperationalStoreNotification.cs
@@ -21,5 +21,14 @@ public Task PersistedGrantsRemovedAsync(IEnumerable<PersistedGrant> persistedGra
             }
             return Task.CompletedTask;
         }
+
+        public Task DeviceCodesRemovedAsync(IEnumerable<DeviceFlowCodes> deviceCodes)
+        {
+            foreach (var deviceCode in deviceCodes) 
+            {
+                Console.WriteLine("cleaned device code");
+            }
+            return Task.CompletedTask;
+        }
     }
 }