写出健壮�?Bash 脚本

Wed, 09 Jan 2013 04:11:00 GMT

许多人用shell脚本完成一些简单�Q务，而且变成了他们生命的一部分。不�q�的是，shell脚本在运行异常时会受到非常大的媄响。在写脚本时��这�c�问题最��化是十分必要的。本文中我将介绍一些让bash脚本变得健壮的技术�?/p>

使用set -u

你因为没有对变量初始化而��脚本崩溃�q�多��次�Q�对于我来说�Q�很多次�?/p>

chroot=$1 ... rm -rf $chroot/usr/share/doc

如果上面的代码你没有�l�参数就�q�行�Q�你不会仅仅删除掉chroot中的文档�Q�而是��系�l�的所有文档都删除。那你应该做些什么呢�Q�好在bash提供�?em>set -u�Q�当你��用未初始化的变量�Ӟ��让bash自动退出。你也可以��用可��L��更��Z��点的set -o nounset�?/p>

david% bash /tmp/shrink-chroot.sh

$chroot=

david% bash -u /tmp/shrink-chroot.sh

/tmp/shrink-chroot.sh: line 3: $1: unbound variable

david%

使用set -e

你写的每一个脚本的开始都应该包含set -e。这告诉bash一但有��M��一个语句返回非真的��|��则退出bash。��?e的好处是避免错误滚雪球般的变成严重错误，能尽早的捕获错误。更加可�ȝ��版本�Q?em>set -o errexit

