Improve database storage put method

database/storage/badger/badger.go

@@ -82,16 +82,19 @@ func (b *Badger) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (b *Badger) Put(r record.Record) error {
+func (b *Badger) Put(r record.Record) (record.Record, error) {
 	data, err := r.MarshalRecord(r)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
 	err = b.db.Update(func(txn *badger.Txn) error {
 		return txn.Set([]byte(r.DatabaseKey()), data)
 	})
-	return err
+	if err != nil {
+		return nil, err
+	}
+	return r, nil
 }
 
 // Delete deletes a record from the database.

database/storage/bbolt/bbolt.go

@@ -86,10 +86,10 @@ func (b *BBolt) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (b *BBolt) Put(r record.Record) error {
+func (b *BBolt) Put(r record.Record) (record.Record, error) {
 	data, err := r.MarshalRecord(r)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
 	err = b.db.Update(func(tx *bbolt.Tx) error {
@@ -100,9 +100,9 @@ func (b *BBolt) Put(r record.Record) error {
 		return nil
 	})
 	if err != nil {
-		return err
+		return nil, err
 	}
-	return nil
+	return r, nil
 }
 
 // PutMany stores many records in the database.

database/storage/bbolt/bbolt_test.go

@@ -65,7 +65,7 @@ func TestBBolt(t *testing.T) {
 	a.SetKey("test:A")
 
 	// put record
-	err = db.Put(a)
+	_, err = db.Put(a)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -100,15 +100,15 @@ func TestBBolt(t *testing.T) {
 	qZ.SetKey("test:z")
 	qZ.CreateMeta()
 	// put
-	err = db.Put(qA)
+	_, err = db.Put(qA)
 	if err == nil {
-		err = db.Put(qB)
+		_, err = db.Put(qB)
 	}
 	if err == nil {
-		err = db.Put(qC)
+		_, err = db.Put(qC)
 	}
 	if err == nil {
-		err = db.Put(qZ)
+		_, err = db.Put(qZ)
 	}
 	if err != nil {
 		t.Fatal(err)

database/storage/fstree/fstree.go

@@ -104,15 +104,15 @@ func (fst *FSTree) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (fst *FSTree) Put(r record.Record) error {
+func (fst *FSTree) Put(r record.Record) (record.Record, error) {
 	dstPath, err := fst.buildFilePath(r.DatabaseKey(), true)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
 	data, err := r.MarshalRecord(r)
 	if err != nil {
-		return err
+		return nil, err
 	}
 
 	err = writeFile(dstPath, data, defaultFileMode)
@@ -120,15 +120,15 @@ func (fst *FSTree) Put(r record.Record) error {
 		// create dir and try again
 		err = os.MkdirAll(filepath.Dir(dstPath), defaultDirMode)
 		if err != nil {
-			return fmt.Errorf("fstree: failed to create directory %s: %s", filepath.Dir(dstPath), err)
+			return nil, fmt.Errorf("fstree: failed to create directory %s: %s", filepath.Dir(dstPath), err)
 		}
 		err = writeFile(dstPath, data, defaultFileMode)
 		if err != nil {
-			return fmt.Errorf("fstree: could not write file %s: %s", dstPath, err)
+			return nil, fmt.Errorf("fstree: could not write file %s: %s", dstPath, err)
 		}
 	}
 
-	return nil
+	return r, nil
 }
 
 // Delete deletes a record from the database.

database/storage/hashmap/map.go

@@ -44,12 +44,12 @@ func (hm *HashMap) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (hm *HashMap) Put(r record.Record) error {
+func (hm *HashMap) Put(r record.Record) (record.Record, error) {
 	hm.dbLock.Lock()
 	defer hm.dbLock.Unlock()
 
 	hm.db[r.DatabaseKey()] = r
-	return nil
+	return r, nil
 }
 
 // PutMany stores many records in the database.

database/storage/hashmap/map_test.go

@@ -57,7 +57,7 @@ func TestHashMap(t *testing.T) {
 	a.SetKey("test:A")
 
 	// put record
-	err = db.Put(a)
+	_, err = db.Put(a)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -86,15 +86,15 @@ func TestHashMap(t *testing.T) {
 	qZ.SetKey("test:z")
 	qZ.CreateMeta()
 	// put
-	err = db.Put(qA)
+	_, err = db.Put(qA)
 	if err == nil {
-		err = db.Put(qB)
+		_, err = db.Put(qB)
 	}
 	if err == nil {
-		err = db.Put(qC)
+		_, err = db.Put(qC)
 	}
 	if err == nil {
-		err = db.Put(qZ)
+		_, err = db.Put(qZ)
 	}
 	if err != nil {
 		t.Fatal(err)

database/storage/injectbase.go

@@ -21,8 +21,8 @@ func (i *InjectBase) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (i *InjectBase) Put(m record.Record) error {
-	return errNotImplemented
+func (i *InjectBase) Put(m record.Record) (record.Record, error) {
+	return nil, errNotImplemented
 }
 
 // PutMany stores many records in the database.

database/storage/interface.go

@@ -9,7 +9,7 @@ import (
 // Interface defines the database storage API.
 type Interface interface {
 	Get(key string) (record.Record, error)
-	Put(m record.Record) error
+	Put(m record.Record) (record.Record, error)
 	Delete(key string) error
 	Query(q *query.Query, local, internal bool) (*iterator.Iterator, error)
 

database/storage/sinkhole/sinkhole.go

@@ -36,8 +36,8 @@ func (s *Sinkhole) Get(key string) (record.Record, error) {
 }
 
 // Put stores a record in the database.
-func (s *Sinkhole) Put(m record.Record) error {
-	return nil
+func (s *Sinkhole) Put(r record.Record) (record.Record, error) {
+	return r, nil
 }
 
 // PutMany stores many records in the database.