Added Save/Load to db

Fixes #3. Fixes #2
This commit is contained in:
sm
2016-02-14 22:20:10 -08:00
parent af4faf0206
commit 4cfaf2124f
4 changed files with 70 additions and 17 deletions
+39 -8
View File
@@ -1,7 +1,8 @@
package vain
import (
"errors"
"encoding/json"
"os"
"strings"
"sync"
)
@@ -18,33 +19,63 @@ func Valid(p string, packages []Package) bool {
type MemStore struct {
l sync.RWMutex
p map[string]Package
dbl sync.Mutex
path string
}
func NewMemStore() *MemStore {
func NewMemStore(path string) *MemStore {
return &MemStore{
p: make(map[string]Package),
path: path,
p: make(map[string]Package),
}
}
func (ms MemStore) Add(p Package) error {
func (ms *MemStore) Add(p Package) error {
ms.l.Lock()
ms.p[p.Path] = p
ms.l.Unlock()
return nil
}
func (ms MemStore) Remove(path string) error {
func (ms *MemStore) Remove(path string) error {
ms.l.Lock()
delete(ms.p, path)
ms.l.Unlock()
return nil
}
func (ms MemStore) Save() error {
return errors.New("save is not implemented")
func (ms *MemStore) Save() error {
// running in-memory only
if ms.path == "" {
return nil
}
ms.dbl.Lock()
defer ms.dbl.Unlock()
f, err := os.Create(ms.path)
if err != nil {
return err
}
defer f.Close()
return json.NewEncoder(f).Encode(ms.p)
}
func (ms MemStore) All() []Package {
func (ms *MemStore) Load() error {
// running in-memory only
if ms.path == "" {
return nil
}
ms.dbl.Lock()
defer ms.dbl.Unlock()
f, err := os.Open(ms.path)
if err != nil {
return err
}
defer f.Close()
return json.NewDecoder(f).Decode(&ms.p)
}
func (ms *MemStore) All() []Package {
r := []Package{}
ms.l.RLock()
for _, p := range ms.p {