Skip to content

Commit

Permalink
Merge pull request #684 from BlitherBoom812/BlitherBoom812-patch-1
Browse files Browse the repository at this point in the history
retry command
  • Loading branch information
DragonAura authored May 7, 2024
2 parents 005e738 + 0d9dd09 commit 7d7d33e
Showing 1 changed file with 213 additions and 56 deletions.
269 changes: 213 additions & 56 deletions dependency/shell/run.sh
Original file line number Diff line number Diff line change
@@ -1,71 +1,228 @@
#!/usr/local

python_dir=/usr/local/PlayerCode/CAPI/python/PyAPI
python_main_dir=/usr/local/PlayerCode/CAPI/python
playback_dir=/usr/local/playback
playback_dir=/usr/local/output
map_dir=/usr/local/map
mkdir -p $playback_dir

if [ $EXPOSED -eq 1 ]; then
nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $TIME --mode $MODE --mapResource $MAP --url $URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
server_pid=$!
else
nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $TIME --mode $MODE --mapResource $MAP --notAllowSpectator --url $URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
server_pid=$!
# initialize
if [[ "${MODE}" == "ARENA" ]]; then
MODE_NUM=0
elif [[ "${MODE}" == "COMPETITION" ]]; then
MODE_NUM=1
fi
sleep 5
for k in {1..2}
do
pushd /usr/local/team$k
if [ $k -eq 1 ]; then
for i in {1..4}

# set default value
: "${TEAM_SEQ_ID:=0}"
: "${TEAM_LABELS:=Student:Tricker}"
: "${TEAM_LABEL:=Student}"
: "${EXPOSED=1}"
: "${MODE_NUM=0}"
: "${GAME_TIME=10}"
: "${CONNECT_IP=172.17.0.1}"


get_current_team_label() {
if [ $TEAM_SEQ_ID -eq $2 ]; then
echo "find current team label: $1"
current_team_label=$1
fi
}

read_array() {
callback=$1
echo "read array: set callback command: $callback"

IFS=':' read -r -a fields <<< "$2"

count=0 # loop count

for field in "${fields[@]}"
do
echo "parse field: $field"
param0=$field

# call command
run_command="$callback $param0 $count"
echo "Call Command: $run_command"
$run_command

count=$((count+1))
done
}


function retry_command {
local command="$1"
local max_attempts=2
local attempt_num=1
local sleep_seconds=10

while [ $attempt_num -le $max_attempts ]; do
echo "Attempt $attempt_num / $max_attempts to run command: $command"

eval $command &
local PID=$!

sleep $sleep_seconds

if kill -0 $PID 2>/dev/null; then
echo "Connected to server successfully."
return 0
else
echo "Failed to connect to server. Retrying..."
((attempt_num++))
fi
done

echo "Failed to connect to server after $max_attempts attempts."
return 1
}



if [ "$TERMINAL" = "SERVER" ]; then
map_path=$map_dir/$MAP_ID.txt
# allow spectator always.
nice -10 ./Server --port 8888 --studentCount 4 --trickerCount 1 --resultFileName $playback_dir/result --gameTimeInSecond $GAME_TIME --mode $MODE_NUM --mapResource $map_path --url $SCORE_URL --token $TOKEN --fileName $playback_dir/video --startLockFile $playback_dir/start.lock > $playback_dir/server.log 2>&1 &
server_pid=$!
echo "server pid: $server_pid"
ls $playback_dir

echo "SCORE URL: $SCORE_URL"
echo "FINISH URL: $FINISH_URL"

echo "waiting..."
sleep 30 # wait connection time
echo "watching..."

if [ -f $playback_dir/start.lock ]; then
ps -p $server_pid
while [ $? -eq 0 ]
do
j=$((i - 1))
if [ -f "./player$i.py" ]; then
cp -r $python_main_dir $python_main_dir$i
cp -f ./player$i.py $python_main_dir$i/PyAPI/AI.py
nice -0 python3 $python_main_dir$i/PyAPI/main.py -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
elif [ -f "./capi$i" ]; then
nice -0 ./capi$i -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
else
echo "ERROR. $i is not found."
fi
sleep 1
ps -p $server_pid > /dev/null 2>&1
done

# update score by finish url

echo "Getting result score..."
result=$(cat $playback_dir/result.json)
score0=$(echo "$result" | grep -oP '(?<="Student":)\d+') # Student score
score1=$(echo "$result" | grep -oP '(?<="Tricker":)\d+') # Tricker score
echo "Result score: Student: $score0, Tricker: $score1"

