Was ist eine richtige Art und Weise COUNT tun (id) in Esqueleto und Yesod

Question

Ich versuche, herauszufinden, wie die folgende Abfrage schreiben Esqueleto mit

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') 

Hier ist, wie mein Modell definiert ist

Person 
    admin Bool 
    vip Bool 
    karma Int 
    hellbanned Bool 

fast alles zu bekommen, mit Ausnahme der COUNT Teil

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

I manag

ich habe es geschafft ed, um eine countRows Funktion zu finden, jedoch habe ich es nicht geschafft, einen Weg zu finden, diese beiden auf eine Art zu kombinieren, die checket.

Ich bin mir auch nicht sicher, ob ich all diese p ^. in jedem Zweig der Where-Klausel brauchen, oder wenn diese irgendwie zusammen kollabiert werden können?

Answer 1

Hier einige alte Code, den ich habe, dass sich ein Graf, ich erinnere mich nicht viel darüber, aber hoffentlich hilft es!

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

Ich habe Adam Bergmark des gefundene Antwort ganz nützlich, aber ich denke, es sollte mehr informativ sein:

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