From 1142585f843bbd70eb7de31c9cbc4e9095610a46 Mon Sep 17 00:00:00 2001
From: George Suntres <gsuntres@pm.me>
Date: Mon, 27 Apr 2026 13:54:51 -0400
Subject: [PATCH] Add AutoCleanup in configuration

---
 main.go | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index c095922..272e623 100644
--- a/main.go
+++ b/main.go
@@ -315,6 +315,7 @@ type MongoStartProps struct {
 	MongoDBPrefix string
 	ContextFields []string
 	WithAudit bool
+	AutoCleanup bool
 	OnAudit *OnAudit
 	AllowTruncatingDoubles bool
 }
@@ -410,11 +411,13 @@ func Start(props *MongoStartProps) error {
 		return err
 	}
 
-	sys.OnExit(func () {
-		if err := client.Client.Disconnect(context.TODO()); err != nil {
-			log.Fatalf("failed to disconnect: %v", err)
-		}
-	})
+	if props.AutoCleanup {
+		sys.OnExit(func () {
+			if err := Stop(); err != nil {
+				log.Fatalf("failed to disconnect: %v", err)
+			}
+		})	
+	}
 
 	// Register defaults.go
 	var structfulDef bson.M