boost的shared_ptr循环引用

boost的智能指针给编程带来了极大的便利，不需要关心内存的释放，不要要调用delete，而且还可以定制delete的方法。其实boost的智能指针是可以当成scope_exit来用的，同样是退出时处理。但是凡事都是有利有弊，boost的shared_ptr如果在循环引用的时候会出现无法释放内存的情况，所谓循环引用就是A智能指针类里存放B的智能指针，B的智能指针类里存放A，将a、b的值互相设置。增加引用计数，在释放的时候由于计数问题，会导致在退出指针域的时候无法进行释放，解决该问题的方案是在类成员中使用弱指针。见如下例子

// circleRef.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"

#include <iostream>
#include <boost/shared_ptr.hpp>
#include <boost/weak_ptr.hpp>

using namespace std;

class ObjectB;

class ObjectA
{
public:
	~ObjectA()
	{
		std::cout<<"dctor ~ObjectA"<<std::endl;
	}

	void setObjectB(boost::shared_ptr<ObjectB> b)
	{
		m_objB = b;
	}
private:
	boost::shared_ptr<ObjectB> m_objB;
};

class ObjectB
{
public:
	~ObjectB()
	{
		cout<<"dctor ~ObjectB"<<endl;
	}

	void setObjectA(boost::shared_ptr<ObjectA> a)
	{
		m_objA = a;
	}
private:
	boost::shared_ptr<ObjectA> m_objA;
};

void test()
{
	boost::shared_ptr<ObjectA> a(new ObjectA);
	boost::shared_ptr<ObjectB> b(new ObjectB);

	a->setObjectB(b);
	b->setObjectA(a);
};

int _tmain(int argc, _TCHAR* argv[])
{
	test();
	printf("test ");
	getchar();
	return 0;
}

程序执行，不会调用各自的析构函数。若将其改为weak_ptr则不会引入多余计数的问题，从而可以保证对象被正确释放，看如下修改后的正确代码：

// circleRef.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"

#include <iostream>
#include <boost/shared_ptr.hpp>
#include <boost/weak_ptr.hpp>

using namespace std;

class ObjectB;

class ObjectA
{
public:
	~ObjectA()
	{
		std::cout<<"dctor ~ObjectA"<<std::endl;
	}

	void setObjectB(boost::shared_ptr<ObjectB> b)
	{
		m_objB = b;
	}
private:
	boost::weak_ptr<ObjectB> m_objB;
};

class ObjectB
{
public:
	~ObjectB()
	{
		cout<<"dctor ~ObjectB"<<endl;
	}

	void setObjectA(boost::shared_ptr<ObjectA> a)
	{
		m_objA = a;
	}
private:
	boost::weak_ptr<ObjectA> m_objA;
};

void test()
{
	boost::shared_ptr<ObjectA> a(new ObjectA);
	boost::shared_ptr<ObjectB> b(new ObjectB);

	a->setObjectB(b);
	b->setObjectA(a);
};

int _tmain(int argc, _TCHAR* argv[])
{
	test();
	printf("test ");
	getchar();
	return 0;
}