fix: use context cancellation for quick stop

ldez · ldez · commit 4562edd25fb8 · 2025-07-10T18:13:04.000+02:00
diff --git a/pkg/goanalysis/runner_action.go b/pkg/goanalysis/runner_action.go
@@ -1,6 +1,7 @@
 package goanalysis
 
 import (
+	"context"
 	"fmt"
 	"runtime/debug"
 
@@ -28,11 +29,12 @@ func (actAlloc *actionAllocator) alloc() *action {
 	return act
 }
 
-func (act *action) waitUntilDependingAnalyzersWorked(stopChan chan struct{}) {
+func (act *action) waitUntilDependingAnalyzersWorked(ctx context.Context, stopChan chan struct{}) {
 	for _, dep := range act.Deps {
 		if dep.Package == act.Package {
 			select {
 			case <-stopChan:
+			case <-ctx.Done():
 				return
 			case <-dep.analysisDoneCh:
 			}
diff --git a/pkg/goanalysis/runner_loadingpackage.go b/pkg/goanalysis/runner_loadingpackage.go
@@ -1,6 +1,7 @@
 package goanalysis
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"go/ast"
@@ -84,15 +85,17 @@ func (lp *loadingPackage) analyze(stopChan chan struct{}, loadMode LoadMode, loa
 		return
 	}
 
-	var actsWg errgroup.Group
+	actsWg, ctx := errgroup.WithContext(context.Background())
 
 	for _, act := range lp.actions {
 		actsWg.Go(func() error {
-			act.waitUntilDependingAnalyzersWorked(stopChan)
+			act.waitUntilDependingAnalyzersWorked(ctx, stopChan)
 
 			select {
 			case <-stopChan:
 				return nil
+			case <-ctx.Done():
+				return nil
 			default:
 			}
 
