转载

Golang Beego Api连接Mysql简单示例（集成Swagger和Convey单测）

1.工程创建步骤如下，

https://beego.me/blog/beego_api

如要集成swagger，请注意使用下面的语句启动beego。

bee run -downdoc=true -gendoc=true

2.连接mysql数据库。

先要安装两个包，

如下，

go get github.com/astaxie/beego/orm
go get github.com/go-sql-driver/mysql

其次有一个坑，就是必须要先建一个别名为default的数据库，才能启动连接。

而且这个建库的操作最好是在models里面，实测的话，框架先加载models，然后才去运行main包里面的init函数，单元测试直接测试models也是只加载models。

另一个问题是RegisterDatabase这个函数只能调一次，对同一个数据库别名来说，只能调一次。所以要先判断要注册的数据库是否已经注册了。如果已注册，就不要再注册了，否则又会报错。所以写了一个class来处理这个问题。

具体的代码如下，

dbinit.go源代码，

package models

import (
    "database/sql"

    "github.com/astaxie/beego"
    "github.com/astaxie/beego/logs"
    "github.com/astaxie/beego/orm"
    _ "github.com/go-sql-driver/mysql"
)

type DBInit struct {
    Db *sql.DB
}

func DBInitNew() *DBInit {
    db, _ := orm.GetDB()
    dbInit := DBInit{Db: db}
    return &dbInit
}

func (dbInit *DBInit) InitDatabase() {
    if dbInit.Db == nil {
        beego.LoadAppConfig("ini", "..//conf//app.conf")
        orm.RegisterDriver("mysql", orm.DRMySQL)
        dbType := beego.AppConfig.String("dbType")
        sqlConn := beego.AppConfig.String("sqlconn")
        maxIdle, _ := beego.AppConfig.Int("maxIdle")
        maxConn, _ := beego.AppConfig.Int("maxConn")
        logs.Info("Connect to [%v] database with conn: [%v] /n", dbType, sqlConn)
        orm.RegisterDataBase("default", dbType, sqlConn, maxIdle, maxConn)
        orm.Debug = true
    }
}

3.代码目录结构如下，