fix: 数据源找不到直接退出提示改正

This commit is contained in:
TsMask committed 2025-02-20 21:11:20 +08:00
1 parent 89a66a1503
commit 333b9fb053
1 file changed
+6 -1
+6 -1
View File
@@ -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 获取数据源名称列表