mirror of
https://github.com/APIParkLab/APIPark.git
synced 2026-06-04 10:13:53 +08:00
update docker build script
This commit is contained in:
Liujian
+2
-1
@@ -7,4 +7,5 @@
|
||||
/.vscode/
|
||||
.air.toml
|
||||
/tmp/
|
||||
/work
|
||||
/work
|
||||
/cmd/
|
||||
|
||||
@@ -746,7 +746,9 @@ func (i *imlServiceModule) Delete(ctx context.Context, id string) error {
|
||||
Id: id,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
if err.Error() != "nil" {
|
||||
return err
|
||||
}
|
||||
}
|
||||
err = client.Subscribe().Offline(ctx, &gateway.SubscribeRelease{
|
||||
Service: id,
|
||||
|
||||
+8
-4
@@ -1,10 +1,14 @@
|
||||
# 名称：apipark通用镜像
|
||||
# 创建时间：2022-10-25
|
||||
FROM centos:7.9.2009
|
||||
MAINTAINER liujian
|
||||
FROM alpine:latest
|
||||
|
||||
RUN sed -i 's|https://dl-cdn.alpinelinux.org/alpine|https://mirrors.aliyun.com/alpine|g' /etc/apk/repositories \
|
||||
&& apk update \
|
||||
&& apk add --no-cache curl tzdata bind-tools
|
||||
|
||||
ENV TZ=Asia/Shanghai
|
||||
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
||||
|
||||
RUN ln -sf /usr/share/zoneinfo/${TZ} /etc/localtime \
|
||||
&& echo "Asia/Shanghai" > /etc/timezone
|
||||
|
||||
ARG APP
|
||||
|
||||
|
||||
@@ -10,6 +10,11 @@ BuildMode=$3
|
||||
if [[ "${BuildMode}" == "" ]];then
|
||||
BuildMode="all"
|
||||
fi
|
||||
|
||||
if [[ "${ARCH}" == "" ]];then
|
||||
ARCH="amd64"
|
||||
fi
|
||||
|
||||
# 编译可执行文件
|
||||
./scripts/build.sh "cmd" "" "${BuildMode}" ${ARCH}
|
||||
|
||||
@@ -22,12 +27,11 @@ mkdir -p scripts/cmd/ && cp cmd/${APP} scripts/cmd/ && cp cmd/apipark_ai_event_l
|
||||
VERSION=$(gen_version)
|
||||
|
||||
|
||||
if [[ "${ARCH}" == "" ]];then
|
||||
ARCH="amd64"
|
||||
fi
|
||||
|
||||
OPTIONS=""
|
||||
if [[ "${ARCH}" == "arm" ]];then
|
||||
SYS_ARCH=$(arch)
|
||||
if [[ (${SYS_ARCH} == "aarch64" || ${SYS_ARCH} == "arm64") && $ARCH == "amd64" ]];then
|
||||
OPTIONS="--platform=linux/amd64"
|
||||
elif [[ ${SYS_ARCH} == "amd64" && $ARCH == "arm64" ]];then
|
||||
OPTIONS="--platform=linux/arm64"
|
||||
fi
|
||||
|
||||
|
||||
@@ -1,11 +1,10 @@
|
||||
#!/bin/sh
|
||||
|
||||
set -e
|
||||
|
||||
source ./init_config.sh
|
||||
|
||||
OLD_IFS="$IFS"
|
||||
IFS=","
|
||||
arr=(${REDIS_ADDR})
|
||||
IFS="$OLD_IFS"
|
||||
|
||||
|
||||
@@ -21,10 +20,11 @@ echo -e "redis:" >> config.yml
|
||||
echo -e " user_name: ${REDIS_USER_NAME}" >> config.yml
|
||||
echo -e " password: ${REDIS_PWD}" >> config.yml
|
||||
echo -e " addr: " >> config.yml
|
||||
for s in ${arr[@]}
|
||||
for s in $REDIS_ADDR
|
||||
do
|
||||
echo -e " - $s" >> config.yml
|
||||
done
|
||||
|
||||
echo -e "nsq:" >> config.yml
|
||||
echo -e " addr: ${NSQ_ADDR}" >> config.yml
|
||||
echo -e " topic_prefix: ${NSQ_TOPIC_PREFIX}" >> config.yml
|
||||
@@ -38,5 +38,31 @@ echo -e " log_period: ${ERROR_PERIOD}" >> config.yml
|
||||
|
||||
cat config.yml
|
||||
nohup ./apipark >> run.log 2>&1 &
|
||||
wait_for_apipark
|
||||
|
||||
nohup ./apipark_ai_event_listen >> run.log 2>&1 &
|
||||
|
||||
if [[ ${Init} == "true" ]];then
|
||||
login_apipark
|
||||
r=$(is_init)
|
||||
if [[ $r == "true" ]];then
|
||||
echo "Already initialized, skipping initialization."
|
||||
exit 0
|
||||
fi
|
||||
wait_for_influxdb
|
||||
|
||||
wait_for_apinto
|
||||
set_cluster
|
||||
|
||||
wait_for_influxdb
|
||||
set_influxdb
|
||||
|
||||
set_loki
|
||||
set_nsq
|
||||
set_openapi_config
|
||||
# 重启apipark
|
||||
kill -9 $(pgrep apipark)
|
||||
nohup ./apipark >> run.log 2>&1 &
|
||||
fi
|
||||
|
||||
tail -F run.log
|
||||
Executable
+292
@@ -0,0 +1,292 @@
|
||||
#!/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"
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user