Fix IPoolObject should only return back to it's owning pool

Author: austinv900

src/Pooling/ArrayPool.cs

@@ -70,8 +70,6 @@ public void Free(object item)
                 return;
             }
 
-            IPoolSource<T> elementSource = _arePooledItems ? PoolFactory.GetSource<T>() : null;
-
             for (int i = 0; i < items.Length; i++)
             {
                 object itm = items[i];
@@ -85,10 +83,6 @@ public void Free(object item)
                     {
                         obj.Source.Free(obj);
                     }
-                    else if (elementSource != null)
-                    {
-                        elementSource.Free(obj);
-                    }
                 }
             }
 

src/Pooling/IPoolObject.cs

@@ -7,7 +7,7 @@ public interface IPoolObject : IDisposable
         /// <summary>
         /// The source pool this item came from
         /// </summary>
-        IPoolSource Source { get; }
+        IPoolSource Source { get; set; }
 
         /// <summary>
         /// Instruct the <see cref="IPoolSource"/> to ignore this object

src/Pooling/ObjectPool.cs

@@ -9,26 +9,37 @@ internal class ObjectPool<T> : IPoolSource<T> where T : class
 
         private readonly Type _poolType;
         private readonly List<T> _pool;
+        private readonly bool _isPooledInterface;
 
         public ObjectPool()
         {
             _poolType = typeof(T);
             _pool = new List<T>();
+            _isPooledInterface = typeof(IPoolObject).IsAssignableFrom(typeof(T));
         }
 
         public T Get()
         {
+            T item;
             lock (_pool)
             {
                 if (_pool.Count == 0)
                 {
-                    return (T)Activator.CreateInstance(typeof(T));
+                    item = (T)Activator.CreateInstance(typeof(T));
                 }
+                else
+                {
+                    item = _pool[0];
+                    _pool.RemoveAt(0);
+                }
+            }
 
-                T item = _pool[0];
-                _pool.RemoveAt(0);
-                return item;
+            if (_isPooledInterface && item is IPoolObject pooled)
+            {
+                pooled.Source = this;
             }
+
+            return item;
         }
 
         public void Free(object item)