stack/components/payments/cmd/connectors/internal/task/state.go

package task

import (
	"context"

	"github.com/formancehq/payments/cmd/connectors/internal/storage"
	"github.com/pkg/errors"
)

type StateResolver interface {
	ResolveTo(ctx context.Context, v any) error
}
type StateResolverFn func(ctx context.Context, v any) error

func (fn StateResolverFn) ResolveTo(ctx context.Context, v any) error {
	return fn(ctx, v)
}

func ResolveTo[State any](ctx context.Context, resolver StateResolver, to *State) (*State, error) {
	err := resolver.ResolveTo(ctx, to)
	if err != nil {
		return nil, err
	}

	return to, nil
}

func MustResolveTo[State any](ctx context.Context, resolver StateResolver, to State) State {
	state, err := ResolveTo(ctx, resolver, &to)
	if errors.Is(err, storage.ErrNotFound) {
		return to
	}

	if err != nil {
		panic(err)
	}

	return *state
}