mirror of
https://github.com/APIParkLab/APIPark.git
synced 2026-06-14 20:41:15 +08:00
Liujian
293 lines
7.4 KiB
Bash
Executable File
293 lines
7.4 KiB
Bash
Executable File
#!/bin/sh
|
||
|
||
Cookie=""
|
||
|
||
if [[ "$ApiparkAddress" == "" ]]; then
|
||
ApiparkAddress="http://127.0.0.1:8288"
|
||
fi
|
||
if [[ "${ApintoAddress}" == "" ]]; then
|
||
ApintoAddress="http://apipark-apinto:9400"
|
||
fi
|
||
if [[ "$InfluxdbAddress" == "" ]]; then
|
||
InfluxdbAddress="http://apipark-influxdb:8086"
|
||
fi
|
||
|
||
|
||
|
||
if [[ "$NSQAddress" == "" ]]; then
|
||
NSQAddress="apipark-nsq:4150"
|
||
fi
|
||
if [[ "$LokiAddress" == "" ]]; then
|
||
LokiAddress="http://apipark-loki:3100"
|
||
fi
|
||
|
||
echo_fail() {
|
||
printf "\e[91m✘ Error:\e[0m $@\n" >&2
|
||
}
|
||
|
||
echo_pass() {
|
||
printf "\e[92m✔ Passed:\e[0m $@\n" >&2
|
||
}
|
||
|
||
echo_warn() {
|
||
printf "\e[93m⚠ Warning:\e[0m $@\n" >&2
|
||
}
|
||
|
||
echo_pause() {
|
||
printf "\e[94m⏸ Pause:\e[0m $1\n" >&2
|
||
}
|
||
|
||
echo_question() {
|
||
printf "\e[95m? Question:\e[0m $@\n" >&2
|
||
}
|
||
|
||
echo_info() {
|
||
printf "\e[96mℹ Info:\e[0m $1\n" >&2
|
||
}
|
||
|
||
echo_point() {
|
||
printf "\e[94m➜ Point:\e[0m $1\n" >&2
|
||
}
|
||
|
||
echo_bullet() {
|
||
printf "\e[94m• Step:\e[0m $1\n" >&2
|
||
}
|
||
|
||
echo_wait() {
|
||
printf "\e[95m⏳ Waiting:\e[0m $1\n" >&2
|
||
}
|
||
|
||
echo_split() {
|
||
echo "" >&2
|
||
echo "" >&2
|
||
echo -e "\e[94m────────────────────────────────────────────────────────────\e[0m" >&2
|
||
}
|
||
|
||
request_apipark() {
|
||
path=$1
|
||
body=$2
|
||
method=$3
|
||
if [[ "$method" == "" ]]; then
|
||
method="POST"
|
||
fi
|
||
if [[ "$Cookie" == "" ]]; then
|
||
cmd="curl -X ${method} -s -i -H \"Content-Type: application/json\" -d '$body' \"${ApiparkAddress}${path}\""
|
||
echo_info "Executing: $cmd" # 打印命令
|
||
response=$(eval "$cmd")
|
||
else
|
||
cmd="curl -X ${method} -s -i -H \"Content-Type: application/json\" -H \"Cookie: $Cookie\" -d '$body' \"${ApiparkAddress}${path}\""
|
||
echo_info "Executing: $cmd" # 打印命令
|
||
response=$(eval "$cmd")
|
||
fi
|
||
echo "$response"
|
||
}
|
||
|
||
request_apinto() {
|
||
path="$1"
|
||
body="$2"
|
||
echo_info "Executing: curl -i -X POST -H \"Content-Type: application/json\" \"${ApintoAddress}${path}\" -d '$body'"
|
||
response=$(curl -i -X POST -H "Content-Type: application/json" "${ApintoAddress}${path}" -d "$body")
|
||
status_code=$(echo "$response" | grep -E 'HTTP/[0-9.]+ [0-9]+' | awk '{print $2}' || echo "0")
|
||
echo_info "$response"
|
||
echo "$status_code"
|
||
}
|
||
|
||
login_apipark() {
|
||
# 执行登录请求并捕获响应头
|
||
body='{"name":"admin","password":"'"${ADMIN_PASSWORD}"'"}'
|
||
response=$(request_apipark "/api/v1/account/login/username" "$body")
|
||
|
||
# 从响应中提取 Set-Cookie 头
|
||
cookie=$(echo "$response" | grep -i "Set-Cookie" | sed 's/Set-Cookie: //;s/;.*//')
|
||
|
||
# 提取 JSON 主体（假设 JSON 在最后一行或响应中可识别）
|
||
json_body=$(echo "$response" | grep '^{.*}$')
|
||
# 提取 code 值
|
||
code=$(echo "$json_body" | sed 's/.*"code":\([0-9]*\).*/\1/')
|
||
|
||
# 检查 code 是否为 0
|
||
if [ "$code" -eq 0 ]; then
|
||
Cookie=$cookie
|
||
echo_pass "login success"
|
||
else
|
||
echo_fail "login failed: $json_body"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
set_cluster() {
|
||
# 设置集群地址
|
||
body='{"manager_address":"'"${ApintoAddress}"'"}'
|
||
path="/api/v1/cluster/reset"
|
||
response=$(request_apipark "$path" "$body" "PUT")
|
||
# 从响应中提取 code
|
||
code=$(echo "${response}" | grep '^{.*}$' | sed 's/.*"code":\([0-9]*\).*/\1/')
|
||
if [ "$code" -eq 0 ]; then
|
||
echo_pass "Set cluster successfully"
|
||
else
|
||
echo_fail "Set cluster failed: ${response}"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
set_loki() {
|
||
# 设置 loki 地址
|
||
body='{"config":{"url":"'"${LokiAddress}"'"}}'
|
||
path="/api/v1/log/loki"
|
||
response=$(request_apipark "$path" "$body")
|
||
# 从响应中提取 code
|
||
code=$(echo "${response}" | grep '^{.*}$' | sed 's/.*"code":\([0-9]*\).*/\1/')
|
||
if [ "$code" -eq 0 ]; then
|
||
echo_pass "Set loki successfully"
|
||
else
|
||
echo_fail "Set loki failed: ${response}"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
set_nsq() {
|
||
body="{
|
||
\"address\": [
|
||
\"${NSQAddress}\"
|
||
],
|
||
\"description\": \"auto init nsqd config\",
|
||
\"driver\": \"nsqd\",
|
||
\"formatter\": {
|
||
\"ai\": [
|
||
\"\$ai_provider\",
|
||
\"\$ai_model\",
|
||
\"\$ai_model_input_token\",
|
||
\"\$ai_model_output_token\",
|
||
\"\$ai_model_total_token\",
|
||
\"\$ai_model_cost\",
|
||
\"\$ai_provider_statuses\"
|
||
],
|
||
\"fields\": [
|
||
\"\$time_iso8601\",
|
||
\"\$request_id\",
|
||
\"\$api\",
|
||
\"\$provider\",
|
||
\"@ai\"
|
||
]
|
||
},
|
||
\"scopes\": [
|
||
\"access_log\"
|
||
],
|
||
\"topic\": \"apipark_ai_event\",
|
||
\"type\": \"json\"
|
||
}"
|
||
|
||
status_code=$(request_apinto "/api/output/ai_event" "$body")
|
||
echo "Status code: $status_code"
|
||
if [ "$status_code" -eq 200 ]; then
|
||
echo_pass "Update nsq successfully"
|
||
else
|
||
echo_fail "Update nsq failed: ${status_code}"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
set_influxdb() {
|
||
if [ -z "$InfluxdbToken" ]; then
|
||
echo_fail "Influxdb token is empty"
|
||
exit 1
|
||
fi
|
||
if [ -z "$InfluxdbOrg" ]; then
|
||
InfluxdbOrg="apipark"
|
||
fi
|
||
body='{"driver":"influxdb-v2","config":{"addr":"'"${InfluxdbAddress}"'","org":"'"${InfluxdbOrg}"'","token":"'${InfluxdbToken}'"}}'
|
||
response=$(request_apipark "/api/v1/monitor/config" "$body")
|
||
# 从响应中提取 code
|
||
code=$(echo "${response}" | grep '^{.*}$' | sed 's/.*"code":\([0-9]*\).*/\1/')
|
||
if [ "$code" -eq 0 ]; then
|
||
echo_pass "Update influxdb config successfully"
|
||
else
|
||
echo_fail "Update influxdb config failed: ${response}"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
retry() {
|
||
max_wait="$1"
|
||
shift
|
||
cmd="$@"
|
||
|
||
sleep_interval=2
|
||
curr_wait=0
|
||
|
||
until $cmd
|
||
do
|
||
if [ "$curr_wait" -ge "$max_wait" ]
|
||
then
|
||
echo "Command '$cmd' failed after $curr_wait seconds."
|
||
return 1
|
||
else
|
||
curr_wait=$((curr_wait + sleep_interval))
|
||
sleep "$sleep_interval"
|
||
fi
|
||
done
|
||
}
|
||
|
||
is_init() {
|
||
path="/api/v1/system/general"
|
||
method="GET"
|
||
response=$(request_apipark "$path" "" "$method")
|
||
# 从响应中提取 site_prefix
|
||
site_prefix=$(echo "${response}" | grep '^{.*}$' | sed 's/.*"site_prefix":"\{0,1\}\([^"]*\)"\{0,1\}.*/\1/')
|
||
if [ -z "$site_prefix" ]; then
|
||
echo_pass "No apipark openapi address set"
|
||
echo false
|
||
else
|
||
echo_pass "Apipark openapi address found: $site_prefix"
|
||
echo true
|
||
fi
|
||
}
|
||
|
||
set_openapi_config() {
|
||
IP=$(dig +short myip.opendns.com @resolver1.opendns.com | grep -E '^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$')
|
||
if [ -z "$IP" ]; then
|
||
echo_fail "Failed to resolve IP address"
|
||
exit 1
|
||
fi
|
||
body='{"site_prefix":"http://'"${IP}"':18288"}'
|
||
response=$(request_apipark "/api/v1/system/general" "$body")
|
||
# 从响应中提取 code
|
||
code=$(echo "${response}" | grep '^{.*}$' | sed 's/.*"code":\([0-9]*\).*/\1/')
|
||
if [ "$code" -eq 0 ]; then
|
||
echo_pass "Update apipark openapi address successfully"
|
||
else
|
||
echo_fail "Update apipark openapi address failed: ${response}"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
wait_for() {
|
||
waitName=$1
|
||
cmd=$2
|
||
echo ${cmd}
|
||
echo_wait "Waiting for ${waitName} to start..."
|
||
retry 30 ${cmd}
|
||
if [ $? -eq 0 ]; then
|
||
echo_pass "${waitName} has been installed successfully"
|
||
else
|
||
echo_fail "${waitName} installation failed"
|
||
exit 1
|
||
fi
|
||
|
||
}
|
||
|
||
wait_for_apipark() {
|
||
wait_for "apipark" "curl -s -o /dev/null ${ApiparkAddress}/api/v1/account/login"
|
||
}
|
||
|
||
wait_for_apinto() {
|
||
wait_for "apinto" "curl -s -o /dev/null ${ApintoAddress}/api/router"
|
||
}
|
||
|
||
wait_for_influxdb() {
|
||
wait_for "influxdb" "curl -s -o /dev/null ${InfluxdbAddress}/api/v2/health"
|
||
}
|
||
|
||
|
||
|