mirror of
https://github.com/dolthub/dolt.git
synced 2026-02-11 10:33:08 -06:00
507b881c21
In order to allow tools like shove to correctly address named DataStores, we need to have the notion of a name be settable at the DataStoreFlags level. Once we've done that, it doesn't really make sense to have API surface for creating DataStores without a name -- though for compatibility, the code will continue to accept an empty string for a DataStore's name
110 lines
2.0 KiB
Go
110 lines
2.0 KiB
Go
package chunks
|
|
|
|
import (
|
|
"flag"
|
|
"sync"
|
|
|
|
"github.com/attic-labs/noms/d"
|
|
"github.com/attic-labs/noms/ref"
|
|
)
|
|
|
|
// An in-memory implementation of store.ChunkStore. Useful mainly for tests.
|
|
type MemoryStore struct {
|
|
data map[ref.Ref]Chunk
|
|
memoryRootTracker
|
|
mu *sync.Mutex
|
|
}
|
|
|
|
func NewMemoryStore() *MemoryStore {
|
|
return &MemoryStore{
|
|
mu: &sync.Mutex{},
|
|
}
|
|
}
|
|
|
|
func (ms *MemoryStore) Get(ref ref.Ref) Chunk {
|
|
ms.mu.Lock()
|
|
defer ms.mu.Unlock()
|
|
if c, ok := ms.data[ref]; ok {
|
|
return c
|
|
}
|
|
return EmptyChunk
|
|
}
|
|
|
|
func (ms *MemoryStore) Has(r ref.Ref) bool {
|
|
ms.mu.Lock()
|
|
defer ms.mu.Unlock()
|
|
if ms.data == nil {
|
|
return false
|
|
}
|
|
_, ok := ms.data[r]
|
|
return ok
|
|
}
|
|
|
|
func (ms *MemoryStore) Put(c Chunk) {
|
|
ms.mu.Lock()
|
|
defer ms.mu.Unlock()
|
|
if ms.data == nil {
|
|
ms.data = map[ref.Ref]Chunk{}
|
|
}
|
|
ms.data[c.Ref()] = c
|
|
}
|
|
|
|
func (ms *MemoryStore) Len() int {
|
|
ms.mu.Lock()
|
|
defer ms.mu.Unlock()
|
|
return len(ms.data)
|
|
}
|
|
|
|
func (l *MemoryStore) Close() error {
|
|
return nil
|
|
}
|
|
|
|
type MemoryStoreFlags struct {
|
|
use *bool
|
|
}
|
|
|
|
func MemoryFlags(prefix string) MemoryStoreFlags {
|
|
return MemoryStoreFlags{
|
|
flag.Bool(prefix+"mem", false, "use a memory-based (ephemeral, and private to this application) chunkstore"),
|
|
}
|
|
}
|
|
|
|
func (f MemoryStoreFlags) CreateStore(ns string) ChunkStore {
|
|
if f.check() {
|
|
return NewMemoryStore()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f MemoryStoreFlags) CreateFactory() Factory {
|
|
if f.check() {
|
|
return &MemoryStoreFactory{f, map[string]*MemoryStore{}}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f MemoryStoreFlags) check() bool {
|
|
return *f.use
|
|
}
|
|
|
|
type MemoryStoreFactory struct {
|
|
flags MemoryStoreFlags
|
|
stores map[string]*MemoryStore
|
|
}
|
|
|
|
func (f *MemoryStoreFactory) CreateStore(ns string) ChunkStore {
|
|
d.Chk.NotNil(f.stores, "Cannot use LevelDBStoreFactory after Shutter().")
|
|
if !f.flags.check() {
|
|
return nil
|
|
}
|
|
if cs, present := f.stores[ns]; present {
|
|
return cs
|
|
}
|
|
f.stores[ns] = NewMemoryStore()
|
|
return f.stores[ns]
|
|
}
|
|
|
|
func (f *MemoryStoreFactory) Shutter() {
|
|
f.stores = map[string]*MemoryStore{}
|
|
}
|