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

This commit is contained in:
TsMask
2025-02-20 21:11:20 +08:00
parent 89a66a1503
commit 333b9fb053

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 获取数据源名称列表