Question

我试图弄清楚如何使用Esqueleto

编写以下查询

SELECT COUNT("person"."id")
FROM "person"
WHERE (("person"."admin" = 't' OR "person"."vip" = 't') // 't' as in True
      OR "person"."karma" >= 5000 AND "person"."hellbanned" = 'f')

这里是我的模型的定义方式

Person
    admin Bool
    vip Bool
    karma Int
    hellbanned Bool

除了COUNT部分

之外，我设法得到了几乎所有东西

select $
  from $ \p -> do
    where_
      ((p ^. PersonAdmin ==. val True) ||. (p ^. PersonVip ==. val True)
      &&. (p ^. PersonKarma >=. val 5000) &&. (p ^. PersonHellbanned ==. val False))
    return $ p ^. PersonId

我设法找到了countRows函数，但是我还没有设法找到一种方法来将这两者结合起来，以便进行类型检查。

我也不确定在where子句的每个分支中是否需要所有p ^.，或者是否可以某种方式将它们折叠在一起？

Answer 1

这里有一些我认为有用的旧代码，我不记得很多，但希望它有所帮助！

selectCount
  :: (From SqlQuery SqlExpr SqlBackend a)
  => (a -> SqlQuery ()) -> Persist Int
selectCount q = do
  res <- select $ from $ (\x -> q x >> return countRows)
  return $ fromMaybe 0 $ (\(Value a) -> a) <$> headMay res

getCount :: RepositoryUri -> Persist Int
getCount ruri =
  selectCount $ \(r `InnerJoin` rs) -> do
    on     $ r ^. R.RepositoryId  ==. rs ^. Repo
    where_ $ r ^. R.Uri ==. val ruri

Answer 2

我发现Adam Bergmark的答案非常有用，但我认为应该提供更多信息：

import Import hiding ( Value )
import Data.Maybe ( maybeToList
                  , listToMaybe )
import Database.Esqueleto
import Database.Esqueleto.Internal.Language (From)

selectCount
  :: (From SqlQuery SqlExpr SqlBackend a)
  => (a -> SqlQuery ()) -> YesodDB App Int
selectCount q = do
  res <- select $ from $ (\x -> q x >> return countRows)
  return $ fromMaybe 0 . listToMaybe . fmap (\(Value v) -> v) $ res

在Esqueleto和Yesod中执行COUNT（id）的正确方法是什么

2 个答案: