diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt index d13b4325..7c21b6bc 100644 --- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt +++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt @@ -22,10 +22,7 @@ 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.stages.HasUnionWithCompatibility -import opensavvy.ktmongo.dsl.aggregation.stages.ProjectStageOperators -import opensavvy.ktmongo.dsl.aggregation.stages.SetStageOperators -import opensavvy.ktmongo.dsl.aggregation.stages.UnsetStageOperators +import opensavvy.ktmongo.dsl.aggregation.stages.* import opensavvy.ktmongo.dsl.options.SortOptionDsl import opensavvy.ktmongo.dsl.path.Field import opensavvy.ktmongo.dsl.query.FilterQuery @@ -70,6 +67,8 @@ interface CoroutineMongoAggregationPipeline : MongoAggregationPi override fun unionWith(other: HasUnionWithCompatibility): CoroutineMongoAggregationPipeline + override fun lookup(block: LookupStageOperators.() -> Unit): CoroutineMongoAggregationPipeline + override fun group(block: AccumulationOperators.() -> Unit): CoroutineMongoAggregationPipeline override fun countTo(field: Field): CoroutineMongoAggregationPipeline diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt index 50757f44..62a0d175 100644 --- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt +++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt @@ -29,10 +29,7 @@ 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.stages.HasUnionWithCompatibility -import opensavvy.ktmongo.dsl.aggregation.stages.ProjectStageOperators -import opensavvy.ktmongo.dsl.aggregation.stages.SetStageOperators -import opensavvy.ktmongo.dsl.aggregation.stages.UnsetStageOperators +import opensavvy.ktmongo.dsl.aggregation.stages.* import opensavvy.ktmongo.dsl.options.SortOptionDsl import opensavvy.ktmongo.dsl.path.Field import opensavvy.ktmongo.dsl.query.FilterQuery @@ -120,6 +117,10 @@ private class CoroutineMongoAggregationPipelineImpl @OptIn(LowLe override fun unionWith(other: HasUnionWithCompatibility): CoroutineMongoAggregationPipelineImpl = super.unionWith(other) as CoroutineMongoAggregationPipelineImpl + @KtMongoDsl + override fun lookup(block: LookupStageOperators.() -> Unit): CoroutineMongoAggregationPipelineImpl = + super.lookup(block) as CoroutineMongoAggregationPipelineImpl + @KtMongoDsl override fun group(block: AccumulationOperators.() -> Unit): CoroutineMongoAggregationPipelineImpl = super.group(block) as CoroutineMongoAggregationPipelineImpl