2015-06-02 10:46:25 -04:00
|
|
|
// Copyright (c) 2015 Pagoda Box Inc
|
|
|
|
//
|
|
|
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v.
|
|
|
|
// 2.0. If a copy of the MPL was not distributed with this file, You can obtain one
|
|
|
|
// at http://mozilla.org/MPL/2.0/.
|
|
|
|
//
|
|
|
|
|
2014-12-01 17:55:19 -05:00
|
|
|
package scribble
|
|
|
|
|
|
|
|
import (
|
2014-12-16 11:59:23 -05:00
|
|
|
"encoding/json"
|
2014-12-16 16:30:28 -05:00
|
|
|
"errors"
|
2014-12-01 17:55:19 -05:00
|
|
|
"fmt"
|
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
|
|
|
"strings"
|
2014-12-16 16:30:28 -05:00
|
|
|
"sync"
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2015-05-14 12:59:47 -04:00
|
|
|
"github.com/pagodabox/golang-hatchet"
|
2014-12-01 18:12:45 -05:00
|
|
|
)
|
|
|
|
|
2014-12-16 11:59:23 -05:00
|
|
|
//
|
|
|
|
const Version = "0.0.1"
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
//
|
|
|
|
type (
|
|
|
|
|
2014-12-16 11:59:23 -05:00
|
|
|
// Driver
|
2014-12-01 17:55:19 -05:00
|
|
|
Driver struct {
|
2014-12-16 16:30:28 -05:00
|
|
|
mutexes map[string]sync.Mutex
|
2014-12-01 17:55:19 -05:00
|
|
|
dir string
|
2014-12-16 16:30:28 -05:00
|
|
|
log hatchet.Logger
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// Transaction represents
|
|
|
|
Transaction struct {
|
|
|
|
Action string
|
|
|
|
Collection string
|
2014-12-16 16:30:28 -05:00
|
|
|
ResourceID string
|
2014-12-01 17:55:19 -05:00
|
|
|
Container interface{}
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
2014-12-16 11:59:23 -05:00
|
|
|
// New
|
|
|
|
func New(dir string, logger hatchet.Logger) (*Driver, error) {
|
|
|
|
fmt.Printf("Creating database directory at '%v'...\n", dir)
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
//
|
|
|
|
if logger == nil {
|
|
|
|
logger = hatchet.DevNullLogger{}
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
scribble := &Driver{
|
|
|
|
dir: dir,
|
|
|
|
mutexes: make(map[string]sync.Mutex),
|
|
|
|
log: logger,
|
|
|
|
}
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
//
|
2014-12-16 11:59:23 -05:00
|
|
|
if err := mkDir(scribble.dir); err != nil {
|
|
|
|
return nil, err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2014-12-16 11:59:23 -05:00
|
|
|
return scribble, nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// Transact
|
2014-12-16 11:59:23 -05:00
|
|
|
func (d *Driver) Transact(trans Transaction) error {
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
//
|
|
|
|
switch trans.Action {
|
|
|
|
case "write":
|
2014-12-16 16:30:28 -05:00
|
|
|
return d.write(trans)
|
2014-12-01 17:55:19 -05:00
|
|
|
case "read":
|
2014-12-16 16:30:28 -05:00
|
|
|
return d.read(trans)
|
2014-12-01 17:55:19 -05:00
|
|
|
case "readall":
|
2014-12-16 16:30:28 -05:00
|
|
|
return d.readAll(trans)
|
2014-12-01 17:55:19 -05:00
|
|
|
case "delete":
|
2014-12-16 16:30:28 -05:00
|
|
|
return d.delete(trans)
|
2014-12-01 17:55:19 -05:00
|
|
|
default:
|
2014-12-16 16:30:28 -05:00
|
|
|
return errors.New(fmt.Sprintf("Unsupported action %+v", trans.Action))
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
return nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// private
|
|
|
|
|
|
|
|
// write
|
2014-12-16 16:30:28 -05:00
|
|
|
func (d *Driver) write(trans Transaction) error {
|
|
|
|
|
|
|
|
mutex := d.getOrCreateMutex(trans.Collection)
|
|
|
|
mutex.Lock()
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
//
|
|
|
|
dir := d.dir + "/" + trans.Collection
|
|
|
|
|
|
|
|
//
|
2014-12-16 16:30:28 -05:00
|
|
|
b, err := json.MarshalIndent(trans.Container, "", "\t")
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2014-12-16 16:30:28 -05:00
|
|
|
if err := mkDir(dir); err != nil {
|
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2014-12-16 16:32:14 -05:00
|
|
|
if err := ioutil.WriteFile(dir + "/" + trans.ResourceID, b, 0666); err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-16 11:59:23 -05:00
|
|
|
}
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
mutex.Unlock()
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
return nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// read
|
2014-12-16 16:30:28 -05:00
|
|
|
func (d *Driver) read(trans Transaction) error {
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
dir := d.dir + "/" + trans.Collection
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
b, err := ioutil.ReadFile(dir + "/" + trans.ResourceID)
|
2014-12-01 17:55:19 -05:00
|
|
|
if err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
2014-12-16 11:59:23 -05:00
|
|
|
if err := json.Unmarshal(b, trans.Container); err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
return nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// readAll
|
2014-12-16 16:30:28 -05:00
|
|
|
func (d *Driver) readAll(trans Transaction) error {
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
dir := d.dir + "/" + trans.Collection
|
|
|
|
|
|
|
|
//
|
|
|
|
files, err := ioutil.ReadDir(dir)
|
|
|
|
|
2014-12-16 11:59:23 -05:00
|
|
|
// an error here just means an empty collection so do nothing
|
2014-12-01 17:55:19 -05:00
|
|
|
if err != nil {
|
|
|
|
}
|
|
|
|
|
|
|
|
var f []string
|
|
|
|
|
|
|
|
for _, file := range files {
|
|
|
|
b, err := ioutil.ReadFile(dir + "/" + file.Name())
|
|
|
|
if err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
f = append(f, string(b))
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
2014-12-16 11:59:23 -05:00
|
|
|
if err := json.Unmarshal([]byte("["+strings.Join(f, ",")+"]"), trans.Container); err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
return nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// delete
|
2014-12-16 16:30:28 -05:00
|
|
|
func (d *Driver) delete(trans Transaction) error {
|
|
|
|
|
|
|
|
mutex := d.getOrCreateMutex(trans.Collection)
|
|
|
|
mutex.Lock()
|
2014-12-01 17:55:19 -05:00
|
|
|
|
|
|
|
dir := d.dir + "/" + trans.Collection
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
err := os.Remove(dir + "/" + trans.ResourceID)
|
2014-12-01 17:55:19 -05:00
|
|
|
if err != nil {
|
2014-12-16 16:30:28 -05:00
|
|
|
return err
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
mutex.Unlock()
|
|
|
|
|
|
|
|
return nil
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
// helpers
|
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
// getOrCreateMutex
|
|
|
|
func (d *Driver) getOrCreateMutex(collection string) sync.Mutex {
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
c, ok := d.mutexes[collection]
|
2014-12-01 17:55:19 -05:00
|
|
|
|
2014-12-16 16:30:28 -05:00
|
|
|
// if the mutex doesn't exist make it
|
2014-12-01 17:55:19 -05:00
|
|
|
if !ok {
|
2014-12-16 16:30:28 -05:00
|
|
|
d.mutexes[collection] = sync.Mutex{}
|
|
|
|
return d.mutexes[collection]
|
2014-12-01 17:55:19 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
|
|
|
// mkDir
|
|
|
|
func mkDir(d string) error {
|
|
|
|
|
|
|
|
//
|
|
|
|
dir, _ := os.Stat(d)
|
|
|
|
|
|
|
|
if dir == nil {
|
|
|
|
err := os.MkdirAll(d, 0755)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|