cleanup handleRequest

cmd/smoketest/main.go

@@ -2,19 +2,21 @@ package main
 
 import (
 	"fmt"
+	"io"
+	"log"
 	"net"
 	"os"
 )
 
 const (
 	connPort = "3030"
-	connType = "tcp"
+	connType = "tcp4"
 )
 
 func main() {
 	l, err := net.Listen(connType, ":"+connPort)
 	if err != nil {
-		fmt.Printf("net.Listen fail: %v\n", err)
+		log.Fatalln(err)
 		os.Exit(1)
 	}
 	defer l.Close()
@@ -24,19 +26,15 @@ func main() {
 	for {
 		conn, err := l.Accept()
 		if err != nil {
-			fmt.Printf("Error accepting: %v\n", err)
+			log.Fatalln(err)
 		}
 		go handleRequest(conn)
 	}
 }
 
 func handleRequest(conn net.Conn) {
-	buf := make([]byte, 1024)
-	_, err := conn.Read(buf)
-	fmt.Printf("read in %s.\nSending it back\n", string(buf))
-	if err != nil {
-		fmt.Printf("Error reading: %v\n", err)
+	defer conn.Close()
+	if _, err := io.Copy(conn, conn); err != nil {
+		log.Fatalln(err)
 	}
-	conn.Write(buf)
-	conn.Close()
 }