diff options
author | huangyi <yi.codeplayer@gmail.com> | 2011-10-23 13:22:18 +0800 |
---|---|---|
committer | huangyi <yi.codeplayer@gmail.com> | 2011-10-23 13:22:18 +0800 |
commit | d126d8fe4159b68feadd4145f0e2a1f90d7ca502 (patch) | |
tree | 450b233a2d7bdf17b64e8b28e102b0da0f18798e /Network/DNS/Response.hs | |
parent | 1cc2f6a8882ce9e1a79ab8789b50a79b2d021d14 (diff) |
make encoder and decoder invertable, and tests to ensure that.
Diffstat (limited to 'Network/DNS/Response.hs')
-rw-r--r-- | Network/DNS/Response.hs | 20 |
1 files changed, 14 insertions, 6 deletions
diff --git a/Network/DNS/Response.hs b/Network/DNS/Response.hs index 36f6203..abcb8ba 100644 --- a/Network/DNS/Response.hs +++ b/Network/DNS/Response.hs @@ -1,6 +1,6 @@ {-# LANGUAGE OverloadedStrings #-} -module Network.DNS.Response (responseIter, parseResponse) where +module Network.DNS.Response (responseIter, parseResponse, runDNSFormat, runDNSFormat_) where import Control.Applicative import Control.Monad @@ -12,13 +12,21 @@ import Network.DNS.Internal import Network.DNS.StateBinary import Data.Enumerator (Enumerator, Iteratee, run_, ($$)) import Data.ByteString (ByteString) +import qualified Data.ByteString.Lazy as BL -responseIter :: Iteratee ByteString IO (DNSFormat, PState) -responseIter = runSGet decodeResponse +runDNSFormat :: BL.ByteString -> Either String (DNSFormat, PState) +runDNSFormat bs = runSGet decodeResponse bs -parseResponse :: Enumerator ByteString IO (a,b) - -> Iteratee ByteString IO (a,b) - -> IO a +runDNSFormat_ :: BL.ByteString -> Either String DNSFormat +runDNSFormat_ bs = fst <$> runDNSFormat bs + +responseIter :: Monad m => Iteratee ByteString m (DNSFormat, PState) +responseIter = iterSGet decodeResponse + +parseResponse :: (Functor m, Monad m) + => Enumerator ByteString m (a,b) + -> Iteratee ByteString m (a,b) + -> m a parseResponse enum iter = fst <$> run_ (enum $$ iter) ---------------------------------------------------------------- |