Merge branch 'master' of https://github.com/trympet/MediatR.IPC

README.md

@@ -75,7 +75,8 @@ Currently, exceptions thrown by request handlers are not serialized, and no type
 
 ## Contributing
 All forms of contribution are welcome! Here is a list of some much needed features.
-- [ ] Request cancellation with `CancellationToken`
+
+- [x] Request cancellation with `CancellationToken`
 - [ ] Dynamic buffers for requests in `MediatorServerBase`
 - [ ] `IPublisher` implementation
 - [ ]  Routing of `INotification` to `INotificationHandler` designated for IPC via DI container.

src/Mediator/IPCMediator.PublicStatics.cs

@@ -10,6 +10,7 @@ namespace MediatR
 {
     public abstract partial class IPCMediator
     {
+        private static readonly Type NotificationType = typeof(INotification);
         private static readonly Type RequestType = typeof(IRequest<>);
         private static readonly Type UnitType = typeof(Unit);
 
@@ -29,9 +30,9 @@ public static IPCBuilder<Assembly, IEnumerable<Type>> RegisterAssemblyTypes(Asse
         {
             var types = assembly.ExportedTypes;
             var requests = assembly.ExportedTypes
-                .Where(t => t
+                .Where<Type>(t => t
                     .GetInterfaces()
-                    .Any(i => i.IsGenericType && i.GetGenericTypeDefinition() == RequestType)
+                    .Any(i => IsRequest(i) || IsNotification(i))
                 );
 
             var builder = new IPCBuilder<Assembly, IEnumerable<Type>>(requests);
@@ -40,6 +41,13 @@ public static IPCBuilder<Assembly, IEnumerable<Type>> RegisterAssemblyTypes(Asse
             return builder;
         }
 
+        private static bool IsRequest(Type i)
+            => i.IsGenericType && i.GetGenericTypeDefinition() == RequestType;
+
+
+        private static bool IsNotification(Type i)
+            => i == NotificationType;
+
         /// <summary>
         /// Registers a request to be handled by the IPC mediator.
         /// </summary>

src/Mediators/MediatorClient.cs

@@ -70,7 +70,7 @@ private object DeserializeResponse(Message response, Type contentType)
 
             if (response.IsNullResponse)
             {
-                return default!;
+                return default(Unit);
             }
 
             return DeserializeContent(response , contentType);