Added unique set

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()
+}