eta_api/models/data_manage/public_chart_classify.go

package data_manage

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

// ChartPublicClassify
// @Description: 公共分类
type ChartPublicClassify struct {
	ChartPublicClassifyId       int       `gorm:"primaryKey" `
	Source                      int       `description:"分类来源: 1-图库; 2-商品价格曲线; 3-相关性图表; 6-拟合方程图表; 7-统计特征"`
	ChartPublicClassifyName     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:"最后修改人姓名"`
	ChartPublicClassifyNamePath string    `description:"分类名称的完整路径，格式为：父级分类名称|当前分类名称"`
	ChartPublicClassifyIdPath   string    `description:"分类的完整路径，格式为：父级ID,当前ID"`
}

func (m ChartPublicClassify) ToChartClassifyItems() *ChartClassifyItems {
	return &ChartClassifyItems{
		ChartClassifyId: m.ChartPublicClassifyId,
		//ChartInfoId:         0,
		ChartClassifyName:   m.ChartPublicClassifyName,
		ChartClassifyNameEn: m.ChartPublicClassifyName,
		ParentId:            m.ParentId,
		Level:               m.Level,
		Sort:                m.Sort,
		UniqueCode:          m.UniqueCode,
		Source:              m.Source,
		//SourceName:          "",
		SysUserId:       m.SysUserId,
		SysUserRealName: m.SysUserRealName,
		//DateType:            0,
		//StartDate:           "",
		//EndDate:             "",
		//ChartType:           0,
		//Calendar:            "",
		//SeasonStartDate:     "",
		//SeasonEndDate:       "",
		Children:            nil,
		Button:              ChartClassifyItemsButton{},
		IsJoinPermission:    0,
		HaveOperaAuth:       false,
		ChartClassifyIdPath: m.ChartPublicClassifyIdPath,
	}
}

// Add
// @Description: 添加分类
// @author: Roc
// @receiver m
// @datetime 2024-12-04 17:00:18
// @return err error
func (m *ChartPublicClassify) 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 *ChartPublicClassify) Update(cols []string) (err error) {
	err = global.DmSQL["data"].Select(cols).Updates(m).Error

	return
}

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

	return
}

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

	return
}

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

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

	return
}

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

	return
}

// GetAllChartClassifyByType 根据类型获取分类列表
func (m ChartPublicClassify) GetAllChartClassifyByType(source int8) (items []*ChartClassifyItems, err error) {
	list, err := m.GetChartClassifyListByType(int8(source))
	if err != nil {
		return
	}

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

	return
}

func (m ChartPublicClassify) GetChartClassifyCount(classifyName string, parentId int, source int) (count int, err error) {
	sql := `SELECT COUNT(1) AS count FROM chart_public_classify WHERE parent_id=? AND chart_public_classify_name=? AND source = ?  `
	err = global.DmSQL["data"].Raw(sql, parentId, classifyName, source).Scan(&count).Error

	return
}

// UpdateChartClassifyNameAndNamePath
// @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 *ChartPublicClassify) UpdateChartClassifyNameAndNamePath(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 chart_public_classify SET chart_public_classify_name_path = REPLACE(chart_public_classify_name_path,?,?) WHERE chart_public_classify_id_path LIKE ?`
		err = tx.Exec(sql, oldClassifyNamePath, newClassifyNamePath, fmt.Sprint(m.ChartPublicClassifyIdPath+`,%`)).Error
		if err != nil {
			return
		}
	}

	return
}

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

	return
}

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

	return
}

func (m *ChartPublicClassify) GetAllChildClassifyIdList() (items []int, err error) {
	sql := ` SELECT chart_public_classify_id FROM chart_public_classify  WHERE chart_public_classify_id_path LIKE ?  ORDER BY create_time DESC `
	err = global.DmSQL["data"].Raw(sql, fmt.Sprint(m.ChartPublicClassifyIdPath+`,%`)).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 *ChartPublicClassify) BatchesDel(classifyIdList []int) (err error) {
	sql := ` DELETE FROM chart_public_classify  WHERE chart_public_classify_id IN (?) `
	err = global.DmSQL["data"].Exec(sql, classifyIdList).Error

	return
}

// GetChartPublicClassifyByParentIdAndName 根据父级ID和名称获取分类
func GetChartPublicClassifyByParentIdAndName(source, parentId int, classifyName string, classifyId int) (item *ChartClassify, err error) {
	o := global.DmSQL["data"]
	sql := `SELECT * FROM chart_public_classify WHERE source = ? AND parent_id = ? AND chart_public_classify_name = ? AND chart_public_classify_id <> ? LIMIT 1`
	err = o.Raw(sql, source, parentId, classifyName, classifyId).First(&item).Error

	return
}