完成xray启动

database/db.go

@@ -1,13 +1,15 @@
 package database
 
 import (
+	"gorm.io/driver/sqlite"
 	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
 	"io/fs"
 	"os"
 	"path"
+	"x-ui/config"
 	"x-ui/database/model"
 )
-import "gorm.io/driver/sqlite"
 
 var db *gorm.DB
 
@@ -35,6 +37,10 @@ func initInbound() error {
 	return db.AutoMigrate(&model.Inbound{})
 }
 
+func initSetting() error {
+	return db.AutoMigrate(&model.Setting{})
+}
+
 func InitDB(dbPath string) error {
 	dir := path.Dir(dbPath)
 	err := os.MkdirAll(dir, fs.ModeDir)
@@ -42,7 +48,17 @@ func InitDB(dbPath string) error {
 		return err
 	}
 
-	c := &gorm.Config{}
+	var gormLogger logger.Interface
+
+	if config.IsDebug() {
+		gormLogger = logger.Discard
+	} else {
+		gormLogger = logger.Default
+	}
+
+	c := &gorm.Config{
+		Logger: gormLogger,
+	}
 	db, err = gorm.Open(sqlite.Open(dbPath), c)
 	if err != nil {
 		return err
@@ -56,6 +72,10 @@ func InitDB(dbPath string) error {
 	if err != nil {
 		return err
 	}
+	err = initSetting()
+	if err != nil {
+		return err
+	}
 
 	return nil
 }
@@ -63,3 +83,7 @@ func InitDB(dbPath string) error {
 func GetDB() *gorm.DB {
 	return db
 }
+
+func IsNotFound(err error) bool {
+	return err == gorm.ErrRecordNotFound
+}