fix lint error in storage folder. (#276)

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2017-08-23 22:25:05 -05:00
committed by GitHub
parent e070248a11
commit 35e2c97f43
7 changed files with 77 additions and 17 deletions

View File

@@ -2,6 +2,7 @@ package leveldb
import (
"fmt"
"log"
"strconv"
"github.com/appleboy/gorush/config"
@@ -18,7 +19,12 @@ func setLevelDB(key string, count int64) {
_ = db.Put([]byte(key), []byte(value), nil)
defer db.Close()
defer func() {
err := db.Close()
if err != nil {
log.Println("LevelDB error:", err.Error())
}
}()
}
func getLevelDB(key string, count *int64) {
@@ -27,7 +33,12 @@ func getLevelDB(key string, count *int64) {
data, _ := db.Get([]byte(key), nil)
*count, _ = strconv.ParseInt(string(data), 10, 64)
defer db.Close()
defer func() {
err := db.Close()
if err != nil {
log.Println("LevelDB error:", err.Error())
}
}()
}
// New func implements the storage interface for gorush (https://github.com/appleboy/gorush)

View File

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