✨ init project

service/example/exa_attachment_category.go

@@ -0,0 +1,66 @@
+package example
+
+import (
+	"errors"
+	"git.echol.cn/loser/xiecheng_server/global"
+	"git.echol.cn/loser/xiecheng_server/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
+}