moredoc/model/category.go

package model

import (
	"time"

	"go.uber.org/zap"
	"gorm.io/gorm"
)

type Category struct {
	Id              int64      `form:"id" json:"id,omitempty" gorm:"primaryKey;autoIncrement;column:id;comment:;"`
	Icon            string     `form:"icon" json:"icon,omitempty" gorm:"column:icon;comment:分类图标;type:varchar(255);size:255;"`
	Cover           string     `form:"cover" json:"cover,omitempty" gorm:"column:cover;comment:分类封面;type:varchar(255);size:255;"`
	ParentId        int64      `form:"parent_id" json:"parent_id,omitempty" gorm:"column:parent_id;type:int(11);size:11;default:0;index:parent_id_title,unique;index:parent_id;comment:上级ID;"`
	Title           string     `form:"title" json:"title,omitempty" gorm:"column:title;type:varchar(64);size:64;index:parent_id_title,unique;comment:分类名称;"`
	DocCount        int        `form:"doc_count" json:"doc_count,omitempty" gorm:"column:doc_count;type:int(11);size:11;default:0;comment:文档统计;"`
	Sort            int        `form:"sort" json:"sort,omitempty" gorm:"column:sort;type:int(11);size:11;default:0;index:idx_sort;comment:排序，值越大越靠前;"`
	Enable          bool       `form:"enable" json:"enable,omitempty" gorm:"column:enable;type:tinyint(1);size:1;index:idx_enable;default:1;"`
	Description     string     `form:"description" json:"description,omitempty" gorm:"column:description;type:text;comment:分类描述;"`
	ShowDescription bool       `form:"show_description" json:"show_description,omitempty" gorm:"column:show_description;type:tinyint(1);size:1;default:0;comment:是否显示描述;"`
	CreatedAt       *time.Time `form:"created_at" json:"created_at,omitempty" gorm:"column:created_at;type:datetime;comment:创建时间;"`
	UpdatedAt       *time.Time `form:"updated_at" json:"updated_at,omitempty" gorm:"column:updated_at;type:datetime;comment:更新时间;"`
}

func (Category) TableName() string {
	return tablePrefix + "category"
}

// CreateCategory 创建Category
func (m *DBModel) CreateCategory(category *Category) (err error) {
	err = m.db.Create(category).Error
	if err != nil {
		m.logger.Error("CreateCategory", zap.Error(err))
		return
	}
	if category.Cover != "" {
		m.setAttachmentType(AttachmentTypeCategoryCover, category.Id, []string{category.Cover})
	}
	return
}

// UpdateCategory 更新Category，如果需要更新指定字段，则请指定updateFields参数
func (m *DBModel) UpdateCategory(category *Category, updateFields ...string) (err error) {
	var existCategory Category
	err = m.db.Where("id = ?", category.Id).Find(&existCategory).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		m.logger.Error("UpdateCategory", zap.Error(err))
		return
	}

	tx := m.db.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
			return
		}
		tx.Commit()
	}()

	// 如果更新了父分类，则需要更新文档统计
	if existCategory.ParentId != category.ParentId {
		var (
			oldParentIds, newParentIds, documentIds []int64
			docCates                                []DocumentCategory
			limit                                   = 100
			page                                    = 1
			modelDocCate                            = &DocumentCategory{}
			modelCate                               = &Category{}
		)

		oldParentIds = m.GetCategoryParentIds(existCategory.Id)
		if category.ParentId > 0 {
			newParentIds = append(newParentIds, category.ParentId)
			newParentIds = append(newParentIds, m.GetCategoryParentIds(category.ParentId)...)
		}

		// 获取需要更新的文档ID
		for {
			m.db.Model(modelDocCate).Select("document_id", "category_id").Where("category_id = ?", existCategory.Id).Limit(limit).Offset((page - 1) * limit).Find(&docCates)
			if len(docCates) == 0 {
				break
			}
			page++
			for _, v := range docCates {
				documentIds = append(documentIds, v.DocumentId)
			}

			if len(oldParentIds) > 0 {
				// 删除旧的分类关联
				err = tx.Model(modelDocCate).Where("category_id in (?) and document_id in (?)", oldParentIds, documentIds).Delete(modelDocCate).Error
				if err != nil {
					m.logger.Error("UpdateCategory", zap.Error(err))
					return
				}

				// 更新旧的分类统计
				err = tx.Model(modelCate).Where("id in (?)", oldParentIds).Update("doc_count", gorm.Expr("doc_count - ?", len(documentIds))).Error
				if err != nil {
					m.logger.Error("UpdateCategory", zap.Error(err))
					return
				}
			}

			if len(newParentIds) > 0 {
				var newDocCates []DocumentCategory
				for _, documentId := range documentIds {
					for _, newCateId := range newParentIds {
						newDocCates = append(newDocCates, DocumentCategory{
							DocumentId: documentId,
							CategoryId: newCateId,
						})
					}
				}

				// 创建新的分类关联
				err = tx.Create(&newDocCates).Error
				if err != nil {
					m.logger.Error("UpdateCategory", zap.Error(err))
					return
				}

				// 更新新的分类统计
				err = tx.Model(modelCate).Where("id in (?)", newParentIds).Update("doc_count", gorm.Expr("doc_count + ?", len(documentIds))).Error
				if err != nil {
					m.logger.Error("UpdateCategory", zap.Error(err))
					return
				}
			}
		}
	}

	db := tx.Model(category)

	updateFields = m.FilterValidFields(Category{}.TableName(), updateFields...)
	if len(updateFields) > 0 { // 更新指定字段
		db = db.Select(updateFields)
	} else {
		db = db.Select(m.GetTableFields(Category{}.TableName()))
	}

	err = db.Where("id = ?", category.Id).Updates(category).Error
	if err != nil {
		m.logger.Error("UpdateCategory", zap.Error(err))
		return
	}

	if category.Cover != "" {
		m.setAttachmentType(AttachmentTypeCategoryCover, category.Id, []string{category.Cover})
	}
	if category.Icon != "" {
		m.setAttachmentType(AttachmentTypeCategoryCover, category.Id, []string{category.Icon})
	}

	return
}

