- set key to key if direct match when inserting into red-black tree. fixes the issue when the key is a struct type, which is how treeset uses the red-black tree #31

trees/redblacktree/redblacktree.go

@@ -72,6 +72,7 @@ func (tree *Tree) Put(key interface{}, value interface{}) {
 			compare := tree.Comparator(key, node.Key)
 			switch {
 			case compare == 0:
+				node.Key = key
 				node.Value = value
 				return
 			case compare < 0: