eagle/internal/pkg/repeat.go

// Package pkg 重复提交模型封装
package pkg

import (
	"fmt"
	"time"

	"github.com/go-eagle/eagle/pkg/log"
	"github.com/go-eagle/eagle/pkg/redis"
	"github.com/go-eagle/eagle/pkg/utils"
)

// CRepeat define struct
type CRepeat struct {
	cRepeatClient redis.CheckRepeat
}

// NewCRepeat create a check repeat
func NewCRepeat() *CRepeat {
	return &CRepeat{
		cRepeatClient: redis.NewCheckRepeat(redis.RedisClient),
	}
}

// getKey return a check repeat key
func (c *CRepeat) getKey(userID int64, check string) string {
	key, err := utils.Md5(fmt.Sprintf("%d:%s", userID, check))
	if err != nil {
		log.Warnf("md5 string err: %v", err)
	}
	return key
}

// Set record a repeat value
func (c *CRepeat) Set(userID int64, check string, value interface{}, expiration time.Duration) error {
	return c.cRepeatClient.Set(c.getKey(userID, check), value, expiration)
}

// SetNX  set
func (c *CRepeat) SetNX(userID int64, check string, value interface{}, expiration time.Duration) (bool, error) {
	return c.cRepeatClient.SetNX(c.getKey(userID, check), value, expiration)
}

// Get get value
func (c *CRepeat) Get(userID int64, check string) (interface{}, error) {
	return c.cRepeatClient.Get(c.getKey(userID, check))
}

// Del delete
func (c *CRepeat) Del(userID int64, check string) int64 {
	return c.cRepeatClient.Del(c.getKey(userID, check))
}