aiforge/services/socketwrap/syncqueue.go

package socketwrap

import (
	"container/list"
	"sync"
)

type SyncQueue struct {
	Queue   *list.List
	Mutex   *sync.RWMutex
	MaxSize int
}

func (q *SyncQueue) Push(value interface{}) {
	q.Mutex.Lock()
	{
		if q.Queue.Len() < q.MaxSize {
			q.Queue.PushBack(value)
		} else {
			q.Queue.PushBack(value)
			q.Queue.Remove(q.Queue.Front())
		}
	}

	q.Mutex.Unlock()
}

func NewSyncQueue(maxSize int) *SyncQueue {
	return &SyncQueue{
		list.New(),
		&sync.RWMutex{},
		maxSize,
	}
}