fix bufio cache bug

proxy.go

@@ -37,7 +37,7 @@ func ygoEndpoint(w http.ResponseWriter, r *http.Request) {
 
 	tcp, err := net.Dial("tcp", "127.0.0.1"+PROXY_PORT)
 	if err != nil {
-		log.Fatal("connect tcp server error: ", err)
+		log.Fatal("Connect tcp server error: ", err)
 	}
 
 	log.Println("Tcp connected")
@@ -51,12 +51,10 @@ func ygoEndpoint(w http.ResponseWriter, r *http.Request) {
 }
 
 func wsProxy(ws *websocket.Conn, tcp *net.Conn, wg *sync.WaitGroup) {
-	writer := bufio.NewWriter(*tcp)
-
 	for {
 		messageType, buffer, err := ws.ReadMessage()
 		if err != nil {
-			log.Println("websocket read message error: ", err)
+			log.Println("Websocket read message error: ", err)
 			break
 		}
 
@@ -71,9 +69,9 @@ func wsProxy(ws *websocket.Conn, tcp *net.Conn, wg *sync.WaitGroup) {
 			break
 		}
 
-		_, err = writer.Write(buffer)
+		_, err = (*tcp).Write(buffer)
 		if err != nil {
-			log.Fatal("websocket send message error: ", err)
+			log.Fatal("Tcp send message error: ", err)
 			break
 		}
 	}
@@ -92,7 +90,7 @@ func tcpProxy(tcp *net.Conn, ws *websocket.Conn, wg *sync.WaitGroup) {
 				continue
 			}
 
-			log.Println("tcp read message error: ", err)
+			log.Println("Tcp read message error: ", err)
 			break
 		}
 
@@ -104,7 +102,7 @@ func tcpProxy(tcp *net.Conn, ws *websocket.Conn, wg *sync.WaitGroup) {
 
 		err = ws.WriteMessage(websocket.BinaryMessage, buffer)
 		if err != nil {
-			log.Fatal("tcp send message error: ", err)
+			log.Fatal("Websocket send message error: ", err)
 			break
 		}
 	}