feat: implement the version with generics

go.mod

@@ -1,6 +1,6 @@
 module github.com/kevwan/stream
 
-go 1.17
+go 1.18
 
 require github.com/stretchr/testify v1.7.0
 

ring.go

@@ -3,25 +3,25 @@ package stream
 import "sync"
 
 // A Ring can be used as fixed size ring.
-type Ring struct {
-	elements []interface{}
+type Ring[T any] struct {
+	elements []T
 	index    int
 	lock     sync.Mutex
 }
 
 // NewRing returns a Ring object with the given size n.
-func NewRing(n int) *Ring {
+func NewRing[T any](n int) *Ring[T] {
 	if n < 1 {
 		panic("n should be greater than 0")
 	}
 
-	return &Ring{
-		elements: make([]interface{}, n),
+	return &Ring[T]{
+		elements: make([]T, n),
 	}
 }
 
 // Add adds v into r.
-func (r *Ring) Add(v interface{}) {
+func (r *Ring[T]) Add(v T) {
 	r.lock.Lock()
 	defer r.lock.Unlock()
 
@@ -30,7 +30,7 @@ func (r *Ring) Add(v interface{}) {
 }
 
 // Take takes all items from r.
-func (r *Ring) Take() []interface{} {
+func (r *Ring[T]) Take() []T {
 	r.lock.Lock()
 	defer r.lock.Unlock()
 
@@ -43,7 +43,7 @@ func (r *Ring) Take() []interface{} {
 		size = r.index
 	}
 
-	elements := make([]interface{}, size)
+	elements := make([]T, size)
 	for i := 0; i < size; i++ {
 		elements[i] = r.elements[(start+i)%len(r.elements)]
 	}

ring_test.go

@@ -9,12 +9,12 @@ import (
 
 func TestNewRing(t *testing.T) {
 	assert.Panics(t, func() {
-		NewRing(0)
+		NewRing[int](0)
 	})
 }
 
 func TestRingLess(t *testing.T) {
-	ring := NewRing(5)
+	ring := NewRing[int](5)
 	for i := 0; i < 3; i++ {
 		ring.Add(i)
 	}
@@ -23,7 +23,7 @@ func TestRingLess(t *testing.T) {
 }
 
 func TestRingMore(t *testing.T) {
-	ring := NewRing(5)
+	ring := NewRing[int](5)
 	for i := 0; i < 11; i++ {
 		ring.Add(i)
 	}
@@ -32,7 +32,7 @@ func TestRingMore(t *testing.T) {
 }
 
 func TestRingAdd(t *testing.T) {
-	ring := NewRing(5051)
+	ring := NewRing[int](5051)
 	wg := sync.WaitGroup{}
 	for i := 1; i <= 100; i++ {
 		wg.Add(1)
@@ -48,7 +48,7 @@ func TestRingAdd(t *testing.T) {
 }
 
 func BenchmarkRingAdd(b *testing.B) {
-	ring := NewRing(500)
+	ring := NewRing[int](500)
 	b.RunParallel(func(pb *testing.PB) {
 		for pb.Next() {
 			for i := 0; i < b.N; i++ {