From cb18ba471f544ef409870f969ec9043f9f3ceeb3 Mon Sep 17 00:00:00 2001
From: Chris Rodgers <chris@teamrodgers.uk>
Date: Wed, 23 May 2018 20:30:39 +0100
Subject: [PATCH] Added locks to ObjectDataCollection

---
 Effort.Extra/ObjectDataCollection.cs | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/Effort.Extra/ObjectDataCollection.cs b/Effort.Extra/ObjectDataCollection.cs
index ed389b4..0bf33ab 100644
--- a/Effort.Extra/ObjectDataCollection.cs
+++ b/Effort.Extra/ObjectDataCollection.cs
@@ -9,6 +9,8 @@ namespace Effort.Extra
     /// </summary>
     internal class ObjectDataCollection : KeyedCollection<Guid, ObjectData>
     {
+        private readonly object locker = new object();
+
         /// <summary>
         /// Extracts the key from the specified element.
         /// </summary>
@@ -28,9 +30,12 @@ protected override Guid GetKeyForItem(ObjectData item)
         /// <param name="data">The data.</param>
         public void AddOrUpdate(ObjectData data)
         {
-            if (data == null) throw new ArgumentNullException(nameof(data));
-            if (Contains(data.Identifier)) Remove(data.Identifier);
-            Add(data);
+            lock (locker)
+            {
+                if (data == null) throw new ArgumentNullException(nameof(data));
+                if (Contains(data.Identifier)) Remove(data.Identifier);
+                Add(data);
+            }
         }
 
         /// <summary>
@@ -41,8 +46,11 @@ public void AddOrUpdate(ObjectData data)
         /// <returns><c>true</c>, if the key exists, otherwise <c>false</c>.</returns>
         public bool TryGetValue(Guid key, out ObjectData data)
         {
-            data = Contains(key) ? this[key] : null;
-            return data != null;
+            lock (locker)
+            {
+                data = Contains(key) ? this[key] : null;
+                return data != null;
+            }
         }
     }
 }
\ No newline at end of file