From 7079e327defe3cf1bf278c078c3d7762bb9ac6aa Mon Sep 17 00:00:00 2001 From: truthhun <1272881215@qq.com> Date: Mon, 24 Jul 2023 10:33:06 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8F=98=E6=9B=B4=E5=88=86?= =?UTF-8?q?=E7=B1=BB=E5=B1=82=E7=BA=A7=E4=B9=8B=E5=90=8E=EF=BC=8C=E6=96=87?= =?UTF-8?q?=E6=A1=A3=E5=88=86=E7=B1=BB=E5=87=BA=E7=8E=B0=E9=94=99=E4=B9=B1?= =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model/category.go | 111 +++++++++++++++++++++++++- web/pages/admin/document/category.vue | 10 +-- 2 files changed, 113 insertions(+), 8 deletions(-) diff --git a/model/category.go b/model/category.go index fb4a7c4..1f756fd 100644 --- a/model/category.go +++ b/model/category.go @@ -11,7 +11,7 @@ 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 int `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;"` + 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:排序，值越大越靠前;"` @@ -41,7 +41,95 @@ func (m *DBModel) CreateCategory(category *Category) (err error) { // UpdateCategory 更新Category，如果需要更新指定字段，则请指定updateFields参数 func (m *DBModel) UpdateCategory(category *Category, updateFields ...string) (err error) { - db := m.db.Model(category) + 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 { // 更新指定字段 @@ -53,6 +141,7 @@ func (m *DBModel) UpdateCategory(category *Category, updateFields ...string) (er err = db.Where("id = ?", category.Id).Updates(category).Error if err != nil { m.logger.Error("UpdateCategory", zap.Error(err)) + return } if category.Cover != "" { @@ -79,7 +168,7 @@ func (m *DBModel) GetCategory(id interface{}, fields ...string) (category Catego } // GetCategoryByParentIdTitle(parentId int, title string, fields ...string) 根据唯一索引获取Category -func (m *DBModel) GetCategoryByParentIdTitle(parentId int, title string, fields ...string) (category Category, err error) { +func (m *DBModel) GetCategoryByParentIdTitle(parentId int64, title string, fields ...string) (category Category, err error) { db := m.db fields = m.FilterValidFields(Category{}.TableName(), fields...) @@ -177,3 +266,19 @@ func (m *DBModel) CountCategory() (count int64, err error) { } 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 +} diff --git a/web/pages/admin/document/category.vue b/web/pages/admin/document/category.vue index dfa3e3f..0d54bcd 100644 --- a/web/pages/admin/document/category.vue +++ b/web/pages/admin/document/category.vue @@ -131,9 +131,9 @@ export default { this.search = { ...this.search, ...search, page: 1 } if ( location.pathname + location.search === -        this.$router.resolve({ -          query: this.search, -        }).href + this.$router.resolve({ + query: this.search, + }).href ) { this.listCategory() } else { @@ -240,7 +240,7 @@ export default { } this.tableListFields = [ { prop: 'title', label: '名称', minWidth: 120, fixed: 'left' }, - { prop: 'icon', label: '图标', width: 48, type: 'image' }, + { prop: 'id', label: 'ID', width: 80, type: 'number', fixed: 'left' }, { prop: 'enable', label: '是否启用', @@ -259,9 +259,9 @@ export default { width: 80, type: 'number', }, + { prop: 'icon', label: '图标', width: 48, type: 'image' }, { prop: 'cover', label: '封面', width: 100, type: 'image' }, { prop: 'doc_count', label: '文档数', width: 80, type: 'number' }, - { prop: 'id', label: 'ID', width: 80, type: 'number' }, { prop: 'description', label: '分类描述', minWidth: 200 }, { prop: 'created_at', label: '创建时间', width: 160, type: 'datetime' }, { prop: 'updated_at', label: '更新时间', width: 160, type: 'datetime' },