From c671aa9aadb1ab0a056fe87039d03b6728c29b3d Mon Sep 17 00:00:00 2001
From: Joel Wetzell <git@jwetzell.com>
Date: Sat, 22 Nov 2025 14:25:38 -0600
Subject: [PATCH] rework udp client output to avoid closed connections

---
 udp-client.go | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/udp-client.go b/udp-client.go
index ba45c36..f91467b 100644
--- a/udp-client.go
+++ b/udp-client.go
@@ -12,6 +12,7 @@ type UDPClient struct {
 	Port   uint16
 	conn   net.Conn
 	router *Router
+	addr   *net.UDPAddr
 }
 
 func init() {
@@ -39,7 +40,7 @@ func init() {
 			portNum, ok := port.(float64)
 
 			if !ok {
-				return nil, fmt.Errorf("net.udp.client port must be uint16")
+				return nil, fmt.Errorf("net.udp.client port must be a number")
 			}
 
 			return &UDPClient{Host: hostString, Port: uint16(portNum), config: config}, nil
@@ -60,28 +61,36 @@ func (uc *UDPClient) RegisterRouter(router *Router) {
 }
 
 func (uc *UDPClient) Run() error {
+
 	addr, err := net.ResolveUDPAddr("udp", fmt.Sprintf("%s:%d", uc.Host, uc.Port))
 	if err != nil {
 		return err
 	}
-	client, err := net.DialUDP("udp", nil, addr)
-	if err != nil {
-		return err
-	}
 
-	uc.conn = client
+	uc.addr = addr
+
 	<-uc.router.Context.Done()
 	slog.Debug("router context done in module", "id", uc.config.Id)
 	return nil
 }
 
 func (uc *UDPClient) Output(payload any) error {
-	if uc.conn != nil {
-		payloadBytes, ok := payload.([]byte)
-		if !ok {
-			return fmt.Errorf("net.udp.client is only able to output bytes")
-		}
-		_, err := uc.conn.Write(payloadBytes)
+
+	client, err := net.DialUDP("udp", nil, uc.addr)
+	if err != nil {
+		return err
+	}
+
+	uc.conn = client
+
+	payloadBytes, ok := payload.([]byte)
+	if !ok {
+		return fmt.Errorf("net.udp.client is only able to output bytes")
+	}
+
+	_, err = uc.conn.Write(payloadBytes)
+
+	if err != nil {
 		return err
 	}
 	return nil