122 lines
3.5 KiB
Shell
122 lines
3.5 KiB
Shell
|
#!/bin/bash
|
|||
|
|
# ═══════════════════════════════════════════════════════════
|
|||
|
|
# 光湖语言系统 · 模块注册脚本
|
|||
|
|
# 服务器上任一模块启动时调用 · 自动写入注册文件
|
|||
|
|
#
|
|||
|
|
# 用法:
|
|||
|
|
# module-register.sh --server GZ006 --func GTW --name "Gatekeeper" --port 8922
|
|||
|
|
# module-register.sh --stop GZ006-GTW-01
|
|||
|
|
# ═══════════════════════════════════════════════════════════
|
|||
|
|
|
|||
|
|
REGISTRY_DIR="/opt/zhuyuan/registry/modules.d"
|
|||
|
|
mkdir -p "$REGISTRY_DIR"
|
|||
|
|
|
|||
|
|
# ── 停止模式 ──
|
|||
|
|
if [ "$1" = "--stop" ]; then
|
|||
|
|
CODE="$2"
|
|||
|
|
if [ -z "$CODE" ]; then
|
|||
|
|
echo "用法: $0 --stop {CODE}" >&2
|
|||
|
|
exit 1
|
|||
|
|
fi
|
|||
|
|
REG_FILE="$REGISTRY_DIR/MOD-${CODE}.json"
|
|||
|
|
if [ -f "$REG_FILE" ]; then
|
|||
|
|
python3 -c "
|
|||
|
|
import json, sys
|
|||
|
|
try:
|
|||
|
|
with open('$REG_FILE', 'r') as f:
|
|||
|
|
data = json.load(f)
|
|||
|
|
data['status'] = 'stopped'
|
|||
|
|
data['stopped_at'] = '$(date -u +%Y-%m-%dT%H:%M:%SZ)'
|
|||
|
|
with open('$REG_FILE', 'w') as f:
|
|||
|
|
json.dump(data, f, indent=2)
|
|||
|
|
print(f'⊢ 模块 {data[\"code\"]} 已标记为已停止')
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f'⚠ 更新注册文件失败: {e}', file=sys.stderr)
|
|||
|
|
"
|
|||
|
|
fi
|
|||
|
|
exit 0
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
# ── 注册模式 ──
|
|||
|
|
SERVER=""
|
|||
|
|
FUNC=""
|
|||
|
|
NAME=""
|
|||
|
|
PORT=""
|
|||
|
|
PROTOCOL="tcp"
|
|||
|
|
DESCRIPTION=""
|
|||
|
|
PID="$$"
|
|||
|
|
|
|||
|
|
while [ $# -gt 0 ]; do
|
|||
|
|
case "$1" in
|
|||
|
|
--server) SERVER="$2"; shift 2 ;;
|
|||
|
|
--func) FUNC="$2"; shift 2 ;;
|
|||
|
|
--name) NAME="$2"; shift 2 ;;
|
|||
|
|
--port) PORT="$2"; shift 2 ;;
|
|||
|
|
--protocol) PROTOCOL="$2"; shift 2 ;;
|
|||
|
|
--description) DESCRIPTION="$2"; shift 2 ;;
|
|||
|
|
--pid) PID="$2"; shift 2 ;;
|
|||
|
|
*) shift ;;
|
|||
|
|
esac
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
if [ -z "$SERVER" ] || [ -z "$FUNC" ] || [ -z "$NAME" ]; then
|
|||
|
|
echo "❌ 缺少必要参数: --server --func --name" >&2
|
|||
|
|
exit 1
|
|||
|
|
fi
|
|||
|
|
|
|||
|
|
# 计算序号（同一服务器+功能下，取最大序号+1）
|
|||
|
|
SEQ=1
|
|||
|
|
for f in "$REGISTRY_DIR"/MOD-${SERVER}-${FUNC}-*.json; do
|
|||
|
|
if [ -f "$f" ]; then
|
|||
|
|
existing_seq=$(basename "$f" .json | grep -oE '[0-9]+$')
|
|||
|
|
if [ -n "$existing_seq" ] && [ "$existing_seq" -ge "$SEQ" ]; then
|
|||
|
|
SEQ=$((existing_seq + 1))
|
|||
|
|
fi
|
|||
|
|
fi
|
|||
|
|
done
|
|||
|
|
|
|||
|
|
# 格式化序号为两位
|
|||
|
|
SEQ_FMT=$(printf "%02d" "$SEQ")
|
|||
|
|
CODE="${SERVER}-${FUNC}-${SEQ_FMT}"
|
|||
|
|
REG_FILE="$REGISTRY_DIR/MOD-${CODE}.json"
|
|||
|
|
TIMESTAMP=$(date -u +%Y-%m-%dT%H:%M:%SZ)
|
|||
|
|
|
|||
|
|
python3 -c "
|
|||
|
|
import json
|
|||
|
|
data = {
|
|||
|
|
'code': '${CODE}',
|
|||
|
|
'server': '${SERVER}',
|
|||
|
|
'function': '${FUNC}',
|
|||
|
|
'name': '${NAME}',
|
|||
|
|
'port': ${PORT:-0},
|
|||
|
|
'protocol': '${PROTOCOL}',
|
|||
|
|
'description': '${DESCRIPTION}',
|
|||
|
|
'pid': ${PID},
|
|||
|
|
'started_at': '${TIMESTAMP}',
|
|||
|
|
'registered_at': '${TIMESTAMP}',
|
|||
|
|
'status': 'running'
|
|||
|
|
}
|
|||
|
|
with open('${REG_FILE}', 'w') as f:
|
|||
|
|
json.dump(data, f, indent=2, ensure_ascii=False)
|
|||
|
|
print(f'✅ 模块已注册: {data[\"code\"]}')
|
|||
|
|
print(f' 名称: ${NAME}')
|
|||
|
|
print(f' 文件: MOD-${CODE}.json')
|
|||
|
|
" 2>/dev/null || {
|
|||
|
|
# Python3 不可用时的降级方案（纯 shell 写 JSON）
|
|||
|
|
cat > "$REG_FILE" << JSONEOF
|
|||
|
|
{
|
|||
|
|
"code": "${CODE}",
|
|||
|
|
"server": "${SERVER}",
|
|||
|
|
"function": "${FUNC}",
|
|||
|
|
"name": "${NAME}",
|
|||
|
|
"port": ${PORT:-0},
|
|||
|
|
"protocol": "${PROTOCOL}",
|
|||
|
|
"description": "${DESCRIPTION}",
|
|||
|
|
"pid": ${PID},
|
|||
|
|
"started_at": "${TIMESTAMP}",
|
|||
|
|
"registered_at": "${TIMESTAMP}",
|
|||
|
|
"status": "running"
|
|||
|
|
}
|
|||
|
|
JSONEOF
|
|||
|
|
echo "✅ 模块已注册: ${CODE}"
|
|||
|
|
}
|