Lifetime tracking for ListenerSet.

Author: Mysticial

Common/Cpp/ListenerSet.h

@@ -10,11 +10,15 @@
 #include <map>
 #include <atomic>
 #include "Common/Cpp/Concurrency/SpinLock.h"
+#include "Common/Cpp/LifetimeSanitizer.h"
 
 //#include <iostream>
 //using std::cout;
 //using std::endl;
 
+#define PA_DEBUG_ListenerSet
+
+
 namespace PokemonAutomation{
 
 
@@ -30,12 +34,18 @@ class ListenerSet{
     }
 
     void add(ListenerType& listener){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         WriteSpinLock lg(m_lock);
 //        std::lock_guard<std::mutex> lg(m_lock);
         m_listeners[&listener]++;
         m_count.store(m_listeners.size(), std::memory_order_relaxed);
     }
     void remove(ListenerType& listener){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         WriteSpinLock lg(m_lock);
 //        std::lock_guard<std::mutex> lg(m_lock);
         auto iter = m_listeners.find(&listener);
@@ -50,6 +60,9 @@ class ListenerSet{
 
     template <typename Lambda>
     void run_lambda_unique(Lambda&& lambda){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         if (empty()){
             return;
         }
@@ -61,6 +74,9 @@ class ListenerSet{
     }
     template <typename Lambda>
     void run_lambda_with_duplicates(Lambda&& lambda){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         if (empty()){
             return;
         }
@@ -77,6 +93,9 @@ class ListenerSet{
 
     template <typename Function, class... Args>
     void run_method_unique(Function function, Args&&... args){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         if (empty()){
             return;
         }
@@ -88,6 +107,9 @@ class ListenerSet{
     }
     template <typename Function, class... Args>
     void run_method_with_duplicates(Function function, Args&&... args){
+#ifdef PA_DEBUG_ListenerSet
+        auto scope = m_sanitizer.check_scope();
+#endif
         if (empty()){
             return;
         }
@@ -110,6 +132,10 @@ class ListenerSet{
     mutable SpinLock m_lock;
 //    mutable std::mutex m_lock;
     std::map<ListenerType*, size_t> m_listeners;
+
+#ifdef PA_DEBUG_ListenerSet
+    LifetimeSanitizer m_sanitizer;
+#endif
 };