WeakContainer.h

Go to the documentation of this file.

 
#pragma once
#include "SpinLock.h"
#include "catapult/functions.h"
#include <list>
#include <memory>
 
namespace catapult { namespace utils {
 
    template<typename T>
    class WeakContainer {
    public:
        WeakContainer() : WeakContainer([](const auto&) {})
        {}
 
        WeakContainer(const consumer<T&>& close) : m_close(close)
        {}
 
    public:
        size_t size() const {
            SpinLockGuard guard(m_lock);
            const_cast<WeakContainer*>(this)->pruneInternal();
            return m_entries.size();
        }
 
        void insert(const std::weak_ptr<T>& pEntry) {
            SpinLockGuard guard(m_lock);
            pruneInternal();
            m_entries.push_back(pEntry);
        }
 
        void clear() {
            SpinLockGuard guard(m_lock);
            for (const auto& pEntry : m_entries) {
                auto pSharedEntry = pEntry.lock();
                if (!pSharedEntry)
                    continue;
 
                m_close(*pSharedEntry);
            }
 
            m_entries.clear();
        }
 
    private:
        void pruneInternal() {
            m_entries.remove_if([](const auto& pExistingEntry) { return !pExistingEntry.lock(); });
        }
 
    private:
        consumer<T&> m_close;
        std::list<std::weak_ptr<T>> m_entries;
        mutable utils::SpinLock m_lock;
    };
}}