eta_api/models/ai_predict_model/ai_predict_model_classify.go

package data_manage

import (
	sql2 "database/sql"
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"fmt"
	"strings"
	"time"
)

// AiPredictModelClassify 同花顺高频数据-分类
type AiPredictModelClassify struct {
	AiPredictModelClassifyId int       `gorm:"primaryKey;column:ai_predict_model_classify_id;"`
	UniqueCode               string    `description:"唯一编码"`
	ClassifyName             string    `description:"分类名称"`
	ClassifyNameEn           string    `description:"英文分类名称"`
	ParentId                 int       `description:"父级ID"`
	Level                    int       `description:"层级"`
	LevelPath                string    `description:"层级路径"`
	Sort                     int       `description:"排序"`
	RootId                   int       `description:"顶级分类ID"`
	SysUserId                int       `description:"创建人ID"`
	SysUserRealName          string    `description:"创建人姓名"`
	CreateTime               time.Time `description:"创建时间"`
	ModifyTime               time.Time `description:"修改时间"`
}

func (m *AiPredictModelClassify) TableName() string {
	return "ai_predict_model_classify"
}

type AiPredictModelClassifyCols struct {
	PrimaryId       string
	UniqueCode      string
	ClassifyName    string
	ClassifyNameEn  string
	ParentId        string
	Level           string
	LevelPath       string
	Sort            string
	RootId          string
	SysUserId       string
	SysUserRealName string
	CreateTime      string
	ModifyTime      string
}

func (m *AiPredictModelClassify) Cols() AiPredictModelClassifyCols {
	return AiPredictModelClassifyCols{
		PrimaryId:       "ai_predict_model_classify_id",
		UniqueCode:      "unique_code",
		ClassifyName:    "classify_name",
		ClassifyNameEn:  "classify_name_en",
		ParentId:        "parent_id",
		Level:           "level",
		LevelPath:       "level_path",
		Sort:            "sort",
		RootId:          "root_id",
		SysUserId:       "sys_user_id",
		SysUserRealName: "sys_user_real_name",
		CreateTime:      "create_time",
		ModifyTime:      "modify_time",
	}
}

func (m *AiPredictModelClassify) Create() (err error) {
	err = global.DbMap[utils.DbNameIndex].Create(m).Error
	return
}

func (m *AiPredictModelClassify) CreateMulti(items []*AiPredictModelClassify) (err error) {
	if len(items) == 0 {
		return
	}
	err = global.DbMap[utils.DbNameIndex].CreateInBatches(items, utils.MultiAddNum).Error
	return
}

func (m *AiPredictModelClassify) Update(cols []string) (err error) {
	err = global.DbMap[utils.DbNameIndex].Select(cols).Updates(m).Error
	return
}

func (m *AiPredictModelClassify) Remove() (err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
	err = o.Exec(sql, m.AiPredictModelClassifyId).Error
	return
}

func (m *AiPredictModelClassify) MultiRemove(ids []int) (err error) {
	if len(ids) == 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.Cols().PrimaryId, utils.GetOrmInReplace(len(ids)))
	err = o.Exec(sql, ids).Error
	return
}

func (m *AiPredictModelClassify) RemoveByCondition(condition string, pars []interface{}) (err error) {
	if condition == "" {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s`, m.TableName(), condition)
	err = o.Exec(sql, pars...).Error
	return
}

func (m *AiPredictModelClassify) GetItemById(id int) (item *AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
	err = o.Raw(sql, id).First(&item).Error
	return
}

func (m *AiPredictModelClassify) GetItemByCondition(condition string, pars []interface{}, orderRule string) (item *AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	order := ``
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s %s LIMIT 1`, m.TableName(), condition, order)
	err = o.Raw(sql, pars...).First(&item).Error
	return
}

func (m *AiPredictModelClassify) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

func (m *AiPredictModelClassify) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

func (m *AiPredictModelClassify) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
	pars = append(pars, startSize, pageSize)
	err = o.Exec(sql, pars...).Find(&items).Error
	return
}

// AiPredictModelClassifyItem 同花顺高频数据信息
type AiPredictModelClassifyItem struct {
	ClassifyId     int                           `description:"分类ID"`
	ClassifyName   string                        `description:"分类名称"`
	ClassifyNameEn string                        `description:"英文分类名称"`
	ParentId       int                           `description:"父级ID"`
	Level          int                           `description:"层级"`
	Sort           int                           `description:"排序"`
	LevelPath      string                        `description:"层级路径"`
	UniqueCode     string                        `description:"唯一编码"`
	Children       []*AiPredictModelClassifyItem `description:"子分类"`
}

func (m *AiPredictModelClassify) Format2Item() (item *AiPredictModelClassifyItem) {
	item = new(AiPredictModelClassifyItem)
	item.ClassifyId = m.AiPredictModelClassifyId
	item.ClassifyName = m.ClassifyName
	item.ClassifyNameEn = m.ClassifyNameEn
	item.ParentId = m.ParentId
	item.Level = m.Level
	item.Sort = m.Sort
	item.LevelPath = m.LevelPath
	item.UniqueCode = m.UniqueCode
	item.Children = make([]*AiPredictModelClassifyItem, 0)
	return
}

type AiPredictModelClassifyAddReq struct {
	ClassifyName string `description:"分类名称"`
	ParentId     int    `description:"父级ID, 第一级传0"`
	Level        int    `description:"层级, 第一级传0, 其余传上一级的层级"`
}

type AiPredictModelClassifyEditReq struct {
	ClassifyId   int    `description:"分类ID"`
	ClassifyName string `description:"分类名称"`
}

func (m *AiPredictModelClassify) GetSortMax(parentId int) (sort int, err error) {
	var sortNull sql2.NullInt64
	sql := fmt.Sprintf(`SELECT MAX(%s) FROM %s WHERE %s = ?`, m.Cols().Sort, m.TableName(), m.Cols().ParentId)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId).Scan(&sortNull).Error
	if err == nil && sortNull.Valid {
		sort = int(sortNull.Int64)
	}

	return
}

type AiPredictModelClassifyRemoveReq struct {
	ClassifyId int `description:"分类ID"`
	IndexId    int `description:"标的ID"`
}

type AiPredictModelClassifyListResp struct {
	AllNodes []*AiPredictModelClassifyListItem `description:"分类节点"`
}

type AiPredictModelClassifyListItem struct {
	NodeType     int                               `description:"节点类型: 0-分类; 1-指标"`
	NodeName     string                            `description:"节点名称"`
	ClassifyId   int                               `description:"分类ID"`
	ClassifyName string                            `description:"分类名称"`
	IndexId      int                               `description:"指标ID"`
	IndexCode    string                            `description:"指标编码"`
	IndexName    string                            `description:"指标名称"`
	ParentId     int                               `description:"父级ID"`
	Level        int                               `description:"层级"`
	Sort         int                               `description:"排序"`
	UniqueCode   string                            `description:"唯一编码, 指标为IndexCode"`
	Children     []*AiPredictModelClassifyListItem `description:"子分类"`
}

type AiPredictModelClassifyMoveReq struct {
	ClassifyId       int `description:"分类ID"`
	ParentClassifyId int `description:"父级分类ID"`
	PrevClassifyId   int `description:"上一个兄弟节点分类ID"`
	NextClassifyId   int `description:"下一个兄弟节点分类ID"`
	ItemId           int `description:"指标ID, 如果指标ID有值，则移动对象为指标，否则认为移动对象为分类"`
	PrevItemId       int `description:"上一个指标ID"`
	NextItemId       int `description:"下一个指标ID"`
}

func GetAiPredictModelClassifyById(classifyId int) (item *AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := `SELECT * FROM ai_predict_model_classify WHERE ai_predict_model_classify_id = ?`
	err = o.Raw(sql, classifyId).First(&item).Error
	return
}

func GetAiPredictModelClassifyByRootIdLevel(rootId int, orderStr string) (items []*AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM ai_predict_model_classify WHERE root_id = ? `
	if orderStr != "" {
		sql += orderStr
	} else {
		sql += ` order by level desc, sort asc, ai_predict_model_classify_id asc`
	}
	err = o.Raw(sql, rootId).Find(&items).Error
	return
}

// UpdateAiPredictModelClassifySortByParentId 根据父类id更新排序
func UpdateAiPredictModelClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` update ai_predict_model_classify set sort = ` + updateSort + ` WHERE parent_id = ? AND sort > ? `
	if classifyId > 0 {
		sql += ` or ( ai_predict_model_classify_id > ` + fmt.Sprint(classifyId) + ` and sort = ` + fmt.Sprint(nowSort) + `)`
	}
	err = o.Exec(sql, parentId, nowSort).Error
	return
}

// GetFirstAiPredictModelClassifyByParentId 获取当前父级分类下，且排序数相同 的排序第一条的数据
func GetFirstAiPredictModelClassifyByParentId(parentId int) (item *AiPredictModelClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM ai_predict_model_classify WHERE parent_id = ? order by sort asc,ai_predict_model_classify_id asc limit 1`
	err = o.Raw(sql, parentId).First(&item).Error
	return
}

func UpdateAiPredictModelClassifyChildByParentClassifyId(classifyIds []int, rootId int, levelStep int) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	var pars []interface{}
	pars = append(pars, rootId, levelStep)
	pars = append(pars, classifyIds)
	// 更新相关联的二级分类的parentId,和classify_name_second
	sql := `UPDATE ai_predict_model_classify SET root_id = ?, level = level+? where ai_predict_model_classify_id IN (` + utils.GetOrmInReplace(len(classifyIds)) + `)`
	err = o.Exec(sql, pars...).Error
	if err != nil {
		return
	}
	return
}

