Huangzj
56 lines
1.7 KiB
Go
56 lines
1.7 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
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
// ZipT is the typed version of Zip.
|
|
//
|
|
// - resultSelectorFn is of type "func(TFirst,TSecond)TResult"
|
|
//
|
|
// NOTE: Zip has better performance than ZipT.
|
|
func (q Query) ZipT(q2 Query,
|
|
resultSelectorFn interface{}) Query {
|
|
resultSelectorGenericFunc, err := newGenericFunc(
|
|
"ZipT", "resultSelectorFn", resultSelectorFn,
|
|
simpleParamValidator(newElemTypeSlice(new(genericType), new(genericType)), newElemTypeSlice(new(genericType))),
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
resultSelectorFunc := func(item1 interface{}, item2 interface{}) interface{} {
|
|
return resultSelectorGenericFunc.Call(item1, item2)
|
|
}
|
|
|
|
return q.Zip(q2, resultSelectorFunc)
|
|
}
|