refine kcp output

pull/432/head
Darien Raymond 8 years ago
parent 14bbb58b12
commit 3643dc37e0
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169

@ -28,8 +28,7 @@ func (v *SimpleSegmentWriter) Write(seg Segment) error {
v.Lock()
defer v.Unlock()
v.buffer.AppendSupplier(seg.Bytes())
v.buffer.Reset(seg.Bytes())
_, err := v.writer.Write(v.buffer.Bytes())
v.buffer.Clear()
return err
}

@ -61,12 +61,13 @@ func (v *DataSegment) Command() Command {
return CommandData
}
func (v *DataSegment) SetData(b []byte) {
func (v *DataSegment) SetData(data []byte) {
if v.Data == nil {
v.Data = buf.NewSmall()
}
v.Data.Clear()
v.Data.Append(b)
v.Data.Reset(func(b []byte) (int, error) {
return copy(b, data), nil
})
}
func (v *DataSegment) Bytes() buf.Supplier {

Loading…
Cancel
Save