add universal tcp message reader and writer

1 files changed, 4 insertions, 7 deletions

diff --git a/internal/client/tcpclient.go b/internal/client/tcpclient.go
index 1a76ad6..3181fd4 100644
--- a/internal/client/tcpclient.go
+++ b/internal/client/tcpclient.go
@@ -3,10 +3,10 @@ package client
 import (
 	"context"
 	"fmt"
-	"io"
 	"log"
 	"net"
 
+	"codeberg.org/snonux/gorum/internal/tcp"
 	"codeberg.org/snonux/gorum/internal/vote"
 )
 
@@ -29,18 +29,15 @@ func tcpClientRun(ctx context.Context, node string, ch <-chan vote.Vote) error {
 		}
 
 		log.Println("tcpclient: sending", message, "to node", node)
-
-		bytes := []byte(fmt.Sprintf("%s\n", message))
-		_, err = conn.Write(bytes)
-		if err != nil {
+		if err := tcp.WriteStr(conn, message); err != nil {
 			return err
 		}
 
-		response, err := io.ReadAll(conn)
+		response, err := tcp.ReadStr(conn)
 		if err != nil {
 			return err
 		}
 
-		log.Println("tcpclient: received", string(response), "from node", node)
+		log.Println("tcpclient: received", response, "from node", node)
 	}
 }