zrok/controller/limits/config.go

package limits

import "time"

const Unlimited = -1

type Config struct {
	Environments int
	Shares       int
	Bandwidth    *BandwidthConfig
	Cycle        time.Duration
	Enforcing    bool
}

type BandwidthConfig struct {
	PerAccount     *BandwidthPerPeriod
	PerEnvironment *BandwidthPerPeriod
	PerShare       *BandwidthPerPeriod
}

type BandwidthPerPeriod struct {
	Period  time.Duration
	Warning *Bandwidth
	Limit   *Bandwidth
}

type Bandwidth struct {
	Rx    int64
	Tx    int64
	Total int64
}

func DefaultConfig() *Config {
	return &Config{
		Environments: Unlimited,
		Shares:       Unlimited,
		Bandwidth: &BandwidthConfig{
			PerAccount: &BandwidthPerPeriod{
				Period: 365 * (24 * time.Hour),
				Warning: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
				Limit: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
			},
			PerEnvironment: &BandwidthPerPeriod{
				Period: 365 * (24 * time.Hour),
				Warning: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
				Limit: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
			},
			PerShare: &BandwidthPerPeriod{
				Period: 365 * (24 * time.Hour),
				Warning: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
				Limit: &Bandwidth{
					Rx:    Unlimited,
					Tx:    Unlimited,
					Total: Unlimited,
				},
			},
		},
		Enforcing: false,
		Cycle:     15 * time.Minute,
	}
}