add MigrateDB func for a single source of truth

1 files changed, 3 insertions, 3 deletions

diff --git a/main.go b/main.go
index 8cb2af22..68f59911 100644
--- a/main.go
+++ b/main.go
@@ -2,7 +2,6 @@ package main
 
 import (
 	"fmt"
-	"github.com/spf13/cobra"
 	"log"
 	"os"
 	"os/signal"
@@ -16,6 +15,8 @@ import (
 	"x-ui/web/global"
 	"x-ui/web/service"
 
+	"github.com/spf13/cobra"
+
 	"github.com/op/go-logging"
 )
 
@@ -215,8 +216,7 @@ func migrateDb() {
 		log.Fatal(err)
 	}
 	fmt.Println("Start migrating database...")
-	inboundService.MigrationRequirements()
-	inboundService.RemoveOrphanedTraffics()
+	inboundService.MigrateDB()
 	fmt.Println("Migration done!")
 }