計算機科学のブログ

コンテキストでの型の操作 Functor型クラス Map型、欠損値、Maybe型

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

lesson/package.yaml

name:                lesson
version:             0.1.0.0
github:              "githubuser/lesson"
license:             BSD3
author:              "Author name here"
maintainer:          "example@example.com"
copyright:           "2022 Author name here"

extra-source-files:
- README.md
- ChangeLog.md

# Metadata used when publishing your package
# synopsis:            Short description of your package
# category:            Web

# To avoid duplicated efforts in documentation and dealing with the
# complications of embedding Haddock markup inside cabal files, it is
# common to point users to the README.md file.
description:         Please see the README on GitHub at <https://github.com/githubuser/lesson#readme>

dependencies:
- base >= 4.7 && < 5
- containers

library:
  source-dirs: src

executables:
  lesson-exe:
    main:                Main.hs
    source-dirs:         app
    ghc-options:
    - -threaded
    - -rtsopts
    - -with-rtsopts=-N
    dependencies:
    - lesson

tests:
  lesson-test:
    main:                Spec.hs
    source-dirs:         test
    ghc-options:
    - -threaded
    - -rtsopts
    - -with-rtsopts=-N
    dependencies:
    - lesson

コード

lesson/app/Main.hs

module Main where

import qualified Data.Map as Map
import Lib
  ( RobotPart (..),
  )
import System.Environment (getArgs)

leftArm =
  RobotPart
    { cost = 2
    }

rightArm =
  RobotPart
    { cost = 3
    }

robotHead =
  RobotPart
    { cost = 4
    }

partsDB :: Map.Map Int RobotPart
partsDB =
  Map.fromList $
    zip
      [1, 2, 3]
      [leftArm, rightArm, robotHead]

main :: IO ()
main = do
  args <- getArgs
  let id = read $ head args
  let part = Map.lookup id partsDB
  print $ cost <$> part

lesson/src/Lib.hs

module Lib
  ( RobotPart (..),
  )
where

data RobotPart = RobotPart
  { cost :: Double
  }

入出力結果(Terminal, Zsh)

% stack exec lesson-exe  
lesson-exe: Prelude.head: empty list
% stack exec lesson-exe 0
Nothing
% stack exec lesson-exe 1
Just 2.0
% stack exec lesson-exe 2
Just 3.0
% stack exec lesson-exe 3
Just 4.0
% stack exec lesson-exe 4
Nothing
%