first commit
This commit is contained in:
commit
e06eb20b2f
28
.gitignore
vendored
Normal file
28
.gitignore
vendored
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
# Compiled Object files, Static and Dynamic libs (Shared Objects)
|
||||||
|
*.o
|
||||||
|
*.a
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Folders
|
||||||
|
_obj
|
||||||
|
_test
|
||||||
|
|
||||||
|
# Architecture specific extensions/prefixes
|
||||||
|
*.[568vq]
|
||||||
|
[568vq].out
|
||||||
|
|
||||||
|
*.cgo1.go
|
||||||
|
*.cgo2.c
|
||||||
|
_cgo_defun.c
|
||||||
|
_cgo_gotypes.go
|
||||||
|
_cgo_export.*
|
||||||
|
|
||||||
|
_testmain.go
|
||||||
|
|
||||||
|
*.exe
|
||||||
|
*.test
|
||||||
|
*.prof
|
||||||
|
|
||||||
|
#
|
||||||
|
.DS_Store
|
||||||
|
TODO
|
22
LICENSE
Normal file
22
LICENSE
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2014 nanobox-core
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
||||||
|
|
218
main.go
Normal file
218
main.go
Normal file
@ -0,0 +1,218 @@
|
|||||||
|
package scribble
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/nanobox-core/nanobox-server/utils"
|
||||||
|
)
|
||||||
|
|
||||||
|
//
|
||||||
|
type (
|
||||||
|
|
||||||
|
// Driver represents
|
||||||
|
Driver struct {
|
||||||
|
channels map[string]chan int
|
||||||
|
dir string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Transaction represents
|
||||||
|
Transaction struct {
|
||||||
|
Action string
|
||||||
|
Collection string
|
||||||
|
Resource string
|
||||||
|
Container interface{}
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
//
|
||||||
|
var (
|
||||||
|
debugging bool
|
||||||
|
)
|
||||||
|
|
||||||
|
// Init
|
||||||
|
func (d *Driver) Init(opts map[string]string) int {
|
||||||
|
fmt.Printf("Creating database directory at '%v'...\n", opts["db_dir"])
|
||||||
|
|
||||||
|
debugging = (opts["debugging"] == "true")
|
||||||
|
|
||||||
|
d.dir = opts["db_dir"]
|
||||||
|
|
||||||
|
//
|
||||||
|
d.channels = make(map[string]chan int)
|
||||||
|
|
||||||
|
// make a ping channel
|
||||||
|
ping := make(chan int)
|
||||||
|
d.channels["ping"] = ping
|
||||||
|
|
||||||
|
//
|
||||||
|
if err := mkDir(d.dir); err != nil {
|
||||||
|
fmt.Printf("Unable to create dir '%v': %v", d.dir, err)
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Transact
|
||||||
|
func (d *Driver) Transact(trans Transaction) {
|
||||||
|
|
||||||
|
//
|
||||||
|
done := d.getOrCreateChan(trans.Collection)
|
||||||
|
|
||||||
|
//
|
||||||
|
switch trans.Action {
|
||||||
|
case "write":
|
||||||
|
go d.write(trans, done)
|
||||||
|
case "read":
|
||||||
|
go d.read(trans, done)
|
||||||
|
case "readall":
|
||||||
|
go d.readAll(trans, done)
|
||||||
|
case "delete":
|
||||||
|
go d.delete(trans, done)
|
||||||
|
default:
|
||||||
|
fmt.Println("Unsupported action ", trans.Action)
|
||||||
|
}
|
||||||
|
|
||||||
|
// wait...
|
||||||
|
<-done
|
||||||
|
}
|
||||||
|
|
||||||
|
// private
|
||||||
|
|
||||||
|
// write
|
||||||
|
func (d *Driver) write(trans Transaction, done chan<- int) {
|
||||||
|
|
||||||
|
//
|
||||||
|
dir := d.dir + "/" + trans.Collection
|
||||||
|
|
||||||
|
//
|
||||||
|
if err := mkDir(dir); err != nil {
|
||||||
|
fmt.Println("Unable to create dir '%v': %v", dir, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
file, err := os.Create(dir + "/" + trans.Resource)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Unable to create file %v/%v: %v", trans.Collection, trans.Resource, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
//
|
||||||
|
b := utils.ToJSONIndent(trans.Container)
|
||||||
|
|
||||||
|
_, err = file.WriteString(string(b))
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Unable to write to file %v: %v", trans.Resource, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// release...
|
||||||
|
done <- 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// read
|
||||||
|
func (d *Driver) read(trans Transaction, done chan<- int) interface{} {
|
||||||
|
|
||||||
|
dir := d.dir + "/" + trans.Collection
|
||||||
|
|
||||||
|
b, err := ioutil.ReadFile(dir + "/" + trans.Resource)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Unable to read file %v/%v: %v", trans.Collection, trans.Resource, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := utils.FromJSON(b, trans.Container); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// release...
|
||||||
|
done <- 0
|
||||||
|
|
||||||
|
return trans.Container
|
||||||
|
}
|
||||||
|
|
||||||
|
// readAll
|
||||||
|
func (d *Driver) readAll(trans Transaction, done chan<- int) {
|
||||||
|
|
||||||
|
dir := d.dir + "/" + trans.Collection
|
||||||
|
|
||||||
|
//
|
||||||
|
files, err := ioutil.ReadDir(dir)
|
||||||
|
|
||||||
|
// if there is an error here it just means there are no evars so dont do anything
|
||||||
|
if err != nil {
|
||||||
|
}
|
||||||
|
|
||||||
|
var f []string
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
b, err := ioutil.ReadFile(dir + "/" + file.Name())
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
f = append(f, string(b))
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
if err := utils.FromJSON([]byte("["+strings.Join(f, ",")+"]"), trans.Container); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// release...
|
||||||
|
done <- 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// delete
|
||||||
|
func (d *Driver) delete(trans Transaction, done chan<- int) {
|
||||||
|
|
||||||
|
dir := d.dir + "/" + trans.Collection
|
||||||
|
|
||||||
|
err := os.Remove(dir + "/" + trans.Resource)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Unable to delete file %v/%v: %v", trans.Collection, trans.Resource, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// release...
|
||||||
|
done <- 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// helpers
|
||||||
|
|
||||||
|
// getChan
|
||||||
|
func (d *Driver) getOrCreateChan(channel string) chan int {
|
||||||
|
|
||||||
|
c, ok := d.channels[channel]
|
||||||
|
|
||||||
|
// if the chan doesn't exist make it
|
||||||
|
if !ok {
|
||||||
|
d.channels[channel] = make(chan int)
|
||||||
|
return d.channels[channel]
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user