add concept of framing to net.tcp

This commit is contained in:
Joel Wetzell
2025-11-18 21:02:48 -06:00
parent 9ad60c8193
commit 86966132ca
5 changed files with 112 additions and 10 deletions

View File

@@ -3,7 +3,8 @@
{ {
"type": "net.tcp.server", "type": "net.tcp.server",
"params": { "params": {
"port": 8000 "port": 8000,
"framing": "LF"
} }
} }
] ]

View File

@@ -0,0 +1,6 @@
package framing
type Framer interface {
Frame([]byte) [][]byte
Clear()
}

View File

@@ -0,0 +1,33 @@
package framing
import (
"bytes"
)
type ByteSeparatorFramer struct {
buffer []byte
separator []byte
}
func NewByteSeparatorFramer(separator []byte) *ByteSeparatorFramer {
return &ByteSeparatorFramer{separator: separator, buffer: []byte{}}
}
func (bsf *ByteSeparatorFramer) Frame(data []byte) [][]byte {
messages := [][]byte{}
bsf.buffer = append(bsf.buffer, data...)
parts := bytes.Split(bsf.buffer, bsf.separator)
if len(parts) > 0 {
bsf.buffer = parts[len(parts)-1]
messages = parts[:len(parts)-1]
}
return messages
}
func (bsf *ByteSeparatorFramer) Clear() {
bsf.buffer = []byte{}
}

View File

@@ -6,12 +6,15 @@ import (
"log/slog" "log/slog"
"net" "net"
"time" "time"
"github.com/jwetzell/showbridge-go/internals/framing"
) )
type TCPClient struct { type TCPClient struct {
config ModuleConfig config ModuleConfig
Host string Host string
Port uint16 Port uint16
framer framing.Framer
} }
func init() { func init() {
@@ -42,7 +45,31 @@ func init() {
return nil, fmt.Errorf("tcp client port must be uint16") return nil, fmt.Errorf("tcp client port must be uint16")
} }
return TCPClient{Host: hostString, Port: uint16(portNum), config: config}, nil framingMethod, ok := params["framing"]
if !ok {
return nil, fmt.Errorf("tcp client requires a framing method")
}
framingMethodString, ok := framingMethod.(string)
if !ok {
return nil, fmt.Errorf("tcp framing method must be a string")
}
var framer framing.Framer
switch framingMethodString {
case "CR":
framer = framing.NewByteSeparatorFramer([]byte{'\r'})
case "LF":
framer = framing.NewByteSeparatorFramer([]byte{'\n'})
case "CRLF":
framer = framing.NewByteSeparatorFramer([]byte{'\r', '\n'})
default:
return nil, fmt.Errorf("unknown framing method: %s", framingMethodString)
}
return TCPClient{framer: framer, Host: hostString, Port: uint16(portNum), config: config}, nil
}, },
}) })
} }
@@ -79,11 +106,17 @@ func (tc TCPClient) Run(ctx context.Context) error {
if err != nil { if err != nil {
slog.Debug("connection closed") slog.Debug("connection closed")
tc.framer.Clear()
break READ break READ
} }
if byteCount > 0 { if tc.framer != nil {
slog.Info(string(buffer[0:byteCount])) if byteCount > 0 {
messages := tc.framer.Frame(buffer[0:byteCount])
for _, message := range messages {
slog.Debug("tcp-client message", "bytes", message)
}
}
} }
} }

View File

@@ -5,11 +5,14 @@ import (
"fmt" "fmt"
"log/slog" "log/slog"
"net" "net"
"github.com/jwetzell/showbridge-go/internals/framing"
) )
type TCPServer struct { type TCPServer struct {
config ModuleConfig config ModuleConfig
Port uint16 Port uint16
framingMethod string
} }
func init() { func init() {
@@ -28,7 +31,18 @@ func init() {
return nil, fmt.Errorf("tcp server port must be uint16") return nil, fmt.Errorf("tcp server port must be uint16")
} }
return TCPServer{Port: uint16(portNum), config: config}, nil framingMethod, ok := params["framing"]
if !ok {
return nil, fmt.Errorf("tcp server requires a framing method")
}
framingMethodString, ok := framingMethod.(string)
if !ok {
return nil, fmt.Errorf("tcp framing method must be a string")
}
return TCPServer{framingMethod: framingMethodString, Port: uint16(portNum), config: config}, nil
}, },
}) })
} }
@@ -44,6 +58,17 @@ func (ts TCPServer) Type() string {
func (ts TCPServer) HandleClient(ctx context.Context, client net.Conn) { func (ts TCPServer) HandleClient(ctx context.Context, client net.Conn) {
slog.Info("handling connection", "remoteAddr", client.RemoteAddr().String()) slog.Info("handling connection", "remoteAddr", client.RemoteAddr().String())
var framer framing.Framer
switch ts.framingMethod {
case "LF":
framer = framing.NewByteSeparatorFramer([]byte{'\n'})
case "CR":
framer = framing.NewByteSeparatorFramer([]byte{'\r'})
case "CRLF":
framer = framing.NewByteSeparatorFramer([]byte{'\r', '\n'})
}
buffer := make([]byte, 1024) buffer := make([]byte, 1024)
for { for {
select { select {
@@ -58,9 +83,13 @@ func (ts TCPServer) HandleClient(ctx context.Context, client net.Conn) {
} }
return return
} }
if framer != nil {
if byteCount > 0 { if byteCount > 0 {
slog.Info(string(buffer[0:byteCount])) messages := framer.Frame(buffer[0:byteCount])
for _, message := range messages {
slog.Debug("tcp-server message", "bytes", message)
}
}
} }
} }