implement 'immediate' as third parameter of Listen() method.

Author: endel

Assets/Plugins/Colyseus/StateListener/StateContainer.cs

@@ -72,7 +72,7 @@ public FallbackPatchListener Listen(Action<PatchObject> callback)
 			return listener;
 		}
 
-		public PatchListener Listen(string segments, Action<DataChange> callback) {
+		public PatchListener Listen(string segments, Action<DataChange> callback, bool immediate = false) {
 			var rawRules = segments.Split ('/');
 			var regexpRules = this.ParseRegexRules (rawRules);
 
@@ -84,6 +84,12 @@ public PatchListener Listen(string segments, Action<DataChange> callback) {
 
 			this.listeners.Add(listener);
 
+			if (immediate) {
+				List<PatchListener> onlyListener = new List<PatchListener>();
+				onlyListener.Add(listener);
+				this.CheckPatches(Compare.GetPatchList(new IndexedDictionary<string, object>(), this.state), onlyListener);
+        	}
+
 			return listener;
 		}
 
@@ -128,23 +134,29 @@ protected Regex[] ParseRegexRules (string[] rules)
 			return regexpRules;
 		}
 
-		private void CheckPatches(PatchObject[] patches)
+		private void CheckPatches(PatchObject[] patches, List<PatchListener> _listeners = null)
 		{
+			if (_listeners == null) 
+			{
+				_listeners = this.listeners;
+			}
 
 			for (var i = patches.Length - 1; i >= 0; i--)
 			{
 				var matched = false;
 
-				for (var j = 0; j < this.listeners.Count; j++)
+				for (var j = 0; j < _listeners.Count; j++)
 				{
-					var listener = this.listeners[j];
+					var listener = _listeners[j];
 					var pathVariables = this.GetPathVariables(patches[i], listener);
 					if (pathVariables != null)
 					{
-						var dataChange = new DataChange ();
-						dataChange.path = pathVariables;
-						dataChange.operation = patches [i].operation;
-						dataChange.value = patches [i].value;
+						DataChange dataChange = new DataChange
+						{
+							path = pathVariables,
+							operation = patches[i].operation,
+							value = patches[i].value
+						};
 
 						listener.callback.Invoke (dataChange);
 						matched = true;

Assets/Tests/StateContainerTest.cs

@@ -1,7 +1,4 @@
-using UnityEngine;
-using UnityEngine.TestTools;
-using NUnit.Framework;
-using System.Collections;
+using NUnit.Framework;
 using System.Collections.Generic;
 
 using Colyseus;
@@ -232,6 +229,31 @@ public void ListenInitialState() {
 		Assert.AreEqual (9, listenCalls);
 	}
 
+	[Test]
+	public void ListenWithImmediate()
+	{
+		var container = new StateContainer(GetRawData());
+		var listenCalls = 0;
+
+		container.Listen("players/:id/position/:attribute", (DataChange change) => {
+			listenCalls++;
+		}, true);
+
+		container.Listen("turn", (DataChange change) => {
+			listenCalls++;
+		}, true);
+
+		container.Listen("game/turn", (DataChange change) => {
+			listenCalls++;
+		}, true);
+
+		container.Listen("messages/:number", (DataChange change) => {
+			listenCalls++;
+		}, true);
+
+		Assert.AreEqual(9, listenCalls);
+	}
+
 	protected IndexedDictionary<string, object> GetRawData () {
 		var data = new IndexedDictionary<string, object> ();
 		var players = new IndexedDictionary<string, object> ();