mirror of
https://github.com/Li4n0/revsuit.git
synced 2026-09-21 22:30:46 +08:00
25 lines
348 B
Go
25 lines
348 B
Go
package database
|
|
|
|
import "gorm.io/gorm"
|
|
|
|
var (
|
|
DB *gorm.DB
|
|
Driver DriverType
|
|
)
|
|
|
|
type DriverType = string
|
|
|
|
const Sqlite = "sqlite"
|
|
const Mysql = "mysql"
|
|
|
|
func InitDB(driver DriverType, dsn string) (err error) {
|
|
Driver = driver
|
|
switch driver {
|
|
case Sqlite:
|
|
DB, err = NewSqlite3(dsn)
|
|
case Mysql:
|
|
DB, err = NewMysql(dsn)
|
|
}
|
|
return err
|
|
}
|