From 4cb160b88c55566b261e17d9effef7df94c72f59 Mon Sep 17 00:00:00 2001
From: fyah <727475183@qq.com>
Date: Sat, 9 May 2026 13:36:12 +0800
Subject: [PATCH] feat: update OTA config to use stream-gateway (port 9000) and
 add build script

---
 config/edge.yaml             |   2 +-
 scripts/build_ota_package.sh | 217 +++++++++++++++++++++++++++++++++++
 2 files changed, 218 insertions(+), 1 deletion(-)
 create mode 100755 scripts/build_ota_package.sh

diff --git a/config/edge.yaml b/config/edge.yaml
index 4b267bf..be01f4f 100644
--- a/config/edge.yaml
+++ b/config/edge.yaml
@@ -15,5 +15,5 @@ infer_fps: 2
 infer_workers: 3
 conf_threshold: 0.2
 dedup_window_sec: 30
-ota_url: http://101.36.73.102:8087
+ota_url: http://101.36.73.102:9000
 version: 1.0.0
diff --git a/scripts/build_ota_package.sh b/scripts/build_ota_package.sh
new file mode 100755
index 0000000..ba3e1d8
--- /dev/null
+++ b/scripts/build_ota_package.sh
@@ -0,0 +1,217 @@
+#!/usr/bin/env bash
+# =============================================================================
+# build_ota_package.sh - OTA 升级包构建脚本
+# 
+# 用法:
+#   ./build_ota_package.sh -v 1.2.0 -n "修复内存泄漏问题"
+#   ./build_ota_package.sh -v 1.2.0 -n "新增模型支持" --upload
+#
+# 功能:
+#   1. 编译 Go 边缘代理二进制文件 (linux/arm64)
+#   2. 打包为 tar.gz 格式
+#   3. 计算 SHA256 校验和
+#   4. 可选: 自动发布到云端 OTA 服务
+# =============================================================================
+
+set -euo pipefail
+
+# ── 默认配置 ─────────────────────────────────────────────────────────────────
+VERSION=""
+RELEASE_NOTES=""
+UPLOAD=false
+OTA_SERVER_URL="http://127.0.0.1:9000"
+OTA_ADMIN_TOKEN=""
+
+# ── 项目路径 ─────────────────────────────────────────────────────────────────
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+EDGE_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
+BUILD_DIR="${EDGE_DIR}/build/ota"
+PACKAGE_DIR="${BUILD_DIR}/package"
+
+# ── 参数解析 ─────────────────────────────────────────────────────────────────
+usage() {
+    echo "用法: $0 -v <版本号> [-n \"发布说明\"] [--upload] [--url <OTA服务器URL>]"
+    echo ""
+    echo "选项:"
+    echo "  -v    版本号 (必填, 格式: X.Y.Z 或 X.Y.Z-beta)"
+    echo "  -n    发布说明 (可选)"
+    echo "  --upload    构建后自动发布到云端 OTA 服务"
+    echo "  --url   OTA 服务器 URL (默认: http://127.0.0.1:9000)"
+    echo "  --token 管理 API Token (可选)"
+    echo ""
+    echo "示例:"
+    echo "  $0 -v 1.2.0 -n \"修复内存泄漏\""
+    echo "  $0 -v 1.2.0 -n \"新增安全帽检测\" --upload --url http://101.36.73.102:9000"
+    exit 1
+}
+
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        -v)
+            VERSION="$2"
+            shift 2
+            ;;
+        -n)
+            RELEASE_NOTES="$2"
+            shift 2
+            ;;
+        --upload)
+            UPLOAD=true
+            shift
+            ;;
+        --url)
+            OTA_SERVER_URL="$2"
+            shift 2
+            ;;
+        --token)
+            OTA_ADMIN_TOKEN="$2"
+            shift 2
+            ;;
+        -h|--help)
+            usage
+            ;;
+        *)
+            echo "未知参数: $1"
+            usage
+            ;;
+    esac
+done
+
+# ── 参数验证 ─────────────────────────────────────────────────────────────────
+if [[ -z "$VERSION" ]]; then
+    echo "错误: 版本号是必填参数 (-v)"
+    usage
+fi
+
+if ! echo "$VERSION" | grep -qE '^[0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9.]+)?$'; then
+    echo "错误: 版本号格式不正确，请使用 X.Y.Z 或 X.Y.Z-beta 格式"
+    exit 1
+fi
+
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "  构建 OTA 升级包"
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "  版本:       $VERSION"
+echo "  发布说明:   ${RELEASE_NOTES:-无}"
+echo "  边缘项目:   ${EDGE_DIR}"
+echo "  构建目录:   ${BUILD_DIR}"
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+
+# ── 清理旧构建 ──────────────────────────────────────────────────────────────
+rm -rf "${BUILD_DIR}"
+mkdir -p "${PACKAGE_DIR}"
+
+# ── 编译 Go 二进制 ──────────────────────────────────────────────────────────
+echo ""
+echo ">>> 准备二进制文件..."
+
+if command -v go &>/dev/null && [[ -f "${EDGE_DIR}/cmd/edge-agent/main.go" ]]; then
+    echo "    检测到 Go 环境，编译二进制 (linux/arm64)..."
+    cd "${EDGE_DIR}"
+    
+    CGO_ENABLED=0 GOOS=linux GOARCH=arm64 \
+        go build -ldflags "-s -w -X main.Version=${VERSION}" \
+        -o "${PACKAGE_DIR}/edge-agent" \
+        ./cmd/edge-agent
+    
+    BINARY_SIZE=$(stat -c%s "${PACKAGE_DIR}/edge-agent")
+    echo "    编译成功: $(numfmt --to=iec "${BINARY_SIZE}")"
+elif [[ -f "${EDGE_DIR}/edge-agent" ]]; then
+    echo "    使用预编译二进制: ${EDGE_DIR}/edge-agent"
+    cp "${EDGE_DIR}/edge-agent" "${PACKAGE_DIR}/edge-agent"
+    BINARY_SIZE=$(stat -c%s "${PACKAGE_DIR}/edge-agent")
+elif [[ -f "${EDGE_DIR}/build/edge-agent" ]]; then
+    echo "    使用预编译二进制: ${EDGE_DIR}/build/edge-agent"
+    cp "${EDGE_DIR}/build/edge-agent" "${PACKAGE_DIR}/edge-agent"
+    BINARY_SIZE=$(stat -c%s "${PACKAGE_DIR}/edge-agent")
+else
+    echo "错误: 未找到 Go 编译器或预编译二进制文件"
+    echo "  解决方案:"
+    echo "  1. 安装 Go: sudo snap install go --classic"
+    echo "  2. 或预先编译二进制: cd ${EDGE_DIR} && CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -o build/edge-agent ./cmd/edge-agent"
+    exit 1
+fi
+
+# ── 设置文件权限 ─────────────────────────────────────────────────────────────
+chmod +x "${PACKAGE_DIR}/edge-agent"
+
+# ── 打包 ─────────────────────────────────────────────────────────────────────
+PACKAGE_FILE="${BUILD_DIR}/edge-agent-${VERSION}.tar.gz"
+echo ""
+echo ">>> 创建升级包..."
+
+cd "${PACKAGE_DIR}"
+tar czf "${PACKAGE_FILE}" edge-agent
+
+PACKAGE_SIZE=$(stat -c%s "${PACKAGE_FILE}")
+echo "    打包完成: ${PACKAGE_FILE} ($(numfmt --to=iec "${PACKAGE_SIZE}"))"
+
+# ── 计算 SHA256 ──────────────────────────────────────────────────────────────
+SHA256=$(sha256sum "${PACKAGE_FILE}" | awk '{print $1}')
+echo "    SHA256:     ${SHA256}"
+
+# ── 生成版本信息文件 ─────────────────────────────────────────────────────────
+cat > "${BUILD_DIR}/version-info.json" << EOF
+{
+    "version": "${VERSION}",
+    "sha256": "${SHA256}",
+    "file_size": ${PACKAGE_SIZE},
+    "release_notes": "${RELEASE_NOTES}",
+    "build_time": "$(date -u +%Y-%m-%dT%H:%M:%SZ)",
+    "arch": "arm64",
+    "binary_size": ${BINARY_SIZE}
+}
+EOF
+
+echo "    版本信息:   ${BUILD_DIR}/version-info.json"
+
+# ── 自动上传 (可选) ─────────────────────────────────────────────────────────
+if [[ "$UPLOAD" == true ]]; then
+    echo ""
+    echo ">>> 发布到云端 OTA 服务: ${OTA_SERVER_URL}..."
+    
+    CURL_ARGS=(
+        "-s"
+        "-X" "POST"
+        "${OTA_SERVER_URL}/api/v1/edge/update/admin/publish"
+        "-F" "version=${VERSION}"
+        "-F" "release_notes=${RELEASE_NOTES}"
+        "-F" "maintenance_only=true"
+        "-F" "target_arch=arm64"
+        "-F" "package=@${PACKAGE_FILE}"
+    )
+    
+    if [[ -n "$OTA_ADMIN_TOKEN" ]]; then
+        CURL_ARGS+=("-H" "Authorization: Bearer ${OTA_ADMIN_TOKEN}")
+    fi
+    
+    RESPONSE=$("${CURL_ARGS[@]}")
+    echo "    服务器响应: ${RESPONSE}"
+    
+    if echo "$RESPONSE" | grep -q '"ok": true'; then
+        echo "    发布成功!"
+    else
+        echo "    警告: 服务器返回非成功响应"
+    fi
+fi
+
+# ── 完成 ─────────────────────────────────────────────────────────────────────
+echo ""
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "  构建完成!"
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+echo "  升级包路径:  ${PACKAGE_FILE}"
+echo "  版本信息:    ${BUILD_DIR}/version-info.json"
+echo ""
+echo "  手动发布到 OTA 服务:"
+echo "    curl -X POST ${OTA_SERVER_URL}/api/v1/edge/update/admin/publish \\"
+echo "      -F \"version=${VERSION}\" \\"
+echo "      -F \"release_notes=${RELEASE_NOTES}\" \\"
+echo "      -F \"package=@${PACKAGE_FILE}\""
+echo ""
+echo "  边缘侧升级命令 (在设备上执行):"
+echo "    curl -o /tmp/update.tar.gz ${OTA_SERVER_URL}/api/v1/edge/update/packages/${VERSION}"
+echo "    sudo systemctl stop edge-agent"
+echo "    sudo tar xzf /tmp/update.tar.gz -C /opt/tianyan-edge/bin/"
+echo "    sudo systemctl start edge-agent"
+echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"