From b1063e9a79337f70d6b4e173ed1e03178fbc243e Mon Sep 17 00:00:00 2001 From: TiganeteaRobert Date: Wed, 10 Aug 2022 12:46:43 +0300 Subject: [PATCH] Fix panic on failure target WriteOversized() and WriteInvalid() (closes #194) --- cmd/cli/cli.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/cmd/cli/cli.go b/cmd/cli/cli.go index a216c2f6..fa73a28c 100644 --- a/cmd/cli/cli.go +++ b/cmd/cli/cli.go @@ -220,6 +220,9 @@ func sourceWriteFunc(t targetiface.Target, ft failureiface.Failure, tr transform if len(messagesToSend) > 0 { err2 := retry.Exponential(5, time.Second, "failureTarget.WriteOversized", func() error { res, err := ft.WriteOversized(t.MaximumAllowedMessageSizeBytes(), messagesToSend) + if err != nil { + return err + } if len(res.Oversized) != 0 || len(res.Invalid) != 0 { log.Fatal("Oversized message transformation resulted in new oversized / invalid messages") } @@ -238,6 +241,9 @@ func sourceWriteFunc(t targetiface.Target, ft failureiface.Failure, tr transform if len(messagesToSend) > 0 { err3 := retry.Exponential(5, time.Second, "failureTarget.WriteInvalid", func() error { res, err := ft.WriteInvalid(messagesToSend) + if err != nil { + return err + } if len(res.Oversized) != 0 || len(res.Invalid) != 0 { log.Fatal("Invalid message transformation resulted in new invalid / oversized messages") }