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
|
{-# LANGUAGE OverloadedStrings #-}
module Main (main) where
import Web.Scotty
import Control.Monad.IO.Class (liftIO)
import qualified Blaze.ByteString.Builder as B
import qualified Data.ByteString as BS
import qualified Data.ByteString.Lazy as BSL
import qualified Data.Text.Lazy as T
main :: IO ()
main = scotty 3000 $ do
post "/echo" $ do
rd <- bodyReader
stream $ ioCopy rd $ return ()
post "/count" $ do
wb <- body -- this must happen before first 'rd'
rd <- bodyReader
let step acc = do
chunk <- rd
putStrLn "got a chunk"
let len = BS.length chunk
if len > 0
then step $ acc + len
else return acc
len <- liftIO $ step 0
text $ T.pack $ "uploaded " ++ show len ++ " bytes, wb len is " ++ show (BSL.length wb)
ioCopy :: IO BS.ByteString -> IO () -> (B.Builder -> IO ()) -> IO () -> IO ()
ioCopy reader close write flush = step >> flush where
step = do chunk <- reader
if (BS.length chunk > 0)
then (write $ B.insertByteString chunk) >> step
else close
|
