eta_api/models/ai_summary/ai_summary_classify.go

package ai_summary

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

	"github.com/rdlucklib/rdluck_tools/paging"
)

type AiSummaryClassify struct {
	AiSummaryClassifyId int       `orm:"column(ai_summary_classify_id);pk" gorm:"primaryKey"` // ai纪要分类id
	ClassifyName        string    // 分类名称
	ParentId            int       // 父级id
	CreateTime          time.Time // 创建时间
	ModifyTime          time.Time // 修改时间
	SysUserId           int       // 创建人id，使用指针表示可为空
	SysUserRealName     string    // 创建人姓名
	Level               int       // 层级
	Sort                int       // 排序字段，越小越靠前，默认值：10
	RootId              int       // 顶级ID
	HasData             int       `description:"是否含有指标数据"`
}

func AddAiSummaryClassify(item *AiSummaryClassify) (lastId int64, err error) {
	o := global.DbMap[utils.DbNameMaster]
	err = o.Create(item).Error
	if err != nil {
		return
	}
	lastId = int64(item.AiSummaryClassifyId)
	return
}

type AiSummaryClassifyItems struct {
	AiSummaryClassifyId int
	Title               string                    `description:"标题" json:"-"`
	ClassifyName        string                    `description:"分类名称"`
	ParentId            int                       `description:"父级id"`
	HasData             int                       `description:"是否含有指标数据"`
	CreateTime          time.Time                 `description:"创建时间"`
	ModifyTime          time.Time                 `description:"修改时间"`
	SysUserId           int                       `description:"创建人id"`
	SysUserRealName     string                    `description:"创建人姓名"`
	Level               int                       `description:"层级"`
	Sort                int                       `description:"排序字段，越小越靠前，默认值：10"`
	AiSummaryId         int                       `description:"纪要id"`
	UniqueCode          string                    `description:"唯一编码"`
	Children            []*AiSummaryClassifyItems `gorm:"-"`
}

type AiSummaryClassifyListResp struct {
	AllNodes []*AiSummaryClassifyItems
}

// GetAiSummaryClassifyByParentId
func GetAiSummaryClassifyByParentId(parentId int) (items []*AiSummaryClassifyItems, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT * FROM ai_summary_classify WHERE parent_id=? order by sort asc,ai_summary_classify_id asc`
	err = o.Raw(sql, parentId).Find(&items).Error
	return
}

// GetAiSummaryInfoByAdminId 获取所有我创建的纪要，用于分类展示
func GetAiSummaryInfoByAdminId(adminId int) (items []*AiSummaryClassifyItems, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT
	a.*,
	b.ai_summary_classify_id,
	b.classify_name 
FROM
	ai_summary AS a
	JOIN ai_summary_classify AS b ON a.classify_id = b.ai_summary_classify_id 
WHERE
	a.sys_user_id = ? 
ORDER BY
	a.sort ASC,
	a.create_time ASC `
	err = o.Raw(sql, adminId).Find(&items).Error
	return
}

// GetAiSummaryClassifyAndInfoByParentId
func GetAiSummaryClassifyAndInfoByParentId(parentId int) (items []*AiSummaryClassifyItems, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT
	0 AS ai_summary_id,
	ai_summary_classify_id,
	classify_name,
	parent_id,
	create_time,
	modify_time,
	sys_user_id,
	sys_user_real_name AS sys_user_real_name,
	sort,
	level
FROM
	ai_summary_classify 
WHERE
	parent_id = ? UNION ALL
SELECT
	ai_summary_id,
	classify_id as ai_summary_classify_id,
	title AS classify_name,
	0 AS parent_id,
	create_time,
	modify_time,
	sys_user_id,
	sys_user_real_name,
	sort,
	0 AS level
FROM
	ai_summary 
WHERE
	classify_id = ? 
ORDER BY
	sort ASC,
	ai_summary_classify_id ASC`
	err = o.Raw(sql, parentId, parentId).Find(&items).Error
	return
}

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

func GetAiSummaryClassifyCount(classifyName string, parentId int) (count int, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT COUNT(1) AS count FROM ai_summary_classify WHERE parent_id=? AND classify_name=? `
	err = o.Raw(sql, parentId, classifyName).Scan(&count).Error
	return
}

// GetAiSummaryClassifyMaxSort 获取ai纪要下最大的排序数
func GetAiSummaryClassifyMaxSort(parentId int) (sort int, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT COALESCE(MAX(sort), 0) AS sort FROM ai_summary_classify WHERE parent_id=? `
	err = o.Raw(sql, parentId).Scan(&sort).Error
	return
}

type EditAiSummaryClassifyReq struct {
	ClassifyName        string `description:"分类名称"`
	AiSummaryClassifyId int    `description:"分类id"`
}

func EditSandboxClassify(classifyId int, classifyName string) (err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `UPDATE ai_summary_classify SET classify_name=?, modify_time=NOW() WHERE ai_summary_classify_id=? `
	err = o.Exec(sql, classifyName, classifyId).Error
	return
}

type AiSummaryClassifyDeleteCheckReq struct {
	AiSummaryClassifyId int `description:"分类id"`
}

func GetAiSummaryInfoCountByClassifyId(classifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameMaster]
	var pars []interface{}
	var sql string
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE ai_summary_classify_cte (ai_summary_classify_id, parent_id) AS (
				SELECT ai_summary_classify_id, parent_id
				FROM ai_summary_classify
				WHERE parent_id = ?
				UNION ALL
				SELECT c.ai_summary_classify_id, c.parent_id
				FROM ai_summary_classify c
				INNER JOIN ai_summary_classify_cte ct ON c.parent_id = ct.ai_summary_classify_id
			)
			SELECT COUNT(1) AS count
			FROM ai_summary a
			WHERE a.classify_id IN (
				SELECT ai_summary_classify_id
				FROM ai_summary_classify_cte
				UNION
				SELECT ai_summary_classify_id
				FROM ai_summary_classify
				WHERE ai_summary_classify_id = ?
			)`
		pars = append(pars, classifyId, classifyId)
	} else {
		sql = ` SELECT COUNT(1) AS count FROM ai_summary AS a
		WHERE a.classify_id IN(
		SELECT t.ai_summary_classify_id FROM 
		(
		SELECT rd.*
		FROM (SELECT * FROM ai_summary_classify WHERE parent_id IS NOT NULL) rd,
			 (SELECT @pid := ?) pd 
		WHERE FIND_IN_SET(parent_id, @pid) > 0 
		  AND @pid := CONCAT(@pid, ',', ai_summary_classify_id) 
		UNION SELECT * FROM ai_summary_classify WHERE ai_summary_classify_id = @pid 
		)AS t
		) `
		pars = append(pars, classifyId)
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

type DeleteAiSummaryClassifyReq struct {
	AiSummaryClassifyId int `description:"分类id"`
	AiSummaryId         int `description:"纪要id"`
}

func DeleteAiSummaryClassify(classifyId int) (err error) {
	o := global.DbMap[utils.DbNameMaster]
	var sql string
	var pars []interface{}
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE ai_summary_classify_cte (ai_summary_classify_id, parent_id) AS (
			SELECT ai_summary_classify_id, parent_id
			FROM ai_summary_classify
			WHERE parent_id = ?
			UNION ALL
			SELECT c.ai_summary_classify_id, c.parent_id
			FROM ai_summary_classify c
			INNER JOIN ai_summary_classify_cte ct ON c.parent_id = ct.ai_summary_classify_id
		)
		DELETE FROM ai_summary_classify
		WHERE ai_summary_classify_id IN (
			SELECT ai_summary_classify_id
			FROM ai_summary_classify_cte
			UNION
			SELECT ai_summary_classify_id
			FROM ai_summary_classify
			WHERE ai_summary_classify_id = ?
		)`
		pars = append(pars, classifyId, classifyId)
	} else {
		sql = ` DELETE FROM ai_summary_classify
				WHERE ai_summary_classify_id IN(
				SELECT t.ai_summary_classify_id FROM
				(
				SELECT rd.*
				FROM (SELECT * FROM ai_summary_classify WHERE parent_id IS NOT NULL) rd,
				(SELECT @pid := ?) pd
				WHERE FIND_IN_SET(parent_id, @pid) > 0
				AND @pid := CONCAT(@pid, ',', ai_summary_classify_id)
				UNION SELECT * FROM ai_summary_classify WHERE ai_summary_classify_id = @pid
				)AS t
				) `
		pars = append(pars, classifyId)
	}
	err = o.Exec(sql, pars...).Error
	return
}

// MoveAiSummaryClassifyReq 移动纪要分类请求参数
type MoveAiSummaryClassifyReq struct {
	AiSummaryClassifyId int `description:"分类id"`
	AiSummaryId         int `description:"纪要ID"`
	ParentClassifyId    int `description:"父级分类id 移动纪要时为目标分类id"`
	PrevId              int `description:"上一个兄弟节点分类id"`
	NextId              int `description:"下一个兄弟节点分类id"`
	PrevType            int `description:"上一个兄弟节点类型 1分类 2纪要 "`
	NextType            int `description:"上一个兄弟节点类型 1分类 2纪要 "`
}

func GetAiSummaryClassifyById(classifyId int) (item *AiSummaryClassify, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT * FROM ai_summary_classify WHERE ai_summary_classify_id=? `
	err = o.Raw(sql, classifyId).First(&item).Error
	return
}

func GetAiSummaryClassifyCountById(classifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT count(1) AS count FROM ai_summary_classify WHERE ai_summary_classify_id=? `
	err = o.Raw(sql, classifyId).Scan(&count).Error
	return
}

// Update 更新沙盘分类基础信息
func (aiSummaryClassify *AiSummaryClassify) Update(cols []string) (err error) {
	o := global.DbMap[utils.DbNameMaster]
	err = o.Select(cols).Updates(aiSummaryClassify).Error
	return
}

// GetFirstAiSummaryClassifyByParentId 获取当前父级沙盘分类下的排序第一条的数据
func GetFirstAiSummaryClassifyByParentId(parentId int) (item *AiSummaryClassify, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT * FROM ai_summary_classify WHERE parent_id=? order by sort asc,ai_summary_classify_id asc limit 1`
	err = o.Raw(sql, parentId).First(&item).Error
	return
}

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

// 获取所有子级分类id
func GetAiSummaryClassifySubcategories(classifyId int) (Ids string, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT GROUP_CONCAT(ai_summary_classify_id) AS ids
FROM (
SELECT @pv := ? AS ai_summary_classify_id
UNION ALL
SELECT sc.ai_summary_classify_id
FROM ai_summary_classify sc
JOIN (SELECT @pv := ?) initial
WHERE sc.parent_id = @pv
) subcategories; `
	err = o.Raw(sql, classifyId, classifyId).Scan(&Ids).Error
	return
}

func GetAiSummaryClassify(aiSummaryClassify int) (aiSummaryClassifyIds string, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := `SELECT GROUP_CONCAT(t.ai_summary_classify_id) AS ai_summary_classify_id FROM (
			SELECT a.ai_summary_classify_id FROM ai_summary_classify AS a 
			WHERE a.ai_summary_classify_id=?
			UNION ALL
			SELECT a.ai_summary_classify_id FROM ai_summary_classify AS a 
			WHERE a.parent_id=? UNION ALL
	SELECT
		ai_summary_classify_id 
	FROM
		ai_summary_classify 
WHERE
	parent_id IN ( SELECT ai_summary_classify_id FROM ai_summary_classify WHERE parent_id = ? )
			)AS t`
	err = o.Raw(sql, aiSummaryClassify, aiSummaryClassify, aiSummaryClassify).Scan(&aiSummaryClassifyIds).Error
	return
}

func GetAiSummaryAllParentByClassifyId(aiSummaryClassifyId int) (ids string, err error) {
	o := global.DbMap[utils.DbNameMaster]
	var sql string
	if utils.DbDriverName == utils.DbDriverByDm {
		sql = `WITH RECURSIVE parent_tree (ai_summary_classify_id, parent_id, level) AS (
    -- 获取起始节点的父级
    SELECT ai_summary_classify_id, parent_id, 1 as level
    FROM ai_summary_classify
    WHERE ai_summary_classify_id = (
        SELECT parent_id 
        FROM ai_summary_classify 
        WHERE ai_summary_classify_id = ?
    )
    
    UNION ALL
    
    -- 递归获取所有上级节点
    SELECT ac.ai_summary_classify_id, ac.parent_id, pt.level + 1
    FROM ai_summary_classify ac
    JOIN parent_tree pt ON ac.ai_summary_classify_id = pt.parent_id
    WHERE ac.parent_id IS NOT NULL
)
SELECT LISTAGG(ai_summary_classify_id, ',') WITHIN GROUP (ORDER BY level DESC) AS ids
FROM parent_tree`
	}else{
	sql = `SELECT
	GROUP_CONCAT(DISTINCT m.ai_summary_classify_id  ORDER BY m.level) AS ids 
FROM
	(
	SELECT
		@id AS _id,(
		SELECT
			@id := parent_id 
		FROM
			ai_summary_classify 
		WHERE
			ai_summary_classify_id = _id 
		) 
	FROM
		(
		SELECT
			@id :=(
			SELECT
				parent_id 
			FROM
				ai_summary_classify 
			WHERE
				ai_summary_classify_id = ? 
			)) vm,
		ai_summary_classify m 
	WHERE
		@id IS NOT NULL 
	) vm
	INNER JOIN ai_summary_classify m 
WHERE
	ai_summary_classify_id = vm._id `
	}
	var sqlIds sql2.NullString
	err = o.Raw(sql, aiSummaryClassifyId).Scan(&sqlIds).Error
	if sqlIds.Valid {
		ids = sqlIds.String
	}
	return
}

type AiSummaryListResp struct {
	Paging *paging.PagingItem
	List   []*AiSummaryItems
}

// GetAiSummaryClassifyAll
func GetAiSummaryClassifyAll() (items []*AiSummaryClassifyItems, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT * FROM ai_summary_classify WHERE parent_id<>0 order by sort asc,ai_summary_classify_id asc`
	err = o.Raw(sql).Find(&items).Error
	return
}

func GetAiSummaryClassifyAllIncludeParent() (items []*AiSummaryClassifyItems, err error) {
	o := global.DbMap[utils.DbNameMaster]
	sql := ` SELECT * FROM ai_summary_classify order by sort asc,ai_summary_classify_id asc`
	err = o.Raw(sql).Find(&items).Error
	return
}