|
| 1 | +# set 命令 |
| 2 | + |
| 3 | +`set`命令是 Bash 脚本的重要环节,却常常被忽视,导致脚本的安全性和可维护性出问题。本文介绍它的基本用法,让你可以更安心地使用 Bash 脚本。 |
| 4 | + |
| 5 | +## 简介 |
| 6 | + |
| 7 | +我们知道,Bash 执行脚本的时候,会创建一个新的 Shell。 |
| 8 | + |
| 9 | +```bash |
| 10 | +$ bash script.sh |
| 11 | +``` |
| 12 | + |
| 13 | +上面代码中,`script.sh`是在一个新的 Shell 里面执行。这个 Shell 就是脚本的执行环境,Bash 默认给定了这个环境的各种参数。 |
| 14 | + |
| 15 | +`set`命令用来修改 Shell 环境的运行参数,也就是可以定制环境。一共有十几个参数可以定制,[官方手册](https://www.gnu.org/software/bash/manual/html_node/The-Set-Builtin.html)有完整清单,本文介绍其中最常用的四个。 |
| 16 | + |
| 17 | +顺便提一下,如果命令行下不带任何参数,直接运行`set`,会显示所有的环境变量和 Shell 函数。 |
| 18 | + |
| 19 | +```bash |
| 20 | +$ set |
| 21 | +``` |
| 22 | + |
| 23 | +## set -u |
| 24 | + |
| 25 | +执行脚本的时候,如果遇到不存在的变量,Bash 默认忽略它。 |
| 26 | + |
| 27 | +```bash |
| 28 | +#!/usr/bin/env bash |
| 29 | + |
| 30 | +echo $a |
| 31 | +echo bar |
| 32 | +``` |
| 33 | + |
| 34 | +上面代码中,`$a`是一个不存在的变量。执行结果如下。 |
| 35 | + |
| 36 | +```bash |
| 37 | +$ bash script.sh |
| 38 | + |
| 39 | +bar |
| 40 | +``` |
| 41 | + |
| 42 | +可以看到,`echo $a`输出了一个空行,Bash 忽略了不存在的`$a`,然后继续执行`echo bar`。大多数情况下,这不是开发者想要的行为,遇到变量不存在,脚本应该报错,而不是一声不响地往下执行。 |
| 43 | + |
| 44 | +`set -u`就用来改变这种行为。脚本在头部加上它,遇到不存在的变量就会报错,并停止执行。 |
| 45 | + |
| 46 | +```bash |
| 47 | +#!/usr/bin/env bash |
| 48 | +set -u |
| 49 | + |
| 50 | +echo $a |
| 51 | +echo bar |
| 52 | +``` |
| 53 | + |
| 54 | +运行结果如下。 |
| 55 | + |
| 56 | +```bash |
| 57 | +$ bash script.sh |
| 58 | +bash: script.sh:行4: a: 未绑定的变量 |
| 59 | +``` |
| 60 | + |
| 61 | +可以看到,脚本报错了,并且不再执行后面的语句。 |
| 62 | + |
| 63 | +`-u`还有另一种写法`-o nounset`,两者是等价的。 |
| 64 | + |
| 65 | +```bash |
| 66 | +set -o nounset |
| 67 | +``` |
| 68 | + |
| 69 | +## set -x |
| 70 | + |
| 71 | +默认情况下,脚本执行后,屏幕只显示运行结果,没有其他内容。如果多个命令连续执行,它们的运行结果就会连续输出。有时会分不清,某一段内容是什么命令产生的。 |
| 72 | + |
| 73 | +`set -x`用来在运行结果之前,先输出执行的那一行命令。 |
| 74 | + |
| 75 | +```bash |
| 76 | +#!/usr/bin/env bash |
| 77 | +set -x |
| 78 | + |
| 79 | +echo bar |
| 80 | +``` |
| 81 | + |
| 82 | +执行上面的脚本,结果如下。 |
| 83 | + |
| 84 | +```bash |
| 85 | +$ bash script.sh |
| 86 | ++ echo bar |
| 87 | +bar |
| 88 | +``` |
| 89 | + |
| 90 | +可以看到,执行`echo bar`之前,该命令会先打印出来,行首以`+`表示。这对于调试复杂的脚本是很有用的。 |
| 91 | + |
| 92 | +`-x`还有另一种写法`-o xtrace`。 |
| 93 | + |
| 94 | +```bash |
| 95 | +set -o xtrace |
| 96 | +``` |
| 97 | + |
| 98 | +## Bash 的错误处理 |
| 99 | + |
| 100 | +如果脚本里面有运行失败的命令(返回值非0),Bash 默认会继续执行后面的命令。 |
| 101 | + |
| 102 | +```bash |
| 103 | +#!/usr/bin/env bash |
| 104 | + |
| 105 | +foo |
| 106 | +echo bar |
| 107 | +``` |
| 108 | + |
| 109 | +上面脚本中,`foo`是一个不存在的命令,执行时会报错。但是,Bash 会忽略这个错误,继续往下执行。 |
| 110 | + |
| 111 | +```bash |
| 112 | +$ bash script.sh |
| 113 | +script.sh:行3: foo: 未找到命令 |
| 114 | +bar |
| 115 | +``` |
| 116 | + |
| 117 | +可以看到,Bash 只是显示有错误,并没有终止执行。 |
| 118 | + |
| 119 | +这种行为很不利于脚本安全和除错。实际开发中,如果某个命令失败,往往需要脚本停止执行,防止错误累积。这时,一般采用下面的写法。 |
| 120 | + |
| 121 | +```bash |
| 122 | +command || exit 1 |
| 123 | +``` |
| 124 | + |
| 125 | +上面的写法表示只要`command`有非零返回值,脚本就会停止执行。 |
| 126 | + |
| 127 | +如果停止执行之前需要完成多个操作,就要采用下面三种写法。 |
| 128 | + |
| 129 | +```bash |
| 130 | +# 写法一 |
| 131 | +command || { echo "command failed"; exit 1; } |
| 132 | + |
| 133 | +# 写法二 |
| 134 | +if ! command; then echo "command failed"; exit 1; fi |
| 135 | + |
| 136 | +# 写法三 |
| 137 | +command |
| 138 | +if [ "$?" -ne 0 ]; then echo "command failed"; exit 1; fi |
| 139 | +``` |
| 140 | + |
| 141 | +另外,除了停止执行,还有一种情况。如果两个命令有继承关系,只有第一个命令成功了,才能继续执行第二个命令,那么就要采用下面的写法。 |
| 142 | + |
| 143 | +```bash |
| 144 | +command1 && command2 |
| 145 | +``` |
| 146 | + |
| 147 | +## set -e |
| 148 | + |
| 149 | +上面这些写法多少有些麻烦,容易疏忽。`set -e`从根本上解决了这个问题,它使得脚本只要发生错误,就终止执行。 |
| 150 | + |
| 151 | +```bash |
| 152 | +#!/usr/bin/env bash |
| 153 | +set -e |
| 154 | + |
| 155 | +foo |
| 156 | +echo bar |
| 157 | +``` |
| 158 | + |
| 159 | +执行结果如下。 |
| 160 | + |
| 161 | +```bash |
| 162 | +$ bash script.sh |
| 163 | +script.sh:行4: foo: 未找到命令 |
| 164 | +``` |
| 165 | + |
| 166 | +可以看到,第4行执行失败以后,脚本就终止执行了。 |
| 167 | + |
| 168 | +`set -e`根据返回值来判断,一个命令是否运行失败。但是,某些命令的非零返回值可能不表示失败,或者开发者希望在命令失败的情况下,脚本继续执行下去。这时可以暂时关闭`set -e`,该命令执行结束后,再重新打开`set -e`。 |
| 169 | + |
| 170 | +```bash |
| 171 | +set +e |
| 172 | +command1 |
| 173 | +command2 |
| 174 | +set -e |
| 175 | +``` |
| 176 | + |
| 177 | +上面代码中,`set +e`表示关闭`-e`选项,`set -e`表示重新打开`-e`选项。 |
| 178 | + |
| 179 | +还有一种方法是使用`command || true`,使得该命令即使执行失败,脚本也不会终止执行。 |
| 180 | + |
| 181 | +```bash |
| 182 | +#!/bin/bash |
| 183 | +set -e |
| 184 | + |
| 185 | +foo || true |
| 186 | +echo bar |
| 187 | +``` |
| 188 | + |
| 189 | +上面代码中,`true`使得这一行语句总是会执行成功,后面的`echo bar`会执行。 |
| 190 | + |
| 191 | +`-e`还有另一种写法`-o errexit`。 |
| 192 | + |
| 193 | +```bash |
| 194 | +set -o errexit |
| 195 | +``` |
| 196 | + |
| 197 | +## set -o pipefail |
| 198 | + |
| 199 | +`set -e`有一个例外情况,就是不适用于管道命令。 |
| 200 | + |
| 201 | +所谓管道命令,就是多个子命令通过管道运算符(`|`)组合成为一个大的命令。Bash 会把最后一个子命令的返回值,作为整个命令的返回值。也就是说,只要最后一个子命令不失败,管道命令总是会执行成功,因此它后面命令依然会执行,`set -e`就失效了。 |
| 202 | + |
| 203 | +请看下面这个例子。 |
| 204 | + |
| 205 | +```bash |
| 206 | +#!/usr/bin/env bash |
| 207 | +set -e |
| 208 | + |
| 209 | +foo | echo a |
| 210 | +echo bar |
| 211 | +``` |
| 212 | + |
| 213 | +执行结果如下。 |
| 214 | + |
| 215 | +```bash |
| 216 | +$ bash script.sh |
| 217 | +a |
| 218 | +script.sh:行4: foo: 未找到命令 |
| 219 | +bar |
| 220 | +``` |
| 221 | + |
| 222 | +上面代码中,`foo`是一个不存在的命令,但是`foo | echo a`这个管道命令会执行成功,导致后面的`echo bar`会继续执行。 |
| 223 | + |
| 224 | +`set -o pipefail`用来解决这种情况,只要一个子命令失败,整个管道命令就失败,脚本就会终止执行。 |
| 225 | + |
| 226 | +```bash |
| 227 | +#!/usr/bin/env bash |
| 228 | +set -eo pipefail |
| 229 | + |
| 230 | +foo | echo a |
| 231 | +echo bar |
| 232 | +``` |
| 233 | + |
| 234 | +运行后,结果如下。 |
| 235 | + |
| 236 | +```bash |
| 237 | +$ bash script.sh |
| 238 | +a |
| 239 | +script.sh:行4: foo: 未找到命令 |
| 240 | +``` |
| 241 | + |
| 242 | +可以看到,`echo bar`没有执行。 |
| 243 | + |
| 244 | +## 总结 |
| 245 | + |
| 246 | +`set`命令的上面这四个参数,一般都放在一起使用。 |
| 247 | + |
| 248 | +```bash |
| 249 | +# 写法一 |
| 250 | +set -euxo pipefail |
| 251 | + |
| 252 | +# 写法二 |
| 253 | +set -eux |
| 254 | +set -o pipefail |
| 255 | +``` |
| 256 | + |
| 257 | +这两种写法建议放在所有 Bash 脚本的头部。 |
| 258 | + |
| 259 | +另一种办法是在执行 Bash 脚本的时候,从命令行传入这些参数。 |
| 260 | + |
| 261 | +```bash |
| 262 | +$ bash -euxo pipefail script.sh |
| 263 | +``` |
| 264 | + |
| 265 | +## 参考链接 |
| 266 | + |
| 267 | +- [The Set Builtin](https://www.gnu.org/software/bash/manual/html_node/The-Set-Builtin.html) |
| 268 | +- [Safer bash scripts with 'set -euxo pipefail'](https://vaneyckt.io/posts/safer_bash_scripts_with_set_euxo_pipefail/) |
| 269 | +- [Writing Robust Bash Shell Scripts](http://www.davidpashley.com/articles/writing-robust-shell-scripts/) |
| 270 | + |
0 commit comments