1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
module Main where
import Control.Exception
import Control.Monad
import System.Mem
import Data.IORef
import Data.ByteString (ByteString, packCStringLen)
import Foreign.Ptr
import Control.DeepSeq
import Data.Compact
import Data.Compact.Internal
import Data.Compact.Serialized
assertFail :: String -> IO ()
assertFail msg = throwIO $ AssertionFailed msg
assertEquals :: (Eq a, Show a) => a -> a -> IO ()
assertEquals expected actual =
if expected == actual then return ()
else assertFail $ "expected " ++ (show expected)
++ ", got " ++ (show actual)
serialize :: NFData a => a -> IO (SerializedCompact a, [ByteString])
serialize val = do
cnf <- compactSized 4096 True val
bytestrref <- newIORef undefined
scref <- newIORef undefined
withSerializedCompact cnf $ \sc -> do
writeIORef scref sc
performMajorGC
bytestrs <- forM (serializedCompactBlockList sc) $ \(ptr, size) -> do
packCStringLen (castPtr ptr, fromIntegral size)
writeIORef bytestrref bytestrs
performMajorGC
bytestrs <- readIORef bytestrref
sc <- readIORef scref
return (sc, bytestrs)
main = do
let val = ("hello", 1, 42, 42, Just 42) ::
(String, Int, Int, Integer, Maybe Int)
(sc, bytestrs) <- serialize val
performMajorGC
mcnf <- importCompactByteStrings sc bytestrs
case mcnf of
Nothing -> assertFail "import failed"
Just cnf -> assertEquals val (getCompact cnf)
|
