timetable-mgr/src/dbAccess/client.go

package dbAccess

import (
	"git.fjla.uk/owlboard/mq-client/helpers"

	"context"
	"fmt"
	"os"
	"time"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	// "go.mongodb.org/mongo-driver/mongo/readpref"
)

// Generate DB Url
var dbUri string = getDbUri()

func getDbUri() string {
	var dbHost string = os.Getenv("OWL_DB_HOST")
	if dbHost == "" {
		dbHost = "localhost"
	}
	var dbPort string = os.Getenv("OWL_DB_PORT")
	if dbPort == "" {
		dbPort = "27017"
	}
	var dbUser string = os.Getenv("OWL_DB_USER")
	if dbUser == "" {
		dbUser = "owl"
	}
	var dbPass string = os.Getenv("OWL_DB_PASS")
	if dbPass == "" {
		dbPass = "twittwoo"
	}
	var uri = "mongodb://" + dbUser + ":" + dbPass + "@" + dbHost + ":" + dbPort
	return uri
}

// Provide the DB Connection to other functions
var MongoClient (*mongo.Client) = initDataAccess()

// Initialise the DB Connection
func initDataAccess() *mongo.Client {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	client, err := mongo.Connect(ctx, options.Client().ApplyURI(dbUri))
	if err != nil {
		fmt.Println(err)
		helpers.Log("Error connecting to database", 4)
	} else {
		helpers.Log("Connected to Database", 1)
	}
	return client
}