state_update.go

package storage

import (
	"github.com/Worldcoin/hubble-commander/models"
	"github.com/pkg/errors"
	bh "github.com/timshannon/badgerhold/v4"
)

func (s *StateTree) addStateUpdate(update *models.StateUpdate) error {
	return s.database.Badger.Insert(bh.NextSequence(), *update)
}

func (s *StateTree) getStateUpdate(id uint64) (*models.StateUpdate, error) {
	var stateUpdate models.StateUpdate
	err := s.database.Badger.Get(id, &stateUpdate)
	if errors.Is(err, bh.ErrNotFound) {
		return nil, errors.WithStack(NewNotFoundError("state update"))
	}
	if err != nil {
		return nil, err
	}
	return &stateUpdate, nil
}

func (s *StateTree) removeStateUpdate(id uint64) error {
	return s.database.Badger.Delete(id, models.StateUpdate{})
}