replace db driver package and driver name for postgres

This commit is contained in:
ZhengYang 2014-08-22 15:46:36 +08:00
parent ab24b84d15
commit 4dc2d67bd2
1 changed files with 10 additions and 3 deletions

View File

@ -89,11 +89,11 @@ import (
"github.com/astaxie/beego" "github.com/astaxie/beego"
"github.com/astaxie/beego/orm" "github.com/astaxie/beego/orm"
_ "github.com/go-sql-driver/mysql" {{.DriverPkg}}
) )
func init() { func init() {
orm.RegisterDataBase("default", "mysql", "{{.conn}}") orm.RegisterDataBase("default", "{{.DriverName}}", "{{.conn}}")
} }
func main() { func main() {
@ -574,9 +574,16 @@ func createapi(cmd *Command, args []string) int {
if conn != "" { if conn != "" {
fmt.Println("create main.go:", path.Join(apppath, "main.go")) fmt.Println("create main.go:", path.Join(apppath, "main.go"))
maingoContent := strings.Replace(apiMainconngo, "{{.Appname}}", packpath, -1)
maingoContent = strings.Replace(maingoContent, "{{.DriverName}}", string(driver), -1)
if driver == "mysql" {
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/go-sql-driver/mysql"`, -1)
} else if driver == "postgres" {
maingoContent = strings.Replace(maingoContent, "{{.DriverPkg}}", `_ "github.com/lib/pq"`, -1)
}
writetofile(path.Join(apppath, "main.go"), writetofile(path.Join(apppath, "main.go"),
strings.Replace( strings.Replace(
strings.Replace(apiMainconngo, "{{.Appname}}", packpath, -1), maingoContent,
"{{.conn}}", "{{.conn}}",
conn.String(), conn.String(),
-1, -1,