递归地列出目录下的所有文件，使用管道

Question

我读完Pipes tutorial，我想编写一个函数以递归方式列出目录中的所有文件。我试着用下面的代码：

enumFiles :: FilePath -> Producer' FilePath (PS.SafeT IO)() 
enumFiles path = 
    PS.bracket (openDirStream path) (closeDirStream) loop 
    where 
    loop :: DirStream -> Producer' FilePath (PS.SafeT IO)() 
    loop ds = PS.liftBase (readDirStream ds) >>= checkName 
     where 
     checkName :: FilePath -> Producer' FilePath (PS.SafeT IO)() 
     checkName "" = return() 
     checkName "." = loop ds 
     checkName ".." = loop ds 
     checkName name = PS.liftBase (getSymbolicLinkStatus newPath) 
         >>= checkStat newPath 
      where newPath = path </> name 

     checkStat path stat 
      | isRegularFile stat = yield path >> loop ds 
      | isDirectory stat = enumFiles path 
      | otherwise = loop ds 

然而，这生产者将尽快达到return()终止。我想我不是以正确的方式撰写它的，但我没有看到做这件事的正确方法。

Answer 1

只需改变这一行：

| isDirectory stat = enumFiles path 

到

| isDirectory stat = enumFiles path >> loop ds 

的代码是缺少这个递归情况下，递归。

您也可以达到打破这种生产成更小的生产商和管道组成：

{-# LANGUAGE RankNTypes #-} 

module Main where 

import qualified Pipes.Prelude as P 
import qualified Pipes.Safe as PS 

import   Control.Monad 
import   Pipes 
import   System.FilePath.Posix 
import   System.Posix.Directory 
import   System.Posix.Files 

readDirStream' :: FilePath -> Producer' FilePath (PS.SafeT IO)() 
readDirStream' dirpath = 
    PS.bracket (openDirStream dirpath) closeDirStream (forever . loop) 
    where 
    loop stream = 
     liftIO (readDirStream stream) >>= yield 

enumFiles :: FilePath -> Producer' FilePath (PS.SafeT IO)() 
enumFiles path = 
    readDirStream' path 
    >-> P.takeWhile (/= "") 
    >-> P.filter (not . flip elem [".", ".."]) 
    >-> P.map (path </>) 
    >-> forever (do 
        entry <- await 
        status <- liftIO $ getSymbolicLinkStatus entry 
        when (isDirectory status) (enumFiles entry) 
        when (isRegularFile status) (yield entry)) 

main :: IO() 
main = 
    PS.runSafeT $ runEffect (enumFiles "/tmp" >-> P.stdoutLn) 

我觉得它往往有助于使用forever从Control.Monad或从Pipe.Prelude代替人工递归组合子之一;它有助于减少像这样的小错别字。然而，正如孩子们所说，你的里程可能会有很大的不同。