core/mr:a little optimization for collector initialization in ForEach function (#1937)

Co-authored-by: notrynosuccess <daihongshan@gmail.com>
master
NoTryNoSuccess 3 years ago committed by GitHub
parent e9501c3fb3
commit fb519fa547
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -102,12 +102,12 @@ func ForEach(generate GenerateFunc, mapper ForEachFunc, opts ...Option) {
options := buildOptions(opts...) options := buildOptions(opts...)
panicChan := &onceChan{channel: make(chan interface{})} panicChan := &onceChan{channel: make(chan interface{})}
source := buildSource(generate, panicChan) source := buildSource(generate, panicChan)
collector := make(chan interface{}, options.workers) collector := make(chan interface{})
done := make(chan lang.PlaceholderType) done := make(chan lang.PlaceholderType)
go executeMappers(mapperContext{ go executeMappers(mapperContext{
ctx: options.ctx, ctx: options.ctx,
mapper: func(item interface{}, writer Writer) { mapper: func(item interface{}, _ Writer) {
mapper(item) mapper(item)
}, },
source: source, source: source,

Loading…
Cancel
Save