快捷導(dǎo)航

C++實(shí)現(xiàn)Go的defer功能(示例代碼)

更新時(shí)間：2021年07月11日 09:39:47 作者：witton

defer和go一樣都是Go語(yǔ)言提供的關(guān)鍵字。defer用于資源的釋放，會(huì)在函數(shù)返回之前進(jìn)行調(diào)用。接下來(lái)通過(guò)本文給大家介紹C++實(shí)現(xiàn)Go的defer功能，感興趣的朋友跟隨小編一起看看吧

在Go語(yǔ)言中有一個(gè)關(guān)鍵字：defer，它的作用就是延遲執(zhí)行后面的函數(shù)，在資源釋放方面特別有用，比如下面一段C/C++的示例代碼：

void test()
{
	FILE* fp = fopen("test.txt", "r");
	if (nullptr == fp)
		return;

	if (...)
	{
		fclose(fp);
		return;
	}
	if (...)
	{
		fclose(fp);
		return;
	}
	if (...)
	{
		fclose(fp);
		return;
	}
	fclose(fp);
}

在每一處返回之前都需要調(diào)用fclose來(lái)關(guān)閉文件句柄，中間的流程中斷越多，越是容易遺漏調(diào)用fclose導(dǎo)致未正常關(guān)閉文件。

C++可以使用shared_ptr，auto_ptr之類的智能指針來(lái)管理分配的內(nèi)存，但是像上面這種情況C++并沒(méi)有現(xiàn)成的可使用的代碼來(lái)處理。而Go語(yǔ)言提供了defer關(guān)鍵字來(lái)解決此類問(wèn)題，Go可以按如下方式來(lái)寫(xiě)：

func test() {
	file, err := os.Open("test.txt")
	if err != nil {
		return
	}
	defer file.Close()
	if ... {
		return
	}
	if ... {
		return
	}
	if ... {
		return
	}
}

只需要使用一句：

defer file.Close()

即可，Go會(huì)自動(dòng)在return之后調(diào)用defer后面的函數(shù)。我們?cè)倏纯聪旅娴氖纠?/p>

package main

import (
	"fmt"
)

func test() (n int, err error) {
	defer fmt.Println("測(cè)試1")
	defer fmt.Println("測(cè)試2")
	defer fmt.Println("測(cè)試3")
	return fmt.Println("test")
}

func main() {
	test()
}

它的輸出為：

test
測(cè)試3
測(cè)試2
測(cè)試1

在這里插入圖片描述

可以看出有多個(gè)defer時(shí)，按照先進(jìn)后出的方式執(zhí)行的。

C++中我們可以利用析構(gòu)函數(shù)來(lái)實(shí)現(xiàn)，而且C++的局部變量析構(gòu)規(guī)則也是按照先進(jìn)后出的方式執(zhí)行的。為此，我們需要定義一個(gè)Defer類：

#include <functional>
typedef std::function<void()> fnDefer;
class Defer
{
public:
	Defer(fnDefer fn) : m_fn(fn)
	{
	}
	~Defer()
	{
		if(m_fn)
			m_fn();
	}
private:
	fnDefer m_fn;
};

這樣，前面的C++示例代碼可以寫(xiě)成：

void test()
{
	FILE* fp = fopen("test.txt", "r");
	if (nullptr == fp)
		return;

	Defer d([&]()
	{
		fclose(fp);
	});
	if (...)
	{
		return;
	}
	if (...)
	{
		return;
	}
	if (...)
	{
		return;
	}
}

不用再在每一處返回前手動(dòng)寫(xiě)代碼關(guān)閉文件了。

但是這里還有一點(diǎn)不便之處就是需要手寫(xiě)一個(gè)lambda表達(dá)式和手動(dòng)定義一個(gè)變量，這個(gè)很好解決，使用宏來(lái)處理。

#define defer1(a,b) a##b
#define defer2(a, b) defer1(a, b)
#define defer(expr) Defer defer2(__Defer__,__COUNTER__) ([&](){expr;})

為了方便在同一函數(shù)多處使用，定義了defer宏來(lái)給變量命不同的名，前面的代碼可以改為：

void test()
{
	FILE* fp = fopen("test.txt", "r");
	if (nullptr == fp)
		return;

	defer(fclose(fp));
	if (...)
	{
		return;
	}
	if (...)
	{
		return;
	}
	if (...)
	{
		return;
	}
}

這樣就實(shí)用且方便得多了。下面給出完整代碼以及測(cè)試用例：

#include <functional>

using namespace std;

typedef std::function<void()> fnDefer;
class Defer
{
public:
	Defer(fnDefer fn) : m_fn(fn)
	{
	}
	~Defer()
	{
		if(m_fn)
			m_fn();
	}
private:
	fnDefer m_fn;
};

#define defer1(a,b) a##b
#define defer2(a, b) defer1(a, b)
#define defer(expr) Defer defer2(__Defer__,__COUNTER__) ([&](){expr;})

class Test
{
public:
	void f(int i)
	{
		printf("f:%d %p\n", i, this);
	}
};

int main(int argc, char *argv[])
{
	Test t;
	printf("test:%p\n", &t);
	defer(t.f(1));
	defer(t.f(2));
	defer(t.f(3));

	return 0;
}

結(jié)果如下：

在這里插入圖片描述