fix

wangpengfei

2023-08-26 2cffaad12f0ed5ff562f49cf7cdf202a8f328c8d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package system
 
import (
    "context"
    "github.com/pkg/errors"
    "gorm.io/gorm"
    sysModel "srm/model/system"
    "srm/service/system"
    "srm/utils"
)
 
const initOrderAuthority = initOrderCasbin + 1
 
type initAuthority struct{}
 
// auto run
func init() {
    system.RegisterInit(initOrderAuthority, &initAuthority{})
}
 
func (i *initAuthority) MigrateTable(ctx context.Context) (context.Context, error) {
    db, ok := ctx.Value("db").(*gorm.DB)
    if !ok {
        return ctx, system.ErrMissingDBContext
    }
    return ctx, db.AutoMigrate(&sysModel.SysAuthority{})
}
 
func (i *initAuthority) TableCreated(ctx context.Context) bool {
    db, ok := ctx.Value("db").(*gorm.DB)
    if !ok {
        return false
    }
    return db.Migrator().HasTable(&sysModel.SysAuthority{})
}
 
func (i initAuthority) InitializerName() string {
    return sysModel.SysAuthority{}.TableName()
}
 
func (i *initAuthority) InitializeData(ctx context.Context) (context.Context, error) {
    db, ok := ctx.Value("db").(*gorm.DB)
    if !ok {
        return ctx, system.ErrMissingDBContext
    }
    entities := []sysModel.SysAuthority{
        {AuthorityId: 888, AuthorityName: "普通用户", ParentId: utils.Pointer[uint](0), DefaultRouter: "dashboard"},
        {AuthorityId: 9528, AuthorityName: "测试角色", ParentId: utils.Pointer[uint](0), DefaultRouter: "dashboard"},
        {AuthorityId: 8881, AuthorityName: "普通用户子角色", ParentId: utils.Pointer[uint](888), DefaultRouter: "dashboard"},
    }
 
    if err := db.Create(&entities).Error; err != nil {
        return ctx, errors.Wrapf(err, "%s表数据初始化失败!", sysModel.SysAuthority{}.TableName())
    }
    // data authority
    if err := db.Model(&entities[0]).Association("DataAuthorityId").Replace(
        []*sysModel.SysAuthority{
            {AuthorityId: 888},
            {AuthorityId: 9528},
            {AuthorityId: 8881},
        }); err != nil {
        return ctx, errors.Wrapf(err, "%s表数据初始化失败!",
            db.Model(&entities[0]).Association("DataAuthorityId").Relationship.JoinTable.Name)
    }
    if err := db.Model(&entities[1]).Association("DataAuthorityId").Replace(
        []*sysModel.SysAuthority{
            {AuthorityId: 9528},
            {AuthorityId: 8881},
        }); err != nil {
        return ctx, errors.Wrapf(err, "%s表数据初始化失败!",
            db.Model(&entities[1]).Association("DataAuthorityId").Relationship.JoinTable.Name)
    }
 
    next := context.WithValue(ctx, i.InitializerName(), entities)
    return next, nil
}
 
func (i *initAuthority) DataInserted(ctx context.Context) bool {
    db, ok := ctx.Value("db").(*gorm.DB)
    if !ok {
        return false
    }
    if errors.Is(db.Where("authority_id = ?", "8881").
        First(&sysModel.SysAuthority{}).Error, gorm.ErrRecordNotFound) { // 判断是否存在数据
        return false
    }
    return true
}