[interval] Fix panic in empty tree Size() method (#17)

zyedidia · May 16, 2022 · 7fb5054 · 7fb5054
1 parent ecd8f7d
commit 7fb5054
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/interval/itree.go b/interval/itree.go
@@ -350,12 +350,13 @@ func (n *node[I, V]) findSmallest() *node[I, V] {
 }
 
 func (n *node[I, V]) size() int {
-	s := 1
-	if n.left != nil {
-		s += n.left.size()
-	}
-	if n.right != nil {
-		s += n.right.size()
+	if n == nil {
+		return 0
 	}
+
+	s := 1
+	s += n.left.size()
+	s += n.right.size()
+
 	return s
 }