feat: Phase 1+2 架构进化 — 连续思考链/主动消息决策/情感状态机/离线自主思考 (86文件)
Phase 1 (基础设施): - ThinkChain 思考链连续性 + 差异化思考提示词 (persistent) - AutonomousToolPolicy 工具安全策略 (safe/unsafe/conditional) - MessageScheduler 自适应消息节奏 (Idle/Available/Busy) - SessionEnrichmentStore 渐进式上下文丰富 (5层) - ConversationBus 事件总线 + ResponseCache (dedup) - pkg/logger 统一日志 + 所有 handler 替换 fmt.Printf - NPE 守卫/链路优化/数据库表修复/Go workspace Phase 2 (人格交互): - EmotionState/EmotionTracker 情感状态机 (5种心情, 情绪衰减) - ProactiveGuard 主动消息多维决策 (静默时段/紧急度/频率/校验) - Gateway↔ai-core 在线状态感知链路 (presence notification) - 离线思考频率控制 + 重连问候 + 离线消息排队 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
+35
-33
@@ -2,7 +2,7 @@ package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"github.com/yourname/cyrene-ai/pkg/logger"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
@@ -23,9 +23,10 @@ import (
|
||||
)
|
||||
|
||||
func main() {
|
||||
logger.SetDefault(logger.New("gateway"))
|
||||
// 自动加载 .env 文件（来自 backend/.env）
|
||||
if err := godotenv.Load("../.env"); err != nil {
|
||||
log.Println("ℹ 未找到 .env 文件，将使用环境变量或默认值")
|
||||
logger.Println("ℹ 未找到 .env 文件，将使用环境变量或默认值")
|
||||
}
|
||||
|
||||
// 加载配置
|
||||
@@ -33,7 +34,7 @@ func main() {
|
||||
|
||||
// 确保上传目录存在
|
||||
if err := os.MkdirAll("./uploads", 0755); err != nil {
|
||||
log.Printf("⚠ 创建上传目录失败: %v", err)
|
||||
logger.Printf("⚠ 创建上传目录失败: %v", err)
|
||||
}
|
||||
|
||||
// 初始化数据库持久化存储 (降级：连接失败不崩溃)
|
||||
@@ -46,49 +47,49 @@ func main() {
|
||||
var ruleEngine *engine.RuleEngine
|
||||
databaseURL := cfg.DatabaseURL()
|
||||
if s, err := store.NewSessionStore(databaseURL); err != nil {
|
||||
log.Printf("⚠ 会话持久化存储初始化失败 (数据库不可用): %v", err)
|
||||
log.Println("⚠ Gateway 将以仅内存模式运行 — 会话数据在重启后丢失")
|
||||
logger.Printf("⚠ 会话持久化存储初始化失败 (数据库不可用): %v", err)
|
||||
logger.Println("⚠ Gateway 将以仅内存模式运行 — 会话数据在重启后丢失")
|
||||
} else {
|
||||
sessionStore = s
|
||||
log.Println("✅ 会话持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 会话持久化存储已启用 (PostgreSQL)")
|
||||
|
||||
// 初始化 users 表
|
||||
if err := store.CreateUsersTable(s.DB()); err != nil {
|
||||
log.Printf("⚠ 创建 users 表失败: %v", err)
|
||||
logger.Printf("⚠ 创建 users 表失败: %v", err)
|
||||
} else {
|
||||
log.Println("✅ Users 表已就绪")
|
||||
logger.Println("✅ Users 表已就绪")
|
||||
}
|
||||
|
||||
// 种子数据：如果没有 admin 用户，创建默认 admin
|
||||
if existingAdmin, err := store.GetUserByUsername(s.DB(), cfg.AdminUsername); err != nil {
|
||||
log.Printf("⚠ 查询管理员用户失败: %v", err)
|
||||
logger.Printf("⚠ 查询管理员用户失败: %v", err)
|
||||
} else if existingAdmin == nil {
|
||||
log.Printf("🔧 未找到管理员用户，创建默认 %s (username: %s)...", cfg.AdminUsername, cfg.AdminUsername)
|
||||
logger.Printf("🔧 未找到管理员用户，创建默认 %s (username: %s)...", cfg.AdminUsername, cfg.AdminUsername)
|
||||
defaultAdminPassword := cfg.AdminPassword
|
||||
passwordHash, err := bcrypt.GenerateFromPassword([]byte(defaultAdminPassword), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
log.Printf("⚠ 管理员密码哈希生成失败: %v", err)
|
||||
logger.Printf("⚠ 管理员密码哈希生成失败: %v", err)
|
||||
} else {
|
||||
if _, err := store.CreateUser(s.DB(), cfg.AdminUsername, string(passwordHash), true); err != nil {
|
||||
log.Printf("⚠ 创建默认管理员失败: %v", err)
|
||||
logger.Printf("⚠ 创建默认管理员失败: %v", err)
|
||||
} else {
|
||||
log.Printf("✅ 默认管理员用户已创建 (username: %s)", cfg.AdminUsername)
|
||||
logger.Printf("✅ 默认管理员用户已创建 (username: %s)", cfg.AdminUsername)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
log.Println("✅ 管理员用户已存在")
|
||||
logger.Println("✅ 管理员用户已存在")
|
||||
}
|
||||
|
||||
// 清理旧的管理员用户 (is_admin=true 但 username 与当前 ADMIN_USERNAME 不同)
|
||||
// 当 .env 中 ADMIN_USERNAME 变更时，旧的 admin 用户会成为孤立的会话持有者
|
||||
if allUsers, err := store.ListUsers(s.DB()); err != nil {
|
||||
log.Printf("⚠ 查询所有用户失败: %v", err)
|
||||
logger.Printf("⚠ 查询所有用户失败: %v", err)
|
||||
} else {
|
||||
for _, u := range allUsers {
|
||||
if u.IsAdmin && u.Username != cfg.AdminUsername {
|
||||
log.Printf("🗑 清理旧管理员用户: %s (id=%d)", u.Username, u.ID)
|
||||
logger.Printf("🗑 清理旧管理员用户: %s (id=%d)", u.Username, u.ID)
|
||||
if err := store.DeleteUser(s.DB(), u.ID); err != nil {
|
||||
log.Printf("⚠ 删除旧管理员用户失败: %s, err=%v", u.Username, err)
|
||||
logger.Printf("⚠ 删除旧管理员用户失败: %s, err=%v", u.Username, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -96,42 +97,42 @@ func main() {
|
||||
|
||||
// 初始化提醒存储（复用同一数据库连接）
|
||||
if rs, err := store.NewReminderStore(s.DB()); err != nil {
|
||||
log.Printf("⚠ 提醒存储初始化失败: %v", err)
|
||||
logger.Printf("⚠ 提醒存储初始化失败: %v", err)
|
||||
} else {
|
||||
reminderStore = rs
|
||||
log.Println("✅ 提醒持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 提醒持久化存储已启用 (PostgreSQL)")
|
||||
}
|
||||
|
||||
// 初始化简报存储（复用同一数据库连接）
|
||||
if bs, err := store.NewBriefingStore(s.DB()); err != nil {
|
||||
log.Printf("⚠ 简报存储初始化失败: %v", err)
|
||||
logger.Printf("⚠ 简报存储初始化失败: %v", err)
|
||||
} else {
|
||||
briefingStore = bs
|
||||
log.Println("✅ 简报持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 简报持久化存储已启用 (PostgreSQL)")
|
||||
}
|
||||
|
||||
// 初始化自动化存储（复用同一数据库连接）
|
||||
if as, err := store.NewAutomationStore(s.DB()); err != nil {
|
||||
log.Printf("⚠ 自动化存储初始化失败: %v", err)
|
||||
logger.Printf("⚠ 自动化存储初始化失败: %v", err)
|
||||
} else {
|
||||
automationStore = as
|
||||
log.Println("✅ 自动化持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 自动化持久化存储已启用 (PostgreSQL)")
|
||||
}
|
||||
|
||||
// 初始化文件存储（复用同一数据库连接）
|
||||
if fs, err := store.NewFileStore(s.DB()); err != nil {
|
||||
log.Printf("⚠ 文件存储初始化失败: %v", err)
|
||||
logger.Printf("⚠ 文件存储初始化失败: %v", err)
|
||||
} else {
|
||||
fileStore = fs
|
||||
log.Println("✅ 文件持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 文件持久化存储已启用 (PostgreSQL)")
|
||||
}
|
||||
|
||||
// 初始化知识库存储（复用同一数据库连接）
|
||||
if ks, err := store.NewKnowledgeStore(s.DB()); err != nil {
|
||||
log.Printf("⚠ 知识库存储初始化失败: %v", err)
|
||||
logger.Printf("⚠ 知识库存储初始化失败: %v", err)
|
||||
} else {
|
||||
knowledgeStore = ks
|
||||
log.Println("✅ 知识库持久化存储已启用 (PostgreSQL)")
|
||||
logger.Println("✅ 知识库持久化存储已启用 (PostgreSQL)")
|
||||
}
|
||||
}
|
||||
|
||||
@@ -139,12 +140,13 @@ func main() {
|
||||
hub := ws.NewHub()
|
||||
hub.SetStore(sessionStore)
|
||||
hub.SetIdleTimeout(cfg.SessionIdleTimeoutMin)
|
||||
hub.SetAICoreConfig(cfg.AICoreURL, cfg.InternalServiceToken)
|
||||
|
||||
// 初始化规则引擎 (需要 Hub)
|
||||
if automationStore != nil {
|
||||
ruleEngine = engine.NewRuleEngine(automationStore, hub)
|
||||
ruleEngine.Start()
|
||||
log.Println("✅ 规则引擎已启动")
|
||||
logger.Println("✅ 规则引擎已启动")
|
||||
}
|
||||
|
||||
// 初始化Gin
|
||||
@@ -192,9 +194,9 @@ func main() {
|
||||
}
|
||||
|
||||
go func() {
|
||||
log.Printf("🚀 Gateway 启动在端口 %s", cfg.Port)
|
||||
logger.Printf("🚀 Gateway 启动在端口 %s", cfg.Port)
|
||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
log.Fatalf("服务启动失败: %v", err)
|
||||
logger.Fatalf("服务启动失败: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
@@ -202,19 +204,19 @@ func main() {
|
||||
quit := make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-quit
|
||||
log.Println("正在关闭服务...")
|
||||
logger.Println("正在关闭服务...")
|
||||
|
||||
hub.StopIoTBroadcast()
|
||||
|
||||
// 关闭数据库连接
|
||||
if sessionStore != nil {
|
||||
if err := sessionStore.Close(); err != nil {
|
||||
log.Printf("⚠ 关闭数据库连接失败: %v", err)
|
||||
logger.Printf("⚠ 关闭数据库连接失败: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
defer cancel()
|
||||
srv.Shutdown(ctx)
|
||||
log.Println("服务已关闭")
|
||||
logger.Println("服务已关闭")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user