// Update updates a document
func Update(collectionName string, doc interface{}) error {
connection := shared.Connection()
if connection == nil {
log.Println("MongoConn", connection)
return errors.New("No connection to database server")
}
var asMap map[string]interface{}
switch value := doc.(type) {
case *map[string]interface{}:
asMap = *(value)
case map[string]interface{}:
asMap = value
default:
return errors.New("Must be a map")
}
_, ok := asMap["_id"]
if !ok {
return errors.New("Doc needs to have _id to be saved")
}
findDoc := bson.M{"_id": asMap["_id"]}
collection := connection.DB(variables.DBName).C(collectionName)
err := collection.Update(findDoc, doc)
if err != nil {
return err
}
return nil
}
// Save tries to insert then tries to save
func Save(collectionName string, doc interface{}) error {
connection := shared.Connection()
if connection == nil {
log.Println("MongoConn", connection)
return errors.New("No connection to database server")
}
err := Insert(collectionName, doc)
if err == nil {
return nil
}
return Update(collectionName, doc)
}
// Get reteives a record given the id
func Get(collectionName, id string) (*map[string]interface{}, error) {
connection := shared.Connection()
if connection == nil {
return nil, errors.New("No connection to database server")
}
var result map[string]interface{}
collection := connection.DB(variables.DBName).C(collectionName)
err := collection.Find(bson.M{"_id": id}).One(&result)
if err != nil {
return nil, err
}
return &result, nil
}
// Insert creates a document
func Insert(collectionName string, doc interface{}) error {
connection := shared.Connection()
if connection == nil {
log.Println("MongoConn", connection)
return errors.New("No connection to database server")
}
collection := connection.DB(variables.DBName).C(collectionName)
err := collection.Insert(doc)
if err != nil {
return err
}
return nil
}
