forked from quic-go/quic-go
34 lines
557 B
Go
34 lines
557 B
Go
package wire
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/quic-go/quic-go/internal/protocol"
|
|
)
|
|
|
|
var pool sync.Pool
|
|
|
|
func init() {
|
|
pool.New = func() any {
|
|
return &StreamFrame{
|
|
Data: make([]byte, 0, protocol.MaxPacketBufferSize),
|
|
fromPool: true,
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetStreamFrame() *StreamFrame {
|
|
f := pool.Get().(*StreamFrame)
|
|
return f
|
|
}
|
|
|
|
func putStreamFrame(f *StreamFrame) {
|
|
if !f.fromPool {
|
|
return
|
|
}
|
|
if protocol.ByteCount(cap(f.Data)) != protocol.MaxPacketBufferSize {
|
|
panic("wire.PutStreamFrame called with packet of wrong size!")
|
|
}
|
|
pool.Put(f)
|
|
}
|