add conn limiter

This commit is contained in:
ginuerzh
2022-09-14 00:14:20 +08:00
parent e7a104651a
commit 45b7ac2021
4 changed files with 34 additions and 16 deletions

View File

@ -0,0 +1,15 @@
package traffic
import "context"
type Limiter interface {
// Wait blocks with the requested n and returns the result value,
// the returned value is less or equal to n.
Wait(ctx context.Context, n int) int
Limit() int
}
type TrafficLimiter interface {
In(key string) Limiter
Out(key string) Limiter
}