package redis import ( "log" "strconv" "github.com/appleboy/gorush/config" "github.com/appleboy/gorush/storage" "gopkg.in/redis.v5" ) // var redisClient *redis.Client // New func implements the storage interface for gorush (https://github.com/appleboy/gorush) func New(config config.ConfYaml) *Storage { return &Storage{ config: config, } } func getInt64(key string, count *int64) { val, _ := redisClient.Get(key).Result() *count, _ = strconv.ParseInt(val, 10, 64) } // Storage is interface structure type Storage struct { config config.ConfYaml } // Init client storage. func (s *Storage) Init() error { redisClient = redis.NewClient(&redis.Options{ Addr: s.config.Stat.Redis.Addr, Password: s.config.Stat.Redis.Password, DB: s.config.Stat.Redis.DB, }) _, err := redisClient.Ping().Result() if err != nil { // redis server error log.Println("Can't connect redis server: " + err.Error()) return err } return nil } // Reset Client storage. func (s *Storage) Reset() { redisClient.Set(storage.TotalCountKey, strconv.Itoa(0), 0) redisClient.Set(storage.IosSuccessKey, strconv.Itoa(0), 0) redisClient.Set(storage.IosErrorKey, strconv.Itoa(0), 0) redisClient.Set(storage.AndroidSuccessKey, strconv.Itoa(0), 0) redisClient.Set(storage.AndroidErrorKey, strconv.Itoa(0), 0) } // AddTotalCount record push notification count. func (s *Storage) AddTotalCount(count int64) { total := s.GetTotalCount() + count redisClient.Set(storage.TotalCountKey, strconv.Itoa(int(total)), 0) } // AddIosSuccess record counts of success iOS push notification. func (s *Storage) AddIosSuccess(count int64) { total := s.GetIosSuccess() + count redisClient.Set(storage.IosSuccessKey, strconv.Itoa(int(total)), 0) } // AddIosError record counts of error iOS push notification. func (s *Storage) AddIosError(count int64) { total := s.GetIosError() + count redisClient.Set(storage.IosErrorKey, strconv.Itoa(int(total)), 0) } // AddAndroidSuccess record counts of success Android push notification. func (s *Storage) AddAndroidSuccess(count int64) { total := s.GetAndroidSuccess() + count redisClient.Set(storage.AndroidSuccessKey, strconv.Itoa(int(total)), 0) } // AddAndroidError record counts of error Android push notification. func (s *Storage) AddAndroidError(count int64) { total := s.GetAndroidError() + count redisClient.Set(storage.AndroidErrorKey, strconv.Itoa(int(total)), 0) } // GetTotalCount show counts of all notification. func (s *Storage) GetTotalCount() int64 { var count int64 getInt64(storage.TotalCountKey, &count) return count } // GetIosSuccess show success counts of iOS notification. func (s *Storage) GetIosSuccess() int64 { var count int64 getInt64(storage.IosSuccessKey, &count) return count } // GetIosError show error counts of iOS notification. func (s *Storage) GetIosError() int64 { var count int64 getInt64(storage.IosErrorKey, &count) return count } // GetAndroidSuccess show success counts of Android notification. func (s *Storage) GetAndroidSuccess() int64 { var count int64 getInt64(storage.AndroidSuccessKey, &count) return count } // GetAndroidError show error counts of Android notification. func (s *Storage) GetAndroidError() int64 { var count int64 getInt64(storage.AndroidErrorKey, &count) return count }