common/pkgs/future/set_void_future.go

package future

import (
	"context"
)

type SetVoidFuture struct {
	err          error
	isCompleted  bool
	completeChan chan any
}

func NewSetVoid() *SetVoidFuture {
	return &SetVoidFuture{
		completeChan: make(chan any),
	}
}

func (f *SetVoidFuture) SetVoid() {
	f.isCompleted = true
	close(f.completeChan)
}

func (f *SetVoidFuture) SetError(err error) {
	f.err = err
	f.isCompleted = true
	close(f.completeChan)
}

func (f *SetVoidFuture) Error() error {
	return f.err
}

func (f *SetVoidFuture) IsComplete() bool {
	return f.isCompleted
}

func (f *SetVoidFuture) Wait(ctx context.Context) error {
	select {
	case <-f.completeChan:
		return f.err

	case <-ctx.Done():
		return ErrContextCancelled
	}
}