fix lint error in storage folder. (#276)
Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user