Skip to content
This repository has been archived by the owner on Jun 6, 2024. It is now read-only.

Commit

Permalink
fix review comments
Browse files Browse the repository at this point in the history
change init script
  • Loading branch information
Binyang2014 committed Sep 17, 2019
1 parent 1998650 commit a68a678
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 15 deletions.
2 changes: 1 addition & 1 deletion src/k8s-job-exit-spec/config/k8s-job-exit-spec.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -613,7 +613,7 @@ spec:

###########################################################################
# Range: [129, 192]
# Owner: PAI_FC
# Owner: PAI_FC and PAI_RUNTIME
# Description: User Container issued failures:
# -> Involuntary failures caused by OS Signal
###########################################################################
Expand Down
42 changes: 28 additions & 14 deletions src/kube-runtime/src/init
Original file line number Diff line number Diff line change
Expand Up @@ -21,31 +21,39 @@ set -o errexit
set -o nounset
set -o pipefail

CHILD_PROCESS="UNKNOWN"

function exit_handler()
{
EXIT_CODE=$?
echo "start execute exit handler" >> ${PAI_INIT_LOG_FILE}
if [[ $EXIT_CODE -eq 0 ]]; then
exit 0
fi

if [[ $EXIT_CODE -eq 200 ]]; then
exit 250
fi
echo "start execute exit handler"
echo "child process is ${CHILD_PROCESS}, exit code is ${EXIT_CODE}"
if [[ $CHILD_PROCESS = "FRAMEWORK_BARRIER" ]]; then
if [[ $EXIT_CODE -eq 200 ]]; then
exit 250
fi

if [[ $EXIT_CODE -eq 201 ]]; then
exit 251
fi
if [[ $EXIT_CODE -eq 201 ]]; then
exit 251
fi

if [[ $EXIT_CODE -eq 210 ]]; then
exit 252
if [[ $EXIT_CODE -eq 210 ]]; then
exit 252
fi
fi

if [[ $EXIT_CODE -eq 10 ]]; then
exit 253
if [[ $CHILD_PROCESS = "PORT_CONFLICT_CHECKER" ]]; then
if [[ $EXIT_CODE -eq 10 ]]; then
exit 253
fi
fi

exit ${EXIT_CODE}
echo "Unknown exit code, platform error"
exit 1
}

trap exit_handler EXIT
Expand Down Expand Up @@ -80,32 +88,38 @@ cd ${PAI_WORK_DIR}

# framework barrier
# priority=0
CHILD_PROCESS="FRAMEWORK_BARRIER"
echo "frameworkbarrier start"
${PAI_INIT_DIR}/frameworkbarrier 2>&1 | tee ${PAI_LOG_DIR}/barrier.log
echo "frameworkbarrier returns $?"
${PAI_INIT_DIR}/frameworkbarrier 2>&1 | tee ${PAI_LOG_DIR}/barrier.log
echo "barrier returns $?"

# error spec
# priority=1
CHILD_PROCESS="ERROR_SPEC"
cp ${PAI_CONFIG_DIR}/runtime-exit-spec.yaml ${PAI_RUNTIME_DIR}
echo "copy exit spec returns $?" >> ${PAI_INIT_LOG_FILE}

# generate runtime env variables
# priority=10
CHILD_PROCESS="ENV_GENERATOR"
python ${PAI_INIT_DIR}/parser.py genenv framework.json > ${PAI_RUNTIME_DIR}/runtime_env.sh
echo "parser.py genenv returns $?"

# generate jobconfig
# priority=11
CHILD_PROCESS="CONFIG_GENERATOR"
python ${PAI_INIT_DIR}/parser.py genconf framework.json > ${PAI_RUNTIME_DIR}/job_config.yaml
echo "parser.py genconf returns $?"

# Init plugins
# priority=12
CHILD_PROCESS="PLUGIN_INITIALIZER"
python ${PAI_INIT_DIR}/initializer.py ${PAI_RUNTIME_DIR}/job_config.yaml ${PAI_WORK_DIR}/plugins ${PAI_RUNTIME_DIR} ${FC_TASKROLE_NAME}
echo "initializer.py returns $?"

# Init plugins
# check port conflict
CHILD_PROCESS="PORT_CONFLICT_CHECKER"
python ${PAI_INIT_DIR}/port.py ${PAI_RUNTIME_DIR}/runtime_env.sh 2>> ${PAI_LOG_DIR}/initializer.log
echo "port.py.py returns $?" >> ${PAI_INIT_LOG_FILE}

Expand Down

0 comments on commit a68a678

Please sign in to comment.