From d674f2da8b5bb6e1f33edd1bfe2a1c0d50e54bfd Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Thu, 29 Jun 2023 23:59:45 +0800
Subject: [PATCH] Implement Add and Del callbacks for Hybrid components.

---
 src/Pools/EcsHybridPool.cs | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/src/Pools/EcsHybridPool.cs b/src/Pools/EcsHybridPool.cs
index 2653295..c620ed1 100644
--- a/src/Pools/EcsHybridPool.cs
+++ b/src/Pools/EcsHybridPool.cs
@@ -18,9 +18,6 @@ namespace DCFApixels.DragonECS
         private int[] _recycledItems;
         private int _recycledItemsCount;
 
-        private IEcsComponentReset<T> _componentResetHandler = EcsComponentResetHandler<T>.instance;
-        private IEcsComponentCopy<T> _componentCopyHandler = EcsComponentCopyHandler<T>.instance;
-
         private List<IEcsPoolEventListener> _listeners = new List<IEcsPoolEventListener>();
 
         #region Properites
@@ -67,6 +64,7 @@ namespace DCFApixels.DragonECS
             }
             this.IncrementEntityComponentCount(entityID);
             _listeners.InvokeOnAdd(entityID);
+            component.OnAddToPool();
             _items[itemIndex] = component;
         }
         public void Set(int entityID, T component)
@@ -90,10 +88,10 @@ namespace DCFApixels.DragonECS
             else
             {//not null
                 _listeners.InvokeOnDel(entityID);
-                if (_items[itemIndex] is IDisposable disposable)
-                    disposable.Dispose();
+                component.OnDelFromPool();
             }
             _listeners.InvokeOnAdd(entityID);
+            component.OnAddToPool();
             _items[itemIndex] = component;
         }
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
@@ -125,8 +123,7 @@ namespace DCFApixels.DragonECS
 #endif
             ref int itemIndex = ref _mapping[entityID];
             T component = _items[itemIndex];
-            if (component is IDisposable disposable)
-                disposable.Dispose();
+            component.OnDelFromPool();
             if (_recycledItemsCount >= _recycledItems.Length)
                 Array.Resize(ref _recycledItems, _recycledItems.Length << 1);
             _recycledItems[_recycledItemsCount++] = itemIndex;