93 lines
1.9 KiB
Go
93 lines
1.9 KiB
Go
|
package sandbox
|
||
|
|
|
||
|
|
import "sync"
|
||
|
|
|
||
|
|
// memoryStore implements a Store in memory.
|
||
|
|
type memoryStore struct {
|
||
|
|
s map[string]*Sandbox
|
||
|
|
sync.RWMutex
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewMemoryStore initializes a new memory store.
|
||
|
|
func NewMemoryStore() Storer {
|
||
|
|
return &memoryStore{
|
||
|
|
s: make(map[string]*Sandbox),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add appends a new sandbox to the memory store.
|
||
|
|
// It overrides the id if it existed before.
|
||
|
|
func (c *memoryStore) Add(id string, cont *Sandbox) {
|
||
|
|
c.Lock()
|
||
|
|
c.s[id] = cont
|
||
|
|
c.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get returns a sandbox from the store by id.
|
||
|
|
func (c *memoryStore) Get(id string) *Sandbox {
|
||
|
|
c.RLock()
|
||
|
|
res := c.s[id]
|
||
|
|
c.RUnlock()
|
||
|
|
return res
|
||
|
|
}
|
||
|
|
|
||
|
|
// Delete removes a sandbox from the store by id.
|
||
|
|
func (c *memoryStore) Delete(id string) {
|
||
|
|
c.Lock()
|
||
|
|
delete(c.s, id)
|
||
|
|
c.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
// List returns a sorted list of sandboxes from the store.
|
||
|
|
// The sandboxes are ordered by creation date.
|
||
|
|
func (c *memoryStore) List() []*Sandbox {
|
||
|
|
sandboxes := History(c.all())
|
||
|
|
sandboxes.sort()
|
||
|
|
return sandboxes
|
||
|
|
}
|
||
|
|
|
||
|
|
// Size returns the number of sandboxes in the store.
|
||
|
|
func (c *memoryStore) Size() int {
|
||
|
|
c.RLock()
|
||
|
|
defer c.RUnlock()
|
||
|
|
return len(c.s)
|
||
|
|
}
|
||
|
|
|
||
|
|
// First returns the first sandbox found in the store by a given filter.
|
||
|
|
func (c *memoryStore) First(filter StoreFilter) *Sandbox {
|
||
|
|
for _, cont := range c.all() {
|
||
|
|
if filter(cont) {
|
||
|
|
return cont
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ApplyAll calls the reducer function with every sandbox in the store.
|
||
|
|
// This operation is asynchronous in the memory store.
|
||
|
|
// NOTE: Modifications to the store MUST NOT be done by the StoreReducer.
|
||
|
|
func (c *memoryStore) ApplyAll(apply StoreReducer) {
|
||
|
|
wg := new(sync.WaitGroup)
|
||
|
|
for _, cont := range c.all() {
|
||
|
|
wg.Add(1)
|
||
|
|
go func(sandbox *Sandbox) {
|
||
|
|
apply(sandbox)
|
||
|
|
wg.Done()
|
||
|
|
}(cont)
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Wait()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *memoryStore) all() []*Sandbox {
|
||
|
|
c.RLock()
|
||
|
|
sandboxes := make([]*Sandbox, 0, len(c.s))
|
||
|
|
for _, cont := range c.s {
|
||
|
|
sandboxes = append(sandboxes, cont)
|
||
|
|
}
|
||
|
|
c.RUnlock()
|
||
|
|
return sandboxes
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ Storer = &memoryStore{}
|