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

1. 脚本工具功能概述

需求描述：

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

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

function get_all_process 返回进程名称列表字符串"nginx httpd mysql datanode"
function get_all_group 返回进程组列表字符串
function get_process_info（该函数可以接收一个参数，参数为进程名称）返回进程详细信息列表字符串，详细信息包括：运行状态、PID、CPU、MEM、启动时间
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

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

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

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
    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
}

if [ !-e $HOME_DIR/$CONFIG_FILE ];then
    echo "$CONFIG_FILE is not exist..please check.."
    exit 1
fi

4. 程序主流程设计

./app_status.sh 执行有三种情况：

无参数列出配置文件中所有进程的运行信息
-g GroupName 列出GroupName组内的所有进程
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 [email protected];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 [email protected];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

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

1. 脚本工具功能概述

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

3. 功能函数

4. 程序主流程设计

继续阅读

Apache (You don't have permission to access / on this server.）

debian9升级4.9.0内核到4.19.2内核过程

centOS7 配置 vsftpd 虚拟用户及权限Vsftpd配置虚拟用户及权限

linux-svn卸载与安装

vsftp虚拟多用户多权限一键部署脚本

Ubuntu14.04 LTS下安装mongodb

Nginx服务优化（1）——隐藏版本号、修改用户与组、网页缓存时间、日志切割、连接超时一、隐藏版本号二、修改用户与组三、配置Nginx网页缓存时间四、实现Nginx日志分割五、配置Nginx实现连接超时六、补充关于时间日期的命令

httpd服务的部署、启动、配置和简单优化一、部署二、启动三、配置文件

配置网页内容访问

手动安装Intel network I217-LM网卡的Linux驱动

禁止ubuntu系统弹出报错界面

Ubuntu Linux下Apache的配置文件

samba服务器的功能

【Linux】UDP广播报文接收速率问题

Linux设备模型（中）之上层容器

PowerPC平台 Linux移植三