1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 01:57:12 -05:00

simplify buffer allocation

This commit is contained in:
Darien Raymond 2018-12-04 11:07:33 +01:00
parent a6d81cc56d
commit c49b93b39e
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169

View File

@ -29,17 +29,19 @@ func (w *BufferToBytesWriter) WriteMultiBuffer(mb MultiBuffer) error {
return WriteAllBytes(w.Writer, mb[0].Bytes())
}
if cap(w.cache) < len(mb) {
w.cache = make([][]byte, 0, len(mb))
}
bs := w.cache
for _, b := range mb {
bs = append(bs, b.Bytes())
}
w.cache = bs
defer func() {
for idx := range w.cache {
w.cache[idx] = nil
for idx := range bs {
bs[idx] = nil
}
w.cache = w.cache[:0]
}()
nb := net.Buffers(bs)