Add trace logger events for Bandwidth toxic

Author: miry

cmd/server/server.go

@@ -12,6 +12,7 @@ import (
 
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
 
 	"github.com/Shopify/toxiproxy/v2"
 	"github.com/Shopify/toxiproxy/v2/collectors"
@@ -68,6 +69,7 @@ func run(cli cliArguments) {
 	}
 
 	logger := setupLogger()
+	log.Logger = logger
 
 	rand.Seed(cli.seed)
 

scripts/test-e2e

@@ -37,7 +37,7 @@ go run ../test/e2e/endpoint.go 2>&1 | sed -e 's/^/[web] /' &
 
 echo "=== Starting Toxiproxy"
 
-LOG_LEVEL=debug $server -proxy-metrics -runtime-metrics 2>&1 | sed -e 's/^/[toxiproxy] /' &
+LOG_LEVEL=trace $server -proxy-metrics -runtime-metrics 2>&1 | sed -e 's/^/[toxiproxy] /' &
 
 echo "=== Wait when services are available"
 

toxics/bandwidth.go

@@ -1,8 +1,11 @@
 package toxics
 
 import (
+	"fmt"
 	"time"
 
+	"github.com/rs/zerolog/log"
+
 	"github.com/Shopify/toxiproxy/v2/stream"
 )
 
@@ -13,10 +16,17 @@ type BandwidthToxic struct {
 }
 
 func (t *BandwidthToxic) Pipe(stub *ToxicStub) {
+	logger := log.With().
+		Str("component", "BandwidthToxic").
+		Str("method", "Pipe").
+		Str("toxic_type", "bandwidth").
+		Str("addr", fmt.Sprintf("%p", t)).
+		Logger()
 	var sleep time.Duration = 0
 	for {
 		select {
 		case <-stub.Interrupt:
+			logger.Trace().Msg("BandwidthToxic was interrupted")
 			return
 		case p := <-stub.Input:
 			if p == nil {
@@ -39,6 +49,7 @@ func (t *BandwidthToxic) Pipe(stub *ToxicStub) {
 					p.Data = p.Data[t.Rate*100:]
 					sleep -= 100 * time.Millisecond
 				case <-stub.Interrupt:
+					logger.Trace().Msg("BandwidthToxic was interrupted during writing data")
 					stub.Output <- p // Don't drop any data on the floor
 					return
 				}
@@ -50,6 +61,7 @@ func (t *BandwidthToxic) Pipe(stub *ToxicStub) {
 				sleep -= time.Since(start)
 				stub.Output <- p
 			case <-stub.Interrupt:
+				logger.Trace().Msg("BandwidthToxic was interrupted during writing data")
 				stub.Output <- p // Don't drop any data on the floor
 				return
 			}