bench.sh

江左梅郎

1. #!/usr/bin/env bash
   
2. #
   
3. # Description: Auto test download & I/O speed script
   
4. #
   
5. # Copyright (C) 2015 - 2018 Teddysun <[email protected]>
   
6. #
   
7. # Thanks: LookBack <[email protected]>
   
8. #
   
9. # URL: https://teddysun.com/444.html
   
10. #
    
11. 
    
12. if  [ ! -e '/usr/bin/wget' ]; then
    
13.     echo "Error: wget command not found. You must be install wget command at first."
    
14.     exit 1
    
15. fi
    
16. 
    
17. # Colors
    
18. RED='\033[0;31m'
    
19. GREEN='\033[0;32m'
    
20. YELLOW='\033[0;33m'
    
21. BLUE='\033[0;36m'
    
22. PLAIN='\033[0m'
    
23. 
    
24. get_opsy() {
    
25.     [ -f /etc/redhat-release ] && awk '{print ($1,$3~/^[0-9]/?$3:$4)}' /etc/redhat-release && return
    
26.     [ -f /etc/os-release ] && awk -F'[= "]' '/PRETTY_NAME/{print $3,$4,$5}' /etc/os-release && return
    
27.     [ -f /etc/lsb-release ] && awk -F'[="]+' '/DESCRIPTION/{print $2}' /etc/lsb-release && return
    
28. }
    
29. 
    
30. next() {
    
31.     printf "%-70s\n" "-" | sed 's/\s/-/g'
    
32. }
    
33. 
    
34. speed_test_v4() {
    
35.     local output=$(LANG=C wget -4O /dev/null -T300 $1 2>&1)
    
36.     local speedtest=$(printf '%s' "$output" | awk '/\/dev\/null/ {speed=$3 $4} END {gsub(/\(|\)/,"",speed); print speed}')
    
37.     local ipaddress=$(printf '%s' "$output" | awk -F'|' '/Connecting to .*\|([^\|]+)\|/ {print $2}')
    
38.     local nodeName=$2
    
39.     printf "${YELLOW}%-32s${GREEN}%-24s${RED}%-14s${PLAIN}\n" "${nodeName}" "${ipaddress}" "${speedtest}"
    
40. }
    
41. 
    
42. speed_test_v6() {
    
43.     local output=$(LANG=C wget -6O /dev/null -T300 $1 2>&1)
    
44.     local speedtest=$(printf '%s' "$output" | awk '/\/dev\/null/ {speed=$3 $4} END {gsub(/\(|\)/,"",speed); print speed}')
    
45.     local ipaddress=$(printf '%s' "$output" | awk -F'|' '/Connecting to .*\|([^\|]+)\|/ {print $2}')
    
46.     local nodeName=$2
    
47.     printf "${YELLOW}%-32s${GREEN}%-24s${RED}%-14s${PLAIN}\n" "${nodeName}" "${ipaddress}" "${speedtest}"
    
48. }
    
49. 
    
50. speed_v4() {
    
51.     speed_test_v4 'http://cachefly.cachefly.net/100mb.test' 'CacheFly'
    
52.     speed_test_v4 'http://speedtest.tokyo.linode.com/100MB-tokyo.bin' 'Linode, Tokyo, JP'
    
53.     speed_test_v4 'http://speedtest.singapore.linode.com/100MB-singapore.bin' 'Linode, Singapore, SG'
    
54.     speed_test_v4 'http://speedtest.london.linode.com/100MB-london.bin' 'Linode, London, UK'
    
55.     speed_test_v4 'http://speedtest.frankfurt.linode.com/100MB-frankfurt.bin' 'Linode, Frankfurt, DE'
    
56.     speed_test_v4 'http://speedtest.fremont.linode.com/100MB-fremont.bin' 'Linode, Fremont, CA'
    
57.     speed_test_v4 'http://speedtest.dal05.softlayer.com/downloads/test100.zip' 'Softlayer, Dallas, TX'
    
58.     speed_test_v4 'http://speedtest.sea01.softlayer.com/downloads/test100.zip' 'Softlayer, Seattle, WA'
    
59.     speed_test_v4 'http://speedtest.fra02.softlayer.com/downloads/test100.zip' 'Softlayer, Frankfurt, DE'
    
60.     speed_test_v4 'http://speedtest.sng01.softlayer.com/downloads/test100.zip' 'Softlayer, Singapore, SG'
    
61.     speed_test_v4 'http://speedtest.hkg02.softlayer.com/downloads/test100.zip' 'Softlayer, HongKong, CN'
    
62. }
    
