【Shell脚本编程（八）】大型脚本开发实战

灰太狼 2023-07-10 08:46 28阅读 0赞

## 1. 脚本工具功能概述 ##

**需求描述：**

1.  实现一个脚本工具，该脚本提供类似supervisor功能，可以对进程进行管理；
2.  一键查看所有进程运行状态；
3.  单个或批量启动进程，单个或批量停止进程；
4.  提供进程分组功能，可以按组查看进程运行状态，可以按组启动或停止该组内所有进程；

## 2. 拆分脚本功能，抽象函数 ##

1.  function get\_all\_process 返回进程名称列表字符串"nginx httpd mysql datanode"
2.  function get\_all\_group 返回进程组列表字符串
3.  function get\_process\_info（该函数可以接收一个参数，参数为进程名称） 返回进程详细信息列表字符串，详细信息包括：运行状态、PID、CPU、MEM、启动时间
4.  function get\_all\_process\_by\_group 返回进程组内的所有进程名称列表字符串

例子：DB组--> "mysql postgresql oracle"

## 3. 功能函数 ##

*  process.cfg

[GROUP_LIST]
    WEB
    DB
    HADOOP
    YARN
    
    [WEB]
    nginx
    httpd
    
    [DB]
    mysql
    postgresql
    oracle
    
    [HADOOP]
    datanode
    namenode
    journalnode
    
    [YARN]
    resourcemanager
    nodemanager
    
    [nginx]
    description="Web Server 1"
    program_name=tail
    parameter=-f /root/lesson/9.1/tmp/web-nginx.conf
    
    [httpd]
    description="Web Server 2"
    program_name=tail
    parameter=-f /root/lesson/9.1/tmp/web-httpd.conf
    
    [mysql]
    description="High Performance DataBase"
    program_name=tail

1.  function get\_all\_group 返回进程组列表字符串

vim app_status.sh
    
    #!/bin/bash
    #
    # Func: Get Porcess Status In process.cfg
    
    # Define Variables
    HOME_DIR="/root/lesson/9.1"
    CONFIG_FILE="process.cfg"
    
    function get_all_group
    {
        
        if [ -e $HOME_DIR/$CONFIG_FILE ];then
            echo "$CONFIG_FILE is not exist..Please Check.."
            exit 1
        else
            G_LIST=`sed -n '/\[GROUP_LIST\]/,/\[.*\]/p' process.cfg | egrep -v "(^$|\[.*\])" `
            echo "$G_LIST"
        fi
    }
    
    groups=`get_all_group`
    
    echo $groups

1.  function get\_all\_process 返回进程名称列表字符串
2.  function get\_process\_info（注：该函数可以接收一个参数，参数为进程名称） 返回进程详细信息列表字符串，详细信息包括：运行状态、PID、CPU、MEM、启动时间
3.  function get\_all\_process\_by\_group 返回进程组内的所有进程名称列表字符串（例子：DB组–> “mysql postgresql oracle”

## 4. 程序主流程设计 ##

./app\_status.sh 执行有三种情况：

1.  无参数 列出配置文件中所有进程的运行信息
2.  \-g GroupName 列出GroupName组内的所有进程
3.  process\_name1 列出指定进程的运行信息

#说明：该函数无需输入任何参数；返回配置文件 process.cfg中所有的组信息，例如WEB、DB等
    function get_all_group
    
    #说明：该函数无需输入任何参数；返回配置文件 process.cfg 中所有的进程信息。
    function get_all_process
    
    #说明：该函数接收一个参数，参数为进程名称；返回值是一个PID的列表，可能有一个PID，也可能有多个
    function get_process_pid_by_name
    
    #说明：该函数接收一个参数，参数为进程PID；返回值是一个进程运行信息的列表，列表包含运行状态、CPU占用率、内存占用率、进程启动时间
    function get_process_info_by_pid
    
    #说明：该函数接收一个参数，参数为组的名称；返回值是0或1，0代表该组在配置文件中，1代表该组不在配置文件中
    function is_group_in_config
    
    #说明：该函数接受一个参数，参数为组名称；返回值是对应组内的所有进程名称列表
    function get_all_process_by_group
    
    #说明：该函数接收一个参数，参数是一个进程名称；返回值是一个组名
    function get_group_by_process_name
    
    #说明：该函数接收两个参数，第一个参数为process_name，第二个参数为组名称；返回值是针对每一个进程PID的运行信息
    funcion format_print
    
    #说明：该函数接收一个参数，参数为进程名称；返回值是0或1，0代表该进程在配置文件中，1代表进程不在配置文件中
    function is_process_in_config

