Mike Schwörer
fe4cdc48af
Some checks failed
Build Docker and Deploy / Run goext test-suite (push) Failing after 25s
66 lines
2.3 KiB
Go
66 lines
2.3 KiB
Go
package wmo
|
|
|
|
import (
|
|
"context"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
|
)
|
|
|
|
func (c *Coll[TData]) InsertOne(ctx context.Context, valueIn TData) (TData, error) {
|
|
for _, hook := range c.marshalHooks {
|
|
valueIn = hook(valueIn)
|
|
}
|
|
|
|
insRes, err := c.coll.InsertOne(ctx, valueIn)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "mongo-query[insert-one] failed").Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
r, err := c.findOneInternal(ctx, bson.M{"_id": insRes.InsertedID}, false)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "mongo-query[insert-one] failed").Str("collection", c.Name()).Build()
|
|
}
|
|
return *r, nil
|
|
}
|
|
|
|
// InsertOneUnchecked behaves the same as InsertOne, but allows arbitrary data to be inserted (valueIn is any instead of TData)
|
|
func (c *Coll[TData]) InsertOneUnchecked(ctx context.Context, valueIn any) (TData, error) {
|
|
insRes, err := c.coll.InsertOne(ctx, valueIn)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "mongo-query[insert-one] failed").Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
r, err := c.findOneInternal(ctx, bson.M{"_id": insRes.InsertedID}, false)
|
|
if err != nil {
|
|
return *new(TData), exerr.Wrap(err, "mongo-query[insert-one] failed").Str("collection", c.Name()).Build()
|
|
}
|
|
return *r, nil
|
|
}
|
|
|
|
func (c *Coll[TData]) InsertMany(ctx context.Context, valueIn []TData) (*mongo.InsertManyResult, error) {
|
|
for _, hook := range c.marshalHooks {
|
|
for i := 0; i < len(valueIn); i++ {
|
|
valueIn[i] = hook(valueIn[i])
|
|
}
|
|
}
|
|
|
|
insRes, err := c.coll.InsertMany(ctx, langext.ArrayToInterface(valueIn))
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "mongo-query[insert-many] failed").Int("len(valueIn)", len(valueIn)).Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
return insRes, nil
|
|
}
|
|
|
|
// InsertManyUnchecked behaves the same as InsertOne, but allows arbitrary data to be inserted (valueIn is []any instead of []TData)
|
|
func (c *Coll[TData]) InsertManyUnchecked(ctx context.Context, valueIn []any) (*mongo.InsertManyResult, error) {
|
|
insRes, err := c.coll.InsertMany(ctx, langext.ArrayToInterface(valueIn))
|
|
if err != nil {
|
|
return nil, exerr.Wrap(err, "mongo-query[insert-many] failed").Int("len(valueIn)", len(valueIn)).Str("collection", c.Name()).Build()
|
|
}
|
|
|
|
return insRes, nil
|
|
}
|