181 lines
3.5 KiB
Go
181 lines
3.5 KiB
Go
/*
|
|
Copyright 2019-2021 Intel Corporation
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package utils
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
// Unknown represents an unknown id.
|
|
Unknown ID = -1
|
|
)
|
|
|
|
// ID is nn integer id, used to identify packages, CPUs, nodes, etc.
|
|
type ID = int
|
|
|
|
// IDSet is an unordered set of integer ids.
|
|
type IDSet map[ID]struct{}
|
|
|
|
// NewIDSet creates a new unordered set of (integer) ids.
|
|
func NewIDSet(ids ...ID) IDSet {
|
|
s := make(map[ID]struct{})
|
|
|
|
for _, id := range ids {
|
|
s[id] = struct{}{}
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
// NewIDSetFromIntSlice creates a new unordered set from an integer slice.
|
|
func NewIDSetFromIntSlice(ids ...int) IDSet {
|
|
s := make(map[ID]struct{})
|
|
|
|
for _, id := range ids {
|
|
s[ID(id)] = struct{}{}
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
// Clone returns a copy of this IdSet.
|
|
func (s IDSet) Clone() IDSet {
|
|
return NewIDSet(s.Members()...)
|
|
}
|
|
|
|
// Add adds the given ids into the set.
|
|
func (s IDSet) Add(ids ...ID) {
|
|
for _, id := range ids {
|
|
s[id] = struct{}{}
|
|
}
|
|
}
|
|
|
|
// Del deletes the given ids from the set.
|
|
func (s IDSet) Del(ids ...ID) {
|
|
if s != nil {
|
|
for _, id := range ids {
|
|
delete(s, id)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Size returns the number of ids in the set.
|
|
func (s IDSet) Size() int {
|
|
return len(s)
|
|
}
|
|
|
|
// Has tests if all the ids are present in the set.
|
|
func (s IDSet) Has(ids ...ID) bool {
|
|
if s == nil {
|
|
return false
|
|
}
|
|
|
|
for _, id := range ids {
|
|
_, ok := s[id]
|
|
if !ok {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Members returns all ids in the set as a randomly ordered slice.
|
|
func (s IDSet) Members() []ID {
|
|
if s == nil {
|
|
return []ID{}
|
|
}
|
|
ids := make([]ID, len(s))
|
|
idx := 0
|
|
for id := range s {
|
|
ids[idx] = id
|
|
idx++
|
|
}
|
|
return ids
|
|
}
|
|
|
|
// SortedMembers returns all ids in the set as a sorted slice.
|
|
func (s IDSet) SortedMembers() []ID {
|
|
ids := s.Members()
|
|
sort.Slice(ids, func(i, j int) bool {
|
|
return ids[i] < ids[j]
|
|
})
|
|
return ids
|
|
}
|
|
|
|
// String returns the set as a string.
|
|
func (s IDSet) String() string {
|
|
return s.StringWithSeparator(",")
|
|
}
|
|
|
|
// StringWithSeparator returns the set as a string, separated with the given separator.
|
|
func (s IDSet) StringWithSeparator(args ...string) string {
|
|
if s == nil || len(s) == 0 {
|
|
return ""
|
|
}
|
|
|
|
var sep string
|
|
|
|
if len(args) == 1 {
|
|
sep = args[0]
|
|
} else {
|
|
sep = ","
|
|
}
|
|
|
|
str := ""
|
|
t := ""
|
|
for _, id := range s.SortedMembers() {
|
|
str = str + t + strconv.Itoa(int(id))
|
|
t = sep
|
|
}
|
|
|
|
return str
|
|
}
|
|
|
|
// MarshalJSON is the JSON marshaller for IDSet.
|
|
func (s IDSet) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(s.String())
|
|
}
|
|
|
|
// UnmarshalJSON is the JSON unmarshaller for IDSet.
|
|
func (s *IDSet) UnmarshalJSON(data []byte) error {
|
|
str := ""
|
|
if err := json.Unmarshal(data, &str); err != nil {
|
|
return fmt.Errorf("invalid IDSet entry '%s': %v", string(data), err)
|
|
}
|
|
|
|
*s = NewIDSet()
|
|
if str == "" {
|
|
return nil
|
|
}
|
|
|
|
for _, idstr := range strings.Split(str, ",") {
|
|
id, err := strconv.ParseUint(idstr, 10, 0)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid IDSet entry '%s': %v", idstr, err)
|
|
}
|
|
s.Add(ID(id))
|
|
}
|
|
|
|
return nil
|
|
}
|