diff options
-rw-r--r-- | ethutil/set.go | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/ethutil/set.go b/ethutil/set.go new file mode 100644 index 000000000..80e2edde2 --- /dev/null +++ b/ethutil/set.go @@ -0,0 +1,32 @@ +package ethutil + +type Settable interface { + AsSet() UniqueSet +} + +type UniqueSet map[interface{}]struct{} + +func NewSet(v ...interface{}) UniqueSet { + set := make(UniqueSet) + for _, val := range v { + set.Insert(val) + } + + return set +} + +func (self UniqueSet) Insert(k interface{}) UniqueSet { + self[k] = struct{}{} + + return self +} + +func (self UniqueSet) Include(k interface{}) bool { + _, ok := self[k] + + return ok +} + +func Set(s Settable) UniqueSet { + return s.AsSet() +} |