fix(plugin): 修复退出时插件信息持久化以及加载插件信息 等问题

Author: Clov614

main.go

@@ -38,6 +38,7 @@ func main() {
 		zerolog.SetGlobalLevel(zerolog.InfoLevel)
 		gin.SetMode(gin.ReleaseMode)
 	}
+	defer logging.Close() // 保存日志
 	//// 在初始化完成后输出所有缓冲日志
 	//logging.Logger.Flush(zerolog.GlobalLevel())
 	//logging.Logger.SetActive(false) // 取消缓存，正常日志输出

rikkabot/adapter/adapter.go

@@ -43,7 +43,7 @@ func (a *Adapter) HandleCovert() {
 		for {
 			select {
 			case <-a.ctx.Done():
-				logging.ErrorWithErr(a.ctx.Err(), "handle covert exit")
+				logging.WarnWithErr(a.ctx.Err(), "handle covert exit")
 				return
 			case msg := <-a.cli.GetMsgChan(): // 转换收到的消息
 				logging.Debug("rikka-bot received message", map[string]interface{}{"sdk-msg": msg})
@@ -57,12 +57,15 @@ func (a *Adapter) HandleCovert() {
 		for {
 			select {
 			case <-a.ctx.Done():
-				logging.ErrorWithErr(a.ctx.Err(), "handle send exit")
+				logging.WarnWithErr(a.ctx.Err(), "handle send exit")
 				return
 			case respMsg := <-sendChan: // 接收到回复消息
 				logging.Debug("rikka-bot send message", map[string]interface{}{"sdk-msg": respMsg})
 				//rnd := rand.New(rand.NewSource(time.Now().UnixNano()))
 				//time.Sleep(time.Duration((rnd.Intn(1000) + 1000)) * time.Millisecond)
+				if respMsg == nil {
+					continue
+				}
 				err := a.sendMsg(respMsg)
 				if err != nil {
 					logging.ErrorWithErr(err, "sendMsg fail skip send")

rikkabot/message/message.go

@@ -34,7 +34,7 @@ type Message struct {
 	SenderName string             `json:"sender_name"`            // 消息发送者用户昵称
 	IsAtMe     bool               `json:"is_at"`                  // 群组中是否艾特本人
 	IsGroup    bool               `json:"is_group"`               // 是否为群聊消息
-	IsFriend   bool               `json:"is_friend"`              // 是否为好友私聊消息
+	IsFriend   bool               `json:"is_friend"`              // 是否为好友发送的消息 （注意 不是私聊消息）
 	IsGH       bool               `json:"is_gh"`                  // 是否为公众号
 	IsMySelf   bool               `json:"is_my_self"`             // 消息是否为自己发送的
 	IsSystem   bool               `json:"is_system"`              // 是否为系统消息

rikkabot/processor/cache/cache.go

@@ -20,8 +20,9 @@ type Cache struct {
 	*cacheExported // 隐藏字段
 	config         *config.CommonConfig
 
-	done chan struct{}
-	wg   sync.WaitGroup
+	done      chan struct{}
+	wg        sync.WaitGroup
+	closeOnce sync.Once
 }
 
 var (
@@ -308,9 +309,12 @@ func (c *Cache) cycleSave() {
 }
 
 func (c *Cache) Close() {
-	close(c.done)
-	c.wg.Wait()
-	logging.Info("cache closed")
+	c.closeOnce.Do(func() {
+		c.handleSave(false)
+		close(c.done)
+		c.wg.Wait()
+		logging.Info("cache closed")
+	})
 }
 
 func (c *Cache) handleSave(firstLoad bool) {

rikkabot/processor/processor.go

@@ -10,17 +10,21 @@ import (
 	"sync/atomic"
 	"time"
 
+	"github.com/Clov614/rikka-bot-wechat/rikkabot/processor/cache"
+
 	"github.com/Clov614/logging"
 	wcf "github.com/Clov614/wcf-rpc-sdk"
 
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/message"
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/plugins"
 
 	/* 下方为插件的导入 */
-	_ "github.com/Clov614/rikka-bot-wechat/rikkabot/plugins/admin" // 管理员模块
-	/* 从上到下对应优先级由高到低 */
+	_ "github.com/Clov614/rikka-bot-wechat/rikkabot/plugins/admin"       // 管理员模块
 	_ "github.com/Clov614/rikka-bot-wechat/rikkabot/plugins/biliDecoder" // bilibili链接解析
-)
+
+	/* 从上到下对应优先级由高到低 */
+	_ "github.com/Clov614/rikka-bot-wechat/rikkabot/plugins/ai" // AI对话
+	/* 从上到下对应优先级由高到低 */)
 
 type Processor struct {
 	ctx        context.Context
@@ -207,11 +211,15 @@ func (p *Processor) Close() {
 				}
 			}
 			close(done)
+			logging.Debug("所有插件关闭完成")
 		}()
 		// 缓存插件设置信息
 		ag := plugins.GetAutoRegister()
 		ag.CachePlugins()
+		logging.Debug("保存插件信息完毕")
 
+		c := cache.GetCache()
+		c.Close() // 保存并关闭缓存
 		select {
 		case <-done:
 		case <-time.After(time.Second * 5): // 设置一个超时时间

rikkabot/processor/processor_test.go

@@ -6,41 +6,44 @@ package processor
 
 import (
 	"context"
-	wcf "github.com/Clov614/wcf-rpc-sdk"
 	"testing"
 	"time"
 
+	wcf "github.com/Clov614/wcf-rpc-sdk"
+
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/message"
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/plugins"
+	"github.com/Clov614/rikka-bot-wechat/rikkabot/processor/cache"
 )
 
-type pTestMatcher struct {
-	isThought bool
-}
-
-func (m *pTestMatcher) Match(ctx context.Context, msg *message.Message) bool {
-	return m.isThought
-}
-
-// TestPlugin 是一个真实的插件，用于测试
+// 定义一个专门用于测试的 TestPlugin 结构体
 type TestPlugin struct {
-	*plugins.Plugin // 嵌入标准的 Plugin 结构
-	executeCount    int
+	*plugins.Plugin     // 嵌入标准的 Plugin 结构
+	executeCount    int // 执行计数器
 }
 
+// AddExecCount 增加执行计数
 func (p *TestPlugin) AddExecCount(delta int) {
 	p.executeCount += delta
 }
 
 // NewTestPlugin 创建一个新的 TestPlugin 实例
 func NewTestPlugin(name string, level plugins.PluginLevel) *TestPlugin {
 	p := &TestPlugin{
-		Plugin: plugins.DefaultPlugin(name).AsEnable(), // 使用默认的 Plugin 初始化 // 设置插件名称
+		Plugin: plugins.DefaultPlugin(name).AsEnable(), // 使用默认的 Plugin 初始化并启用
 	}
 	p.Plugin.PluginOpt.Level = level // 设置插件级别
 	return p
 }
 
+type pTestMatcher struct {
+	isThought bool
+}
+
+func (m *pTestMatcher) Match(ctx context.Context, msg *message.Message) bool {
+	return m.isThought
+}
+
 // TestProcessor_RegisterAndStart 测试插件注册和启动流程
 func TestProcessor_RegisterAndStart(t *testing.T) {
 	plugin := NewTestPlugin("test RegistAndStart-01", plugins.MediumLevel)
@@ -153,3 +156,102 @@ func TestProcessor_MessageFlow(t *testing.T) {
 		t.Logf("Sending message: %s", outputMsg.Content)
 	}
 }
+
+// TestProcessor_CloseAndCache 测试 Processor 的 Close 方法、cache 的联动以及重启后的加载
+func TestProcessor_CloseAndCache(t *testing.T) {
+	// 准备阶段：创建并注册插件
+	pluginName := "testCloseAndCache-01"
+	plugin := NewTestPlugin(pluginName, plugins.MediumLevel)
+	plugin.AsAction(&plugins.ActionHandler{
+		Name:    "test action",
+		Matcher: &pTestMatcher{true},
+		Action: func(ctx context.Context, recvMsg *message.Message) (reply message.Message, ok bool, err error) {
+			plugin.AddExecCount(1) // 增加执行计数
+			return *recvMsg, true, nil
+		},
+	})
+	plugins.GetAutoRegister().RegisterPlugin(plugin) // 直接获取并注册
+
+	// 阶段 1: 首次运行并关闭，测试缓存
+	func() {
+		ctx, cancel := context.WithCancel(context.Background())
+		defer cancel()
+		inputChan := make(chan *message.Message)
+		sendChan := make(chan *message.Message, 10)
+		processor := NewProcessor(ctx, wcf.NewClient(10, false, false))
+
+		processor.Start(inputChan, sendChan)
+		inputChan <- &message.Message{Content: "test message"} // 触发插件执行
+		<-time.After(time.Second * 1)                          // 等待执行
+
+		processor.Close() // 关闭并触发缓存
+
+		// 验证插件执行次数和缓存状态
+		if plugin.executeCount != 1 {
+			t.Errorf("Expected plugin executeCount to be 1, got: %d", plugin.executeCount)
+		}
+		cachedPlugin, ok := cache.GetCache().GetPluginInfo(pluginName)
+		if !ok {
+			t.Fatalf("Plugin %s not found in cache", pluginName)
+		}
+		// 验证缓存的是 PluginOpt
+		cachedOpt, ok := cachedPlugin.(plugins.PluginOpt)
+		if !ok {
+			t.Fatalf("Expected cached plugin info to be PluginOpt, got: %T", cachedPlugin)
+		}
+		if !cachedOpt.Enable {
+			t.Errorf("Expected plugin %s to be enabled in cache", pluginName)
+		}
+	}() // 使用匿名函数隔离作用域
+
+	// 阶段 2: 重启并验证加载
+	func() {
+		ctx, cancel := context.WithCancel(context.Background())
+		defer cancel()
+		inputChan := make(chan *message.Message)
+		sendChan := make(chan *message.Message, 10)
+		processor := NewProcessor(ctx, wcf.NewClient(10, false, false))
+
+		// 注意：这里不需要重新注册插件，因为 AutoRegister 会从 cache 加载
+		processor.Start(inputChan, sendChan)
+
+		// 验证插件已从 cache 加载
+		registeredPlugins := processor.LevelLayer[plugins.MediumLevel].GetPlugins()
+		// 遍历找到我们的测试插件
+		var plugin2 *TestPlugin
+		for _, p := range registeredPlugins {
+			if (*p).GetName() == pluginName {
+				var ok bool
+				plugin2, ok = (*p).(*TestPlugin)
+				if !ok {
+					t.Fatalf("Expected plugin '%s' to be *TestPlugin, got: %T", pluginName, *p)
+				}
+				break
+			}
+		}
+
+		if plugin2 == nil {
+			t.Fatalf("Plugin '%s' not found in registered plugins", pluginName)
+		}
+
+		// 从缓存中恢复
+		cachedOpt, ok := cache.GetCache().GetPluginInfo(pluginName)
+		if !ok {
+			t.Fatalf("Plugin %s not found in cache", pluginName)
+		}
+		plugin2.PluginOpt = cachedOpt.(plugins.PluginOpt)
+
+		if !plugin2.PluginOpt.Enable {
+			t.Errorf("Expected plugin %s to be enabled after restart", pluginName)
+		}
+
+		// 再次触发插件执行，验证其功能
+		inputChan <- &message.Message{Content: "test message"}
+		<-time.After(time.Second * 1)
+		if plugin2.executeCount != 2 { // 执行次数应为 2
+			t.Errorf("Expected plugin executeCount to be 2, got: %d", plugin2.executeCount)
+		}
+
+		processor.Close() // 关闭
+	}() // 使用匿名函数隔离作用域
+}

rikkabot/rikkabot.go

@@ -4,6 +4,9 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"sync"
+	"time"
+
 	"github.com/Clov614/logging"
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/config"
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/message"
@@ -12,8 +15,6 @@ import (
 	"github.com/Clov614/rikka-bot-wechat/rikkabot/utils/timeutil"
 	wcf "github.com/Clov614/wcf-rpc-sdk"
 	"github.com/google/uuid"
-	"sync"
-	"time"
 )
 
 type RikkaBot struct {
@@ -155,35 +156,32 @@ func (r *RikkaBot) Start() {
 	r.EnableProcess = true // 防止生产者阻塞
 }
 
+var onceExit sync.Once
+
 // Exit 主动退出 rikkabot
 func (r *RikkaBot) Exit() {
-	logging.Info("rikka bot exited")
-	r.EventPool.Close()
-	r.Processor.Close()
-	r.cancel()
-	r.cli.Close()
+	onceExit.Do(func() {
+		logging.Info("rikka bot exited")
+		r.EventPool.Close()
+		r.Processor.Close()
+		r.cli.Close()
+		r.cancel()
+	})
 }
 
 // ExitWithErr 异常退出 rikkabot
 func (r *RikkaBot) ExitWithErr(code int, msg string) {
-	logging.Info("rikka bot exited")
 	logging.Error("异常退出")
 	logging.Error(msg, map[string]interface{}{"exit code": code})
-	r.EventPool.Close()
-	r.Processor.Close()
-	r.cancel()
-	r.cli.Close()
+	r.Exit()
 }
 
-var onceExit sync.Once
-
 // Block 当发生错误，该方法会立即返回，否则会一直阻塞
 func (r *RikkaBot) Block() error {
 	<-r.ctx.Done()
-	onceExit.Do(func() { r.Exit() })
-	logging.Close() // 关闭日志文件
-	logging.Info("主程序将在5s后退出...")
-	time.Sleep(5 * time.Second) // 增加一个固定延迟，简单的确保退出 todo perf -> wg
+	r.Exit()
+	logging.Info("主程序将在10s后退出...")
+	time.Sleep(10 * time.Second) // 增加一个固定延迟，简单的确保退出 todo perf -> wg
 	return r.err
 }