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

environment limit and relax actions (#276)

Browse Source
This commit is contained in:
Michael Quigley
2023-03-27 14:01:31 -04:00
committed by Kenneth Bingham
parent 427ef7e3f9
commit f78ead73f7
4 changed files with 64 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
controller/limits/shareLimitAction.go
View File

@ -17,18 +17,18 @@ func newShareLimitAction(str *store.Store, edge *rest_management_api_client.Ziti
return &shareLimitAction{str, edge}
}
func (a *shareLimitAction) HandleShare(s *store.Share, _, _ int64, _ *BandwidthPerPeriod, trx *sqlx.Tx) error {
logrus.Infof("limiting '%v'", s.Token)
func (a *shareLimitAction) HandleShare(shr *store.Share, _, _ int64, _ *BandwidthPerPeriod, trx *sqlx.Tx) error {
logrus.Infof("limiting '%v'", shr.Token)
env, err := a.str.GetEnvironment(s.EnvironmentId, trx)
env, err := a.str.GetEnvironment(shr.EnvironmentId, trx)
if err != nil {
return err
}
if err := zrokEdgeSdk.DeleteServicePolicyDial(env.ZId, s.Token, a.edge); err != nil {
if err := zrokEdgeSdk.DeleteServicePolicyDial(env.ZId, shr.Token, a.edge); err != nil {
return err
}
logrus.Infof("removed service dial policy for '%v'", s.Token)
logrus.Infof("removed dial service policy for '%v'", shr.Token)
return nil
}