.github
alerting
client
config
core
docs
example
jsonpath
k8s
k8stest
metric
pattern
security
static
vendor
cloud.google.com
github.com
golang.org
google.golang.org
appengine
protobuf
encoding
internal
descfmt
descopts
detrand
encoding
errors
fieldsort
filedesc
filetype
flags
genid
impl
mapsort
pragma
set
ints.go
strs
version
proto
reflect
runtime
types
AUTHORS
CONTRIBUTORS
LICENSE
PATENTS
gopkg.in
k8s.io
sigs.k8s.io
modules.txt
watchdog
.dockerignore
.gitattributes
.gitignore
Dockerfile
LICENSE.md
README.md
config.yaml
go.mod
go.sum
gzip.go
main.go
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
// Copyright 2018 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package set provides simple set data structures for uint64s.
|
|
package set
|
|
|
|
import "math/bits"
|
|
|
|
// int64s represents a set of integers within the range of 0..63.
|
|
type int64s uint64
|
|
|
|
func (bs *int64s) Len() int {
|
|
return bits.OnesCount64(uint64(*bs))
|
|
}
|
|
func (bs *int64s) Has(n uint64) bool {
|
|
return uint64(*bs)&(uint64(1)<<n) > 0
|
|
}
|
|
func (bs *int64s) Set(n uint64) {
|
|
*(*uint64)(bs) |= uint64(1) << n
|
|
}
|
|
func (bs *int64s) Clear(n uint64) {
|
|
*(*uint64)(bs) &^= uint64(1) << n
|
|
}
|
|
|
|
// Ints represents a set of integers within the range of 0..math.MaxUint64.
|
|
type Ints struct {
|
|
lo int64s
|
|
hi map[uint64]struct{}
|
|
}
|
|
|
|
func (bs *Ints) Len() int {
|
|
return bs.lo.Len() + len(bs.hi)
|
|
}
|
|
func (bs *Ints) Has(n uint64) bool {
|
|
if n < 64 {
|
|
return bs.lo.Has(n)
|
|
}
|
|
_, ok := bs.hi[n]
|
|
return ok
|
|
}
|
|
func (bs *Ints) Set(n uint64) {
|
|
if n < 64 {
|
|
bs.lo.Set(n)
|
|
return
|
|
}
|
|
if bs.hi == nil {
|
|
bs.hi = make(map[uint64]struct{})
|
|
}
|
|
bs.hi[n] = struct{}{}
|
|
}
|
|
func (bs *Ints) Clear(n uint64) {
|
|
if n < 64 {
|
|
bs.lo.Clear(n)
|
|
return
|
|
}
|
|
delete(bs.hi, n)
|
|
}
|