Go-API接口访问频率限制DEMO-向日葵屋

Go-API接口访问频率限制DEMO

末蓝、 2022-09-02 12:57 267阅读 0赞

代码片段如下：

package utils
import (
    "time"
    "base.domain.com/global"
    "github.com/go-redis/redis"
)
// Limiter 定义属性
type Limiter struct {
    redisClient *redis.Client
}
// NewLimiter 创建新的limiter
func NewLimiter() *Limiter {
    ipConfig := global.GetConfig("ip")
    redis := NewRedis(ipConfig.GetString("IP_FREQUENCE_LIMIT_REDIS"))
    return &Limiter{redisClient: redis.Client}
}
// Allow 通过redis的value判断第几次访问并返回是否允许访问
func (l *Limiter) Allow(key string, visitNumber int64, visitPeriod time.Duration) bool {
    currentLength := l.redisClient.LLen(key).Val()
    if currentLength >= visitNumber {
        return false
    }
    if v := l.redisClient.Exists(key).Val(); v == 0 {
        pipe := l.redisClient.TxPipeline()
        pipe.RPush(key, key)
        pipe.Expire(key, visitPeriod)
        _, _ = pipe.Exec()
    } else {
        l.redisClient.RPushX(key, key)
    }
    return true
}
// DuplicateSubmit 限制表单重复提交
func (l *Limiter) DuplicateSubmit(key string, visitNumber int64, visitPeriod time.Duration) bool {
    return l.redisClient.SetNX(key, visitNumber, visitPeriod).Val()
}