41 lines
1.2 KiB
Go
41 lines
1.2 KiB
Go
|
package models
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 用户的自定义权限关联表
|
|||
|
|
type MgtLinkUserPmn struct {
|
|||
|
|
UserId uint `gorm:"column:user_id;primaryKey" json:"user_id"` // 用户ID
|
|||
|
|
AppId uint `gorm:"column:app_id;primaryKey" json:"app_id"` // 应用ID
|
|||
|
|
PmnId uint `gorm:"column:pmn_id;primaryKey" json:"pmn_id"` // 权限ID
|
|||
|
|
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime" json:"created_at,omitempty"` // 创建时间
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (MgtLinkUserPmn) TableName() string {
|
|||
|
|
return "mgt_link_user_pmn"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// func init() {
|
|||
|
|
// database.AppendMigrate(&MgtLinkUserPmn{})
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
// AfterDelete钩子：仅当该用户在该应用下不再拥有任何权限时，移除用户-应用关联
|
|||
|
|
func (l *MgtLinkUserPmn) AfterDelete(tx *gorm.DB) error {
|
|||
|
|
var count int64
|
|||
|
|
if err := tx.Model(&MgtLinkUserPmn{}).
|
|||
|
|
Where("app_id = ? AND user_id = ?", l.AppId, l.UserId).
|
|||
|
|
Count(&count).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if count == 0 {
|
|||
|
|
if err := tx.Where("app_id = ? AND user_id = ?", l.AppId, l.UserId).
|
|||
|
|
Delete(&MgtLinkUserApp{}).Error; err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|