redis.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  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. GetRedisTTL(key string) time.Duration
  21. Incrby(key string, num int) (interface{}, error)
  22. Do(commandName string, args ...interface{}) (reply interface{}, err error)
  23. SAdd(key string, args ...interface{}) (err error)
  24. SRem(key string, args ...interface{}) (err error)
  25. SIsMember(key string, args interface{}) (bool, error)
  26. ZAdd(key string, members ...*redis.Zset) error
  27. ZRangeWithScores(key string) ([]*redis.Zset, error)
  28. Expire(key string, duration time.Duration) error
  29. RedisClient() client.UniversalClient
  30. Keys(pattern string) ([]string, error)
  31. }
  32. func initRedis(redisType string, conf string) (redisClient RedisClient, err error) {
  33. switch redisType {
  34. case "cluster": // 集群
  35. redisClient, err = redis.InitClusterRedis(conf)
  36. default: // 默认走单机
  37. redisClient, err = redis.InitStandaloneRedis(conf)
  38. }
  39. return
  40. }
  41. // redis没有key的错误
  42. const RedisNoKeyErr = "redis: nil"