tweaks for non-zrok traffic

This commit is contained in:
Michael Quigley
2023-06-16 13:18:46 -04:00
parent 1578ecca98
commit 36d267256f
4 changed files with 64 additions and 48 deletions

View File

@@ -209,14 +209,18 @@ mainLoop:
for {
select {
case usage := <-a.queue:
if err := a.enforce(usage); err != nil {
logrus.Errorf("error running enforcement: %v", err)
}
if time.Since(lastCycle) > a.cfg.Cycle {
if err := a.relax(); err != nil {
logrus.Errorf("error running relax cycle: %v", err)
if usage.ShareToken != "" {
if err := a.enforce(usage); err != nil {
logrus.Errorf("error running enforcement: %v", err)
}
lastCycle = time.Now()
if time.Since(lastCycle) > a.cfg.Cycle {
if err := a.relax(); err != nil {
logrus.Errorf("error running relax cycle: %v", err)
}
lastCycle = time.Now()
}
} else {
logrus.Warn("not enforcing for usage with no share token: %v", usage.String())
}
case <-time.After(a.cfg.Cycle):