This commit is contained in:
Michael Quigley 2024-06-05 21:11:23 -04:00
parent 83e9e84dad
commit cb4afd4a0f
No known key found for this signature in database
GPG Key ID: 9B60314A9DD20A62

View File

@ -323,7 +323,7 @@ func (a *Agent) enforce(u *metrics.Usage) error {
return err return err
} }
exceededLc, rxBytes, txBytes, err := a.isOverLimitClass(u, ul.toBandwidthArray(sdk.BackendMode(shr.BackendMode))) exceededLc, rxBytes, txBytes, err := a.hasExceededBandwidthLimit(u, ul.toBandwidthArray(sdk.BackendMode(shr.BackendMode)))
if err != nil { if err != nil {
return errors.Wrap(err, "error checking limit classes") return errors.Wrap(err, "error checking limit classes")
} }
@ -369,9 +369,7 @@ func (a *Agent) enforce(u *metrics.Usage) error {
lcId := exceededLc.GetLimitClassId() lcId := exceededLc.GetLimitClassId()
je.LimitClassId = &lcId je.LimitClassId = &lcId
} }
_, err := a.str.CreateBandwidthLimitJournalEntry(je, trx) if _, err := a.str.CreateBandwidthLimitJournalEntry(je, trx); err != nil {
if err != nil {
return err return err
} }
acct, err := a.str.GetAccount(int(u.AccountId), trx) acct, err := a.str.GetAccount(int(u.AccountId), trx)
@ -504,7 +502,7 @@ func (a *Agent) relax() error {
return nil return nil
} }
func (a *Agent) isOverLimitClass(u *metrics.Usage, bwcs []store.BandwidthClass) (store.BandwidthClass, int64, int64, error) { func (a *Agent) hasExceededBandwidthLimit(u *metrics.Usage, bwcs []store.BandwidthClass) (store.BandwidthClass, int64, int64, error) {
periodBw := make(map[int]struct { periodBw := make(map[int]struct {
rx int64 rx int64
tx int64 tx int64