1
0
mirror of https://blitiri.com.ar/repos/chasquid synced 2025-12-19 14:57:04 +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:
Alberto Bertogli
2016-05-06 22:49:16 +01:00
parent 17c536f5e3
commit 724d915e95
2 changed files with 72 additions and 0 deletions

32
internal/set/set.go Normal file
View 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
}