eeva/servant
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fmapRouter.

Browse source
This commit is contained in:
Matthias Fischmann 2015-10-12 17:06:43 +02:00
parent 80396c8e47
commit 96f10add65
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
servant-server/src/Servant/Server/Internal/Router.hs
View file

@ -4,7 +4,7 @@ import Data.Map (Map)
import qualified Data.Map as M import qualified Data.Map as M
import Data.Monoid ((<>)) import Data.Monoid ((<>))
import Data.Text (Text) import Data.Text (Text)
import Network.Wai (Request, pathInfo) import Network.Wai (Request, Response, pathInfo)
import Servant.Server.Internal.PathInfo import Servant.Server.Internal.PathInfo
import Servant.Server.Internal.RoutingApplication import Servant.Server.Internal.RoutingApplication
@ -21,6 +21,13 @@ data Router =
| Choice Router Router | Choice Router Router
-- ^ left-biased choice between two routers -- ^ left-biased choice between two routers
fmapRouter :: (RouteResult Response -> RouteResult Response) -> Router -> Router
fmapRouter f (LeafRouter a) = LeafRouter $ \req cont -> a req (cont . f)
fmapRouter f (StaticRouter m) = StaticRouter (fmapRouter f <$> m)
fmapRouter f (DynamicRouter d) = DynamicRouter (fmapRouter f <$> d)
fmapRouter f (Choice r1 r2) = Choice (fmapRouter f r1) (fmapRouter f r2)
fmapRouter f (WithRequest g) = WithRequest (fmapRouter f . g)
-- | Smart constructor for the choice between routers. -- | Smart constructor for the choice between routers.
-- We currently optimize the following cases: -- We currently optimize the following cases:
-- --
@ -69,4 +76,3 @@ runRouter (Choice r1 r2) request respond =
then runRouter r2 request $ \ mResponse2 -> then runRouter r2 request $ \ mResponse2 ->
respond (mResponse1 <> mResponse2) respond (mResponse1 <> mResponse2)
else respond mResponse1 else respond mResponse1