From 238fa5a1e605805e2016404b3fe1ff4305c50d9f Mon Sep 17 00:00:00 2001 From: matst80 Date: Thu, 21 Nov 2024 19:52:33 +0100 Subject: [PATCH] bufferd write --- tcp-connection.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tcp-connection.go b/tcp-connection.go index dc7d962..797313d 100644 --- a/tcp-connection.go +++ b/tcp-connection.go @@ -65,11 +65,13 @@ func NewConnection(address string) *Connection { } func SendFrame(conn net.Conn, data *FrameWithPayload) error { - err := binary.Write(conn, binary.LittleEndian, data.Frame) + w := bufio.NewWriter(conn) + err := binary.Write(w, binary.LittleEndian, data.Frame) if err != nil { return err } - _, err = conn.Write(data.Payload) + _, err = w.Write(data.Payload) + w.Flush() return err } @@ -102,6 +104,7 @@ func (c *Connection) Call(msg FrameType, data []byte) (*FrameWithPayload, error) } defer conn.Close() + defer close(ch) select { case ret := <-ch: