+{-# LANGUAGE
+ BangPatterns
+ , UnicodeSyntax
+ #-}
module Network.HTTP.Lucu.ResponseWriter
( responseWriter
)
import Data.Sequence (ViewR(..))
import Network.HTTP.Lucu.Config
import Network.HTTP.Lucu.Format
+import Network.HTTP.Lucu.HandleLike
import Network.HTTP.Lucu.Headers
import Network.HTTP.Lucu.HttpVersion
import Network.HTTP.Lucu.Interaction
import Network.HTTP.Lucu.Postprocess
import Network.HTTP.Lucu.Response
import Prelude hiding (catch)
-import System.IO
+import System.IO (stderr)
-responseWriter :: Config -> Handle -> InteractionQueue -> ThreadId -> IO ()
+responseWriter :: HandleLike h => Config -> h -> InteractionQueue -> ThreadId -> IO ()
responseWriter !cnf !h !tQueue !readerTID
= awaitSomethingToWrite
`catches`
[ Handler (( \ _ -> return () ) :: IOException -> IO ())
, Handler ( \ ThreadKilled -> return () )
- , Handler ( \ BlockedIndefinitely -> hPutStrLn stderr "requestWriter: blocked indefinitely" )
+ , Handler ( \ BlockedIndefinitelyOnSTM -> hPutStrLn stderr "requestWriter: blocked indefinitely" )
, Handler (( \ e -> hPutStrLn stderr (show e) ) :: SomeException -> IO ())
]
where
awaitSomethingToWrite :: IO ()
awaitSomethingToWrite
= {-# SCC "awaitSomethingToWrite" #-}
- do action
- <- atomically $!
- -- キューが空でなくなるまで待つ
- do queue <- readTVar tQueue
- -- GettingBody 状態にあり、Continue が期待され
- -- ã\81¦ã\82\90ã\81¦ã\80\81ã\81\9dã\82\8cã\81\8cã\81¾ã\81 é\80\81ä¿¡å\89\8dã\81ªã\81®ã\81§ã\81\82ã\82\8cã\81°ã\80\81
- -- Continue を送信する。
- case S.viewr queue of
- EmptyR -> retry
- _ :> itr -> do state <- readItr itr itrState id
+ join $!
+ atomically $!
+ -- キューが空でなくなるまで待つ
+ do queue <- readTVar tQueue
+ -- GettingBody 状態にあり、Continue が期待されてゐ
+ -- ã\81¦ã\80\81ã\81\9dã\82\8cã\81\8cã\81¾ã\81 é\80\81ä¿¡å\89\8dã\81ªã\81®ã\81§ã\81\82ã\82\8cã\81°ã\80\81Continue ã\82\92é\80\81
+ -- 信する。
+ case S.viewr queue of
+ EmptyR -> retry
+ _ :> itr -> do state <- readItr itr itrState id
- if state == GettingBody then
- writeContinueIfNecessary itr
- else
- if state >= DecidingBody then
- writeHeaderOrBodyIfNecessary itr
- else
- retry
- action
+ if state == GettingBody then
+ writeContinueIfNecessary itr
+ else
+ if state >= DecidingBody then
+ writeHeaderOrBodyIfNecessary itr
+ else
+ retry
writeContinueIfNecessary :: Interaction -> STM (IO ())
- writeContinueIfNecessary itr
+ writeContinueIfNecessary !itr
= {-# SCC "writeContinueIfNecessary" #-}
- itr `seq`
do expectedContinue <- readItr itr itrExpectedContinue id
if expectedContinue then
do wroteContinue <- readItr itr itrWroteContinue id
retry
writeHeaderOrBodyIfNecessary :: Interaction -> STM (IO ())
- writeHeaderOrBodyIfNecessary itr
+ writeHeaderOrBodyIfNecessary !itr
-- DecidingBody 以降の状態にあり、まだヘッダを出力する前であ
-- れば、ヘッダを出力する。ヘッダ出力後であり、bodyToSend が
-- 空でなければ、それを出力する。空である時は、もし状態が
-- Done であれば後処理をする。
= {-# SCC "writeHeaderOrBodyIfNecessary" #-}
- itr `seq`
do wroteHeader <- readItr itr itrWroteHeader id
if not wroteHeader then
return $! writeBodyChunk itr
writeContinue :: Interaction -> IO ()
- writeContinue itr
+ writeContinue !itr
= {-# SCC "writeContinue" #-}
- itr `seq`
do let cont = Response {
resVersion = HttpVersion 1 1
, resStatus = Continue
awaitSomethingToWrite
writeHeader :: Interaction -> IO ()
- writeHeader itr
+ writeHeader !itr
= {-# SCC "writeHeader" #-}
- itr `seq`
do res <- atomically $! do writeItr itr itrWroteHeader True
readItr itr itrResponse id
hPutResponse h res
awaitSomethingToWrite
writeBodyChunk :: Interaction -> IO ()
- writeBodyChunk itr
+ writeBodyChunk !itr
= {-# SCC "writeBodyChunk" #-}
- itr `seq`
do willDiscardBody <- atomically $! readItr itr itrWillDiscardBody id
willChunkBody <- atomically $! readItr itr itrWillChunkBody id
chunk <- atomically $! do chunk <- readItr itr itrBodyToSend id
unless willDiscardBody
$ do if willChunkBody then
do hPutStr h (fmtHex False 0 $! fromIntegral $! C8.length chunk)
- C8.hPut h (C8.pack "\r\n")
- C8.hPut h chunk
- C8.hPut h (C8.pack "\r\n")
+ hPutLBS h (C8.pack "\r\n")
+ hPutLBS h chunk
+ hPutLBS h (C8.pack "\r\n")
else
- C8.hPut h chunk
+ hPutLBS h chunk
hFlush h
awaitSomethingToWrite
finishBodyChunk :: Interaction -> IO ()
- finishBodyChunk itr
+ finishBodyChunk !itr
= {-# SCC "finishBodyChunk" #-}
- itr `seq`
do willDiscardBody <- atomically $! readItr itr itrWillDiscardBody id
willChunkBody <- atomically $! readItr itr itrWillChunkBody id
when (not willDiscardBody && willChunkBody)
- $ C8.hPut h (C8.pack "0\r\n\r\n") >> hFlush h
+ $ hPutLBS h (C8.pack "0\r\n\r\n") >> hFlush h
finalize :: Interaction -> IO ()
- finalize itr
+ finalize !itr
= {-# SCC "finalize" #-}
- itr `seq`
do finishBodyChunk itr
willClose <- atomically $!
do queue <- readTVar tQueue