mirror of
https://github.com/samuelncui/yatm.git
synced 2026-01-03 11:45:21 +00:00
21 lines
327 B
Go
21 lines
327 B
Go
package resource
|
|
|
|
import (
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func NewDBConn(dialect, dsn string) (*gorm.DB, error) {
|
|
|
|
var dialector gorm.Dialector
|
|
switch dialect {
|
|
case "mysql":
|
|
dialector = mysql.Open(dsn)
|
|
case "sqlite":
|
|
dialector = sqlite.Open(dsn)
|
|
}
|
|
|
|
return gorm.Open(dialector)
|
|
}
|