12

internal/global/global.go

@@ -0,0 +1,26 @@
+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
+}