perf(lsp): use shared timeout for parallel diagnostics collection

Christian Rocha created 2 weeks ago

Change summary

internal/agent/tools/diagnostics.go | 12 +++++++++++-
1 file changed, 11 insertions(+), 1 deletion(-)

Detailed changes

internal/agent/tools/diagnostics.go

@@ -7,6 +7,7 @@ import (
 	"log/slog"
 	"sort"
 	"strings"
+	"sync"
 	"time"
 
 	"charm.land/fantasy"
@@ -72,12 +73,21 @@ func waitForLSPDiagnostics(
 		return
 	}
 
+	waitCtx, cancel := context.WithTimeout(ctx, timeout)
+	defer cancel()
+
+	var wg sync.WaitGroup
 	for client := range manager.Clients().Seq() {
 		if !client.HandlesFile(filepath) {
 			continue
 		}
-		client.WaitForDiagnostics(ctx, timeout)
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			client.WaitForDiagnostics(waitCtx, timeout)
+		}()
 	}
+	wg.Wait()
 }
 
 // notifyLSPs notifies LSP servers that a file has changed and waits for