diff options
Diffstat (limited to 'Utility/JSONStream.hs')
| -rw-r--r-- | Utility/JSONStream.hs | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/Utility/JSONStream.hs b/Utility/JSONStream.hs index c5d4e1c8d..45978ef06 100644 --- a/Utility/JSONStream.hs +++ b/Utility/JSONStream.hs @@ -5,30 +5,32 @@ - License: BSD-2-clause -} -{-# LANGUAGE GADTs #-} +{-# LANGUAGE GADTs, OverloadedStrings #-} module Utility.JSONStream ( JSONChunk(..), start, add, + addNestedObject, end ) where import Data.Aeson import qualified Data.Text as T import qualified Data.ByteString.Lazy as B +import qualified Data.ByteString.Lazy.UTF8 as BU8 import Data.Char import Data.Word data JSONChunk v where - JSONChunk :: ToJSON v => [(String, v)] -> JSONChunk [(String, v)] AesonObject :: Object -> JSONChunk Object + JSONChunk :: ToJSON v => [(String, v)] -> JSONChunk [(String, v)] encodeJSONChunk :: JSONChunk v -> B.ByteString +encodeJSONChunk (AesonObject o) = encode o encodeJSONChunk (JSONChunk l) = encode $ object $ map mkPair l where mkPair (s, v) = (T.pack s, toJSON v) -encodeJSONChunk (AesonObject o) = encode o {- Aeson does not support building up a larger JSON object piece by piece - with streaming output. To support streaming, a hack: @@ -43,11 +45,21 @@ start a add :: JSONChunk a -> B.ByteString add a - | not (B.null b) && B.head b == startchar = B.cons addchar (B.drop 1 b) + | not (B.null b) && B.head b == startchar = + B.cons addchar (B.drop 1 b) | otherwise = bad b where b = start a +addNestedObject :: String -> B.ByteString -> B.ByteString +addNestedObject s b = B.concat + [ ",\"" + , BU8.fromString s + , "\":" + , b + , "}" + ] + end :: B.ByteString end = endchar `B.cons` sepchar `B.cons` B.empty |