eta_api/models/data_manage/public_edb_classify.go

package data_manage

import (
	"eta_gn/eta_api/global"
	"fmt"
	"time"
)

// EdbPublicClassify
// @Description: 公共分类
type EdbPublicClassify struct {
	EdbPublicClassifyId       int       `gorm:"primaryKey" `
	ClassifyType              uint8     `description:"分类类型，0：普通指标分类，1：预测指标分类，2：指标加工的分类（普通指标的计算指标）"`
	EdbPublicClassifyName     string    `description:"分类名称"`
	ParentId                  int       `description:"父级id"`
	HasData                   int       `description:"是否含有指标数据"`
	RootId                    int       `description:"顶级id"`
	CreateTime                time.Time `description:"创建时间"`
	ModifyTime                time.Time `description:"修改时间"`
	SysUserId                 int       `description:"创建人id"`
	SysUserRealName           string    `description:"创建人姓名"`
	Level                     int       `description:"层级"`
	UniqueCode                string    `description:"唯一编码"`
	Sort                      int       `description:"排序字段，越小越靠前，默认值：10"`
	LastModifyUserId          int       `description:"最后修改人ID"`
	LastModifyUserRealName    string    `description:"最后修改人姓名"`
	EdbPublicClassifyNamePath string    `description:"分类名称的完整路径，格式为：父级分类名称|当前分类名称"`
	EdbPublicClassifyIdPath   string    `description:"分类的完整路径，格式为：父级ID,当前ID"`
}

func (m EdbPublicClassify) ToEdbClassifyItems() *EdbClassifyItems {
	return &EdbClassifyItems{
		ClassifyId: m.EdbPublicClassifyId,
		//EdbInfoId:        0,
		ClassifyType:   m.ClassifyType,
		ClassifyName:   m.EdbPublicClassifyName,
		ClassifyNameEn: m.EdbPublicClassifyName,
		ParentId:       m.ParentId,
		RootId:         m.RootId,
		Level:          m.Level,
		Sort:           m.Sort,
		UniqueCode:     m.UniqueCode,
		//Source:           0,
		//SourceName:       "",
		SysUserId:       m.SysUserId,
		SysUserRealName: m.SysUserRealName,
		//StartDate:        "",
		//EdbCode:          "",
		//EdbType:          0,
		//Children:         nil,
		//Button:           EdbClassifyItemsButton{},
		//IsJoinPermission: 0,
		//HaveOperaAuth:    false,
		ClassifyIdPath: m.EdbPublicClassifyIdPath,
	}
}

// Add
// @Description: 添加分类
// @author: Roc
// @receiver m
// @datetime 2024-12-04 17:00:18
// @return err error
func (m *EdbPublicClassify) Add() (err error) {
	err = global.DmSQL["data"].Create(m).Error

	return
}

// Update
// @Description: 更新分类基础信息
// @author: Roc
// @receiver m
// @datetime 2024-12-04 17:01:51
// @param cols []string
// @return err error
func (m *EdbPublicClassify) Update(cols []string) (err error) {
	err = global.DmSQL["data"].Select(cols).Updates(m).Error

	return
}

// GetEdbClassifyById
// @Description: 根据id获取分类
// @author: Roc
// @receiver m
// @datetime 2024-12-04 16:56:15
// @param classifyId int
// @return item *EdbPublicClassify
// @return err error
func (m EdbPublicClassify) GetEdbClassifyById(classifyId int) (item *EdbPublicClassify, err error) {
	sql := `SELECT * FROM edb_public_classify WHERE edb_public_classify_id=? `
	err = global.DmSQL["data"].Raw(sql, classifyId).First(&item).Error

	return
}

// GetEdbClassifyListByParentId
// @Description: 根据父级id获取下级分类列表
// @author: Roc
// @receiver m
// @datetime 2024-12-03 13:57:48
// @param parentId int
// @param classifyType int8
// @return items []*EdbPublicClassify
// @return err error
func (m EdbPublicClassify) GetEdbClassifyListByParentId(parentId int, classifyType int8) (items []*EdbPublicClassify, err error) {
	o := global.DmSQL["data"]
	sql := `SELECT * FROM edb_public_classify WHERE parent_id=? AND classify_type = ? order by sort asc,edb_public_classify_id asc `
	err = o.Raw(sql, parentId, classifyType).Find(&items).Error

	return
}

// GetEdbClassifyItemsByParentId
// @Description: 根据父级id获取下级分类列表（结构型的）
// @author: Roc
// @receiver m
// @datetime 2024-12-03 13:56:18
// @param parentId int
// @param classifyType int8
// @return items []*EdbClassifyItems
// @return err error
func (m EdbPublicClassify) GetEdbClassifyItemsByParentId(parentId int, classifyType int8) (items []*EdbClassifyItems, err error) {
	list, err := m.GetEdbClassifyListByParentId(parentId, classifyType)
	if err != nil {
		return
	}

	for _, v := range list {
		items = append(items, v.ToEdbClassifyItems())
	}

	return
}

