345 lines
7.5 KiB
Bash
345 lines
7.5 KiB
Bash
|
#!/bin/bash
|
|||
|
|
#trap 'rm -rf /home/ceiec/oam;rm -rf /home/ceiec/collector;' EXIT
|
|||
|
|
#带颜色输出内容
|
|||
|
|
#$1 颜色 red green yellow 警告 red 成功 green 提示/输入 yellow
|
|||
|
|
#$2 输出语句
|
|||
|
|
#$3 -n y/n 是否换行输出
|
|||
|
|
function pinfo(){
|
|||
|
|
LINE=""
|
|||
|
|
if [[ "$#" -gt 2 ]]; then
|
|||
|
|
#statements
|
|||
|
|
if [[ "$3" = "n" ]]; then
|
|||
|
|
#statements
|
|||
|
|
LINE="-n"
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
case $1 in
|
|||
|
|
"red")
|
|||
|
|
echo -e $LINE "\e[1;31m "${2}"\e[0m"
|
|||
|
|
;;
|
|||
|
|
"green")
|
|||
|
|
echo -e $LINE "\e[1;32m "${2}"\e[0m"
|
|||
|
|
;;
|
|||
|
|
"yellow")
|
|||
|
|
echo -e $LINE "\e[1;33m "${2}"\e[0m"
|
|||
|
|
;;
|
|||
|
|
"blue")
|
|||
|
|
echo -e $LINE "\e[1;34m "${2}"\e[0m"
|
|||
|
|
;;
|
|||
|
|
*)
|
|||
|
|
echo $LINE $2
|
|||
|
|
;;
|
|||
|
|
esac
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
READ_NOTNULL_TEMP=""
|
|||
|
|
function read_notNull(){
|
|||
|
|
echo -n "$1"
|
|||
|
|
stty erase '^H'
|
|||
|
|
read READ_NOTNULL_TEMP
|
|||
|
|
length=$#
|
|||
|
|
if [[ length -gt 1 ]]; then #参数大于1个，验证输入是否正确
|
|||
|
|
#statements
|
|||
|
|
FLAG=0
|
|||
|
|
while [[ true ]]; do
|
|||
|
|
#statements
|
|||
|
|
for i in "$@"; do
|
|||
|
|
if [[ "$READ_NOTNULL_TEMP" = "$i" ]]; then
|
|||
|
|
#statements
|
|||
|
|
FLAG=1
|
|||
|
|
break
|
|||
|
|
fi
|
|||
|
|
done
|
|||
|
|
if [[ "$FLAG" = 1 ]]; then
|
|||
|
|
#statements
|
|||
|
|
break
|
|||
|
|
else
|
|||
|
|
echo "input failed,please input again！"
|
|||
|
|
echo -n "$1"
|
|||
|
|
read READ_NOTNULL_TEMP
|
|||
|
|
fi
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
|
|||
|
|
else
|
|||
|
|
while [ -z "$READ_NOTNULL_TEMP" ]
|
|||
|
|
do
|
|||
|
|
echo "input is empty,please input again！"
|
|||
|
|
echo -n "$1"
|
|||
|
|
read READ_NOTNULL_TEMP
|
|||
|
|
done
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
function modify_file(){
|
|||
|
|
if [ $# != 2 ]
|
|||
|
|
then
|
|||
|
|
echo "usage: modify_file [prop_name] [prop_value]"
|
|||
|
|
exit 0
|
|||
|
|
fi
|
|||
|
|
prop_name="$1"
|
|||
|
|
prop_value="$2"
|
|||
|
|
echo "modify_file $PROP_FILE: $prop_name $prop_value"
|
|||
|
|
if [ -z "$(cat $PROP_FILE |grep $prop_name)" ]
|
|||
|
|
then
|
|||
|
|
echo "" >> $PROP_FILE
|
|||
|
|
echo "$prop_name=$prop_value" >> $PROP_FILE
|
|||
|
|
else
|
|||
|
|
sed -i "s|^$prop_name.*|$prop_name=$prop_value|" $PROP_FILE
|
|||
|
|
fi
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#当前目录
|
|||
|
|
temp=`dirname $0`
|
|||
|
|
CURDIR=`cd $temp;pwd`
|
|||
|
|
|
|||
|
|
#设备类型
|
|||
|
|
DEVICE='ADC-A016'
|
|||
|
|
#java打包程序路径
|
|||
|
|
PROC_PATH=${CURDIR}"/process/oam.zip"
|
|||
|
|
#脚本及配置文件路径
|
|||
|
|
SCRIPT_HOME=${CURDIR}"/A016/master/script"
|
|||
|
|
#安装路径
|
|||
|
|
OAM_HOME="/home/ceiec"
|
|||
|
|
#配置文件路径
|
|||
|
|
CONFIG_FILE="/home/ceiec/oam/conf/config.properties"
|
|||
|
|
read_notNull 'please input device type[ASEM-T102/ADC-A016](t/a):' t a
|
|||
|
|
device_type="$READ_NOTNULL_TEMP"
|
|||
|
|
|
|||
|
|
is_master="1"
|
|||
|
|
#安装类型标志 1 A106 master 2 A106 slave 3 T102
|
|||
|
|
role=-1
|
|||
|
|
DB_FILE="oam.db.A016"
|
|||
|
|
if [ "t" = "$device_type" ];then
|
|||
|
|
DEVICE="ASEM-T102"
|
|||
|
|
SCRIPT_HOME=${CURDIR}"/T102/script"
|
|||
|
|
DB_FILE="oam.db.T102"
|
|||
|
|
role=3
|
|||
|
|
else
|
|||
|
|
role=1
|
|||
|
|
read_notNull 'please input install as master or slave (m/s)' m s
|
|||
|
|
flag="$READ_NOTNULL_TEMP"
|
|||
|
|
#echo "$flag"
|
|||
|
|
if [ "s" = "$flag" ];then
|
|||
|
|
is_master="0"
|
|||
|
|
role=2
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
#echo "is_master:$is_master"
|
|||
|
|
#安装java程序
|
|||
|
|
if [ "$is_master" = "1" ];then
|
|||
|
|
pinfo green "install oam program..."
|
|||
|
|
unzip $PROC_PATH -d $OAM_HOME >/dev/null
|
|||
|
|
if [ $? = 1 ];then
|
|||
|
|
pinfo red "install FAILD."
|
|||
|
|
exit 0
|
|||
|
|
fi
|
|||
|
|
PROP_FILE="$CONFIG_FILE"
|
|||
|
|
modify_file 'equipment_name' $DEVICE
|
|||
|
|
|
|||
|
|
cp ${CURDIR}"/db/$DB_FILE" ${OAM_HOME}"/oam/db/oam.db"
|
|||
|
|
read_notNull 'start oam right now?(y/n)' y n
|
|||
|
|
start="$READ_NOTNULL_TEMP"
|
|||
|
|
if [ "$start" = 'y' ];then
|
|||
|
|
pinfo green "now start program..."
|
|||
|
|
bash ${OAM_HOME}"/oam/shell/startup.sh" >/dev/null 2>$1
|
|||
|
|
if [ $? = 0 ];then
|
|||
|
|
pinfo green "start SUCCESS"
|
|||
|
|
else
|
|||
|
|
pinfo red "start FAILD.check \"$OAM_HOME/oam/log/gloam.log\" for help..."
|
|||
|
|
exit 0
|
|||
|
|
fi
|
|||
|
|
else
|
|||
|
|
echo "please start program with \"$OAM_HOME/oam/shell/startup.sh\""
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
enter_password=""
|
|||
|
|
function readPasswd(){
|
|||
|
|
enter_password=""
|
|||
|
|
stty -echo cbreak erase '^H'
|
|||
|
|
while true
|
|||
|
|
do
|
|||
|
|
character=$(dd if=/dev/tty bs=1 count=1 2> /dev/null)
|
|||
|
|
case $character in
|
|||
|
|
$(echo -e "\n"))
|
|||
|
|
break
|
|||
|
|
;;
|
|||
|
|
$(echo -e "\b"))
|
|||
|
|
if [ -z "$enter_password" ]; then
|
|||
|
|
echo -n -e "\b \b"
|
|||
|
|
enter_password=$(echo "$password" | sed 's/.$//g')
|
|||
|
|
fi
|
|||
|
|
;;
|
|||
|
|
*)
|
|||
|
|
enter_password=$enter_password$character
|
|||
|
|
echo -n '*'
|
|||
|
|
;;
|
|||
|
|
esac
|
|||
|
|
done
|
|||
|
|
stty -cbreak echo
|
|||
|
|
echo ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
enter_info=""
|
|||
|
|
function readInfo_notNull(){
|
|||
|
|
if [ $# != 1 ];then
|
|||
|
|
echo "usage:readInfo_notNull [hint]"
|
|||
|
|
exit 1
|
|||
|
|
fi
|
|||
|
|
hint="$1"
|
|||
|
|
stty erase '^H'
|
|||
|
|
read -p "$hint:" enter_info
|
|||
|
|
while [ -z "enter_info" ];do
|
|||
|
|
read -p "this parameter is must,please input again:" enter_info
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
NODE_ID=""
|
|||
|
|
function getNodeIdByIP(){
|
|||
|
|
if [ $# != 1 ];then
|
|||
|
|
echo "usage:getNodeIdByIP [node_ip]"
|
|||
|
|
exit 1
|
|||
|
|
fi
|
|||
|
|
node_ip="$1"
|
|||
|
|
NODE_ID=$(sqlite3 ${CURDIR}"/db/$DB_FILE" <<EOF |awk '{print $0}'
|
|||
|
|
select id from node_table where ip="$node_ip";
|
|||
|
|
EOF
|
|||
|
|
)
|
|||
|
|
echo "NODE_ID=$NODE_ID"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#安装脚本程序方法
|
|||
|
|
function installScript(){
|
|||
|
|
if [ $# != 1 ];then
|
|||
|
|
echo "usage:installScript [script_dir]"
|
|||
|
|
fi
|
|||
|
|
script_dir=$1
|
|||
|
|
|
|||
|
|
read -p "please input script file home[default /home/ceiec]:" script_home
|
|||
|
|
if [ -z "$script_home" ];then
|
|||
|
|
script_home="/home/ceiec"
|
|||
|
|
else
|
|||
|
|
if [ ! -d "$script_home" ];then
|
|||
|
|
mkdir -p "$script_home"
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
#echo "$script_home"
|
|||
|
|
unzip ${script_dir}"/collector.zip" -d "$script_home" >/dev/null
|
|||
|
|
if [ $? != 0 ];then
|
|||
|
|
pinfo red "script file unzip FAILD"
|
|||
|
|
exit 1
|
|||
|
|
fi
|
|||
|
|
cd ${script_home}"/collector"
|
|||
|
|
#修改主机配置文件config
|
|||
|
|
if [[ -f config ]];then
|
|||
|
|
for i in 1 2 3 4
|
|||
|
|
do
|
|||
|
|
ipinfo=$(sqlite3 ${CURDIR}"/db/"${DB_FILE} <<EOF |awk -va="$i" 'NR==a{print $0}'
|
|||
|
|
select ip||","||user||","||pwd from node_table;
|
|||
|
|
EOF
|
|||
|
|
)
|
|||
|
|
#写入配置
|
|||
|
|
PROP_FILE="$script_home/collector/config"
|
|||
|
|
modify_file 'ip'${i} "$ipinfo"
|
|||
|
|
if [ "t" = "$device_type" ];then
|
|||
|
|
break
|
|||
|
|
fi
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
#key src_dir
|
|||
|
|
srcDir="$script_home"
|
|||
|
|
|
|||
|
|
#key dest_dir
|
|||
|
|
destDir="$script_home"
|
|||
|
|
|
|||
|
|
#写入配置
|
|||
|
|
PROP_FILE="$script_home/collector/config"
|
|||
|
|
modify_file 'src_dir' "$srcDir"
|
|||
|
|
modify_file 'dest_dir' "$destDir"
|
|||
|
|
fi
|
|||
|
|
#为脚本授权
|
|||
|
|
chmod +x *.sh
|
|||
|
|
if [ -f expect_scp ];then
|
|||
|
|
chmod +x expect_scp
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
#新建定时任务
|
|||
|
|
if [ -z "$(grep "$script_home/collector/getSingleMacInfo.sh" /var/spool/cron/root | grep -v 'grep' 2>/dev/null)" ];then
|
|||
|
|
echo "*/5 * * * * $script_home/collector/getSingleMacInfo.sh" >>/var/spool/cron/root
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
if [[ -f ${script_home}"/collector/merge.sh" && -z "$(grep "$script_home/collector/merge.sh" /var/spool/cron/root |grep -v 'grep' 2>/dev/null)" ]];then
|
|||
|
|
echo "*/1 * * * * $script_home/collector/merge.sh">>/var/spool/cron/root
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
#处理configs目录下配置文件
|
|||
|
|
cd ${script_home}"/collector/configs"
|
|||
|
|
<<UNUSE
|
|||
|
|
cpu="cpu_cpu.cfg"
|
|||
|
|
disk="disk_disk.cfg"
|
|||
|
|
mem="memory_memory.cfg"
|
|||
|
|
net="net_net.cfg"
|
|||
|
|
sys="systeminfo_servicessysinfo.cfg"
|
|||
|
|
UNUSE
|
|||
|
|
#获取node_table的id
|
|||
|
|
readInfo_notNull "please input the localhost"
|
|||
|
|
#只针对ADC-A016
|
|||
|
|
ip="$enter_info"
|
|||
|
|
getNodeIdByIP "$ip"
|
|||
|
|
while [ -z "$NODE_ID" ];do
|
|||
|
|
#获取node_table的id
|
|||
|
|
readInfo_notNull "this ip is not exist,please input the device ip again"
|
|||
|
|
#只针对ADC-A016
|
|||
|
|
ip="$enter_info"
|
|||
|
|
getNodeIdByIP "$ip"
|
|||
|
|
|
|||
|
|
done
|
|||
|
|
node_id="$NODE_ID"
|
|||
|
|
for file in *
|
|||
|
|
do
|
|||
|
|
temp=$(awk 'NR==1{print $0}' $file|cut -d'=' -f2)
|
|||
|
|
value=${node_id}${temp}
|
|||
|
|
PROP_FILE=$(readlink -f "$file")
|
|||
|
|
modify_file 'pubInfo' "$value"
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
#处理device_config 脚本
|
|||
|
|
if [ ! -f '/usr/local/bin/device_config' ];then
|
|||
|
|
cp $script_dir/device_config /usr/local/bin
|
|||
|
|
if [ $? = 0 ];then
|
|||
|
|
chmod +x /usr/local/bin/device_config
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#安装脚本程序
|
|||
|
|
case $role in
|
|||
|
|
1 )
|
|||
|
|
installScript "$SCRIPT_HOME"
|
|||
|
|
;;
|
|||
|
|
2 )
|
|||
|
|
SCRIPT_HOME="${CURDIR}/A016/slave"
|
|||
|
|
installScript "$SCRIPT_HOME"
|
|||
|
|
;;
|
|||
|
|
3 )
|
|||
|
|
SCRIPT_HOME="${CURDIR}/T102/script"
|
|||
|
|
installScript "$SCRIPT_HOME"
|
|||
|
|
;;
|
|||
|
|
|
|||
|
|
* )
|
|||
|
|
pinfo red "`basename $0` exec error "
|
|||
|
|
exit 1
|
|||
|
|
;;
|
|||
|
|
esac
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|