extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-08-15 18:42:32 +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/share.go
View File

@ -3,6 +3,7 @@ package controller
import (
"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"
@ -12,10 +13,10 @@ import (
)
type shareHandler struct {
cfg *LimitsConfig
cfg *limits.Config
}
func newShareHandler(cfg *LimitsConfig) *shareHandler {
func newShareHandler(cfg *limits.Config) *shareHandler {
return &shareHandler{cfg: cfg}
}
@ -144,7 +145,7 @@ func (h *shareHandler) Handle(params share.ShareParams, principal *rest_model_zr
}
func (h *shareHandler) checkLimits(principal *rest_model_zrok.Principal, envs []*store.Environment, tx *sqlx.Tx) error {
if !principal.Limitless && h.cfg.Shares > Unlimited {
if !principal.Limitless && h.cfg.Shares > limits.Unlimited {
total := 0
for i := range envs {
shrs, err := str.FindSharesForEnvironment(envs[i].Id, tx)