Chore: adjust batch

This commit is contained in:
Dreamacro
2021-07-23 00:30:23 +08:00
parent 4578b2c826
commit 09697b7679
3 changed files with 8 additions and 13 deletions

View File

@ -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
}