From 827210dd155890fcc051d07772c06f41f629eb0f Mon Sep 17 00:00:00 2001
From: Chris Xue <chrisxue815@gmail.com>
Date: Thu, 29 Aug 2019 16:36:26 +0100
Subject: [PATCH] Reduce RingBuffer heap allocation

---
 queue/ring.go      | 8 ++++----
 queue/ring_test.go | 6 ++++++
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/queue/ring.go b/queue/ring.go
index dfd0b53..ed1de3c 100644
--- a/queue/ring.go
+++ b/queue/ring.go
@@ -41,7 +41,7 @@ type node struct {
 	data     interface{}
 }
 
-type nodes []*node
+type nodes []node
 
 // RingBuffer is a MPMC buffer that achieves threadsafety with CAS operations
 // only.  A put on full or get on empty call will block until an item
@@ -64,7 +64,7 @@ func (rb *RingBuffer) init(size uint64) {
 	size = roundUp(size)
 	rb.nodes = make(nodes, size)
 	for i := uint64(0); i < size; i++ {
-		rb.nodes[i] = &node{position: i}
+		rb.nodes[i] = node{position: i}
 	}
 	rb.mask = size - 1 // so we don't have to do this with every put/get operation
 }
@@ -93,7 +93,7 @@ L:
 			return false, ErrDisposed
 		}
 
-		n = rb.nodes[pos&rb.mask]
+		n = &rb.nodes[pos&rb.mask]
 		seq := atomic.LoadUint64(&n.position)
 		switch dif := seq - pos; {
 		case dif == 0:
@@ -146,7 +146,7 @@ L:
 			return nil, ErrDisposed
 		}
 
-		n = rb.nodes[pos&rb.mask]
+		n = &rb.nodes[pos&rb.mask]
 		seq := atomic.LoadUint64(&n.position)
 		switch dif := seq - (pos + 1); {
 		case dif == 0:
diff --git a/queue/ring_test.go b/queue/ring_test.go
index 6948dde..21b020c 100644
--- a/queue/ring_test.go
+++ b/queue/ring_test.go
@@ -396,3 +396,9 @@ func BenchmarkRBGet(b *testing.B) {
 		rb.Get()
 	}
 }
+
+func BenchmarkRBAllocation(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		NewRingBuffer(1024)
+	}
+}