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

@@ -1,6 +1,8 @@
package boltdb
import (
"log"
"github.com/appleboy/gorush/config"
"github.com/appleboy/gorush/storage"
@@ -35,14 +37,32 @@ func (s *Storage) Reset() {
func (s *Storage) setBoltDB(key string, count int64) {
db, _ := storm.Open(s.config.Stat.BoltDB.Path)
db.Set(s.config.Stat.BoltDB.Bucket, key, count)
defer db.Close()
err := db.Set(s.config.Stat.BoltDB.Bucket, key, count)
if err != nil {
log.Println("BoltDB set error:", err.Error())
}
defer func() {
err := db.Close()
if err != nil {
log.Println("BoltDB error:", err.Error())
}
}()
}
func (s *Storage) getBoltDB(key string, count *int64) {
db, _ := storm.Open(s.config.Stat.BoltDB.Path)
db.Get(s.config.Stat.BoltDB.Bucket, key, count)
defer db.Close()
err := db.Get(s.config.Stat.BoltDB.Bucket, key, count)
if err != nil {
log.Println("BoltDB get error:", err.Error())
}
defer func() {
err := db.Close()
if err != nil {
log.Println("BoltDB error:", err.Error())
}
}()
}
// AddTotalCount record push notification count.

View File

@@ -13,7 +13,8 @@ func TestBoltDBEngine(t *testing.T) {
config := c.BuildDefaultPushConf()
boltDB := New(config)
boltDB.Init()
err := boltDB.Init()
assert.Nil(t, err)
boltDB.Reset()
boltDB.AddTotalCount(10)