Make sure we exit the preflight hook if the context is cancelled

Signed-off-by: Mihai Todor <[email protected]>

internal/impl/kafka/enterprise/redpanda_migrator_input.go

@@ -180,6 +180,7 @@ func init() {
 
 					// Make multiple attempts until the output connects in the background.
 					// TODO: It would be nicer to somehow get notified when the output is ready.
+				loop:
 					for {
 						if err = kafka.FranzSharedClientUse(outputResource, res, func(details *kafka.FranzSharedClientInfo) error {
 							for _, topic := range topics {
@@ -206,7 +207,11 @@ func init() {
 							break
 						}
 
-						time.Sleep(time.Millisecond * 100)
+						select {
+						case <-time.After(100 * time.Millisecond):
+						case <-ctx.Done():
+							break loop
+						}
 					}
 				},
 				func(res *service.Resources) {