bug fixed #76

diff --git a/models/models.go b/models/models.go
index f1d43531be84151aab0ae30713970360bb406f80..ab13548ef7a2b24e4b533ed22678407021368e73 100644 (file)
--- a/models/models.go
+++ b/models/models.go
@@ -8,6 +8,7 @@ import (
        "fmt"
        "os"
        "path"
+       "strings"
 
        _ "github.com/go-sql-driver/mysql"
        _ "github.com/lib/pq"
@@ -55,8 +56,18 @@ func NewTestEngine(x *xorm.Engine) (err error) {
                x, err = xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8",
                        DbCfg.User, DbCfg.Pwd, DbCfg.Host, DbCfg.Name))
        case "postgres":
-               x, err = xorm.NewEngine("postgres", fmt.Sprintf("user=%s password=%s dbname=%s sslmode=%s",
-                       DbCfg.User, DbCfg.Pwd, DbCfg.Name, DbCfg.SslMode))
+               var host, port = "127.0.0.1", "5432"
+               fields := strings.Split(DbCfg.Host, ":")
+               if len(fields) > 0 {
+                       host = fields[0]
+               }
+               if len(fields) > 1 {
+                       port = fields[1]
+               }
+               cnnstr := fmt.Sprintf("user=%s password=%s host=%s port=%s dbname=%s sslmode=%s",
+                       DbCfg.User, DbCfg.Pwd, host, port, DbCfg.Name, DbCfg.SslMode)
+               //fmt.Println(cnnstr)
+               x, err = xorm.NewEngine("postgres", cnnstr)
        case "sqlite3":
                if !EnableSQLite3 {
                        return fmt.Errorf("Unknown database type: %s", DbCfg.Type)