Add node iterators for list

zyedidia · Mar 20, 2023 · 4410d23 · 4410d23
1 parent 7927e1c
commit 4410d23
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/list/list.go b/list/list.go
@@ -88,3 +88,21 @@ func (n *Node[V]) EachReverse(fn func(val V)) {
 		node = node.Prev
 	}
 }
+
+// EachNode calls 'fn' on every node from this node onward in the list.
+func (n *Node[V]) EachNode(fn func(n *Node[V])) {
+	node := n
+	for node != nil {
+		fn(node)
+		node = node.Next
+	}
+}
+
+// EachReverseNode calls 'fn' on every node from this node backward in the list.
+func (n *Node[V]) EachReverseNode(fn func(n *Node[V])) {
+	node := n
+	for node != nil {
+		fn(node)
+		node = node.Prev
+	}
+}