bootstrap_storage.go

package unleash

import (
	"encoding/json"
	"fmt"
	"io"

	"github.com/Unleash/unleash-client-go/v4/api"
)

type BootstrapStorage struct {
	backingStore DefaultStorage
	Reader       io.Reader
}

func (bs *BootstrapStorage) Load() error {
	if len(bs.backingStore.data) > 0 || bs.Reader == nil {
		return nil
	}

	dec := json.NewDecoder(bs.Reader)
	clientFeatures := api.FeatureResponse{}
	if err := dec.Decode(&clientFeatures); err != nil {
		return err
	}

	bs.backingStore.data = clientFeatures.FeatureMap()
	return nil
}

func (bs *BootstrapStorage) Init(backupPath string, appName string) {
	bs.backingStore.Init(backupPath, appName)
	err := bs.Load()

	if err != nil {
		fmt.Printf("Could not load bootstrap storage, because: %s", err.Error())
		return
	}
}

func (bs *BootstrapStorage) Reset(data map[string]interface{}, persist bool) error {
	return bs.backingStore.Reset(data, persist)
}

func (bs *BootstrapStorage) Persist() error {
	return bs.backingStore.Persist()
}

func (bs *BootstrapStorage) Get(key string) (interface{}, bool) {
	return bs.backingStore.Get(key)
}

func (bs *BootstrapStorage) List() []interface{} {
	return bs.backingStore.List()
}