general/mongo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add UpdateSet, Relaxed in client, Fix tests

Browse Source
This commit is contained in:
George Suntres
2026-04-22 10:22:23 -04:00
parent 45f9ac558f
commit 188c5a1be1
9 changed files with 163 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
discrimination.go
View File

@@ -6,6 +6,8 @@ import (
"context"
"go.mongodb.org/mongo-driver/v2/bson"
"git.gsuntres.com/general/commons"
)
func (c *MongoClient) DiscriminatorCheckAndApplyToData(ctx context.Context, name string, data map[string]any) error {
@@ -31,6 +33,24 @@ func (c *MongoClient) DiscriminatorCheckAndApplyToData(ctx context.Context, name
return nil
}
func (c *MongoClient) DiscriminatorOmitInData(name string, data bson.M) error {
cdef, ok := c.Registry[name]
if ok && cdef.Discriminator != nil {
if data == nil {
data = map[string]any{}
}
log.Printf("Making sure discriminator is not in data for %s", name)
_, ok := data[cdef.Discriminator.Field]
if ok {
delete(data, cdef.Discriminator.Field)
}
}
return nil
}
func (c *MongoClient) DiscriminatorCheckAndApplyToFilter(ctx context.Context, name string, filter bson.M) error {
cdef, ok := c.Registry[name]
if ok && cdef.Discriminator != nil {
@@ -48,8 +68,13 @@ func (c *MongoClient) DiscriminatorCheckAndApplyToFilter(ctx context.Context, na
// update payload
v := vAny.(string)
filter[cdef.Discriminator.Field] = bson.M{"$eq": v}
if commons.StringIsBlank(v) {
return fmt.Errorf("discriminator field required for %s", name)
}
filter[cdef.Discriminator.Field] = bson.M{"eq": v}
}
return nil
}