detect DATABASE_URL that consist of postgres

Author: jackii

Dockerfile

@@ -1,7 +1,7 @@
 ############################
 # STEP 1 build executable binary
 ############################
-FROM golang:alpine AS builder
+FROM golang:1.19.10-alpine3.18 AS builder
 RUN apk update && apk add --no-cache git gcc libc-dev openssl && go install github.com/gobuffalo/packr/packr@latest
 WORKDIR $GOPATH/src/securitybunker/databunker/src/
 COPY src/go.mod ./deps

src/storage/pgsql-storage.go

@@ -8,6 +8,7 @@ import (
 	"io/ioutil"
 	"log"
 	"net/http"
+	"net/url"
 	"os"
 	"strconv"
 	"strings"
@@ -24,6 +25,22 @@ type PGSQLDB struct {
 }
 
 func (dbobj PGSQLDB) getConnectionString(dbname *string) string {
+	databaseURL := os.Getenv("DATABASE_URL")
+	if len(databaseURL) > 0 {
+		u, err := url.Parse(databaseURL)
+		if err == nil && u.Scheme == "postgres" {
+			// Extract user info, host, port, and dbname from the URL
+			user := u.User.Username()
+			pass, _ := u.User.Password()
+			host := u.Hostname()
+			port := u.Port()
+			dbname := strings.TrimPrefix(u.Path, "/")
+
+			return fmt.Sprintf("user='%s' password='%s' host='%s' port='%s' dbname='%s'",
+				user, pass, host, port, dbname)
+		}
+	}
+
 	user := os.Getenv("PGSQL_USER_NAME")
 	pass := os.Getenv("PGSQL_USER_PASS")
 	host := os.Getenv("PGSQL_HOST")

src/storage/storage.go

@@ -3,6 +3,7 @@ package storage
 import (
 	"go.mongodb.org/mongo-driver/bson"
 	"net/http"
+	"net/url"
 	"os"
 )
 
@@ -104,8 +105,18 @@ type BackendDB interface {
 }
 
 func getDBObj() BackendDB {
-	host := os.Getenv("MYSQL_HOST")
 	var db BackendDB
+	databaseURL := os.Getenv("DATABASE_URL")
+	// Check if DATABASE_URL is set and is a PostgreSQL URL
+	if len(databaseURL) > 0 {
+		u, err := url.Parse(databaseURL)
+		if err == nil && u.Scheme == "postgres" {
+			db = &PGSQLDB{}
+			return db
+		}
+	}
+
+	host := os.Getenv("MYSQL_HOST")
 	if len(host) > 0 {
 		db = &MySQLDB{}
 		return db