mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-06-12 16:53:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support elements to a pipeline

Browse Source
This commit is contained in:
Nikita Tokarchuk
2019-05-16 11:07:01 +02:00
parent b7e92a5055
commit a86455df2d
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mongox/common/common.go
+17 -5
View File
@@ -30,8 +30,20 @@ func createAggregateLoad(db *mongox.Database, target interface{}, composed *quer
collection := db.GetCollectionOf(target) collection := db.GetCollectionOf(target)
opts := options.Aggregate() opts := options.Aggregate()
pipelineHead := primitive.A{primitive.M{"$match": composed.M()}} pipeline := primitive.A{}
pipelineTail := primitive.A{}
if !composed.Empty() {
pipeline = append(pipeline, primitive.M{"$match": primitive.M{"$expr": composed.M()}})
}
if composed.Sorter() != nil {
pipeline = append(pipeline, primitive.M{"$sort": composed.Sorter()})
}
if composed.Skipper() != nil {
pipeline = append(pipeline, primitive.M{"$skip": *composed.Skipper()})
}
if composed.Limiter() != nil {
pipeline = append(pipeline, primitive.M{"$limit": *composed.Limiter()})
}
el := reflect.ValueOf(target).Elem() el := reflect.ValueOf(target).Elem()
elType := el.Type() elType := el.Type()
@@ -114,7 +126,7 @@ func createAggregateLoad(db *mongox.Database, target interface{}, composed *quer
lookupPipeline = append(lookupPipeline, primitive.M{"$limit": 1}) lookupPipeline = append(lookupPipeline, primitive.M{"$limit": 1})
} }
pipelineTail = append(pipelineTail, primitive.M{ pipeline = append(pipeline, primitive.M{
"$lookup": primitive.M{ "$lookup": primitive.M{
"from": lookupCollection.Name(), "from": lookupCollection.Name(),
"let": lookupVars, "let": lookupVars,
@@ -127,7 +139,7 @@ func createAggregateLoad(db *mongox.Database, target interface{}, composed *quer
continue continue
} }
pipelineTail = append(pipelineTail, primitive.M{ pipeline = append(pipeline, primitive.M{
"$unwind": primitive.M{ "$unwind": primitive.M{
"preserveNullAndEmptyArrays": true, "preserveNullAndEmptyArrays": true,
"path": "$" + jsonName, "path": "$" + jsonName,
@@ -136,5 +148,5 @@ func createAggregateLoad(db *mongox.Database, target interface{}, composed *quer
} }
} }
return collection.Aggregate(db.Context(), append(pipelineHead, pipelineTail...), opts) return collection.Aggregate(db.Context(), pipeline, opts)
} }