mqttim

commit 8ee8bb3e245874cde667086b1712ec0f9c78bf5d
parent 1866cb9607a9f84c9d90b2b6c72942490c9c8487
Author: Natasha Kerensikova <natgh@instinctive.eu>
Date:   Wed, 19 Mar 2025 18:39:55 +0000

MQTT keepalive
Diffstat:
M
main.go
 | 
26
++++++++++++++++++++++----

1 file changed, 22 insertions(+), 4 deletions(-)
diff --git a/main.go b/main.go
@@ -42,10 +42,11 @@ type LogConfig struct {
 }
 
 type MqttConfig struct {
-	Server   string
-	Session  string
-	UserName string
-	Password string
+	Server    string
+	Session   string
+	UserName  string
+	Password  string
+	Keepalive int
 }
 
 type Config struct {
@@ -177,6 +178,7 @@ func main() {
 	go subscribeAll(m, ircQueue)
 	go mqttReader(m, l, ircQueue, &config)
 	go ircSender(&config.Irc, i, ircQueue, cmdQueue)
+	go mqttKeepalive(m, ircQueue, config.Mqtt.Keepalive)
 	i.Loop()
 }
 
@@ -186,6 +188,22 @@ func dup(src []byte) []byte {
 	return res
 }
 
+func mqttKeepalive(m *mqtt.Client, c chan<- Msg, keepalive int) {
+	if keepalive <= 0 {
+		return
+	}
+
+	period := time.Duration(keepalive) * time.Second
+
+	for {
+		time.Sleep(period)
+
+		if err := m.Ping(nil); err != nil {
+			c <- errMsg("mqttPing", err)
+		}
+	}
+}
+
 func mqttReader(m *mqtt.Client, l *mqttLogger, c chan<- Msg, config *Config) {
 	var big *mqtt.BigMessage