// GetCategory 根据id获取Category
func (m *DBModel) GetCategory(id interface{}, fields ...string) (category Category, err error) {
	db := m.db

	fields = m.FilterValidFields(Category{}.TableName(), fields...)
	if len(fields) > 0 {
		db = db.Select(fields)
	}

	err = db.Where("id = ?", id).First(&category).Error
	return
}

// GetCategoryByParentIdTitle(parentId int, title string, fields ...string) 根据唯一索引获取Category
func (m *DBModel) GetCategoryByParentIdTitle(parentId int64, title string, fields ...string) (category Category, err error) {
	db := m.db

	fields = m.FilterValidFields(Category{}.TableName(), fields...)
	if len(fields) > 0 {
		db = db.Select(fields)
	}

	db = db.Where("parent_id = ?", parentId)

	db = db.Where("title = ?", title)

	err = db.First(&category).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		m.logger.Error("GetCategoryByParentIdTitle", zap.Error(err))
		return
	}
	return
}

type OptionGetCategoryList struct {
	Page         int
	Size         int
	WithCount    bool                      // 是否返回总数
	Ids          []interface{}             // id列表
	SelectFields []string                  // 查询字段
	QueryRange   map[string][2]interface{} // map[field][]{min,max}
	QueryIn      map[string][]interface{}  // map[field][]{value1,value2,...}
	QueryLike    map[string][]interface{}  // map[field][]{value1,value2,...}
}

// GetCategoryList 获取Category列表
func (m *DBModel) GetCategoryList(opt *OptionGetCategoryList) (categoryList []Category, total int64, err error) {
	db := m.db.Model(&Category{})
	tableName := Category{}.TableName()

	db = m.generateQueryIn(db, tableName, opt.QueryIn)
	db = m.generateQueryLike(db, tableName, opt.QueryLike)

	if len(opt.Ids) > 0 {
		db = db.Where("id in (?)", opt.Ids)
	}

	if opt.WithCount {
		err = db.Count(&total).Error
		if err != nil {
			m.logger.Error("GetCategoryList", zap.Error(err))
			return
		}
	}

	opt.SelectFields = m.FilterValidFields(tableName, opt.SelectFields...)
	if len(opt.SelectFields) > 0 {
		db = db.Select(opt.SelectFields)
	}

	db = db.Offset((opt.Page - 1) * opt.Size).Limit(opt.Size)

	err = db.Order("parent_id asc, sort desc, title asc").Find(&categoryList).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		m.logger.Error("GetCategoryList", zap.Error(err))
	}
	return
}

// DeleteCategory 删除数据
// 分类下存在文档的分类，则不允许删除
// 查找子分类，如果子分类下存在文档，则不允许删除
func (m *DBModel) DeleteCategory(ids []int64) (err error) {
	var (
		children    []Category
		childrenIds []int64
	)

	m.db.Select("id").Where("parent_id in (?) and doc_count = ?", ids, 0).Find(&children)
	if len(children) > 0 {
		for _, v := range children {
			childrenIds = append(childrenIds, v.Id)
		}
		if err = m.DeleteCategory(childrenIds); err != nil {
			return
		}
	}

	err = m.db.Where("id in (?) and doc_count = ?", ids, 0).Delete(&Category{}).Error
	if err != nil {
		m.logger.Error("DeleteCategory", zap.Error(err))
	}
	return
}

func (m *DBModel) CountCategory() (count int64, err error) {
	err = m.db.Model(&Category{}).Count(&count).Error
	if err != nil {
		m.logger.Error("CountCategory", zap.Error(err))
	}
	return
}

// 查询指定分类ID的所有父级分类ID
func (m *DBModel) GetCategoryParentIds(id int64) (ids []int64) {
	var category Category
	err := m.db.Model(&Category{}).Select("id", "parent_id").Where("id = ?", id).Find(&category).Error
	if err != nil && err != gorm.ErrRecordNotFound {
		m.logger.Error("GetCategoryParentIds", zap.Error(err), zap.Int64("id", id))
		return
	}

	if category.ParentId > 0 {
		ids = append(ids, category.ParentId)
		ids = append(ids, m.GetCategoryParentIds(category.ParentId)...)
	}
	return
}