string.go 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226
  1. // Copyright 2019 github.com. All rights reserved.
  2. // Use of this source code is governed by github.com.
  3. package redis
  4. import (
  5. "time"
  6. )
  7. func (p *Redis) Get(key string) (res string, err error) {
  8. // 安全检查
  9. if p == nil {
  10. return res, errRedis
  11. }
  12. if p.cluster {
  13. // 客户端安全检查
  14. if p.cclient == nil {
  15. return res, errRedisCClient
  16. }
  17. return p.cclient.Get(key).Result()
  18. }
  19. // 客户端安全检查
  20. if p.client == nil {
  21. return res, errRedisClient
  22. }
  23. return p.client.Get(key).Result()
  24. }
  25. func (p *Redis) Set(key string, value interface{}) (res string, err error) {
  26. // 安全检查
  27. if p == nil {
  28. return res, errRedis
  29. }
  30. if p.cluster {
  31. // 客户端安全检查
  32. if p.cclient == nil {
  33. return res, errRedisCClient
  34. }
  35. return p.cclient.Set(key, value, 0).Result()
  36. }
  37. // 客户端安全检查
  38. if p.client == nil {
  39. return res, errRedisClient
  40. }
  41. return p.client.Set(key, value, 0).Result()
  42. }
  43. func (p *Redis) SetEx(key string, seconds int64, value interface{}) (res string, err error) {
  44. // 安全检查
  45. if p == nil {
  46. return res, errRedis
  47. }
  48. if p.cluster {
  49. // 客户端安全检查
  50. if p.cclient == nil {
  51. return res, errRedisCClient
  52. }
  53. return p.cclient.Set(key, value, time.Duration(seconds)*time.Second).Result()
  54. }
  55. // 客户端安全检查
  56. if p.client == nil {
  57. return res, errRedisClient
  58. }
  59. return p.client.Set(key, value, time.Duration(seconds)*time.Second).Result()
  60. }
  61. func (p *Redis) MGet(keys ...string) (res []string, err error) {
  62. // 安全检查
  63. if p == nil {
  64. return res, errRedis
  65. }
  66. vals := make([]interface{}, len(keys))
  67. if p.cluster {
  68. // 客户端安全检查
  69. if p.cclient == nil {
  70. return res, errRedisCClient
  71. }
  72. if vals, err = p.cclient.MGet(keys...).Result(); err != nil {
  73. return res, err
  74. }
  75. // 转为[]string
  76. return strings(vals, err)
  77. }
  78. // 客户端安全检查
  79. if p.client == nil {
  80. return res, errRedisClient
  81. }
  82. if vals, err = p.client.MGet(keys...).Result(); err != nil {
  83. return res, err
  84. }
  85. // 转为[]string
  86. return strings(vals, err)
  87. }
  88. func (p *Redis) SetBit(key string, offset int64, value int) (res int64, err error) {
  89. // 安全检查
  90. if p == nil {
  91. return res, errRedis
  92. }
  93. if p.cluster {
  94. // 客户端安全检查
  95. if p.cclient == nil {
  96. return res, errRedisCClient
  97. }
  98. return p.cclient.SetBit(key, offset, value).Result()
  99. }
  100. // 客户端安全检查
  101. if p.client == nil {
  102. return res, errRedisClient
  103. }
  104. return p.client.SetBit(key, offset, value).Result()
  105. }
  106. func (p *Redis) GetBit(key string, offset int64) (res int64, err error) {
  107. // 安全检查
  108. if p == nil {
  109. return res, errRedis
  110. }
  111. if p.cluster {
  112. // 客户端安全检查
  113. if p.cclient == nil {
  114. return res, errRedisCClient
  115. }
  116. return p.cclient.GetBit(key, offset).Result()
  117. }
  118. // 客户端安全检查
  119. if p.client == nil {
  120. return res, errRedisClient
  121. }
  122. return p.client.GetBit(key, offset).Result()
  123. }
  124. func (p *Redis) SetNx(key string, value interface{}) (res bool, err error) {
  125. // 安全检查
  126. if p == nil {
  127. return res, errRedis
  128. }
  129. if p.cluster {
  130. // 客户端安全检查
  131. if p.cclient == nil {
  132. return res, errRedisCClient
  133. }
  134. return p.cclient.SetNX(key, value, 0).Result()
  135. }
  136. // 客户端安全检查
  137. if p.client == nil {
  138. return res, errRedisClient
  139. }
  140. return p.client.SetNX(key, value, 0).Result()
  141. }
  142. func (p *Redis) SetNxEx(key string, value interface{}, seconds int64) (res bool, err error) {
  143. // 安全检查
  144. if p == nil {
  145. return res, errRedis
  146. }
  147. if p.cluster {
  148. // 客户端安全检查
  149. if p.cclient == nil {
  150. return res, errRedisCClient
  151. }
  152. return p.cclient.SetNX(key, value, time.Duration(seconds)*time.Second).Result()
  153. }
  154. // 客户端安全检查
  155. if p.client == nil {
  156. return res, errRedisClient
  157. }
  158. return p.client.SetNX(key, value, time.Duration(seconds)*time.Second).Result()
  159. }
  160. func (p *Redis) Append(key, value string) (res int64, err error) {
  161. // 安全检查
  162. if p == nil {
  163. return res, errRedis
  164. }
  165. if p.cluster {
  166. // 客户端安全检查
  167. if p.cclient == nil {
  168. return res, errRedisCClient
  169. }
  170. return p.cclient.Append(key, value).Result()
  171. }
  172. // 客户端安全检查
  173. if p.client == nil {
  174. return res, errRedisClient
  175. }
  176. return p.client.Append(key, value).Result()
  177. }