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

Implement insert using command mechanism

Browse source
This commit is contained in:
Victor Denisov 2016-05-22 17:38:07 -07:00
parent 726109588b
commit d123460b40
1 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
Database/MongoDB/Query.hs
View file

@ -412,8 +412,28 @@ insert' :: (MonadIO m) => [InsertOption] -> Collection -> [Document] -> Action m
insert' opts col docs = do insert' opts col docs = do
db <- thisDatabase db <- thisDatabase
docs' <- liftIO $ mapM assignId docs docs' <- liftIO $ mapM assignId docs
p <- asks mongoPipe
let sd = P.serverData p
if (maxWireVersion sd < 2)
then do
write (Insert (db <.> col) opts docs') write (Insert (db <.> col) opts docs')
return $ map (valueAt "_id") docs' return $ map (valueAt "_id") docs'
else do
doc <- runCommand $
[ "insert" =: col
, "ordered" =: (KeepGoing `notElem` opts)
, "documents" =: docs'
]
liftIO $ putStrLn $ show doc
case (look "writeErrors" doc, look "writeConcernError" doc) of
(Nothing, Nothing) -> return $ map (valueAt "_id") docs'
(Just err, Nothing) -> do
liftIO $ throwIO $ WriteFailure (maybe 0 id $ lookup "ok" doc) (show err)
(Nothing, Just err) -> do
liftIO $ throwIO $ WriteFailure (maybe 0 id $ lookup "ok" doc) (show err)
(Just err, Just writeConcernErr) -> do
liftIO $ throwIO $ WriteFailure (maybe 0 id $ lookup "ok" doc) (show err ++ show writeConcernErr)
assignId :: Document -> IO Document assignId :: Document -> IO Document
-- ^ Assign a unique value to _id field if missing -- ^ Assign a unique value to _id field if missing