try unsubscribe feature added to signal center

erkerkiii · Dec 29, 2024 · 710c85f · 710c85f
1 parent ee79bca
commit 710c85f
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/Gum.Signal/Core/SignalCenter.cs b/Gum.Signal/Core/SignalCenter.cs
@@ -34,6 +34,28 @@ public void Unsubscribe<T>(Action<T> action)
 			}
 		}
 
+		public bool TryUnsubscribe<T>(Action<T> action)
+		{
+			bool isSuccessful = false;
+			lock (_lock)
+			{
+				for (int index = 0; index < _entries.Count; index++)
+				{
+					Entry entry = _entries[index];
+
+					if (entry.Delegate as Action<T> != action || !_entries.Contains(entry))
+					{
+						continue;
+					}
+
+					_entries.Remove(entry);
+					isSuccessful = true;
+				}
+			}
+
+			return isSuccessful;
+		}
+
 		public void Fire<T>(T signal)
 		{
 			int hashCode = typeof(T).GetHashCode();

diff --git a/Tests/SignalTests/SignalCenterTests.cs b/Tests/SignalTests/SignalCenterTests.cs
@@ -39,6 +39,23 @@ void FooAction(FooSignal _) { }
 			Assert.IsFalse(_signalCenter.Exists<FooSignal>(FooAction));
 		}
 
+		[Test]
+		public void TryUnsubscribe()
+		{
+			void BarAction(BarSignal _) { }
+			void FooAction(FooSignal _) { }
+
+			_signalCenter.Subscribe<BarSignal>(BarAction);
+			Assert.IsTrue(_signalCenter.Exists<BarSignal>(BarAction));
+			Assert.IsFalse(_signalCenter.Exists<FooSignal>(FooAction));
+			bool isUnsubscribedFromBar = _signalCenter.TryUnsubscribe<BarSignal>(BarAction);
+			bool isUnsubscribedFromFoo = _signalCenter.TryUnsubscribe<FooSignal>(FooAction);
+			Assert.IsFalse(_signalCenter.Exists<FooSignal>(FooAction));
+			Assert.IsFalse(_signalCenter.Exists<BarSignal>(BarAction));
+			Assert.IsTrue(isUnsubscribedFromBar);
+			Assert.IsFalse(isUnsubscribedFromFoo);
+		}
+
 		[Test]
 		public void Subscribe_And_Fire()
 		{