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

server/source/authority.go

@@ -0,0 +1,36 @@
+package information
+
+import (
+	"gin-vue-admin/global"
+	"gin-vue-admin/model"
+	"github.com/gookit/color"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+var Authority = new(authority)
+
+type authority struct{}
+
+var authorities = []model.SysAuthority{
+	{CreatedAt: time.Now(), UpdatedAt: time.Now(), AuthorityId: "888", AuthorityName: "普通用户", ParentId: "0", DefaultRouter: "dashboard"},
+	{CreatedAt: time.Now(), UpdatedAt: time.Now(), AuthorityId: "8881", AuthorityName: "普通用户子角色", ParentId: "888", DefaultRouter: "dashboard"},
+	{CreatedAt: time.Now(), UpdatedAt: time.Now(), AuthorityId: "9528", AuthorityName: "测试角色", ParentId: "0", DefaultRouter: "dashboard"},
+}
+
+//@author: [SliverHorn](https://github.com/SliverHorn)
+//@description: sys_authorities 表数据初始化
+func (a *authority) Init() error {
+	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		if tx.Where("authority_id IN ? ", []string{"888", "9528"}).Find(&[]model.SysAuthority{}).RowsAffected == 2 {
+			color.Danger.Println("\n[Mysql] --> sys_authorities 表的初始数据已存在!")
+			return nil
+		}
+		if err := tx.Create(&authorities).Error; err != nil { // 遇到错误时回滚事务
+			return err
+		}
+		color.Info.Println("\n[Mysql] --> sys_authorities 表初始数据成功!")
+		return nil
+	})
+}