mirror of
https://github.com/openziti/zrok.git
synced 2024-11-22 16:13:47 +01:00
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package limits
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/openziti/zrok/controller/store"
|
|
"github.com/openziti/zrok/sdk/golang/sdk"
|
|
)
|
|
|
|
type configBandwidthClass struct {
|
|
periodInMinutes int
|
|
bw *Bandwidth
|
|
limitAction store.LimitAction
|
|
}
|
|
|
|
func newConfigBandwidthClasses(cfg *BandwidthPerPeriod) []store.BandwidthClass {
|
|
return []store.BandwidthClass{
|
|
&configBandwidthClass{
|
|
periodInMinutes: int(cfg.Period.Minutes()),
|
|
bw: cfg.Warning,
|
|
limitAction: store.WarningLimitAction,
|
|
},
|
|
&configBandwidthClass{
|
|
periodInMinutes: int(cfg.Period.Minutes()),
|
|
bw: cfg.Limit,
|
|
limitAction: store.LimitLimitAction,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (bc *configBandwidthClass) IsGlobal() bool {
|
|
return true
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetLimitClassId() int {
|
|
return -1
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetShareMode() sdk.ShareMode {
|
|
return ""
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetBackendMode() sdk.BackendMode {
|
|
return ""
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetPeriodMinutes() int {
|
|
return bc.periodInMinutes
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetRxBytes() int64 {
|
|
return bc.bw.Rx
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetTxBytes() int64 {
|
|
return bc.bw.Tx
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetTotalBytes() int64 {
|
|
return bc.bw.Total
|
|
}
|
|
|
|
func (bc *configBandwidthClass) GetLimitAction() store.LimitAction {
|
|
return bc.limitAction
|
|
}
|
|
|
|
func (bc *configBandwidthClass) String() string {
|
|
if out, err := json.Marshal(bc.bw); err == nil {
|
|
return fmt.Sprintf("Config<period: %d, %s, action: %s>", bc.periodInMinutes, string(out), bc.limitAction)
|
|
}
|
|
return "<<ERROR>>"
|
|
}
|