middlewares

queue.go

@@ -1,10 +1,14 @@
 package laniakea
 
-import "fmt"
+import (
+	"fmt"
+	"sync"
+)
 
 type Queue[T any] struct {
-	queue []T
 	size  uint64
+	mu    sync.RWMutex
+	queue []T
 }
 
 func CreateQueue[T any](size uint64) *Queue[T] {
@@ -23,11 +27,13 @@ func (q *Queue[T]) Enqueue(el T) error {
 }
 
 func (q *Queue[T]) Peak() T {
+	q.mu.RLock()
+	defer q.mu.RUnlock()
 	return q.queue[0]
 }
 
 func (q *Queue[T]) IsEmpty() bool {
-	return len(q.queue) == 0
+	return q.Length() == 0
 }
 
 func (q *Queue[T]) IsFull() bool {
@@ -35,16 +41,26 @@ func (q *Queue[T]) IsFull() bool {
 }
 
 func (q *Queue[T]) Length() uint64 {
+	q.mu.RLock()
+	defer q.mu.RUnlock()
 	return uint64(len(q.queue))
 }
 
 func (q *Queue[T]) Dequeue() T {
+	q.mu.RLock()
 	el := q.queue[0]
+	q.mu.RUnlock()
+
 	if q.Length() == 1 {
+		q.mu.Lock()
 		q.queue = make([]T, 0)
+		q.mu.Unlock()
 		return el
 	}
+
+	q.mu.Lock()
 	q.queue = q.queue[1:]
+	q.mu.Unlock()
 	return el
 }