Merge pull request #77 from harshavardhana/pr_out_add_tiedot_integration_package_first_cut_more_improvements_needed

pkg/utils/database/database.go

@@ -0,0 +1,51 @@
+package database
+
+import (
+	"github.com/HouzuoGuo/tiedot/db"
+)
+
+func NewDatabase() *Database {
+	return &Database{}
+}
+
+func (data *Database) setdbdir(dirname string) {
+	data.DBdir = dirname
+}
+
+func (data *Database) GetDBHandle(dirname string) error {
+	var err error
+	data.setdbdir(dirname)
+	data.DBhandle, err = db.OpenDB(data.DBdir)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (data *Database) InitCollection(colls ...string) {
+	for _, str := range colls {
+		data.DBhandle.Create(str)
+	}
+}
+
+func (data *Database) GetCollections() []string {
+	var colls []string
+	for _, name := range data.DBhandle.AllCols() {
+		colls = append(colls, name)
+	}
+	return colls
+}
+
+func (data *Database) getCollectionHandle(coll string) *db.Col {
+	return data.DBhandle.Use(coll)
+}
+
+func (data *Database) GetCollectionData(coll string, docid int) (map[string]interface{}, error) {
+	collHandle := data.getCollectionHandle(coll)
+	return collHandle.Read(docid)
+}
+
+func (data *Database) InsertToCollection(coll string, model map[string]interface{}) (docid int, err error) {
+	collHandle := data.getCollectionHandle(coll)
+	return collHandle.Insert(model)
+}

pkg/utils/database/database_test.go

@@ -0,0 +1,36 @@
+package database
+
+import (
+	"os"
+	"testing"
+
+	. "gopkg.in/check.v1"
+)
+
+func Test(t *testing.T) { TestingT(t) }
+
+type MySuite struct{}
+
+var _ = Suite(&MySuite{})
+
+func (s *MySuite) Testing(c *C) {
+	d := NewDatabase()
+	d.GetDBHandle("/tmp/testdata")
+	defer os.RemoveAll("/tmp/testdata")
+
+	d.InitCollection("Matrix")
+
+	data := map[string]interface{}{
+		"version":  "1.4",
+		"url":      "golang.org",
+		"language": "Go",
+	}
+
+	docId, err1 := d.InsertToCollection("Matrix", data)
+	c.Assert(err1, IsNil)
+
+	retdata, err2 := d.GetCollectionData("Matrix", docId)
+
+	c.Assert(err2, IsNil)
+	c.Assert(data, DeepEquals, retdata)
+}

pkg/utils/database/structure.go

@@ -0,0 +1,10 @@
+package database
+
+import (
+	"github.com/HouzuoGuo/tiedot/db"
+)
+
+type Database struct {
+	DBdir    string
+	DBhandle *db.DB
+}