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

add trx to action interfaces; implement limit and relax for share (#276)

Browse Source
This commit is contained in:
Michael Quigley
2023-03-27 13:51:48 -04:00
committed by Kenneth Bingham
parent bf48ddec76
commit 427ef7e3f9
12 changed files with 72 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
controller/limits/shareLimitAction.go
View File

@ -1,8 +1,10 @@
package limits
import (
"github.com/jmoiron/sqlx"
"github.com/openziti/edge/rest_management_api_client"
"github.com/openziti/zrok/controller/store"
"github.com/openziti/zrok/controller/zrokEdgeSdk"
"github.com/sirupsen/logrus"
)
@ -15,7 +17,18 @@ func newShareLimitAction(str *store.Store, edge *rest_management_api_client.Ziti
return &shareLimitAction{str, edge}
}
func (a *shareLimitAction) HandleShare(s *store.Share, rxBytes, txBytes int64, limit *BandwidthPerPeriod) error {
func (a *shareLimitAction) HandleShare(s *store.Share, _, _ int64, _ *BandwidthPerPeriod, trx *sqlx.Tx) error {
logrus.Infof("limiting '%v'", s.Token)
env, err := a.str.GetEnvironment(s.EnvironmentId, trx)
if err != nil {
return err
}
if err := zrokEdgeSdk.DeleteServicePolicyDial(env.ZId, s.Token, a.edge); err != nil {
return err
}
logrus.Infof("removed service dial policy for '%v'", s.Token)
return nil
}