Skip to content

Commit

Permalink
optimize: support for more operational commands in seata-server.sh (#…
Browse files Browse the repository at this point in the history 
…5524)
  • Loading branch information
@doubleDimple
doubleDimple authored Aug 29, 2023
1 parent 436d674 commit b4e4711
Showing 1 changed file with 259 additions and 3 deletions.
262 changes: 259 additions & 3 deletions distribution/bin/seata-server.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,266 @@ JAVA_OPT="${JAVA_OPT} -Dspring.config.additional-location=${BASEDIR}/conf/ -Dspr
JAVA_OPT="${JAVA_OPT} -jar ${BASEDIR}/target/seata-server.jar"

CMD_LINE_ARGS=$@
NEW_ARGS=$(echo "${CMD_LINE_ARGS}" | sed -e 's/^start//g' -e 's/^restart//g' -e 's/^ //g')

show_usage() {
echo " Usage: sh seata-server.sh(for linux and mac) or cmd seata-server.bat(for"
echo " windows) [options]"
echo " Options:"
echo " --host, -h"
echo " The ip to register to registry center."
echo " --port, -p"
echo " The port to listen."
echo " Default: 0"
echo " --storeMode, -m"
echo " log store mode : file, db, redis"
echo " --serverNode, -n"
echo " server node id, such as 1, 2, 3.it will be generated according to the"
echo " snowflake by default"
echo " --seataEnv, -e"
echo " The name used for multi-configuration isolation."
echo " --sessionStoreMode, -ssm"
echo " session log store mode : file, db, redis"
echo " --lockStoreMode, -lsm"
echo " lock log store mode : file, db, redis"
echo " --help"
}

echo "Affected JVM parameters:$JAVA_OPT"

# start
echo "$JAVACMD ${JAVA_OPT} ${CMD_LINE_ARGS}" > ${BASEDIR}/logs/start.out 2>&1 &
nohup $JAVACMD ${JAVA_OPT} ${CMD_LINE_ARGS} >> ${BASEDIR}/logs/start.out 2>&1 &
echo "seata-server is starting, you can check the ${BASEDIR}/logs/start.out"
#echo "$JAVACMD ${JAVA_OPT} ${CMD_LINE_ARGS}" > ${BASEDIR}/logs/start.out 2>&1 &
#nohup $JAVACMD ${JAVA_OPT} ${CMD_LINE_ARGS} >> ${BASEDIR}/logs/start.out 2>&1 &
#echo "seata-server is starting, you can check the ${BASEDIR}/logs/start.out"

function validate_host() {
local host=$1
local re_ip="^([0-9]{1,3}\.){3}[0-9]{1,3}$"
if [[ ! $host =~ $re_ip ]]; then
echo "Invalid host: $host"
show_usage
exit 1
fi
}

function validate_port() {
local port="$1"
if ! [[ "$port" =~ ^[0-9]+$ ]]; then
echo "Error: Invalid port: $port"
show_usage
exit 1
fi
return 0
}

function validate_mode() {
local mode="$1"
if [[ "$mode" != 'file' && "$mode" != 'db' && "$mode" != 'redis' ]]; then
echo "Error: Invalid storeMode: $mode"
show_usage
exit 1
fi
return 0
}

function validate_serverNode() {
local serverNode="$1"
if ! [[ "$serverNode" =~ ^[0-9]+$ ]]; then
echo "Error: Invalid serverNode: $serverNode"
show_usage
exit 1
fi
return 0
}

while [[ $# -gt 0 ]]; do
key="$1"
case "$key" in
start|stop|restart)
;;
-h|--host)
if [[ -n "$2" ]]; then
validate_host "$2"
shift
else
echo "Error: Host value is required"
show_usage
exit 1
fi
;;
-p|--port)
if [[ -n "$2" ]]; then
validate_port "$2"
shift
else
echo "Error: Port value is required"
show_usage
exit 1
fi
;;
-m|--storeMode)
if [[ -n "$2" ]]; then
validate_mode "$2"
shift
else
echo "Error: storeMode value is required"
show_usage
exit 1
fi
;;
-ssm|--sessionStoreMode)
if [[ -n "$2" ]]; then
validate_mode "$2"
shift
else
echo "Error: sessionStoreMode value is required"
show_usage
exit 1
fi
;;
-lsm|--lockStoreMode)
if [[ -n "$2" ]]; then
validate_mode "$2"
shift
else
echo "Error: lockStoreMode value is required"
show_usage
exit 1
fi
;;
-e|--seataEnv)
if [[ -n "$2" ]]; then
shift
else
echo "Error: seataEnv value is required"
show_usage
exit 1
fi
;;
-n|--serverNode)
if [[ -n "$2" ]]; then
validate_serverNode "$2"
shift
else
echo "Error: serverNode value is required"
show_usage
exit 1
fi
;;
--help)
show_usage
exit 1
;;
*)
echo "Error: Unknown argument: $key"
show_usage
exit 1
;;
esac
shift
done

