Add timeseries, discriminator

insert.go

@@ -2,18 +2,10 @@ package mongo
 
 import (
 	"context"
-	"time"
-	"fmt"
-
+	
 	"go.mongodb.org/mongo-driver/v2/bson"
-	"github.com/matoous/go-nanoid/v2"
-
-	// "git.gsuntres.com/boxtep/boxtep/core"
 )
 
-const alphabet = "0123456789abcdefghijclmnopqrstuvwxyz"
-
-const maxLen = 10
 
 // InsertOneWithStruct can be used to insert defined structs.
 func (c *MongoClient) InsertOneFromStruct(ctx context.Context, database, name string, data any) (bson.M, error) {
@@ -31,6 +23,10 @@ func (c *MongoClient) InsertOne(ctx context.Context, database, name string, data
 
 	prepareForInsert(data, c.GetIdPrefix(name))
 
+	if err := c.DiscriminatorCheckAndApplyToData(ctx, name, data); err != nil {
+		return nil, err
+	}
+
 	if _, err := collection.InsertOne(ctx, data); err != nil {
 		return nil, err
 	}
@@ -47,37 +43,4 @@ func prepareForInsert(data bson.M, idPrefix string) {
 	ensureUpdatedAt(data)
 }
 
-// ensureId adds the id property when missing or when it's an empty string.
-func ensureId(data bson.M, idPrefix string) string {
-	maybeId, hasId := data["_id"]
 
-	var id, finalId string
-	if !hasId || maybeId == "" {
-		id, _ = gonanoid.Generate(alphabet, maxLen)
-		if idPrefix != "" {
-			finalId = fmt.Sprintf("%s_%s", idPrefix, id)
-		} else {
-			finalId = id
-		}
-
-		data["_id"] = finalId
-	}
-
-	return finalId
-}
-
-func ensureCreatedAt(data bson.M) time.Time {
-	now := time.Now().UTC().Truncate(time.Millisecond)
-
-	data["created_at"] = now
-
-	return now
-}
-
-func ensureUpdatedAt(data bson.M) time.Time {
-	now := time.Now().UTC().Truncate(time.Millisecond)
-
-	data["updated_at"] = now
-
-	return now
-}