# detect two team seqs:
echo "Parsing TEAM_LABELS: $TEAM_LABELS"
TEAM_SEQ_ID=0
read_array get_current_team_label $TEAM_LABELS
if [[ "${current_team_label}" == "Student" ]]; then
echo "Parse Success: 1st team is Student"
finish_payload='{"result": {"status": "Finished", "scores": ['${score0}', '${score1}']}}'
elif [[ "${current_team_label}" == "Tricker" ]]; then
echo "Parse Success: 1st team is Tricker"
finish_payload='{"result": {"status": "Finished", "scores": ['${score1}', '${score0}']}}'
else
echo "Parse Failure: 1st team is Unknown"
finish_payload='{"result": {"status": "Crashed", "scores": [0, 0]}}'
fi

if [[ -n $finish_payload ]]; then
echo "FINISH_URL: $FINISH_URL, payload: $finish_payload. Start update score..."
curl $FINISH_URL -X POST -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d "${finish_payload}" > $playback_dir/send.log 2>&1
else
echo "Payload not set."
fi

# Congratulations! You have finished the competition!!!!!
# touch $playback_dir/finish.lock
echo "Finish!"
else
for i in {5..5}
do
j=$((i - 1))
if [ -f "./player$i.py" ]; then
echo "Failed to start game."
touch $playback_dir/finish.lock
touch temp.lock
mv -f temp.lock $playback_dir/video.thuaipb
kill -9 $server_pid
finish_payload='{"result": {"status": "Crashed", "scores": [0, 0]}}'
curl $FINISH_URL -X POST -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d "${finish_payload}" > $playback_dir/send.log 2>&1
fi


elif [ "$TERMINAL" = "CLIENT" ]; then
echo "Client Mode! Team Label data - $TEAM_LABEL"

# parse team label name
current_team_label=$TEAM_LABEL

# k is an enum (1,2), 1 = Student, 2 = Tricker
if [ "$current_team_label" = "Student" ]; then
k=1
elif [ "$current_team_label" = "Tricker" ]; then
k=2
else
echo "Error: Invalid Team Label"
exit
fi
pushd /usr/local/code
if [ $k -eq 1 ]; then
for i in {1..4}
do
j=$((i - 1)) # student player id from 0 to 3
code_name=Student$i
if [ -f "./$code_name.py" ]; then
echo "find ./$code_name.py"
cp -r $python_main_dir $python_main_dir$i
cp -f ./$code_name.py $python_main_dir$i/PyAPI/AI.py

command="nice -0 python3 $python_main_dir$i/PyAPI/main.py -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"

retry_command "$command" &

ps -aux |grep main.py

elif [ -f "./$code_name" ]; then
echo "find ./$code_name"

command="nice -0 ./$code_name -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"

ps -aux |grep $code_name

else
echo "ERROR. $code_name is not found."
fi
done
else
i=5
j=4 # tricker id is 4
code_name=Tricker
if [ -f "./$code_name.py" ]; then
echo "find ./$code_name.py"
cp -r $python_main_dir $python_main_dir$i
cp -f ./player$i.py $python_main_dir$i/PyAPI/AI.py
nice -0 python3 $python_main_dir$i/PyAPI/main.py -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
elif [ -f "./capi$i" ]; then
nice -0 ./capi$i -I 127.0.0.1 -P 8888 -p $j > $playback_dir/team$k-player$j.log 2>&1 &
cp -f ./$code_name.py $python_main_dir$i/PyAPI/AI.py

command="nice -0 python3 $python_main_dir$i/PyAPI/main.py -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"

retry_command "$command"

ps -aux |grep main.py
elif [ -f "./$code_name" ]; then
echo "find ./$code_name"

command="nice -0 ./$code_name -I $CONNECT_IP -P $PORT -p $j > $playback_dir/team$k-player$j.log 2>&1"

retry_command "$command"

ps -aux |grep $code_name
else
echo "ERROR. $i is not found."
echo "ERROR. $code_name is not found."
fi
done
fi
popd
done
fi

sleep 10
# curl $CONNECT_IP:$PORT

if [ -f $playback_dir/start.lock ]; then
ps -p $server_pid
while [ $? -eq 0 ]
do
sleep 1
ps -p $server_pid > /dev/null 2>&1
done
# result=$(cat /usr/local/playback/result.json)
# score0=$(echo "$result" | grep -oP '(?<="Student":)\d+')
# score1=$(echo "$result" | grep -oP '(?<="Tricker":)\d+')
# curl $URL -X PUT -H "Content-Type: application/json" -H "Authorization: Bearer $TOKEN" -d '{"result":[{"team_id":0, "score":'${score0}'}, {"team_id":1, "score":'${score1}'}], "mode":'${MODE}'}'> $playback_dir/send.log 2>&1
touch $playback_dir/finish.lock
echo "Finish"
sleep $((GAME_TIME * 2))

popd
else
echo "Failed to start game."
touch $playback_dir/finish.lock
touch temp.lock
mv -f temp.lock $playback_dir/video.thuaipb
kill -9 $server_pid
echo "VALUE ERROR: TERMINAL is neither SERVER nor CLIENT."
fi

0 comments on commit 7d7d33e

Please sign in to comment.