From 57e8a634b83f3f536b12e5830b8cd113a0a4a4f4 Mon Sep 17 00:00:00 2001
From: pixel <303176530@qq.com>
Date: Mon, 7 Sep 2020 13:18:02 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BC=80=E5=90=AF=E5=88=9D=E5=A7=8B=E5=8C=96?=
 =?UTF-8?q?=E9=85=8D=E7=BD=AE=20=E6=9A=82=E6=97=B6=E4=BF=AE=E6=94=B9?=
 =?UTF-8?q?=E4=B8=BA=20=E9=85=8D=E7=BD=AE=E6=96=87=E4=BB=B6=E6=8E=A7?=
 =?UTF-8?q?=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/config.yaml      | 1 +
 server/config/config.go | 1 +
 server/main.go          | 6 +++++-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/server/config.yaml b/server/config.yaml
index abf6d5df..037c17f9 100644
--- a/server/config.yaml
+++ b/server/config.yaml
@@ -52,6 +52,7 @@ system:
   env: 'public'  # Change to "develop" to skip authentication for development mode
   addr: 8888
   db-type: "mysql"  # support mysql/sqlite
+  need-init-data: false
 
 # captcha configuration
 captcha:
diff --git a/server/config/config.go b/server/config/config.go
index d91cf21d..9a1197ac 100644
--- a/server/config/config.go
+++ b/server/config/config.go
@@ -18,6 +18,7 @@ type System struct {
 	Env           string `mapstructure:"env" json:"env" yaml:"env"`
 	Addr          int    `mapstructure:"addr" json:"addr" yaml:"addr"`
 	DbType        string `mapstructure:"db-type" json:"dbType" yaml:"db-type"`
+	NeedInitData  bool   `mapstructure:"need-init-data" json:"needInitData" yaml:"need-init-data"`
 }
 
 type JWT struct {
diff --git a/server/main.go b/server/main.go
index e0ad59e1..9bbb166b 100644
--- a/server/main.go
+++ b/server/main.go
@@ -3,6 +3,8 @@ package main
 import (
 	"gin-vue-admin/core"
 	"gin-vue-admin/global"
+	"gin-vue-admin/gva/init_data"
+
 	//"gin-vue-admin/gva/init_data"
 	"gin-vue-admin/initialize"
 	//"runtime"
@@ -25,7 +27,9 @@ func main() {
 		initialize.Mysql()
 	}
 	initialize.DBTables()
-	//init_data.InitData() // 打开注释即可初始化数据  使用过后一定要再次注释或删除
+	if global.GVA_CONFIG.System.NeedInitData {
+		init_data.InitData() // 通过配置文件初始化数据 默认为 false 首次运行需要将 ./config.yaml中 system下的 need-init-data 修改为true
+	}
 	// 程序结束前关闭数据库链接
 	db, _ := global.GVA_DB.DB()
 	defer db.Close()
-- 
GitLab