一种错误处理宏的应用

在日常的一些代码中，常常会出现需要进行一系列的初始化和释放操作的情况，而在每次的初始化之后一般都要判断指针是否为空、内存分配是否失败等情况，如果失败的话就要返回，就像下面这个函数一样：

DWORD Function() { DWORD dwError = 0; CObjectA *pA = new CObjectA(); if (pA == NULL) return err_1; CObjectB *pB = new CObjectB(); if (pB == NULL) { delete pA; return err_2; } …… if (pB) delete pB; if (pA) delete pA; return dwError; } 

如果过程中初始化和释放的步骤较多，代码就会显得比较臃肿，而且由于return语句较多，也不是一种很好的风格，因为一般来讲，我们还是喜欢单入口和单出口的函数，这样显得可读性会更好。

有没有更好的写法呢？前几天我的一个同事提出了一种不错的解决方案，通过定义一系列的宏来解决这个问题，我看了一下，觉得很不错，于是又在这个方法之上进行了一些扩展。比如，如果我们希望对指针类型的返回值进行一系列的检查，可以进行如下的定义：

#define BR_SUCCEEDED(ret) (((DWORD)(ret)) == 0) #define BR_FAILED(ret) (((DWORD)(ret)) != 0) #define BR_RETURNCODE_NAME dwError #define __BR_ERRORLABLE Error #define BR_ERRORLABLE() __BR_ERRORLABLE #define BR_DECLARE_ERRORHANDLER / Error: #define BR_DECLARE_ERRORHANDLER_EX(n) / BR_ERRORLABLE()##n: #define BR_DEFAULT_ERRORRETURN / return BR_RETURNCODE_NAME; #define BR_DEFAULT_VOIDRETURN / return; // 检查申请内存返回的指针，出错时跳转到错误处理 #define BR_CPR(ret) / if ((ret) == NULL) / { / BR_RETURNCODE_NAME = ERR_UNKNOWN; / goto __BR_ERRORLABLE; / } // 检查申请内存返回的指针，出错时设置自定义返回值并跳转到错误处理 #define BR_CPR2(ret, code) / if ((ret) == NULL) / { / BR_RETURNCODE_NAME = (DWORD)(code); / goto __BR_ERRORLABLE; / } // 检查申请内存返回的指针，出错时设置自定义返回值并跳转到指定错误处理 #define BR_CPR3(ret, code, n) / if ((ret) == NULL) / { / BR_RETURNCODE_NAME = (DWORD)(code); / goto BR_ERRORLABLE()##n; / }  

这样，我们最初的代码就可以写成这样：

DWORD Function() { DWORD dwError = 0; CObjectA *pA = new CObjectA(); BR_CPR2(pA, err_1); CObjectB *pB = new CObjectB(); BR_CPR3(pA, err_1, 2); …… BR_DECLARE_ERRORHANDLER_EX(3) if (pB) delete pB; BR_DECLARE_ERRORHANDLER_EX(2) if (pA) delete pA; BR_DECLARE_ERRORHANDLER BR_DEFAULT_ERRORRETURN; } 

看看，是不是顺眼了很多？我们同样可以针对BOOL等类型的返回值写出类似的宏。