backup/showbridge-go
1
0
Fork 0
mirror of https://github.com/jwetzell/showbridge-go.git synced 2026-04-26 21:05:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

rework udp client output to avoid closed connections

Browse Source
This commit is contained in:
Joel Wetzell
2025-11-22 14:25:38 -06:00
parent 1ae6dc3719
commit c671aa9aad
1 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
udp-client.go
View File

@@ -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