cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases27 Wiki Activity

respond early

Browse Source
This commit is contained in:
Cory Schwartz 2022-04-19 19:21:27 -07:00
parent fc42d03ea4
commit 9558e9f0ee
1 changed files with 4 additions and 1 deletions

5
gateway/node.go
View File

@ -165,9 +165,12 @@ func (gw *Node) checkTimestamp(at time.Time) error {
func (gw *Node) limit(ctx context.Context, tokens int) error { func (gw *Node) limit(ctx context.Context, tokens int) error {
ctx2, cancel := context.WithTimeout(ctx, gw.rateLimitTimeout) ctx2, cancel := context.WithTimeout(ctx, gw.rateLimitTimeout)
defer cancel() defer cancel()
if !gw.rateLimiter.AllowN(time.Now(), tokens) {
return fmt.Errorf("server busy")
}
err := gw.rateLimiter.WaitN(ctx2, tokens) err := gw.rateLimiter.WaitN(ctx2, tokens)
if err != nil { if err != nil {
return fmt.Errorf("rate limited: %w", err) return fmt.Errorf("server busy, cannot complete before timeout %w", err)
} }
return nil return nil
} }