我正在尝试将我写了一段时间的日志文件解析器转换回管道,我遇到了一个问题。我将简化解析器本身的细节,因为这与问题无关。我有一个看起来像这样的日志文件:
200 GET
404 POST
500 GET
FOO
301 PUT
302 GET
201 POST
所以解析代码非常简单:
data SimpleLogEntry = SimpleLogEntry {
status :: Int
, method :: String
} deriving (Show, Eq)
parseHTTPStatus :: Parser Int
parseHTTPStatus = validate <$> decimal
where validate d = if (d >= 200 && d < 999) then d else 100
parseHTTPMethod :: Parser String
parseHTTPMethod =
(stringCI "GET" *> return "Get")
<|> (stringCI "POST" *> return "Post")
<|> (stringCI "PUT" *> return "Put")
<|> return "Unknown"
parseLogLine :: Parser SimpleLogEntry
parseLogLine = fmap SimpleLogEntry
parseHTTPStatus
<*> (space *> parseHTTPMethod)
到现在为止还挺好。这是我在管道中实现的方法:
import Prelude hiding (lines)
import Control.Applicative
import Control.Monad.IO.Class (liftIO)
import Control.Monad.Trans.Resource (runResourceT, ResourceT)
import Data.Attoparsec.ByteString.Char8
import qualified Data.ByteString as B
import qualified Data.ByteString.Char8 as B8
import Data.Conduit
import qualified Data.Conduit.Attoparsec as CA
import qualified Data.Conduit.Binary as CB
import qualified Data.Conduit.List as CL
logLines:: Source (ResourceT IO) B.ByteString
logLines = CB.sourceFile "~/test.log" $= CB.lines
parseEntry :: ConduitM B8.ByteString SimpleLogEntry (ResourceT IO) ()
parseEntry = CA.conduitParserEither parseLogLine =$= awaitForever go
where
go (Left err) = liftIO $ putStrLn ("Got an error: " ++ CA.errorMessage err)
go (Right (_, logEntry)) = yield logEntry
sink :: Sink SimpleLogEntry (ResourceT IO) ()
sink = CL.mapM_ (\t -> liftIO $ putStrLn $ "Got a status: " ++ (show . status) t)
main :: IO ()
main = runResourceT $ logLines $= parseEntry $$ sink
运行时main
我得到这个输出:
Got a status: 200
Got a status: 404
Got a status: 500
Got an error: Failed reading: takeWhile1
我无法理解为什么管道会在此时终止,而不是像我想做的那样继续解析文件的下一行。阅读文档Data.Conduit.Attoparsec
,这似乎正是该用例conduitParserEither
的设计目的。
更新
根据@Fabian,事实证明这conduitParserEither
并不是我真正想要的。parseEntry
这是我想做的定义:
parseEntry' :: ConduitM B8.ByteString SimpleLogEntry (ResourceT IO) ()
parseEntry' = (CL.map (parseOnly parseLogLine)) =$= awaitForever go
where
go (Left err) = liftIO $ putStrLn ("Got an error: " ++ err)
go (Right logEntry) = yield logEntry