专栏名称: 程序员大咖
为程序员提供最优质的博文、最精彩的讨论、最实用的开发资源;提供最新最全的编程学习资料:PHP、Objective-C、Java、Swift、C/C++函数库、.NET Framework类库、J2SE API等等。并不定期奉送各种福利。
目录
相关文章推荐
程序猿  ·  离谱!下载 DeepSeek 将判 20 ... ·  昨天  
程序员小灰  ·  清华大学:DeepSeek从入门到精通(2025) ·  昨天  
OSC开源社区  ·  谷歌安卓系统“假开源、真垄断”? ·  4 天前  
程序猿  ·  DeepSeek招人年薪最高154万 ·  5 天前  
51好读  ›  专栏  ›  程序员大咖

编写Linux Shell脚本的最佳实践

程序员大咖  · 公众号  · 程序员  · 2017-08-11 10:24

正文

点击上方“ 程序员大咖 ”,选择“置顶公众号”

关键时刻,第一时间送达!


前言


由于工作需要,最近重新开始拾掇shell脚本。虽然绝大部分命令自己平时也经常使用,但是在写成脚本的时候总觉得写的很难看。而且当我在看其他人写的脚本的时候,总觉得难以阅读。毕竟shell脚本这个东西不算是正经的编程语言,他更像是一个工具,用来杂糅不同的程序供我们调用。因此很多人在写的时候也是想到哪里写到哪里,基本上都像是一段超长的main函数,不忍直视。同时,由于历史原因,shell有很多不同的版本,而且也有很多有相同功能的命令需要我们进行取舍,以至于代码的规范很难统一。


考虑到上面的这些原因,我查阅了一些相关的文档,发现这些问题其实很多人都考虑过,而且也形成了一些不错的文章,但是还是有点零散。因此我就在这里把这些文章稍微整理了一下, 作为以后我自己写脚本的技术规范。



代码风格规范


开头有“蛇棒”


所谓shebang其实就是在很多脚本的第一行出现的以”#!”开头的注释,他指明了当我们没有指定解释器的时候默认的解释器,一般可能是下面这样:

#!/bin/bas

当然,解释器有很多种,除了bash之外,我们可以用下面的命令查看本机支持的解释器:

#!/bin/$ cat /etc/shells      #/etc/shells: valid login shells      /bin/sh      /bin/dash      /bin/bash      /bin/rbash      /usr/bin/screen

当我们直接使用./a.sh来执行这个脚本的时候,如果没有shebang,那么它就会默认用$SHELL指定的解释器,否则就会用shebang指定的解释器。


不过,上面这种写法可能不太具备适应性,一般我们会用下面的方式来指定:

#!/usr/bin/env bash

这种方式是我们推荐的使用方式。


代码有注释


注释,显然是一个常识,不过这里还是要再强调一下,这个在shell脚本里尤为重要。因为很多单行的shell命令不是那么浅显易懂,没有注释的话在维护起来会让人尤其的头大。
注释的意义不仅在于解释用途,而在于告诉我们注意事项,就像是一个README。
具体的来说,对于shell脚本,注释一般包括下面几个部分:


  • shebang

  • 脚本的参数

  • 脚本的用途

  • 脚本的注意事项

  • 脚本的写作时间,作者,版权等

  • 各个函数前的说明注释

  • 一些较复杂的单行命令注释


参数要规范


这一点很重要,当我们的脚本需要接受参数的时候,我们一定要先判断参数是否合乎规范,并给出合适的回显,方便使用者了解参数的使用。


最少,最少,我们至少得判断下参数的个数吧:

if [[ $# != 2 ]];then       echo "Parameter incorrect."       exit 1   
fi

变量和魔数


一般情况下我们会将一些重要的环境变量定义在开头,确保这些变量的存在。

source /etc/profileexport PATH=”/usr/local/bin:/usr/bin:/bin:/usr/local/sbin:/usr/sbin:/sbin:/apps/bin/”

这种定义方式有一个很常见的用途,最典型的应用就是,当我们本地安装了很多java版本时,我们可能需要指定一个java来用。那么这时我们就会在脚本开头重新定义JAVA_HOME以及PATH变量来进行控制。


同时,一段好的代码通常是不会有很多硬编码在代码里的“魔数”的。如果一定要有,通常是用一个变量的形式定义在开头,然后调用的时候直接调用这个变量,这样方便日后的修改。


缩进有规矩


对于shell脚本,缩进是个大问题。因为很多需要缩进的地方(比如if,for语句)都不长,所有很多人都懒得去缩进,而且很多人不习惯用函数,导致缩进功能被弱化。
其实正确的缩进是很重要的,尤其是在写函数的时候,否则我们在阅读的时候很容易把函数体跟直接执行的命令搞混。


常见的缩进方法主要有”soft tab”和”hard tab”两种。


  • 所谓soft tab就是使用n个空格进行缩进(n通常是2或4)

  • 所谓hard tab当然就是指真实的””字符
    这里不去撕哪种方式最好,只能说各有各的优劣。反正我习惯用hard tab。
    对于if和for语句之类的,我们最好不要把then,do这些关键字单独写一行,这样看上去比较丑。。。


命名有标准


所谓命名规范,基本包含下面这几点:


  • 文件名规范,以.sh结尾,方便识别

  • 变量名字要有含义,不要拼错

  • 统一命名风格,写shell一般用小写字母加下划线


编码要统一


在写脚本的时候尽量使用UTF-8编码,能够支持中文等一些奇奇怪怪的字符。不过虽然能写中文,但是在写注释以及打log的时候还是尽量英文,毕竟很多机器还是没有直接支持中文的,打出来可能会有乱码。


这里还尤其需要注意一点,就是当我们是在windows下用utf-8编码来写shell脚本的时候,一定要注意这个utf-8是否是有BOM的。默认情况下windows判断utf-8格式是通过在文件开头加上三个EF BB BF字节来判断的,但是在Linux中默认是无BOM的。因此如果我们是在windows下写脚本的时候,一定要注意将编码改成Utf-8无BOM,一般用notepad++之类的编辑器都能改。否则,在Linux下运行的时候就会识别到开头的三个字符,从而报一些无法识别命令的错。


权限记得加


这一点虽然很小,但是我个人却经常忘记,不加执行权限会导致无法直接执行,有点讨厌。。。


日志和回显


日志的重要性不必多说,能够方便我们回头纠错,在大型的项目里是非常重要的。
如果这个脚本是供用户直接在命令行使用的,那么我们最好还要能够在执行时实时回显执行过程,方便用户掌控。


有时候为了提高用户体验,我们会在回显中添加一些特效,比如颜色啊,闪烁啊之类的,具体可以参考
ANSI/VT100 Control sequences 这篇文章的介绍。


密码要移除


不要把密码硬编码在脚本里,不要把密码硬编码在脚本里,不要把密码硬编码在脚本里。


重要的事情说三遍,尤其是当脚本托管在类似Github这类平台中时。。。


太长要分行


在调用某些程序的时候,参数可能会很长,这时候为了保证较好的阅读体验,我们可以用反斜杠来分行:

./configure    –prefix=/usr    –sbin-path=/usr/sbin/nginx    –conf-path=/etc/nginx/nginx.conf 

注意在反斜杠前有个空格。


编码细节规范


代码有效率


在使用命令的时候要了解命令的具体做法,尤其当数据处理量大的时候,要时刻考虑该命令是否会影响效率。


比如下面的两个sed命令:

sed -n '1p' filesed -n '1p;1q' file

他们的作用一样,都是获取文件的第一行。但是第一条命令会读取整个文件,而第二条命令只读取第一行。当文件很大的时候,仅仅是这样一条命令不一样就会造成巨大的效率差异。


当然,这里只是为了举一个例子,这个例子真正正确的用法应该是使用head -n1 file命令。。。


勤用双引号


几乎所有的大佬都推荐在使用”$”来获取变量的时候最好加上双引号。


不加上双引号在很多情况下都会造成很大的麻烦,为什么呢?举一个例子:

#!/bin/sh   #已知当前文件夹有一个a.sh的文件   var="*.sh"   
echo $var  
echo "$var"

他的运行结果如下:

a.sh*.sh

为啥会这样呢?其实可以解释为他执行了下面的命令:

echo *.shecho "*.sh"

在很多情况下,在将变量作为参数的时候,一定要注意上面这一点,仔细体会其中的差异。上面只是一个非常小的例子,实际应用的时候由于这个细节导致的问题实在是太多了。。。


巧用main函数


我们知道,像java,C这样的编译型语言都会有一个函数入口,这种结构使得代码可读性很强,我们知道哪些直接执行,那些是函数。但是脚本不一样,脚本属于解释性语言,从第一行直接执行到最后一行,如果在这当中命令与函数糅杂在一起,那就非常难读了。


用python的朋友都知道,一个合乎标准的python脚本大体上至少是这样的:

#!/usr/bin/env python   def func1():       pass   def func2():       pass   if __name__=='__main__':       func1()       func2()

他用一个很巧妙的方法实现了我们习惯的main函数,使得代码可读性更强。

在shell中,我们也有类似的小技巧:

#!/usr/bin/env bash   func1(){       #do sth   }   func2(){       #do sth   }   main(){       func1       func2   }   main "$@"


我们可以采用这种写法,同样实现类似的main函数,使得脚本的结构化程度更好。

考虑作用域


shell中默认的变量作用域都是全局的,比如下面的脚本:

#!/usr/bin/env bash   var=1
func(){       var=2
  }   func   echo $var

他的输出结果就是2而不是1,这样显然不符合我们的编码习惯,很容易造成一些问题。


因此,相比直接使用全局变量,我们最好使用local readonly这类的命令,其次我们可以使用declare来声明变量。这些方式都比使用全局方式定义要好。


函数返回值


在使用函数的时候一定要注意,shell中函数的返回值只能是整数,估计是因为一般情况下一个函数的返回值通常表示这个函数的运行状态,所以一般都是0或者是1就够了,因此就设计成了这样。不过,如果非得想传递字符串,也可以通过下面变通的方法:

func(){       echo "2333"   }   res=$(func)   echo "This is from $res."

这样,通过echo或者print之类的就可以做到传一些额外参数的目的。


间接引用值


什么叫间接引用?比如下面这个场景:

VAR1="2323232"
VAR2="VAR1"

我们有一个变量VAR1,又有一个变量VAR2,这个VAR2的值是VAR1的名字,那么我们现在想通过VAR2来获取VAR1的值,这时候应该怎么办呢?


比较土鳖的方法是这样:

echo ${!VAR1}

这个用法的确可行,但是看起来十分的不舒服,很难只管的去理解,我们并不推荐。而且事实上我们本身就不推荐使用eval这个命令。


比较舒服的写法是下面这样:

echo ${!VAR1}

通过在变量名前加一个!就可以做到简单的间接引用了。


不过需要注意的是,用上面的方法,我们只能够做到取值,而不能做到赋值。

巧用heredocs


所谓heredocs,也可以算是一种多行输入的方法,即在”<


使用heredocs,我们可以非常方便的生成一些模板文件:

cat>>/etc/rsyncd.conf << EOF   log file = /usr/local/logs/rsyncd.log   transfer logging = yes   log format = %t %a %m %f %b   syslog facility = local3   EOF

学会查路径


很多情况下,我们会先获取当前脚本的路径,然后一这个路径为基准,去找其他的路径。通常我们是直接用pwd以期获得脚本的路径。


不过其实这样是不严谨的,pwd获得的是当前shell的执行路径,而不是当前脚本的执行路径。


正确的做法应该是下面这两种:

script_dir=$(cd $(dirname $0) && pwd)script_dir=$(dirname $(readlink -f $0 ))

应当先cd进当前脚本的目录然后再pwd,或者直接读取当前脚本的所在路径。


代码要简短


这里的简短不单单是指代码长度,而是只用到的命令数。原则上我们应当做到,能一条命令解决的问题绝不用两条命令解决。这不仅牵涉到代码的可读性,而且也关乎代码的执行效率。


最最经典的例子如下:

cat /etc/passwd | grep root   grep root /etc/passw

cat命令最为人不齿的用法就是这样,用的没有任何意义,明明一条命令可以解决,他非得加根管道。。。


其实代码简短在还能某种程度上能保证效率的提升,比如下面的例子:

#method1   find . -name '*.txt' |xargs sed -i s/233/666/g   find . -name '*.txt' |xargs sed -i s/235






请到「今天看啥」查看全文