yintai2/app/common/dao/base.go

package dao

import (
	"yintai-api/base/global"

	"gorm.io/gorm"
)

func getDB(db ...*gorm.DB) *gorm.DB {
	if len(db) > 0 {
		return db[0]
	}

	return global.GetDB()
}