vim app_status.sh
    
    #!/bin/bash
    #
    # Func: Get Porcess Status In process.cfg
    
    # Define Variables
    HOME_DIR="/root/lesson/9.1"
    CONFIG_FILE="process.cfg"
    this_pid=$$
    
    function get_all_group
    {
        
        G_LIST=`sed -n '/\[GROUP_LIST\]/,/\[.*\]/p' $HOME_DIR/$CONFIG_FILE | egrep -v "(^$|\[.*\])" `
        echo "$G_LIST"
    }
    
    function get_all_process
    {
        
        for g in `get_all_group`
        do
            P_LIST=`sed -n "/\[$g\]/,/\[.*\]/p" $HOME_DIR/$CONFIG_FILE | egrep -v "(^$|\[.*\])"`
            echo "$P_LIST"
        done
    }
    
    function get_process_pid_by_name #process_name
    {
        
        if [ $# -ne 1 ];then
            return 1
        else
            pids=`ps -ef | grep $1 | grep -v grep | grep -v $0 | awk '{print $2}'`
            echo $pids
        fi
    }
    
    function get_process_info_by_pid #process_pid
    {
        
        num=1
        if [ `ps -ef | awk -v pid=$1 '$2==pid{print }' | wc -l` -eq 1 ];then
            pro_status="RUNNING"
        else
            pro_status="STOPED"
        fi
        pro_cpu=`ps aux | awk -v pid=$1 '$2==pid{print $3}'`
        pro_mem=`ps aux | awk -v pid=$1 '$2==pid{print $4}'`
        pro_start_time=`ps -p $1 -o lstart | grep -v STARTED`
    }
    
    function is_group_in_config
    {
        
        for gn in `get_all_group`;do
            if [ $gn == $1];then
                return
            fi
        done
        echo "Group $1 is not in process.cfg"
        return 1
    }
    
    function is_process_in_config
    {
        
        for pn in `get_all_process`;do
            if [ $pn == $1];then
                return
            fi
        done
        echo "Process $1 is not in process.cfg"
        return 1
    }
    
    function get_all_process_by_group
    {
        
        is_group_in_config $1
        if [ $? -eq 0 ];then
            p_list=`sed -n "/\[$1\]/,/\[.*\]/p" $HOME_DIR/CONFIG_FILE | egrep -v "(^$|^#|\[.*\])"`
            echo $p_list
        else
            echo "GroupName $1 is not in process.cfg"
        fi
    }
    
    function get_group_by_process_name
    {
        
        for gn in `get_all_group`;do
            for pn in `get_all_process_by_group $gn`;do
                if [ $pn == $1 ];then
                    echo "$gn"
                fi
            done
        done
    }
    
    function format_print
    {
        
        ps -ef | grep $1 | grep -v grep | grep -v $this_pid &> /dev/null
        if [ $? -eq 0 ];then
            pids=`get_process_pid_by_name $1`
            for pid in $pids;do
                get_process_info_by_pid $pid
                awk -v p_name=$1 \
                    -v g_name=$2 \
                    -v p_status=$pro_status \
                    -v p_pid=$pid \
                    -v p_cpu=$pro_cpu \
                    -v p_mem=$pro_mem \
                    -v p_start_time="$pro_start_time" \
                    'BEGIN{printf "%-10s%-12s%-5s%-5s%-5s%-5s%-15s\n",p_name,g_name,p_status,p_pid,p_cpu,p_mem,p_start_time}'
            done
        else
            awk -v p_name=$1 -v g_name=$2 'BEGIN{printf "%-10s%-12s%-5s%-5s%-5s%-5s%-15s\n",p_name,g_name,"STOPED","NULL","NULL","NULL","NULL"}'
        fi
    }
    
    awk 'BEGIN{printf "%-20s%-10s%-10s%-6s%-7s%-10s%-20s\n","ProcessName---------","GroupName---","Status----","PID---","CPU----","MEMORY----","StartTime---"}'
    
    if [ $# -gt 0 ];then
        if [ "$1" == "-g" ];then
            shift
            for gn in $@;do
                is_group_in_config $gn || continue
                for pn in `get_all_process_by_group $gn`;do
                    is_process_in_config $pn && format_print $pn $gn
                done
            done
        else
            for pn in $@;do
                gn=`get_group_by_process_name $pn`
                is_process_in_config $pn && format_print $pn $gn
            done
        fi
    else
        for pn in `get_all_process`;do
            gn=`get_group_by_process_name $pn`
            is_process_in_config $pn && format_print $pn $gn
        done
    fi