382 lines
9.9 KiB
Bash
382 lines
9.9 KiB
Bash
#!/bin/bash
|
|
|
|
RED="31m" ## 姨妈红
|
|
GREEN="32m" ## 水鸭青
|
|
YELLOW="33m" ## 鸭屎黄
|
|
PURPLE="35m" ## 基佬紫
|
|
BLUE="36m" ## 天依蓝
|
|
BlinkGreen="32;5m" ##闪烁的绿色
|
|
BlinkRed="31;5m" ##闪烁的红色
|
|
BackRed="41m" ## 背景红色
|
|
SplitLine="----------------------" #会被sys函数中的方法重写
|
|
|
|
hostArchVersion=""
|
|
hostArch=""
|
|
#### CollectSystemInfo ####
|
|
serverName=""
|
|
serverIpPbV4=""
|
|
serverIpInV4=""
|
|
serverIpPbV6=""
|
|
serverIpInV6=""
|
|
location=""
|
|
provider=""
|
|
managePort=""
|
|
cpuBrand=""
|
|
cpuCore=""
|
|
memoryTotal=""
|
|
diskTotal=""
|
|
diskUsage=""
|
|
osInfo=""
|
|
osKernelInfo=""
|
|
tcpControl=""
|
|
virtualization=""
|
|
ioSpeed=""
|
|
machineId=""
|
|
|
|
### tmp usage
|
|
ioavg=""
|
|
public_ipv4=""
|
|
country=""
|
|
region=""
|
|
city=""
|
|
org=""
|
|
#### CollectSystemInfo ####
|
|
|
|
######## 颜色函数方法很精妙 ############
|
|
colorEcho() {
|
|
# shellcheck disable=SC2145
|
|
echo -e "\033[${1}${@:2}\033[0m" 1>&2
|
|
}
|
|
# 判断命令是否存在
|
|
command_exists() {
|
|
local cmd="$1"
|
|
if eval type type >/dev/null 2>&1; then
|
|
eval type "$cmd" >/dev/null 2>&1
|
|
elif command >/dev/null 2>&1; then
|
|
command -v "$cmd" >/dev/null 2>&1
|
|
else
|
|
which "$cmd" >/dev/null 2>&1
|
|
fi
|
|
local rt=$?
|
|
return ${rt}
|
|
}
|
|
|
|
FunctionStart() {
|
|
colorEcho ${PURPLE} ${SplitLine}
|
|
colorEcho ${PURPLE} ${SplitLine}
|
|
echo ""
|
|
}
|
|
|
|
FunctionSuccess() {
|
|
colorEcho ${GREEN} ${SplitLine}
|
|
echo ""
|
|
}
|
|
|
|
FunctionEnd() {
|
|
echo ""
|
|
colorEcho ${BlinkGreen} ${SplitLine}
|
|
echo ""
|
|
echo ""
|
|
}
|
|
####### 获取系统版本及64位或32位信息
|
|
check_sys() {
|
|
# 获取当前终端的宽度，动态调整分割线的长度
|
|
shellwidth=$(stty size | awk '{print $2}')
|
|
if [[ $shellwidth -gt 1 ]]; then
|
|
SplitLine=$(yes "-" | sed ${shellwidth}'q' | tr -d '\n')
|
|
fi
|
|
|
|
sys_bit=$(uname -m)
|
|
case $sys_bit in
|
|
i[36]86)
|
|
os_bit="32"
|
|
hostArch="386"
|
|
;;
|
|
x86_64)
|
|
os_bit="64"
|
|
hostArch="amd64"
|
|
;;
|
|
*armv6*)
|
|
os_bit="arm"
|
|
hostArch="arm6"
|
|
;;
|
|
*armv7*)
|
|
os_bit="arm"
|
|
hostArch="arm7"
|
|
;;
|
|
*aarch64* | *armv8*)
|
|
os_bit="arm64"
|
|
hostArch="arm64"
|
|
;;
|
|
*)
|
|
colorEcho ${RED} "
|
|
哈哈……这个 辣鸡脚本 不支持你的系统。 (-_-) \n
|
|
备注: 仅支持 Ubuntu 16+ / Debian 8+ / CentOS 7+ 系统
|
|
" && exit 1
|
|
;;
|
|
esac
|
|
|
|
}
|
|
|
|
GoIOTest() {
|
|
(LANG=C dd if=/dev/zero of=benchtest_$$ bs=512k count=$1 conv=fdatasync && rm -f benchtest_$$) 2>&1 | awk -F, '{io=$NF} END { print io}' | sed 's/^[ \t]*//;s/[ \t]*$//'
|
|
}
|
|
|
|
calc_size() {
|
|
local raw=$1
|
|
local total_size=0
|
|
local num=1
|
|
local unit="KB"
|
|
if ! [[ ${raw} =~ ^[0-9]+$ ]]; then
|
|
echo ""
|
|
return
|
|
fi
|
|
if [ "${raw}" -ge 1073741824 ]; then
|
|
num=1073741824
|
|
unit="TB"
|
|
elif [ "${raw}" -ge 1048576 ]; then
|
|
num=1048576
|
|
unit="GB"
|
|
elif [ "${raw}" -ge 1024 ]; then
|
|
num=1024
|
|
unit="MB"
|
|
elif [ "${raw}" -eq 0 ]; then
|
|
echo "${total_size}"
|
|
return
|
|
fi
|
|
total_size=$(awk 'BEGIN{printf "%.1f", '$raw' / '$num'}')
|
|
echo "${total_size} ${unit}"
|
|
}
|
|
|
|
GethostArchINfo() {
|
|
[ -f /etc/redhat-release ] && awk '{print $0}' /etc/redhat-release && return
|
|
[ -f /etc/os-release ] && awk -F'[= "]' '/PRETTY_NAME/{print $3,$4,$5}' /etc/os-release && return
|
|
[ -f /etc/lsb-release ] && awk -F'[="]+' '/DESCRIPTION/{print $2}' /etc/lsb-release && return
|
|
}
|
|
|
|
StartIOTest() {
|
|
|
|
FunctionStart
|
|
colorEcho ${BLUE} "start IO speed test !"
|
|
|
|
freespace=$(df -m . | awk 'NR==2 {print $4}')
|
|
if [ -z "${freespace}" ]; then
|
|
freespace=$(df -m . | awk 'NR==3 {print $3}')
|
|
fi
|
|
if [ ${freespace} -gt 1024 ]; then
|
|
writemb=2048
|
|
io1=$(GoIOTest ${writemb})
|
|
colorEcho $YELLOW "I/O Speed(1st run) : $io1)"
|
|
io2=$(GoIOTest ${writemb})
|
|
colorEcho $YELLOW "I/O Speed(2st run) : $io2)"
|
|
io3=$(GoIOTest ${writemb})
|
|
colorEcho $YELLOW "I/O Speed(3st run) : $io3)"
|
|
ioraw1=$(echo $io1 | awk 'NR==1 {print $1}')
|
|
[ "$(echo $io1 | awk 'NR==1 {print $2}')" == "GB/s" ] && ioraw1=$(awk 'BEGIN{print '$ioraw1' * 1024}')
|
|
ioraw2=$(echo $io2 | awk 'NR==1 {print $1}')
|
|
[ "$(echo $io2 | awk 'NR==1 {print $2}')" == "GB/s" ] && ioraw2=$(awk 'BEGIN{print '$ioraw2' * 1024}')
|
|
ioraw3=$(echo $io3 | awk 'NR==1 {print $1}')
|
|
[ "$(echo $io3 | awk 'NR==1 {print $2}')" == "GB/s" ] && ioraw3=$(awk 'BEGIN{print '$ioraw3' * 1024}')
|
|
ioall=$(awk 'BEGIN{print '$ioraw1' + '$ioraw2' + '$ioraw3'}')
|
|
ioavg=$(awk 'BEGIN{printf "%.1f", '$ioall' / 3}')
|
|
colorEcho $YELLOW "I/O Speed(average) : $ioavg MB/s)"
|
|
else
|
|
echo " $(_red "Not enough space for I/O Speed test!")"
|
|
fi
|
|
|
|
FunctionSuccess
|
|
FunctionEnd
|
|
}
|
|
|
|
Check_Virtualization() {
|
|
|
|
FunctionStart
|
|
colorEcho ${BLUE} "start to check host virtualization !"
|
|
|
|
command_exists "dmesg" && virtualx="$(dmesg 2>/dev/null)"
|
|
if command_exists "dmidecode"; then
|
|
sys_manu="$(dmidecode -s system-manufacturer 2>/dev/null)"
|
|
sys_product="$(dmidecode -s system-product-name 2>/dev/null)"
|
|
sys_ver="$(dmidecode -s system-version 2>/dev/null)"
|
|
else
|
|
sys_manu=""
|
|
sys_product=""
|
|
sys_ver=""
|
|
fi
|
|
if grep -qa docker /proc/1/cgroup; then
|
|
virt="Docker"
|
|
elif grep -qa lxc /proc/1/cgroup; then
|
|
virt="LXC"
|
|
elif grep -qa container=lxc /proc/1/environ; then
|
|
virt="LXC"
|
|
elif [[ -f /proc/user_beancounters ]]; then
|
|
virt="OpenVZ"
|
|
elif [[ "${virtualx}" == *kvm-clock* ]]; then
|
|
virt="KVM"
|
|
elif [[ "${sys_product}" == *KVM* ]]; then
|
|
virt="KVM"
|
|
elif [[ "${cpuName}" == *KVM* ]]; then
|
|
virt="KVM"
|
|
elif [[ "${cpuName}" == *QEMU* ]]; then
|
|
virt="KVM"
|
|
elif [[ "${virtualx}" == *"VMware Virtual Platform"* ]]; then
|
|
virt="VMware"
|
|
elif [[ "${sys_product}" == *"VMware Virtual Platform"* ]]; then
|
|
virt="VMware"
|
|
elif [[ "${virtualx}" == *"Parallels Software International"* ]]; then
|
|
virt="Parallels"
|
|
elif [[ "${virtualx}" == *VirtualBox* ]]; then
|
|
virt="VirtualBox"
|
|
elif [[ -e /proc/xen ]]; then
|
|
if grep -q "control_d" "/proc/xen/capabilities" 2>/dev/null; then
|
|
virt="Xen-Dom0"
|
|
else
|
|
virt="Xen-DomU"
|
|
fi
|
|
elif [ -f "/sys/hypervisor/type" ] && grep -q "xen" "/sys/hypervisor/type"; then
|
|
virt="Xen"
|
|
elif [[ "${sys_manu}" == *"Microsoft Corporation"* ]]; then
|
|
if [[ "${sys_product}" == *"Virtual Machine"* ]]; then
|
|
if [[ "${sys_ver}" == *"7.0"* || "${sys_ver}" == *"Hyper-V" ]]; then
|
|
virt="Hyper-V"
|
|
else
|
|
virt="Microsoft Virtual Machine"
|
|
fi
|
|
fi
|
|
else
|
|
virt="Dedicated"
|
|
fi
|
|
|
|
FunctionSuccess
|
|
FunctionEnd
|
|
}
|
|
|
|
GetIpv4Info() {
|
|
|
|
FunctionStart
|
|
colorEcho ${BLUE} "start to get system public ip info !"
|
|
|
|
org="$(wget -q -T10 -O- ipinfo.io/org)"
|
|
city="$(wget -q -T10 -O- ipinfo.io/city)"
|
|
country="$(wget -q -T10 -O- ipinfo.io/country)"
|
|
region="$(wget -q -T10 -O- ipinfo.io/region)"
|
|
public_ipv4="$(wget -q -T10 -O- ipinfo.io/ip)"
|
|
|
|
FunctionSuccess
|
|
FunctionEnd
|
|
|
|
}
|
|
|
|
GenerateSystemInfo() {
|
|
FunctionStart
|
|
colorEcho $BLUE "start to collect system info !"
|
|
|
|
check_sys
|
|
|
|
cpuName=$(awk -F: '/model name/ {name=$2} END {print name}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//')
|
|
cores=$(awk -F: '/processor/ {core++} END {print core}' /proc/cpuinfo)
|
|
freq=$(awk -F'[ :]' '/cpu MHz/ {print $4;exit}' /proc/cpuinfo)
|
|
ccache=$(awk -F: '/cache size/ {cache=$2} END {print cache}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//')
|
|
cpu_aes=$(grep -i 'aes' /proc/cpuinfo)
|
|
cpu_virt=$(grep -Ei 'vmx|svm' /proc/cpuinfo)
|
|
tram=$(
|
|
LANG=C
|
|
free | awk '/Mem/ {print $2}'
|
|
)
|
|
tram=$(calc_size $tram)
|
|
uram=$(
|
|
LANG=C
|
|
free | awk '/Mem/ {print $3}'
|
|
)
|
|
uram=$(calc_size $uram)
|
|
swap=$(
|
|
LANG=C
|
|
free | awk '/Swap/ {print $2}'
|
|
)
|
|
swap=$(calc_size $swap)
|
|
uswap=$(
|
|
LANG=C
|
|
free | awk '/Swap/ {print $3}'
|
|
)
|
|
uswap=$(calc_size $uswap)
|
|
up=$(awk '{a=$1/86400;b=($1%86400)/3600;c=($1%3600)/60} {printf("%d days, %d hour %d min\n",a,b,c)}' /proc/uptime)
|
|
if command_exists "w"; then
|
|
load=$(
|
|
LANG=C
|
|
w | head -1 | awk -F'load average:' '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//'
|
|
)
|
|
elif command_exists "uptime"; then
|
|
load=$(
|
|
LANG=C
|
|
uptime | head -1 | awk -F'load average:' '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//'
|
|
)
|
|
fi
|
|
opsy=$(GethostArchINfo)
|
|
arch=$(uname -m)
|
|
if command_exists "getconf"; then
|
|
lbit=$(getconf LONG_BIT)
|
|
else
|
|
echo ${arch} | grep -q "64" && lbit="64" || lbit="32"
|
|
fi
|
|
kern=$(uname -r)
|
|
disk_total_size=$(
|
|
LANG=C
|
|
df -t simfs -t ext2 -t ext3 -t ext4 -t btrfs -t xfs -t vfat -t ntfs -t swap --total 2>/dev/null | grep total | awk '{ print $2 }'
|
|
)
|
|
disk_total_size=$(calc_size $disk_total_size)
|
|
disk_used_size=$(
|
|
LANG=C
|
|
df -t simfs -t ext2 -t ext3 -t ext4 -t btrfs -t xfs -t vfat -t ntfs -t swap --total 2>/dev/null | grep total | awk '{ print $3 }'
|
|
)
|
|
disk_used_size=$(calc_size $disk_used_size)
|
|
tcpctrl=$(sysctl net.ipv4.tcp_congestion_control | awk -F ' ' '{print $3}')
|
|
|
|
FunctionSuccess
|
|
|
|
# todo
|
|
# StartIOTest
|
|
|
|
GetIpv4Info
|
|
|
|
Check_Virtualization
|
|
|
|
local machineNumber=""
|
|
|
|
if [[ $(cat /etc/hostname | cut -d"-" -f 3 | grep -c '^[0-9][0-9]') -gt 0 ]]; then
|
|
machineNumber=$(cat /etc/hostname | cut -d"-" -f 3)
|
|
else
|
|
machineNumber=99
|
|
fi
|
|
|
|
export serverName="${city}-${hostArch}-${machineNumber}"
|
|
export serverIpPbV4="$public_ipv4"
|
|
export serverIpInV4=""
|
|
export serverIpPbV6=""
|
|
export serverIpInV6=""
|
|
export location="$city $region $country"
|
|
export provider="$org"
|
|
export managePort="$(netstat -ntulp | grep sshd | grep -w tcp | awk '{print$4}' | cut -d":" -f2)"
|
|
export cpuCore="$cores @ $freq MHz"
|
|
export cpuBrand="$cpuName"
|
|
export memoryTotal="$tram"
|
|
export diskTotal="$disk_total_size"
|
|
export diskUsage="$disk_used_size"
|
|
export archInfo="$arch ($lbit Bit)"
|
|
export osInfo="$opsy"
|
|
export osKernelInfo="$kern"
|
|
export tcpControl="$tcpctrl"
|
|
export virtualization="$virt"
|
|
export ioSpeed="$ioavg MB/s"
|
|
export machineId="$(cat /host/etc/machine-id)"
|
|
|
|
FunctionEnd
|
|
}
|
|
|
|
main() {
|
|
|
|
GenerateSystemInfo
|
|
|
|
}
|
|
|
|
main && env && java ${JAVA_OPTS} -jar /wdd/agent.jar
|