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