buffer msgio

License: MIT
Signed-off-by: Jeromy <jeromyj@gmail.com>

Godeps/_workspace/src/github.com/jbenet/go-msgio/msgio.go

 package msgio
 
 import (
+	"bufio"
 	"errors"
 	"io"
 	"sync"
@@ -75,7 +76,8 @@ type ReadWriteCloser interface {
 
 // writer is the underlying type that implements the Writer interface.
 type writer struct {
-	W io.Writer
+	W   io.Writer
+	buf *bufio.Writer
 
 	lock sync.Locker
 }
@@ -83,7 +85,7 @@ type writer struct {
 // NewWriter wraps an io.Writer with a msgio framed writer. The msgio.Writer
 // will write the length prefix of every message written.
 func NewWriter(w io.Writer) WriteCloser {
-	return &writer{W: w, lock: new(sync.Mutex)}
+	return &writer{W: w, buf: bufio.NewWriter(w), lock: new(sync.Mutex)}
 }
 
 func (s *writer) Write(msg []byte) (int, error) {
@@ -100,8 +102,13 @@ func (s *writer) WriteMsg(msg []byte) (err error) {
 	if err := WriteLen(s.W, len(msg)); err != nil {
 		return err
 	}
-	_, err = s.W.Write(msg)
-	return err
+
+	_, err = s.buf.Write(msg)
+	if err != nil {
+		return err
+	}
+
+	return s.buf.Flush()
 }
 
 func (s *writer) Close() error {