Compare commits
3 Commits
0785baf4b7
...
c228caa309
Author | SHA1 | Date | |
---|---|---|---|
c228caa309 | |||
f8c0f0eddb | |||
2f56894b6a |
@ -5,7 +5,6 @@ when:
|
|||||||
|
|
||||||
steps:
|
steps:
|
||||||
test:
|
test:
|
||||||
group: test
|
|
||||||
image: golang:1.22-alpine
|
image: golang:1.22-alpine
|
||||||
commands:
|
commands:
|
||||||
- go test -v ./...
|
- go test -v ./...
|
||||||
|
10
functions.go
10
functions.go
@ -1,6 +1,6 @@
|
|||||||
package sets
|
package sets
|
||||||
|
|
||||||
// Union returns a union of the given sets (left ∪ right)
|
// Union returns a union of the given sets (left ∪ right).
|
||||||
func Union[T comparable](left, right *Set[T]) *Set[T] {
|
func Union[T comparable](left, right *Set[T]) *Set[T] {
|
||||||
left.mux.RLock()
|
left.mux.RLock()
|
||||||
right.mux.RLock()
|
right.mux.RLock()
|
||||||
@ -23,7 +23,7 @@ func Union[T comparable](left, right *Set[T]) *Set[T] {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Intersection returns an intersection of the given sets (left ∩ right)
|
// Intersection returns an intersection of the given sets (left ∩ right).
|
||||||
func Intersection[T comparable](left, right *Set[T]) *Set[T] {
|
func Intersection[T comparable](left, right *Set[T]) *Set[T] {
|
||||||
left.mux.RLock()
|
left.mux.RLock()
|
||||||
right.mux.RLock()
|
right.mux.RLock()
|
||||||
@ -43,7 +43,7 @@ func Intersection[T comparable](left, right *Set[T]) *Set[T] {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Diff returns the relative complement of sets (left ∖ right)
|
// Diff returns the relative complement of sets (left ∖ right).
|
||||||
func Diff[T comparable](left, right *Set[T]) *Set[T] {
|
func Diff[T comparable](left, right *Set[T]) *Set[T] {
|
||||||
left.mux.RLock()
|
left.mux.RLock()
|
||||||
right.mux.RLock()
|
right.mux.RLock()
|
||||||
@ -63,7 +63,7 @@ func Diff[T comparable](left, right *Set[T]) *Set[T] {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// SymmetricDiff returns the symmetric difference between sets (left ⊖ right)
|
// SymmetricDiff returns the symmetric difference between sets (left ⊖ right).
|
||||||
func SymmetricDiff[T comparable](left, right *Set[T]) *Set[T] {
|
func SymmetricDiff[T comparable](left, right *Set[T]) *Set[T] {
|
||||||
left.mux.RLock()
|
left.mux.RLock()
|
||||||
right.mux.RLock()
|
right.mux.RLock()
|
||||||
@ -88,7 +88,7 @@ func SymmetricDiff[T comparable](left, right *Set[T]) *Set[T] {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Equal checks whether the sets are equal (left = right)
|
// Equal checks whether the sets are equal (left = right).
|
||||||
func Equal[T comparable](left, right *Set[T]) bool {
|
func Equal[T comparable](left, right *Set[T]) bool {
|
||||||
left.mux.RLock()
|
left.mux.RLock()
|
||||||
right.mux.RLock()
|
right.mux.RLock()
|
||||||
|
19
set.go
19
set.go
@ -4,13 +4,13 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Set represents a set of values
|
// Set represents a set of values.
|
||||||
type Set[T comparable] struct {
|
type Set[T comparable] struct {
|
||||||
store map[T]struct{}
|
store map[T]struct{}
|
||||||
mux sync.RWMutex
|
mux sync.RWMutex
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new set
|
// New creates a new set.
|
||||||
func New[T comparable](data ...T) *Set[T] {
|
func New[T comparable](data ...T) *Set[T] {
|
||||||
set := &Set[T]{
|
set := &Set[T]{
|
||||||
store: map[T]struct{}{},
|
store: map[T]struct{}{},
|
||||||
@ -22,7 +22,12 @@ func New[T comparable](data ...T) *Set[T] {
|
|||||||
return set
|
return set
|
||||||
}
|
}
|
||||||
|
|
||||||
// Size returns number of elements in set
|
// NewFromSlice creates a new set from a slice.
|
||||||
|
func NewFromSlice[T comparable](data []T) *Set[T] {
|
||||||
|
return New[T](data...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Size returns number of elements in set.
|
||||||
func (s *Set[T]) Size() int {
|
func (s *Set[T]) Size() int {
|
||||||
s.mux.RLock()
|
s.mux.RLock()
|
||||||
defer s.mux.RUnlock()
|
defer s.mux.RUnlock()
|
||||||
@ -30,7 +35,7 @@ func (s *Set[T]) Size() int {
|
|||||||
return len(s.store)
|
return len(s.store)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Contains checks whether the value is contained in the set
|
// Contains checks whether the value is contained in the set.
|
||||||
func (s *Set[T]) Contains(val T) bool {
|
func (s *Set[T]) Contains(val T) bool {
|
||||||
s.mux.RLock()
|
s.mux.RLock()
|
||||||
defer s.mux.RUnlock()
|
defer s.mux.RUnlock()
|
||||||
@ -39,7 +44,7 @@ func (s *Set[T]) Contains(val T) bool {
|
|||||||
return ok
|
return ok
|
||||||
}
|
}
|
||||||
|
|
||||||
// Insert inserts a value into the set if the value was not already present
|
// Insert inserts a value into the set if the value was not already present.
|
||||||
func (s *Set[T]) Insert(val T) bool {
|
func (s *Set[T]) Insert(val T) bool {
|
||||||
s.mux.Lock()
|
s.mux.Lock()
|
||||||
defer s.mux.Unlock()
|
defer s.mux.Unlock()
|
||||||
@ -51,7 +56,7 @@ func (s *Set[T]) Insert(val T) bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete removes a value from the set if the value was already present
|
// Delete removes a value from the set if the value was already present.
|
||||||
func (s *Set[T]) Delete(val T) bool {
|
func (s *Set[T]) Delete(val T) bool {
|
||||||
s.mux.Lock()
|
s.mux.Lock()
|
||||||
defer s.mux.Unlock()
|
defer s.mux.Unlock()
|
||||||
@ -63,7 +68,7 @@ func (s *Set[T]) Delete(val T) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// Slice returns a slice which contains the elements from the set
|
// Slice returns a slice which contains the elements from the set.
|
||||||
func (s *Set[T]) Slice() []T {
|
func (s *Set[T]) Slice() []T {
|
||||||
s.mux.RLock()
|
s.mux.RLock()
|
||||||
defer s.mux.RUnlock()
|
defer s.mux.RUnlock()
|
||||||
|
Loading…
Reference in New Issue
Block a user