mirror of
https://blitiri.com.ar/repos/chasquid
synced 2025-12-18 14:47:03 +00:00
Implement a string set data type
We will need a few string sets in different places, create it as a separate package for convenience. For now, it only implements strings, but it may be extended to other data types in the future.
This commit is contained in:
32
internal/set/set.go
Normal file
32
internal/set/set.go
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
// Package set implement sets for various types. Well, only string for now :)
|
||||||
|
package set
|
||||||
|
|
||||||
|
type String struct {
|
||||||
|
m map[string]struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewString(values ...string) *String {
|
||||||
|
s := &String{}
|
||||||
|
s.Add(values...)
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *String) Add(values ...string) {
|
||||||
|
if s.m == nil {
|
||||||
|
s.m = map[string]struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v := range values {
|
||||||
|
s.m[v] = struct{}{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *String) Has(value string) bool {
|
||||||
|
// We explicitly allow s to be nil *in this function* to simplify callers'
|
||||||
|
// code. Note that Add will not tolerate it, and will panic.
|
||||||
|
if s == nil || s.m == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := s.m[value]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
40
internal/set/set_test.go
Normal file
40
internal/set/set_test.go
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package set
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestString(t *testing.T) {
|
||||||
|
s1 := &String{}
|
||||||
|
|
||||||
|
// Test that Has works on a new set.
|
||||||
|
if s1.Has("x") {
|
||||||
|
t.Error("'x' is in the empty set")
|
||||||
|
}
|
||||||
|
|
||||||
|
s1.Add("a")
|
||||||
|
s1.Add("b", "ccc")
|
||||||
|
|
||||||
|
expectStrings(s1, []string{"a", "b", "ccc"}, []string{"notin"}, t)
|
||||||
|
|
||||||
|
s2 := NewString("a", "b", "c")
|
||||||
|
expectStrings(s2, []string{"a", "b", "c"}, []string{"notin"}, t)
|
||||||
|
|
||||||
|
// Test that Has works (and not panics) on a nil set.
|
||||||
|
var s3 *String
|
||||||
|
if s3.Has("x") {
|
||||||
|
t.Error("'x' is in the nil set")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func expectStrings(s *String, in []string, notIn []string, t *testing.T) {
|
||||||
|
for _, str := range in {
|
||||||
|
if !s.Has(str) {
|
||||||
|
t.Errorf("String %q not in set, it should be", str)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, str := range notIn {
|
||||||
|
if s.Has(str) {
|
||||||
|
t.Errorf("String %q is in the set, should not be", str)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user