mh_goadmin_server/app/admin/models/category.go

package models

import (
	"fmt"
	"go-admin/app/admin/models/common"
	orm "go-admin/common/global"
	"go-admin/logger"
	"gorm.io/gorm"
)

type Category struct {
	Model

	Name                  string `json:"name"`                    // 分类名称
	Number                string `json:"number"`                  // 编号
	Display               int8   `json:"display"`                 // 1 展示 0 隐藏
	Pid                   uint32 `json:"pid" gorm:"index"`        // 父分类的编号
	CooperativeBusinessId uint32 `json:"cooperative_business_id"` // 合作商id
}

var (
	Display   = 1
	UnDisplay = 0
)

func (c *Category) TableName() string {
	return "erp_category"
}

func (c *Category) BeforeCreate(tx *gorm.DB) error {
	if c.Number == "" {
		n, err := GenerateNumber(c.CooperativeBusinessId, c.Pid)
		if err != nil {
			logger.Error("before create category err", logger.Field("c", c), logger.Field("err", err))
			return err
		}
		c.Number = n
	}
	return nil
}

// GenerateNumber 生成分类编码
func GenerateNumber(cid uint32, pid uint32) (string, error) {
	var count int64
	var err error

	err = orm.Eloquent.Model(Category{}).
		Unscoped().
		Scopes(common.ScopeBusiness(cid)).
		Where("pid", pid).
		Count(&count).
		Error
	if err != nil {
		return "", err
	}
	if pid == 0 {
		return fmt.Sprintf("%03d", count+1), nil
	} else {
		var parent Category
		err = orm.Eloquent.Model(Category{}).
			Scopes(common.ScopeBusiness(cid)).
			Where("id", pid).
			First(&parent).Error
		if err != nil {
			return "", err
		}
		return fmt.Sprintf("%s%03d", parent.Number, count+1), nil
	}
}

// GetCategoryById 通过id获取分类
func GetCategoryById(id uint32) (*Category, error) {
	//var c *Category
	c := new(Category)
	err := orm.Eloquent.Model(c).Where("id", id).First(c).Error
	if err != nil {
		return nil, err
	}

	return c, nil
}

func comparePrefix(a, b string) bool {
	// 获取字符串 a 的长度
	aLen := len(a)

	// 如果 b 的长度比 a 短，则无法比较
	if len(b) < aLen {
		return false
	}

	// 截取 b 的前 aLen 个字符
	bPrefix := b[:aLen]

	// 比较 a 和 b 的前缀部分
	return a == bPrefix
}

// CheckCouponCategory 判断优惠券分类和商品分类是否对应
func CheckCouponCategory(num string, id uint32) bool {
	categoryInfo, err := GetCategoryById(id)
	if err != nil || err == RecordNotFound {
		return false
	}

	return comparePrefix(num, categoryInfo.Number)
}

type CategoryModel struct {
	Category
	SubCategory []*CategoryModel `json:"sub_category" gorm:"-"` //子分类
}

func ScopeOnlyDisplay(db *gorm.DB) *gorm.DB {
	return db.Where("display", Display)
}

// GetCategoryList 分类列表
func GetCategoryList(bid uint32, all bool) ([]*CategoryModel, error) {
	var list = make([]*CategoryModel, 0)
	m := orm.Eloquent.Model(Category{})

	var scopes = []func(db *gorm.DB) *gorm.DB{common.ScopeBusiness(bid)}
	if !all {
		scopes = append(scopes, ScopeOnlyDisplay)
	}

	err := m.Scopes(scopes...).Where("pid = ?", 0).Find(&list).Error
	if err != nil {
		return nil, err
	}
	for _, top := range list {
		top.SubCategory = findChildCategory(top, all)
	}

	return list, nil
}

// 查询子分类
func findChildCategory(prev *CategoryModel, all bool) []*CategoryModel {
	var cs []*CategoryModel
	m := orm.Eloquent.Model(Category{})
	if !all {
		m = m.Scopes(ScopeOnlyDisplay)
	}
	err := m.Where("pid", prev.ID).Find(&cs).Error
	if err != nil {
		return nil
	}

	for _, c := range cs {
		c.SubCategory = findChildCategory(c, all)
	}

	return cs
}

// CheckCommodityByCategory 检查某个分类下是否有商品
func CheckCommodityByCategory(categoryId uint32) bool {
	var count int64
	err := orm.Eloquent.Table("erp_commodity").Where("erp_category_id = ? and count > 0", categoryId).
		Count(&count).Error
	if err != nil {
		logger.Error("CheckCommodityByCategory err:", logger.Field("err", err))
	}

	return count > 0
}

// IsExistingCategory 查询分类资料中某个名称的分类是否存在
func IsExistingCategory(categoryName string) bool {
	if categoryName == "" {
		return true
	}
	// 查询分类名称是否存在的逻辑
	var count int64
	orm.Eloquent.Debug().Model(&Category{}).
		Where("name = ?", categoryName).
		Count(&count)

	return count > 0
}

// GetSubcategoryIds 获取指定分类的所有子分类ID
func GetSubcategoryIds(categoryID uint32) ([]uint32, error) {
	var subCategories []Category
	err := orm.Eloquent.Model(&Category{}).Where("pid = ?", categoryID).Find(&subCategories).Error
	if err != nil {
		return nil, err
	}

	var subCategoryIDs []uint32
	for _, subCategory := range subCategories {
		subCategoryIDs = append(subCategoryIDs, subCategory.ID)
		// 递归获取子分类的子分类
		childIDs, err := GetSubcategoryIds(subCategory.ID)
		if err != nil {
			return nil, err
		}
		subCategoryIDs = append(subCategoryIDs, childIDs...)
	}

	return subCategoryIDs, nil
}

// TransformErpCategoryIds 转换 ErpCategoryId，将子分类ID添加进来并去重
func TransformErpCategoryIds(categoryIds []uint32) ([]uint32, error) {
	categoryIDSet := make(map[uint32]struct{}) // 用于去重的集合

	// 遍历传入的 ErpCategoryId
	for _, categoryID := range categoryIds {
		// 获取当前分类及其所有子分类ID
		subCategoryIDs, err := GetSubcategoryIds(categoryID)
		if err != nil {
			return nil, err
		}

		// 将当前分类ID添加到集合中
		categoryIDSet[categoryID] = struct{}{}
		// 将子分类ID添加到集合中
		for _, subID := range subCategoryIDs {
			categoryIDSet[subID] = struct{}{}
		}
	}

	// 将集合转换为切片
	var result []uint32
	for id := range categoryIDSet {
		result = append(result, id)
	}

	return result, nil
}