feat: insertOne() now returns the document with its ID

.gitignore

@@ -22,3 +22,4 @@ go.work
 
 # MacOS
 .DS_Store
+.vscode/launch.json

crud.go

@@ -4,6 +4,7 @@ import (
 	"context"
 
 	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
@@ -12,10 +13,13 @@ import (
 func (r *Repo[T]) InsertOne(ctx context.Context, doc T, opts ...*options.InsertOneOptions) (T, error) {
 	doc.BeforeInsert(ctx)
 	defer doc.AfterInsert(ctx)
-	_, err := r.collection.InsertOne(ctx, doc, opts...)
+	res, err := r.collection.InsertOne(ctx, doc, opts...)
 	if err != nil {
 		return *new(T), err
 	}
+	if id, ok := res.InsertedID.(primitive.ObjectID); ok {
+		doc.SetID(id)
+	}
 	return doc, nil
 }
 

crud_test.go

@@ -24,6 +24,7 @@ func TestRepo_InsertOne(t *testing.T) {
 	var doc TestUser
 	err = db.Collection(testColl).FindOne(ctx, bson.M{"age": 6}).Decode(&doc)
 	require.NoError(t, err)
+	require.NotNil(t, doc.ID)
 	require.NotNil(t, doc.CreatedAt)
 	require.NotNil(t, doc.UpdatedAt)
 }

document.go

@@ -3,12 +3,14 @@ package modm
 import (
 	"context"
 
+	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 // Document represents an interface for common document operations.
 type Document interface {
+	SetID(id primitive.ObjectID)
 	BeforeInsert(ctx context.Context)
 	AfterInsert(ctx context.Context)
 	BeforeUpdate(ctx context.Context)

field.go

@@ -15,6 +15,10 @@ type DefaultField struct {
 	UpdatedAt time.Time          `bson:"updated_at,omitempty" json:"updated_at"`
 }
 
+func (df *DefaultField) SetID(id primitive.ObjectID) {
+	df.ID = id
+}
+
 // DefaultUpdatedAt sets the default value for the UpdatedAt field.
 func (df *DefaultField) DefaultUpdatedAt() {
 	df.UpdatedAt = time.Now()