standalone_redis.go 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277
  1. package redis
  2. import (
  3. "context"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "github.com/go-redis/redis/v8"
  8. "strconv"
  9. "time"
  10. )
  11. // StandaloneRedisClient
  12. // @Description: 单机redis客户端
  13. type StandaloneRedisClient struct {
  14. redisClient *redis.Client
  15. }
  16. func InitStandaloneRedis(config string) (standaloneRedis *StandaloneRedisClient, err error) {
  17. var cf map[string]string
  18. err = json.Unmarshal([]byte(config), &cf)
  19. if err != nil {
  20. return
  21. }
  22. //if _, ok := cf["key"]; !ok {
  23. // cf["key"] = DefaultKey
  24. //}
  25. if _, ok := cf["conn"]; !ok {
  26. err = errors.New("config has no conn key")
  27. return
  28. }
  29. // db库
  30. dbNum := 0
  31. // 如果指定了db库
  32. if _, ok := cf["dbNum"]; ok {
  33. dbNum, err = strconv.Atoi(cf["dbNum"])
  34. if err != nil {
  35. return
  36. }
  37. }
  38. // 密码
  39. if _, ok := cf["password"]; !ok {
  40. cf["password"] = ""
  41. }
  42. client := redis.NewClient(&redis.Options{
  43. Addr: cf["conn"],
  44. Password: cf["password"],
  45. DB: dbNum,
  46. //PoolSize: 10, //连接池最大socket连接数,默认为10倍CPU数, 10 * runtime.NumCPU(暂不配置)
  47. })
  48. _, err = client.Ping(context.TODO()).Result()
  49. if err != nil {
  50. err = errors.New("redis 链接失败:" + err.Error())
  51. return
  52. }
  53. standaloneRedis = &StandaloneRedisClient{redisClient: client}
  54. return
  55. }
  56. // Get
  57. // @Description: 根据key获取数据(其实是返回的字节编码)
  58. // @receiver rc
  59. // @param key
  60. // @return interface{}
  61. func (rc *StandaloneRedisClient) Get(key string) interface{} {
  62. data, err := rc.redisClient.Get(context.TODO(), key).Bytes()
  63. if err != nil {
  64. return nil
  65. }
  66. return data
  67. }
  68. func (rc *StandaloneRedisClient) SetEX(key string, val interface{}, timeout time.Duration) bool {
  69. _, err := rc.redisClient.SetEX(context.TODO(), key, val, timeout).Result()
  70. if err != nil {
  71. return false
  72. }
  73. return true
  74. }
  75. // RedisBytes
  76. // @Description: 根据key获取字节编码数据
  77. // @receiver rc
  78. // @param key
  79. // @return data
  80. // @return err
  81. func (rc *StandaloneRedisClient) RedisBytes(key string) (data []byte, err error) {
  82. data, err = rc.redisClient.Get(context.TODO(), key).Bytes()
  83. return
  84. }
  85. // RedisString
  86. // @Description: 根据key获取字符串数据
  87. // @receiver rc
  88. // @param key
  89. // @return data
  90. // @return err
  91. func (rc *StandaloneRedisClient) RedisString(key string) (data string, err error) {
  92. data, err = rc.redisClient.Get(context.TODO(), key).Result()
  93. return
  94. }
  95. // RedisInt
  96. // @Description: 根据key获取int数据
  97. // @receiver rc
  98. // @param key
  99. // @return data
  100. // @return err
  101. func (rc *StandaloneRedisClient) RedisInt(key string) (data int, err error) {
  102. data, err = rc.redisClient.Get(context.TODO(), key).Int()
  103. return
  104. }
  105. // Put
  106. // @Description: put一个数据到redis
  107. // @receiver rc
  108. // @param key
  109. // @param val
  110. // @param timeout
  111. // @return error
  112. func (rc *StandaloneRedisClient) Put(key string, val interface{}, timeout time.Duration) error {
  113. var err error
  114. err = rc.redisClient.SetEX(context.TODO(), key, val, timeout).Err()
  115. if err != nil {
  116. return err
  117. }
  118. err = rc.redisClient.HSet(context.TODO(), DefaultKey, key, true).Err()
  119. return err
  120. }
  121. // SetNX
  122. // @Description: 设置一个会过期时间的值
  123. // @receiver rc
  124. // @param key
  125. // @param val
  126. // @param timeout
  127. // @return bool
  128. func (rc *StandaloneRedisClient) SetNX(key string, val interface{}, timeout time.Duration) bool {
  129. result, err := rc.redisClient.SetNX(context.TODO(), key, val, timeout).Result()
  130. if err != nil {
  131. return false
  132. }
  133. return result
  134. }
  135. // Delete
  136. // @Description: 删除redis中的键值对
  137. // @receiver rc
  138. // @param key
  139. // @return error
  140. func (rc *StandaloneRedisClient) Delete(key string) error {
  141. var err error
  142. err = rc.redisClient.Del(context.TODO(), key).Err()
  143. if err != nil {
  144. return err
  145. }
  146. err = rc.redisClient.HDel(context.TODO(), DefaultKey, key).Err()
  147. return err
  148. }
  149. // IsExist
  150. // @Description: 根据key判断是否写入缓存中
  151. // @receiver rc
  152. // @param key
  153. // @return bool
  154. func (rc *StandaloneRedisClient) IsExist(key string) bool {
  155. result, err := rc.redisClient.Exists(context.TODO(), key).Result()
  156. if err != nil {
  157. return false
  158. }
  159. if result == 0 {
  160. _ = rc.redisClient.HDel(context.TODO(), DefaultKey, key).Err()
  161. return false
  162. }
  163. return true
  164. }
  165. // LPush
  166. // @Description: 写入list
  167. // @receiver rc
  168. // @param key
  169. // @param val
  170. // @return error
  171. func (rc *StandaloneRedisClient) LPush(key string, val interface{}) error {
  172. data, _ := json.Marshal(val)
  173. err := rc.redisClient.LPush(context.TODO(), key, data).Err()
  174. return err
  175. }
  176. // Brpop
  177. // @Description: 从list中读取
  178. // @receiver rc
  179. // @param key
  180. // @param callback
  181. func (rc *StandaloneRedisClient) Brpop(key string, callback func([]byte)) {
  182. values, err := rc.redisClient.BRPop(context.TODO(), 1*time.Second, key).Result()
  183. if err != nil {
  184. return
  185. }
  186. if len(values) < 2 {
  187. fmt.Println("assert is wrong")
  188. return
  189. }
  190. callback([]byte(values[1]))
  191. }
  192. // GetRedisTTL
  193. // @Description: 获取key的过期时间
  194. // @receiver rc
  195. // @param key
  196. // @return time.Duration
  197. func (rc *StandaloneRedisClient) GetRedisTTL(key string) time.Duration {
  198. value, err := rc.redisClient.TTL(context.TODO(), key).Result()
  199. if err != nil {
  200. return 0
  201. }
  202. return value
  203. }
  204. // Incrby
  205. // @Description: 设置自增值
  206. // @receiver rc
  207. // @param key
  208. // @param num
  209. // @return interface{}
  210. // @return error
  211. func (rc *StandaloneRedisClient) Incrby(key string, num int) (interface{}, error) {
  212. return rc.redisClient.IncrBy(context.TODO(), key, int64(num)).Result()
  213. }
  214. // Do
  215. // @Description: cmd执行redis命令
  216. // @receiver rc
  217. // @param commandName
  218. // @param args
  219. // @return reply
  220. // @return err
  221. func (rc *StandaloneRedisClient) Do(commandName string, args ...interface{}) (reply interface{}, err error) {
  222. newArgs := []interface{}{commandName}
  223. newArgs = append(newArgs, args...)
  224. return rc.redisClient.Do(context.TODO(), newArgs...).Result()
  225. }
  226. func (rc *StandaloneRedisClient) RunLUA(command string, ctx context.Context, keys []string, args ...interface{}) bool {
  227. script := redis.NewScript(command)
  228. result, err := script.Run(ctx, rc.redisClient, keys, args).Int()
  229. if err != nil {
  230. return false
  231. }
  232. if result == 1 {
  233. return true
  234. }
  235. return false
  236. }