mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-24 06:28:42 +01:00
46 lines
763 B
Go
46 lines
763 B
Go
package utils
|
|
|
|
import (
|
|
"encoding/json"
|
|
"sync"
|
|
)
|
|
|
|
type RingBuffer[T any] struct {
|
|
Size int
|
|
data []T
|
|
start int
|
|
mutex sync.RWMutex
|
|
}
|
|
|
|
func NewRingBuffer[T any](size int) *RingBuffer[T] {
|
|
return &RingBuffer[T]{
|
|
Size: size,
|
|
data: make([]T, 0, size),
|
|
}
|
|
}
|
|
|
|
func (r *RingBuffer[T]) Push(data T) {
|
|
r.mutex.Lock()
|
|
defer r.mutex.Unlock()
|
|
if len(r.data) == r.Size {
|
|
r.data[r.start] = data
|
|
r.start = (r.start + 1) % r.Size
|
|
} else {
|
|
r.data = append(r.data, data)
|
|
}
|
|
}
|
|
|
|
func (r *RingBuffer[T]) Data() []T {
|
|
r.mutex.RLock()
|
|
defer r.mutex.RUnlock()
|
|
if len(r.data) == r.Size {
|
|
return append(r.data[r.start:], r.data[:r.start]...)
|
|
} else {
|
|
return r.data
|
|
}
|
|
}
|
|
|
|
func (r *RingBuffer[T]) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(r.Data())
|
|
}
|