redis.go 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. package utils
  2. import (
  3. "eta/eta_api/utils/redis"
  4. client "github.com/go-redis/redis/v8"
  5. "time"
  6. )
  7. type RedisClient interface {
  8. Get(key string) interface{}
  9. GetStr(key string) string
  10. GetUInt64(key string) (uint64, error)
  11. RedisBytes(key string) (data []byte, err error)
  12. RedisString(key string) (data string, err error)
  13. RedisInt(key string) (data int, err error)
  14. Put(key string, val interface{}, timeout time.Duration) error
  15. SetNX(key string, val interface{}, timeout time.Duration) bool
  16. Delete(key string) error
  17. IsExist(key string) bool
  18. LPush(key string, val interface{}) error
  19. Brpop(key string, callback func([]byte))
  20. BrpopWithTimeout(key string, timeout time.Duration, callback func([]byte)) error
  21. GetRedisTTL(key string) time.Duration
  22. Incrby(key string, num int) (interface{}, error)
  23. Do(commandName string, args ...interface{}) (reply interface{}, err error)
  24. SAdd(key string, args ...interface{}) (err error)
  25. SRem(key string, args ...interface{}) (err error)
  26. SIsMember(key string, args interface{}) (bool, error)
  27. ZAdd(key string, members ...*redis.Zset) error
  28. ZRangeWithScores(key string) ([]*redis.Zset, error)
  29. Expire(key string, duration time.Duration) error
  30. RedisClient() client.UniversalClient
  31. Keys(pattern string) ([]string, error)
  32. }
  33. func initRedis(redisType string, conf string) (redisClient RedisClient, err error) {
  34. switch redisType {
  35. case "cluster": // 集群
  36. redisClient, err = redis.InitClusterRedis(conf)
  37. default: // 默认走单机
  38. redisClient, err = redis.InitStandaloneRedis(conf)
  39. }
  40. return
  41. }
  42. // redis没有key的错误
  43. const RedisNoKeyErr = "redis: nil"