Update README and refactor MongoDB connection to use configurable URI

README.md

@@ -113,7 +113,11 @@ Mount data volume to `/usr/src/app/data/`.
 
 The module processes logs and returns them by request.
 
-Mount data volumes to `/usr/src/app/data/auth/`, `/usr/src/app/data/db/` and `/usr/src/app/data/source/`.
+Mount data volumes to `/usr/src/app/data/auth/` and `/usr/src/app/data/source/`.
+
+|Var                    |Description
+|-----------------------|-----------
+|MONGODB_URI            |MongoDB connection string
 
 ## Release
 

logs_import/db/db.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 
 	"github.com/abaldeweg/warehouse-server/logs_import/entity"
+	"github.com/spf13/viper"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
@@ -18,7 +19,7 @@ type DBHandler struct {
 
 // NewDBHandler creates a new DBHandler.
 func NewDBHandler() (*DBHandler, error) {
-	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")
+	clientOptions := options.Client().ApplyURI(viper.Get("MONGODB_URI").(string))
 	client, err := mongo.Connect(context.TODO(), clientOptions)
 	if err != nil {
 		return nil, err

logs_import/main.go

@@ -3,10 +3,16 @@ package main
 import (
 	"time"
 
+	"github.com/abaldeweg/warehouse-server/framework/config"
 	"github.com/abaldeweg/warehouse-server/logs_import/cmd"
+	"github.com/spf13/viper"
 )
 
 func main() {
+	config.LoadAppConfig()
+
+	viper.SetDefault("MONGODB_URI", "mongodb://localhost:27017")
+
 	go cmd.Execute()
 
 	for {

logs_web/db/db.go

@@ -6,20 +6,21 @@ import (
 	"time"
 
 	"github.com/abaldeweg/warehouse-server/logs_web/entity"
+	"github.com/spf13/viper"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
 // DBHandler handles database operations for logs.
 type DBHandler struct {
-	client *mongo.Client
+	client     *mongo.Client
 	collection *mongo.Collection
 }
 
 // NewDBHandler creates a new DBHandler.
 func NewDBHandler() (*DBHandler, error) {
-	clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")
+	clientOptions := options.Client().ApplyURI(viper.Get("MONGODB_URI").(string))
 	client, err := mongo.Connect(context.TODO(), clientOptions)
 	if err != nil {
 		return nil, err

logs_web/main.go

@@ -3,10 +3,16 @@ package main
 import (
 	"log"
 
+	"github.com/abaldeweg/warehouse-server/framework/config"
 	"github.com/abaldeweg/warehouse-server/logs_web/router"
+	"github.com/spf13/viper"
 )
 
 func main() {
+	config.LoadAppConfig()
+
+	viper.SetDefault("MONGODB_URI", "mongodb://localhost:27017")
+
 	r := router.Routes()
 	log.Fatal(r.Run(":8080"))
 }