app

type Limit ¶

type Limit struct {
	Executor bitrate.Bitrate
	Address  bitrate.Bitrate
	Updated  bool
}

type Limiter ¶

type Limiter struct {
	// contains filtered or unexported fields
}

Limiter keeps track of the bandwidth being used by individual debuglets as well as the maximum bandwidth any debuglet is allowed to use on a executor level as well as per destination.

func NewLimiter(l *zap.Logger) *Limiter

func (l *Limiter) ExecutorCapacity() bitrate.Bitrate

func (l *Limiter) GetAddrLimit(ID uuid.UUID, addr string) (bitrate.Bitrate, bool, error)

func (l *Limiter) GetExecLimit(ID uuid.UUID) (bitrate.Bitrate, bool, error)

func (l *Limiter) GetLimit(ID uuid.UUID, addr string) (Limit, error)

func (l *Limiter) InsertDebuglet(ID uuid.UUID, minimum, maximum bitrate.Bitrate, addrs []string) error

func (l *Limiter) RemoveDebuglet(ID uuid.UUID)

func (l *Limiter) SetAddrCapacity(addr string, c bitrate.Bitrate)

func (l *Limiter) SetExecutorCapacity(c bitrate.Bitrate)

func (l *Limiter) Wait(ctx context.Context, direction TransferDirection, ID uuid.UUID, addr string, size bitrate.Bitrate) error

type TransferDirection ¶

type TransferDirection = int

const (
	TransferIn TransferDirection = iota
	TransferOut
)

type UsageLimits ¶

type UsageLimits struct {
	ExecutorRatelimit    bitrate.Bitrate
	ExecutorBurst        bitrate.Bitrate
	DestinationRatelimit bitrate.Bitrate
	DestinationBurst     bitrate.Bitrate
}

type UsageTracker ¶

type UsageTracker struct {
	// contains filtered or unexported fields
}

UsageTracker tracks the amount of bandwidth that is being used by an individual debuglet. The idea is to set the executor and destination specific limit that is allowed by the debuglet and then UsageTracker.Wait will block the caller for the amount of time to ratelimit the bandwidth usage.

func NewUsageTracker(l *zap.Logger) *UsageTracker

func (u *UsageTracker) Upsert(destination string, limits UsageLimits)

func (u *UsageTracker) Wait(ctx context.Context, dir TransferDirection, destination string, size bitrate.Bitrate) error