// GetEdbClassifyListByParentId
// @Description: 根据父级id获取下级分类列表
// @author: Roc
// @receiver m
// @datetime 2024-12-03 13:57:48
// @param parentId int
// @param classifyType int8
// @return items []*EdbPublicClassify
// @return err error
func (m EdbPublicClassify) GetEdbClassifyListByType(classifyType int8) (items []*EdbPublicClassify, err error) {
	o := global.DmSQL["data"]
	sql := `SELECT * FROM edb_public_classify WHERE classify_type = ? order by sort asc,edb_public_classify_id asc `
	err = o.Raw(sql, classifyType).Find(&items).Error

	return
}

// GetAllEdbClassifyByType 根据类型获取分类列表
func (m EdbPublicClassify) GetAllEdbClassifyByType(classifyType int8) (items []*EdbClassifyItems, err error) {
	list, err := m.GetEdbClassifyListByType(int8(classifyType))
	if err != nil {
		return
	}

	for _, v := range list {
		items = append(items, v.ToEdbClassifyItems())
	}

	return
}

func (m EdbPublicClassify) GetEdbClassifyCount(classifyName string, parentId int, classifyType uint8) (count int, err error) {
	sql := `SELECT COUNT(1) AS count FROM edb_public_classify WHERE parent_id=? AND edb_public_classify_name=? AND classify_type = ?  `
	err = global.DmSQL["data"].Raw(sql, parentId, classifyName, classifyType).Scan(&count).Error

	return
}

// UpdateEdbClassifyNameAndNamePath
// @Description: 更新分类信息
// @author: Roc
// @receiver m
// @datetime 2024-12-05 09:27:33
// @param cols []string
// @param oldClassifyNamePath string
// @param newClassifyNamePath string
// @return err error
func (m *EdbPublicClassify) UpdateEdbClassifyNameAndNamePath(cols []string, oldClassifyNamePath, newClassifyNamePath string) (err error) {
	tx := global.DmSQL["data"].Begin()
	defer func() {
		if err != nil {
			_ = tx.Rollback()
			return
		}
		_ = tx.Commit()
	}()

	// 变更分类信息
	err = tx.Select(cols).Updates(m).Error
	if err != nil {
		return
	}

	// 更改子分类的完整的路径
	if oldClassifyNamePath != `` && newClassifyNamePath != `` {
		sql := `UPDATE edb_public_classify SET edb_public_classify_name_path = REPLACE(edb_public_classify_name_path,?,?) WHERE edb_public_classify_id_path LIKE ?`
		err = tx.Exec(sql, oldClassifyNamePath, newClassifyNamePath, fmt.Sprint(m.EdbPublicClassifyIdPath+`,%`)).Error
		if err != nil {
			return
		}
	}

	return
}

// GetEdbClassifyMaxSort 获取分类下最大的排序数
func (m EdbPublicClassify) GetEdbClassifyMaxSort(parentId int, classifyType uint8) (sort int, err error) {
	sql := `SELECT COALESCE(Max(sort), 0) AS sort FROM edb_public_classify WHERE parent_id=? AND classify_type=? `
	err = global.DmSQL["data"].Raw(sql, parentId, classifyType).Scan(&sort).Error

	return
}

// GetEdbInfoCountByClassifyIdList
// @Description: 根据公共分类id集合获取名下指标数量
// @author: Roc
// @receiver m
// @datetime 2024-12-05 09:59:56
// @param classifyIdList []int
// @return count int
// @return err error
func (m EdbPublicClassify) GetEdbInfoCountByClassifyIdList(classifyIdList []int) (count int, err error) {
	if len(classifyIdList) <= 0 {
		return
	}
	sql := `SELECT COUNT(1) total FROM edb_info  WHERE edb_public_classify_id IN (?) `
	err = global.DmSQL["data"].Raw(sql, classifyIdList).Scan(&count).Error

	return
}

func (m *EdbPublicClassify) GetAllChildClassifyIdList() (items []int, err error) {
	sql := ` SELECT edb_public_classify_id FROM edb_public_classify  WHERE edb_public_classify_id_path LIKE ?  ORDER BY create_time DESC `
	err = global.DmSQL["data"].Raw(sql, fmt.Sprint(m.EdbPublicClassifyIdPath+`,%`)).Scan(&items).Error

	return
}

// BatchesDel
// @Description: 根据分类id批量删除
// @author: Roc
// @receiver m
// @datetime 2024-12-05 11:13:06
// @param classifyIdList []int
// @return err error
func (m *EdbPublicClassify) BatchesDel(classifyIdList []int) (err error) {
	sql := ` DELETE FROM edb_public_classify  WHERE edb_public_classify_id IN (?) `
	err = global.DmSQL["data"].Exec(sql, classifyIdList).Error

	return
}