aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--ethutil/set.go32
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()
+}