From 4db4ec1621767513ed2bd99d835c3e2f1c2b0e7e Mon Sep 17 00:00:00 2001
From: obscuren <geffobscura@gmail.com>
Date: Mon, 15 Sep 2014 01:10:50 +0200
Subject: Added unique set

---
 ethutil/set.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 ethutil/set.go

(limited to 'ethutil/set.go')

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()
+}
-- 
cgit v1.2.3


From 33a0dec8a157b9687ca6038f4deb011f3f1f7bdc Mon Sep 17 00:00:00 2001
From: obscuren <geffobscura@gmail.com>
Date: Mon, 15 Sep 2014 15:42:12 +0200
Subject: Improved catching up and refactored

---
 ethutil/set.go | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'ethutil/set.go')

diff --git a/ethutil/set.go b/ethutil/set.go
index 80e2edde2..7955edac0 100644
--- a/ethutil/set.go
+++ b/ethutil/set.go
@@ -4,9 +4,13 @@ type Settable interface {
 	AsSet() UniqueSet
 }
 
-type UniqueSet map[interface{}]struct{}
+type Stringable interface {
+	String() string
+}
+
+type UniqueSet map[string]struct{}
 
-func NewSet(v ...interface{}) UniqueSet {
+func NewSet(v ...Stringable) UniqueSet {
 	set := make(UniqueSet)
 	for _, val := range v {
 		set.Insert(val)
@@ -15,14 +19,14 @@ func NewSet(v ...interface{}) UniqueSet {
 	return set
 }
 
-func (self UniqueSet) Insert(k interface{}) UniqueSet {
-	self[k] = struct{}{}
+func (self UniqueSet) Insert(k Stringable) UniqueSet {
+	self[k.String()] = struct{}{}
 
 	return self
 }
 
-func (self UniqueSet) Include(k interface{}) bool {
-	_, ok := self[k]
+func (self UniqueSet) Include(k Stringable) bool {
+	_, ok := self[k.String()]
 
 	return ok
 }
-- 
cgit v1.2.3