From 9ff094b6f5b1ed401cbe280f83a15aa7df14c565 Mon Sep 17 00:00:00 2001
From: wenjianzhang <zwj777@live.com>
Date: Fri, 3 Nov 2023 17:36:04 +0800
Subject: [PATCH] =?UTF-8?q?fix=F0=9F=90=9B:=20Fixed=20data=20migration=20i?=
 =?UTF-8?q?ssue=20during=20multi-tenant=20configuration?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/migrate/server.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/cmd/migrate/server.go b/cmd/migrate/server.go
index fa25741d..b1667ca6 100644
--- a/cmd/migrate/server.go
+++ b/cmd/migrate/server.go
@@ -63,6 +63,18 @@ func migrateModel() error {
 		host = "*"
 	}
 	db := sdk.Runtime.GetDbByKey(host)
+	if db == nil {
+		if len(sdk.Runtime.GetDb()) == 1 && host == "*" {
+			for k, v := range sdk.Runtime.GetDb() {
+				db = v
+				host = k
+				break
+			}
+		}
+	}
+	if db == nil {
+		return fmt.Errorf("未找到数据库配置")
+	}
 	if config.DatabasesConfig[host].Driver == "mysql" {
 		//初始化数据库时候用
 		db.Set("gorm:table_options", "ENGINE=InnoDB CHARSET=utf8mb4")