Add allowDiskUse option for aggregates

Merge pull request #120 from acondolu/allow-disk-use
This commit is contained in:
Victor Denisov 2021-01-03 20:53:11 -08:00 committed by GitHub
commit 952aed5a1f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1338,16 +1338,28 @@ aggregate :: (MonadIO m, MonadFail m) => Collection -> Pipeline -> Action m [Doc
aggregate aColl agg = do aggregate aColl agg = do
aggregateCursor aColl agg def >>= rest aggregateCursor aColl agg def >>= rest
data AggregateConfig = AggregateConfig {} data AggregateConfig = AggregateConfig
deriving Show { allowDiskUse :: Bool -- ^ Enable writing to temporary files (aggregations have a 100Mb RAM limit)
}
deriving Show
instance Default AggregateConfig where instance Default AggregateConfig where
def = AggregateConfig {} def = AggregateConfig
{ allowDiskUse = False
}
aggregateCommand :: Collection -> Pipeline -> AggregateConfig -> Document
aggregateCommand aColl agg AggregateConfig {..} =
[ "aggregate" =: aColl
, "pipeline" =: agg
, "cursor" =: ([] :: Document)
, "allowDiskUse" =: allowDiskUse
]
aggregateCursor :: (MonadIO m, MonadFail m) => Collection -> Pipeline -> AggregateConfig -> Action m Cursor aggregateCursor :: (MonadIO m, MonadFail m) => Collection -> Pipeline -> AggregateConfig -> Action m Cursor
-- ^ Runs an aggregate and unpacks the result. See <http://docs.mongodb.org/manual/core/aggregation/> for details. -- ^ Runs an aggregate and unpacks the result. See <http://docs.mongodb.org/manual/core/aggregation/> for details.
aggregateCursor aColl agg _ = do aggregateCursor aColl agg cfg = do
response <- runCommand ["aggregate" =: aColl, "pipeline" =: agg, "cursor" =: ([] :: Document)] response <- runCommand (aggregateCommand aColl agg cfg)
getCursorFromResponse aColl response getCursorFromResponse aColl response
>>= either (liftIO . throwIO . AggregateFailure) return >>= either (liftIO . throwIO . AggregateFailure) return