shell脚本整合json数值输出从而获取硬件信息

需要安装必要的软件，这里我为了保障脚本合理的进行，使用的是真机环境，没有使用虚拟机

[root@sys-idc-pxe01 ~]# yum install jq lsscsi MegaCli
Loaded plugins: fastestmirror, security
Loading mirror speeds from cached hostfile
Setting up Install Process
Package jq-1.3-2.el6.x86_64 already installed and latest version
Package lsscsi-0.23-3.el6.x86_64 already installed and latest version
Package MegaCli-8.00.40-1.i386 already installed and latest version
Nothing to do

我的shell风格比较偏向模块化，所以使用函数会多一些，这样方便理解，相信大家也能看得懂。

我不太喜欢python风格的代码格式（虽然我python写的也很不错），看起来给人的感觉太紧凑，太乱。据说是前辈们出版书籍的时候为了省纸，才出现的那种风格，后辈怀着对前辈的敬仰就延续了下来，其实前辈也没有那么写过那种风格的代码，不知道是不是真的。

[root@sys-idc-pxe01 ~]# cat ./get_server_info.sh 
#!/bin/sh
#description: get server hardware info
#author: liuxin
#date: 20180122
#需要安装jq工具 yum install jq
#用于存放该服务器的所有信息，个人喜欢把全局变量写到外面
#写到函数里面，没有加local的变量也是全局变量
INFO="{}"
#定义一个工具函数，用于生成json数值，后面将会频繁用到
function create_json()
{
#utility function
local key=$1
local value="$2"
local json=""
#if value is string
if [ -z "$(echo "$value" |egrep "/[|/]|/{|/}")" ]
then
json=$(jq -n {"$key":"/"$value/""})
#if value is json, object
elif [ "$(echo "$value" |jq -r type)" == "object" ]
then
json=$(jq -n {"$key":"$value"})
#if value is array
elif [ "$(echo "$value" |jq -r type)" == "array" ]
then
json=$(jq -n "{$key:$value}")
else
echo "value type error..."
exit 1
return 0
fi
echo $json
return 0
}
#获取CPU信息
function get_cpu()
{
#获取cpu信息，去掉空格和制表符和空行，以便于for循环
local cpu_model_1=$(dmidecode -s processor-version |grep '@' |tr -d " " |tr -s "/n" |tr -d "/t")
local cpu_info="{}"
local i=0
#因为去掉了空格和制表符，以下默认使用换行符分隔
for line in $(echo "$cpu_model_1")
do
local cpu_model="$line"
local cpu1=$(create_json "cpu_model" "$cpu_model")
#获取每块cpu的信息，这里只记录了型号
local cpu=$(create_json "cpu_$i" "$cpu1")
local cpu_info=$(jq -n "$cpu_info + $cpu")
i=$[ $i + 1]
done
#将cpu的信息整合成一个json，key是cpu
local info=$(create_json "cpu" "$cpu_info")
#将信息加入到全局变量中
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_mem()
{
#generate json {Locator:{sn:sn,size:size}}
local mem_info="{}"
#获取每个内存的信息，包括Size:|Locator:|Serial Number:
local mem_info_1=$(dmidecode -t memory |egrep 'Size:|Locator:|Serial Number:' |grep -v 'Bank Locator:' |awk '
{
if (NR%3==1 && $NF=="MB")
{
size=$2;
getline (NR+1);
locator=$2;
getline (NR+2);
sn=$NF;
printf("%s,%s,%s/n",locator,size,sn)
}
}')
#根据上面的信息，将信息过滤并整合成json
local i=0
for line in $(echo "$mem_info_1")
do
local locator=$(echo $line |awk -F , '{print $1}')
local      sn=$(echo $line |awk  -F , '{print $3}')
local    size=$(echo $line |awk -F , '{print $2}')
local mem1=$(create_json "locator" "$locator")
local mem2=$(create_json "sn" "$sn")
local mem3=$(create_json "size" "$size")
local mem4=$(jq -n "$mem1 + $mem2 + $mem3")
#每条内存的信息，key是内存从0开始的序号
local mem=$(create_json "mem_$i" "$mem4")
#将这些内存的信息组合到一个json中
mem_info=$(jq -n "$mem_info + $mem")
i=$[ $i + 1 ]
done
#给这些内存的信息设置key，mem
local info=$(create_json "mem" "$mem_info")
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_megacli_disk()
{
#设置megacli工具的路径，此条可以根据情况更改
local raid_tool="/opt/MegaRAID/MegaCli/MegaCli64"
#将硬盘信息获取，保存下来，省去每次都执行的操作
$raid_tool pdlist aall >/tmp/megacli_pdlist.txt
local disk_info="{}"
#获取硬盘的必要信息
local disk_info_1=$(cat /tmp/megacli_pdlist.txt |egrep 'Enclosure Device ID:|Slot Number:|PD Type:|Raw Size:|Inquiry Data:|Media Type:'|awk '
{
if(NR%6==1 && $1$2=="EnclosureDevice")
{
E=$NF;
getline (NR+1);
S=$NF;
getline (NR+2);
pdtype=$NF;
getline (NR+3);
size=$3$4;
getline (NR+4);
sn=$3$4$5$6;
getline (NR+5);
mediatype=$3$4$5$6;
printf("%s,%s,%s,%s,%s,%s/n",E,S,pdtype,size,sn,mediatype)
}
}')
#将获取到的硬盘信息进行整合，生成json
local i=0
for line in $(echo $disk_info_1)
do
#local       key=$(echo $line |awk -F , '{printf("ES%s_%s/n",$1,$2)}')
local         E=$(echo $line |awk -F , '{print $1}')
local         S=$(echo $line |awk -F , '{print $2}')
local    pdtype=$(echo $line |awk -F , '{print $3}')
local      size=$(echo $line |awk -F , '{print $4}')
local        sn=$(echo $line |awk -F , '{print $5}')
local mediatype=$(echo $line |awk -F , '{print $6}')
local disk1=$(create_json "pdtype" "$pdtype")
local disk1_1=$(create_json "enclosure_id" "$E")
local disk1_2=$(create_json "slot_id" "$S")
local disk2=$(create_json "size" "$size")
local disk3=$(create_json "sn" "$sn")
local disk4=$(create_json "mediatype" "$mediatype")
local disk5=$(jq -n "$disk1 + $disk1_1 + $disk1_2 + $disk2 + $disk3 + $disk4")
local disk=$(create_json "disk_$i" "$disk5")
disk_info=$(jq -n "$disk_info + $disk")
i=$[ $i + 1 ]
done
#echo $disk_info
local info=$(create_json "disk" "$disk_info")
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_hba_disk()
{
#对于hba卡的硬盘，使用smartctl获取硬盘信息
local disk_tool="smartctl"
local disk_info="{}"
#lsscsi 需要使用yum install lsscsi 安装
local disk_info_1=$(lsscsi -g |grep -v 'enclosu' |awk '{printf("%s,%s,%s,%s/n",$1,$2,$(NF-1),$NF)}')
local i=0
for line in $(echo $disk_info_1)
do
local         E=$(echo $line |awk -F , '{print $1}' |awk -F ':' '{print $1}' |tr -d '/[|/]')
local         S=$(echo $line |awk -F , '{print $NF}' |egrep -o [0-9]*)
local        sd=$(echo $line |awk -F , '{print $(NF-1)}')
$disk_tool -i $sd >/tmp/disk_info.txt
local    pdtype="SATA"
if [ "$(cat /tmp/disk_info.txt |grep 'Transport protocol:' |awk '{print $NF}')" == "SAS" ]
then
local    pdtype="SAS"
fi
local      size=$(cat /tmp/disk_info.txt |grep 'User Capacity:' |awk '{printf("%s%s/n",$(NF-1),$NF)}' |tr -d '/[|/]')
local        sn=$(cat /tmp/disk_info.txt |grep 'Serial Number:' |awk '{print $NF}')
local mediatype="disk"
local disk1=$(create_json "pdtype" "$pdtype")
local disk1_1=$(create_json "enclosure_id" "$E")
local disk1_2=$(create_json "slot_id" "$S")
local disk2=$(create_json "size" "$size")
local disk3=$(create_json "sn" "$sn")
local disk4=$(create_json "mediatype" "$mediatype")
local disk5=$(jq -n "$disk1 + $disk1_1 + $disk1_2 + $disk2 + $disk3 + $disk4")
local disk=$(create_json "disk_$i" "$disk5")
disk_info=$(jq -n "$disk_info + $disk")
i=$[ $i + 1 ]
done
#echo $disk_info
local info=$(create_json "disk" "$disk_info")
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_disk()
{
#根据获取到的硬盘控制器类型，来判断使用什么工具采集硬盘信息
if [ "$(echo "$INFO" |jq -r .disk_ctrl.disk_ctrl_0.type)" == "raid" ]
then
get_megacli_disk
elif [ "$(echo "$INFO" |jq -r .disk_ctrl.disk_ctrl_0.type)" == "hba" ]
then
get_hba_disk
else
local info=$(create_json "disk" "error")
INFO=$(jq -n "$INFO + $info")
fi
#hp机器比较特殊，这里我没有做hp机器硬盘信息采集，有兴趣的朋友可以自行添加上
#if hp machine
return 0
}
function get_diskController()
{
local disk_ctrl="{}"
#if LSI Controller
local disk_ctrl_1="$(lspci -nn |grep LSI)"
local i=0
#以换行符分隔
IFS_OLD=$IFS && IFS=$'/n'
for line in $(echo "$disk_ctrl_1")
do
#echo $line
local   ctrl_id=$(echo "$line" |awk -F ']:' '{print $1}' |awk '{print $NF}' |tr -d '/[|/]')
case "$ctrl_id" in
#根据控制器的id或进行判断是raid卡还是hba卡，因为品牌比较多，后续可以在此处进行扩展添加
0104)
# 获取Logic以后的字符串，并进行拼接
local ctrl_name=$(echo "${line##*"Logic"}" |awk '{printf("%s_%s_%s/n",$1,$2,$3)}')
local     ctrl1=$(create_json "id" "$ctrl_id")
local     ctrl2=$(create_json "type" "raid")
local     ctrl3=$(create_json "name" "$ctrl_name")
;;
0100|0107)
local ctrl_name=$(echo "${line##*"Logic"}" |awk '{printf("%s_%s_%s/n",$1,$3,$4)}')
local     ctrl1=$(create_json "id" "$ctrl_id")
local     ctrl2=$(create_json "type" "hba")
local     ctrl3=$(create_json "name" "$ctrl_name")
;;
*)
local     ctrl1=$(create_json "id" "----")
local     ctrl2=$(create_json "type" "----")
local     ctrl3=$(create_json "name" "----")
;;
esac
local ctrl_tmp=$(jq -n "$ctrl1 + $ctrl2 + $ctrl3")
local ctrl=$(create_json "disk_ctrl_$i" "$ctrl_tmp")
disk_ctrl=$(jq -n "$disk_ctrl + $ctrl")
i=$[ $i + 1 ]
done
IFS=$IFS_OLD
local info=$(create_json "disk_ctrl" "$disk_ctrl")
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_netcard()
{
local netcard_info="{}"
local netcard_info_1="$(lspci -nn |grep Ether)"
local i=0
#echo "$netcard_info_1"
IFS_OLD=$IFS && IFS=$'/n'
for line in $(echo "$netcard_info_1")
do
local     net_id=$(echo $line |egrep -o '[0-9a-z]{4}:[0-9a-z]{4}')
local   net_id_1=$(echo $net_id |awk -F : '{print $1}')
case "$net_id_1" in
8086)
local net_name=$(echo "${line##*": "}" |awk '{printf("%s_%s_%s_%s/n",$1,$3,$4,$5)}')
local     type=$(echo $line |egrep -o SFP || echo "TP")
local     net1=$(create_json "id" "$net_id")
local     net2=$(create_json "name" "$net_name")
local     net3=$(create_json "type" "$type")
;;
14e4)
local net_name=$(echo "${line##*": "}" |awk '{printf("%s_%s_%s_%s/n",$1,$3,$4,$5)}')
local     type=$(echo $line |egrep -o SFP || echo "TP")
local     net1=$(create_json "id" "$net_id")
local     net2=$(create_json "name" "$net_name")
local     net3=$(create_json "type" "$type")
;;
*)
local net_name=$(echo "${line##*": "}" |awk '{printf("%s_%s_%s_%s/n",$1,$3,$4,$5)}')
local     type=$(echo $line |egrep -o SFP || echo "TP")
local     net1=$(create_json "id" "$net_id")
local     net2=$(create_json "name" "$net_name")
local     net3=$(create_json "type" "$type")
;;
esac
local net1=$(jq -n "$net1 + $net2 + $net3")
#echo $net
local net2=$(create_json "net_$i" "$net1")
netcard_info=$(jq -n "$netcard_info + $net2")
i=$[ $i + 1 ]
done
IFS=$IFS_OLD
local info=$(create_json "net" "$netcard_info")
INFO=$(jq -n "$INFO + $info")
return 0
}
function get_server()
{
local product=$(dmidecode -s system-product-name |grep -v '^#' |tr -d ' ' |head -n1)
local manufacturer=$(dmidecode -s system-manufacturer |grep -v '^#' |tr -d ' ' |head -n1)
local server1=$(create_json "manufacturer" "$manufacturer")
local server2=$(create_json "product" "$product")
local server3=$(jq -n "$server1 + $server2")
local info=$(create_json "basic_info" "$server3")
INFO=$(jq -n "$INFO + $info")
return 0
}
ALL_INFO=""
function get_all()
{
#因为硬盘信息的获取依赖硬盘控制器的信息，所以get_diskController要放到get_disk前面
get_server
get_cpu
get_mem
get_diskController
get_disk
get_netcard
local sn=$(dmidecode -s system-serial-number |grep -v '^#' |tr -d ' ' |head -n1)
ALL_INFO=$(create_json "$sn" "$INFO")
return 0
}
function main()
{
get_all
echo $ALL_INFO
return 0
}
#-------------------------------------------------
main