diff --git a/output_tcp_test.go b/output_tcp_test.go index c987e3b..05cd738 100644 --- a/output_tcp_test.go +++ b/output_tcp_test.go @@ -44,17 +44,25 @@ func startTCP(cb func([]byte)) net.Listener { go func() { for { conn, _ := listener.Accept() - + go func() { - scanner := bufio.NewScanner(conn) - - scanner.Split(scanBytes) - - for scanner.Scan() { - cb(scanner.Bytes()) - } + reader := bufio.NewReader(conn) + for { + cb(func() { + buf,err := reader.ReadBytes('ΒΆ') + new_buf_len := len(buf) - 2 + new_buf := make([]byte, new_buf_len) + copy(new_buf, buf[:new_buf_len]) + if err != nil { + if err != io.EOF { + log.Printf("error: %s\n", err) + } + } + return new_buf + } conn.Close() + }() } }()