Chore: adjust batch

adapter/provider/healthcheck.go

@@ -59,7 +59,7 @@ func (hc *HealthCheck) touch() {
 }
 
 func (hc *HealthCheck) check() {
-	b := batch.New(batch.WithConcurrencyNum(10))
+	b, _ := batch.New(context.Background(), batch.WithConcurrencyNum(10))
 	for _, proxy := range hc.proxies {
 		p := proxy
 		b.Go(p.Name(), func() (interface{}, error) {

common/batch/batch.go

@@ -89,7 +89,9 @@ func (b *Batch) Result() map[string]Result {
 	return copy
 }
 
-func New(opts ...Option) *Batch {
+func New(ctx context.Context, opts ...Option) (*Batch, context.Context) {
+	ctx, cancel := context.WithCancel(ctx)
+
 	b := &Batch{
 		result: map[string]Result{},
 	}
@@ -98,14 +100,6 @@ func New(opts ...Option) *Batch {
 		o(b)
 	}
 
-	return b
-}
-
-func WithContext(ctx context.Context, opts ...Option) (*Batch, context.Context) {
-	ctx, cancel := context.WithCancel(ctx)
-
-	b := New(opts...)
 	b.cancel = cancel
-
 	return b, ctx
 }

common/batch/batch_test.go

@@ -11,7 +11,7 @@ import (
 )
 
 func TestBatch(t *testing.T) {
-	b := New()
+	b, _ := New(context.Background())
 
 	now := time.Now()
 	b.Go("foo", func() (interface{}, error) {
@@ -37,7 +37,8 @@ func TestBatch(t *testing.T) {
 }
 
 func TestBatchWithConcurrencyNum(t *testing.T) {
-	b := New(
+	b, _ := New(
+		context.Background(),
 		WithConcurrencyNum(3),
 	)
 
@@ -61,7 +62,7 @@ func TestBatchWithConcurrencyNum(t *testing.T) {
 }
 
 func TestBatchContext(t *testing.T) {
-	b, ctx := WithContext(context.Background())
+	b, ctx := New(context.Background())
 
 	b.Go("error", func() (interface{}, error) {
 		time.Sleep(time.Millisecond * 100)