mirror of https://github.com/veypi/OneAuth.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
707 B
Go
41 lines
707 B
Go
7 months ago
|
//
|
||
|
|
// db.go
|
||
|
|
// Copyright (C) 2024 veypi <i@veypi.com>
|
||
|
|
// 2025-03-04 16:08:06
|
||
|
|
// Distributed under terms of the MIT license.
|
||
|
|
//
|
||
|
|
|
||
|
|
package cfg
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/glebarez/sqlite"
|
||
|
|
"gorm.io/driver/mysql"
|
||
|
|
"gorm.io/driver/postgres"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
var db *gorm.DB
|
||
|
|
|
||
|
|
func DB() *gorm.DB {
|
||
|
|
if db == nil {
|
||
|
|
var err error
|
||
|
|
var conn gorm.Dialector
|
||
|
|
if Config.DB == "mysql" {
|
||
|
|
conn = mysql.Open(Config.DSN)
|
||
|
|
} else if Config.DB == "postgres" {
|
||
|
|
conn = postgres.Open(Config.DSN)
|
||
|
|
} else {
|
||
|
|
conn = sqlite.Open(Config.DSN)
|
||
|
|
}
|
||
|
|
db, err = gorm.Open(conn,
|
||
|
|
&gorm.Config{
|
||
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return db
|
||
|
|
}
|