zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: notrynosuccess <daihongshan@gmail.com>
This commit is contained in:
NoTryNoSuccess
2022-05-27 17:19:40 +08:00
committed by GitHub
parent e9501c3fb3
commit fb519fa547
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/mr/mapreduce.go
View File

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