我是否可以使用命名记录设置管道 CSV 的列顺序



我的代码在下面。似乎数据以随机顺序输出,这是有道理的,因为NamedRecord只是一个Map。有没有办法执行特定订单?

我认为一种选择是创建一个 Conduit (Named a) m (Row Text) 类型的管道,但这似乎是一个足够常见的用例,应该已经存在一些标准。

processor :: Monad m => Conduit (Named FalconRow) m (Named HefRow)
processor = do
  value <- await 
  case value of
    Nothing -> return ()
    Just v -> do
      let transformed = (fixRow . getNamed) v
      CL.sourceList $ Prelude.map Named transformed
runTranslation :: IO ()
runTranslation = runResourceT $ 
  transformCSV defCSVSettings 
               (sourceFile "input.csv") 
               processor
               (sinkFile "output.csv")

我最终只是为此制作了自己的函数:

sortColumns :: (Monad m, DefaultOrdered a, ToNamedRecord a) => Conduit (Named a) m (Row ByteString)
sortColumns = CL.map doSort
doSort :: (DefaultOrdered a, ToNamedRecord a) => Named a -> Row ByteString
doSort value =
  let columnMap = toNamedRecord $ getNamed value
      headers = toList $ headerOrder $ getNamed value
      bsValue = map (title -> findWithDefault "" title columnMap) headers
  in  bsValue

最新更新