隨著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 [email protected]
    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中，則預設第一個引數為help
cmd=$1
if [ -z $cmd ] || [ -z "${cmdHelpMap[$cmd]}" ]
then
    if [ $# -ne 0 ]
    then
        println ${redColor}無效引數:$*
    fi
    cmd="help"
    inputArgsNum=1
fi

#引數處理
parameterHandling $cmd

#執行相應處理函式
$cmd $*

指令碼直接儲存為deploy.sh，以下是一些簡單操作提示資訊的效果圖: