Article Image
Article Image
read

最近寫 Elixir 時,有個慣用的 functional 小手法,順手記錄一下。

1. 無差別包裝 List

當處理的對象可能是 nillist,或是單個元素時,可以直接將它們包裝成單層的 list 進行後續操作。

Elixir 提供了 List.wrap 讓你直接將 nil 及 單個元素包裝成 list。對於原本就是 list 的對象則直接回傳:

# Elixir
List.wrap(nil)    # => []
List.wrap(1)      # => [1]
List.wrap([2, 3]) # => [2, 3]

值得一提的是包裝 nil 會回傳空的 list,這對稍後要介紹的連續技起了重要的作用。

Ruby 則可以用 Array()

# Ruby
Array(nil)    # => []
Array(1)      # => [1]
Array([2, 3]) # => [2, 3]

在 JavaScript 裡就得要自己拼了,以 Ramda 為例:

let wrap = R.compose(R.filter(R.identity), R.unnest, R.of)

wrap(null)    // => []
wrap(1)       // => [1]
wrap([2, 3])  // => [2, 3]

這裡用 R.unnest 而非 R.flatten,是因為它不會把串列全部攤平至只剩一層。這個特性在你想要處理的「元素」本身就是一個陣列時會排上用場。

2. 攤開 map 的結果。

flat_map 可以將 map 的結果攤開到外層串列中。這個看範例會比較好懂:

[1, 2, 3]
|> Enum.flat_map(fn x -> [x, x] end)

# => [1, 1, 2, 2, 3, 3]

flat_map 是函數式語言/函式庫基本的操作之一,應該都找得到。Ramda 裡叫 R.chain

3. 連續技:優雅的消失

結合上面兩個函式,先把含有 nil 的集合 wrap 成一個個的串列,再用 flat_mapflatten 攤開。這個時候就會發現 nil 在操作過程中直接消失了,不需要 filter,更不用 if/else 判斷。例如:

class = %{
  teacher: %{name: "John", age: 40},
  assistant: nil,
  students: [%{name: "Amber", age: 20}, %{name: "William", age: 22}]
}

class
|> Map.values
|> Enum.flat_map(&List.wrap/1)

# => [%{age: 20, name: "Amber"}, %{age: 22, name: "William"}, %{age: 40, name: "John"}]

實際應用時,通常會直接將上層集合傳到 flat_map 裡,並將處理的函式拿出來另外宣告,這樣可以用 pattern matching 對上層結構做更細緻的處理。

Happy hacking!

Blog Logo

taiansu


Published

Image

Tai-An Su

So Wrong, It's Right.

Back to Overview