WeakPtr.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
 
AST_NAMESPACE_BEGIN
 
 
template<typename _Object>
class WeakPtr
{
public:
    WeakPtr()
        :m_object{ nullptr }
    {
    }
    WeakPtr(std::nullptr_t)
        :m_object{ nullptr }
    {
    }
    WeakPtr(_Object* obj)
        :m_object{ obj }
    {
        _incWeakRef();
    }
    WeakPtr(const WeakPtr& ptr)
        :WeakPtr{ ptr.m_object }
    {
    }
    ~WeakPtr()
    {
        _decWeakRef();
    }
    WeakPtr& operator=(_Object* obj)
    {
        if (obj != m_object) {
            _decWeakRef();
            m_object = obj;
            _incWeakRef();
        }
        return *this;
    }
    WeakPtr& operator=(std::nullptr_t)
    {
        reset();
        return *this;
    }
    WeakPtr& operator=(const WeakPtr& ptr)
    {
        if (this != &ptr) {
            this->operator=(ptr.m_object);
        }
        return *this;
    }
    _Object* lock() const
    {
        return get();
    }
    _Object* get() const
    {
        if (expired()) {
            return nullptr;
        }
        return m_object;
    }
    bool expired() const
    {
        return !m_object || m_object->isDestructed();
    }
    void reset()
    {
        _decWeakRef();
        m_object = nullptr;
    }
private:
    void _incWeakRef()
    {
        if(m_object)
            m_object->incWeakRef();
    }
    void _decWeakRef()
    {
        if(m_object)
            m_object->decWeakRef();
    }
protected:
    _Object* m_object{nullptr};
};
 
 
 
AST_NAMESPACE_END