Memoize requestBody IO action.
This commit is contained in:
parent
1e6fbd111d
commit
83f55259fa
1 changed files with 10 additions and 4 deletions
|
@ -10,13 +10,14 @@ module Servant.Server.Internal where
|
|||
import Control.Applicative
|
||||
import Control.Monad.Trans.Either
|
||||
import Data.Aeson
|
||||
import Data.Maybe (catMaybes)
|
||||
import Data.IORef
|
||||
import Data.Maybe (catMaybes, fromMaybe)
|
||||
import Data.Monoid
|
||||
import Data.Proxy
|
||||
import Data.String
|
||||
import Data.String.Conversions
|
||||
import Data.Text (Text)
|
||||
import Data.Text.Encoding (decodeUtf8)
|
||||
import Data.Text (Text)
|
||||
import GHC.TypeLits
|
||||
import Network.HTTP.Types hiding (Header)
|
||||
import Network.Wai
|
||||
|
@ -25,7 +26,12 @@ import Servant.Common.Text
|
|||
|
||||
toApplication :: RoutingApplication -> Application
|
||||
toApplication ra request respond = do
|
||||
ra request (routingRespond . routeResult)
|
||||
reqBodyRef <- newIORef Nothing
|
||||
let memoReqBody = fromMaybe <$> (do
|
||||
r <- requestBody request
|
||||
writeIORef reqBodyRef $ Just r
|
||||
return r ) <*> readIORef reqBodyRef
|
||||
ra request{ requestBody = memoReqBody } (routingRespond . routeResult)
|
||||
where
|
||||
routingRespond :: Either RouteMismatch Response -> IO ResponseReceived
|
||||
routingRespond (Left NotFound) =
|
||||
|
|
Loading…
Reference in a new issue