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]
 }