lc_veterans/app/system/dao/role.go

package dao

// Role 角色
type Role struct {
	ID        int    `gorm:"primary_key" json:"id"`                             //编号
	TenantId  string `gorm:"type:varchar(12);default '000000'" json:"tenantId"` //租户id
	ParentId  int    `gorm:"type:int" json:"parentId"`                          //父主键
	RoleName  string `gorm:"type:varchar(255)" json:"roleName"`                 //角色别名
	Sort      int    `gorm:"type:int" json:"sort"`                              //排序
	RoleAlias string `gorm:"type:varchar(255)" json:"roleAlias"`                //角色别名
	IsDeleted int    `gorm:"type:int" json:"isDeleted"`                         //是否删除
}

func (Role) TableName() string {
	return "role"
}

func (c *Role) Get() error {
	return Db.Model(&c).Where("id = ?", c.ID).Find(&c).Error
}

func (c *Role) GetRole() error {
	return Db.Model(&c).Where("id = ? and is_deleted = 0", c.ID).Find(&c).Error
}

func (c Role) GetRoles(offset, limit int) ([]Role, int, error) {
	var Roles []Role
	var counts int64
	db := Db.Model(&c)

	if c.RoleName != "" {
		db = db.Where("role_name like ?", "%"+c.RoleName+"%")
	}
	if c.RoleAlias != "" {
		db = db.Where("role_alias like ?", "%"+c.RoleAlias+"%")
	}
	if c.TenantId != "" {
		db = db.Where("tenant_id like ?", "%"+c.TenantId+"%")
	}
	err := db.Where("is_deleted = 0").Offset(offset).Limit(limit).Find(&Roles).Error
	db1 := Db.Model(&c)
	db1.Where("is_deleted = 0").Count(&counts)
	return Roles, int(counts), err
}

func (c *Role) Save() error {
	return Db.Model(&c).Save(&c).Error
}

func (c *Role) Update() error {
	return Db.Model(&c).Updates(&c).Error
}

func (c *Role) Remove() error {
	return Db.Model(&c).Updates(map[string]interface{}{"is_deleted": c.IsDeleted}).Error
}

func (c *Role) IsExistChild() bool {
	var count int64
	_ = Db.Model(&c).Where("parent_id = ? and is_deleted = 0", c.ParentId).Count(&count).Error
	return count > 0
}

func (c *Role) UpdatePwd(pwd string) error {
	return Db.Model(&c).Updates(map[string]interface{}{"password": pwd}).Error
}

func (c *Role) GetAll() ([]Role, error) {
	var Roles []Role
	err := Db.Model(&c).Where("is_deleted = 0").Find(&Roles).Error
	return Roles, err
}

func (c *Role) UpdateRoles(RoleIds []string, roleIds string) error {
	err := Db.Model(&c).Where("id in ?", RoleIds).Updates(map[string]interface{}{"role_id": roleIds}).Error
	return err
}

func (c *Role) GetByParentId() ([]Role, error) {
	var Roles []Role
	err := Db.Model(&c).Where("is_deleted = 0 AND parent_id = ?", c.ParentId).Find(&Roles).Error
	return Roles, err
}