coopgo-platform
/
gorush
7
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

fixed #402 item get value error fix (#404) 

- bager rename badger
- Value method not usable, therfore changed to ValueCopy
This commit is contained in:
Cem KIY 2019-04-13 17:39:33 +03:00 committed by Bo-Yi Wu
parent 7b0750c807
commit 5b526a73bd
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
storage/badger/badger.go
View File

@ -29,7 +29,7 @@ type Storage struct {
// Init client storage. // Init client storage.
func (s *Storage) Init() error { func (s *Storage) Init() error {
s.name = "bager" s.name = "badger"
s.opts = badger.DefaultOptions s.opts = badger.DefaultOptions
s.opts.Dir = os.TempDir() + "badger" s.opts.Dir = os.TempDir() + "badger"
s.opts.ValueDir = os.TempDir() + "badger" s.opts.ValueDir = os.TempDir() + "badger"
@ -70,7 +70,7 @@ func (s *Storage) setBadger(key string, count int64) {
} }
} }
func (s *Storage) getBager(key string, count *int64) { func (s *Storage) getBadger(key string, count *int64) {
db, err := badger.Open(s.opts) db, err := badger.Open(s.opts)
if err != nil { if err != nil {
@ -90,7 +90,8 @@ func (s *Storage) getBager(key string, count *int64) {
if err != nil { if err != nil {
return err return err
} }
val, err := item.Value() dst := []byte{}
val, err := item.ValueCopy(dst)
if err != nil { if err != nil {
return err return err
} }
@ -143,7 +144,7 @@ func (s *Storage) AddAndroidError(count int64) {
// GetTotalCount show counts of all notification. // GetTotalCount show counts of all notification.
func (s *Storage) GetTotalCount() int64 { func (s *Storage) GetTotalCount() int64 {
var count int64 var count int64
s.getBager(storage.TotalCountKey, &count) s.getBadger(storage.TotalCountKey, &count)
return count return count
} }
@ -151,7 +152,7 @@ func (s *Storage) GetTotalCount() int64 {
// GetIosSuccess show success counts of iOS notification. // GetIosSuccess show success counts of iOS notification.
func (s *Storage) GetIosSuccess() int64 { func (s *Storage) GetIosSuccess() int64 {
var count int64 var count int64
s.getBager(storage.IosSuccessKey, &count) s.getBadger(storage.IosSuccessKey, &count)
return count return count
} }
@ -159,7 +160,7 @@ func (s *Storage) GetIosSuccess() int64 {
// GetIosError show error counts of iOS notification. // GetIosError show error counts of iOS notification.
func (s *Storage) GetIosError() int64 { func (s *Storage) GetIosError() int64 {
var count int64 var count int64
s.getBager(storage.IosErrorKey, &count) s.getBadger(storage.IosErrorKey, &count)
return count return count
} }
@ -167,7 +168,7 @@ func (s *Storage) GetIosError() int64 {
// GetAndroidSuccess show success counts of Android notification. // GetAndroidSuccess show success counts of Android notification.
func (s *Storage) GetAndroidSuccess() int64 { func (s *Storage) GetAndroidSuccess() int64 {
var count int64 var count int64
s.getBager(storage.AndroidSuccessKey, &count) s.getBadger(storage.AndroidSuccessKey, &count)
return count return count
} }
@ -175,7 +176,7 @@ func (s *Storage) GetAndroidSuccess() int64 {
// GetAndroidError show error counts of Android notification. // GetAndroidError show error counts of Android notification.
func (s *Storage) GetAndroidError() int64 { func (s *Storage) GetAndroidError() int64 {
var count int64 var count int64
s.getBager(storage.AndroidErrorKey, &count) s.getBadger(storage.AndroidErrorKey, &count)
return count return count
} }