From a21aad82c8162ce4c75f82761df0a2ba361e7b75 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Wed, 9 Feb 2022 16:16:43 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8A=A0=E8=BD=BD=E9=94=99?=
 =?UTF-8?q?=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/migrate.go |  7 ++++++-
 engine/load.go | 37 +++++++++++++++++++------------------
 2 files changed, 25 insertions(+), 19 deletions(-)

diff --git a/cmd/migrate.go b/cmd/migrate.go
index d7d6294f0f..39958cb79b 100644
--- a/cmd/migrate.go
+++ b/cmd/migrate.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"fmt"
+	"os"
 
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
@@ -18,7 +19,11 @@ var migrateCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		Boot()
 		// 加载数据模型
-		engine.Load(config.Conf)
+		err := engine.Load(config.Conf)
+		if err != nil {
+			fmt.Printf(color.RedString("加载失败: %s\n", err.Error()))
+			os.Exit(1)
+		}
 
 		if name != "" {
 			mod, has := gou.Models[name]
diff --git a/engine/load.go b/engine/load.go
index 626bb46cb1..efbef1fe52 100644
--- a/engine/load.go
+++ b/engine/load.go
@@ -32,6 +32,10 @@ func Load(cfg config.Config) (err error) {
 	// 第一步: 加载应用信息
 	app.Load(cfg)
 
+	// 加密密钥函数
+	gou.LoadCrypt(fmt.Sprintf(`{"key":"%s"}`, cfg.DB.AESKey), "AES")
+	gou.LoadCrypt(`{}`, "PASSWORD")
+
 	// 第二步: 建立数据库 & 会话连接
 	share.DBConnect(cfg.DB)           // 创建数据库连接
 	share.SessionConnect(cfg.Session) // 创建会话服务器链接
@@ -49,51 +53,48 @@ func Load(cfg config.Config) (err error) {
 	// 第四步: 加载共享库 & JS 处理器
 	err = share.Load(cfg) // 加载共享库 lib
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
 	err = script.Load(cfg) // 加载JS处理器 script
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
 
 	// 第五步: 加载数据模型等
 	err = model.Load(cfg) // 加载数据模型 model
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
+
 	err = flow.Load(cfg) // 加载业务逻辑 Flow
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
 	err = plugin.Load(cfg) // 加载业务插件 plugin
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
+
 	err = table.Load(cfg) // 加载数据表格 table
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
+
 	err = chart.Load(cfg) // 加载分析图表 chart
 	if err != nil {
-		return err
-	}
-	err = page.Load(cfg) // 加载页面 page
-	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
 
-	importer.Load(cfg)  // 加载数据导入 imports
-	workflow.Load(cfg)  // 加载工作流  workflow
+	page.Load(cfg)     // 加载页面 page 忽略错误
+	importer.Load(cfg) // 加载数据导入 imports
+	workflow.Load(cfg) // 加载工作流  workflow
+
 	err = api.Load(cfg) // 加载业务接口 API
 	if err != nil {
-		return err
+		exception.Err(err, 400).Throw()
 	}
 
 	server.Load(cfg) // 加载服务
-
-	// 加密密钥函数
-	gou.LoadCrypt(fmt.Sprintf(`{"key":"%s"}`, cfg.DB.AESKey), "AES")
-	gou.LoadCrypt(`{}`, "PASSWORD")
 	return nil
 }