gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

preallocate the message buffers of the ipv4.Message passed to ReadBatch (#3541)

Browse Source
This commit is contained in:
Marten Seemann
2022-08-30 14:11:43 +03:00
committed by GitHub
parent 0b26365dae
commit 5cd5341545
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sys_conn_oob.go
View File

@@ -122,10 +122,15 @@ func newConn(c OOBCapablePacketConn) (*oobConn, error) {
bc = ipv4.NewPacketConn(c) bc = ipv4.NewPacketConn(c)
} }
msgs := make([]ipv4.Message, batchSize)
for i := range msgs {
// preallocate the [][]byte
msgs[i].Buffers = make([][]byte, 1)
}
oobConn := &oobConn{ oobConn := &oobConn{
OOBCapablePacketConn: c, OOBCapablePacketConn: c,
batchConn: bc, batchConn: bc,
messages: make([]ipv4.Message, batchSize), messages: msgs,
readPos: batchSize, readPos: batchSize,
} }
for i := 0; i < batchSize; i++ { for i := 0; i < batchSize; i++ {
@@ -142,7 +147,7 @@ func (c *oobConn) ReadPacket() (*receivedPacket, error) {
buffer := getPacketBuffer() buffer := getPacketBuffer()
buffer.Data = buffer.Data[:protocol.MaxPacketBufferSize] buffer.Data = buffer.Data[:protocol.MaxPacketBufferSize]
c.buffers[i] = buffer c.buffers[i] = buffer
c.messages[i].Buffers = [][]byte{c.buffers[i].Data} c.messages[i].Buffers[0] = c.buffers[i].Data
} }
c.readPos = 0 c.readPos = 0