Shell map实现Docker简单部署脚本

随着docker的流行，docker的应用越来越多，工作中肯定少不了，但个人感觉docker的有些命令需要的参数有点多，比如运行一个容器，–name给容器起名，-d后台运行等等。根据个人习惯，一般很多参数都是固定的，没必要每次操作都敲一遍，工作效率特别低。特别是对于不熟悉docker的人员，比如运维人员，这时候给他们提供一个部署脚本是很有必要的，因为越是能简化他们的工作，他们找我们解决问题的机会就更少。
前段时间实现了一个很简单的部署脚本，可以简化docker安装、容器运行等工作，根据传给脚本的第一个参数作为命令而进行不同的处理。对于不同的命令在shell中使用case来进行匹配不同的处理，后来发现随着命令的增多，case匹配的代码越来越长，而且需要修改多处地方，修改很费劲，毕竟处理实现功能处理外，还需要提供一些帮助信息。
这两天一直想着怎么改进这脚本怎么写，让它可以很好的扩展，添加新命令更简单。经过一个晚上的思考及在网上搜索，最终主要使用map简化case操作，最后写成的脚本个人还是挺满意的，添加一个命令时不会修改原有的代码，只需要在同一个地方添加新代码，在map中添加帮助信息，另外再对应添加一个处理函数就好。具体实现如下，脚本中只是添加了几个简单命令：

#!/bin/bash#声明一个命令帮助容器，运行脚本时输入的第一个参数当作命令，根据命令做不同处理declare -A cmdHelpMap=()redColor="\033[31m"normalColor="\033[0m"yellowColor="\033[33m"scriptName=$0       #运行的脚本名字inputArgsNum=$#     #运行脚本时输入参数的个数#打印信息，对打印信息进行查找脚本名字，红色打印需要传递的参数function println(){    findScriptName=0  #是否找到脚本名字    index=0;    #遍历打印信息    for arg in $@    do        if [ ${arg} = ${scriptName} ]        then            findScriptName=1            echo -en "\n\t"        fi        if [ ${index} -gt ${inputArgsNum} ]        then            echo -en ${redColor}        fi        echo -en "${arg} "        if [ ${findScriptName} -eq 1 ]        then            let index++        fi    done    echo -e ${normalColor}}#参数处理，判断输入参数个数与帮助信息中需要的参数个数是否相等，不相等则打印帮助信息并退出function parameterHandling(){    findScriptName=0    argsNum=0;    key=$1    helpMsg=${cmdHelpMap[$key]}     #遍历帮助信息总计需要输入的参数个数，帮助信息中出现脚本名字后开始统计，    #自动以空格分割帮助信息进行遍历    for msg in ${helpMsg}    do        if [ ${findScriptName} -eq 1 ]        then            let argsNum++        fi        if [ ${msg} = ${scriptName} ]        then            findScriptName=1        fi    done    #参数个数不相等处理    if [ ${inputArgsNum} -ne ${argsNum} ]    then        println ${helpMsg}        exit    fi}#确认操作function confirm(){    echo -ne "${yellowColor}$* ?[y/n]${normalColor}"    read input    if [ -z ${input} ]    then        exit    fi    if [ ${input} != "y" ]    then        exit    fi}##############################################################################################添加命令只需要两步：#1、以命令为key，往cmdHelpMap中添加帮助信息#2、以命令为函数名定义命令处理函数#例如以下的help命令#help命令cmdHelpMap["help"]="查看帮助信息: ${scriptName} help"function help(){    inputArgsNum=0    println "Usage:"    preFunc=${FUNCNAME[1]}    number=1;    for key in ${!cmdHelpMap[@]}    do        println ${number}.${cmdHelpMap[$key]}        let number++    done}#install命令cmdHelpMap["install"]="安装docker: ${scriptName} install"function install(){    docker -v    if [ $? -eq 0 ]; then        confirm Docker已安装，确认是否继续安装    fi    curl -sSL https://get.daocloud.io/docker | sh}#run命令cmdHelpMap["run"]="运行容器: ${scriptName} run [imageName:imageTag] [hostPort:containerPort]"function run(){    image=$2    containerName=${image/:/-}    port=$3    docker run -d --name ${containerName} --restart=always -p ${port} ${image}    if [ $? -ne 0 ]; then        println ${redColor}运行${containerName}失败, 请检查镜像${image}是否存在        exit    fi}#rmi命令cmdHelpMap["rmi"]="删除镜像: ${scriptName} rmi [imageName:imageTag]"function rmi(){    image=$2    confirm 确定删除镜像${image}    docker rmi ${image}    if [ $? -ne 0 ]; then        println ${redColor}删除镜像失败, 请检查镜像${image}是否存在    fi}#rm命令cmdHelpMap["rm"]="删除容器: ${scriptName} rm [imageName-imageTag]"function rm(){    containerName=$2    docker rm -f ${containerName}    if [ $? -ne 0 ]; then        println ${redColor}删除容器失败, 请检查容器${containerName}是否存在    fi}##############################################################################################如果第一个参数为空或者不存在map中，则默认第一个参数为helpcmd=$1if [ -z $cmd ] || [ -z "${cmdHelpMap[$cmd]}" ]then    if [ $# -ne 0 ]    then        println ${redColor}无效参数:$*    fi    cmd="help"    inputArgsNum=1fi#参数处理parameterHandling $cmd#执行相应处理函数$cmd $*

脚本直接保存为deploy.sh，以下是一些简单操作提示信息的效果图: