在 elixir 裡可以用 Stream 來表示無限長的序列,例如 0,1,2,3... 就可以表示為:
iex> s = Stream.iterate(0, & &1 + 1)
#Function<62.50989570/2 in Stream.unfold/2>
如果我們想要計算這個數列中每5個數的和,就可以使用 Stream.transform 函式:
iex> s1 = Stream.transform(s, {0, 0}, fn x, {sum, count} ->
...> if count == 5 do
...> {[sum], {x, 1}}
...> else
...> {[], {sum + x, count + 1}}
...> end
...> end)
#Function<60.50989570/2 in Stream.transform/3>
iex> Enum.take(s1, 10)
[10, 35, 60, 85, 110, 135, 160, 185, 210, 235]
它是 Enum.flat_map_reduce
的 Stream 版本。