viktor
852468f976
# Conflicts: # exerr/constructor.go # exerr/dataCategory.go # exerr/dataSeverity.go # exerr/dataType.go # exerr/exerr.go # go.mod # mongoext/registry.go # reflectext/primStrSer.go # rfctime/date.go # rfctime/rfc3339.go # rfctime/rfc3339Nano.go # rfctime/seconds.go # rfctime/unix.go # rfctime/unixMilli.go # rfctime/unixNano.go # wmo/collection.go # wmo/queryInsert.go
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
package wmo
|
|
|
|
import (
|
|
"context"
|
|
"git.blackforestbytes.com/BlackForestBytes/goext/exerr"
|
|
"git.blackforestbytes.com/BlackForestBytes/goext/langext"
|
|
"go.mongodb.org/mongo-driver/v2/mongo"
|
|
"go.mongodb.org/mongo-driver/v2/mongo/options"
|
|
)
|
|
|
|
func (c *Coll[TData]) Aggregate(ctx context.Context, pipeline mongo.Pipeline, opts ...options.Lister[options.AggregateOptions]) ([]TData, error) {
|
|
|
|
for _, ppl := range c.extraModPipeline {
|
|
pipeline = langext.ArrConcat(pipeline, ppl(ctx))
|
|
}
|
|
|
|
cursor, err := c.coll.Aggregate(ctx, pipeline, opts...)
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "mongo-aggregation failed").Any("pipeline", pipeline).Any("options", opts).Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
defer func() { _ = cursor.Close(ctx) }()
|
|
|
|
res, err := c.decodeAll(ctx, cursor)
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "failed to decode values").Build()
|
|
}
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (c *Coll[TData]) AggregateOneOpt(ctx context.Context, pipeline mongo.Pipeline, opts ...options.Lister[options.AggregateOptions]) (*TData, error) {
|
|
|
|
for _, ppl := range c.extraModPipeline {
|
|
pipeline = langext.ArrConcat(pipeline, ppl(ctx))
|
|
}
|
|
|
|
cursor, err := c.coll.Aggregate(ctx, pipeline, opts...)
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "mongo-aggregation failed").Any("pipeline", pipeline).Any("options", opts).Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
defer func() { _ = cursor.Close(ctx) }()
|
|
|
|
if cursor.Next(ctx) {
|
|
v, err := c.decodeSingle(ctx, cursor)
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "failed to decode single value").Build()
|
|
}
|
|
return &v, nil
|
|
}
|
|
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *Coll[TData]) AggregateOne(ctx context.Context, pipeline mongo.Pipeline, opts ...options.Lister[options.AggregateOptions]) (TData, error) {
|
|
|
|
for _, ppl := range c.extraModPipeline {
|
|
pipeline = langext.ArrConcat(pipeline, ppl(ctx))
|
|
}
|
|
|
|
cursor, err := c.coll.Aggregate(ctx, pipeline, opts...)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "mongo-aggregation failed").Any("pipeline", pipeline).Any("options", opts).Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
defer func() { _ = cursor.Close(ctx) }()
|
|
|
|
if cursor.Next(ctx) {
|
|
v, err := c.decodeSingle(ctx, cursor)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "failed to decode single value").Build()
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
return *new(TData), exerr.Wrap(mongo.ErrNoDocuments, "no document in result").Build()
|
|
}
|