AskaEth
3ba31b58ff
- 引用计数真实维护（documents 表唯一写入面）：增减点只有四处 —— history 的 insert / UpdateMessageDocument / DeleteMessage / DeleteSession，且**全在同一事务**内（计数写失败 ⇒ 消息写入整体回滚，有钉住用例）；UpdateMessageDocument 按新旧名字**集合差**增减（三阶段回写是常态，重算会重复计数 ⇒ 文件永远回收不掉）。 - GC：候选 = ref_count = 0；判据 = last_referenced_at（NULL 回落 created_at）早于 now − 保留期；动作 = 删 sidecar + 原件 + 元数据行（删行带 AND ref_count = 0 原子再确认）；路径双检；汇总日志（含实占字节）；单件失败 WARN 并继续；删除顺序 = 先文件后行（失败则行留着下轮重试，不留孤儿）。 - 组合根：启动期清扫一次（不阻塞启动）+ ticker 常驻（可停，close 顺序在关库之前）；env DOCUMENT_GC_INTERVAL 默认 1h / DOCUMENT_RETENTION 默认 168h，**任一 ≤0 = 关闭 GC**（RETENTION=0 不解释成立即回收，关闭时记 WARN）；.env.example 已同步（AGENTS §10）。 - 主会话独立验收：隔离 worktree（S7 的 18 文件，排除 F2 在途的 mcp/vision）build/vet 绿、全树 59 包 0 FAIL、GC/引用用例 20 PASS、双平台交叉编译绿；**变异亲自复现**（候选查询去掉 ref_count = 0 ⇒ TestGCDoesNotDeleteReferencedDocument 红：candidates = 1, want 0）。 - 未做（任务书允许的「只做后端」）：面板手动「清理未引用附件」入口 —— 后端能力已就绪（documents.NewGC + GC.Sweep），接成端点属小切片（含 admin 路由 + 审计 document.gc + 契约登记）。 - 已知窄窗口（已在 gc.go 文件头登记）：一轮 GC 与一次落库几乎同时发生时可能删掉刚被引用的文件（行不删、记 WARN）；坏 document 列只记 WARN 不阻塞消息写入 ⇒ 计数可能偏低（只会让文件**晚**回收，绝不早删）。
124 lines
4.2 KiB
Go
124 lines
4.2 KiB
Go
// cyrene 命令是 Cyrene v2 的唯一二进制入口：它把全部子系统（配置、日志、数据库、
|
||
// bus、人格、模型适配器、工具分发器、记忆栈、agent 循环、编排、执行、IoT、渠道
|
||
// 会话与适配器）组装进同一个进程，在单端口（:8080）上同时提供网关与内嵌前端，
|
||
// 并在收到终止信号时优雅退出。
|
||
//
|
||
// 构建：`go build -ldflags="-s -w" -trimpath -o cyrene ./cmd/cyrene/`。
|
||
// 前端必须先构建，SPA 才能被内嵌：`cd frontend/web && npm run build`。
|
||
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"io/fs"
|
||
"log/slog"
|
||
"os"
|
||
"path/filepath"
|
||
"time"
|
||
// 内嵌 IANA 时区数据库，让配置的 TZ（config.TZ，默认 "Asia/Shanghai"）
|
||
// 在任何平台都能解析：Windows 不自带系统 tzdata，Go 也刻意不捆绑，缺了它
|
||
// 循环的时间注入在 Windows 上会静默失效（internal/runtime/loop/time.go 会记
|
||
// "timezone load failed"）。代价约 450 KB，换来二进制自包含。
|
||
_ "time/tzdata"
|
||
|
||
capreg "git.yeij.top/AskaEth/Cyrene_v2/internal/capability/registry"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/logger"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/platform"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/webui"
|
||
)
|
||
|
||
// envPath 是启动时合并的 .env 文件（仅开发用；生产走 systemd/env_file）。
|
||
// .env 不存在不算错误。
|
||
const envPath = ".env"
|
||
|
||
// shutdownTimeout 是优雅退场的时长上界。
|
||
const shutdownTimeout = 15 * time.Second
|
||
|
||
// externalDistCandidates 是前端覆盖目录的候选清单：按序检查，都不存在才回落到
|
||
// 内嵌 dist；命中的第一个生效。路径相对进程工作目录。
|
||
func externalDistCandidates() []string {
|
||
return []string{
|
||
filepath.Join("frontend", "web", "dist"),
|
||
filepath.Join("frontend", "dist"),
|
||
}
|
||
}
|
||
|
||
// resolveFrontendFS 返回用于伺服 SPA 的文件系统：存在外部 dist 目录时用它
|
||
// （部署期热覆盖），否则用内嵌 dist。
|
||
func resolveFrontendFS(log *slog.Logger) fs.FS {
|
||
for _, dir := range externalDistCandidates() {
|
||
if info, err := os.Stat(dir); err == nil && info.IsDir() {
|
||
log.Info("serving external frontend dist", "dir", dir)
|
||
return os.DirFS(dir)
|
||
}
|
||
}
|
||
log.Info("serving embedded frontend dist")
|
||
return webui.Assets()
|
||
}
|
||
|
||
func main() {
|
||
if err := run(); err != nil {
|
||
fmt.Fprintln(os.Stderr, "cyrene:", err)
|
||
os.Exit(1)
|
||
}
|
||
}
|
||
|
||
// run 启动应用、拉起全部服务与网关，然后阻塞等待终止信号或网关失败。
|
||
func run() error {
|
||
cfg := config.MustLoad(envPath)
|
||
format := logger.FormatText
|
||
if cfg.Env == "production" {
|
||
format = logger.FormatJSON
|
||
}
|
||
logRing := logger.NewRingBuffer(logger.DefaultRingCapacity)
|
||
log := logger.New(logger.Options{Level: cfg.LogLevel, Format: string(format), Ring: logRing})
|
||
log = logger.WithSource(log, "main")
|
||
|
||
ctx, stop := platform.SignalContext()
|
||
defer stop()
|
||
|
||
a, err := newApp(ctx, cfg, log)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// 可观测性日志追尾：logger 的环形缓冲在上面（网关之前）就建好了，
|
||
// 现在网关已存在，把它交给网关。
|
||
wireGatewayLogBuffer(a, logRing)
|
||
|
||
if err := a.automation.Start(ctx); err != nil {
|
||
return fmt.Errorf("start automations: %w", err)
|
||
}
|
||
a.backthink.Start(ctx)
|
||
// 文档回收（切片 S7）：启动期清扫一次 + 此后按 DOCUMENT_GC_INTERVAL 定时清扫。
|
||
// 它不等这一次跑完就去拉网关（见 wire_document_gc.go 的取舍 1）。
|
||
a.startDocumentGC(ctx)
|
||
a.log.Info("cyrene subsystems started",
|
||
"personas", a.personas.CountPersonas(),
|
||
"tools", len(a.caps.List(capreg.KindTool)),
|
||
"channels", len(a.caps.List(capreg.KindChannel)))
|
||
a.log.Info("cyrene ready", "data_dir", resolveDataDir(cfg))
|
||
|
||
errCh, err := a.gw.Start(ctx)
|
||
if err != nil {
|
||
return fmt.Errorf("start gateway: %w", err)
|
||
}
|
||
|
||
select {
|
||
case <-ctx.Done():
|
||
a.log.Info("cyrene signal received, shutting down")
|
||
case gwErr := <-errCh:
|
||
if gwErr != nil {
|
||
shutCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
|
||
defer cancel()
|
||
a.close(shutCtx)
|
||
return fmt.Errorf("gateway failed: %w", gwErr)
|
||
}
|
||
}
|
||
|
||
shutCtx, cancel := context.WithTimeout(context.Background(), shutdownTimeout)
|
||
defer cancel()
|
||
a.close(shutCtx)
|
||
return nil
|
||
}
|