在C++中,有三种类型的循环语句:for, while, 和do...while, 但是在一般应用中作循环时, 我们可能用for和while要多一些,do...while相对不受重视。
但是,最近在读我们项目的代码时,却发现了do...while的一些十分聪明的用法,不是用来做循环,而是用作其他来提高代码的健壮性。
1. do...while(0)消除goto语句。
通常,如果在一个函数中开始要分配一些资源,然后在中途执行过程中如果遇到错误则退出函数,当然,退出前先释放资源,我们的代码可能是这样:
version 1
bool
Execute()
{
// 分配资源
int * p = new int ;
bool bOk( true );
// 执行并进行错误处理
bOk = func1();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
bOk = func2();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
bOk = func3();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
// ..........
// 执行成功,释放资源并返回
delete p;
p = NULL;
return true ;
}
{
// 分配资源
int * p = new int ;
bool bOk( true );
// 执行并进行错误处理
bOk = func1();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
bOk = func2();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
bOk = func3();
if ( ! bOk)
{
delete p;
p = NULL;
return false ;
}
// ..........
// 执行成功,释放资源并返回
delete p;
p = NULL;
return true ;
}
这里一个最大的问题就是代码的冗余,而且我每增加一个操作,就需要做相应的错误处理,非常不灵活。于是我们想到了goto:
version 2
bool
Execute()
{
// 分配资源
int * p = new int ;
bool bOk( true );
// 执行并进行错误处理
bOk = func1();
if ( ! bOk) goto errorhandle;
bOk = func2();
if ( ! bOk) goto errorhandle;
bOk = func3();
if ( ! bOk) goto errorhandle;
// ..........
// 执行成功,释放资源并返回
delete p;
p = NULL;
return true ;
errorhandle:
delete p;
p = NULL;
return false ;
}
{
// 分配资源
int * p = new int ;
bool bOk( true );
// 执行并进行错误处理
bOk = func1();
if ( ! bOk) goto errorhandle;
bOk = func2();
if ( ! bOk) goto errorhandle;
bOk = func3();
if ( ! bOk) goto errorhandle;
// ..........
// 执行成功,释放资源并返回
delete p;
p = NULL;
return true ;
errorhandle:
delete p;
p = NULL;
return false ;
}
代码冗余是消除了,但是我们引入了C++中身份比较微妙的goto语句,虽然正确的使用goto可以大大提高程序的灵活性与简洁性,但太灵活的东西往往是很危险的,它会让我们的程序捉摸不定,那么怎么才能避免使用goto语句,又能消除代码冗余呢,请看do...while(0)循环:
version3
bool
Execute()
{
// 分配资源
int * p = new int ;
bool bOk( true );
do
{
// 执行并进行错误处理
bOk = func1();
if ( ! bOk) break ;
bOk = func2();
if ( ! bOk) break ;
bOk = func3();
if ( ! bOk) break ;
// ..........
} while ( 0 );
// 释放资源
delete p;
p = NULL;
return bOk;
}
{
// 分配资源
int * p = new int ;
bool bOk( true );
do
{
// 执行并进行错误处理
bOk = func1();
if ( ! bOk) break ;
bOk = func2();
if ( ! bOk) break ;
bOk = func3();
if ( ! bOk) break ;
// ..........
} while ( 0 );
// 释放资源
delete p;
p = NULL;
return bOk;
}
“漂亮!”, 看代码就行了,啥都不用说了...
为了宏展开的时候不会出错。
如过直接放在花括号会出错的,不如
1
|
#define DO_SOMETHING {int time = 100 ;make_love(i);}
|
下面是使用的地方
1
2
|
if
(you->strong())DO_SOMETHING;
else
..;
|
展开后是这样的
1
2
3
4
|
if
(you->strong())
{
int
time
= 100 ;make_love(i);}
;
else
..;
|
那么编译就会出错,可以验证使用d0{}while(0)就不会出错
linux内核源码经常见这玩意
1
2
3
4
|
#define AB1 a; b; // x, 下面语句b不能被执行: if (cond) AB1;
#define AB2 { a; b; } // x, 下面语句编译出错:if (cond) AB2; else ...;
#define AB3 a, b // x, 有运算符优先级问题
#define AB4 do { a; b; } while (0)
|
前面几个都是有问题的,只有do while(0)解决得比较好。