database - 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