使用-e把你从检查错误中解放出来。如果你忘记了检查，bash会替你做�q��g事。不�q�你也没有办法��?em>$?来获取命令执行状态了�Q�因为bash无法获得��M��?的返回倹{��你可以使用另一�U�结构：

command

if [ "$?"-ne 0]; then echo "command failed"; exit 1; fi

可以替换成：

command || { echo "command failed"; exit 1; }

或者��用：

if ! command; then echo "command failed"; exit 1; fi

如果你必��M��用返回非0值的命��o�Q�或者你对返回值�ƈ不感兴趣呢？你可以��?command || true �Q�或者你有一�D�很长的代码�Q�你可以暂时关闭错误��查功能，不过我徏议你谨慎使用�?/p>

set +e

command1

command2

set -e

相关文档指出�Q�bash默认�q�回��道中最后一个命令的��|��也许是你不想要的那个。比如执�?false | true ��会被认为命令成功执行。如果你惌��q�样的命令被认�ؓ是执行失败，可以使用 set -o pipefail

�E�序防�M - 考虑意料之外的事

你的脚本也许会被攑ֈ�“意外”的�̎户下�q�行�Q�像�~�少文�g或者目录没有被创徏�{�情��c��你可以做一些预防这些错误事情。比如，当你创徏一个目录后�Q�如果父目录不存在，mkdir 命��o会返回一个错误。如果你创徏目录时给mkdir命��o加上-p选项�Q�它会在创徏需要的目录前，把需要的父目录创建出来。另一个例子是 rm 命��o。如果你要删除一个不存在的文�Ӟ��它会“吐槽”�q�且你的脚本会停止工作。（因�ؓ你��用了-e选项�Q�对吧？�Q�你可以使用-f选项来解册��个问题，在文件不存在的时候让脚本�l�箋工作�?nbsp;

准备好处理文件名中的�I�格

有些��Z��在文件名或者命令行参数中��用空��|��你需要在�~�写脚本时时刻记得这件事。你需要时刻记得用引号包围变量�?/p>

if [ $filename = "foo" ];

�?em>$filename变量包含�I�格时就会挂掉。可以这栯��冻I��

if [ "$filename" = "foo" ];

使用$@变量�Ӟ��你也需要��用引��P��因�ؓ�I�格隔开的两个参��C��被解释成两个独立的部分�?/p>

david% foo() { for i in $@; do echo $i; done }; foo bar "baz quux"

bar

baz

quux

david% foo() { for i in "$@"; do echo $i; done }; foo bar "baz quux"

bar

baz quux

我没有想��C�Q何不能��?em>"$@"的时候，所以当你有疑问的时候，使用引号��没有错误�?/p>

如果你同时��用find和xargs�Q�你应该使用 -print0 来让字符分割文�g名，而不是换行符分割�?/p>

david% touch "foo bar"

david% find | xargs ls

ls: ./foo: No such file or directory

ls: bar: No such file or directory

david% find -print0 | xargs -0 ls

./foo bar

讄��的陷�?/h1>
当你�~�写的脚本挂掉后�Q�文件系�l�处于未知状态。比如锁文�g状态、��时文件状态或者更��C��一个文件后在更��C��一个文件前挂掉。如果你能解册��些问题，无论�? 删除锁文�Ӟ��又或者在脚本遇到问题时回滚到已知状态，你都是非常棒的。幸�q�的是，bash提供了一�U�方法，当bash接收��C��个UNIX信号�Ӟ��q�行一�? 命��o或者一个函数。可以��?strong>trap命��o�?/p>
trap command signal [signal ...]

你可以链接多个信��P��列表可以使用kill -l获得�Q�，但是��Z��清理�D�局�Q�我们只使用其中的三个：INT�Q?em>TERM�?em>EXIT。你可以使用-as来让traps恢复到初始状态�?/p>

信号描述

INT	Interrupt - 当有��Z��用Ctrl-C�l�止脚本时被触发
TERM	Terminate - 当有��Z��用kill杀死脚本进�E�时被触�?/p>
EXIT	Exit - �q�是一个伪信号�Q�当脚本正常退出或者set -e后因为出错而退出时被触�?/p>

当你使用锁文件时�Q�可以这样写�Q?/p>

if [ ! -e $lockfile ]; then

touch $lockfile

critical-section

rm $lockfile

else

echo "critical-section is already running"

当最重要的部�?critical-section)正在�q�行�Ӟ��如果杀��M��脚本�q�程�Q�会发生什么呢�Q�锁文�g会被扔在那，而且你的脚本在它被删除以前再也不会运行了。解��x��法：

if [ ! -e $lockfile ]; then

trap " rm -f $lockfile; exit" INT TERM EXIT

touch $lockfile

critical-section

rm $lockfile

trap - INT TERM EXIT

else

echo "critical-section is already running"

现在当你杀死进�E�时�Q�锁文�g一同被删除。注意在trap命��o中明��地退��Z��脚本�Q�否则脚本会�l�箋执行trap后面的命令�?/p>

竟态条�?(wikipedia)

在上面锁文�g的例子中�Q�有一个竟态条件是不得不指出的�Q�它存在于判断锁文�g和创建锁文�g之间。一个可行的解决�Ҏ��是��用IO重定向和bash的noclobber(wikipedia)模式�Q�重定向��C��存在的文件。我们可以这么做�Q?/p>

if ( set -o noclobber; echo "$$" > "$lockfile") 2> /dev/null;

then

trap 'rm -f "$lockfile"; exit $?' INT TERM EXIT

critical-section

rm -f "$lockfile"

trap - INT TERM EXIT

else

echo "Failed to acquire lockfile: $lockfile"

echo "held by $(cat $lockfile)"

更复杂一点儿的问题是你要更新一大堆文�g�Q�当它们更新�q�程中出现问题时�Q�你是否能让脚本挂得更加优雅一些。你想确认那些正��更��C��Q�哪些根本没有变化。比如你需要一个添加用��L��脚本�?/p>

add_to_passwd $user

cp -a /etc/skel /home/$user

chown $user /home/$user -R

当磁盘空间不��x��者进�E�中途被杀死，�q�个脚本��׃��出现问题。在�q�种情况下，你也许希望用戯�̎户不存在�Q�而且他的文�g也应该被删除�?/p>

rollback() {

del_from_passwd $user

if [ -e /home/$user ]; then

rm -rf /home/$user

exit

}

trap rollback INT TERM EXIT

add_to_passwd $user

cp -a /etc/skel /home/$user

chown $user /home/$user -R

trap - INT TERM EXIT

在脚本最后需要��用trap关闭rollback调用�Q�否则当脚本正常退出的时候rollback��会被调用，那么脚本�{�于什么都没做�?/p>

保持原子�?/h1>
又是你需要一�ơ更新目录中的一大堆文�g�Q�比如你需要将URL重写到另一个网站的域名。你也许会写�Q?/p>
for file in $(find /var/www -type f -name ".html"); do

perl -pi -e 's/www.example.net/www.example.com/' $file

done

如果修改��C��半是脚本出现问题�Q�一部分使用www.example.com�Q�而另一部分使用www.example.net。你可以使用备䆾和trap解决�Q�但在升�U�过�E�中你的�|�站URL是不一致的�?/p>
解决�Ҏ��是将�q�个改变做成一个原子操作。先�Ҏ��据做一个副本，在副本中更新URL�Q�再用副本替换掉现在工作的版本。你需要确认副本和工作版本目录在同一个磁盘分��Z��Q�这样你��可以利用Linux�pȝ��的优势，它移动目录仅仅是更新目录指向的inode节点�?/p>
cp -a /var/www /var/www-tmp

for file in $(find /var/www-tmp -type -f -name ".html"); do

perl -pi -e 's/www.example.net/www.example.com/' $file

done

mv /var/www /var/www-old

mv /var/www-tmp /var/www

�q�意味着如果更新�q�程出问题，�U�上�pȝ��不会受媄响。线上系�l�受影响的时间降低�ؓ两次mv操作的时��_��q�个旉��非常短，因�ؓ文�g�pȝ��仅更新inode而不用真正的复制所有的数据�?/p>
�q�种技术的�~�点是你需要两倍的��盘�I�间�Q�而且那些长时间打开文�g的进�E�需要比较长的时间才能升�U�到新文件版本，��更新完成后重新启动这些进�E�。对 �? apache服务器来说这不是问题�Q�因为它每次都重新打开文�g。你可以使用lsof命��o查看当前正打开的文件。优势是你有了一个先前的备䆾�Q�当你需要还�? �Ӟ��它就�z�上用场了�?/p>

转自: http://www.linuxidc.com/Linux/2012-03/56614.htm

��果�?/a> 2013-01-09 12:11 发表评论

91精品国产综合久久婷婷,91精品国产综合久久婷婷,国产成人精品久久免费动漫

写出健壮�?Bash 脚本

使用set -u

使用set -e

�E�序防�M - 考虑意料之外的事

准备好处理文件名中的�I�格

信号描述

竟态条�?(wikipedia)