65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
|
//go:build 386
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
* Copyright (C) THL A29 Limited, a Tencent company. All rights reserved.
|
|||
|
|
*
|
|||
|
|
* SPDX-License-Identifier: Apache-2.0
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
package lockfree
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"sync"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type e[T any] struct {
|
|||
|
|
c uint64
|
|||
|
|
val T
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ringBuffer 具体对象的存放区域，通过数组（定长切片）实现环状数据结构
|
|||
|
|
// 其中e为具体对象，非指针，这样可以一次性进行内存申请
|
|||
|
|
type ringBuffer[T any] struct {
|
|||
|
|
sync.RWMutex
|
|||
|
|
// 增加默认的对象以便于return，处理data race问题
|
|||
|
|
tDefault T
|
|||
|
|
buf []e[T]
|
|||
|
|
capMask uint64
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func newRingBuffer[T any](cap int) *ringBuffer[T] {
|
|||
|
|
x := ringBuffer[T]{
|
|||
|
|
capMask: uint64(cap) - 1,
|
|||
|
|
buf: make([]e[T], cap),
|
|||
|
|
}
|
|||
|
|
return &x
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (r *ringBuffer[T]) write(c uint64, v T) {
|
|||
|
|
x := &r.buf[c&r.capMask]
|
|||
|
|
r.Lock()
|
|||
|
|
defer r.Unlock()
|
|||
|
|
x.val = v
|
|||
|
|
x.c = c + 1
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (r *ringBuffer[T]) element(c uint64) e[T] {
|
|||
|
|
return r.buf[c&r.capMask]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (r *ringBuffer[T]) contains(c uint64) (T, *uint64, bool) {
|
|||
|
|
x := &r.buf[c&r.capMask]
|
|||
|
|
r.RLock()
|
|||
|
|
defer r.RUnlock()
|
|||
|
|
if x.c == c+1 {
|
|||
|
|
v := x.val
|
|||
|
|
return v, &x.c, true
|
|||
|
|
}
|
|||
|
|
return r.tDefault, &x.c, false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (r *ringBuffer[T]) cap() uint64 {
|
|||
|
|
return r.capMask + 1
|
|||
|
|
}
|