Skip to content

K
k3s
Unverified Commit 470a1818 authored by Clayton Coleman's avatar Clayton Coleman
Browse files
Options

Framer does not need to allocate a slice for each execution

parent 159797ba
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
pkg/util/framer/framer.go
View file @ 470a1818
...@@ -25,6 +25,7 @@ import ( ...@@ -25,6 +25,7 @@ import (
type lengthDelimitedFrameWriter struct { type lengthDelimitedFrameWriter struct {
w io.Writer w io.Writer
h [4]byte
} }
func NewLengthDelimitedFrameWriter(w io.Writer) io.Writer { func NewLengthDelimitedFrameWriter(w io.Writer) io.Writer {
...@@ -34,13 +35,12 @@ func NewLengthDelimitedFrameWriter(w io.Writer) io.Writer { ...@@ -34,13 +35,12 @@ func NewLengthDelimitedFrameWriter(w io.Writer) io.Writer {
// Write writes a single frame to the nested writer, prepending it with the length in // Write writes a single frame to the nested writer, prepending it with the length in
// in bytes of data (as a 4 byte, bigendian uint32). // in bytes of data (as a 4 byte, bigendian uint32).
func (w *lengthDelimitedFrameWriter) Write(data []byte) (int, error) { func (w *lengthDelimitedFrameWriter) Write(data []byte) (int, error) {
header := [4]byte{} binary.BigEndian.PutUint32(w.h[:], uint32(len(data)))
binary.BigEndian.PutUint32(header[:], uint32(len(data))) n, err := w.w.Write(w.h[:])
n, err := w.w.Write(header[:])
if err != nil { if err != nil {
return 0, err return 0, err
} }
if n != len(header) { if n != len(w.h) {
return 0, io.ErrShortWrite return 0, io.ErrShortWrite
} }
return w.w.Write(data) return w.w.Write(data)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment