mirror of
https://github.com/openziti/zrok.git
synced 2025-06-20 01:37:52 +02:00
tweaks for non-zrok traffic
This commit is contained in:
parent
1578ecca98
commit
36d267256f
@ -209,6 +209,7 @@ mainLoop:
|
|||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case usage := <-a.queue:
|
case usage := <-a.queue:
|
||||||
|
if usage.ShareToken != "" {
|
||||||
if err := a.enforce(usage); err != nil {
|
if err := a.enforce(usage); err != nil {
|
||||||
logrus.Errorf("error running enforcement: %v", err)
|
logrus.Errorf("error running enforcement: %v", err)
|
||||||
}
|
}
|
||||||
@ -218,6 +219,9 @@ mainLoop:
|
|||||||
}
|
}
|
||||||
lastCycle = time.Now()
|
lastCycle = time.Now()
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
logrus.Warn("not enforcing for usage with no share token: %v", usage.String())
|
||||||
|
}
|
||||||
|
|
||||||
case <-time.After(a.cfg.Cycle):
|
case <-time.After(a.cfg.Cycle):
|
||||||
if err := a.relax(); err != nil {
|
if err := a.relax(); err != nil {
|
||||||
|
@ -46,12 +46,12 @@ func (a *Agent) Start() error {
|
|||||||
case event := <-a.events:
|
case event := <-a.events:
|
||||||
if usage, err := Ingest(event.Data()); err == nil {
|
if usage, err := Ingest(event.Data()); err == nil {
|
||||||
if err := a.cache.addZrokDetail(usage); err != nil {
|
if err := a.cache.addZrokDetail(usage); err != nil {
|
||||||
logrus.Error(err)
|
logrus.Errorf("unable to add zrok detail for: %v: %v", usage.String(), err)
|
||||||
}
|
}
|
||||||
shouldAck := true
|
shouldAck := true
|
||||||
for _, snk := range a.snks {
|
for _, snk := range a.snks {
|
||||||
if err := snk.Handle(usage); err != nil {
|
if err := snk.Handle(usage); err != nil {
|
||||||
logrus.Error(err)
|
logrus.Errorf("error handling usage: %v", err)
|
||||||
if shouldAck {
|
if shouldAck {
|
||||||
shouldAck = false
|
shouldAck = false
|
||||||
}
|
}
|
||||||
@ -59,11 +59,14 @@ func (a *Agent) Start() error {
|
|||||||
}
|
}
|
||||||
if shouldAck {
|
if shouldAck {
|
||||||
if err := event.Ack(); err != nil {
|
if err := event.Ack(); err != nil {
|
||||||
logrus.Error("unable to Ack message", err)
|
logrus.Errorf("unable to ack handled message: %v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
logrus.Error(err)
|
logrus.Errorf("unable to ingest '%v': %v", event.Data(), err)
|
||||||
|
if err := event.Ack(); err != nil {
|
||||||
|
logrus.Errorf("unable to ack unparseable message: %v", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -83,14 +83,6 @@ mainLoop:
|
|||||||
msgLoop:
|
msgLoop:
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case event := <-s.msgs:
|
|
||||||
if event.Body != nil {
|
|
||||||
s.events <- &ZitiEventAMQP{
|
|
||||||
data: ZitiEventJson(event.Body),
|
|
||||||
msg: event,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
case err, ok := <-s.errs:
|
case err, ok := <-s.errs:
|
||||||
if err != nil || !ok {
|
if err != nil || !ok {
|
||||||
logrus.Error(err)
|
logrus.Error(err)
|
||||||
@ -99,6 +91,21 @@ mainLoop:
|
|||||||
|
|
||||||
case <-s.close:
|
case <-s.close:
|
||||||
break mainLoop
|
break mainLoop
|
||||||
|
|
||||||
|
case event, ok := <-s.msgs:
|
||||||
|
if !ok {
|
||||||
|
logrus.Warn("selecting on msg !ok")
|
||||||
|
break msgLoop
|
||||||
|
}
|
||||||
|
if event.Body != nil {
|
||||||
|
s.events <- &ZitiEventAMQP{
|
||||||
|
data: ZitiEventJson(event.Body),
|
||||||
|
msg: event,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
logrus.Warn("event body was nil!")
|
||||||
|
break msgLoop
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -22,6 +22,7 @@ func newInfluxWriter(cfg *InfluxConfig) *influxWriter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (w *influxWriter) Handle(u *Usage) error {
|
func (w *influxWriter) Handle(u *Usage) error {
|
||||||
|
if u.ShareToken != "" {
|
||||||
out := fmt.Sprintf("share: %v, circuit: %v", u.ShareToken, u.ZitiCircuitId)
|
out := fmt.Sprintf("share: %v, circuit: %v", u.ShareToken, u.ZitiCircuitId)
|
||||||
|
|
||||||
envId := fmt.Sprintf("%d", u.EnvironmentId)
|
envId := fmt.Sprintf("%d", u.EnvironmentId)
|
||||||
@ -56,6 +57,7 @@ func (w *influxWriter) Handle(u *Usage) error {
|
|||||||
} else {
|
} else {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user