package repository

import (
	"context"

	"golang.org/x/sync/errgroup"
)

// RunWorkers runs count instances of workerFunc using an errgroup.Group.
// After all workers have terminated, finalFunc is run. If an error occurs in
// one of the workers, it is returned. FinalFunc is always run, regardless of
// any other previous errors.
func RunWorkers(ctx context.Context, count int, workerFunc func() error, finalFunc func()) error {
	wg, ctx := errgroup.WithContext(ctx)

	// run workers
	for i := 0; i < count; i++ {
		wg.Go(workerFunc)
	}

	// wait for termination
	err := wg.Wait()

	// make sure finalFunc is run
	finalFunc()

	// if the workers returned an error, return it to the caller
	if err != nil {
		return err
	}

	return nil
}