eta_api/models/ai_summary/ai_summary_classify.go

package ai_summary

import (
	"fmt"
	"github.com/beego/beego/v2/client/orm"
	"time"
)

type AiSummaryClassify struct {
	AiSummaryClassifyId int       `orm:"column(ai_summary_classify_id);pk"` // 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 := orm.NewOrm()
	lastId, err = o.Insert(item)
	return
}

type AiSummaryClassifyItems struct {
	AiSummaryClassifyId int
	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"`
	Children            []*AiSummaryClassifyItems
}

type AiSummaryClassifyListResp struct {
	AllNodes []*AiSummaryClassifyItems
}

// GetAiSummaryClassifyByParentId
func GetAiSummaryClassifyByParentId(parentId int) (items []*AiSummaryClassifyItems, err error) {
	o := orm.NewOrm()
	sql := ` SELECT * FROM ai_summary_classify WHERE parent_id=? order by sort asc,ai_summary_classify_id asc`
	_, err = o.Raw(sql, parentId).QueryRows(&items)
	return
}

// GetAiSummaryInfoByAdminId 获取所有我创建的纪要，用于分类展示
func GetAiSummaryInfoByAdminId(adminId int) (items []*AiSummaryClassifyItems, err error) {
	o := orm.NewOrm()
	sql := ` SELECT ai_summary_id,ai_summary_classify_id,name AS classify_name,
             sys_user_id,sys_user_real_name
            FROM ai_summary where sys_user_id = ? ORDER BY sort asc,create_time ASC `
	_, err = o.Raw(sql, adminId).QueryRows(&items)
	return
}

// GetAiSummaryClassifyAndInfoByParentId
func GetAiSummaryClassifyAndInfoByParentId(parentId int) (items []*AiSummaryClassifyItems, err error) {
	o := orm.NewOrm()
	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,
	'' 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).QueryRows(&items)
	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 := orm.NewOrm()
	sql := `SELECT COUNT(1) AS count FROM ai_summary_classify WHERE parent_id=? AND classify_name=? `
	err = o.Raw(sql, parentId, classifyName).QueryRow(&count)
	return
}

// GetAiSummaryClassifyMaxSort 获取ai纪要下最大的排序数
func GetAiSummaryClassifyMaxSort(parentId int) (sort int, err error) {
	o := orm.NewOrm()
	sql := `SELECT Max(sort) AS sort FROM ai_summary_classify WHERE parent_id=? `
	err = o.Raw(sql, parentId).QueryRow(&sort)
	return
}

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

func EditSandboxClassify(classifyId int, classifyName string) (err error) {
	o := orm.NewOrm()
	sql := `UPDATE ai_summary_classify SET classify_name=?, modify_time=NOW() WHERE ai_summary_classify_id=? `
	_, err = o.Raw(sql, classifyName, classifyId).Exec()
	return
}

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

func GetAiSummaryInfoCountByClassifyId(classifyId int) (count int, err error) {
	o := orm.NewOrm()
	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
				) `
	err = o.Raw(sql, classifyId).QueryRow(&count)
	return
}

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

func DeleteAiSummaryClassify(classifyId int) (err error) {
	o := orm.NewOrm()
	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
				) `
	_, err = o.Raw(sql, classifyId).Exec()
	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 := orm.NewOrm()
	sql := `SELECT * FROM ai_summary_classify WHERE ai_summary_classify_id=? `
	err = o.Raw(sql, classifyId).QueryRow(&item)
	return
}

func GetAiSummaryClassifyCountById(classifyId int) (count int, err error) {
	o := orm.NewOrm()
	sql := `SELECT count(1) AS count FROM ai_summary_classify WHERE ai_summary_classify_id=? `
	err = o.Raw(sql, classifyId).QueryRow(&count)
	return
}

// Update 更新沙盘分类基础信息
func (aiSummaryClassify *AiSummaryClassify) Update(cols []string) (err error) {
	o := orm.NewOrm()
	_, err = o.Update(aiSummaryClassify, cols...)
	return
}

// GetFirstAiSummaryClassifyByParentId 获取当前父级沙盘分类下的排序第一条的数据
func GetFirstAiSummaryClassifyByParentId(parentId int) (item *AiSummaryClassify, err error) {
	o := orm.NewOrm()
	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).QueryRow(&item)
	return
}

// UpdateSandboxClassifySortByParentId 根据沙盘父类id更新排序
func UpdateAiSummaryClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string) (err error) {
	o := orm.NewOrm()
	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.Raw(sql, parentId, nowSort).Exec()
	return
}

// 获取所有子级分类id
func GetAiSummaryClassifySubcategories(classifyId int) (Ids string, err error) {
	o := orm.NewOrm()
	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).QueryRow(&Ids)
	return
}