Use thread control

smartcontractkit · Oct 17, 2023 · 996acf0 · 996acf0
1 parent a5f915f
commit 996acf0
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/core/services/ocr2/plugins/ocr2keeper/evm21/registry_check_pipeline.go b/core/services/ocr2/plugins/ocr2keeper/evm21/registry_check_pipeline.go
@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"math/big"
 	"strings"
-	"sync"
 
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/common/hexutil"
@@ -41,16 +40,11 @@ func (r *EvmRegistry) CheckUpkeeps(ctx context.Context, keys ...ocr2keepers.Upke
 		}
 	}
 
-	var wg sync.WaitGroup
-	wg.Add(1)
-	defer wg.Wait()
-
 	chResult := make(chan checkResult, 1)
 
-	go func() {
-		defer wg.Done()
+	r.threadCtrl.Go(func(ctx context.Context) {
 		r.doCheck(ctx, keys, chResult)
-	}()
+	})
 
 	select {
 	case rs := <-chResult: