feat(foundation): Added BootstrapFunc and TerminateFunc

Signed-off-by: Flc゛ <[email protected]>

foundation/provider.go

@@ -16,3 +16,23 @@ func (*BaseProvider) Bootstrap(ctx context.Context) (context.Context, error) {
 func (*BaseProvider) Terminate(ctx context.Context) (context.Context, error) {
 	return ctx, nil
 }
+
+type BootstrapFunc func(context.Context) (context.Context, error)
+
+func (f BootstrapFunc) Bootstrap(ctx context.Context) (context.Context, error) {
+	return f(ctx)
+}
+
+func (f BootstrapFunc) Terminate(ctx context.Context) (context.Context, error) {
+	return ctx, nil
+}
+
+type TerminateFunc func(context.Context) (context.Context, error)
+
+func (f TerminateFunc) Bootstrap(ctx context.Context) (context.Context, error) {
+	return ctx, nil
+}
+
+func (f TerminateFunc) Terminate(ctx context.Context) (context.Context, error) {
+	return f(ctx)
+}

foundation/provider_test.go

@@ -0,0 +1,40 @@
+package foundation
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+type providerFuncKey struct{}
+
+func TestProvider_BootstrapFunc(t *testing.T) {
+	p := BootstrapFunc(func(ctx context.Context) (context.Context, error) {
+		return context.WithValue(ctx, providerFuncKey{}, "test"), nil
+	})
+
+	ctx := context.Background()
+	ctx, err := p.Bootstrap(ctx)
+	assert.NoError(t, err)
+	assert.Equal(t, "test", ctx.Value(providerFuncKey{}))
+
+	ctx, err = p.Terminate(ctx)
+	assert.NoError(t, err)
+	assert.Equal(t, "test", ctx.Value(providerFuncKey{}))
+}
+
+func TestProvider_BootstrapFuncError(t *testing.T) {
+	p := TerminateFunc(func(ctx context.Context) (context.Context, error) {
+		return context.WithValue(ctx, providerFuncKey{}, "test"), nil
+	})
+
+	ctx := context.Background()
+	ctx, err := p.Bootstrap(ctx)
+	assert.NoError(t, err)
+	assert.Nil(t, ctx.Value(providerFuncKey{}))
+
+	ctx, err = p.Terminate(ctx)
+	assert.NoError(t, err)
+	assert.Equal(t, "test", ctx.Value(providerFuncKey{}))
+}