27 lines
323 B
Go
27 lines
323 B
Go
package global
|
|
|
|
import (
|
|
"sync"
|
|
"unR2/internal/config"
|
|
"unR2/internal/ent"
|
|
)
|
|
|
|
var (
|
|
dbOnce sync.Once
|
|
db *ent.Client
|
|
AppConfig = &config.Conf
|
|
)
|
|
|
|
func SetDB(client *ent.Client) {
|
|
dbOnce.Do(func() {
|
|
db = client
|
|
})
|
|
}
|
|
|
|
func DB() *ent.Client {
|
|
if db == nil {
|
|
panic("DB is not initialized")
|
|
}
|
|
return db
|
|
}
|