eta_mini_ht_api/models/config/permission.go

package config

import (
	"eta/eta_mini_ht_api/models"
	"gorm.io/gorm"
	"gorm.io/gorm/clause"
	"time"
)

const (
	Columns = "permission_id,name,parent_id,risk_level,sort"
)

// Permission 定义权限表的结构
type Permission struct {
	Id           int       `gorm:"primaryKey;autoIncrement"`
	PermissionId int       `gorm:"type:int"`
	Name         string    `gorm:"type:varchar(255)"`
	ParentId     int       `gorm:"default:null"`
	RiskLevel    string    `gorm:"type:varchar(10)"`
	Sort         int       `gorm:"default:null"`
	CreateTime   time.Time `gorm:"default:null"`
	UpdateTime   time.Time `gorm:"default:null;onUpdate:CURRENT_TIMESTAMP"`
}

// BeforeCreate 钩子用于在创建记录前设置 CreateTime
func (p *Permission) BeforeCreate(tx *gorm.DB) (err error) {
	p.CreateTime = time.Now()
	return
}

func BatchInsertOrUpdate(list []Permission) (err error) {
	db := models.Main()
	OnConflictFunc := clause.OnConflict{
		Columns:   []clause.Column{{Name: "permission_id"}},
		DoUpdates: clause.AssignmentColumns([]string{"name", "parent_id", "sort"}),
	}
	// 执行批量插入或更新操作
	err = db.Clauses(OnConflictFunc).Create(&list).Error
	return
}
func GetPermissionByName(name string) (permission Permission, err error) {
	db := models.Main()
	err = db.Model(&Permission{}).Select(Columns).Where("name = ?", name).First(&permission).Error
	return
}
func GetPermissionList() (list []Permission, err error) {
	db := models.Main()
	err = db.Model(&Permission{}).Select(Columns).Order("parent_id asc").Order("sort asc").Find(&list).Error
	return
}
func GetPermissionListByIds(permissionIds []int) (list []Permission, err error) {
	db := models.Main()
	//一级分类不能挂风险等级
	err = db.Model(&Permission{}).Select(Columns).Where("permission_id in ? and (risk_level !='' and risk_level is NOT NULL and parent_id !=0 )", permissionIds).Find(&list).Error
	return
}
func GetPermissionListWithRisk() (list []Permission, err error) {
	db := models.Main()
	//一级分类不能挂风险等级
	err = db.Model(&Permission{}).Select(Columns).Where("risk_level is NOT NULL and risk_level != '' and parent_id !=0").Find(&list).Error
	return
}