tcp jonny

tcp-connection.go

@@ -1,7 +1,6 @@
 package main
 
 import (
-	"bufio"
 	"encoding/binary"
 	"fmt"
 	"log"
@@ -65,13 +64,13 @@ func NewConnection(address string) *Connection {
 }
 
 func SendFrame(conn net.Conn, data *FrameWithPayload) error {
-	w := bufio.NewWriter(conn)
-	err := binary.Write(w, binary.LittleEndian, data.Frame)
+
+	err := binary.Write(conn, binary.LittleEndian, data.Frame)
 	if err != nil {
 		return err
 	}
-	_, err = w.Write(data.Payload)
-	w.Flush()
+	_, err = conn.Write(data.Payload)
+
 	return err
 }
 
@@ -117,15 +116,14 @@ func (c *Connection) Call(msg FrameType, data []byte) (*FrameWithPayload, error)
 func WaitForFrame(conn net.Conn, resultChan chan<- FrameWithPayload) error {
 	var err error
 	var frame Frame
-	r := bufio.NewReader(conn)
 
-	err = binary.Read(r, binary.LittleEndian, &frame)
+	err = binary.Read(conn, binary.LittleEndian, &frame)
 	if err != nil {
 		return err
 	}
 	if frame.IsValid() {
 		payload := make([]byte, frame.Length)
-		_, err = r.Read(payload)
+		_, err = conn.Read(payload)
 		if err != nil {
 			return err
 		}