63. 
    
64. speed_v6() {
    
65.     speed_test_v6 'http://speedtest.atlanta.linode.com/100MB-atlanta.bin' 'Linode, Atlanta, GA'
    
66.     speed_test_v6 'http://speedtest.dallas.linode.com/100MB-dallas.bin' 'Linode, Dallas, TX'
    
67.     speed_test_v6 'http://speedtest.newark.linode.com/100MB-newark.bin' 'Linode, Newark, NJ'
    
68.     speed_test_v6 'http://speedtest.singapore.linode.com/100MB-singapore.bin' 'Linode, Singapore, SG'
    
69.     speed_test_v6 'http://speedtest.tokyo.linode.com/100MB-tokyo.bin' 'Linode, Tokyo, JP'
    
70.     speed_test_v6 'http://speedtest.sjc03.softlayer.com/downloads/test100.zip' 'Softlayer, San Jose, CA'
    
71.     speed_test_v6 'http://speedtest.wdc01.softlayer.com/downloads/test100.zip' 'Softlayer, Washington, WA'
    
72.     speed_test_v6 'http://speedtest.par01.softlayer.com/downloads/test100.zip' 'Softlayer, Paris, FR'
    
73.     speed_test_v6 'http://speedtest.sng01.softlayer.com/downloads/test100.zip' 'Softlayer, Singapore, SG'
    
74.     speed_test_v6 'http://speedtest.tok02.softlayer.com/downloads/test100.zip' 'Softlayer, Tokyo, JP'
    
75. }
    
76. 
    
77. io_test() {
    
78.     (LANG=C dd if=/dev/zero of=test_$ bs=64k count=16k conv=fdatasync && rm -f test_$ ) 2>&1 | awk -F, '{io=$NF} END { print io}' | sed 's/^[ \t]*//;s/[ \t]*$//'
    
79. }
    
80. 
    
81. calc_disk() {
    
82.     local total_size=0
    
83.     local array=$@
    
84.     for size in "${array[@]}"
    
85.     do
    
86.         [ "${size}" == "0" ] && size_t=0 || size_t=`echo ${size:0:${#size}-1}`
    
87.         [ "`echo ${size:(-1)}`" == "K" ] && size=0
    
88.         [ "`echo ${size:(-1)}`" == "M" ] && size=$( awk 'BEGIN{printf "%.1f", '$size_t' / 1024}' )
    
89.         [ "`echo ${size:(-1)}`" == "T" ] && size=$( awk 'BEGIN{printf "%.1f", '$size_t' * 1024}' )
    
90.         [ "`echo ${size:(-1)}`" == "G" ] && size=${size_t}
    
91.         total_size=$( awk 'BEGIN{printf "%.1f", '$total_size' + '$size'}' )
    
92.     done
    
93.     echo ${total_size}
    
94. }
    
95. 
    
96. cname=$( awk -F: '/model name/ {name=$2} END {print name}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' )
    
97. cores=$( awk -F: '/model name/ {core++} END {print core}' /proc/cpuinfo )
    
98. freq=$( awk -F: '/cpu MHz/ {freq=$2} END {print freq}' /proc/cpuinfo | sed 's/^[ \t]*//;s/[ \t]*$//' )
    
99. tram=$( free -m | awk '/Mem/ {print $2}' )
    
100. uram=$( free -m | awk '/Mem/ {print $3}' )
     
101. swap=$( free -m | awk '/Swap/ {print $2}' )
     
102. uswap=$( free -m | awk '/Swap/ {print $3}' )
     
103. 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 )
     
104. load=$( w | head -1 | awk -F'load average:' '{print $2}' | sed 's/^[ \t]*//;s/[ \t]*$//' )
     
105. opsy=$( get_opsy )
     
106. arch=$( uname -m )
     
107. lbit=$( getconf LONG_BIT )
     
108. kern=$( uname -r )
     
109. ipv6=$( wget -qO- -t1 -T2 ipv6.icanhazip.com )
     
110. disk_size1=($( LANG=C df -hPl | grep -wvE '\-|none|tmpfs|devtmpfs|by-uuid|chroot|Filesystem' | awk '{print $2}' ))
     
111. disk_size2=($( LANG=C df -hPl | grep -wvE '\-|none|tmpfs|devtmpfs|by-uuid|chroot|Filesystem' | awk '{print $3}' ))
     
112. disk_total_size=$( calc_disk "${disk_size1[@]}" )
     
113. disk_used_size=$( calc_disk "${disk_size2[@]}" )
     
114. 
     
115. clear
     
116. next
     
117. echo -e "CPU model            : ${BLUE}$cname${PLAIN}"
     
118. echo -e "Number of cores      : ${BLUE}$cores${PLAIN}"
     
119. echo -e "CPU frequency        : ${BLUE}$freq MHz${PLAIN}"
     
120. echo -e "Total size of Disk   : ${BLUE}$disk_total_size GB ($disk_used_size GB Used)${PLAIN}"
     
121. echo -e "Total amount of Mem  : ${BLUE}$tram MB ($uram MB Used)${PLAIN}"
     
122. echo -e "Total amount of Swap : ${BLUE}$swap MB ($uswap MB Used)${PLAIN}"
     
123. echo -e "System uptime        : ${BLUE}$up${PLAIN}"
     
124. echo -e "Load average         : ${BLUE}$load${PLAIN}"
     
125. echo -e "OS                   : ${BLUE}$opsy${PLAIN}"
     
126. echo -e "Arch                 : ${BLUE}$arch ($lbit Bit)${PLAIN}"
     
127. echo -e "Kernel               : ${BLUE}$kern${PLAIN}"
     
128. next
     
129. io1=$( io_test )
     
130. echo -e "I/O speed(1st run)   : ${YELLOW}$io1${PLAIN}"
     
131. io2=$( io_test )
     
132. echo -e "I/O speed(2nd run)   : ${YELLOW}$io2${PLAIN}"
     
133. io3=$( io_test )
     
134. echo -e "I/O speed(3rd run)   : ${YELLOW}$io3${PLAIN}"
     
135. ioraw1=$( echo $io1 | awk 'NR==1 {print $1}' )
     
136. [ "`echo $io1 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw1=$( awk 'BEGIN{print '$ioraw1' * 1024}' )
     
137. ioraw2=$( echo $io2 | awk 'NR==1 {print $1}' )
     
138. [ "`echo $io2 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw2=$( awk 'BEGIN{print '$ioraw2' * 1024}' )
     
139. ioraw3=$( echo $io3 | awk 'NR==1 {print $1}' )
     
140. [ "`echo $io3 | awk 'NR==1 {print $2}'`" == "GB/s" ] && ioraw3=$( awk 'BEGIN{print '$ioraw3' * 1024}' )
     
141. ioall=$( awk 'BEGIN{print '$ioraw1' + '$ioraw2' + '$ioraw3'}' )
     
142. ioavg=$( awk 'BEGIN{printf "%.1f", '$ioall' / 3}' )
     
143. echo -e "Average I/O speed    : ${YELLOW}$ioavg MB/s${PLAIN}"
     
144. next
     
145. printf "%-32s%-24s%-14s\n" "Node Name" "IPv4 address" "Download Speed"
     
146. speed_v4 && next
     
147. if [[ "$ipv6" != "" ]]; then
     
148.     printf "%-32s%-24s%-14s\n" "Node Name" "IPv6 address" "Download Speed"
     
149.     speed_v6 && next
     
150. fi

复制代码