fix boltdb and memory client lint.

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2016-08-01 10:57:00 +08:00
parent 4494310984
commit dd1fb29e6b
2 changed files with 26 additions and 0 deletions

View File

@ -21,14 +21,17 @@ func New(config config.ConfYaml) *Storage {
} }
} }
// Storage is interface structure
type Storage struct { type Storage struct {
config config.ConfYaml config config.ConfYaml
} }
// Init client storage.
func (s *Storage) Init() error { func (s *Storage) Init() error {
return nil return nil
} }
// Reset Client storage.
func (s *Storage) Reset() { func (s *Storage) Reset() {
s.setBoltDB(TotalCountKey, 0) s.setBoltDB(TotalCountKey, 0)
s.setBoltDB(IosSuccessKey, 0) s.setBoltDB(IosSuccessKey, 0)
@ -49,31 +52,37 @@ func (s *Storage) getBoltDB(key string, count *int64) {
defer db.Close() defer db.Close()
} }
// AddTotalCount record push notification count.
func (s *Storage) AddTotalCount(count int64) { func (s *Storage) AddTotalCount(count int64) {
total := s.GetTotalCount() + count total := s.GetTotalCount() + count
s.setBoltDB(TotalCountKey, total) s.setBoltDB(TotalCountKey, total)
} }
// AddIosSuccess record counts of success iOS push notification.
func (s *Storage) AddIosSuccess(count int64) { func (s *Storage) AddIosSuccess(count int64) {
total := s.GetIosSuccess() + count total := s.GetIosSuccess() + count
s.setBoltDB(IosSuccessKey, total) s.setBoltDB(IosSuccessKey, total)
} }
// AddIosError record counts of error iOS push notification.
func (s *Storage) AddIosError(count int64) { func (s *Storage) AddIosError(count int64) {
total := s.GetIosError() + count total := s.GetIosError() + count
s.setBoltDB(IosErrorKey, total) s.setBoltDB(IosErrorKey, total)
} }
// AddAndroidSuccess record counts of success Android push notification.
func (s *Storage) AddAndroidSuccess(count int64) { func (s *Storage) AddAndroidSuccess(count int64) {
total := s.GetAndroidSuccess() + count total := s.GetAndroidSuccess() + count
s.setBoltDB(AndroidSuccessKey, total) s.setBoltDB(AndroidSuccessKey, total)
} }
// AddAndroidError record counts of error Android push notification.
func (s *Storage) AddAndroidError(count int64) { func (s *Storage) AddAndroidError(count int64) {
total := s.GetAndroidError() + count total := s.GetAndroidError() + count
s.setBoltDB(AndroidErrorKey, total) s.setBoltDB(AndroidErrorKey, total)
} }
// GetTotalCount show counts of all notification.
func (s *Storage) GetTotalCount() int64 { func (s *Storage) GetTotalCount() int64 {
var count int64 var count int64
s.getBoltDB(TotalCountKey, &count) s.getBoltDB(TotalCountKey, &count)
@ -81,6 +90,7 @@ func (s *Storage) GetTotalCount() int64 {
return count return count
} }
// GetIosSuccess show success counts of iOS notification.
func (s *Storage) GetIosSuccess() int64 { func (s *Storage) GetIosSuccess() int64 {
var count int64 var count int64
s.getBoltDB(IosSuccessKey, &count) s.getBoltDB(IosSuccessKey, &count)
@ -88,6 +98,7 @@ func (s *Storage) GetIosSuccess() int64 {
return count return count
} }
// GetIosError show error counts of iOS notification.
func (s *Storage) GetIosError() int64 { func (s *Storage) GetIosError() int64 {
var count int64 var count int64
s.getBoltDB(IosErrorKey, &count) s.getBoltDB(IosErrorKey, &count)
@ -95,6 +106,7 @@ func (s *Storage) GetIosError() int64 {
return count return count
} }
// GetAndroidSuccess show success counts of Android notification.
func (s *Storage) GetAndroidSuccess() int64 { func (s *Storage) GetAndroidSuccess() int64 {
var count int64 var count int64
s.getBoltDB(AndroidSuccessKey, &count) s.getBoltDB(AndroidSuccessKey, &count)
@ -102,6 +114,7 @@ func (s *Storage) GetAndroidSuccess() int64 {
return count return count
} }
// GetAndroidError show error counts of Android notification.
func (s *Storage) GetAndroidError() int64 { func (s *Storage) GetAndroidError() int64 {
var count int64 var count int64
s.getBoltDB(AndroidErrorKey, &count) s.getBoltDB(AndroidErrorKey, &count)

View File

@ -30,61 +30,74 @@ func New() *Storage {
} }
} }
// Storage is interface structure
type Storage struct { type Storage struct {
stat *statApp stat *statApp
} }
// Init client storage.
func (s *Storage) Init() error { func (s *Storage) Init() error {
return nil return nil
} }
// Reset Client storage.
func (s *Storage) Reset() { func (s *Storage) Reset() {
} }
// AddTotalCount record push notification count.
func (s *Storage) AddTotalCount(count int64) { func (s *Storage) AddTotalCount(count int64) {
atomic.AddInt64(&s.stat.TotalCount, count) atomic.AddInt64(&s.stat.TotalCount, count)
} }
// AddIosSuccess record counts of success iOS push notification.
func (s *Storage) AddIosSuccess(count int64) { func (s *Storage) AddIosSuccess(count int64) {
atomic.AddInt64(&s.stat.Ios.PushSuccess, count) atomic.AddInt64(&s.stat.Ios.PushSuccess, count)
} }
// AddIosError record counts of error iOS push notification.
func (s *Storage) AddIosError(count int64) { func (s *Storage) AddIosError(count int64) {
atomic.AddInt64(&s.stat.Ios.PushError, count) atomic.AddInt64(&s.stat.Ios.PushError, count)
} }
// AddAndroidSuccess record counts of success Android push notification.
func (s *Storage) AddAndroidSuccess(count int64) { func (s *Storage) AddAndroidSuccess(count int64) {
atomic.AddInt64(&s.stat.Android.PushSuccess, count) atomic.AddInt64(&s.stat.Android.PushSuccess, count)
} }
// AddAndroidError record counts of error Android push notification.
func (s *Storage) AddAndroidError(count int64) { func (s *Storage) AddAndroidError(count int64) {
atomic.AddInt64(&s.stat.Android.PushError, count) atomic.AddInt64(&s.stat.Android.PushError, count)
} }
// GetTotalCount show counts of all notification.
func (s *Storage) GetTotalCount() int64 { func (s *Storage) GetTotalCount() int64 {
count := atomic.LoadInt64(&s.stat.TotalCount) count := atomic.LoadInt64(&s.stat.TotalCount)
return count return count
} }
// GetIosSuccess show success counts of iOS notification.
func (s *Storage) GetIosSuccess() int64 { func (s *Storage) GetIosSuccess() int64 {
count := atomic.LoadInt64(&s.stat.Ios.PushSuccess) count := atomic.LoadInt64(&s.stat.Ios.PushSuccess)
return count return count
} }
// GetIosError show error counts of iOS notification.
func (s *Storage) GetIosError() int64 { func (s *Storage) GetIosError() int64 {
count := atomic.LoadInt64(&s.stat.Ios.PushError) count := atomic.LoadInt64(&s.stat.Ios.PushError)
return count return count
} }
// GetAndroidSuccess show success counts of Android notification.
func (s *Storage) GetAndroidSuccess() int64 { func (s *Storage) GetAndroidSuccess() int64 {
count := atomic.LoadInt64(&s.stat.Android.PushSuccess) count := atomic.LoadInt64(&s.stat.Android.PushSuccess)
return count return count
} }
// GetAndroidError show error counts of Android notification.
func (s *Storage) GetAndroidError() int64 { func (s *Storage) GetAndroidError() int64 {
count := atomic.LoadInt64(&s.stat.Android.PushError) count := atomic.LoadInt64(&s.stat.Android.PushError)