| 1 | module main |
| 2 | |
| 3 | struct SquareIterator { |
| 4 | arr []int |
| 5 | mut: |
| 6 | idx int |
| 7 | } |
| 8 | |
| 9 | fn (mut iter SquareIterator) next[K]() ?K { |
| 10 | if iter.idx >= iter.arr.len { |
| 11 | return none |
| 12 | } |
| 13 | defer { |
| 14 | iter.idx++ |
| 15 | } |
| 16 | return iter.arr[iter.idx] * iter.arr[iter.idx] |
| 17 | } |
| 18 | |
| 19 | fn main() { |
| 20 | nums := [1, 2, 3, 4, 5] |
| 21 | iter := SquareIterator{ |
| 22 | arr: nums |
| 23 | } |
| 24 | |
| 25 | for squared in iter { |
| 26 | println(squared) |
| 27 | } |
| 28 | } |
| 29 |