Return last id.

db.go

@@ -190,7 +190,7 @@ type Database interface {
 
 // Collection methods.
 type Collection interface {
-	Append(...interface{}) error
+	Append(...interface{}) ([]Id, error)
 
 	Count(...interface{}) (int, error)
 

mongo/mongo.go

@@ -80,7 +80,11 @@ func (c *MongoDataSourceCollection) Truncate() error {
 }
 
 // Appends an item to the collection.
-func (c *MongoDataSourceCollection) Append(items ...interface{}) error {
+func (c *MongoDataSourceCollection) Append(items ...interface{}) ([]db.Id, error) {
+
+	var err error
+
+	ids := []db.Id{}
 
 	parent := reflect.TypeOf(c.collection)
 	method, _ := parent.MethodByName("Insert")
@@ -91,16 +95,26 @@ func (c *MongoDataSourceCollection) Append(items ...interface{}) error {
 	itop := len(items)
 
 	for i := 0; i < itop; i++ {
+		id := db.Id(bson.NewObjectId().Hex())
+
+		switch items[i].(type) {
+		case map[string] interface{}:
+			if items[i].(map[string]interface{})["_id"] == nil {
+				items[i].(map[string]interface{})["_id"] = id
+			}
+		}
+
 		args[i+1] = reflect.ValueOf(toInternal(items[i]))
+		ids = append(ids, id)
 	}
 
 	exec := method.Func.Call(args)
 
 	if exec[0].Interface() != nil {
-		return exec[0].Interface().(error)
+		err = exec[0].Interface().(error)
 	}
 
-	return nil
+	return ids, err
 }
 
 // Compiles terms into something *mgo.Session can understand.

mongo/mongo_test.go

@@ -310,7 +310,7 @@ func TestDataTypes(t *testing.T) {
 
 	data := getTestData()
 
-	err = col.Append(data)
+	_, err = col.Append(data)
 
 	if err != nil {
 		t.Errorf("Could not append test data.")
@@ -320,7 +320,7 @@ func TestDataTypes(t *testing.T) {
 
 	item := col.Find()
 
-	err = col.Append(item)
+	_, err = col.Append(item)
 
 	if err == nil {
 		t.Errorf("Expecting duplicated-key error.")
@@ -328,12 +328,14 @@ func TestDataTypes(t *testing.T) {
 
 	delete(item, "_id")
 
-	err = col.Append(item)
+	ids, err := col.Append(item)
 
 	if err != nil {
 		t.Errorf("Could not append second element.")
 	}
 
+	fmt.Printf("%v\n", ids)
+
 	// Testing rows
 
 	items := col.FindAll()

mysql/mysql.go

@@ -671,7 +671,7 @@ func (t *MysqlTable) Find(terms ...interface{}) db.Item {
 }
 
 // Inserts rows into the currently active table.
-func (t *MysqlTable) Append(items ...interface{}) error {
+func (t *MysqlTable) Append(items ...interface{}) ([]db.Id, error) {
 
 	itop := len(items)
 
@@ -684,7 +684,6 @@ func (t *MysqlTable) Append(items ...interface{}) error {
 
 		for field, value := range item.(db.Item) {
 			fields = append(fields, field)
-			//values = append(values, fmt.Sprintf("%v", value))
 			values = append(values, toInternal(value))
 		}
 

mysql/mysql_test.go

@@ -310,7 +310,7 @@ func TestDataTypes(t *testing.T) {
 
 	data := getTestData()
 
-	err = col.Append(data)
+	_, err = col.Append(data)
 
 	if err != nil {
 		t.Errorf("Could not append test data.")
@@ -319,7 +319,7 @@ func TestDataTypes(t *testing.T) {
 	// Getting and reinserting.
 	item := col.Find()
 
-	err = col.Append(item)
+	_, err = col.Append(item)
 
 	if err == nil {
 		t.Errorf("Expecting duplicated-key error.")
@@ -327,7 +327,7 @@ func TestDataTypes(t *testing.T) {
 
 	delete(item, "id")
 
-	err = col.Append(item)
+	_, err = col.Append(item)
 
 	if err != nil {
 		t.Errorf("Could not append second element.")