fix lint error in storage folder. (#276)

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

storage/buntdb/buntdb.go

@@ -2,6 +2,7 @@ package buntdb
 
 import (
 	"fmt"
+	"log"
 	"strconv"
 
 	"github.com/appleboy/gorush/config"
@@ -39,22 +40,44 @@ func (s *Storage) Reset() {
 func (s *Storage) setBuntDB(key string, count int64) {
 	db, _ := buntdb.Open(s.config.Stat.BuntDB.Path)
 
-	db.Update(func(tx *buntdb.Tx) error {
-		tx.Set(key, fmt.Sprintf("%d", count), nil)
+	err := db.Update(func(tx *buntdb.Tx) error {
+		if _, _, err := tx.Set(key, fmt.Sprintf("%d", count), nil); err != nil {
+			return err
+		}
 		return nil
 	})
-	defer db.Close()
+
+	if err != nil {
+		log.Println("BuntDB update error:", err.Error())
+	}
+
+	defer func() {
+		err := db.Close()
+		if err != nil {
+			log.Println("BuntDB error:", err.Error())
+		}
+	}()
 }
 
 func (s *Storage) getBuntDB(key string, count *int64) {
 	db, _ := buntdb.Open(s.config.Stat.BuntDB.Path)
 
-	db.View(func(tx *buntdb.Tx) error {
+	err := db.View(func(tx *buntdb.Tx) error {
 		val, _ := tx.Get(key)
 		*count, _ = strconv.ParseInt(val, 10, 64)
 		return nil
 	})
-	defer db.Close()
+
+	if err != nil {
+		log.Println("BuntDB get error:", err.Error())
+	}
+
+	defer func() {
+		err := db.Close()
+		if err != nil {
+			log.Println("BuntDB error:", err.Error())
+		}
+	}()
 }
 
 // AddTotalCount record push notification count.

storage/buntdb/buntdb_test.go

@@ -14,11 +14,13 @@ func TestBuntDBEngine(t *testing.T) {
 	config := c.BuildDefaultPushConf()
 
 	if _, err := os.Stat(config.Stat.BuntDB.Path); os.IsNotExist(err) {
-		os.RemoveAll(config.Stat.BuntDB.Path)
+		err := os.RemoveAll(config.Stat.BuntDB.Path)
+		assert.Nil(t, err)
 	}
 
 	buntDB := New(config)
-	buntDB.Init()
+	err := buntDB.Init()
+	assert.Nil(t, err)
 	buntDB.Reset()
 
 	buntDB.AddTotalCount(10)