Merge remote-tracking branch 'origin/eventengine-poc' into eventengin…

…e/transition-unit-tests
pubnub · Aug 2, 2023 · af2633b · af2633b
2 parents 51d8738 + 57f4fb2
commit af2633b
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 2 deletions.
diff --git a/src/Api/PubnubApi/EventEngine/Core/EffectDispatcher.cs b/src/Api/PubnubApi/EventEngine/Core/EffectDispatcher.cs
@@ -8,19 +8,23 @@ public class EffectDispatcher {
 		private readonly Dictionary<System.Type, IEffectHandler> effectInvocationHandlerMap =
 			new Dictionary<System.Type, IEffectHandler>();
 
+		public event System.Action<IEffectInvocation> OnEffectDispatch;
+
 		/// <summary>
 		/// Dispatch an invocation i.e. call a registered effect handler.
 		/// </summary>
-		public async Task Dispatch<T>(T invocation) where T : IEffectInvocation {
+		public async Task Dispatch(IEffectInvocation invocation) {
 			if (!effectInvocationHandlerMap.ContainsKey(invocation.GetType())) {
 				throw new ArgumentException($"No handler for {invocation.GetType().Name} found.");
 			}
+
+			OnEffectDispatch?.Invoke(invocation);
 
 			if (invocation is IEffectCancelInvocation) {
 				await effectInvocationHandlerMap[invocation.GetType()].Cancel();
 			} else
 			{
-				var handler = ((IEffectHandler<T>)effectInvocationHandlerMap[invocation.GetType()]);
+				var handler = effectInvocationHandlerMap[invocation.GetType()];
 				if (handler.IsBackground(invocation))
 					handler.Run(invocation).Start();
 				else

diff --git a/src/Api/PubnubApi/EventEngine/Core/Engine.cs b/src/Api/PubnubApi/EventEngine/Core/Engine.cs
@@ -11,6 +11,25 @@ public abstract class Engine {
 		private Task currentTransitionLoop = Utils.EmptyTask;
 
 		private readonly IEffectInvocation[] emptyInvocationList = new IEffectInvocation[0];
+
+		/// <summary>
+		/// Subscribe to receive notification on effect dispatch
+		/// </summary>
+		public event System.Action<IEffectInvocation> OnEffectDispatch
+		{
+			add => dispatcher.OnEffectDispatch += value;
+			remove => dispatcher.OnEffectDispatch -= value;
+		}
+
+		/// <summary>
+		/// Subscribe to receive notification on state transition
+		/// </summary>
+		public event System.Action<TransitionResult> OnStateTransition;
+
+		/// <summary>
+		/// Subscribe to receive notification on event being queued
+		/// </summary>
+		public event System.Action<IEvent> OnEventQueued;
 
 		public Engine() {
 			eventQueue.OnEventQueued += OnEvent;
@@ -22,12 +41,14 @@ public Engine() {
 
 		private async void OnEvent(EventQueue q)
 		{
+			OnEventQueued?.Invoke(q.Peek());
 			await currentTransitionLoop;
 			currentTransitionLoop = eventQueue.Loop(async e => currentState = await Transition(e));
 		}
 
 		private async Task<State> Transition(IEvent e) {
 			var stateInvocationPair = currentState.Transition(e);
+			OnStateTransition?.Invoke(stateInvocationPair);
 
 			if (stateInvocationPair is null) {
 				return currentState;

diff --git a/src/Api/PubnubApi/EventEngine/Core/EventQueue.cs b/src/Api/PubnubApi/EventEngine/Core/EventQueue.cs
@@ -33,6 +33,14 @@ private IEvent Dequeue()
             }
         }
 
+        public IEvent Peek()
+        {
+            lock (lockObj)
+            {
+                return eventQueue.Peek();
+            }
+        }
+
         public async Task Loop<T>(System.Func<IEvent, Task<T>> function)
         {
             while (Count > 0)

diff --git a/src/Api/PubnubApi/EventEngine/Core/Utils.cs b/src/Api/PubnubApi/EventEngine/Core/Utils.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Threading.Tasks;
 using System.Collections.Generic;
+using System.Reflection;
 
 namespace PubnubApi.EventEngine.Core
 {
@@ -17,6 +18,20 @@ internal static IEffectInvocation[] AsArray(this IEffectInvocation invocation)
         {
             return new IEffectInvocation[] { invocation };
         }
+
+        internal static bool IsBackground(this IEffectHandler handler, IEffectInvocation invocation)
+        {
+            return (bool)handler.GetType()
+                .GetMethod("IsBackground")
+                .Invoke(handler, new object[] { invocation });
+        }
+
+        internal static Task Run(this IEffectHandler handler, IEffectInvocation invocation)
+        {
+            return (Task)handler.GetType()
+                .GetMethod("Run")
+                .Invoke(handler, new object[] { invocation });
+        }
     }
 
     public class TransitionResult