Haskell:如何在命令行参数中读取int？

Question

我试图从命令行获取一个int值并将其传递给该disp函数.

import System(getArgs)

main = do
    args <- getArgs
    disp $ read $ head args :: Int

disp n = take n $ repeat 'S'

ghc给出的错误是

 Couldn't match expected type `Int' with actual type `[Char]'
 In the expression: disp $ read $ head args :: Int
 In the expression:
   do { args <- getArgs;
            disp $ read $ head args :: Int }
 In an equation for `main':
     main
       = do { args <- getArgs;
                  disp $ read $ head args :: Int }

谢谢.

Answer 1

问题在于优先级:类型签名总是尝试应用于整个表达式(仅使用括号作用域).所以你的disp $ read $ head args :: Int解析(disp $ read $ head args) :: Int,显然是不正确的.您可以像这样使用括号:

disp (read $ head args :: Int)

或省略类型签名,因为GHC可以在这种情况下推断它:

disp $ read $ head args

此代码仍然无法正常工作,因为您处于IO monad中,因此您需要生成IO操作.您可以通过打印结果来执行此操作,例如:

putStrLn $ disp $ read $ head args

Answer 2

您可以封装拉动整数命令行参数,如下所示:

getIntArg :: IO Int
getIntArg = fmap (read . head) getArgs

这是有效的,因为Monads是Functors.或者你也可以做同样的事情liftM.

那样你的main功能就变成了:

main = do
    n <- getIntArg
    disp n

如果您添加某种类型的打印功能,disp如其他答案中所述.

Answer 3

只需删除您在那里添加的显式类型,它就会起作用.对类型推断有信心.:)添加print $ ...或类似的东西,以纠正新的错误.

会发生什么,take已知的类型,所以disp预期的参数类型也是已知的.是的Int.因此适用read将适用.

做得少,做得更多.