xhtml.Table: Better use of iterators

spotlightpa · May 21, 2024 · dc01120 · dc01120
1 parent cb1d9ff
commit dc01120
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/internal/xhtml/table.go b/internal/xhtml/table.go
@@ -14,8 +14,7 @@ func Tables(root *html.Node) iter.Seq2[*html.Node, TableNodes] {
 		tables := SelectSlice(root, WithAtom(atom.Table))
 		for _, tblNode := range tables {
 			var tbl TableNodes
-			rows := SelectSlice(tblNode, WithAtom(atom.Tr))
-			for _, row := range rows {
+			for row := range SelectAll(tblNode, WithAtom(atom.Tr)) {
 				tds := SelectSlice(row, func(n *html.Node) bool {
 					return n.DataAtom == atom.Td || n.DataAtom == atom.Th
 				})