Vergara_Tech/mongodb
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

mongodb returns count as a double: fix count function

Browse source
This commit is contained in:
Scott R. Parish 2010-03-09 15:18:11 -06:00
parent c1b8a21366
commit cbc5128d47
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
Database/MongoDB.hs
View file

@ -468,16 +468,17 @@ fromUpdateFlags flags = List.foldl (.|.) 0 $
flip fmap flags $ (1 `shiftL`) . fromEnum flip fmap flags $ (1 `shiftL`) . fromEnum
-- | Return the number of documents in /FullCollection/. -- | Return the number of documents in /FullCollection/.
count :: Connection -> FullCollection -> IO Int64 count :: Connection -> FullCollection -> IO Integer
count c col = countMatching c col empty count c col = countMatching c col empty
-- | Return the number of documents in /FullCollection/ matching /Selector/ -- | Return the number of documents in /FullCollection/ matching /Selector/
countMatching :: Connection -> FullCollection -> Selector -> IO Int64 countMatching :: Connection -> FullCollection -> Selector -> IO Integer
countMatching c col sel = do countMatching c col sel = do
let (db, col') = splitFullCol col let (db, col') = splitFullCol col
res <- runCommand c db $ toBsonDoc [("count", toBson col'), res <- runCommand c db $ toBsonDoc [("count", toBson col'),
("query", toBson sel)] ("query", toBson sel)]
return $ fromBson $ fromLookup $ List.lookup (s2L "n") res let cnt = (fromBson $ fromLookup $ List.lookup (s2L "n") res :: Double)
return $ truncate $ cnt
-- | Delete documents matching /Selector/ from the given /FullCollection/. -- | Delete documents matching /Selector/ from the given /FullCollection/.
delete :: Connection -> FullCollection -> Selector -> IO RequestID delete :: Connection -> FullCollection -> Selector -> IO RequestID