V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
The Go Programming Language
http://golang.org/
Go Playground
Go Projects
Revel Web Framework
rv54ntjwfm3ug8
V2EX  ›  Go 编程语言

Go 如何根据请求头限制请求频率?

  •  
  •   rv54ntjwfm3ug8 · 2021-05-02 08:53:53 +08:00 via iPhone · 3155 次点击
    这是一个创建于 1332 天前的主题,其中的信息可能已经有所发展或是发生改变。
    以前一直在写 Rust,Rust 可以用 actix-ratelimit 这个轮子非常方便地根据请求头中的 CF-Connecting-IP 限制频率。请问 Go 中有类似的轮子或者实现方法吗?顺便求推荐一个 Go 语言 Web 框架。
    14 条回复    2021-05-08 09:42:41 +08:00
    LeeReamond
        1
    LeeReamond  
       2021-05-02 10:02:49 +08:00 via Android   ❤️ 3
    怪问题。用 rust 做 web 说明你是一个底层开发,但是却问了一个调包侠都不屑问的简单问题。而且为什么要局限在 cloudflare 的 header ?
    janxin
        2
    janxin  
       2021-05-02 10:13:25 +08:00
    @LeeReamond 用 Rust 的也可能是调包侠 =.= 当然也可能是 DRY 的忠实信徒
    treblex
        3
    treblex  
       2021-05-02 10:44:08 +08:00 via iPhone
    找一个存储方案,然后写个中间件,获取 ip,写记录,判断是否符合规则,abort 请求
    rv54ntjwfm3ug8
        4
    rv54ntjwfm3ug8  
    OP
       2021-05-02 10:45:07 +08:00 via iPhone   ❤️ 1
    @LeeReamond #1 刚接触 Go,想问问有什么好的解决方法。不太喜欢重复造轮子,所以想问问有没有轮子。
    herozzm
        5
    herozzm  
       2021-05-02 10:56:05 +08:00 via Android   ❤️ 1
    将记录写入 redis,然后去出来对比就行了
    lesismal
        6
    lesismal  
       2021-05-02 11:24:26 +08:00
    请求头只是个 key,所以请求头跟问题本身无关。只问怎么限流就行了

    多节点就 redis 之类的:
    https://github.com/go-redis/redis_rate

    单节点就进程内,官方扩展包:
    https://github.com/golang/time
    或者其他第三方实现挺多的、自己手撸也不难
    LeeReamond
        7
    LeeReamond  
       2021-05-02 11:47:19 +08:00 via Android
    @theklf4 应该也不算重复造轮子,毕竟只是个计数器而已,你有发帖的时间应该也写完了。。
    Wincer
        8
    Wincer  
       2021-05-02 12:48:54 +08:00   ❤️ 1
    框架可以试试 gin,https://github.com/julianshen/gin-limiter 这个是基于 gin 框架的限流中间件。
    Akkuman
        9
    Akkuman  
       2021-05-02 14:23:50 +08:00 via Android
    信号量
    com314159
        10
    com314159  
       2021-05-02 14:57:18 +08:00
    1 、 取 http 头中的 ip, 怎么取看你用的什么框架, 如 gin 请自己看看文档
    2 、 使用限流器, 如 go 官方的也行, 限流
    matrix67
        11
    matrix67  
       2021-05-02 16:06:03 +08:00
    @LeeReamond #1 哈哈哈 调包侠 有点形象!!!我就是调包侠
    BeautifulSoap
        12
    BeautifulSoap  
       2021-05-02 16:58:35 +08:00 via Android
    这种程度的东西还能被叫轮子?
    直接一个 map,里面存 ip 和对应的计数不就行了,一丁点代码的事,有必要找轮子吗(当然你需要考虑线程安全,不过 lz 既然写 rust 的这方面想必信手捏来)
    如果你的服务是分布的或不想纠结线程安全,那就用 redis 代替内存中的 map,代码更加简单
    wyx119911
        13
    wyx119911  
       2021-05-02 20:51:28 +08:00
    回包加上 cookie 记录。同时加上 HttpOnly 和过期时间,请求过来时候判断下就好了
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   4911 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 26ms · UTC 01:10 · PVG 09:10 · LAX 17:10 · JFK 20:10
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.