#!/usr/bin/env bash
# 获取服务目录
xxx_dir=$(cd $(dirname "${BASH_SOURCE[0]}") && pwd)
# 端口检测间隔
w_interval=3
# 启动后端口检测次数
max_retried_times=50
REDIS_INSTALL_DIR=/bin/
REDIS_CONFIG_FILE=/etc/redis/redis.conf
MONGO_INSTALL_DIR=/usr/
MONGO_CONFIG_FILE=/etc/mongodb.conf
NGINX_INSTALL_DIR=/usr/sbin/
NGINX_CONFIG_DIR=/etc/nginx/nginx.conf
CONSUL_INSTALL_DIR=/opt/consul
HYBOX=/opt/xxx/hybox/bin
ES=/opt/elasticsearch/bin
declare -A monitor_third_service
monitor_third_service=(
[redis]=6379 # 数据结构服务器
[mongo]=27017 # mongo数据库
[mysql]=3306 # mysql数据库
[web]=80 # 平台前端
[consul]=8500 # 分析控制服务
[hybox]=29200 # yyy
[elasticsearch]=9200 # es
#[mysql_]=3307 # yyymysql
[solr]=8983 # solr
#[soffice]=8100
)
declare -a control_service
control_service=(
lts
jobtracker
yarn_tasktracker
comparison
tldw
graph
analyzer
case_management
chart_management
#tactic
data_gov
gateway
#dictionary
)
# 控制服务的列表
declare -A service_install_dir
service_install_dir=(
[lts]=$xxx_dir/lts-admin
[jobtracker]=$xxx_dir/jobtracker
[yarn_tasktracker]=$xxx_dir/yarn-tasktracker
[comparison]=$xxx_dir/comparison
[tldw]=$xxx_dir/service
[analyzer]=$xxx_dir/analyzer
[case_management]=$xxx_dir/case-management
[tactic]=$xxx_dir/tactic
[graph]=$xxx_dir/graphdb
[chart_management]=$xxx_dir/chart-management
[data_gov]=$xxx_dir/dg_back
[gateway]=$xxx_dir/gateway
[dictionary]=$xxx_dir/dictionary
)
declare -A service_control_script
service_control_script=(
[lts]=./bin/lts-admin.sh
[jobtracker]=./bin/tldw-jobtracker.sh
[yarn_tasktracker]=./bin/tldw-yarn-tasktracker.sh
[comparison]=./bin/tldw-comparison.sh
[tldw]=./bin/tldw-platform-server.sh
[analyzer]=./bin/tldw-pc.sh
[case_management]=./bin/tldw-casemanager.sh
[tactic]=./bin/tactic-server.sh
[graph]=./bin/tldw-graph-server.sh
[chart_management]=./bin/tldw-chartmanager.sh
[data_gov]=./bin/tldw-data-gov.sh
[gateway]=./bin/tldw-gateway.sh
[dictionary]=./bin/dictionary.sh
)
#服务监控列表，格式为[服务]=端口
declare -A service_port
service_port=(
[lts]=8720 # 分布式任务调度框架，管理服务
[jobtracker]=8721 # 分布式任务调度框架，作业管理
[comparison]=8731 # 比对服务
[yarn_tasktracker]=17030 # 分布式任务调度框架，任务执行
[graph]=8999 # 图服务
[tldw]=8080 # 平台后台
[analyzer]=8051 # 分析服务
[case_management]=8060 # 案件管理
[tactic]=8099 # 战法服务
[chart_management]=8810 # 图表管理
[data_gov]=18080 # 案件数据治理
[gateway]=8070
[dictionary]=8820 # 查询工具
)
# 输出格式化
my_echo()
{
local RED="\033[31m"
local GREEN="\033[32m"
local YELLOW="\033[33m"
local END="\033[0m"
if [ $# -eq 1 ];then
echo -e "${GREEN}$1${END}"
elif [ $# -eq 2 ];then
if [[ ! $1 =~ ^(1|2)$ ]];then
echo -e "${RED}$1 is not a insufficent argument, the argument must be 1 or 2, other argument must not permitted.$END"
return 1
fi
case $1 in
1)
echo -e "${YELLOW}${2}${END}"
;;
2)
echo -e "${RED}${2}${END}"
;;
*)
echo -e "${RED}Insufficent argument${END}"
;;
esac
else
echo -e "${RED}Input arguments are not 2, please make sure of it${END}"
return 1
fi
}
#帮助信息输出
help()
{
echo ""
echo "Usage:"
echo "1 Check service status"
my_echo "./start_all.sh status"
echo "2 Control tldw service"
my_echo "./start_all.sh service <operation> <service> "
echo "operation variable is one of start|stop|restart"
echo "service variable is service in list ( ${control_service[@]} )，you can edit zhe control_service array to decision which service has been controlled"
my_echo "Example:./start_all.sh service restart graph restart graphdb service"
my_echo "Otherwise:You can use ./start_all.sh service start|stop|restart all to control all service"
echo "3 Control thirdpart service"
my_echo "./start_all.sh start_mongo|start_redis|start_nginx|start_mysql|start_consul "
echo "You can use command upward to start thirdpart service"
my_echo "./start_all.sh stop_mongo|stop_redis|stop_nginx|stop_mysql|stop_consul "
echo "You can use command upward to stop thirdpart service"
echo "You can also use command './start_all.sh start_thirdpart|stop_thirdpart' to control all thirdpart service"
echo ""
echo "使用方法:"
echo "1 检查服务状态"
my_echo "./start_all.sh status"
echo "2 天网服务控制"
my_echo "./start_all.sh service <operation> <service>"
echo "operation 变量填写start|stop|restart"
echo "service 变量填写列表（${control_service[@]} ）中的服务名 ，你可以通过修改脚本中的control_service数组来觉得控制哪些服务"
my_echo "例如:./start_all.sh service restart graph 重启图库服务"
my_echo "此外:你可以使用./start_all.sh service start|stop|restart all 来控制所有服务"
echo "3 第三方服务控制"
my_echo "./start_all.sh start_mongo|start_redis|start_nginx|start_mysql|start_consul "
echo "你可以使用上面的命令启动第三方服务"
my_echo "./start_all.sh stop_mongo|stop_redis|stop_nginx|stop_mysql|stop_consul "
echo "你可以使用上面的命令停止第三方服务"
echo "你也可以使用'./start_all.sh start_thirdpart|stop_thirdpart' 命令同时控制多个第三方服务"
}
exception()
{
my_echo 2 "Error Function!"
help
exit 1
}
# 检测服务状态并返回状态码
service_exists()
{
service_name=$1
port=$2
pid=$(netstat -nptl | awk '{print $0, $4}' \
| grep -E :"$port"$ | awk '{print $7}' | awk -F "/" '{print $1}')
if [ "$pid" != "" ];then
my_echo "[OK]$(hostname) - $service_name (port: $port) started ok, pid is $pid"
return 0
fi
my_echo 1 "[STARTING]$(hostname) - $service_name (port: $port) ... ...\r\c"
return 2
}
#检测服务状态
check_service()
{
service_name=$1
port=$2
pid=$(netstat -nptl | awk '{print $0, $4}' \
| grep -E :"$port"$ | awk '{print $7}' | awk -F "/" '{print $1}')
if [ "$pid" != "" ];then
printf "\033[32m%-8s %-20s %-10s %-10s\n\033[0m" "[OK]" "$service_name" "$port" "$pid"
else
printf "\033[31m%-8s %-20s %-10s %-10s\n\033[0m" "[NOT OK]" "$service_name" "$port" "not exist"
fi
}
# 检查所有服务状态
check_services()
{
echo "==================thirdpart service=================="
printf "%-8s %-20s %-10s %-10s\n" "Status" "Service Name" "Port" "PID"
for s_ in $(echo ${!monitor_third_service[*]});do
check_service ${s_} ${monitor_third_service[${s_}]}
done
echo "====================tldw service====================="
printf "%-8s %-20s %-10s %-10s\n" "Status" "Service Name" "Port" "PID"
for s_ in $(echo ${control_service[*]});do
check_service ${s_} ${service_port[${s_}]}
done
}
# 多次检测服务状态，直到确认服务启动
check_long_time_service()
{
local count=0
s_name=$1
s_port=$2
if [ "x$(service_exists ${s_name} ${s_port} > /dev/null 2>&1;echo $?)" == "x0" ];then
if service_exists ${s_name} ${s_port}
then
return 0
fi
fi
echo ""
while [ "x$count" != "x${max_retried_times}" ]
do
sleep 5
count=$((count + 1))
my_echo "retried times $count/$max_retried_times \c"
if service_exists ${s_name} ${s_port}
then
return 0
fi
done
# when failed
my_echo ""
my_echo 2 "[STARTED FAILED] $s_name $s_port."
return 1
}
# 获取所有服务pid
get_all_services_pids()
{
pids=""
for s_ in ${!service_port[*]};do
port=${service_port[${s_}]}
pid=$(netstat -nptl | awk '{print $0, $4}' \
| grep -E :"$port"$ | awk '{print $7}' | awk -F "/" '{print $1}')
pids=${pids}" "${pid}
done
echo "[INFO]all active pids are "$pids
}
# 获取所有服务状态
status()
{
check_services
}
service(){
local operation=$1
local service=$2
if [[ ! ${control_service[@]} =~ $service && $service != "all" ]];then
my_echo 2 "$service is not valid service"
return 1
fi
if [ ! -d ${service_install_dir[$service]} ]; then
my_echo 2 "$service dir ${service_install_dir[$service]} not exist,please check!"
return 1
fi
if [[ ! $operation =~ ^(start|stop|restart)$ ]];then
my_echo 2 "$operation is not valid operation"
return 1
fi
if [[ $service == "all" ]];then
for service in $(echo ${control_service[*]})
do
service $operation $service
done
else
cd ${service_install_dir[$service]} && ${service_control_script[$service]} $operation
if [[ $operation =~ ^(start|restart)$ ]];then
check_long_time_service $service ${service_port[$service]}
fi
fi
}
##***修改配置文件模块***
edit_properties(){
config_file=$1
config_key=$2
config_new_value=$3
if [ ! -f $config_file ];then
echo "can not find file $config_file"
return 1
fi
grep -i $config_key $config_file >/dev/null
if [ $? -ne 0 ] ;then
echo "can not find propertied $config_key in file $config_file"
return 1
fi
sed -i "s/^$config_key.*/$config_key=$config_new_value/g" $config_file
echo "change $config_key in $config_file to $config_new_value"
return 0
}
# ***服务依赖组件控制模块***
# Start redis server
start_redis()
{
if service_exists redis ${monitor_third_service[redis]}
then
return 0
fi
$REDIS_INSTALL_DIR/redis-server $REDIS_CONFIG_FILE
check_long_time_service redis ${monitor_third_service[redis]}
}
stop_redis()
{
ps aux|grep redis-server|grep -v grep|awk '{print $2}'|xargs -i kill {}
}
# Start mongodb
start_mongo()
{
if service_exists mongo ${monitor_third_service[mongo]}
then
return 0
fi
$MONGO_INSTALL_DIR/bin/mongod -config $MONGO_CONFIG_FILE
check_long_time_service mongo ${monitor_third_service[mongo]}
}
stop_mongo()
{
$MONGO_INSTALL_DIR/bin/mongod -f $MONGO_CONFIG_FILE --shutdown
}
# mysql control
start_mysql()
{
systemctl start mysqld
}
stop_mysql()
{
systemctl stop mysqld
}
# Start tldw web server
start_web()
{
if service_exists web ${monitor_third_service[web]}
then
return 0
fi
$NGINX_INSTALL_DIR/nginx -c $NGINX_CONFIG_DIR
check_long_time_service web ${monitor_third_service[web]}
}
stop_web()
{
$NGINX_INSTALL_DIR/nginx -s stop
}
start_consul()
{
if service_exists consul ${monitor_third_service[consul]}
then
return 0
fi
cd $CONSUL_INSTALL_DIR&& nohup ./consul agent -dev -client 0.0.0.0 >/dev/null &
sleep 3
check_long_time_service consul ${monitor_third_service[consul]}
}
stop_consul()
{
$CONSUL_INSTALL_DIR/consul leave
}
start_hybox()
{
(cd $HYBOX;./proc_manager.sh start)
}
stop_hybox()
{
(cd $HYBOX;./proc_manager.sh stop)
}
start_hybox_mysql()
{
systemctl start mysqld
sleep 1
}
stop_hybox_mysql()
{
systemctl stop mysqld
sleep 1
}
start_es()
{
#(su elastic;$ES/elasticsearch -d;exit)
su es -c "$ES/elasticsearch -d"
}
stop_es()
{
kill -9 $(ps -ef |grep "elasticsearch"|grep -v grep| awk '{print $2}')
}
start_solr()
{
systemctl start solr
check_long_time_service solr ${monitor_third_service[solr]}
}
stop_solr()
{
systemctl stop solr
}
start_soffice()
{
cd /opt/openoffice4/program && ./soffice -headless -accept="socket,host=127.0.0.1,port=8100;urp;" -nofirststartwizard &
sleep 3
check_long_time_service soffice ${monitor_third_service[soffice]}
}
stop_soffice()
{
ps -ef |grep "[^]]soffice"| awk '{print $2}'|xargs kill -9
}
start_thirdpart()
{
start_redis
start_mongo
start_mysql
start_web
start_consul
#start_hybox_mysql
start_hybox
start_es
start_solr
#start_soffice
}
stop_thirdpart()
{
stop_redis
stop_mongo
stop_mysql
stop_web
stop_consul
#stop_hybox_mysql
stop_hybox
stop_es
stop_solr
#stop_soffice
}
if [ $# -eq 0 ];then
help
exit 1
fi
command -v $1 2>&1 1>/dev/null || exception
##脚本主方法，通过eval动态调用脚本中的函数
eval $* 2>/dev/null
浙公网安备 33010602011771号