optimize init byte.Buffer

Edgio · Apr 2, 2017 · 5d08e48 · 5d08e48
1 parent 994e3a1
commit 5d08e48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ipfix/marshal_test.go b/ipfix/marshal_test.go
@@ -81,7 +81,7 @@ var mockDecodedMsg = Message{
 }
 
 func TestJSONMarshal(t *testing.T) {
-	buf := bytes.NewBufferString("")
+	buf := new(bytes.Buffer)
 	msg := TestMessage{}
 
 	b, err := mockDecodedMsg.JSONMarshal(buf)
@@ -102,7 +102,7 @@ func TestJSONMarshal(t *testing.T) {
 }
 
 func TestJSONMarshalDataSets(t *testing.T) {
-	buf := bytes.NewBufferString("")
+	buf := new(bytes.Buffer)
 	msg := TestMessage{}
 
 	b, _ := mockDecodedMsg.JSONMarshal(buf)
@@ -137,7 +137,7 @@ func TestJSONMarshalDataSets(t *testing.T) {
 }
 
 func BenchmarkJSONMarshal(b *testing.B) {
-	buf := bytes.NewBufferString("")
+	buf := new(bytes.Buffer)
 
 	for i := 0; i < b.N; i++ {
 		mockDecodedMsg.JSONMarshal(buf)

diff --git a/vflow/ipfix.go b/vflow/ipfix.go
@@ -184,7 +184,7 @@ func (i *IPFIX) ipfixWorker(wQuit chan struct{}) {
 		decodedMsg *ipfix.Message
 		mirror     IPFIXUDPMsg
 		msg        = IPFIXUDPMsg{body: ipfixBuffer.Get().([]byte)}
-		buf        = bytes.NewBufferString("")
+		buf        = new(bytes.Buffer)
 		err        error
 		ok         bool
 		b          []byte