// GetAiPredictModelIndexCountByClassifyId 获取目录下(包含子目录)的指标数量
func GetAiPredictModelIndexCountByClassifyId(classifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	var pars []interface{}
	var sql string
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE ai_predict_model_classify_cte (ai_predict_model_classify_id, parent_id) AS (
			SELECT ai_predict_model_classify_id, parent_id
			FROM ai_predict_model_classify
			WHERE parent_id = ?
			UNION ALL
			SELECT c.ai_predict_model_classify_id, c.parent_id
			FROM ai_predict_model_classify c
			INNER JOIN ai_predict_model_classify_cte ct ON c.parent_id = ct.ai_predict_model_classify_id
		)
		SELECT COUNT(1) AS count
		FROM ai_predict_model_index a
			WHERE a.classify_id IN (
				SELECT ai_predict_model_classify_id
				FROM ai_predict_model_classify_cte
				UNION
				SELECT ai_predict_model_classify_id
				FROM ai_predict_model_classify
				WHERE ai_predict_model_classify_id = ?
			)`
		pars = append(pars, classifyId, classifyId, classifyId)
	}else{
		sql = `SELECT COUNT(1) AS count FROM ai_predict_model_index AS a
			WHERE a.classify_id IN(
			SELECT t.ai_predict_model_classify_id FROM 
			(
			SELECT rd.*
			FROM (SELECT * FROM ai_predict_model_classify WHERE parent_id IS NOT NULL) rd,
				(SELECT @pid := ?) pd 
			WHERE FIND_IN_SET(parent_id, @pid) > 0 
			AND @pid := CONCAT(@pid, ',', ai_predict_model_classify_id) 
			UNION SELECT * FROM ai_predict_model_classify WHERE ai_predict_model_classify_id = @pid
			)AS t
		)`
		pars = append(pars, classifyId)
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

// GetAiPredictModelClassifyCountByClassifyId 获取目录下子目录数量
func GetAiPredictModelClassifyCountByClassifyId(chartClassifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	var pars []interface{}
	var sql string
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE ai_predict_model_classify_cte (ai_predict_model_classify_id, parent_id) AS (
			SELECT ai_predict_model_classify_id, parent_id
			FROM ai_predict_model_classify
			WHERE parent_id = ?
			UNION ALL
			SELECT c.ai_predict_model_classify_id, c.parent_id
			FROM ai_predict_model_classify c
			INNER JOIN ai_predict_model_classify_cte ct ON c.parent_id = ct.ai_predict_model_classify_id
		)
		SELECT COUNT(1) AS count
		FROM (
			SELECT ai_predict_model_classify_id
			FROM ai_predict_model_classify_cte
			UNION
			SELECT ai_predict_model_classify_id
			FROM ai_predict_model_classify
			WHERE ai_predict_model_classify_id = ?
		) AS t
		WHERE t.ai_predict_model_classify_id <> ?`
	pars = append(pars, chartClassifyId, chartClassifyId, chartClassifyId)
	}else{
		sql = `SELECT COUNT(1) AS count FROM (
			SELECT rd.*
			FROM (SELECT * FROM ai_predict_model_classify WHERE parent_id IS NOT NULL) rd,
				 (SELECT @pid := ?) pd 
			WHERE FIND_IN_SET(parent_id, @pid) > 0 
			  AND @pid := CONCAT(@pid, ',', ai_predict_model_classify_id) 
			UNION SELECT * FROM ai_predict_model_classify WHERE ai_predict_model_classify_id = @pid
			)AS t
			WHERE t.ai_predict_model_classify_id <> ?`
		pars = append(pars, chartClassifyId, chartClassifyId)
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

// RemoveAiPredictModelClassify 删除分类及子分类
func RemoveAiPredictModelClassify(classifyId int) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	var pars []interface{}
	var sql string
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE ai_predict_model_classify_cte (ai_predict_model_classify_id, parent_id) AS (
				SELECT ai_predict_model_classify_id, parent_id
				FROM ai_predict_model_classify
				WHERE parent_id = ?
				UNION ALL
				SELECT c.ai_predict_model_classify_id, c.parent_id
				FROM ai_predict_model_classify c
				INNER JOIN ai_predict_model_classify_cte ct ON c.parent_id = ct.ai_predict_model_classify_id
			)
			DELETE FROM ai_predict_model_classify
			WHERE ai_predict_model_classify_id IN (
				SELECT ai_predict_model_classify_id
				FROM ai_predict_model_classify_cte
				UNION
				SELECT ai_predict_model_classify_id
				FROM ai_predict_model_classify
				WHERE ai_predict_model_classify_id = ?
			)`
		pars = append(pars, classifyId, classifyId)
	}else{
		sql = `DELETE FROM ai_predict_model_classify
				WHERE ai_predict_model_classify_id IN(
				SELECT t.ai_predict_model_classify_id FROM
				(
				SELECT rd.*
				FROM (SELECT * FROM ai_predict_model_classify WHERE parent_id IS NOT NULL) rd,
				(SELECT @pid := ?) pd
				WHERE FIND_IN_SET(parent_id, @pid) > 0
				AND @pid := CONCAT(@pid, ',', ai_predict_model_classify_id)
				UNION SELECT * FROM ai_predict_model_classify WHERE ai_predict_model_classify_id = @pid
				)AS t
			)`
		pars = append(pars, classifyId)
	}
	err = o.Exec(sql, pars...).Error
	return
}

// GetAiPredictClassifyByParentIdAndName 根据父级ID和名称获取分类
func GetAiPredictClassifyByParentIdAndName(parentId int, classifyName string, classifyId int) (item *AiPredictModelClassify, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := `SELECT * FROM ai_predict_model_classify WHERE parent_id = ? AND classify_name = ? AND ai_predict_model_classify_id <> ? LIMIT 1`
	//err = o.Raw(sql, parentId, classifyName, classifyId).QueryRow(&item)
	err = global.DbMap[utils.DbNameIndex].Raw(sql, parentId, classifyName, classifyId).First(&item).Error
	return
}