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

关于 time/rate 中取消返回 token 的问题(bug?)

  •  
  •   AugOmin · 2022-09-02 09:46:42 +08:00 · 1103 次点击
    这是一个创建于 804 天前的主题,其中的信息可能已经有所发展或是发生改变。

    官方包golang.org/x/time中 cancel 的部分,返回 token 的代码( CancelAt )

    https://github.com/golang/time/blob/e5dcc9cfc0b9553953e355dde5bdf4ff9f82f742/rate/rate.go#L168

    	// calculate tokens to restore
    	// The duration between lim.lastEvent and r.timeToAct tells us how many tokens were reserved
    	// after r was obtained. These tokens should not be restored.
        // 这行代码多减了一次最近申请的预支的 token ,用测试代码说明一下
    	restoreTokens := float64(r.tokens) - r.limit.tokensFromDuration(r.lim.lastEvent.Sub(r.timeToAct))
    	if restoreTokens <= 0 {
    		return
    	}
    	// advance time to now
    	now, _, tokens := r.lim.advance(now)
    	// calculate new number of tokens
    	tokens += restoreTokens
    	if burst := float64(r.lim.burst); tokens > burst {
    		tokens = burst
    	}
    
    

    测试代码

    t0 := time.Now()
    
    l := NewLimiter(1, 10)
    l.ReserveN(t0, 5)       //桶里还剩 5 个 token
    r := l.ReserveN(t0, 10) // 桶里还剩 -5 个 token
    fmt.Printf("%+v\n", l)
    
    l.ReserveN(t0, 2) // 桶里还有 -7 个
    fmt.Printf("%+v\n", l)
    //
    r.CancelAt(t0) // -7 + (10 - 2) = 1
    fmt.Printf("%+v\n", l)
    
    l.ReserveN(t0, 8) // -7
    fmt.Printf("%+v\n", l)
    // 这样到了第七秒就只消费了 5+8+2=15 ,而取消之前应该是 5+10+2=17 ,取消之后 token 的总数变少了
    
    1 条回复    2024-08-30 20:02:38 +08:00
    oluoluo
        1
    oluoluo  
       75 天前
    确实好像是 bug
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   5685 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 27ms · UTC 06:28 · PVG 14:28 · LAX 22:28 · JFK 01:28
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.