Using config.toml with database configuration params

Author: romanoff

db.go

@@ -6,18 +6,13 @@ import (
 	_ "github.com/lib/pq"
 )
 
-const (
-	DB_USER     = "user"
-	DB_PASSWORD = "password"
-	DB_NAME     = "dbname"
-	DB_HOST     = "127.0.0.1"
-	DB_PORT     = "5432"
-	DB_SSLMODE  = "disable"
-)
-
 func GetDbConnection() (*sql.DB, error) {
-	dbinfo := fmt.Sprintf("postgres://%s:%s@%s:%s/%s?sslmode=%s",
-		DB_USER, DB_PASSWORD, DB_HOST, DB_PORT, DB_NAME, DB_SSLMODE)
+	config, err := ParseDbConfig(".")
+	if err != nil {
+		return nil, err
+	}
+	dbinfo := fmt.Sprintf("postgres://%s:%s@%s:%v/%s?sslmode=%s",
+		config.User, config.Password, config.Host, config.Port, config.Database, config.SslMode)
 	db, err := sql.Open("postgres", dbinfo)
 	if err != nil {
 		return nil, err

parser.go

@@ -2,7 +2,9 @@ package main
 
 import (
 	"bytes"
+	"errors"
 	"fmt"
+	"github.com/BurntSushi/toml"
 	"github.com/xeipuuv/gojsonschema"
 	"io/ioutil"
 	"strings"
@@ -92,3 +94,38 @@ func ParseSqlTemplate(path string, route *Route) error {
 	route.SqlTemplate = tmpl
 	return nil
 }
+
+type DbConfig struct {
+	User     string
+	Password string
+	Database string
+	Host     string
+	Port     int
+	SslMode  string
+}
+
+func ParseDbConfig(path string) (*DbConfig, error) {
+	conf := &DbConfig{}
+	content, err := ioutil.ReadFile(path + "/config.toml")
+	if err != nil {
+		return nil, errors.New(fmt.Sprintf("Error while reading config.toml configuration: %v", err))
+	}
+	_, err = toml.Decode(string(content), &conf)
+	if err != nil {
+		return nil, err
+	}
+	if conf.User == "" {
+		conf.User = "postgres"
+	}
+	if conf.Host == "" {
+		conf.Host = "127.0.0.1"
+	}
+	if conf.Port == 0 {
+		conf.Port = 5432
+	}
+	if conf.SslMode == "" {
+		conf.SslMode = "disable"
+	}
+	return conf, nil
+
+}

testapp/config.toml

@@ -1,7 +1,6 @@
-[database]
 user = "user"
 password = "password"
 database = "dbname"
 host = "127.0.0.1"
-port = "5434"
+port = 5434
 sslmode = "disable"