Haskell Servant 将自定义数据传递给身份验证处理程序

Question

我正在使用自定义 monad（带有阅读器）轻松地将数据库池等数据传递给我的处理程序（在使用自定义 monad 之前，我曾经将连接作为 fn 参数传递）。

这就是我定义我的自定义 monad 的方式：

newtype Controller a = Controller
    { runController :: ReaderT ServerEnvironment Handler a
    } deriving ( Functor, Applicative, Monad, MonadReader ServerEnvironment, 
                 MonadError ServantErr, MonadIO )

这个ServerEnvironment 只是我用来携带我的数据的自定义数据类型。

问题是对于我的AuthHandler，我必须专门使用以下功能：

r -> Handler usr

作为身份验证处理程序，我不能使用我的自定义处理程序：

r -> Controller usr

我也无法传递我的ConnectionPool，因为签名不能是：

ConnPool -> r -> Handler usr

那么，如何在不使用全局IO 状态的情况下将额外数据传递给servant 中的身份验证处理程序？

Answer 1

您放入上下文中的AuthHandler 不必在顶层定义！通常，您需要在main 中执行此操作，以便您可以访问您创建的数据库连接等：

type API = 
  ... :<|> (AuthProtect "myProtection" :> ...) :<|> ...

type instance AuthServerData (AuthProtect "myProtection") = User

server :: ServerEnvironment -> Server API
server env = ...

setupEnv :: IO ServerEnvironment
setupEnv = ..

-- This is essentially a 'Controller'.
authenticate :: ServerEnvironment -> Handler User
authenticate conn = ...

main :: IO ()
main = do
  env <- setupEnv
  -- Now, because we have access to the env, we can turn our
  -- 'authenticate' into the right type before putting it
  -- in the context
  let ctx = authenticate env :. EmptyContext
  run 8080 $ serveWithContext myAPI (server conn) ctx