Migrate the gva project to the cmd folder, delete other redundant database configurations, optimize the initialization method of viper, zap, and gorm

2020-10-22 16:28:51 +08:00
parent 45f2bd4a15
commit dd65a4a716
28 changed files with 1010 additions and 698 deletions
--- a/server/cmd/datas/init.go
+++ b/server/cmd/datas/init.go
@@ -0,0 +1,108 @@
+package datas
+
+import (
+	"gin-vue-admin/model"
+	gormadapter "github.com/casbin/gorm-adapter/v3"
+	"github.com/gookit/color"
+	"gorm.io/gorm"
+	"os"
+)
+
+func InitMysqlData(db *gorm.DB) {
+	var err error
+	err = InitSysApi(db)
+	err = InitSysUser(db)
+	err = InitExaCustomer(db)
+	err = InitCasbinModel(db)
+	err = InitSysAuthority(db)
+	err = InitSysBaseMenus(db)
+	err = InitAuthorityMenu(db)
+	err = InitSysDictionary(db)
+	err = InitSysAuthorityMenus(db)
+	err = InitSysDataAuthorityId(db)
+	err = InitSysDictionaryDetail(db)
+	err = InitExaFileUploadAndDownload(db)
+	if err != nil {
+		color.Warn.Printf("[Mysql]-->初始化数据失败,err: %v\n", err)
+		os.Exit(0)
+	}
+	color.Info.Println("[Mysql]-->初始化数据成功")
+}
+
+func InitMysqlTables(db *gorm.DB) {
+	var err error
+	err = db.AutoMigrate(
+		model.SysApi{},
+		model.SysUser{},
+		model.ExaFile{},
+		model.ExaCustomer{},
+		model.SysBaseMenu{},
+		model.SysWorkflow{},
+		model.SysAuthority{},
+		model.JwtBlacklist{},
+		model.ExaFileChunk{},
+		model.SysDictionary{},
+		model.ExaSimpleUploader{},
+		model.SysOperationRecord{},
+		model.SysWorkflowStepInfo{},
+		model.SysDictionaryDetail{},
+		model.SysBaseMenuParameter{},
+		model.ExaFileUploadAndDownload{},
+	)
+	if err != nil {
+		color.Warn.Printf("[Mysql]-->初始化数据表失败,err: %v\n", err)
+		os.Exit(0)
+	}
+	color.Info.Println("[Mysql]-->初始化数据表成功")
+}
+
+func InitPostgresqlData(db *gorm.DB) {
+	var err error
+	err = InitSysApi(db)
+	err = InitSysUser(db)
+	err = InitExaCustomer(db)
+	err = InitCasbinModel(db)
+	err = InitSysAuthority(db)
+	err = InitSysBaseMenus(db)
+	err = InitAuthorityMenu(db)
+	err = InitSysAuthorityMenus(db)
+	err = InitSysDataAuthorityId(db)
+	err = InitSysDictionaryDetail(db)
+	err = InitExaFileUploadAndDownload(db)
+	err = InitSysDictionaryToPostgresql(db)
+	if err != nil {
+		color.Error.Printf("[Postgresql]-->初始化数据失败,err: %v\n", err)
+		os.Exit(0)
+	}
+	color.Info.Println("[Postgresql]-->初始化数据成功")
+}
+
+func InitPostgresqlTables(db *gorm.DB) {
+	var err error
+	if !db.Migrator().HasTable("casbin_rule") {
+		err = db.Migrator().CreateTable(&gormadapter.CasbinRule{})
+	}
+	err = db.AutoMigrate(
+		model.SysApi{},
+		model.SysUser{},
+		model.ExaFile{},
+		model.ExaCustomer{},
+		model.SysBaseMenu{},
+		model.SysWorkflow{},
+		model.SysAuthority{},
+		model.JwtBlacklist{},
+		model.ExaFileChunk{},
+		model.ExaSimpleUploader{},
+		model.SysOperationRecord{},
+		model.SysWorkflowStepInfo{},
+		model.SysDictionaryDetail{},
+		model.SysBaseMenuParameter{},
+		model.ExaFileUploadAndDownload{},
+		SysDictionaryToPostgresql{},
+	)
+	if err != nil {
+		color.Error.Printf("[Postgresql]-->初始化数据表失败,err: %v\n", err)
+		os.Exit(0)
+	}
+	color.Info.Println("[Postgresql]-->初始化数据表成功")
+}