From 333b9fb0534eb764df4b11a9234e5b9c2ef05fc6 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 20 Feb 2025 21:11:20 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E6=95=B0=E6=8D=AE=E6=BA=90=E6=89=BE?=
 =?UTF-8?q?=E4=B8=8D=E5=88=B0=E7=9B=B4=E6=8E=A5=E9=80=80=E5=87=BA=E6=8F=90?=
 =?UTF-8?q?=E7=A4=BA=E6=94=B9=E6=AD=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/framework/database/db/db.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/framework/database/db/db.go b/src/framework/database/db/db.go
index 8cc3a52b..59dec95d 100644
--- a/src/framework/database/db/db.go
+++ b/src/framework/database/db/db.go
@@ -131,7 +131,12 @@ func DB(source string) *gorm.DB {
 	if source == "" {
 		source = config.Get("database.defaultDataSourceName").(string)
 	}
-	return dbMap[source]
+	db := dbMap[source]
+	if db == nil {
+		logger.Fatalf("not database source: %s", source)
+		return nil
+	}
+	return db
 }
 
 // Names 获取数据源名称列表