package dbtools

import (
 "slices"

 "github.com/google/blueprint/syncmap"
)

type KeyValueStore interface {
 Put(key []byte, value []byte) error
 Get(key []byte) ([]byte, error)
 Close() error
}

type InMemKeyValueStore struct {
 data syncmap.SyncMap[string, []byte]
}

func (s *InMemKeyValueStore) Close() error {
 return nil
}

func (s *InMemKeyValueStore) Put(key []byte, value []byte) error {
 s.data.LoadOrStore(string(key), slices.Clone(value))
 return nil
}

func (s *InMemKeyValueStore) Get(key []byte) ([]byte, error) {
 if ret, ok := s.data.Load(string(key)); !ok {
  return nil, nil
 } else {
  return ret, nil
 }
}