groupjoin.go

package go2linq

import (
	"iter"
	"sync"

	"github.com/solsw/errorhelper"
	"github.com/solsw/generichelper"
)

// [GroupJoin] correlates the elements of two sequences based on equality of keys and groups the results.
// [generichelper.DeepEqual] is used to compare keys.
// 'inner' is enumerated on the first iteration over the result.
//
// [GroupJoin]: https://learn.microsoft.com/dotnet/api/system.linq.enumerable.groupjoin
func GroupJoin[Outer, Inner, Key, Result any](outer iter.Seq[Outer], inner iter.Seq[Inner], outerKeySelector func(Outer) Key,
	innerKeySelector func(Inner) Key, resultSelector func(Outer, iter.Seq[Inner]) Result) (iter.Seq[Result], error) {
	if outer == nil || inner == nil {
		return nil, errorhelper.CallerError(ErrNilSource)
	}
	if outerKeySelector == nil || innerKeySelector == nil || resultSelector == nil {
		return nil, errorhelper.CallerError(ErrNilSelector)
	}
	r, err := GroupJoinEq(outer, inner, outerKeySelector, innerKeySelector, resultSelector, generichelper.DeepEqual[Key])
	if err != nil {
		return nil, errorhelper.CallerError(err)
	}
	return r, nil
}

// [GroupJoinEq] correlates the elements of two sequences based on key equality and groups the results.
// 'keyEqual' is used to compare keys. 'inner' is enumerated on the first iteration over the result.
//
// [GroupJoinEq]: https://learn.microsoft.com/dotnet/api/system.linq.enumerable.groupjoin
func GroupJoinEq[Outer, Inner, Key, Result any](outer iter.Seq[Outer], inner iter.Seq[Inner],
	outerKeySelector func(Outer) Key, innerKeySelector func(Inner) Key,
	resultSelector func(Outer, iter.Seq[Inner]) Result, keyEqual func(Key, Key) bool) (iter.Seq[Result], error) {
	if outer == nil || inner == nil {
		return nil, errorhelper.CallerError(ErrNilSource)
	}
	if outerKeySelector == nil || innerKeySelector == nil || resultSelector == nil {
		return nil, errorhelper.CallerError(ErrNilSelector)
	}
	if keyEqual == nil {
		return nil, errorhelper.CallerError(ErrNilEqual)
	}
	return func(yield func(Result) bool) {
			var once sync.Once
			var ilk *Lookup[Key, Inner]
			for o := range outer {
				once.Do(func() { ilk, _ = ToLookupEq(inner, innerKeySelector, keyEqual) })
				if !yield(resultSelector(o, ilk.Item(outerKeySelector(o)))) {
					return
				}
			}
		},
		nil
}