summaryrefslogtreecommitdiffhomepage
path: root/Network/DNS/StateBinary.hs
diff options
context:
space:
mode:
Diffstat (limited to 'Network/DNS/StateBinary.hs')
-rw-r--r--Network/DNS/StateBinary.hs10
1 files changed, 6 insertions, 4 deletions
diff --git a/Network/DNS/StateBinary.hs b/Network/DNS/StateBinary.hs
index 6898d3b..6b3fb6c 100644
--- a/Network/DNS/StateBinary.hs
+++ b/Network/DNS/StateBinary.hs
@@ -4,10 +4,9 @@ module Network.DNS.StateBinary where
import Blaze.ByteString.Builder
import Control.Applicative
import Control.Monad.State
-import Data.Monoid
-import Data.Attoparsec
+import Data.Attoparsec.ByteString
+import qualified Data.Attoparsec.ByteString.Lazy as AL
import Data.Attoparsec.Enumerator
-import qualified Data.Attoparsec.Lazy as AL
import Data.ByteString (ByteString)
import qualified Data.ByteString as BS (unpack, length)
import qualified Data.ByteString.Lazy as BL (ByteString)
@@ -17,10 +16,13 @@ import Data.IntMap (IntMap)
import qualified Data.IntMap as IM (insert, lookup, empty)
import Data.Map (Map)
import qualified Data.Map as M (insert, lookup, empty)
+import Data.Monoid
import Data.Word
import Network.DNS.Types
import Prelude hiding (lookup, take)
+import qualified Data.Attoparsec.Types as T (Parser)
+
----------------------------------------------------------------
type SPut = State WState Write
@@ -83,7 +85,7 @@ wsPush dom pos = do
----------------------------------------------------------------
-type SGet = StateT PState Parser
+type SGet = StateT PState (T.Parser ByteString)
data PState = PState {
psDomain :: IntMap Domain