30 lines
820 B
Go
30 lines
820 B
Go
package limit
|
||
|
||
import "context"
|
||
|
||
type LimitMethodConf interface {
|
||
// 获取需要限流的接口方法名
|
||
GetLimitMethods() map[string]struct{}
|
||
}
|
||
|
||
// 限流服务接口
|
||
type LimitService interface {
|
||
// 查询option值
|
||
GetOption() LimitOption
|
||
// 修改option结构体内的参数，支持实时更新
|
||
SetOption(option LimitOption)
|
||
// 直接消费，如果请求过于频繁，则直接拒绝，返回error
|
||
Consume() error
|
||
// 阻塞消费，如果请求过于频繁，通过ctx设置的Deadline或者Timeout来等待一段时间，若资源不够，亦返回error
|
||
ConsumeWithContext(ctx context.Context) error
|
||
// 克隆复制一个新的限流服务接口
|
||
Clone() LimitService
|
||
}
|
||
|
||
type LimitOption struct {
|
||
// 令牌发放的频率，单位毫秒
|
||
LimitMillisecond int64
|
||
// 每次发放的个数
|
||
Burst int
|
||
}
|