Update dependencies
Signed-off-by: Markus Lehtonen <markus.lehtonen@intel.com>
This commit is contained in:
		
							
								
								
									
										180
									
								
								vendor/github.com/intel/goresctrl/pkg/utils/idset.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										180
									
								
								vendor/github.com/intel/goresctrl/pkg/utils/idset.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							@@ -0,0 +1,180 @@
 | 
			
		||||
/*
 | 
			
		||||
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
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user