cool util for starting alot of gorutines with error handling

workload.go

package workload

import (
	"context"
	"golang.org/x/sync/errgroup"
	"time"
)

type Worker func() error

func All(workers []Worker, timeout time.Duration) error {
	ctx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()
	errs, ctx := errgroup.WithContext(ctx)

	for _, worker := range workers {
		errs.Go(worker)
	}

	return errs.Wait()
}

type Builder struct {
	workers []Worker
}

func NewBuilder() *Builder {
	return &Builder{[]Worker{}}
}

func (b *Builder) Add(worker Worker) {
	b.workers = append(b.workers, worker)
}

func (b *Builder) RunAll(times ...time.Duration) error {
	if len(b.workers) == 0 {
		return nil
	}

	var timeout time.Duration
	if len(times) == 0 {
		timeout = time.Second * 5
	} else {
		timeout = times[0]
	}

	return All(b.workers, timeout)
}