You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
520 B
30 lines
520 B
package dorm
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/beego/beego/v2/client/orm"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
func NewBeegoTidbClient(config *ConfigBeegoClient) (*BeegoClient, error) {
|
|
|
|
var err error
|
|
c := &BeegoClient{config: config}
|
|
|
|
err = orm.RegisterDriver("TiDB", orm.DRTiDB)
|
|
if err != nil {
|
|
return nil, errors.New(fmt.Sprintf("加载驱动失败：%v", err))
|
|
}
|
|
|
|
var db *sql.DB
|
|
o, err := orm.NewOrmWithDB("TiDB", "default", db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c.Db = &o
|
|
|
|
return c, nil
|
|
}
|