go-gorm / dbresolver

Multiple databases, read-write splitting FOR GORM

Home Page:https://gorm.io/docs/dbresolver.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Use dr.Call() without datas into db.Register(config, datas...)

Harv3009 opened this issue · comments

Hi! Can i use dr.Call without datas in db.Register(config, datas...) ?

Callback is called if resolvers is not empty or called db.Register()

func (dr *DBResolver) Call(fc func(connPool gorm.ConnPool) error) error {
	if dr.DB != nil {
		for _, r := range dr.resolvers {
			if err := r.call(fc); err != nil {
				return err
			}
		}
	} else {
		dr.compileCallbacks = append(dr.compileCallbacks, fc)
	}

	return nil
}

I wanted to close the connections with dr.Call after db.Use(dbresolver)