自动初始化数据库功能后端部分完成

server/source/data_authorities.go

@@ -0,0 +1,41 @@
+package information
+
+import (
+	"gin-vue-admin/global"
+
+	"github.com/gookit/color"
+	"gorm.io/gorm"
+)
+
+var DataAuthorities = new(dataAuthorities)
+
+type dataAuthorities struct{}
+
+type DataAuthority struct {
+	AuthorityId   string `gorm:"column:sys_authority_authority_id"`
+	DataAuthority string `gorm:"column:data_authority_id_authority_id"`
+}
+
+var infos = []DataAuthority{
+	{"888", "888"},
+	{"888", "8881"},
+	{"888", "9528"},
+	{"9528", "8881"},
+	{"9528", "9528"},
+}
+
+//@author: [SliverHorn](https://github.com/SliverHorn)
+//@description: sys_data_authority_id 表数据初始化
+func (d *dataAuthorities) Init() error {
+	return global.GVA_DB.Table("sys_data_authority_id").Transaction(func(tx *gorm.DB) error {
+		if tx.Where("sys_authority_authority_id IN ('888', '9528') ").Find(&[]DataAuthority{}).RowsAffected == 5 {
+			color.Danger.Println("\n[Mysql] --> sys_data_authority_id 表初始数据已存在!")
+			return nil
+		}
+		if err := tx.Create(&infos).Error; err != nil { // 遇到错误时回滚事务
+			return err
+		}
+		color.Info.Println("\n[Mysql] --> sys_data_authority_id 表初始数据成功!")
+		return nil
+	})
+}