any-sync/client/storage/storageservice.go

package storage

import (
	"context"
	"github.com/anytypeio/go-anytype-infrastructure-experiments/client/badgerprovider"
	"github.com/anytypeio/go-anytype-infrastructure-experiments/common/app"
	"github.com/anytypeio/go-anytype-infrastructure-experiments/common/commonspace/spacestorage"
	"github.com/dgraph-io/badger/v3"
)

type storageService struct {
	keys storageServiceKeys
	db   *badger.DB
}

type ClientStorage interface {
	spacestorage.SpaceStorageProvider
	app.ComponentRunnable
	AllSpaceIds() (ids []string, err error)
}

func New() ClientStorage {
	return &storageService{}
}

func (s *storageService) Init(a *app.App) (err error) {
	provider := a.MustComponent(badgerprovider.CName).(badgerprovider.BadgerProvider)
	s.db = provider.Badger()
	s.keys = newStorageServiceKeys()
	return
}

func (s *storageService) Name() (name string) {
	return spacestorage.CName
}

func (s *storageService) SpaceStorage(id string) (spacestorage.SpaceStorage, error) {
	return newSpaceStorage(s.db, id)
}

func (s *storageService) SpaceExists(id string) bool {
	return s.db.View(func(txn *badger.Txn) error {
		_, err := getTxn(txn, newSpaceKeys(id).HeaderKey())
		if err != nil {
			return err
		}
		return nil
	}) == nil
}

func (s *storageService) CreateSpaceStorage(payload spacestorage.SpaceStorageCreatePayload) (spacestorage.SpaceStorage, error) {
	return createSpaceStorage(s.db, payload)
}

func (s *storageService) AllSpaceIds() (ids []string, err error) {
	err = s.db.View(func(txn *badger.Txn) error {
		opts := badger.DefaultIteratorOptions
		opts.PrefetchValues = false
		opts.Prefix = s.keys.SpacePrefix()

		it := txn.NewIterator(opts)
		defer it.Close()

		for it.Rewind(); it.Valid(); it.Next() {
			item := it.Item()
			id := item.Key()
			if len(id) <= len(s.keys.SpacePrefix())+1 {
				continue
			}
			id = id[len(s.keys.SpacePrefix())+1:]
			ids = append(ids, string(id))
		}
		return nil
	})
	return
}

func (s *storageService) Run(ctx context.Context) (err error) {
	return nil
}

func (s *storageService) Close(ctx context.Context) (err error) {
	return s.db.Close()
}