extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-08-15 10:32:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

roughed in limits model that incorporates bandwidth limit specs (#235)

Browse Source
This commit is contained in:
Michael Quigley
2023-03-10 14:25:29 -05:00
parent 86380d1836
commit 83ab21f00c
5 changed files with 65 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
controller/enable.go
View File

@ -5,6 +5,7 @@ import (
"encoding/json"
"github.com/go-openapi/runtime/middleware"
"github.com/jmoiron/sqlx"
"github.com/openziti/zrok/controller/limits"
"github.com/openziti/zrok/controller/store"
"github.com/openziti/zrok/controller/zrokEdgeSdk"
"github.com/openziti/zrok/rest_model_zrok"
@ -14,10 +15,10 @@ import (
)
type enableHandler struct {
cfg *LimitsConfig
cfg *limits.Config
}
func newEnableHandler(cfg *LimitsConfig) *enableHandler {
func newEnableHandler(cfg *limits.Config) *enableHandler {
return &enableHandler{cfg: cfg}
}
@ -100,7 +101,7 @@ func (h *enableHandler) Handle(params environment.EnableParams, principal *rest_
}
func (h *enableHandler) checkLimits(principal *rest_model_zrok.Principal, tx *sqlx.Tx) error {
if !principal.Limitless && h.cfg.Environments > Unlimited {
if !principal.Limitless && h.cfg.Environments > limits.Unlimited {
envs, err := str.FindEnvironmentsForAccount(int(principal.ID), tx)
if err != nil {
return errors.Errorf("unable to find environments for account '%v': %v", principal.Email, err)