C++的defer实现

关于defer

  在和朋友交谈时候，无意间了解到Go语言的defer，发现挺有意思的。和智能指针类似，当出了作用域后，被defer修饰的操作才会执行。下面贴一段伪代码：

	{
		std::ofstream ofs;
		defer ofs.close(); // 当出了作用域后，会自己执行ofs.close来关闭文件
		ofs.open("test.txt");
		ofs.write(xxx);
	}

  当出了作用域后，defer修饰的操作才会执行。当逻辑代码出现大量的return或者try_catch时候，用上defer无疑会省去大量的重复代码。

实现一

出了作用域后才执行，最容易想到的就是智能指针了

using defer = std::shared_ptr<void>;
{
		std::ofstream ofs;
		defer (nullptr, [&](...){
			std::cout << "ofs_closed" << std::endl;
			ofs.close();
		});	
		ofs.open("test.txt");
		ofs.write(xxx);
}

实现二

类管理一个std::function对象，析构时候执行function

class Defer
{
public:
	explicit Defer(std::function<void()> && func)
	{
		m_func = std::move(func);
	}

	~Defer()
	{
		m_func();
	}

private:
	std::function<void()> m_func;
};

#define defer_name(name, count) name##count##private
#define defer_link(class, count) defer_name(class, count)
#define defer(expr) auto defer_link(Defer, __COUNTER__) = Defer(expr)

{
		std::ofstream ofs;
		defer ([&]{
			std::cout << "ofs_closed" << std::endl;
			ofs.close();
		});	
		ofs.open("test.txt");
		ofs.write(xxx);
}