首页 > 代码库 > 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)如果我们需要从其他地方导入其他函数

上面我们的maintest1是在同一个目录中,所以我们在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)  我们将模块所在的绝对路径exportPATH中:

:

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脚本的模块化和脚本复用