cache.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. package cache
  2. import (
  3. "context"
  4. "github.com/go-redis/redis/v8"
  5. "log-server/global"
  6. "time"
  7. )
  8. const (
  9. TemporaryTaskPcCode = "%s:temporaryTaskPcCode:%s"
  10. )
  11. type Cache struct {
  12. }
  13. func (s *Cache) DelBatheHsCache(ctx context.Context, key string) (err error) {
  14. keys, err := global.GVA_REDIS.HKeys(ctx, key).Result()
  15. if err != nil {
  16. if err == redis.Nil {
  17. return nil
  18. }
  19. return
  20. }
  21. global.GVA_REDIS.HDel(ctx, key, keys...)
  22. //for _, v := range keys {
  23. // global.GVA_REDIS.HDel(ctx, key, v)
  24. //}
  25. return
  26. }
  27. func (s *Cache) DelBatheCache(ctx context.Context, key string) (err error) {
  28. keys, err := global.GVA_REDIS.Keys(ctx, key).Result()
  29. if err != nil {
  30. if err == redis.Nil {
  31. return nil
  32. }
  33. return
  34. }
  35. global.GVA_REDIS.Del(ctx, keys...)
  36. //for _, v := range keys {
  37. // fmt.Println(v)
  38. // global.GVA_REDIS.Del(ctx, v)
  39. //}
  40. return
  41. }
  42. func (s *Cache) GetCacheKeys(ctx context.Context, keys string) (key []string, err error) {
  43. key, err = global.GVA_REDIS.Keys(ctx, keys).Result()
  44. return
  45. }
  46. func (s *Cache) SetCacheNum(ctx context.Context, key string) (err error) {
  47. bl, err := s.ExistsKey(ctx, key)
  48. if err != nil {
  49. return err
  50. }
  51. if !bl {
  52. err = global.GVA_REDIS.Set(ctx, key, 1, time.Hour*48).Err()
  53. return err
  54. } else {
  55. err = global.GVA_REDIS.Incr(ctx, key).Err()
  56. return err
  57. }
  58. }
  59. func (s *Cache) SetCacheStr(ctx context.Context, key string, value interface{}) (err error) {
  60. err = global.GVA_REDIS.Set(ctx, key, value, time.Hour*30).Err()
  61. return err
  62. }
  63. func (s *Cache) GetCacheNum(ctx context.Context, key string) (num int, err error) {
  64. num, err = global.GVA_REDIS.Get(ctx, key).Int()
  65. if err != nil {
  66. if err == redis.Nil {
  67. return 0, nil
  68. }
  69. return 0, err
  70. }
  71. return num, err
  72. }
  73. func (s *Cache) GetCacheStr(ctx context.Context, key string) (str string, err error) {
  74. str, err = global.GVA_REDIS.Get(ctx, key).Result()
  75. if err != nil {
  76. if err == redis.Nil {
  77. return "", nil
  78. }
  79. return "", err
  80. }
  81. return str, err
  82. }
  83. func (s *Cache) ExistsKey(ctx context.Context, key string) (bool, error) {
  84. isNull, err := global.GVA_REDIS.Exists(ctx, key).Result()
  85. if err != nil {
  86. return false, err
  87. }
  88. if isNull == 0 {
  89. return false, nil
  90. } else {
  91. return true, nil
  92. }
  93. }