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

末蓝、 2022-09-02 12:57 178阅读 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()
    }

有一定的缺陷哈，慎用。。。

发表评论取消回复

表情：

评论列表（有 0 条评论，178人围观）

还没有评论，来说两句吧...

相关阅读

相关 java sringboot 限制接口外部访问频率 ,通过apo切面实现每秒钟接口访问次数不得超过1次(基于ConcurrentHashMap)

使用AOP切面来实现接口访问频率限制。具体实现方法如下：创建一个注解@LimitRequestFrequency，用于标记需要进行访问频率限制的接口方法。该注解需要包含一个

曾经终败给现在/ 2023年10月10日 16:49/ 0 赞/ 17 阅读

相关 spring boot + redis 接口访问频率限制

@Component public class BlackInterceper implements HandlerInterceptor {

迷南。/ 2023年01月08日 02:21/ 0 赞/ 170 阅读

相关 django利用django-ratelimit设置接口请求频率限制

一、概述平台中需要编写接口供第三方调用，需要控制调用频率，需求为5s内调用一次后不得再次调用。 Django官方插件库中有个django-ratelimit插件可以

ゞ浴缸里的玫瑰/ 2022年12月27日 12:42/ 0 赞/ 180 阅读

相关 nginx 访问频率限制配置

一、概述由于业务需求，需要显示某个api接口，限制1分钟访问1次。二、nginx配置先来安装一个nginx yum install -y ngi

我会带着你远行/ 2022年12月27日 12:42/ 0 赞/ 155 阅读

相关 spring boot 限制接口访问次数

首先自定义一个元注解 import java.lang.annotation.Documented; import java.lang.annotation.

╰+攻爆jí腚メ/ 2022年10月01日 06:41/ 0 赞/ 336 阅读

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

代码片段如下： package utils import ( "time" "base.domain.com/glob

末蓝、/ 2022年09月02日 12:57/ 0 赞/ 179 阅读

相关 Spring mvc通过访问者IP限制接口访问频率

我们有的时候系统里面有的用户接口为了控制访问频率，需要使用到，笔记走一波环境： spring spring mvc Guava:谷歌高并发工具，下面是maven

心已赠人/ 2022年05月21日 13:10/ 0 赞/ 391 阅读

相关 Django——对IP访问频率进行限制

\\\\REST\_FRAMEWORK 配置 > 对使用 rest\_framework 框架的项目来说，可以使用框架的设置来对api的访问频率进行限制仅包含相

左手的ㄟ右手/ 2022年05月16日 05:24/ 0 赞/ 225 阅读

相关 DJango - 接口访问方法 Demo

目的 > 理解 Django 交互方法 > test1 > > > 访问接口，然后直接返回数据 (json) > > test2 > > > GET 与 PO

- 日理万妓/ 2022年05月14日 14:34/ 0 赞/ 211 阅读

相关 nginx配置限制同一个ip的访问频率

nginx可以通过`ngx_http_limit_conn_module`和`ngx_http_limit_req_module`配置来限制ip在同一时间段的访问次数.

刺骨的言语ヽ痛彻心扉/ 2022年02月01日 11:13/ 0 赞/ 1176 阅读