72 lines
1.8 KiB
Bash
Executable File
72 lines
1.8 KiB
Bash
Executable File
#!/bin/sh
|
|
set -eu
|
|
|
|
: "${CONSUL_HTTP_ADDR:?need CONSUL_HTTP_ADDR}"
|
|
: "${SERVICE_NAME:?need SERVICE_NAME}"
|
|
: "${SERVICE_ADDR:?need SERVICE_ADDR}"
|
|
: "${SERVICE_PORT:?need SERVICE_PORT}"
|
|
|
|
SERVICE_ID="${SERVICE_ID:-${SERVICE_NAME}-${SERVICE_ADDR}-${SERVICE_PORT}}"
|
|
SERVICE_TAGS="${SERVICE_TAGS:-}"
|
|
CHECK_TYPE="${CHECK_TYPE:-tcp}" # tcp|http
|
|
CHECK_PATH="${CHECK_PATH:-/healthz}" # http 模式才用
|
|
CHECK_INTERVAL="${CHECK_INTERVAL:-10s}"
|
|
CHECK_TIMEOUT="${CHECK_TIMEOUT:-2s}"
|
|
DEREG_AFTER="${DEREG_AFTER:-1m}"
|
|
|
|
# 组装 Tags 的 JSON 数组
|
|
if [ -n "$SERVICE_TAGS" ]; then
|
|
# 逗号分隔转 JSON 数组
|
|
TAGS_JSON=$(printf '%s' "$SERVICE_TAGS" | awk -F, '
|
|
BEGIN{printf "["}
|
|
{for(i=1;i<=NF;i++){gsub(/^ *| *$/, "", $i); printf "%s\"%s\"", (i>1?",":""), $i}}
|
|
END{printf "]"}
|
|
')
|
|
else
|
|
TAGS_JSON="[]"
|
|
fi
|
|
|
|
# 组装 Check JSON
|
|
if [ "$CHECK_TYPE" = "http" ]; then
|
|
CHECK_JSON=$(cat <<EOF
|
|
{
|
|
"Name": "http-${SERVICE_PORT}",
|
|
"HTTP": "http://${SERVICE_ADDR}:${SERVICE_PORT}${CHECK_PATH}",
|
|
"Method": "GET",
|
|
"Interval": "${CHECK_INTERVAL}",
|
|
"Timeout": "${CHECK_TIMEOUT}",
|
|
"DeregisterCriticalServiceAfter": "${DEREG_AFTER}"
|
|
}
|
|
EOF
|
|
)
|
|
else
|
|
CHECK_JSON=$(cat <<EOF
|
|
{
|
|
"Name": "tcp-${SERVICE_PORT}",
|
|
"TCP": "${SERVICE_ADDR}:${SERVICE_PORT}",
|
|
"Interval": "${CHECK_INTERVAL}",
|
|
"Timeout": "${CHECK_TIMEOUT}",
|
|
"DeregisterCriticalServiceAfter": "${DEREG_AFTER}"
|
|
}
|
|
EOF
|
|
)
|
|
fi
|
|
|
|
# 注册 payload
|
|
cat > /tmp/service.json <<JSON
|
|
{
|
|
"Name": "${SERVICE_NAME}",
|
|
"ID": "${SERVICE_ID}",
|
|
"Address": "${SERVICE_ADDR}",
|
|
"Port": ${SERVICE_PORT},
|
|
"Tags": ${TAGS_JSON},
|
|
"Checks": [ ${CHECK_JSON} ]
|
|
}
|
|
JSON
|
|
|
|
echo "[registrar] registering ${SERVICE_ID} -> ${SERVICE_ADDR}:${SERVICE_PORT} ..."
|
|
curl -fsS -X PUT -d @/tmp/service.json "${CONSUL_HTTP_ADDR}/v1/agent/service/register"
|
|
echo "[registrar] done."
|
|
# 阻塞防退出（可选）
|
|
tail -f /dev/null
|