From 3f43e15cc2e82e5d96fcda8491a037f9ba77d5ff Mon Sep 17 00:00:00 2001
From: Cedric Verstraeten <cedric@verstraeten.io>
Date: Mon, 21 Oct 2024 20:41:31 +0200
Subject: [PATCH] fix database client

---
 machinery/src/config/main.go   | 2 +-
 machinery/src/database/main.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/machinery/src/config/main.go b/machinery/src/config/main.go
index 41041fa..a6caf8f 100644
--- a/machinery/src/config/main.go
+++ b/machinery/src/config/main.go
@@ -59,7 +59,7 @@ func OpenConfig(configDirectory string, configuration *models.Configuration) {
 		// Write to mongodb
 		client := database.New()
 
-		db := client.Database(database.DatabaseName)
+		db := client.Client.Database(database.DatabaseName)
 		collection := db.Collection("configuration")
 
 		var globalConfig models.Config
diff --git a/machinery/src/database/main.go b/machinery/src/database/main.go
index c9bf1ff..f59fef7 100644
--- a/machinery/src/database/main.go
+++ b/machinery/src/database/main.go
@@ -19,7 +19,7 @@ var TIMEOUT = 10 * time.Second
 var _init_ctx sync.Once
 var _instance *DB
 
-var DatabaseName = os.Getenv("MONGODB_DATABASE_STORAGE")
+var DatabaseName = os.Getenv("MONGODB_DATABASE_FACTORY")
 
 func New() *DB {