function start_server() {
echo "$JAVACMD ${JAVA_OPT} ${NEW_ARGS}" > ${BASEDIR}/logs/start.out 2>&1 &
nohup $JAVACMD ${JAVA_OPT} ${NEW_ARGS} >> ${BASEDIR}/logs/start.out 2>&1 &
echo "The seata-server is starting, you can check the ${BASEDIR}/logs/start.out"
}

function stop_server() {

PID=`ps aux | grep -i 'seata-server' | grep java | grep -v grep | awk '{print $2}'`

if [ -z "$PID" ]; then
echo "No seata-server running."
exit 1;
fi
echo "The seata-server(${PID}) is running..."
kill ${PID}
sleep 4
echo "Send shutdown request to seata-server(${PID}) OK"

}

function replace_old_arg() {
local old_arg="$1"
local new_arg="$2"
for i in "${!OLD_ARGS_ARRAY[@]}"
do
if [[ "${OLD_ARGS_ARRAY[$i]}" == "$old_arg" ]]; then
OLD_ARGS_ARRAY[$i]="$new_arg"
found=1
for j in $(seq $((i+1)) "${#OLD_ARGS_ARRAY[@]}")
do
if [[ "${OLD_ARGS_ARRAY[$j]}" == "$old_arg" ]]; then
unset OLD_ARGS_ARRAY[$j]
else
break
fi
done
if [[ "$i+1" -lt "${#OLD_ARGS_ARRAY[@]}" && "${OLD_ARGS_ARRAY[$i+1]}" != -* ]]; then
OLD_ARGS_ARRAY[$i+1]="${new_arg_value# }"
fi
break
fi
done
}

function restart_server() {

PID=`ps aux | grep -i 'seata-server' | grep java | grep -v grep | awk '{print $2}'`
#filtered
OLD_ARGS=`ps -p $PID -o args= | grep -v "^$0" | sed -E 's/.*seata-server.jar(.*)/\1/g'`
#Unfiltered
#OLD_ARGS=`ps -p $PID -o args= | grep -v "^$0"`
#echo "previous parameters ${OLD_ARGS}"
IFS=' ' read -r -a OLD_ARGS_ARRAY <<< "${OLD_ARGS}"
IFS=' ' read -r -a NEW_ARGS_ARRAY <<< "${NEW_ARGS}"

for new_arg in "${NEW_ARGS_ARRAY[@]}"
do
found=0
if [[ "$new_arg" == "-p" || "$new_arg" == "--port" ]]; then
replace_old_arg "-p" "$new_arg"
elif [[ "$new_arg" == "-h" || "$new_arg" == "--host" ]]; then
replace_old_arg "-h" "$new_arg"
elif [[ "$new_arg" == "-m" || "$new_arg" == "--storeMode" ]]; then
replace_old_arg "-m" "$new_arg"
elif [[ "$new_arg" == "-n" || "$new_arg" == "--serverNode" ]]; then
replace_old_arg "-n" "$new_arg"
elif [[ "$new_arg" == "-e" || "$new_arg" == "--seataEnv" ]]; then
replace_old_arg "-e" "$new_arg"
elif [[ "$new_arg" == "-ssm" || "$new_arg" == "--sessionStoreMode" ]]; then
replace_old_arg "-ssm" "$new_arg"
elif [[ "$new_arg" == "-lsm" || "$new_arg" == "--lockStoreMode" ]]; then
replace_old_arg "-lsm" "$new_arg"
fi
if [[ "$found" == 0 ]]; then
OLD_ARGS_ARRAY+=("$new_arg")
fi
done

NEW_ARGS=$(printf "%s " "${OLD_ARGS_ARRAY[@]}")
stop_server
echo "The seata-server restarting..."
start_server
}

if [ -z "${CMD_LINE_ARGS}" ]; then
start_server
else
case "${CMD_LINE_ARGS}" in
start*)
start_server
;;
stop*)
stop_server
;;
restart*)
restart_server
;;
*)
start_server
;;
esac
fi

0 comments on commit b4e4711

Please sign in to comment.