32 lines
480 B
Go
32 lines
480 B
Go
package traffic
|
|
|
|
import (
|
|
limiter "github.com/go-gost/core/limiter/traffic"
|
|
)
|
|
|
|
type limitGenerator struct {
|
|
in int
|
|
out int
|
|
}
|
|
|
|
func newLimitGenerator(in, out int) *limitGenerator {
|
|
return &limitGenerator{
|
|
in: in,
|
|
out: out,
|
|
}
|
|
}
|
|
|
|
func (p *limitGenerator) In() limiter.Limiter {
|
|
if p == nil || p.in <= 0 {
|
|
return nil
|
|
}
|
|
return NewLimiter(p.in)
|
|
}
|
|
|
|
func (p *limitGenerator) Out() limiter.Limiter {
|
|
if p == nil || p.out <= 0 {
|
|
return nil
|
|
}
|
|
return NewLimiter(p.out)
|
|
}
|