mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 01:07:35 +00:00
fix(update): 重启交还 guard、OTA 支持 release/包两种来源、修三处真 bug
## 一、重启不再由 update 自己 systemctl（用户口径：应该通知 guard）
原 internal/update 直接 `systemctl restart homeagent`。三个问题：
① 杀掉整个 cgroup（unit 里 KillMode=control-group）⇒ **guard 跟着死**，
而 guard 存在的意义就是不让自己死。
② 丢失 exit 42 语义：guard 的 failures/failbackRound/failbackDone 全丢。
③ 自毁竞态：发命令的进程正是被重启杀掉的进程 ⇒ CombinedOutput 可能报错，
**一次成功的升级被误判失败而回滚**。
现在 internal/update 只到"产物就绪"：预检 → 备份 → 写 pending → 原子替换。
重启由调用方走**既有**通道（restartCh → exit 42 → guard 用新二进制重拉）：
· webui h.kernelCtl.RequestKernelRestart（本来就有，且是白名单内）
· pluginmgr p.sdk.KernelControl().RequestKernelRestart（见下面第三条）
· CLI 外部管理命令，restart unit 后按**版本自报**核对是否真生效
## 二、OTA 支持两种来源（release 自动更新 / 指定安装包）
新增 internal/update/source.go：
Source{Kind: FromPackage, Path} → 直接用本地包
Source{Kind: FromRelease} → 查最新 Release → 下载 → 校验 digest
ParseSource(pkg, release) 统一判空规则（pkg 优先），避免三端各写一套。
两条来源解析后进**同一条** Apply，不存在两份升级实现。
## 三、修三处已提交代码里的真 bug
① **HEAD 编译不过**：d17e261 提交了 kernelmod/update/mod.go（引用 ks.Update），
但 kernelboot/kernelstate.go 的字段改动从未提交。干净 worktree 实测
`ks.Update undefined` —— 推上去的仓库是坏的。本次补上。
② **kernelDataDir 算错一级**：Dir(Dir("/home/newqqagent/plugins")) = "/home"，
而第 0 层 update 模块从 ks.DataDir=/home/newqqagent 读 pending
⇒ **agent 发起的升级一旦启动失败就永不回滚**（静默失效）。改为 Dir 一次。
③ **pluginmgr 的 KernelControl 恒为 nil**：它不在 kernelControlAllowed 里，
于是 `if kc := p.sdk.KernelControl(); kc != nil` 永不成立 ⇒
换完二进制**从不重启** ⇒ pending 留着 ⇒ 下次重启被判"未确认"而回滚。
修法**不是**把 pluginmgr 加进完整白名单 —— 那等于让 LLM 能关机整个服务
（kernel_control_gate_test 明确拦这个）。而是拆出**仅重启**授权：
kernelControlAllowed = {webui} 完整（含关机）
kernelRestartAllowed = {pluginmgr} 仅重启（shutdown 直接报错）
两者的危险程度差一个量级：重启几秒后 guard 自己拉回来，关机不会自己回来。
## 四、EventBus：插件请求更新不必 import internal/update
新增 events.EventUpdateRequest，pluginmgr 订阅并受理：
scope=kernel → kernelUpdateApply；scope=plugin → pluginUpdate(apply)。
它在 internal/plugins/webui 的 SSE 守卫里显式登记为**内核内部控制事件**
（不该推给浏览器：前端要的是 update_progress/update_result，不是"谁发了请求"）。
## 五、顺手修掉的静默失败
· webui /kernel/update 原实现：**先**写 202 Accepted，**再**解析 body，
之后每个错误都直接 `return` —— 用户看到"已受理"然后什么都没发生。
改为同步 Apply（失败可如实报 HTTP 状态），成功后才请求重启。
· pluginmgr 拿不到控制通道时不再假装成功，而是回报 applied=true + 明确原因
（否则用户以为已升级，进程却还在跑旧镜像）。
· update 事件 publishResult 的 errMsg 原先是死代码（`_ = errMsg`），
按 payload 约定写回 error 字段。
This commit is contained in:
JianFeeeee
@ -20,8 +20,12 @@
|
||||
// ## 两阶段（用户口径）
|
||||
//
|
||||
// apply=false（默认）→ 只查 GitHub Release 与本地版本比较，不改动任何东西
|
||||
// apply=true → 下载/解包 → 校验 → 备份 → 原子替换 → 重启 → 版本复核
|
||||
// → 失败自动回滚（DoUpgrade 内部完成）
|
||||
// apply=true → 解包 → 校验 → 备份 → 原子替换（**不重启**）
|
||||
// → 请求 guard 重拉 → 新内核启动时由 update 模块确认或回滚
|
||||
//
|
||||
// ★ 重启为什么不在 internal/update 里做：worker 是被守护方，只有 guard 知道
|
||||
// 这次退出该不该重拉（exit 42 → 重拉且不计失败轮次）。详见
|
||||
// cmd/homed/kernel_control.go。
|
||||
|
||||
package pluginmgr
|
||||
|
||||
@ -39,33 +43,37 @@ func (p *Plugin) kernelUpdateCheck() (interface{}, error) {
|
||||
cur := currentKernelVersion(p)
|
||||
info := update.CheckRelease(cur)
|
||||
return map[string]interface{}{
|
||||
"apply": false,
|
||||
"current": info.Current,
|
||||
"latest": info.Latest,
|
||||
"apply": false,
|
||||
"current": info.Current,
|
||||
"latest": info.Latest,
|
||||
"has_update": info.HasUpdate,
|
||||
"asset": info.Asset,
|
||||
"release": info.LatestURL,
|
||||
"reason": info.Reason,
|
||||
"hint": "确认后传 apply=true 执行升级（会替换二进制并重启内核，连接断开数秒）",
|
||||
"asset": info.Asset,
|
||||
"release": info.LatestURL,
|
||||
"reason": info.Reason,
|
||||
"hint": "确认后传 apply=true 执行升级（会替换二进制并重启内核，连接断开数秒）",
|
||||
}, nil
|
||||
}
|
||||
|
||||
// kernelUpdateApply 是 kernel_update 的执行半边（apply=true）。
|
||||
//
|
||||
// 只接受**本机已有的包**（pkg 参数）：从 URL 现场下载需要处理
|
||||
// 校验/断点/重试，那是 CLI 的职责（homed-upgrade），agent 不应在
|
||||
// 推理过程中发起长下载（会占满工具调用超时）。
|
||||
func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) {
|
||||
if pkgPath == "" {
|
||||
return map[string]interface{}{
|
||||
"error": "apply=true 需要 pkg（本机已存在的安装包路径；下载请跑 homed-upgrade -pkg <url>）",
|
||||
}, nil
|
||||
}
|
||||
st, err := os.Stat(pkgPath)
|
||||
if err != nil || st.IsDir() {
|
||||
return map[string]interface{}{
|
||||
"error": fmt.Sprintf("pkg 无效：%s", pkgPath),
|
||||
}, nil
|
||||
// 来源二选一（update.ParseSource）：
|
||||
//
|
||||
// · pkg 非空 → 用本机已有的安装包
|
||||
// · release=true → 从仓库最新 GitHub Release 自动下载
|
||||
//
|
||||
// ★ 顺序（此前是反的，会出真事故）：
|
||||
//
|
||||
// 先 Apply（备份→pending→替换），**再**请求重启。
|
||||
// 反过来（先请求重启）会在新二进制还没就位时就重拉，
|
||||
// 或重启恰好发生在替换中途 ⇒ 跑起来的是半新半旧的文件。
|
||||
//
|
||||
// ★ 重启走既有的 p.sdk.KernelControl()（restartCh → exit 42 → guard 重拉）。
|
||||
// 它依赖 kernelControlAllowed 白名单 —— pluginmgr 已加入（否则该接口恒为 nil，
|
||||
// 以前写在那里等于静默不重启）。
|
||||
func (p *Plugin) kernelUpdateApply(pkgPath string, release bool) (interface{}, error) {
|
||||
src, err := update.ParseSource(pkgPath, release)
|
||||
if err != nil {
|
||||
return map[string]interface{}{"error": err.Error()}, nil
|
||||
}
|
||||
|
||||
tmp, err := os.MkdirTemp("", "kernel-upgrade-*")
|
||||
@ -74,39 +82,62 @@ func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) {
|
||||
}
|
||||
defer os.RemoveAll(tmp)
|
||||
|
||||
root, err := update.ExtractPkg(pkgPath, tmp)
|
||||
root, rel, err := src.Resolve(tmp)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("解包: %w", err)
|
||||
return map[string]interface{}{"error": "取包失败: " + err.Error()}, nil
|
||||
}
|
||||
newBin := update.FindFile(root, "homed")
|
||||
newDG := update.FindFile(root, "depgraph.bin")
|
||||
if newBin == "" {
|
||||
return nil, fmt.Errorf("包内找不到 homed")
|
||||
return map[string]interface{}{"error": "包内找不到 homed"}, nil
|
||||
}
|
||||
|
||||
// ★ 升级会让本进程消失：先把"已受理"回给模型，再执行替换。
|
||||
//
|
||||
// 否则模型看到的是工具调用超时/断线，无法区分"升级成功"和"工具坏了"。
|
||||
// 返回后由本 goroutine 继续跑 DoUpgrade（替换+重启+回滚）。
|
||||
if kc := p.sdk.KernelControl(); kc != nil {
|
||||
_ = kc.RequestKernelRestart("kernel_update apply")
|
||||
// 走事件总线：升级的每一步都让 WebUI 看得到。
|
||||
var bus *events.Bus
|
||||
if p.sdk != nil {
|
||||
bus = p.sdk.EventBus()
|
||||
}
|
||||
go func() {
|
||||
// 走事件总线：升级的每一步（含重启前的最后几步）都让 WebUI 看得到。
|
||||
// 顺序上 publish 紧贴动作之前 —— 进程消失后事件就发不出去了。
|
||||
bus := (*events.Bus)(nil)
|
||||
if p.sdk != nil {
|
||||
bus = p.sdk.EventBus()
|
||||
|
||||
out, err := update.Apply(root, kernelBinPath(p), newDG, kernelDataDir(p), bus)
|
||||
if err != nil {
|
||||
return map[string]interface{}{"error": err.Error()}, nil
|
||||
}
|
||||
|
||||
// 产物已就位 ⇒ 走**既有**重启通道（KernelControlAPI → restartCh → exit 42
|
||||
// → guard 用新二进制重拉）。不自己 systemctl，也不另开通道。
|
||||
//
|
||||
// ★ 顺序关键：Apply 完成**之后**才请求重启。反过来会在新二进制还没
|
||||
// 就位时重拉，或重启恰好落在替换中途。
|
||||
if kc := p.sdk.KernelControl(); kc != nil {
|
||||
if rerr := kc.RequestKernelRestart("kernel_update apply"); rerr != nil {
|
||||
return map[string]interface{}{
|
||||
"applied": true,
|
||||
"error": "新内核已就位，但请求重启失败: " + rerr.Error(),
|
||||
"hint": "手动重启服务即可；pending 标记会让首次启动自动确认或回滚",
|
||||
}, nil
|
||||
}
|
||||
if err := update.DoUpgradeBus(root, kernelBinPath(p), newDG, p.pluginDir, kernelDataDir(p), bus); err != nil {
|
||||
// DoUpgrade 内部失败已自行回滚；这里只留一条诊断日志
|
||||
fmt.Fprintf(os.Stderr, "[pluginmgr] kernel_update upgrade failed (rolled back): %v\n", err)
|
||||
}
|
||||
}()
|
||||
return map[string]interface{}{
|
||||
"accepted": true,
|
||||
"message": "升级已受理：正在替换二进制并重启。完成后用 kernel_update（apply=false）确认新版本。",
|
||||
}, nil
|
||||
} else {
|
||||
// 拿不到通道时**如实报告**，不要静默当成功：否则用户以为已升级，
|
||||
// 而进程还在跑旧镜像；pending 标记会在下次重启时被判"未确认"→ 回滚。
|
||||
return map[string]interface{}{
|
||||
"applied": true,
|
||||
"error": "新内核已就位，但本插件没有被授予内核控制通道（kernelControlAllowed），无法请求重启",
|
||||
"hint": "手动重启服务完成升级；或改用 WebUI 的 /api/v1/kernel/update",
|
||||
}, nil
|
||||
}
|
||||
|
||||
res := map[string]interface{}{
|
||||
"applied": true,
|
||||
"from": out.From.Version,
|
||||
"to": out.To.Version,
|
||||
"backup": out.BackupDir,
|
||||
"message": "新内核已就位，正在由 guard 重拉。重启后若新版本无法启动，第 0 层 update 模块会自动回滚。",
|
||||
}
|
||||
if rel != nil {
|
||||
res["release"] = rel.Latest
|
||||
res["asset"] = rel.Asset
|
||||
}
|
||||
return res, nil
|
||||
}
|
||||
|
||||
// kernelDataDir 推导数据目录（升级确认标记的落点）。
|
||||
@ -114,8 +145,14 @@ func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) {
|
||||
// 从 pluginDir 反推：<dataDir>/plugins ⇒ 上一级即 dataDir。
|
||||
// 与内核 --data 的值一致，update 模块（第 0 层）才能读到同一个标记。
|
||||
func kernelDataDir(p *Plugin) string {
|
||||
// <dataDir>/plugins ⇒ 上一级即 dataDir。
|
||||
//
|
||||
// ★ 只 Dir 一次。此前写成 Dir(Dir(...))，把 /home/newqqagent/plugins
|
||||
// 算成 /home —— pending 标记落到 /home/kernel-upgrade-pending.json，
|
||||
// 而第 0 层 update 模块从 ks.DataDir 读 ⇒ **永远读不到标记** ⇒
|
||||
// agent 发起的升级一旦启动失败就**不会回滚**（静默失效）。
|
||||
if p.pluginDir != "" {
|
||||
return filepath.Dir(filepath.Dir(p.pluginDir))
|
||||
return filepath.Dir(p.pluginDir)
|
||||
}
|
||||
return "/home/newqqagent"
|
||||
}
|
||||
@ -142,3 +179,47 @@ func kernelBinPath(p *Plugin) string {
|
||||
}
|
||||
return "homed"
|
||||
}
|
||||
|
||||
// onUpdateRequest 处理来自其他插件的更新请求（EventBus）。
|
||||
//
|
||||
// 请求方不必知道"怎么更新"，只需发一个事件：
|
||||
//
|
||||
// scope: "kernel" | "plugin"
|
||||
// name: 插件名（scope=plugin 时必填）
|
||||
// pkg: 本机安装包路径（可选）
|
||||
// release: bool，从仓库最新 Release 自动取包（可选）
|
||||
//
|
||||
// 为什么必须**异步**：升级会替换二进制并（对内核）请求重启，
|
||||
// 本进程随后消失。事件处理器是同步投递的（见 events.Bus.Publish），
|
||||
// 在处理器里同步做完会让发布方卡住直到进程被杀。
|
||||
func (p *Plugin) onUpdateRequest(evt *events.Event) {
|
||||
if evt == nil || evt.Payload == nil {
|
||||
return
|
||||
}
|
||||
scope, _ := evt.Payload["scope"].(string)
|
||||
name, _ := evt.Payload["name"].(string)
|
||||
pkgPath, _ := evt.Payload["pkg"].(string)
|
||||
release, _ := evt.Payload["release"].(bool)
|
||||
|
||||
go func() {
|
||||
var err error
|
||||
switch scope {
|
||||
case "kernel":
|
||||
_, err = p.kernelUpdateApply(pkgPath, release)
|
||||
case "plugin":
|
||||
if name == "" {
|
||||
err = fmt.Errorf("plugin 更新请求缺 name")
|
||||
break
|
||||
}
|
||||
// 复用既有 plugin_update 的 apply 通路（范围即该插件名）
|
||||
_, err = p.pluginUpdate(true, name)
|
||||
default:
|
||||
err = fmt.Errorf("未知更新范围 %q（来源 %s）", scope, evt.Source)
|
||||
}
|
||||
if err != nil {
|
||||
// 事件是单向的，没有返回值可取 ⇒ 失败必须留日志，
|
||||
// 否则请求方看到的就是"发了、没反应"。
|
||||
fmt.Fprintf(os.Stderr, "[pluginmgr] update_request (%s/%s) failed: %v\n", scope, name, err)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
@ -70,27 +70,31 @@ func TestKernelUpdate_CheckShape(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
// 判据 4：kernel_update apply 缺 pkg ⇒ 明确报错并指向 CLI。
|
||||
func TestKernelUpdate_ApplyRequiresPkg(t *testing.T) {
|
||||
// 判据 4：kernel_update apply 没给来源 ⇒ 明确报错（不许"猜一个默认动作"）。
|
||||
//
|
||||
// 旧判据要求错误里出现 homed-upgrade —— 那是"agent 不该在推理中发起长下载"
|
||||
// 时期的限制。现在支持 release=true 自动取包（下载放在内核侧统一实现），
|
||||
// 所以判据改为：必须说清"两种来源二选一"。
|
||||
func TestKernelUpdate_ApplyRequiresSource(t *testing.T) {
|
||||
p := &Plugin{pluginDir: t.TempDir()}
|
||||
out, err := p.kernelUpdateApply("")
|
||||
out, err := p.kernelUpdateApply("", false)
|
||||
if err != nil {
|
||||
t.Fatalf("应返回可呈现的错误: %v", err)
|
||||
}
|
||||
m := out.(map[string]interface{})
|
||||
if !strings.Contains(fmtSprint(m["error"]), "pkg") {
|
||||
t.Errorf("错误应说明需要 pkg，实际: %v", m["error"])
|
||||
msg := fmtSprint(m["error"])
|
||||
if !strings.Contains(msg, "pkg") {
|
||||
t.Errorf("错误应说明需要 pkg，实际: %v", msg)
|
||||
}
|
||||
// 并应提示下载走 CLI（agent 不该在推理中发起长下载）
|
||||
if !strings.Contains(fmtSprint(m["error"]), "homed-upgrade") {
|
||||
t.Errorf("错误应指向 homed-upgrade，实际: %v", m["error"])
|
||||
if !strings.Contains(msg, "Release") && !strings.Contains(msg, "release") {
|
||||
t.Errorf("错误应说明可从 Release 自动更新，实际: %v", msg)
|
||||
}
|
||||
}
|
||||
|
||||
// 判据 5：kernel_update apply 的 pkg 是目录/不存在 ⇒ 拒绝。
|
||||
func TestKernelUpdate_ApplyRejectsBadPkg(t *testing.T) {
|
||||
p := &Plugin{pluginDir: t.TempDir()}
|
||||
out, _ := p.kernelUpdateApply(t.TempDir()) // 目录不是包
|
||||
out, _ := p.kernelUpdateApply(t.TempDir(), false) // 目录不是包
|
||||
m := out.(map[string]interface{})
|
||||
if _, hasErr := m["error"]; !hasErr {
|
||||
t.Errorf("目录不应被当作安装包，实际: %v", m)
|
||||
|
||||
@ -21,9 +21,10 @@ import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/JianFeeeee/HologramAgent/internal/events"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/plugin"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/update"
|
||||
sdk "github.com/JianFeeeee/HologramAgent/internal/sdk"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/update"
|
||||
)
|
||||
|
||||
// platformBinary 按当前 OS/ARCH 选择正确的插件二进制文件名。
|
||||
@ -97,6 +98,9 @@ type Plugin struct {
|
||||
httpAddr string // 本实例的监听地址（默认 defaultHTTPAddr；来自 settings）
|
||||
sdk *sdk.PluginSDK
|
||||
pluginDir string
|
||||
|
||||
// unsubUpdateReq 注销 update_request 订阅（Stop 时调用）。
|
||||
unsubUpdateReq func()
|
||||
// maxPackageBytes 是单个插件包 **压缩后** 的上限（0 = 不限）。
|
||||
//
|
||||
// 为什么是配置项而不是常量：内嵌运行时（node+pi 之类）就要 ~100MB，
|
||||
@ -495,12 +499,26 @@ func (p *Plugin) registerTools(s *sdk.PluginSDK) {
|
||||
return p.pluginInfo(name)
|
||||
})
|
||||
|
||||
// ── 订阅：其他插件的更新请求（EventBus，避免它们直接 import internal/update）──
|
||||
//
|
||||
// 为什么用事件而不是让插件调函数：
|
||||
// · 请求方只说"我要更新"，不必知道怎么拉源码/编译/替换/回滚
|
||||
// · 外置插件根本 import 不到内核内部包
|
||||
// · 请求与进度都在总线上，WebUI 能显示"谁请求 → 到哪步 → 结果如何"
|
||||
//
|
||||
// 注意：内核升级的**重启**仍走既有 KernelControlAPI（restartCh → exit 42
|
||||
// → guard 重拉），本订阅只负责"受理请求 + 执行替换"。
|
||||
if bus := s.EventBus(); bus != nil {
|
||||
p.unsubUpdateReq = bus.Subscribe(events.EventUpdateRequest, p.onUpdateRequest)
|
||||
}
|
||||
|
||||
s.RegisterTool("kernel_update", sdk.ToolDef{
|
||||
Name: "kernel_update",
|
||||
Description: "检查/升级**内核**（homed 二进制）。两阶段：默认（apply=false）只查 GitHub Release " +
|
||||
"与本地版本比较，不改动任何东西；看过结果后传 apply=true + pkg（本机已存在的安装包路径）" +
|
||||
"真正升级：校验 → 备份 → 原子替换 → 重启 → 版本复核 → 失败自动回滚。" +
|
||||
"升级会让内核重启，连接断开数秒。从 URL 下载新包请用 CLI 的 homed-upgrade。",
|
||||
"与本地版本比较，不改动任何东西；看过结果后传 apply=true，并给出包来源二选一：" +
|
||||
"pkg=本机安装包路径，或 release=true 从最新 GitHub Release 自动下载。" +
|
||||
"升级流程：校验 → 备份 → 原子替换 → 由 guard 重拉 → 新内核启动时确认，失败自动回滚。" +
|
||||
"升级会让内核重启，连接断开数秒。",
|
||||
Parameters: map[string]interface{}{
|
||||
"type": "object",
|
||||
"properties": map[string]interface{}{
|
||||
@ -510,7 +528,11 @@ func (p *Plugin) registerTools(s *sdk.PluginSDK) {
|
||||
},
|
||||
"pkg": map[string]interface{}{
|
||||
"type": "string",
|
||||
"description": "安装包在本机的路径（.tar.gz）。仅 apply=true 时需要",
|
||||
"description": "安装包在本机的路径（.tar.gz）。与 release 二选一；给了 pkg 就优先用它",
|
||||
},
|
||||
"release": map[string]interface{}{
|
||||
"type": "boolean",
|
||||
"description": "true=从仓库最新 GitHub Release 自动下载安装包。未给 pkg 时可用",
|
||||
},
|
||||
},
|
||||
},
|
||||
@ -519,8 +541,9 @@ func (p *Plugin) registerTools(s *sdk.PluginSDK) {
|
||||
}, func(args map[string]interface{}) (interface{}, error) {
|
||||
apply, _ := args["apply"].(bool)
|
||||
pkgPath, _ := args["pkg"].(string)
|
||||
release, _ := args["release"].(bool)
|
||||
if apply {
|
||||
return p.kernelUpdateApply(pkgPath)
|
||||
return p.kernelUpdateApply(pkgPath, release)
|
||||
}
|
||||
return p.kernelUpdateCheck()
|
||||
})
|
||||
@ -732,7 +755,6 @@ func (p *Plugin) pluginUpdate(apply bool, scope string) (interface{}, error) {
|
||||
}, nil
|
||||
}
|
||||
|
||||
|
||||
// ======== HTTP API ========
|
||||
|
||||
func (p *Plugin) startHTTPServer() {
|
||||
|
||||
@ -156,6 +156,13 @@ func TestEveryPublishedEventHasAnOutwardPath(t *testing.T) {
|
||||
// 插件间握手事件：clawhubadapter → skillmgr 移交 SKILL 条目，
|
||||
// 属内核内部的职责链，前端无需感知。
|
||||
"skill_detected": "插件间握手（clawhubadapter → skillmgr），非对话流程",
|
||||
// update_request 是**内核内部的控制事件**（插件 → pluginmgr）：
|
||||
// "我要更新"，由 pluginmgr 的订阅者受理并执行替换。
|
||||
//
|
||||
// 它不该进 SSE：前端要的是**结果**（update_progress / update_result
|
||||
// 已在 subTypes 里），而不是"谁发了一个请求"。把它推给浏览器只会
|
||||
// 让界面多出一条没有下文的消息。
|
||||
"update_request": "插件间请求更新（pluginmgr 订阅处理），非 UI 通知",
|
||||
// tools_ready：全部插件加载完毕、工具注册齐的信号，供内核侧订阅者
|
||||
// （如 seq 的开机序列）用。它是**内核内部**的就绪信号，
|
||||
// 不是对话流程事件，前端无需感知。
|
||||
|
||||
@ -2,23 +2,36 @@ package webui
|
||||
|
||||
// 内核更新的 HTTP 端点（GET 检查 / POST 升级）。
|
||||
//
|
||||
// ## 与 agent 工具同底层的架构约束
|
||||
// ## 三端共享同一底层（架构约束）
|
||||
//
|
||||
// agent 的 kernel_update 工具、本端点、CLI（homed-upgrade）三者共享
|
||||
// internal/update 的 CheckRelease / DoUpgrade —— 三份实现必然漂移，
|
||||
// 会出现"WebUI 说有新版、agent 说没有"的分裂状态。
|
||||
// internal/update 的 CheckRelease / Apply —— 三份实现必然漂移，会出现
|
||||
// "WebUI 说有新版、agent 说没有"的分裂状态。
|
||||
// 本文件只做 HTTP 形状转换，不含业务逻辑。
|
||||
//
|
||||
// ## 与 kernel/restart 的关系
|
||||
// ## 包来源二选一
|
||||
//
|
||||
// 升级 = 换二进制 + 重启。重启通道（kernelCtl）已存在且经过生产验证，
|
||||
// 升级走同一通道触发重启，不自建第二条。
|
||||
// {"release": true} → 从仓库最新 GitHub Release 自动下载
|
||||
// {"pkg": "/path/to/pkg.tar.gz"} → 用本机已有的安装包（优先）
|
||||
//
|
||||
// ## 升级 = 换产物 + 重启，两件事**分开做**
|
||||
//
|
||||
// Apply（本进程内，同步） ：校验 → 备份 → pending → 原子替换
|
||||
// 重启（交给内核控制通道） ：kernelCtl.RequestKernelRestart → exit 42
|
||||
// → guard 用新二进制重拉（不计失败轮次）
|
||||
//
|
||||
// ★ 为什么 Apply 必须同步返回：替换是**可失败**的动作（空间不足、权限、
|
||||
// 包损坏）。异步的话这些错误全被吞掉，用户只看到"已受理"然后毫无变化。
|
||||
// 而重启可以异步 —— 它让本进程消失，本来就没法回给它结果。
|
||||
//
|
||||
// ★ 为什么不用 systemctl restart：见 cmd/homed/kernel_control.go ——
|
||||
// worker 是被守护方，只有 guard 知道这次退出该不该重拉。
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
|
||||
"github.com/JianFeeeee/HologramAgent/internal/events"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/update"
|
||||
@ -38,45 +51,85 @@ func (h *Handler) handleKernelUpdate(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
// handleKernelUpdateCheck 处理 GET —— 只检查不改动。
|
||||
func (h *Handler) handleKernelUpdateCheck(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
writeJSONError(w, http.StatusMethodNotAllowed, "只接受 GET")
|
||||
return
|
||||
}
|
||||
cur := currentKernelVersion()
|
||||
writeJSON(w, http.StatusOK, update.CheckRelease(cur))
|
||||
writeJSON(w, http.StatusOK, update.CheckRelease(currentKernelVersion()))
|
||||
}
|
||||
|
||||
// handleKernelUpdateApply 处理 POST /api/v1/kernel/update —— 执行升级。
|
||||
//
|
||||
// body: {"pkg": "/path/to/pkg.tar.gz"}（可选；缺省用最近一次检查到的
|
||||
// release 资产 URL 现场下载）。
|
||||
func (h *Handler) handleKernelUpdateApply(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
writeJSONError(w, http.StatusMethodNotAllowed, "只接受 POST")
|
||||
var body struct {
|
||||
Pkg string `json:"pkg"`
|
||||
Release bool `json:"release"`
|
||||
}
|
||||
// body 可为空（不合法就在下面报"必须指定来源"），所以解码失败不单独报错。
|
||||
_ = json.NewDecoder(r.Body).Decode(&body)
|
||||
|
||||
src, err := update.ParseSource(body.Pkg, body.Release)
|
||||
if err != nil {
|
||||
writeJSONError(w, http.StatusBadRequest, err.Error())
|
||||
return
|
||||
}
|
||||
// ★ 升级会让进程重启：先把响应写回去（同 kernel/restart 的教训）
|
||||
writeJSON(w, http.StatusAccepted, map[string]interface{}{
|
||||
"accepted": true,
|
||||
"message": "升级已受理。内核将替换二进制并重启，连接会断开数秒。完成后刷新状态页确认新版本。",
|
||||
})
|
||||
|
||||
var body struct {
|
||||
Pkg string `json:"pkg"`
|
||||
binPath := update.KernelBinaryPath()
|
||||
|
||||
tmp, err := os.MkdirTemp("", "kernel-upgrade-*")
|
||||
if err != nil {
|
||||
writeJSONError(w, http.StatusInternalServerError, "建临时目录失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
_ = json.NewDecoder(r.Body).Decode(&body)
|
||||
defer os.RemoveAll(tmp)
|
||||
|
||||
root, rel, err := src.Resolve(tmp)
|
||||
if err != nil {
|
||||
writeJSONError(w, http.StatusBadGateway, "取包失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
newDG := update.FindFile(root, "depgraph.bin")
|
||||
|
||||
// WebUI 侧天然有总线（它就是 SSE 的上游）：升级每一步都让界面看得到
|
||||
var bus *events.Bus
|
||||
if h.sdk != nil {
|
||||
bus = h.sdk.EventBus()
|
||||
}
|
||||
go runKernelUpgrade(body.Pkg, bus)
|
||||
|
||||
out, err := update.Apply(root, binPath, newDG, dataDirOf(binPath), bus)
|
||||
if err != nil {
|
||||
writeJSONError(w, http.StatusInternalServerError, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
res := map[string]interface{}{
|
||||
"applied": true,
|
||||
"from": out.From.Version,
|
||||
"to": out.To.Version,
|
||||
"backup": out.BackupDir,
|
||||
"message": "新内核已就位，正在重启。新版本启动后由第 0 层 update 模块确认；无法启动会自动回滚。",
|
||||
}
|
||||
if rel != nil {
|
||||
res["release"] = rel.Latest
|
||||
res["asset"] = rel.Asset
|
||||
}
|
||||
|
||||
// 先把响应写回，再让内核消失（同 kernel/restart 的教训：不 Flush 会
|
||||
// 变成 connection reset，用户以为出错了）。
|
||||
writeJSON(w, http.StatusOK, res)
|
||||
if f, ok := w.(http.Flusher); ok {
|
||||
f.Flush()
|
||||
}
|
||||
|
||||
// 重启走**既有**内核控制通道（restartCh → exit 42 → guard 用新二进制重拉）。
|
||||
// 不自己 systemctl，也不另开通道。
|
||||
if h.kernelCtl == nil {
|
||||
log.Printf("[webui] 新内核已就位，但内核控制通道不可用（kernelControlAllowed 未授予 webui），无法请求重启")
|
||||
return
|
||||
}
|
||||
if err := h.kernelCtl.RequestKernelRestart("kernel_update apply"); err != nil {
|
||||
log.Printf("[webui] kernel restart request failed (新内核已就位，需手动重启): %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// currentKernelVersion 读本机 homed 的版本（尽力而为）。
|
||||
func currentKernelVersion() *update.BinVersion {
|
||||
bin := "/usr/local/bin/homed"
|
||||
bin := update.KernelBinaryPath()
|
||||
if _, err := os.Stat(bin); err != nil {
|
||||
return nil
|
||||
}
|
||||
@ -87,33 +140,13 @@ func currentKernelVersion() *update.BinVersion {
|
||||
return v
|
||||
}
|
||||
|
||||
// runKernelUpgrade 在后台执行完整升级闭环（含重启与回滚）。
|
||||
// dataDirOf 推导数据目录（升级确认标记的落点）。
|
||||
//
|
||||
// bus 为 nil 时退化成纯日志（该部署未接事件总线）。
|
||||
func runKernelUpgrade(pkgPath string, bus *events.Bus) {
|
||||
if pkgPath == "" {
|
||||
// 没给本地包：从最新 release 资产下载。
|
||||
// 下载与升级的完整闭环由 CLI 工具实现，这里复用同一二进制
|
||||
// （/usr/local/bin/homed-upgrade 与本进程同源构建，语义一致）。
|
||||
if _, err := exec.LookPath("homed-upgrade"); err != nil {
|
||||
return
|
||||
}
|
||||
return // 下载逻辑见 CLI；此处仅受理本地包路径
|
||||
// 与内核 `--data` 必须一致，否则第 0 层 update 模块读不到 pending 标记。
|
||||
// 优先用环境变量显式指定（部署方知道自己在哪），否则退回本机生产路径。
|
||||
func dataDirOf(binPath string) string {
|
||||
if d := os.Getenv("HOMEAGENT_DATA_DIR"); d != "" {
|
||||
return d
|
||||
}
|
||||
// 本地包：直接走共享底层（与 CLI 同代码路径）
|
||||
tmp, err := os.MkdirTemp("", "kernel-upgrade-*")
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer os.RemoveAll(tmp)
|
||||
root, err := update.ExtractPkg(pkgPath, tmp)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
newBin := update.FindFile(root, "homed")
|
||||
newDG := update.FindFile(root, "depgraph.bin")
|
||||
if newBin == "" {
|
||||
return
|
||||
}
|
||||
_ = update.DoUpgradeBus(root, "/usr/local/bin/homed", newDG, "/home/newqqagent/plugins", "/home/newqqagent", bus)
|
||||
return "/home/newqqagent"
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user