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
	Set(n int)
}

type Options struct {
	Network string
	Addr    string
	Client  string
	Src     string
}

type Option func(opts *Options)

func NetworkOption(network string) Option {
	return func(opts *Options) {
		opts.Network = network
	}
}

func AddrOption(addr string) Option {
	return func(opts *Options) {
		opts.Addr = addr
	}
}

func ClientOption(client string) Option {
	return func(opts *Options) {
		opts.Client = client
	}
}

func SrcOption(src string) Option {
	return func(opts *Options) {
		opts.Src = src
	}
}

type TrafficLimiter interface {
	In(ctx context.Context, key string, opts ...Option) Limiter
	Out(ctx context.Context, key string, opts ...Option) Limiter
}