#!/bin/bash
########################################################################
#
# FUNCTION : main
# DESCRIPTION: copy dir to
# CALLS : 无
# CALLED BY : 无
# INPUT : 无
# OUTPUT : 无
# LOCAL VAR : 无
# USE GLOBVAR: 无
# RETURN : 无
# CHANGE DIR : 无
#######################################################################
#####################################################
# 用户相关变量
######################################################
SERVICE_USER=des
SERVICE_GROUP=des
MODULE_NAME=console-home
MODULE_PATH=des
#####################################################
#
# 安装路径
#
#####################################################
PRODUCT_PATH=/opt/des/console-des
TOMCAT_PATH=/opt/des/tomcat1212
RETURN_CODE_SUCCESS=0
RETURN_CODE_ERROR=1
#对应缺省：目录700 文件600权限
umask 0077
LOGMAXSIZE=5120
BASE_LOGGER_PATH=/var/log/${MODULE_PATH}
######################################################################
# FUNCTION : chkUser
# DESCRIPTION : 检查当前用户是否是$SERVICE_USER
# CALLS : 无
# CALLED BY : 任何需要调用此函数的地方
# INPUT : $1 想要检查的用户名
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 0 成功
# 1 失败
######################################################################
chkUser()
{
logger_without_echo "check current user"
local curUser=$(/usr/bin/whoami | /usr/bin/awk '{print $1}')
if [ "$curUser" = "$SERVICE_USER" ]; then
logger_without_echo "check current user success"
return 0
else
die "${MODULE_NAME} can only run by ${SERVICE_USER}"
fi
}
######################################################################
# FUNCTION : initLogDir
# DESCRIPTION : 创建日志目录
# CALLS : 无
# CALLED BY : 本脚本初始化日志
# INPUT : 无
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 无
######################################################################
initLogDir()
{
if [ -e "$LOGGER_PATH" ]; then
return 0
else
mkdir -p ${LOGGER_PATH}
echo "init log dir success."
fi
}
######################################################################
# FUNCTION : status
# DESCRIPTION : 检查当前进程状态
# CALLS : 无
# CALLED BY : 任何需要调用此函数的地方
# INPUT : $1 想要检查的进程名
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 0 成功
# 1 失败
######################################################################
internalStatus()
{
# do some work, such as send one msg to process
local pid=$(ps -ww -eo pid,cmd | grep -w "${MODULE_NAME}.main.proc" |grep -w java | grep -vwE "grep|vi|vim|tail|cat" | awk '{print $1}' | head -1)
RETVAL=${RETURN_CODE_ERROR}
[ -n "$pid" ] && RETVAL=${RETURN_CODE_SUCCESS}
if [ "$RETVAL" -eq ${RETURN_CODE_SUCCESS} ]; then
logger_without_echo "normal"
else
logger_without_echo "abnormal"
fi
return "$RETVAL"
}
######################################################################
# FUNCTION : logger_without_echo
# DESCRIPTION : 记录日志到对应文件中，不输出到终端。
# CALLS : 无
# CALLED BY : 无
# INPUT : 无
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 无
######################################################################
logger_without_echo()
{
local logsize=0
if [ -e "$LOGGER_FILE" ]; then
logsize=`ls -lk ${LOGGER_FILE} | awk -F " " '{print $5}'`
else
touch ${LOGGER_FILE}
chown ${SERVICE_USER}: ${LOGGER_FILE}
chmod 600 ${LOGGER_FILE}
fi
if [ "$logsize" -gt "$LOGMAXSIZE" ]; then
# 每次删除10000行，约300K
sed -i '1,10000d' "$LOGGER_FILE"
fi
echo "[` date -d today +\"%Y-%m-%d %H:%M:%S\"`,000] $*" >>"$LOGGER_FILE"
}
######################################################################
# FUNCTION : logger
# DESCRIPTION : 记录日志到对应文件中，同时输出到终端。
# CALLS : 无
# CALLED BY : 无
# INPUT : 无
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 无
######################################################################
logger()
{
logger_without_echo $*
echo "$*"
}
######################################################################
# FUNCTION : die
# DESCRIPTION : 记录日志并退出程序。
# CALLS : 无
# CALLED BY : 无
# INPUT : 无
# OUTPUT : 无
# READ GLOBVAR : 无
# WRITE GLOBVAR: 无
# RETURN : 无
######################################################################
die()
{
logger "$*"
exit ${RETURN_CODE_ERROR}
}
ERR_GET_LOCK=101
ERR_PARAMETERS=102
#######################################################################
# shell文件锁封装函数，
# 参数1：文件锁路径，
# 参数2，需要上锁的实际执行动作，
# 参数3~n，传给实际动作的所有参数
lockWrapCall()
{
local lockFile=$1
local action=$2
[ -n "$lockFile" -a -n "$action" ] || return $ERR_PARAMETERS
local dirName=$(dirname $lockFile)
[ -d "$dirName" ] || return $ERR_PARAMETERS
shift 2
# 定义信号捕捉流程，异常停止进程时清除文件锁
trap 'rm -f $lockFile; logger "trap a stop singal"' 1 2 3 15
####################################
## 文件锁，只允许一个进程执行
####################################
{
flock -no 100
if [ $? -eq 1 ]; then
local lockPid=$(cat $lockFile)
lockPid=$(echo $lockPid)
if [ -z "$lockPid" ]; then
logger "can't get lock file:$lockFile, lockPid is empty, no need to run $action"
return $ERR_GET_LOCK
else
lockPid=$(echo $lockPid)
local openPids=$(lsof -Fp $lockFile)
if echo "$openPids" | grep "^p${lockPid}$" > /dev/null; then
logger "can't get lock file:$lockFile, lockPid:$lockPid is running, no need to run $action"
return $ERR_GET_LOCK
fi
fi
logger "success get lock file:$lockFile, lockPid:$lockPid is not running"
fi
echo $$ > $lockFile
$action "$@"
local ret=$?
# 删除文件锁，使得上述动作参数的子进程不再持有锁
rm -f $lockFile
return $ret
} 100<>$lockFile
# 恢复为默认信号处理
trap '-' 1 2 3 15
}
