zip.go

package flinx

// 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 Zip[T, V, O any](resultSelector func(T, V) O) func(q Query[T], q2 Query[V]) Query[O] {
	return func(q Query[T], q2 Query[V]) Query[O] {
		return Query[O]{
			Iterate: func() Iterator[O] {
				next1 := q.Iterate()
				next2 := q2.Iterate()

				return func() (item O, ok bool) {
					item1, ok1 := next1()
					item2, ok2 := next2()

					if ok1 && ok2 {
						return resultSelector(item1, item2), true
					}
					var o O
					return o, false
				}
			},
		}
	}

}