mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 14:57:38 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
package plugin
|
||
|
||
import (
|
||
"encoding/json"
|
||
"sync"
|
||
|
||
"github.com/JianFeeeee/HologramAgent/internal/events"
|
||
"github.com/JianFeeeee/HologramAgent/internal/plugin/proc"
|
||
pubsdk "github.com/JianFeeeee/homeagentsdk/sdk"
|
||
)
|
||
|
||
// EventRing 是 Bus 与 proc.EvtRing 之间的适配层。
|
||
//
|
||
// 把内核的事件总线接到共享内存事件环：Bus.Publish → handler
|
||
// 把事件序列化写入 EvtRing slot → eventfd 通知子进程。
|
||
// 不改 Bus 自身结构（保护零 API 变动）。
|
||
type EventRing struct {
|
||
ring *proc.EvtRing
|
||
bus *events.Bus
|
||
efd int
|
||
|
||
// unsubs 保存全部已注册订阅的取消函数。
|
||
//
|
||
// ★ 为什么必须留着：这些 handler 会 ring.WritePush（写共享内存）。
|
||
// 而 Host.Close() 会 freeShm 解除整块映射 —— 若那时 handler 还在 Bus 上，
|
||
// 一条事件就会让 handler 写已解除映射的内存：SIGSEGV。
|
||
// 注意 Bus.safeCall 的 recover **捕不到** SIGSEGV（它是 runtime 致命错误，
|
||
// 不是 panic），所以这不是「最坏情况只丢一条事件」，而是整个内核进程被杀。
|
||
//
|
||
// 此前 handleEvents 把 EvtRingSubscribe 返回的取消函数直接丢弃
|
||
// （且 EventsUnsubscribe 是空实现），于是每个订阅过的插件都在 Bus 上
|
||
// 永久留了一个写共享内存的 handler —— 内核关停时必炸。
|
||
// 现在改为在这里登记，由 Close 统一退订（内核关停、以及插件自己的
|
||
// events.unsubscribe 都走这里）。
|
||
mu sync.Mutex
|
||
unsubs []func()
|
||
}
|
||
|
||
func NewEventRing(ring *proc.EvtRing, efd int, bus *events.Bus) *EventRing {
|
||
return &EventRing{ring: ring, bus: bus, efd: efd}
|
||
}
|
||
|
||
// Close 退订本适配层注册到 Bus 的全部 handler。
|
||
//
|
||
// 必须在 Host.Close()（munmap 共享段）**之前**调用；见 unsubs 的说明。
|
||
// 幂等：重复调用安全（退订函数本身在 Bus 侧是「找不到就什么都不做」）。
|
||
func (er *EventRing) Close() {
|
||
er.mu.Lock()
|
||
unsubs := er.unsubs
|
||
er.unsubs = nil
|
||
er.mu.Unlock()
|
||
|
||
for _, fn := range unsubs {
|
||
if fn != nil {
|
||
fn()
|
||
}
|
||
}
|
||
}
|
||
|
||
// Subscribe 在 Bus 上注册一个把事件分发到事件环的 handler，返回取消函数。
|
||
//
|
||
// 不改 Bus 自身结构——handler 把事件序列化后写入环并 post eventfd，
|
||
// Bus 侧按 EventType 精确匹配分发（与现有逻辑完全一致）。
|
||
func (er *EventRing) Subscribe(eventType pubsdk.EventType) func() {
|
||
return er.bus.Subscribe(events.EventType(eventType), func(evt *events.Event) {
|
||
payload, err := json.Marshal(evt)
|
||
if err != nil {
|
||
return
|
||
}
|
||
er.ring.WritePush(pubsdk.EventType(evt.Type), payload)
|
||
proc.EvtfdNotify(er.efd)
|
||
})
|
||
}
|
||
|
||
// EvtRingSubscribe 实现 proc.EvtRingSubscriber 接口。
|
||
// 按事件类型列表订阅，返回统一取消函数。
|
||
func (er *EventRing) EvtRingSubscribe(types []pubsdk.EventType) func() {
|
||
unsubscribes := make([]func(), 0, len(types))
|
||
for _, t := range types {
|
||
unsubscribes = append(unsubscribes, er.Subscribe(t))
|
||
}
|
||
return func() {
|
||
for _, fn := range unsubscribes {
|
||
fn()
|
||
}
|
||
}
|
||
}
|
||
|
||
// EvtRingSubscribeTracked 与 EvtRingSubscribe 相同，但把取消函数登记到
|
||
// unsubs，供 Close 统一退订。内核的 events.subscribe 走这条。
|
||
func (er *EventRing) EvtRingSubscribeTracked(types []pubsdk.EventType) func() {
|
||
un := er.EvtRingSubscribe(types)
|
||
if un == nil {
|
||
return nil
|
||
}
|
||
er.mu.Lock()
|
||
er.unsubs = append(er.unsubs, un)
|
||
er.mu.Unlock()
|
||
return un
|
||
}
|