mongodb/Database/MongoDB/Util.hs

43 lines
986 B
Haskell

module Database.MongoDB.Util
(
putI8, putI32, putI64, putNull, putS,
getI8, getI32, getI64, getC, getS, getNull,
)
where
import Control.Monad
import Data.Binary
import Data.Binary.Get
import Data.Binary.Put
import qualified Data.ByteString.Lazy as L
import qualified Data.ByteString.Lazy.UTF8 as L8
import Data.Char (chr, ord)
import Data.Int
getC = liftM chr getI8
getI8 = liftM fromIntegral getWord8
getI32 :: Get Int32
getI32 = liftM fromIntegral getWord32le
getI64 :: Get Int64
getI64 = liftM fromIntegral getWord64le
getS :: Get (Integer, L8.ByteString)
getS = getLazyByteStringNul >>= \s -> return (fromIntegral $ L.length s + 1, s)
getNull = do {'\0' <- getC; return ()}
putI8 :: (Integral i) => i -> Put
putI8 = putWord8 . fromIntegral
putI32 :: Int32 -> Put
putI32 = putWord32le . fromIntegral
putI64 :: Int64 -> Put
putI64 = putWord64le . fromIntegral
putNull = putI8 0
putS :: L8.ByteString -> Put
putS s = putLazyByteString s >> putNull