mirror of
https://github.com/tailscale/tailscale.git
synced 2025-01-10 18:13:41 +00:00
37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
|
// Copyright (c) Tailscale Inc & AUTHORS
|
||
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
|
||
|
|
package syncs
|
||
|
|
|
||
|
|
// TODO(raggi): this implementation is still imperfect as it will still result
|
||
|
|
// in cross CPU sharing periodically, we instead really want a per-CPU shard
|
||
|
|
// key, but the limitations of calling platform code make reaching for even the
|
||
|
|
// getcpu vdso very painful. See https://github.com/golang/go/issues/18802, and
|
||
|
|
// hopefully one day we can replace with a primitive that falls out of that
|
||
|
|
// work.
|
||
|
|
|
||
|
|
// ShardValue contains a value sharded over a set of shards.
|
||
|
|
// In order to be useful, T should be aligned to cache lines.
|
||
|
|
// Users must organize that usage in One and All is concurrency safe.
|
||
|
|
// The zero value is not safe for use; use [NewShardValue].
|
||
|
|
type ShardValue[T any] struct {
|
||
|
|
shards []T
|
||
|
|
|
||
|
|
//lint:ignore U1000 unused under tailscale_go builds.
|
||
|
|
pool shardValuePool
|
||
|
|
}
|
||
|
|
|
||
|
|
// Len returns the number of shards.
|
||
|
|
func (sp *ShardValue[T]) Len() int {
|
||
|
|
return len(sp.shards)
|
||
|
|
}
|
||
|
|
|
||
|
|
// All yields a pointer to the value in each shard.
|
||
|
|
func (sp *ShardValue[T]) All(yield func(*T) bool) {
|
||
|
|
for i := range sp.shards {
|
||
|
|
if !yield(&sp.shards[i]) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|