Fix memory leak in TreeSet and TreeMap

2 files changed, 8 insertions, 0 deletions

diff --git a/gee/treemap.vala b/gee/treemap.vala
index 8b3917e..a300a3e 100644
--- a/gee/treemap.vala
+++ b/gee/treemap.vala
@@ -121,6 +121,10 @@ public class Gee.TreeMap<K,V> : Gee.AbstractMap<K,V> {
 		this.value_equal_func = value_equal_func;
 	}
 
+	~TreeMap () {
+		clear ();
+	}
+
 	private void rotate_right (ref Node<K, V> root) {
 		Node<K,V> pivot = (owned) root.left;
 		pivot.color = root.color;
diff --git a/gee/treeset.vala b/gee/treeset.vala
index 3548cf1..d380b87 100644
--- a/gee/treeset.vala
+++ b/gee/treeset.vala
@@ -63,6 +63,10 @@ public class Gee.TreeSet<G> : AbstractSet<G>, SortedSet<G> {
 		this.compare_func = compare_func;
 	}
 
+	~TreeSet () {
+		clear ();
+	}
+
 	/**
 	 * {@inheritDoc}
 	 */