代码拉取完成,页面将自动刷新
package dbgorm
import (
"sync"
"gorm.io/gorm"
)
// 本地驱动信息存储,以及对应的lock锁
var _default_dervice map[string]func(dsn string) gorm.Dialector = map[string]func(dsn string) gorm.Dialector{}
var _default_dervice_lock sync.Mutex
// 设置数据库驱动信息
// 如:连接MySQL数据库:dbgorm.SetDervice("mysql",mysql.Open)
// 如果数据库驱动已存在的话,此处会直接跳过本次设置
//
// name 驱动名称
// fun 驱动打开函数
func SetDervice(name string, fun func(dsn string) gorm.Dialector) {
if _, ok := _default_dervice[name]; ok {
// 如果驱动已存在则直接跳过本次插入
return
}
_default_dervice_lock.Lock()
defer _default_dervice_lock.Unlock()
_default_dervice[name] = fun
}
// 设置数据库驱动信息
// 如:连接MySQL数据库:dbgorm.SetDervices(mysql.Open)
// 如果数据库驱动已存在的话,此处会直接跳过本次设置
//
// fun 驱动打开函数
func SetDervices(fun ...func(dsn string) gorm.Dialector) {
_default_dervice_lock.Lock()
defer _default_dervice_lock.Unlock()
for i := 0; i < len(fun); i++ {
name := fun[i]("").Name()
_default_dervice[name] = fun[i]
}
}
// 设置数据库驱动信息
// 如:连接MySQL数据库:dbgorm.SetDervice("mysql",mysql.Open)
// 此操作为强制设置驱动,如果没有则会进行设置,如果有则会进行替换
//
// name 驱动名称
// fun 驱动打开函数
func ReplaceDervice(name string, fun func(dsn string) gorm.Dialector) {
_default_dervice_lock.Lock()
defer _default_dervice_lock.Unlock()
_default_dervice[name] = fun
}
// 获取已设置的数据库驱动列表
func GetDerviceName() []string {
dervice_name := []string{}
for i, _ := range _default_dervice {
dervice_name = append(dervice_name, i)
}
return dervice_name
}
// 判断某驱动名称是否已设置
//
// name 驱动名称
func HasDervice(name string) bool {
if _, ok := _default_dervice[name]; ok {
return true
}
return false
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。