Allows database information to come from environment

Author: stevenleeg

gobb/gobb.sample.conf

@@ -21,6 +21,11 @@ username=db_username
 password=db_password
 database=db_name
 hostname=localhost
+port=5432
+
+;; These options are used for docker port redirection.
+;; env_hostname=POSTGRES_PORT_5432_TCP_PORT
+;; env_port=POSTGRES_PORT_5432_TCP_ADDR
 
 ;; If you have a Google Analytics account, put your information
 ;; in this section (optional)

models/connection.go

@@ -6,6 +6,7 @@ import (
 	"github.com/coopernurse/gorp"
 	_ "github.com/lib/pq"
 	"github.com/stevenleeg/gobb/config"
+	"os"
 )
 
 var db_map *gorp.DbMap
@@ -21,6 +22,17 @@ func GetDbSession() *gorp.DbMap {
 	db_hostname, _ := config.Config.GetString("database", "hostname")
 	db_port, _ := config.Config.GetString("database", "port")
 
+	db_env_hostname, _ := config.Config.GetString("database", "env_hostname")
+	db_env_port, _ := config.Config.GetString("database", "env_port")
+
+	// Allow database information to come from environment variables
+	if db_env_hostname != "" {
+		db_hostname = os.Getenv(db_env_hostname)
+	}
+	if db_env_port != "" {
+		db_port = os.Getenv(db_env_port)
+	}
+
 	if db_port == "" {
 		db_port = "5432"
 	}