mirror of
https://github.com/slackhq/nebula.git
synced 2026-05-16 04:47:38 +02:00
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package batch
|
||
|
||
import (
|
||
"io"
|
||
|
||
"github.com/slackhq/nebula/udp"
|
||
"github.com/slackhq/nebula/util"
|
||
)
|
||
|
||
// Passthrough is a RxBatcher that doesn't batch anything, it just accumulates and then sends packets.
|
||
type Passthrough struct {
|
||
out io.Writer
|
||
slots [][]byte
|
||
// arena is injected; see TCPCoalescer.arena for the contract.
|
||
arena *util.Arena
|
||
cursor int
|
||
}
|
||
|
||
const passthroughBaseNumSlots = 128
|
||
|
||
// DefaultPassthroughArenaCap is the recommended arena capacity for a
|
||
// standalone Passthrough batcher: 128 slots × udp.MTU ≈ 1.1 MiB.
|
||
const DefaultPassthroughArenaCap = passthroughBaseNumSlots * udp.MTU
|
||
|
||
func NewPassthrough(w io.Writer, slots int, arena *util.Arena) *Passthrough {
|
||
return &Passthrough{
|
||
out: w,
|
||
slots: make([][]byte, 0, slots),
|
||
arena: arena,
|
||
}
|
||
}
|
||
|
||
func (p *Passthrough) Reserve(sz int) []byte {
|
||
return p.arena.Reserve(sz)
|
||
}
|
||
|
||
func (p *Passthrough) Commit(pkt []byte) error {
|
||
p.slots = append(p.slots, pkt)
|
||
return nil
|
||
}
|
||
|
||
func (p *Passthrough) Flush() error {
|
||
var firstErr error
|
||
for _, s := range p.slots {
|
||
_, err := p.out.Write(s)
|
||
if err != nil && firstErr == nil {
|
||
firstErr = err
|
||
}
|
||
}
|
||
clear(p.slots)
|
||
p.slots = p.slots[:0]
|
||
p.arena.Reset()
|
||
return firstErr
|
||
}
|