mirror of
https://github.com/tailscale/tailscale.git
synced 2025-10-28 05:00:08 +00:00
util/ringbuffer: rename to ringlog
I need a ringbuffer in the more traditional sense, one that has a notion of item removal as well as tail loss on overrun. This implementation is really a clearable log window, and is used as such where it is used. Updates #cleanup Updates tailscale/corp#31762 Signed-off-by: James Tucker <james@tailscale.com>
This commit is contained in:
committed by
James Tucker
parent
4b9a1a0087
commit
d42f0b6a21
78
util/ringlog/ringlog.go
Normal file
78
util/ringlog/ringlog.go
Normal file
@@ -0,0 +1,78 @@
|
||||
// Copyright (c) Tailscale Inc & AUTHORS
|
||||
// SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
// Package ringlog contains a limited-size concurrency-safe generic ring log.
|
||||
package ringlog
|
||||
|
||||
import "sync"
|
||||
|
||||
// New creates a new [RingLog] containing at most max items.
|
||||
func New[T any](max int) *RingLog[T] {
|
||||
return &RingLog[T]{
|
||||
max: max,
|
||||
}
|
||||
}
|
||||
|
||||
// RingLog is a concurrency-safe fixed size log window containing entries of [T].
|
||||
type RingLog[T any] struct {
|
||||
mu sync.Mutex
|
||||
pos int
|
||||
buf []T
|
||||
max int
|
||||
}
|
||||
|
||||
// Add appends a new item to the [RingLog], possibly overwriting the oldest
|
||||
// item in the log if it is already full.
|
||||
//
|
||||
// It does nothing if rb is nil.
|
||||
func (rb *RingLog[T]) Add(t T) {
|
||||
if rb == nil {
|
||||
return
|
||||
}
|
||||
rb.mu.Lock()
|
||||
defer rb.mu.Unlock()
|
||||
if len(rb.buf) < rb.max {
|
||||
rb.buf = append(rb.buf, t)
|
||||
} else {
|
||||
rb.buf[rb.pos] = t
|
||||
rb.pos = (rb.pos + 1) % rb.max
|
||||
}
|
||||
}
|
||||
|
||||
// GetAll returns a copy of all the entries in the ring log in the order they
|
||||
// were added.
|
||||
//
|
||||
// It returns nil if rb is nil.
|
||||
func (rb *RingLog[T]) GetAll() []T {
|
||||
if rb == nil {
|
||||
return nil
|
||||
}
|
||||
rb.mu.Lock()
|
||||
defer rb.mu.Unlock()
|
||||
out := make([]T, len(rb.buf))
|
||||
for i := range len(rb.buf) {
|
||||
x := (rb.pos + i) % rb.max
|
||||
out[i] = rb.buf[x]
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// Len returns the number of elements in the ring log. Note that this value
|
||||
// could change immediately after being returned if a concurrent caller
|
||||
// modifies the log.
|
||||
func (rb *RingLog[T]) Len() int {
|
||||
if rb == nil {
|
||||
return 0
|
||||
}
|
||||
rb.mu.Lock()
|
||||
defer rb.mu.Unlock()
|
||||
return len(rb.buf)
|
||||
}
|
||||
|
||||
// Clear will empty the ring log.
|
||||
func (rb *RingLog[T]) Clear() {
|
||||
rb.mu.Lock()
|
||||
defer rb.mu.Unlock()
|
||||
rb.pos = 0
|
||||
rb.buf = nil
|
||||
}
|
||||
Reference in New Issue
Block a user