planner/plan/storage/memory/localid.go

package memory

import (
	"errors"
	"sync"

	"go-mod.ewintr.nl/planner/plan/storage"
)

type LocalID struct {
	ids   map[string]int
	mutex sync.RWMutex
}

func NewLocalID() *LocalID {
	return &LocalID{
		ids: make(map[string]int),
	}
}

func (ml *LocalID) FindAll() (map[string]int, error) {
	ml.mutex.RLock()
	defer ml.mutex.RUnlock()

	return ml.ids, nil
}

func (ml *LocalID) Find(id string) (int, error) {
	ml.mutex.RLock()
	defer ml.mutex.RUnlock()

	lid, ok := ml.ids[id]
	if !ok {
		return 0, storage.ErrNotFound
	}

	return lid, nil
}

func (ml *LocalID) FindOrNext(id string) (int, error) {
	lid, err := ml.Find(id)
	switch {
	case errors.Is(err, storage.ErrNotFound):
		return ml.Next()
	case err != nil:
		return 0, err
	default:
		return lid, nil
	}
}

func (ml *LocalID) Next() (int, error) {
	ml.mutex.RLock()
	defer ml.mutex.RUnlock()

	cur := make([]int, 0, len(ml.ids))
	for _, i := range ml.ids {
		cur = append(cur, i)
	}

	localID := storage.NextLocalID(cur)

	return localID, nil
}

func (ml *LocalID) Store(id string, localID int) error {
	ml.mutex.Lock()
	defer ml.mutex.Unlock()

	ml.ids[id] = localID

	return nil
}

func (ml *LocalID) Delete(id string) error {
	ml.mutex.Lock()
	defer ml.mutex.Unlock()

	if _, ok := ml.ids[id]; !ok {
		return storage.ErrNotFound
	}

	delete(ml.ids, id)

	return nil
}