bug fixed #76
This commit is contained in:
parent
8c266f2df5
commit
6cee65db5a
1 changed files with 13 additions and 2 deletions
|
@ -8,6 +8,7 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
|
"strings"
|
||||||
|
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "github.com/go-sql-driver/mysql"
|
||||||
_ "github.com/lib/pq"
|
_ "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",
|
x, err = xorm.NewEngine("mysql", fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8",
|
||||||
DbCfg.User, DbCfg.Pwd, DbCfg.Host, DbCfg.Name))
|
DbCfg.User, DbCfg.Pwd, DbCfg.Host, DbCfg.Name))
|
||||||
case "postgres":
|
case "postgres":
|
||||||
x, err = xorm.NewEngine("postgres", fmt.Sprintf("user=%s password=%s dbname=%s sslmode=%s",
|
var host, port = "127.0.0.1", "5432"
|
||||||
DbCfg.User, DbCfg.Pwd, DbCfg.Name, DbCfg.SslMode))
|
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":
|
case "sqlite3":
|
||||||
if !EnableSQLite3 {
|
if !EnableSQLite3 {
|
||||||
return fmt.Errorf("Unknown database type: %s", DbCfg.Type)
|
return fmt.Errorf("Unknown database type: %s", DbCfg.Type)
|
||||||
|
|
Reference in a new issue