From 1a9e5735a49633d8f566cec77b10c86603c9b633 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Tue, 12 Nov 2024 10:15:30 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20DB=E7=A9=BA=E6=95=B0=E6=8D=AE=E6=BA=90?=
 =?UTF-8?q?=E6=97=B6=E9=80=89=E6=8B=A9=E9=BB=98=E8=AE=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/framework/datasource/datasource.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/framework/datasource/datasource.go b/src/framework/datasource/datasource.go
index 5b1a99c7..736e48d2 100644
--- a/src/framework/datasource/datasource.go
+++ b/src/framework/datasource/datasource.go
@@ -117,6 +117,9 @@ func DefaultDB() *gorm.DB {
 
 // 获取数据源
 func DB(source string) *gorm.DB {
+	if source == "" {
+		source = config.Get("gorm.defaultDataSourceName").(string)
+	}
 	return dbMap[source]
 }