在Haskell中解析key="value"格式命令行参数的更优方案是什么？

现有实现可优化点 #

你的现有实现已经能覆盖核心需求，不过存在几个可以调整的细节：

• 依赖Data.List.Extra的非base库函数，如果想要减少依赖可以用base自带的函数替换
• 嵌套case写法可以用Monad/Maybe组合子简化，可读性更高
• 未明确多同名参数、引号不闭合等边界情况的行为定义

更简洁的实现版本 #

下面是只依赖base库的简化实现，逻辑和你的原始版本完全一致：

import Data.List (isPrefixOf, stripPrefix)
import Data.Maybe (fromMaybe, find)
import Text.Read (readMaybe)

obtainArg :: String -> String -> [String] -> String
obtainArg defaultValue keyWord args = fromMaybe defaultValue $ do
  -- 找到匹配key前缀的参数
  arg <- find ((keyWord ++ "=") `isPrefixOf`) args
  -- 摘除key=前缀，拿到值部分
  valRaw <- stripPrefix (keyWord ++ "=") arg
  -- 尝试解析带引号的字符串，解析失败直接返回原始值
  pure $ fromMaybe valRaw (readMaybe valRaw)

如果需要兼容旧版本GHC（base <4.19），可以自己实现stripPrefix：

stripPrefix :: String -> String -> Maybe String
stripPrefix [] s = Just s
stripPrefix _ [] = Nothing
stripPrefix (p:ps) (s:ss)
  | p == s = stripPrefix ps ss
  | otherwise = Nothing

实践建议 #

• 边界行为明确：
  • 当出现多个同key参数时，现有实现取第一个匹配项，如果需要取最后一项，把find换成倒序查找的逻辑即可
  • 当带引号的参数引号不闭合时，readMaybe会返回Nothing，直接返回原始字符串，如果你需要这种情况抛出错误提示用户，可以加对应的错误处理逻辑
  • 如果需要支持值内包含等号的场景（比如argx=a=b=c），现有逻辑已经支持，会返回a=b=c
• 如果你的CLI程序逻辑比较复杂，不要重复造轮子，直接使用Haskell生态成熟的命令行解析库，这类库原生支持key=value格式、默认值配置、类型校验、自动生成帮助文本等能力，能大幅减少出错概率
• 建议补充单元测试覆盖常见场景：无匹配参数返回默认值、无引号普通值、带空格的带引号值、值包含转义引号、多同key参数等

内容的提问来源于stack exchange，提问作者Iván Renison