From 0cd0c2fa0b4b55c0d13edd7f4d705cf21848b30c Mon Sep 17 00:00:00 2001 From: Jieke Choo Date: Sun, 10 Mar 2024 01:35:09 +0800 Subject: [PATCH] feat: set MQTT client OnConnect before connect to broker (#1556) Signed-off-by: Jieke Choo --- pkg/transforms/mqttsecret.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pkg/transforms/mqttsecret.go b/pkg/transforms/mqttsecret.go index 2796d86fa..c2e788096 100644 --- a/pkg/transforms/mqttsecret.go +++ b/pkg/transforms/mqttsecret.go @@ -336,3 +336,10 @@ func (sender *MQTTSecretSender) ConnectToBroker(lc logger.LoggingClient, sp boot sender.preConnected = true } } + +// SetOnConnect sets the OnConnect Handler before client is connected so client can be captured. +func (sender *MQTTSecretSender) SetOnConnectHandler(onConnect MQTT.OnConnectHandler) { + + sender.opts.SetOnConnectHandler(onConnect) + +}