mirror of
https://github.com/whyour/qinglong.git
synced 2026-09-15 19:57:07 +08:00
whyour
a94e665054
* fix: harden task lifecycle and scheduler readiness * fix: confine log writes to the configured log directory * fix: verify complete build artifacts and untracked inputs * fix: reconcile scheduler state and make stop win startup races * fix: isolate cron generations and serialize scheduler recovery
275 lines
8.1 KiB
Bash
Executable File
275 lines
8.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
create_token() {
|
|
local token_command="ts-node-transpile-only ${dir_root}/back/token.ts"
|
|
local token_file="${dir_root}/static/build/token.js"
|
|
if [[ -f $token_file ]]; then
|
|
token_command="node ${token_file}"
|
|
fi
|
|
__ql_token__=$(eval "$token_command")
|
|
}
|
|
|
|
get_token() {
|
|
local token_data expiration token_value
|
|
local currentTimeStamp=${EPOCHSECONDS:-$(date +%s)}
|
|
if [[ -f "$file_auth_token" ]] && token_data=$(jq -er \
|
|
'select((.expiration | type) == "number" and (.value | type) == "string" and (.value | length) > 0 and (.value | test("[\r\n]") | not)) | .expiration, .value' \
|
|
"$file_auth_token" 2>/dev/null); then
|
|
expiration=${token_data%%$'\n'*}
|
|
token_value=${token_data#*$'\n'}
|
|
if [[ "$expiration" =~ ^[1-9][0-9]{0,10}$ && -n "$token_value" && "$token_value" != *$'\n'* && "$token_value" != *$'\r'* ]] && \
|
|
(( currentTimeStamp < expiration )); then
|
|
__ql_token__=$token_value
|
|
return 0
|
|
fi
|
|
fi
|
|
create_token
|
|
}
|
|
|
|
# Read the status response once; preserve multiline error messages and the
|
|
# legacy code/message variables used by the shell API callers.
|
|
ql_parse_status_response() {
|
|
# Both task status and statistics endpoints normally return this exact body.
|
|
# Match the whole document; all other JSON and malformed responses still
|
|
# use jq. Keep its missing-message result for existing callers.
|
|
if [[ "$1" == '{"code":200}' ]]; then
|
|
code=200
|
|
message=null
|
|
return 0
|
|
fi
|
|
local parsed
|
|
code=""
|
|
message=""
|
|
parsed=$(jq -r '.code, .message' <<< "$1") || return $?
|
|
code=${parsed%%$'\n'*}
|
|
message=${parsed#*$'\n'}
|
|
}
|
|
|
|
add_cron_api() {
|
|
local currentTimeStamp=$(date +%s)
|
|
if [[ $# -eq 1 ]]; then
|
|
local schedule=$(echo "$1" | awk -F ":" '{print $1}')
|
|
local command=$(echo "$1" | awk -F ":" '{print $2}')
|
|
local name=$(echo "$1" | awk -F ":" '{print $3}')
|
|
local sub_id=$(echo "$1" | awk -F ":" '{print $4}')
|
|
else
|
|
local schedule="$1"
|
|
local command="$2"
|
|
local name="$3"
|
|
local sub_id="$4"
|
|
fi
|
|
|
|
if [[ ! $sub_id ]]; then
|
|
sub_id="null"
|
|
fi
|
|
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons?t=$currentTimeStamp" \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{\"name\":\"${name//\"/\\\"}\",\"command\":\"${command//\"/\\\"}\",\"schedule\":\"$schedule\",\"sub_id\":$sub_id}" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '%s -> 添加成功' "$name"
|
|
else
|
|
t '%s -> 添加失败(%s)' "$name" "$message"
|
|
fi
|
|
}
|
|
|
|
update_cron_api() {
|
|
local currentTimeStamp=$(date +%s)
|
|
if [[ $# -eq 1 ]]; then
|
|
local schedule=$(echo "$1" | awk -F ":" '{print $1}')
|
|
local command=$(echo "$1" | awk -F ":" '{print $2}')
|
|
local name=$(echo "$1" | awk -F ":" '{print $3}')
|
|
local id=$(echo "$1" | awk -F ":" '{print $4}')
|
|
else
|
|
local schedule="$1"
|
|
local command="$2"
|
|
local name="$3"
|
|
local id="$4"
|
|
fi
|
|
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons?t=$currentTimeStamp" \
|
|
-X 'PUT' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{\"name\":\"${name//\"/\\\"}\",\"command\":\"${command//\"/\\\"}\",\"schedule\":\"$schedule\",\"id\":\"$id\"}" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '%s -> 更新成功' "$name"
|
|
else
|
|
t '%s -> 更新失败(%s)' "$name" "$message"
|
|
fi
|
|
}
|
|
|
|
update_cron_command_api() {
|
|
local currentTimeStamp=$(date +%s)
|
|
if [[ $# -eq 1 ]]; then
|
|
local command=$(echo "$1" | awk -F ":" '{print $1}')
|
|
local id=$(echo "$1" | awk -F ":" '{print $2}')
|
|
else
|
|
local command="$1"
|
|
local id="$2"
|
|
fi
|
|
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons?t=$currentTimeStamp" \
|
|
-X 'PUT' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{\"command\":\"${command//\"/\\\"}\",\"id\":\"$id\"}" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '%s -> 更新成功' "$command"
|
|
else
|
|
t '%s -> 更新失败(%s)' "$command" "$message"
|
|
fi
|
|
}
|
|
|
|
del_cron_api() {
|
|
local ids="$1"
|
|
local currentTimeStamp=$(date +%s)
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons?t=$currentTimeStamp" \
|
|
-X 'DELETE' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "[$ids]" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '成功'
|
|
else
|
|
t '失败(%s)' "$message"
|
|
fi
|
|
}
|
|
|
|
update_cron() {
|
|
local ids="$1"
|
|
local status="$2"
|
|
local pid="${3:-''}"
|
|
local logPath="$4"
|
|
local lastExecutingTime="${5:-0}"
|
|
local runningTime="${6:-0}"
|
|
local exitCode="${7:-}"
|
|
local currentTimeStamp=${EPOCHSECONDS:-$(date +%s)}
|
|
local dataRaw="{\"ids\":[$ids],\"status\":\"$status\",\"pid\":\"$pid\",\"log_path\":\"$logPath\",\"last_execution_time\":$lastExecutingTime,\"last_running_time\":$runningTime"
|
|
if [[ -n $exitCode ]]; then
|
|
dataRaw="${dataRaw},\"exit_code\":$exitCode"
|
|
fi
|
|
dataRaw="${dataRaw}}"
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons/status?t=$currentTimeStamp" \
|
|
-X 'PUT' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "$dataRaw" \
|
|
--compressed
|
|
)
|
|
ql_parse_status_response "$api" || true
|
|
if [[ $code != 200 ]]; then
|
|
if [[ ! $message ]]; then
|
|
message="$api"
|
|
fi
|
|
echo -e "${message}"
|
|
fi
|
|
}
|
|
|
|
notify_api() {
|
|
local title="$1"
|
|
local content="$2"
|
|
local currentTimeStamp=$(date +%s)
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/system/notify?t=$currentTimeStamp" \
|
|
-X 'PUT' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{\"title\":\"${title//\"/\\\"}\",\"content\":\"${content//\"/\\\"}\"}" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '通知发送成功🎉'
|
|
else
|
|
t '通知失败(%s)' "$message"
|
|
fi
|
|
}
|
|
|
|
find_cron_api() {
|
|
local params="$1"
|
|
local currentTimeStamp=$(date +%s)
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/crons/detail?$params&t=$currentTimeStamp" \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--compressed
|
|
)
|
|
data=$(echo "$api" | jq -r .data)
|
|
if [[ $data == 'null' ]]; then
|
|
echo -e ""
|
|
else
|
|
name=$(echo "$api" | jq -r .data.name)
|
|
echo -e "$name"
|
|
fi
|
|
}
|
|
|
|
update_auth_config() {
|
|
local body="$1"
|
|
local tip="$2"
|
|
local currentTimeStamp=$(date +%s)
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port}/open/system/auth/reset?t=$currentTimeStamp" \
|
|
-X 'PUT' \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{$body}" \
|
|
--compressed
|
|
)
|
|
code=$(echo "$api" | jq -r .code)
|
|
message=$(echo "$api" | jq -r .message)
|
|
if [[ $code == 200 ]]; then
|
|
t '%s成功🎉' "$tip"
|
|
else
|
|
t '%s失败(%s)' "$tip" "$message"
|
|
fi
|
|
}
|
|
|
|
record_cron_stat() {
|
|
local ref_id="$1"
|
|
local exit_code="${2:-0}"
|
|
local elapsed="${3:-0}"
|
|
[[ $ref_id ]] && [[ $ref_id -gt 0 ]] 2>/dev/null || return
|
|
|
|
local api=$(
|
|
curl -s --noproxy "*" "http://localhost:${ql_port:-5700}/open/dashboard/record" \
|
|
-X POST \
|
|
-H "Authorization: Bearer ${__ql_token__}" \
|
|
-H "Content-Type: application/json;charset=UTF-8" \
|
|
--data-raw "{\"ref_id\":$ref_id,\"code\":$exit_code,\"elapsed\":$elapsed}" \
|
|
--compressed
|
|
)
|
|
ql_parse_status_response "$api" || true
|
|
if [[ $code != 200 ]]; then
|
|
if [[ ! $message ]]; then
|
|
message="$api"
|
|
fi
|
|
echo -e "${message}"
|
|
fi
|
|
}
|
|
|
|
get_token
|