feat: 增强 LSP 服务并发安全性并优化多文件诊断逻辑

- **LSP 服务增强**：
    - **多文件诊断推送**：重构 `UpdateSession` 和 `rebuildPackage` 逻辑，支持跨文件聚合错误信息。现在当包内任一文件发生变化时，会全量更新并推送该包下所有相关文件的诊断信息。
    - **并发安全写入**：在 `lsp-server` 中引入 `sync.Mutex` 保护标准输出，并采用 Goroutine 异步处理消息，提升服务器响应性能与稳定性。
    - **异常恢复**：在消息处理循环中增加 `recover` 机制，防止特定脚本解析异常导致整个 LSP 进程崩溃。
- **补全逻辑优化**：
    - **切片安全性增强**：在 `core/ast/query.go` 中进行成员补全时，对 `Params` 切片头进行克隆操作，确保剥离接收者（Receiver）的操作仅影响局部副本，避免污染原始类型定义。
- **重构与清理**：
    - 简化了 `UpdateSession` 的内部流程，移除冗余的诊断过滤逻辑，统一采用包级重新构建模式。

Author: ExplodingDragon

cmd/lsp-server/main.go

@@ -8,6 +8,7 @@ import (
 	"io"
 	"os"
 	"strings"
+	"sync"
 
 	engine "gopkg.d7z.net/go-mini/core"
 	"gopkg.d7z.net/go-mini/core/lspserv"
@@ -33,6 +34,14 @@ func main() {
 	server := lspserv.NewLSPServer(executor)
 	reader := bufio.NewReader(os.Stdin)
 
+	var mu sync.Mutex
+	writeMessage := func(msg interface{}) {
+		mu.Lock()
+		defer mu.Unlock()
+		body, _ := json.Marshal(msg)
+		fmt.Printf("Content-Length: %d\r\n\r\n%s", len(body), body)
+	}
+
 	for {
 		msg, err := readMessage(reader)
 		if err != nil {
@@ -43,7 +52,14 @@ func main() {
 			continue
 		}
 
-		handleMessage(server, msg)
+		go func(m *rpcMessage) {
+			defer func() {
+				if r := recover(); r != nil {
+					fmt.Fprintf(os.Stderr, "LSP Panic recovered: %v\n", r)
+				}
+			}()
+			handleMessage(server, m, writeMessage)
+		}(msg)
 	}
 }
 
@@ -80,12 +96,7 @@ func readMessage(r *bufio.Reader) (*rpcMessage, error) {
 	return &msg, nil
 }
 
-func writeMessage(msg interface{}) {
-	body, _ := json.Marshal(msg)
-	fmt.Printf("Content-Length: %d\r\n\r\n%s", len(body), body)
-}
-
-func handleMessage(server *lspserv.LSPServer, msg *rpcMessage) {
+func handleMessage(server *lspserv.LSPServer, msg *rpcMessage, writeMessage func(interface{})) {
 	switch msg.Method {
 	case "initialize":
 		writeMessage(rpcMessage{
@@ -125,14 +136,16 @@ func handleMessage(server *lspserv.LSPServer, msg *rpcMessage) {
 			code = params.ContentChanges[0].Text
 		}
 
-		diagnostics, _ := server.UpdateSession(uri, code)
+		allDiagnostics, _ := server.UpdateSession(uri, code)
 
-		// 发送异步诊断推送
-		writeMessage(rpcMessage{
-			JSONRPC: "2.0",
-			Method:  "textDocument/publishDiagnostics",
-			Params:  json.RawMessage(fmt.Sprintf(`{"uri":"%s","diagnostics":%s}`, uri, mustMarshal(diagnostics))),
-		})
+		// 为包内所有受影响的文件发布诊断
+		for fURI, diags := range allDiagnostics {
+			writeMessage(rpcMessage{
+				JSONRPC: "2.0",
+				Method:  "textDocument/publishDiagnostics",
+				Params:  json.RawMessage(fmt.Sprintf(`{"uri":"%s","diagnostics":%s}`, fURI, mustMarshal(diags))),
+			})
+		}
 
 	case "textDocument/completion":
 		var params struct {

core/ast/query.go

@@ -833,10 +833,13 @@ func getMemberCompletions(ctx *ValidContext, obj Expr) []CompletionItem {
 			items = append(items, CompletionItem{Label: string(f), Kind: "field", Type: t})
 		}
 		for m, t := range st.Methods {
+			// t 是结构体副本，但 Params 是切片（引用）。
+			// 为了绝对安全，我们克隆一份切片头。
 			sig := t
 			// 剥离接收者以便在补全中显示正确的参数列表
 			if objType != "Package" && objType != TypeModule {
 				if len(sig.Params) > 0 {
+					// 这里的赋值只修改 local sig 的切片头，不会写回 st.Methods
 					sig.Params = sig.Params[1:]
 				}
 			}

core/lspserv/server.go

@@ -37,24 +37,21 @@ func NewLSPServer(e *engine.MiniExecutor) *LSPServer {
 	}
 }
 
-// UpdateSession 更新或创建脚本会话并返回诊断信息
-func (s *LSPServer) UpdateSession(uri, code string) ([]Diagnostic, error) {
-	// 1. 尝试获取包名（简单正则或初步解析）
+// UpdateSession 更新或创建脚本会话并返回诊断信息。
+func (s *LSPServer) UpdateSession(uri, code string) (map[string][]Diagnostic, error) {
 	pkgName := "main"
 	converter := ffigo.NewGoToASTConverter()
 	node, _ := converter.ConvertSourceTolerant(uri, code)
 	if prog, ok := node.(*ast.ProgramStmt); ok && prog.Package != "" {
 		pkgName = prog.Package
 	}
 
-	// 2. 存储文件会话
 	s.sessions.Store(uri, &fileSession{pkgName: pkgName, code: code})
 
-	// 3. 触发包级重新聚合分析
-	return s.rebuildPackage(pkgName, uri)
+	return s.rebuildPackage(pkgName)
 }
 
-func (s *LSPServer) rebuildPackage(pkgName, targetURI string) ([]Diagnostic, error) {
+func (s *LSPServer) rebuildPackage(pkgName string) (map[string][]Diagnostic, error) {
 	val, _ := s.packages.LoadOrStore(pkgName, &packageState{files: make(map[string]string)})
 	pkg := val.(*packageState)
 	pkg.mu.Lock()
@@ -69,10 +66,9 @@ func (s *LSPServer) rebuildPackage(pkgName, targetURI string) ([]Diagnostic, err
 		return true
 	})
 
-	// 聚合所有文件到单个 ProgramStmt
 	var combinedNode *ast.ProgramStmt
 	converter := ffigo.NewGoToASTConverter()
-	diagnostics := make([]Diagnostic, 0)
+	allDiagnostics := make(map[string][]Diagnostic)
 
 	for uri, code := range pkg.files {
 		node, errs := converter.ConvertSourceTolerant(uri, code)
@@ -81,19 +77,17 @@ func (s *LSPServer) rebuildPackage(pkgName, targetURI string) ([]Diagnostic, err
 		for _, err := range errs {
 			var scanErr scanner.Error
 			if errors.As(err, &scanErr) {
-				if scanErr.Pos.Filename == targetURI {
-					diagnostics = append(diagnostics, Diagnostic{
-						Range: Range{
-							Start: Position{Line: scanErr.Pos.Line - 1, Character: scanErr.Pos.Column - 1},
-							End:   Position{Line: scanErr.Pos.Line - 1, Character: scanErr.Pos.Column},
-						},
-						Severity: 1,
-						Source:   "go-mini-syntax",
-						Message:  scanErr.Msg,
-					})
-				}
-			} else if err != nil && uri == targetURI {
-				diagnostics = append(diagnostics, Diagnostic{
+				allDiagnostics[uri] = append(allDiagnostics[uri], Diagnostic{
+					Range: Range{
+						Start: Position{Line: scanErr.Pos.Line - 1, Character: scanErr.Pos.Column - 1},
+						End:   Position{Line: scanErr.Pos.Line - 1, Character: scanErr.Pos.Column},
+					},
+					Severity: 1,
+					Source:   "go-mini-syntax",
+					Message:  scanErr.Msg,
+				})
+			} else if err != nil {
+				allDiagnostics[uri] = append(allDiagnostics[uri], Diagnostic{
 					Range:    FromInternalPos(&ast.Position{L: 1, C: 1}),
 					Severity: 1,
 					Source:   "go-mini-syntax",
@@ -106,69 +100,56 @@ func (s *LSPServer) rebuildPackage(pkgName, targetURI string) ([]Diagnostic, err
 			if combinedNode == nil {
 				combinedNode = prog
 			} else {
-				// 合并符号（借用我们在 cmd/exec 中定义的逻辑思路）
 				mergeProgramStmts(combinedNode, prog)
 			}
 		}
 	}
 
 	if combinedNode == nil {
-		return diagnostics, nil
+		return allDiagnostics, nil
 	}
 
 	// 运行校验以生成持久化 Scope
 	prog, errs := s.executor.NewMiniProgramByProgramTolerant(combinedNode)
 	pkg.combined = prog
 
-	// 将当前触发文件的诊断信息返回
+	// 收集所有文件的诊断信息
 	for _, err := range errs {
 		if astErr, ok := err.(*ast.MiniAstError); ok {
 			for _, log := range astErr.Logs {
 				loc := log.Node.GetBase().Loc
-				// 严格过滤文件路径，确保错误显示在正确的文件中
-				if loc != nil && loc.F == targetURI {
+				if loc != nil && loc.F != "" {
 					diag := Diagnostic{
 						Range:    FromInternalPos(loc),
 						Severity: 1,
 						Source:   "go-mini",
 						Message:  log.Message,
 					}
-					diagnostics = append(diagnostics, diag)
+					allDiagnostics[loc.F] = append(allDiagnostics[loc.F], diag)
 				}
 			}
 		} else {
-			// 处理其他类型的运行时或执行错误
+			// 处理运行时错误
 			var vme *runtime.VMError
 			if errors.As(err, &vme) {
-				if len(vme.Frames) > 0 && vme.Frames[0].Filename == targetURI {
+				if len(vme.Frames) > 0 {
+					f := vme.Frames[0]
 					diag := Diagnostic{
 						Range: FromInternalPos(&ast.Position{
-							L: vme.Frames[0].Line,
-							C: vme.Frames[0].Column,
+							L: f.Line,
+							C: f.Column,
 						}),
 						Severity: 1,
 						Source:   "go-mini-runtime",
 						Message:  vme.Message,
 					}
-					for _, f := range vme.Frames {
-						diag.RelatedInformation = append(diag.RelatedInformation, DiagnosticRelatedInformation{
-							Location: Location{
-								URI: f.Filename,
-								Range: FromInternalPos(&ast.Position{
-									L: f.Line, C: f.Column,
-									EL: f.Line, EC: f.Column + 1,
-								}),
-							},
-							Message: fmt.Sprintf("at %s()", f.Function),
-						})
-					}
-					diagnostics = append(diagnostics, diag)
+					allDiagnostics[f.Filename] = append(allDiagnostics[f.Filename], diag)
 				}
 			}
 		}
 	}
 
-	return diagnostics, nil
+	return allDiagnostics, nil
 }
 
 func mergeProgramStmts(dest, src *ast.ProgramStmt) {