67 lines
2.0 KiB
Go
67 lines
2.0 KiB
Go
package example
|
|
|
|
import (
|
|
"errors"
|
|
"git.echol.cn/loser/lckt/global"
|
|
"git.echol.cn/loser/lckt/model/example"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type AttachmentCategoryService struct{}
|
|
|
|
// AddCategory 创建/更新的分类
|
|
func (a *AttachmentCategoryService) AddCategory(req *example.ExaAttachmentCategory) (err error) {
|
|
// 检查是否已存在相同名称的分类
|
|
if (!errors.Is(global.GVA_DB.Take(&example.ExaAttachmentCategory{}, "name = ? and pid = ?", req.Name, req.Pid).Error, gorm.ErrRecordNotFound)) {
|
|
return errors.New("分类名称已存在")
|
|
}
|
|
if req.ID > 0 {
|
|
if err = global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Where("id = ?", req.ID).Updates(&example.ExaAttachmentCategory{
|
|
Name: req.Name,
|
|
Pid: req.Pid,
|
|
}).Error; err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if err = global.GVA_DB.Create(&example.ExaAttachmentCategory{
|
|
Name: req.Name,
|
|
Pid: req.Pid,
|
|
}).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DeleteCategory 删除分类
|
|
func (a *AttachmentCategoryService) DeleteCategory(id *int) error {
|
|
var childCount int64
|
|
global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Where("pid = ?", id).Count(&childCount)
|
|
if childCount > 0 {
|
|
return errors.New("请先删除子级")
|
|
}
|
|
return global.GVA_DB.Where("id = ?", id).Unscoped().Delete(&example.ExaAttachmentCategory{}).Error
|
|
}
|
|
|
|
// GetCategoryList 分类列表
|
|
func (a *AttachmentCategoryService) GetCategoryList() (res []*example.ExaAttachmentCategory, err error) {
|
|
var fileLists []example.ExaAttachmentCategory
|
|
err = global.GVA_DB.Model(&example.ExaAttachmentCategory{}).Find(&fileLists).Error
|
|
if err != nil {
|
|
return res, err
|
|
}
|
|
return a.getChildrenList(fileLists, 0), nil
|
|
}
|
|
|
|
// getChildrenList 子类
|
|
func (a *AttachmentCategoryService) getChildrenList(categories []example.ExaAttachmentCategory, parentID uint) []*example.ExaAttachmentCategory {
|
|
var tree []*example.ExaAttachmentCategory
|
|
for _, category := range categories {
|
|
if category.Pid == parentID {
|
|
category.Children = a.getChildrenList(categories, category.ID)
|
|
tree = append(tree, &category)
|
|
}
|
|
}
|
|
return tree
|
|
}
|