最近写YAFtp遇到如下情景
但是,最近在读我们项目的代码时,却发现了do...while的一些十分聪明的用法,不是用来做循环,而是用作其他来提高代码的健壮性。
void* retr_thread(void* arg)
{
char* filename = (char*) arg;
DATA_FD_LOCK(client_conn->datafd_mutex);
FILE* fp = fopen(filename, "r");
if (fp == NULL)
{
free(filename);
DATA_FD_UNLOCK(client_conn->datafd_mutex);
return NULL;
}
if (client_conn->datafd == -1)
{
fclose(fp);
free(filename);
DATA_FD_UNLOCK(client_conn->datafd_mutex);
return NULL;
}
char buf[1024];
while(!feof(fp))
{
size_t n = fread(buf, sizeof(*buf), sizeof(buf), fp);
send(client_conn->datafd, buf, n, 0);
}
/* do clean */
free(filename);
close(client_conn->datafd);
client_conn->datafd = -1;
DATA_FD_UNLOCK(client_conn->datafd_mutex);
pthread_exit(NULL);
return NULL;
}
针对不同异常情况,需要作出重复的资源释放。今日在网上看到一种比goto更加优雅的做法,转载如下。
do...while(0)的妙用
在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;
}
“漂亮!”, 看代码就行了,啥都不用说了...