40 lines
756 B
Go
40 lines
756 B
Go
|
package stream
|
|||
|
|
|
|||
|
|
import "runtime"
|
|||
|
|
|
|||
|
|
//Map 对任意数组操作，将A包装成B
|
|||
|
|
func (s *Stream) Map(mappper func(interface{}) interface{}) *Stream {
|
|||
|
|
ret := make([]interface{}, len(s.list))
|
|||
|
|
if s.isParallel {
|
|||
|
|
count := runtime.NumCPU()
|
|||
|
|
ch1 := make(chan interface{}, count)
|
|||
|
|
ch2 := make(chan interface{})
|
|||
|
|
for i := 0; i < count; i++ {
|
|||
|
|
go func() {
|
|||
|
|
for true {
|
|||
|
|
if value, ok := <-ch1; !ok {
|
|||
|
|
break
|
|||
|
|
} else {
|
|||
|
|
ch2 <- mappper(value)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
go func() {
|
|||
|
|
for _, v := range s.list {
|
|||
|
|
ch1 <- v
|
|||
|
|
}
|
|||
|
|
close(ch1)
|
|||
|
|
}()
|
|||
|
|
for i := 0; i < len(s.list); i++ {
|
|||
|
|
ret[i] = <-ch2
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
for i, v := range s.list {
|
|||
|
|
ret[i] = mappper(v)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
s.list = ret
|
|||
|
|
return s
|
|||
|
|
}
|