Dbservice.go

package realbase

import (
	// "fmt"
	// "gopkg.in/igm/sockjs-go.v2/sockjs"
	// "net/http"
	"gopkg.in/mgo.v2/bson"
)

func constructMessage(doc bson.M, operation string) bson.M {
	message := make(map[string]interface{})

	message["origin"] = "db"
	message["data"] = doc
	message["operation"] = operation

	return message
}

type DbService struct {
	DataStore DataStore
}

func GetDbService(connectionString, dbName, colName string) *DbService {
	dataStore := DataStore{connectionString, dbName, colName}
	return &DbService{dataStore}
}

func (dbService *DbService) Insert(doc bson.M) error {
	err := dbService.DataStore.GetCollection().Insert(doc)

	message := constructMessage(doc, "insert")

	GetMessageService().BroadcastJSON(message)

	return err
}