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