From f1d14141e4dce6f94d04e7906eb7d599f8ab3252 Mon Sep 17 00:00:00 2001
From: Martin Gallauner <martin.gallauner@pm.me>
Date: Fri, 5 Jul 2024 07:31:54 +0200
Subject: [PATCH] adds db connection string

---
 internal/database.go | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/internal/database.go b/internal/database.go
index 6a264b0..2138f63 100644
--- a/internal/database.go
+++ b/internal/database.go
@@ -9,13 +9,14 @@ import (
 )
 
 type DatabaseConfig struct {
-	Host     string `env:"POSTGRES_HOST" validate:"hostname"` 
-	DbUser   string `env:"POSTGRES_USER"`
-	Password string `env:"POSTGRES_PASSWORD"`
-	Dbname   string `env:"POSTGRES_DBNAME"`
-	Port     int `env:"POSTGRES_PORT"`
-	Sslmode  string `env:"POSTGRES_SSLMODE"`
-	Timezone string `env:"TIMEZONE"`
+	Host        string `env:"POSTGRES_HOST" validate:"hostname"`
+	DbUser      string `env:"POSTGRES_USER"`
+	Password    string `env:"POSTGRES_PASSWORD"`
+	Dbname      string `env:"POSTGRES_DBNAME"`
+	Port        int    `env:"POSTGRES_PORT"`
+	Sslmode     string `env:"POSTGRES_SSLMODE"`
+	Timezone    string `env:"TIMEZONE"`
+	DatabaseUrl string `env:"DATABASE_URL"`
 }
 
 func ReadDatabaseConfig() (DatabaseConfig, error) {
@@ -34,9 +35,10 @@ func SetupDatabase(config DatabaseConfig) (*gorm.DB, error) {
 	if err != nil {
 		return nil, err
 	}
-	connString := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s", 
-	config.Host, config.DbUser, config.Password, config.Dbname, config.Port, config.Sslmode, config.Timezone)
-	return SetupDatabaseWithDSN(connString)
+	// commented out to use the connection string provided by Digital Ocean
+	/* connString := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s",
+		config.Host, config.DbUser, config.Password, config.Dbname, config.Port, config.Sslmode, config.Timezone) */
+	return SetupDatabaseWithDSN(config.DatabaseUrl)
 }
 
 func SetupDatabaseWithDSN(connString string) (*gorm.DB, error) {