Replace IORefs with interruptibleFor

This commit is contained in:
Victor Denisov 2016-11-01 22:53:27 -07:00
parent d9db9bca52
commit c6bd17f1f4

View file

@ -651,19 +651,11 @@ update' ordered col updateDocs = do
updates
let lens = map length chunks
let lSums = 0 : (zipWith (+) lSums lens)
errorDetected <- liftIO $ newIORef False
ctx <- ask
blocks <- forM (zip lSums chunks) $ \b -> liftIO $ do
ed <- readIORef errorDetected
if ed && ordered
then return $ UpdateResult True 0 Nothing [] [] []
else do
blocks <- liftIO $ interruptibleFor ordered (zip lSums chunks) $ \b -> liftIO $ do
ur <- runReaderT (updateBlock ordered col b) ctx
when (failed ur) $ do
writeIORef errorDetected True
return ur
`catch` \(e :: SomeException) -> do
writeIORef errorDetected True
return $ UpdateResult True 0 Nothing [] [] [] -- TODO probably should be revised
let failedTotal = or $ map failed blocks
let updatedTotal = sum $ map nMatched blocks
@ -720,14 +712,8 @@ updateBlockLegacy :: (MonadIO m)
updateBlockLegacy ordered col (prevCount, docs) = do
db <- thisDatabase
ctx <- ask
errorDetected <- liftIO $ newIORef False
results <-
liftIO $ forM (zip [prevCount, (prevCount + 1) ..] docs) $ \(i, updateDoc) -> do
ed <- readIORef errorDetected
if ed && ordered
then do
return $ UpdateResult True 0 Nothing [] [] []
else do
results <- liftIO $
interruptibleFor ordered (zip [prevCount, (prevCount + 1) ..] docs) $ \(i, updateDoc) -> do
let doc = (at "u" updateDoc) :: Document
let sel = (at "q" updateDoc) :: Document
let upsrt = if at "upsert" updateDoc then [Upsert] else []
@ -752,13 +738,10 @@ updateBlockLegacy ordered col (prevCount, docs) = do
let c = fromMaybe defaultCode eCode
if wtimeout
then do
writeIORef errorDetected True
return $ UpdateResult True 0 Nothing [] [] [WriteConcernError c errV]
else do
writeIORef errorDetected True
return $ UpdateResult True 0 Nothing [] [WriteError i c errV] []
`catch` \(e :: SomeException) -> do
writeIORef errorDetected True
return $ UpdateResult True 0 Nothing [] [WriteError i 0 (show e)] []
return $ foldl1' mergeUpdateResults results