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

2022-09-02 09:46:42 +08:00
 AugOmin

官方包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 的总数变少了
926 次点击
所在节点    Go 编程语言
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/877175

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX