From 6efe7331272b5eb3c81c40f7b3e730323aaa6870 Mon Sep 17 00:00:00 2001 From: Lawrence Liu Date: Tue, 11 Nov 2025 10:37:46 +0800 Subject: [PATCH] =?UTF-8?q?feat(decision):=20auto-reload=20prompt=20templa?= =?UTF-8?q?tes=20when=20starting=20trader=20(#833)=20*=20feat:=20=E5=90=AF?= =?UTF-8?q?=E5=8A=A8=E4=BA=A4=E6=98=93=E5=91=98=E6=97=B6=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E9=87=8D=E6=96=B0=E5=8A=A0=E8=BD=BD=E7=B3=BB=E7=BB=9F=E6=8F=90?= =?UTF-8?q?=E7=A4=BA=E8=AF=8D=E6=A8=A1=E6=9D=BF=20##=20=E6=94=B9=E5=8A=A8?= =?UTF-8?q?=E5=86=85=E5=AE=B9=20-=20=E5=9C=A8=20handleStartTrader=20?= =?UTF-8?q?=E4=B8=AD=E8=B0=83=E7=94=A8=20decision.ReloadPromptTemplates()?= =?UTF-8?q?=20-=20=E6=AF=8F=E6=AC=A1=E5=90=AF=E5=8A=A8=E4=BA=A4=E6=98=93?= =?UTF-8?q?=E5=91=98=E6=97=B6=E4=BB=8E=E7=A1=AC=E7=9B=98=E9=87=8D=E6=96=B0?= =?UTF-8?q?=E5=8A=A0=E8=BD=BD=20prompts/=20=E7=9B=AE=E5=BD=95=E4=B8=8B?= =?UTF-8?q?=E7=9A=84=E6=89=80=E6=9C=89=20.txt=20=E6=A8=A1=E6=9D=BF?= =?UTF-8?q?=E6=96=87=E4=BB=B6=20-=20=E6=B7=BB=E5=8A=A0=E5=AE=8C=E6=95=B4?= =?UTF-8?q?=E7=9A=84=E5=8D=95=E5=85=83=E6=B5=8B=E8=AF=95=E5=92=8C=E7=AB=AF?= =?UTF-8?q?=E5=88=B0=E7=AB=AF=E9=9B=86=E6=88=90=E6=B5=8B=E8=AF=95=20##=20?= =?UTF-8?q?=E6=B5=8B=E8=AF=95=E8=A6=86=E7=9B=96=20-=20=E5=8D=95=E5=85=83?= =?UTF-8?q?=E6=B5=8B=E8=AF=95=EF=BC=9A=E6=A8=A1=E6=9D=BF=E5=8A=A0=E8=BD=BD?= =?UTF-8?q?=E3=80=81=E8=8E=B7=E5=8F=96=E3=80=81=E9=87=8D=E6=96=B0=E5=8A=A0?= =?UTF-8?q?=E8=BD=BD=E5=8A=9F=E8=83=BD=20-=20=E9=9B=86=E6=88=90=E6=B5=8B?= =?UTF-8?q?=E8=AF=95=EF=BC=9A=E6=96=87=E4=BB=B6=E4=BF=AE=E6=94=B9=20?= =?UTF-8?q?=E2=86=92=20=E9=87=8D=E6=96=B0=E5=8A=A0=E8=BD=BD=20=E2=86=92=20?= =?UTF-8?q?=E5=86=B3=E7=AD=96=E5=BC=95=E6=93=8E=E4=BD=BF=E7=94=A8=E6=96=B0?= =?UTF-8?q?=E5=86=85=E5=AE=B9=E7=9A=84=E5=AE=8C=E6=95=B4=E6=B5=81=E7=A8=8B?= =?UTF-8?q?=20-=20=E5=B9=B6=E5=8F=91=E6=B5=8B=E8=AF=95=EF=BC=9A=E9=AA=8C?= =?UTF-8?q?=E8=AF=81=E5=A4=9A=20goroutine=20=E5=9C=BA=E6=99=AF=E4=B8=8B?= =?UTF-8?q?=E7=9A=84=E7=BA=BF=E7=A8=8B=E5=AE=89=E5=85=A8=E6=80=A7=20-=20Ra?= =?UTF-8?q?ce=20detector=20=E6=B5=8B=E8=AF=95=E9=80=9A=E8=BF=87=20##=20?= =?UTF-8?q?=E7=94=A8=E6=88=B7=E4=BD=93=E9=AA=8C=E6=94=B9=E8=BF=9B=20-=20?= =?UTF-8?q?=E4=BF=AE=E6=94=B9=20prompt=20=E6=96=87=E4=BB=B6=E5=90=8E?= =?UTF-8?q?=E6=97=A0=E9=9C=80=E9=87=8D=E5=90=AF=E6=9C=8D=E5=8A=A1=20-=20?= =?UTF-8?q?=E5=8F=AA=E9=9C=80=E5=81=9C=E6=AD=A2=E4=BA=A4=E6=98=93=E5=91=98?= =?UTF-8?q?=E5=86=8D=E5=90=AF=E5=8A=A8=E5=8D=B3=E5=8F=AF=E5=BA=94=E7=94=A8?= =?UTF-8?q?=E6=96=B0=E7=9A=84=20prompt=20-=20=E6=8E=A7=E5=88=B6=E5=8F=B0?= =?UTF-8?q?=E4=BC=9A=E8=BE=93=E5=87=BA=E9=87=8D=E6=96=B0=E5=8A=A0=E8=BD=BD?= =?UTF-8?q?=E6=88=90=E5=8A=9F=E7=9A=84=E6=97=A5=E5=BF=97=E6=8F=90=E7=A4=BA?= =?UTF-8?q?=20*=20feat:=20=E5=9C=A8=E9=87=8D=E6=96=B0=E5=8A=A0=E8=BD=BD?= =?UTF-8?q?=E6=97=A5=E5=BF=97=E4=B8=AD=E6=98=BE=E7=A4=BA=E5=BD=93=E5=89=8D?= =?UTF-8?q?=E4=BD=BF=E7=94=A8=E7=9A=84=E6=A8=A1=E6=9D=BF=E5=90=8D=E7=A7=B0?= =?UTF-8?q?=20*=20feat:=20fallback=20=E5=88=B0=20default=20=E6=A8=A1?= =?UTF-8?q?=E6=9D=BF=E6=97=B6=E6=98=8E=E7=A1=AE=E6=98=BE=E7=A4=BA=E5=8E=9F?= =?UTF-8?q?=E5=9B=A0=20*=20fix:=20correct=20GetTraderConfig=20return=20typ?= =?UTF-8?q?e=20to=20get=20SystemPromptTemplate=20*=20refactor:=20extract?= =?UTF-8?q?=20reloadPromptTemplatesWithLog=20as=20reusable=20method?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/server.go | 22 +- decision/prompt_manager_test.go | 285 +++++++++++++++++++++ decision/prompt_reload_integration_test.go | 243 ++++++++++++++++++ 3 files changed, 549 insertions(+), 1 deletion(-) create mode 100644 decision/prompt_manager_test.go create mode 100644 decision/prompt_reload_integration_test.go diff --git a/api/server.go b/api/server.go index bc6f0986..c758da3d 100644 --- a/api/server.go +++ b/api/server.go @@ -792,12 +792,15 @@ func (s *Server) handleStartTrader(c *gin.Context) { traderID := c.Param("id") // 校验交易员是否属于当前用户 - _, _, _, err := s.database.GetTraderConfig(userID, traderID) + traderRecord, _, _, err := s.database.GetTraderConfig(userID, traderID) if err != nil { c.JSON(http.StatusNotFound, gin.H{"error": "交易员不存在或无访问权限"}) return } + // 获取模板名称 + templateName := traderRecord.SystemPromptTemplate + trader, err := s.traderManager.GetTrader(traderID) if err != nil { c.JSON(http.StatusNotFound, gin.H{"error": "交易员不存在"}) @@ -811,6 +814,9 @@ func (s *Server) handleStartTrader(c *gin.Context) { return } + // 重新加载系统提示词模板（确保使用最新的硬盘文件） + s.reloadPromptTemplatesWithLog(templateName) + // 启动交易员 go func() { log.Printf("▶️ 启动交易员 %s (%s)", traderID, trader.GetName()) @@ -2318,3 +2324,17 @@ func (s *Server) handleGetPublicTraderConfig(c *gin.Context) { c.JSON(http.StatusOK, result) } + +// reloadPromptTemplatesWithLog 重新加载提示词模板并记录日志 +func (s *Server) reloadPromptTemplatesWithLog(templateName string) { + if err := decision.ReloadPromptTemplates(); err != nil { + log.Printf("⚠️ 重新加载提示词模板失败: %v", err) + return + } + + if templateName == "" { + log.Printf("✓ 已重新加载系统提示词模板 [当前使用: default (未指定，使用默认)]") + } else { + log.Printf("✓ 已重新加载系统提示词模板 [当前使用: %s]", templateName) + } +} diff --git a/decision/prompt_manager_test.go b/decision/prompt_manager_test.go new file mode 100644 index 00000000..56f905ba --- /dev/null +++ b/decision/prompt_manager_test.go @@ -0,0 +1,285 @@ +package decision + +import ( + "os" + "path/filepath" + "testing" +) + +func TestPromptManager_LoadTemplates(t *testing.T) { + // 创建临时目录用于测试 + tempDir := t.TempDir() + + tests := []struct { + name string + setupFiles map[string]string // 文件名 -> 内容 + expectedCount int + expectedNames []string + shouldError bool + }{ + { + name: "加载单个模板文件", + setupFiles: map[string]string{ + "default.txt": "你是专业的加密货币交易AI。", + }, + expectedCount: 1, + expectedNames: []string{"default"}, + shouldError: false, + }, + { + name: "加载多个模板文件", + setupFiles: map[string]string{ + "default.txt": "默认策略", + "conservative.txt": "保守策略", + "aggressive.txt": "激进策略", + }, + expectedCount: 3, + expectedNames: []string{"default", "conservative", "aggressive"}, + shouldError: false, + }, + { + name: "空目录", + setupFiles: map[string]string{}, + expectedCount: 0, + expectedNames: []string{}, + shouldError: false, + }, + { + name: "忽略非.txt文件", + setupFiles: map[string]string{ + "default.txt": "正确的模板", + "readme.md": "应该被忽略", + "config.json": "应该被忽略", + }, + expectedCount: 1, + expectedNames: []string{"default"}, + shouldError: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + // 为每个测试用例创建独立的子目录 + testDir := filepath.Join(tempDir, tt.name) + if err := os.MkdirAll(testDir, 0755); err != nil { + t.Fatalf("创建测试目录失败: %v", err) + } + + // 设置测试文件 + for filename, content := range tt.setupFiles { + filePath := filepath.Join(testDir, filename) + if err := os.WriteFile(filePath, []byte(content), 0644); err != nil { + t.Fatalf("创建测试文件失败 %s: %v", filename, err) + } + } + + // 创建新的 PromptManager + pm := NewPromptManager() + + // 执行测试 + err := pm.LoadTemplates(testDir) + + // 检查错误 + if (err != nil) != tt.shouldError { + t.Errorf("LoadTemplates() error = %v, shouldError %v", err, tt.shouldError) + return + } + + // 检查加载的模板数量 + if len(pm.templates) != tt.expectedCount { + t.Errorf("加载的模板数量 = %d, 期望 %d", len(pm.templates), tt.expectedCount) + } + + // 检查模板名称 + for _, expectedName := range tt.expectedNames { + if _, exists := pm.templates[expectedName]; !exists { + t.Errorf("缺少预期的模板: %s", expectedName) + } + } + + // 验证模板内容 + for filename, expectedContent := range tt.setupFiles { + if filepath.Ext(filename) != ".txt" { + continue + } + templateName := filename[:len(filename)-4] // 去掉 .txt + template, err := pm.GetTemplate(templateName) + if err != nil { + t.Errorf("获取模板 %s 失败: %v", templateName, err) + continue + } + if template.Content != expectedContent { + t.Errorf("模板内容不匹配\n期望: %s\n实际: %s", expectedContent, template.Content) + } + } + }) + } +} + +func TestPromptManager_GetTemplate(t *testing.T) { + pm := NewPromptManager() + pm.templates = map[string]*PromptTemplate{ + "default": { + Name: "default", + Content: "默认策略内容", + }, + "aggressive": { + Name: "aggressive", + Content: "激进策略内容", + }, + } + + tests := []struct { + name string + templateName string + expectError bool + expectedContent string + }{ + { + name: "获取存在的模板", + templateName: "default", + expectError: false, + expectedContent: "默认策略内容", + }, + { + name: "获取不存在的模板", + templateName: "nonexistent", + expectError: true, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + template, err := pm.GetTemplate(tt.templateName) + + if (err != nil) != tt.expectError { + t.Errorf("GetTemplate() error = %v, expectError %v", err, tt.expectError) + return + } + + if !tt.expectError && template.Content != tt.expectedContent { + t.Errorf("模板内容 = %s, 期望 %s", template.Content, tt.expectedContent) + } + }) + } +} + +func TestPromptManager_ReloadTemplates(t *testing.T) { + tempDir := t.TempDir() + + // 初始文件 + if err := os.WriteFile(filepath.Join(tempDir, "default.txt"), []byte("初始内容"), 0644); err != nil { + t.Fatalf("创建初始文件失败: %v", err) + } + + pm := NewPromptManager() + if err := pm.LoadTemplates(tempDir); err != nil { + t.Fatalf("初始加载失败: %v", err) + } + + // 验证初始内容 + template, _ := pm.GetTemplate("default") + if template.Content != "初始内容" { + t.Errorf("初始内容不正确: %s", template.Content) + } + + // 修改文件内容 + if err := os.WriteFile(filepath.Join(tempDir, "default.txt"), []byte("更新后内容"), 0644); err != nil { + t.Fatalf("更新文件失败: %v", err) + } + + // 添加新文件 + if err := os.WriteFile(filepath.Join(tempDir, "new.txt"), []byte("新模板内容"), 0644); err != nil { + t.Fatalf("创建新文件失败: %v", err) + } + + // 重新加载 + if err := pm.ReloadTemplates(tempDir); err != nil { + t.Fatalf("重新加载失败: %v", err) + } + + // 验证更新后的内容 + template, err := pm.GetTemplate("default") + if err != nil { + t.Fatalf("获取 default 模板失败: %v", err) + } + if template.Content != "更新后内容" { + t.Errorf("重新加载后内容不正确: got %s, want '更新后内容'", template.Content) + } + + // 验证新模板 + newTemplate, err := pm.GetTemplate("new") + if err != nil { + t.Fatalf("获取 new 模板失败: %v", err) + } + if newTemplate.Content != "新模板内容" { + t.Errorf("新模板内容不正确: %s", newTemplate.Content) + } + + // 验证模板数量 + if len(pm.templates) != 2 { + t.Errorf("重新加载后模板数量 = %d, 期望 2", len(pm.templates)) + } +} + +func TestPromptManager_GetAllTemplateNames(t *testing.T) { + pm := NewPromptManager() + pm.templates = map[string]*PromptTemplate{ + "default": {Name: "default", Content: "默认策略"}, + "conservative": {Name: "conservative", Content: "保守策略"}, + "aggressive": {Name: "aggressive", Content: "激进策略"}, + } + + names := pm.GetAllTemplateNames() + + if len(names) != 3 { + t.Errorf("GetAllTemplateNames() 返回数量 = %d, 期望 3", len(names)) + } + + // 验证所有名称都存在 + nameMap := make(map[string]bool) + for _, name := range names { + nameMap[name] = true + } + + expectedNames := []string{"default", "conservative", "aggressive"} + for _, expectedName := range expectedNames { + if !nameMap[expectedName] { + t.Errorf("缺少预期的模板名称: %s", expectedName) + } + } +} + +func TestReloadPromptTemplates_GlobalFunction(t *testing.T) { + // 保存原始的 promptsDir + originalDir := promptsDir + defer func() { + promptsDir = originalDir + // 恢复原始模板 + globalPromptManager.ReloadTemplates(originalDir) + }() + + // 创建临时目录 + tempDir := t.TempDir() + promptsDir = tempDir + + // 创建测试文件 + if err := os.WriteFile(filepath.Join(tempDir, "test.txt"), []byte("测试内容"), 0644); err != nil { + t.Fatalf("创建测试文件失败: %v", err) + } + + // 调用全局重新加载函数 + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("ReloadPromptTemplates() 失败: %v", err) + } + + // 验证全局管理器已更新 + template, err := GetPromptTemplate("test") + if err != nil { + t.Fatalf("获取模板失败: %v", err) + } + + if template.Content != "测试内容" { + t.Errorf("模板内容不正确: got %s, want '测试内容'", template.Content) + } +} diff --git a/decision/prompt_reload_integration_test.go b/decision/prompt_reload_integration_test.go new file mode 100644 index 00000000..909b3dbb --- /dev/null +++ b/decision/prompt_reload_integration_test.go @@ -0,0 +1,243 @@ +package decision + +import ( + "os" + "path/filepath" + "strings" + "testing" +) + +// TestPromptReloadEndToEnd 端到端测试：验证从文件修改到决策引擎使用的完整流程 +func TestPromptReloadEndToEnd(t *testing.T) { + // 保存原始的 promptsDir + originalDir := promptsDir + defer func() { + promptsDir = originalDir + // 恢复原始模板 + globalPromptManager.ReloadTemplates(originalDir) + }() + + // 创建临时目录模拟 prompts/ 目录 + tempDir := t.TempDir() + promptsDir = tempDir + + // 步骤1: 创建初始 prompt 文件 + initialContent := "# 初始交易策略\n你是一个保守的交易AI。" + if err := os.WriteFile(filepath.Join(tempDir, "test_strategy.txt"), []byte(initialContent), 0644); err != nil { + t.Fatalf("创建初始文件失败: %v", err) + } + + // 步骤2: 首次加载（模拟系统启动） + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("首次加载失败: %v", err) + } + + // 步骤3: 验证初始内容 + template, err := GetPromptTemplate("test_strategy") + if err != nil { + t.Fatalf("获取初始模板失败: %v", err) + } + if template.Content != initialContent { + t.Errorf("初始内容不匹配\n期望: %s\n实际: %s", initialContent, template.Content) + } + + // 步骤4: 使用 buildSystemPrompt 验证模板被正确使用 + systemPrompt := buildSystemPrompt(10000.0, 10, 5, "test_strategy") + if !strings.Contains(systemPrompt, initialContent) { + t.Errorf("buildSystemPrompt 未包含模板内容\n生成的 prompt:\n%s", systemPrompt) + } + + // 步骤5: 模拟用户修改文件（这是用户在硬盘上修改 prompt） + updatedContent := "# 更新的交易策略\n你是一个激进的交易AI，追求高风险高收益。" + if err := os.WriteFile(filepath.Join(tempDir, "test_strategy.txt"), []byte(updatedContent), 0644); err != nil { + t.Fatalf("更新文件失败: %v", err) + } + + // 步骤6: 模拟交易员启动时调用 ReloadPromptTemplates() + t.Log("模拟交易员启动，调用 ReloadPromptTemplates()...") + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("重新加载失败: %v", err) + } + + // 步骤7: 验证新内容已生效 + reloadedTemplate, err := GetPromptTemplate("test_strategy") + if err != nil { + t.Fatalf("获取重新加载的模板失败: %v", err) + } + if reloadedTemplate.Content != updatedContent { + t.Errorf("重新加载后内容不匹配\n期望: %s\n实际: %s", updatedContent, reloadedTemplate.Content) + } + + // 步骤8: 验证 buildSystemPrompt 使用了新内容 + newSystemPrompt := buildSystemPrompt(10000.0, 10, 5, "test_strategy") + if !strings.Contains(newSystemPrompt, updatedContent) { + t.Errorf("buildSystemPrompt 未包含更新后的模板内容\n生成的 prompt:\n%s", newSystemPrompt) + } + + // 步骤9: 验证旧内容不再存在 + if strings.Contains(newSystemPrompt, "保守的交易AI") { + t.Errorf("buildSystemPrompt 仍包含旧的模板内容") + } + + t.Log("✅ 端到端测试通过：文件修改 -> 重新加载 -> 决策引擎使用新内容") +} + +// TestPromptReloadWithCustomPrompt 测试自定义 prompt 与模板重新加载的交互 +func TestPromptReloadWithCustomPrompt(t *testing.T) { + // 保存原始的 promptsDir + originalDir := promptsDir + defer func() { + promptsDir = originalDir + globalPromptManager.ReloadTemplates(originalDir) + }() + + // 创建临时目录 + tempDir := t.TempDir() + promptsDir = tempDir + + // 创建基础模板 + baseContent := "基础策略：稳健交易" + if err := os.WriteFile(filepath.Join(tempDir, "base.txt"), []byte(baseContent), 0644); err != nil { + t.Fatalf("创建文件失败: %v", err) + } + + // 加载模板 + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("加载失败: %v", err) + } + + // 测试1: 基础模板 + 自定义 prompt（不覆盖） + customPrompt := "个性化规则：只交易 BTC" + result := buildSystemPromptWithCustom(10000.0, 10, 5, customPrompt, false, "base") + if !strings.Contains(result, baseContent) { + t.Errorf("未包含基础模板内容") + } + if !strings.Contains(result, customPrompt) { + t.Errorf("未包含自定义 prompt") + } + + // 测试2: 覆盖基础 prompt + result = buildSystemPromptWithCustom(10000.0, 10, 5, customPrompt, true, "base") + if strings.Contains(result, baseContent) { + t.Errorf("覆盖模式下仍包含基础模板内容") + } + if !strings.Contains(result, customPrompt) { + t.Errorf("覆盖模式下未包含自定义 prompt") + } + + // 测试3: 重新加载后效果 + updatedBase := "更新的基础策略：激进交易" + if err := os.WriteFile(filepath.Join(tempDir, "base.txt"), []byte(updatedBase), 0644); err != nil { + t.Fatalf("更新文件失败: %v", err) + } + + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("重新加载失败: %v", err) + } + + result = buildSystemPromptWithCustom(10000.0, 10, 5, customPrompt, false, "base") + if !strings.Contains(result, updatedBase) { + t.Errorf("重新加载后未包含更新的基础模板内容") + } + if strings.Contains(result, baseContent) { + t.Errorf("重新加载后仍包含旧的基础模板内容") + } +} + +// TestPromptReloadFallback 测试模板不存在时的降级机制 +func TestPromptReloadFallback(t *testing.T) { + // 保存原始的 promptsDir + originalDir := promptsDir + defer func() { + promptsDir = originalDir + globalPromptManager.ReloadTemplates(originalDir) + }() + + // 创建临时目录 + tempDir := t.TempDir() + promptsDir = tempDir + + // 只创建 default 模板 + defaultContent := "默认策略" + if err := os.WriteFile(filepath.Join(tempDir, "default.txt"), []byte(defaultContent), 0644); err != nil { + t.Fatalf("创建文件失败: %v", err) + } + + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("加载失败: %v", err) + } + + // 测试1: 请求不存在的模板，应该降级到 default + result := buildSystemPrompt(10000.0, 10, 5, "nonexistent") + if !strings.Contains(result, defaultContent) { + t.Errorf("请求不存在的模板时，未降级到 default") + } + + // 测试2: 空模板名，应该使用 default + result = buildSystemPrompt(10000.0, 10, 5, "") + if !strings.Contains(result, defaultContent) { + t.Errorf("空模板名时，未使用 default") + } +} + +// TestConcurrentPromptReload 测试并发场景下的 prompt 重新加载 +func TestConcurrentPromptReload(t *testing.T) { + // 保存原始的 promptsDir + originalDir := promptsDir + defer func() { + promptsDir = originalDir + globalPromptManager.ReloadTemplates(originalDir) + }() + + // 创建临时目录 + tempDir := t.TempDir() + promptsDir = tempDir + + // 创建测试文件 + if err := os.WriteFile(filepath.Join(tempDir, "test.txt"), []byte("测试内容"), 0644); err != nil { + t.Fatalf("创建文件失败: %v", err) + } + + if err := ReloadPromptTemplates(); err != nil { + t.Fatalf("初始加载失败: %v", err) + } + + // 并发测试：同时读取和重新加载 + done := make(chan bool) + + // 启动多个读取 goroutine + for i := 0; i < 10; i++ { + go func() { + for j := 0; j < 100; j++ { + _, _ = GetPromptTemplate("test") + } + done <- true + }() + } + + // 启动多个重新加载 goroutine + for i := 0; i < 3; i++ { + go func() { + for j := 0; j < 10; j++ { + _ = ReloadPromptTemplates() + } + done <- true + }() + } + + // 等待所有 goroutine 完成 + for i := 0; i < 13; i++ { + <-done + } + + // 验证最终状态正确 + template, err := GetPromptTemplate("test") + if err != nil { + t.Errorf("并发测试后获取模板失败: %v", err) + } + if template.Content != "测试内容" { + t.Errorf("并发测试后模板内容错误: %s", template.Content) + } + + t.Log("✅ 并发测试通过：多个 goroutine 同时读取和重新加载模板，无数据竞争") +}