waiterr

waiterr is a Go package that wraps sync.WaitGroup with enhanced error handling capabilities. It allows you to run multiple goroutines, collect any errors they return, and wait for their completion, either returning the first error encountered or aggregating all errors.

Features

Go(f func() error): Runs a function f in a new goroutine, storing any error it returns.
Wait() error: Waits for all goroutines to complete and returns a combined error of all non-nil errors.
WaitForError() error: Waits for the first error to be returned by any goroutine and immediately returns that error. If all goroutines complete without error, it returns nil.
Unwrap() []error: Returns a slice of all non-nil errors encountered by the goroutines.

Installation

To install waiterr, use go get:

go get codeberg.org/danjones000/waiterr

Usage

Here's a basic example of how to use waiterr:

package main

import (
	"errors"
	"fmt"
	"time"

	"codeberg.org/danjones000/waiterr"
)

func main() {
	we := waiterr.New()

	we.Go(func() error {
		time.Sleep(100 * time.Millisecond)
		fmt.Println("Goroutine 1 finished")
		return nil
	})

	we.Go(func() error {
		time.Sleep(50 * time.Millisecond)
		fmt.Println("Goroutine 2 finished with an error")
		return errors.New("something went wrong in goroutine 2")
	})

	we.Go(func() error {
		time.Sleep(150 * time.Millisecond)
		fmt.Println("Goroutine 3 finished")
		return nil
	})

	// Wait for all goroutines and get all errors
	if err := we.Wait(); err != nil {
		fmt.Printf("All goroutines finished. Combined error: %v
", err)
	}

	// You can also get the first error immediately
	we2 := waiterr.New()
	we2.Go(func() error {
		time.Sleep(100 * time.Millisecond)
		return errors.New("first error from we2")
	})
	we2.Go(func() error {
		time.Sleep(50 * time.Millisecond)
		return errors.New("second error from we2")
	})

	if err := we2.WaitForError(); err != nil {
		fmt.Printf("First error from we2: %v
", err)
	}

	// Get all unwrapped errors
	unwrappedErrors := we.Unwrap()
	if len(unwrappedErrors) > 0 {
		fmt.Println("Unwrapped errors:")
		for i, err := range unwrappedErrors {
			fmt.Printf("  %d: %v
", i+1, err)
		}
	}
}

Using WithContext

package main

import (
	"context"
	"fmt"
	"time"

	"codeberg.org/danjones000/waiterr"
)

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	we, ctx := waiterr.WithContext(ctx)

	we.Go(func() error {
		select {
		case <-time.After(100 * time.Millisecond):
			fmt.Println("Task completed")
			return nil
		case <-ctx.Done():
			fmt.Println("Task cancelled")
			return ctx.Err()
		}
	})

	_ = we.Wait()
	// Output:
	// Task completed
}

Contributing

Please refer to the CONTRIBUTING.md file for guidelines on contributing to this project, including code style, commit messages, and Git workflow.

License

This project is licensed under the MIT License. See the LICENSE file for details.

Go Version

Go 1.25.3

Dependencies

github.com/nalgeon/be v0.3.0 for testing.

3.1 KiB Raw Blame History