計算機科学のブログ

コンテキストでの型の操作 Functor型クラス fmap関数、Map型、lookup関数、欠測値、Maybe型

入門Haskellプログラミング (Will Kurt(著)、株式会社クイープ(監修、翻訳)、翔泳社)のUNIT5(コンテキストでの型の操作)、LESSON 27(Functor型クラス)、27.5(練習問題)Q27-3の解答を求めてみる。

コード

import qualified Data.Map as Map
import Data.Char (toLower)

data RobotPart = RobotPart {cost :: Double} deriving Show

leftArm :: RobotPart
leftArm = RobotPart { cost = 1000 }

rightArm :: RobotPart
rightArm = RobotPart { cost = 1025}

robotHead :: RobotPart
robotHead = RobotPart { cost = 5092.25}

partsDB :: Map.Map Int RobotPart
partsDB =
    let keys = [1, 2, 3]
        vals = [leftArm, rightArm, robotHead]
        keyVals = zip keys vals
    in Map.fromList keyVals

printCost :: Maybe Double -> IO ()
printCost (Just cost) = print cost
printCost Nothing = putStrLn "データベースにIDはありませんでした。"

main :: IO ()
main = do
    putStr "IDを入力: "
    line <- getLine
    let id = read line
    let part = Map.lookup id partsDB
    let c = fmap cost part
    printCost c
    main

入出力結果(Terminal, Zsh)

% runghc sample3.hs
IDを入力: 0
データベースにIDはありませんでした。
IDを入力: 1
1000.0
IDを入力: 2
1025.0
IDを入力: 3
5092.25
IDを入力: 4
データベースにIDはありませんでした。
IDを入力: a
sample3.hs: Prelude.read: no parse
%