Add timeseries, discriminator

discrimination.go

@@ -0,0 +1,47 @@
+package mongo
+
+import (
+	"log"
+	"fmt"
+	"context"
+
+	"go.mongodb.org/mongo-driver/v2/bson"
+)
+
+func (c *MongoClient) DiscriminatorCheckAndApplyToData(ctx context.Context, name string, data map[string]any) error {
+	cdef, ok := c.Registry[name]
+	if ok && cdef.Discriminator != nil {
+		log.Printf("Discriminator found for %s; will use it", name)
+
+		// get from context
+		vAny := ctx.Value(cdef.Discriminator.CtxField)
+		if vAny == nil {
+			return fmt.Errorf("discriminator field required for %s", name)
+		}
+
+		// update payload
+		v := vAny.(string)
+		data[cdef.Discriminator.Field] = v
+	}
+
+	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 {
+		log.Printf("Discriminator found for %s; will use it", name)
+
+		// get from context
+		vAny := ctx.Value(cdef.Discriminator.CtxField)
+		if vAny == nil {
+			return fmt.Errorf("discriminator field required for %s", name)
+		}
+
+		// update payload
+		v := vAny.(string)
+		filter[cdef.Discriminator.Field] = bson.M{"$eq": v}
+	}
+
+	return nil
+}