mirror of
https://github.com/ipfs/kubo.git
synced 2026-03-05 16:28:06 +08:00
msgio proper write
This commit is contained in:
parent
9d05b35231
commit
aed70170dc
2
Godeps/Godeps.json
generated
2
Godeps/Godeps.json
generated
@ -106,7 +106,7 @@
|
||||
},
|
||||
{
|
||||
"ImportPath": "github.com/jbenet/go-msgio",
|
||||
"Rev": "634f25d82673bbdb0b87b8d7e1b7a3df59ede671"
|
||||
"Rev": "7bdc5b738564871e1c0d5ca9449900d0d6773713"
|
||||
},
|
||||
{
|
||||
"ImportPath": "github.com/jbenet/go-multiaddr",
|
||||
|
||||
10
Godeps/_workspace/src/github.com/jbenet/go-msgio/msgio.go
generated
vendored
10
Godeps/_workspace/src/github.com/jbenet/go-msgio/msgio.go
generated
vendored
@ -17,7 +17,7 @@ const lengthSize = 4
|
||||
type Writer interface {
|
||||
|
||||
// Write writes passed in buffer as a single message.
|
||||
Write([]byte) error
|
||||
Write([]byte) (int, error)
|
||||
|
||||
// WriteMsg writes the msg in the passed in buffer.
|
||||
WriteMsg([]byte) error
|
||||
@ -79,8 +79,12 @@ func NewWriter(w io.Writer) WriteCloser {
|
||||
return &writer{W: w, lock: new(sync.Mutex)}
|
||||
}
|
||||
|
||||
func (s *writer) Write(msg []byte) (err error) {
|
||||
return s.WriteMsg(msg)
|
||||
func (s *writer) Write(msg []byte) (int, error) {
|
||||
err := s.WriteMsg(msg)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return len(msg), nil
|
||||
}
|
||||
|
||||
func (s *writer) WriteMsg(msg []byte) (err error) {
|
||||
|
||||
5
Godeps/_workspace/src/github.com/jbenet/go-msgio/msgio_test.go
generated
vendored
5
Godeps/_workspace/src/github.com/jbenet/go-msgio/msgio_test.go
generated
vendored
@ -20,10 +20,13 @@ func TestReadWrite(t *testing.T) {
|
||||
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||
for i := range msgs {
|
||||
msgs[i] = randbuf.RandBuf(r, r.Intn(1000))
|
||||
err := writer.Write(msgs[i])
|
||||
n, err := writer.Write(msgs[i])
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if n != len(msgs[i]) {
|
||||
t.Fatal("wrong length:", n, len(msgs[i]))
|
||||
}
|
||||
}
|
||||
|
||||
if err := writer.Close(); err != nil {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user