diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt index 7c21b6bc..a26eb8bf 100644 --- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt +++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt @@ -22,6 +22,8 @@ package opensavvy.ktmongo.coroutines import opensavvy.ktmongo.api.MongoAggregationPipeline import opensavvy.ktmongo.dsl.LowLevelApi import opensavvy.ktmongo.dsl.aggregation.AccumulationOperators +import opensavvy.ktmongo.dsl.aggregation.AggregationOperators +import opensavvy.ktmongo.dsl.aggregation.Value import opensavvy.ktmongo.dsl.aggregation.stages.* import opensavvy.ktmongo.dsl.options.SortOptionDsl import opensavvy.ktmongo.dsl.path.Field @@ -51,6 +53,8 @@ interface CoroutineMongoAggregationPipeline : MongoAggregationPi override fun match(filter: FilterQuery.() -> Unit): CoroutineMongoAggregationPipeline + override fun matchExpr(filter: AggregationOperators.() -> Value): CoroutineMongoAggregationPipeline + override fun sample(size: Int): CoroutineMongoAggregationPipeline override fun set(block: SetStageOperators.() -> Unit): CoroutineMongoAggregationPipeline diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt index 62a0d175..f13a651f 100644 --- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt +++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt @@ -25,10 +25,7 @@ import opensavvy.ktmongo.dsl.BsonContext import opensavvy.ktmongo.dsl.DangerousMongoApi import opensavvy.ktmongo.dsl.KtMongoDsl import opensavvy.ktmongo.dsl.LowLevelApi -import opensavvy.ktmongo.dsl.aggregation.AbstractPipeline -import opensavvy.ktmongo.dsl.aggregation.AccumulationOperators -import opensavvy.ktmongo.dsl.aggregation.AggregationPipeline -import opensavvy.ktmongo.dsl.aggregation.PipelineChainLink +import opensavvy.ktmongo.dsl.aggregation.* import opensavvy.ktmongo.dsl.aggregation.stages.* import opensavvy.ktmongo.dsl.options.SortOptionDsl import opensavvy.ktmongo.dsl.path.Field @@ -85,6 +82,10 @@ private class CoroutineMongoAggregationPipelineImpl @OptIn(LowLe override fun match(filter: FilterQuery.() -> Unit): CoroutineMongoAggregationPipelineImpl = super.match(filter) as CoroutineMongoAggregationPipelineImpl + @KtMongoDsl + override fun matchExpr(filter: AggregationOperators.() -> Value): CoroutineMongoAggregationPipelineImpl = + super.matchExpr(filter) as CoroutineMongoAggregationPipelineImpl + @KtMongoDsl override fun sample(size: Int): CoroutineMongoAggregationPipelineImpl = super.sample(size) as CoroutineMongoAggregationPipelineImpl