lzhr/v2fly
1
0
Fork 0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-12-22 10:08:15 -05:00
Code Issues Releases Wiki Activity

simplify buffer allocation

Browse Source
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
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
common/buf/writer.go
View File

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