Haskell 仆人将自定义数据传递给授权处理程序

Question

我正在使用自定义 monad（使用 reader）轻松地将数据库池等数据传递给我的处理程序（在使用自定义 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

Haskell 仆人将自定义数据传递给授权处理程序

Haskell Servant passing custom data to auth handler

database

authentication

haskell

pool

servant