From 0d52bd0a349776dc0c48908b17f3c24715806564 Mon Sep 17 00:00:00 2001
From: youys <1272586223@qq.com>
Date: Fri, 1 Aug 2025 14:27:09 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20init=20db=E8=B0=83=E6=95=B4=E4=BD=8D?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/cmd/server/main.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/server/cmd/server/main.go b/server/cmd/server/main.go
index 59feda4..dc61dee 100644
--- a/server/cmd/server/main.go
+++ b/server/cmd/server/main.go
@@ -9,6 +9,7 @@ import (
 	"github.com/go-kratos/kratos/v2/transport/http"
 	"os"
 	"vgpu/internal/conf"
+	"vgpu/internal/database"
 
 	_ "go.uber.org/automaxprocs"
 )
@@ -32,6 +33,12 @@ func init() {
 func main() {
 	flag.Parse()
 	var ctx = context.Background()
+
+	if err := initDatabase(); err != nil {
+		log.Errorf("数据库初始化失败: %v", err)
+		os.Exit(1)
+	}
+
 	app, cleanup, err := initApp(flagconf, ctx)
 	if err != nil {
 		panic(err)
@@ -62,3 +69,15 @@ func newApp(ctx context.Context, logger log.Logger, gs *grpc.Server, hs *http.Se
 func getNodeSelectors(c *conf.Bootstrap) map[string]string {
 	return c.NodeSelectors
 }
+
+func initDatabase() error {
+	config, err := database.LoadConfig(flagconf)
+	log.Infof("config: %+v", config)
+	if err != nil {
+		log.Errorf("Failed to load config: %v", err)
+		return err
+	}
+	database.InitDB(&config.Database)
+	log.Infof("初始化%s成功", config.Database.Driver)
+	return nil
+}