首页 > 代码库 > Shell脚本的模块化和脚本复用
Shell脚本的模块化和脚本复用
日常工作中编写shell脚本是再平常不过的了，那编写过这么多的shell，当我们遇见曾经写过的相关的处理函数时，如何来复用？直接拷贝相应的代码还是“引用过来”，或者是在编写一个功能相对复杂的shell时，如何做到模块化,如何组织复杂的函数调用？
如何导入复用和模块化，就是这里要一起学习的。
我们会按照以下几个部分来学习：
1、shell执行方式不同的效果不同
2、导入模块
1、shell执行方式不同的效果不同
第一种：sh test.sh
这种方式会在我们当前进程中，再新建一个子进行去执行这个test.sh 脚本。我们能获取到的东西，也就是这个test.sh执行完的结果或者输出。
root@docker-host-03:~#more test.sh #!/bin/bash s="thisis a test file" root@docker-host-03:~#sh test.sh root@docker-host-03:~#echo $s root@docker-host-03:~#sh test.sh root@docker-host-03:~#echo $? 0 |
第二种：sourcetest.sh 或者. Test.sh
这种方式相当于是将test.sh的内容拷贝到当前进程中来执行。这样我们能获取到的内容就不仅仅是这个test.sh执行完成的结果。我们还能获取到在这个脚本中定义的全局变量，定义的功能函数。因为我们是将脚本的内容拷贝到当前文件中。
那这种方式就和我们很多编程语言中的模块导入类似。我们来验证下是不是。
root@docker-host-03:~#. test.sh root@docker-host-03:~#echo $s this is atest file root@docker-host-03:~#s=0 root@docker-host-03:~#echo $s 0 root@docker-host-03:~#source test.sh root@docker-host-03:~#echo $s this is a test file #我们可以看到这里，可以获取到变量s的内容。 |
2、导入模块
2.1)先导入一个最简单的
基于上面第一点我们提到的两种shell的执行方式中的第二种。我们利用这种“拷贝”的特性就能够实现将以前写过的功能，进行导入
比如我们写了很简单格式化输出log的函数：
root@docker-host-03:~# cat test1.sh #!/bin/bash
MODULENAME=$(basename $0) LOGFILE=/tmp/logfile-`date +%Y%m%d`
log_info() { #[2017-03-31 12:00:00 ] - TextName - The log message Localdatetime=`date "+%Y-%m-%d %H:%M:%S"` if[ "$1" ];then echo "[ ${datetime} ] - ${MODULENAME} - $1 " | tee -a${LOGFILE} else return 1 fi } |
上面这个log_info 函数会格式化刷出我们的日志内容。我们可以在多处复用。
我们这里的Main.sh脚本和test1.sh在同一个目录里面。我们main.sh先导入test1.sh的内容，然后使用test1.sh中已有的变量，功能函数。我们来看下main.sh的内容。
root@docker-host-03:~# cat main.sh #!/bin/bash
#我们在这里通过. Test1.sh 或是source test1.sh的方式将test1.sh导入进来。 . test1.sh main() { #我们在这里就可以直接使用log_info函数了。 log_info ‘This is the main shell‘ } main root@docker-host-03:~# ./main.sh [ 2017-03-31 15:11:23 ] - main.sh - This isthe main shell root@docker-host-03:~# more/tmp/logfile-20170331 [ 2017-03-31 15:11:39 ] - main.sh - This isthe main shell
|
2.2）如果我们需要从其他地方导入其他函数
上面我们的main和test1是在同一个目录中，所以我们在main 中使用的是相对路径。如果我们需要导入的“模块”test1.sh 是放在其他的目录？或者说我们的模块存在层级关系？我们有应该如何来引入呢？
1）因我们在引入的时候，系统默认会在环境变量$PATH中去寻找我们要导入的文件。一种做法是我们将我们需要用到的公共模块放到$PATH包含的路径中。或者我们添加一个公用的路径到$PATH环境变量中，然后将我们需要用到的公有模块放到公用文件夹下。
如：
root@docker-host-03:~/test_shell# cat main.sh #!/bin/bash
. test1.sh . Config.sh main() { log_info ‘This is the main shell‘ readConf echo${conf} } main root@docker-host-03:~/test_shell# echo$PATH /usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games #在这里我们可以看到我们将两个模块放在了PATH的路径下面。 root@docker-host-03:~/test_shell# ls/usr/local/bin/test1.sh /usr/local/bin/test1.sh root@docker-host-03:~/test_shell# ls -l/usr/local/bin/Config.sh -rw-r--r-- 1 root root 59 Mar 31 15:41/usr/local/bin/Config.sh
|
这里我直接将文件拷贝到了$PATH有包含的目录中。
2） 我们将模块所在的绝对路径export到PATH中：
如:
root@docker-host-03:~/test_shell/module#pwd /root/test_shell/module root@docker-host-03:~/test_shell/module#tree . ├── Config.sh └── test1.sh
0 directories, 2 files root@docker-host-03:~/test_shell/module# cd../ root@docker-host-03:~/test_shell# moremain.sh #!/bin/bash
#这里我们给了一个绝对路径 MODUPATH=/root/test_shell/module export PATH=$PATH:${MODUPATH}
. test1.sh . Config.sh main() { log_info ‘This is the main shell‘ readConf echo${conf} } Main
|
3） 动态的加载模块
当然如果我们在编写一个功能相对复杂的shell的时候，我们将会将这个一个整个复杂的带有结构关系的shell打包，拷贝到任何一个地方去执行。这样我们就没有办法事先知道我们的绝对路径是怎样的。比如这样的目录结构：
root@docker-host-03:~/test_shell#pwd /root/test_shell root@docker-host-03:~/test_shell#tree . ├── main.sh ├── module │ ├── Config.sh │ └── test1.sh └── module2 └── mail.sh
2directories, 4 files root@docker-host-03:~/test_shell#more main.sh #!/bin/bash
#这里动态的获取到我们的绝对路径，并把模块的路径拼凑出来。 MODUPATH=$(dirname$(readlink -f $0))/module echo${MODUPATH} exportPATH=$PATH:${MODUPATH}
.test1.sh .Config.sh main() { log_info ‘This is the main shell‘ readConf echo ${conf} } Main
|
4）如何避免重复多次的导入某个模块
我们要说source，其实就是将内容拷贝到同一个地方执行，那我们可以在模块中加入一个判断条件。如果发现我们模块中的某个变量已经存在，我们就return出去，不再导入后续的内容。
比如这样：
root@docker-host-03:~/test_shell#more module/test1.sh #!/bin/bash
#这里我注释掉的内容就是。而且这个变量名称，尽量按照某种规则确保是唯一的变量名称。 #if [${m_log} ];then # return 0 #fi #m_log="m_log" #这一行输出是为了测试是否多次导入了。 echo"TEST IS" MODULENAME=$(basename$0) LOGFILE=/tmp/logfile-`date+%Y%m%d`
log_info() { #[ 2017-03-31 12:00:00 ] - TextName - The logmessage datetime=`date "+%Y-%m-%d%H:%M:%S"` if [ "$1" ];then echo "[ ${datetime} ] - ${MODULENAME}- $1 " | tee -a ${LOGFILE} else return 1 fi }
|
这里我们在注释掉的情况下，看看多次导入是什么反应。
root@docker-host-03:~/test_shell# more main.sh #!/bin/bash
MODUPATH=$(dirname $(readlink -f$0))/module echo ${MODUPATH} export PATH=$PATH:${MODUPATH}
. test1.sh . Config.sh main() { log_info ‘This is the main shell‘ readConf echo${conf} } main root@docker-host-03:~/test_shell# moremodule/Config.sh #!/bin/bash
. test1.sh readConf() { conf="Read the config file" } root@docker-host-03:~/test_shell# ./main.sh /root/test_shell/module TEST IS TEST IS [ 2017-03-31 16:51:09 ] - main.sh - This isthe main shell Read the config file #我们可以看到这里多次输出了TEST IS内容。 |
#将我们上面注释掉的内容去除，现在虽多次导入，但实质上只会导入一次。这样也会避免多次循环引用。
root@docker-host-03:~/test_shell# ./main.sh /root/test_shell/module TEST IS [ 2017-03-31 16:56:31 ] - main.sh - This isthe main shell Read the config file #当我们去掉了注释部分，这里就不会出现多次重复载入了。 |
-----------------------------小结-----------------
其实这里的模块复用，其实就是通过source 或则. 的方式将原本的模块内容“拷贝过来”，只是在使用的过程当中，需要注意模块路径的问题，还有层级，和多次调用的问题。
至于功能复杂的shell，我想基本也不会复杂到无法接受，基本按照一定的组织发方式就能变得清晰。并且提高复用。
大家可以自己积累一些比较常用的函数，需要用到的时候，导入，不用重复编写。比如日志，邮件告警等等。
----------------------------------------------------
本文出自 “从头开始” 博客，请务必保留此出处http://atong.blog.51cto.com/2393905/1912179
Shell脚本的模块化和脚本复用