Merge pull request #29 from qmuntal/reduce-alloc

Avoid using direct binary.Write when encoding a document
qmuntal · Oct 31, 2020 · 8cd922e · 8cd922e
2 parents 3e49166 + 5b79d3c
commit 8cd922e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/encode.go b/encode.go
@@ -131,11 +131,12 @@ func (e *Encoder) encodeBinary(doc *Document) error {
 	if err != nil {
 		return err
 	}
-	binary.Write(e.w, binary.LittleEndian, jsonText)
-	binary.Write(e.w, binary.LittleEndian, headerPadding)
+	e.w.Write(jsonText)
+	e.w.Write(headerPadding)
 	binary.Write(e.w, binary.LittleEndian, &binHeader)
 	if binBuffer != nil {
-		binary.Write(e.w, binary.LittleEndian, binBuffer.Data)
+		e.w.Write(binBuffer.Data)
 	}
-	return binary.Write(e.w, binary.LittleEndian, binPadding)
+	_, err = e.w.Write(binPadding)
+	return err
 }