68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/appleboy/gorush/gorush"
|
||
|
|
"sync/atomic"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Storage implements the storage interface for gorush (https://github.com/appleboy/gorush)
|
||
|
|
func New(stat gorush.StatusApp) *Storage {
|
||
|
|
return &Storage{
|
||
|
|
stat: stat,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type Storage struct {
|
||
|
|
stat gorush.StatusApp
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) addTotalCount(count int64) {
|
||
|
|
atomic.AddInt64(&s.stat.TotalCount, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) addIosSuccess(count int64) {
|
||
|
|
atomic.AddInt64(&s.stat.Ios.PushSuccess, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) addIosError(count int64) {
|
||
|
|
atomic.AddInt64(&s.stat.Ios.PushError, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) addAndroidSuccess(count int64) {
|
||
|
|
atomic.AddInt64(&s.stat.Android.PushSuccess, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) addAndroidError(count int64) {
|
||
|
|
atomic.AddInt64(&s.stat.Android.PushError, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) getTotalCount() int64 {
|
||
|
|
count := atomic.LoadInt64(&s.stat.TotalCount)
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) getIosSuccess() int64 {
|
||
|
|
count := atomic.LoadInt64(&s.stat.Ios.PushSuccess)
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) getIosError() int64 {
|
||
|
|
count := atomic.LoadInt64(&s.stat.Ios.PushError)
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) getAndroidSuccess() int64 {
|
||
|
|
count := atomic.LoadInt64(&s.stat.Android.PushSuccess)
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Storage) getAndroidError() int64 {
|
||
|
|
count := atomic.LoadInt64(&s.stat.Android.PushError)
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|