openbilibili/app/admin/main/up/dao/upcrm/dao.go

package upcrm

import (
	"context"

	"go-common/app/admin/main/up/conf"
	"go-common/library/log"
	bm "go-common/library/net/http/blademaster"

	"github.com/jinzhu/gorm"
)

//Dao upcrm dao
type Dao struct {
	// config
	conf *conf.Config
	// db
	crmdb      *gorm.DB
	httpClient *bm.Client
}

//New new dao
func New(c *conf.Config) *Dao {
	var d = &Dao{
		conf: c,
	}
	crmdb, err := gorm.Open("mysql", c.DB.Upcrm.DSN)
	if crmdb == nil {
		log.Error("connect to db fail, err=%v", err)
		return nil
	}
	crmdb.SingularTable(true)
	d.crmdb = crmdb
	d.crmdb.LogMode(c.IsTest)
	return d
}

//SetHTTPClient set http client
func (d *Dao) SetHTTPClient(client *bm.Client) {
	d.httpClient = client
}

//GetDb get current gorm db
func (d *Dao) GetDb() *gorm.DB {
	return d.crmdb
}

// BeginTran begin transcation.
func (d *Dao) BeginTran(c context.Context) (tx *gorm.DB) {
	return d.crmdb.Begin()
}