use vald errgroup

vdaas · Sep 19, 2023 · 4c22bcc · 4c22bcc
1 parent 9a9e622
commit 4c22bcc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/index/job/correction/service/corrector.go b/pkg/index/job/correction/service/corrector.go
@@ -35,8 +35,8 @@ import (
 	"github.com/vdaas/vald/internal/net/grpc/codes"
 	"github.com/vdaas/vald/internal/net/grpc/status"
 	"github.com/vdaas/vald/internal/sync"
+	"github.com/vdaas/vald/internal/sync/errgroup"
 	"github.com/vdaas/vald/pkg/index/job/correction/config"
-	stdeg "golang.org/x/sync/errgroup"
 )
 
 type Corrector interface {
@@ -139,11 +139,11 @@ func (c *correct) correct(ctx context.Context) (err error) {
 				return err
 			}
 
-			seg, ctx := stdeg.WithContext(ctx)
+			seg, ctx := errgroup.WithContext(ctx)
 			concurrency := c.cfg.Corrector.GetStreamListConcurrency()
 			seg.SetLimit(concurrency)
 
-			bolteg, ctx := stdeg.WithContext(ctx)
+			bolteg, ctx := errgroup.WithContext(ctx)
 			bolteg.SetLimit(2048)
 
 			finalize := func() error {