A method of propagating cancellation signals to a context

propagate_cancel.go

package signalprop

import (
	"context"
)

func PropagateCancel(ctx context.Context, done chan struct{}) (context.Context, context.CancelFunc) {
	ctx, cancel := context.WithCancel(ctx)
	go func() {
		select {
		case <-ctx.Done():
		case <-done:
			cancel()
		}
	}()
	return ctx, cancel
}