do 區塊與 <- 綁定
February 12, 2022在〈解決巢狀運算的 Monad〉最後說到,do 與 <- 綁定其實可以用在 Monad,而不只是 IO,這邊就實際來看看,像 Maybe、list 可以如何與 do 及 <- 一起使用。
Maybe、do 與 <-
假設從某個運算取得了 maybeName::Maybe String,又從另一個運算取得了 maybeBirth::Maybe Int,現在打算將它們進行串接,取得一個字串,你會怎麼做?如果使用模式比對,你得判斷是不是有值,你可能會想到 Maybe 具有 Applicative 行為,它已經實作了有無值的判斷,這是個好主意!你可以使用 (++) <$> something <*> (show <$> another) 來得到想要的結果。
不過,你有更好的做法:
account :: Maybe String -> Maybe Int -> Maybe String
account maybeName maybeBirth = do
name <- maybeName
birth <- maybeBirth
return (name ++ (show birth))
這麼一來就可以如此使用:
ghci> account (Just "Justin") (Just 526)
Just "Justin526"
ghci>
如果不使用 do 區塊與 <- 綁定,就得逐層地寫出 >>= 與 lambda 函式:
account :: Maybe String -> Maybe Int -> Maybe String
account maybeName maybeBirth =
maybeName >>= (\name -> maybeBirth >>= (\birth -> return (name ++ (show birth))))
括號只是為了讓你便於識別 lambda 函式罷了,來去掉括號並略為排版:
account :: Maybe String -> Maybe Int -> Maybe String
account maybeName maybeBirth =
maybeName >>= \name ->
maybeBirth >>= \birth ->
return (name ++ (show birth))
就結論而言,do 區塊就是銜接 Monad 的簡便方式,這也說明了,為什麼 do 區塊最後必須傳回 Monad。
List、do 與 <-
將一開始的 account 函式型態宣告拿掉,也就是只寫為:
account mName mBirth = do
name <- mName
birth <- mBirth
return (name ++ (show birth))
由編譯器為你判定型態,你覺得結果會怎樣?
ghci> :t account
account :: (Monad m, Show a) => m [Char] -> m a -> m [Char]
ghci> account ["Justin"] [526]
["Justin526"]
ghci> account ["Justin"] []
[]
ghci> account [] [526]
[]
ghci>
實際上,account 只用到了 Monad 定義的 return 方法,判斷為 (Monad m, Show a) => m [Char] -> m a -> m [Char] 是最寬鬆的結果,如此一來,account 不僅可適用 Maybe,也可以適用 list,就像上面示範的,給 account 的 list 只有一個元素時很好理解,就是將唯一的元素綁定到名稱上,如果給 account 的 list 不只有一個元素時會如何呢?
ghci> account ["Justin", "Monica"] [804]
["Justin804","Monica804"]
ghci> account ["Justin", "Monica"] [526, 723]
["Justin526","Justin723","Monica526","Monica723"]
ghci> account ["Justin"] [526, 723]
["Justin526","Justin723"]
ghci>
難以理解嗎?重新看看不用 do 時會怎麼寫,就應該能知道為什麼對 list 做 do 與 <-,會有這樣的結果:
account mName mBirth =
mName >>= \name ->
mBirth >>= \birth ->
return (name ++ (show birth))
我們對 mName 套用 >>=,記得嗎?list 在實作 >>= 的方式是 concat (fmap f xs),也就是說對 mName 套用 >>=,結果就是 concat (fmap f mName),如果 mName 實際上超過一個元素,就會逐一取得並套用 f 後串接起來,後 f 是個 Lambda,它對 mBirth 套用 >>=,如果 mBrith 實際上超過一個元素,就會逐一取得並套用 f 後串接起來,最後就得到與 [name ++ (show birth) | name <- ["Justin", "Monica"], birth <- [526, 723]] 這個 Comprehension 表示式的相同結果。
將以下函式與上頭的函式對照一下,你就更容易記得,對 list 使用 do 與 <- 會是什麼結果了:
account mName mBirth =
[name ++ (show birth) |
name <- mName,
birth <- mBirth]
也就是說,對 list 使用 do 與 <- 時,將 <- 想成是 Comprehension 表示式的 <- 就是了,實際上,Comprehension 表示式是個語法蜜糖,跟使用 do 一樣,Comprehension 表示式最後也是使用 >>= 來進行運算。
Control.Monad 模組
方才看而,一開始的 account 函式可以讓編譯器自動推斷為 (Monad m, Show a) => m [Char] -> m a -> m [Char],因此,account 也可以適用 IO:
ghci> account getLine (return 526)
Justin
"Justin526"
ghci>
這讓我們想起來,在〈來寫些迴圈吧!?〉使用過的一些函式,記得嗎?那些函式是定義在 Control.Monad 模組,實際上它們並不只能用在 IO,像 forM 型態是 Monad m => [a] -> (a -> m b) -> m [b],除了使用在 IO,也可以使用在 Maybe 或 list:
ghci> import Control.Monad
ghci> forM [1, 2, 3, 4] (\x -> Just (x + 10))
Just [11,12,13,14]
ghci> forM [1, 2, 3, 4] (\x -> [x])
[[1,2,3,4]]
ghci> forM [1, 2, 3, 4] (\x -> [x, x])
[[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]
ghci>


