diff --git a/server/config.yaml b/server/config.yaml index abf6d5dfe1f52756ccf4a6e0acfc05db6ee01f3b..037c17f9b8c6f9fe761bd02d0f4ebe636d2c662d 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 d91cf21d02ae9b50e6b5b3a9c2fe7a291a5bdb4f..9a1197acf3c98f697c96af4f109e648c7cd373e2 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 e0ad59e18aeddb35de1ee67d9a0dcb72ee840bb5..9bbb166b0ebc56f7ea8600a6a9f2d074ffb35b63 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()