C++11新特性之智能指针

很多人谈到c++，说它特别难，可能有一部分就是因为c++的内存管理吧，不像java那样有虚拟机动态的管理内存，在程序运行过程中可能就会出现内存泄漏，然而这种问题其实都可以通过c++11引入的智能指针来解决，相反我还认为这种内存管理还是c++语言的优势，因为尽在掌握。

c++11引入了三种智能指针：

• std::shared_ptr
• std::weak_ptr
• std::unique_ptr

shared_ptr

shared_ptr 使用了引用计数，每一个 shared_ptr 的拷贝都指向相同的内存，每次拷贝都会触发引用计数 + 1，每次生命周期结束析构的时候引用计数 - 1，在最后一个 shared_ptr 析构的时候，内存才会释放。

使用方法如下：

struct ClassWrapper {

    ClassWrapper() {
        cout << "construct" << endl;
        data = new int[10];
    }

    ~ClassWrapper() {
        cout << "deconstruct" << endl;
        if (data != nullptr) {
            delete[] data;
        }
    }

    void Print() {
        cout << "print" << endl;
    }

    int* data;
};

void Func(std::shared_ptr<ClassWrapper> ptr) {
    ptr->Print();
}

int main() {
    auto smart_ptr = std::make_shared<ClassWrapper>();
    auto ptr2 = smart_ptr; // 引用计数+1
    ptr2->Print();
    Func(smart_ptr); // 引用计数+1
    smart_ptr->Print();
    ClassWrapper *p = smart_ptr.get(); // 可以通过get获取裸指针
    p->Print();
    return 0;
}

智能指针还可以自定义删除器，在引用计数为0的时候自动调用删除器来释放对象的内存，代码如下：

std::shared_ptr<int> ptr(new int, [](int *p){ delete p; });

关于shared_ptr有几点需要注意：

• 不要用一个裸指针初始化多个shared_ptr，会出现double_free导致程序崩溃。
• 通过shared_from_this()返回this指针，不要把this指针作为shared_ptr返回出来，因为this指针本质就是裸指针，通过this返回可能会导致重复析构，不能把this指针交给智能指针管理。

class A {
    shared_ptr<A> GetSelf() {
        return shared_from_this();
        // return shared_ptr<A>(this); 错误，会导致double free
    }
};

• 尽量使用make_shared，少用new。
• 不要delete get() 返回来的裸指针。
• 不是new出来的空间要自定义删除器。
• 要避免循环引用，循环引用导致内存永远不会被释放，造成内存泄漏。

using namespace std;
struct A;
struct B;

struct A {
    std::shared_ptr<B> bptr;
    ~A() {
        cout << "A delete" << endl;
    }
};

struct B {
    std::shared_ptr<A> aptr;
    ~B() {
        cout << "B delete" << endl;
    }
};

int main() {
    auto aaptr = std::make_shared<A>();
    auto bbptr = std::make_shared<B>();
    aaptr->bptr = bbptr;
    bbptr->aptr = aaptr;
    return 0;
}

上面代码，产生了循环引用，导致aptr和bptr的引用计数为2，离开作用域后aptr和bptr的引用计数 - 1，但是永远不会为0，导致指针永远不会析构，产生了内存泄漏，如何解决这种问题呢，答案是使用weak_ptr。

weak_ptr

weak_ptr是用来监视shared_ptr的生命周期，它不管理shared_ptr内部的指针，它的拷贝的析构都不会影响引用计数，纯粹是作为一个旁观者监视shared_ptr中管理的资源是否存在，可以用来返回this指针和解决循环引用问题。

• 作用1： 返回this指针，上面介绍的shared_from_this()其实就是通过weak_ptr返回的this指针。

• 作用2： 解决循环引用问题。

struct A;
struct B;

struct A {
    std::shared_ptr<B> bptr;
    ~A() {
        cout << "A delete" << endl;
    }
    void Print() {
        cout << "A" << endl;
    }
};

struct B {
    std::weak_ptr<A> aptr; // 这里改成weak_ptr
    ~B() {
        cout << "B delete" << endl;
    }
    void PrintA() {
        if (!aptr.expired()) { // 监视shared_ptr的生命周期
            auto ptr = aptr.lock();
            ptr->Print();
        }
    }
};

int main() {
    auto aaptr = std::make_shared<A>();
    auto bbptr = std::make_shared<B>();
    aaptr->bptr = bbptr;
    bbptr->aptr = aaptr;
    bbptr->PrintA();
    return 0;
}

输出：

A
A delete
B delete

unique_ptr

std::unique_ptr是一个独占型的智能指针，它不允许其它智能指针共享其内部指针，也不允许unique_ptr的拷贝和赋值。使用方法和shared_ptr类似，区别是不可以拷贝：

using namespace std;

struct A {
    ~A() {
    cout << "A delete" << endl;
}
void Print() {
    cout << "A" << endl;
}
};


int main() {
    auto ptr = std::unique_ptr<A>(new A);
    auto tptr = std::make_unique<A>(); // error, c++11还不行，需要c++14
    std::unique_ptr<A> tem = ptr; // error, unique_ptr不允许移动
    ptr->Print();
    return 0;
}

unique_ptr也可以像shared_ptr一样自定义删除器，使用方法和shared_ptr相同。

---

智能指针可以自动管理内存，shared_ptr有引用计数，unique_ptr为唯一智能指针，但是究竟什么时候使用呢？

按我的理解，unique_ptr它就是一个典型的RAII类，与unique_lock，lock_guard等类似，主要用于解决函数可能有多个返回点，及可能存在异常退出的问题，保证资源可以安全释放且代码干净，一般用在不涉及多线程，且生命周期的管理很明确的地方(虽然也可以把管理权move走)。而shared_ptr常在多线程场景中，比方说多个线程都拥有对这个资源的生命周期管理权，任何线程只要还需要用它就必须保证有效。而weak_ptr典型例子循环引用就不说了，主要说它的本质在于和shared_ptr的不同，持有shared_ptr的线程，不仅有资源的使用权也有资源的管理权；而持有weak_ptr的线程只有使用权，不应该由它管理资源的生命周期，所以此时该线程需要某种方式来判断资源的有效性，有效才能用，那就是expired方法了。weak_ptr的典型例子比如观察者模式中，被观察对象要维护一个观察者序列来notify，如果用裸指针来存储那根本无法得知资源是否有效，如果用shared_ptr来存储，那观察者就没法销毁了，所以用weak_ptr来存储，可以使用资源，但不该管理其生命周期。