首页 > 代码库 > boost的shared_ptr循环引用

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;
}