Huangzj
36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
package linq
|
|
|
|
// Zip applies a specified function to the corresponding elements
|
|
// of two collections, producing a collection of the results.
|
|
//
|
|
// The method steps through the two input collections, applying function
|
|
// resultSelector to corresponding elements of the two collections.
|
|
// The method returns a collection of the values that are returned by resultSelector.
|
|
// If the input collections do not have the same number of elements,
|
|
// the method combines elements until it reaches the end of one of the collections.
|
|
// For example, if one collection has three elements and the other one has four,
|
|
// the result collection has only three elements.
|
|
func (q Query) Zip(
|
|
q2 Query,
|
|
resultSelector func(interface{}, interface{}) interface{},
|
|
) Query {
|
|
|
|
return Query{
|
|
Iterate: func() Iterator {
|
|
next1 := q.Iterate()
|
|
next2 := q2.Iterate()
|
|
|
|
return func() (item interface{}, ok bool) {
|
|
item1, ok1 := next1()
|
|
item2, ok2 := next2()
|
|
|
|
if ok1 && ok2 {
|
|
return resultSelector(item1, item2), true
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
},
|
|
}
|
|
}
|