eta_api/models/ai_summary/ai_summary_classify.go

package ai_summary

import (
	"eta_gn/eta_api/global"
	"eta_gn/eta_api/utils"
	"fmt"
	"github.com/rdlucklib/rdluck_tools/paging"
	"time"
)

type AiSummaryClassify struct {
	AiSummaryClassifyId int       `gorm:"primaryKey;column:ai_summary_classify_id;type:int(11);not null"`
	ClassifyName        string    `gorm:"column:classify_name;type:varchar(255);not null;default:''"` // 分类名称
	ParentId            int       `gorm:"column:parent_id;type:int(11);not null;default:0"`           // 父级id
	CreateTime          time.Time `gorm:"column:create_time;type:datetime"`                           // 创建时间
	ModifyTime          time.Time `gorm:"column:modify_time;type:datetime"`                           // 修改时间
	SysUserId           int       `gorm:"index:idx_sys_user_id;column:sys_user_id;type:int(11)"`      // 创建人id
	SysUserRealName     string    `gorm:"column:sys_user_real_name;type:varchar(50)"`                 // 创建人姓名
	Level               int       `gorm:"column:level;type:int(11);default:0"`                        // 层级
	Sort                int       `gorm:"column:sort;type:int(9) unsigned;default:10"`                // 排序字段，越小越靠前，默认值：10
	RootId              int       `gorm:"column:root_id;type:int(11);not null;default:0"`             // 顶级ID
	HasData             int       `gorm:"column:has_data;type:tinyint(4);default:0"`                  // 是否存在指标数据:1有，2：无
}

func AddAiSummaryClassify(item *AiSummaryClassify) (lastId int64, err error) {
	err = global.DEFAULT_DmSQL.Create(item).Error
	lastId = int64(item.AiSummaryClassifyId)
	return
}

type AiSummaryClassifyItems struct {
	AiSummaryClassifyId int                       `gorm:"primaryKey;column:ai_summary_classify_id;type:int(11);not null"`
	ClassifyName        string                    `gorm:"column:classify_name;type:varchar(255);not null;default:''"` // 分类名称
	ParentId            int                       `gorm:"column:parent_id;type:int(11);not null;default:0"`           // 父级id
	CreateTime          time.Time                 `gorm:"column:create_time;type:datetime"`                           // 创建时间
	ModifyTime          time.Time                 `gorm:"column:modify_time;type:datetime"`                           // 修改时间
	SysUserId           int                       `gorm:"index:idx_sys_user_id;column:sys_user_id;type:int(11)"`      // 创建人id
	SysUserRealName     string                    `gorm:"column:sys_user_real_name;type:varchar(50)"`                 // 创建人姓名
	Level               int                       `gorm:"column:level;type:int(11);default:0"`                        // 层级
	Sort                int                       `gorm:"column:sort;type:int(9) unsigned;default:10"`                // 排序字段，越小越靠前，默认值：10
	RootId              int                       `gorm:"column:root_id;type:int(11);not null;default:0"`             // 顶级ID
	HasData             int                       `gorm:"column:has_data;type:tinyint(4);default:0"`                  // 是否存在指标数据:1有，2：无
	AiSummaryId         int                       `description:"纪要id" gorm:"ai_summary_id"`
	UniqueCode          string                    `description:"唯一编码" gorm:"-"`
	Children            []*AiSummaryClassifyItems `gorm:"-"`
}

type AiSummaryClassifyListResp struct {
	AllNodes []*AiSummaryClassifyItems
}

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

// GetAiSummaryInfoByAdminId 获取所有我创建的纪要，用于分类展示
func GetAiSummaryInfoByAdminId(adminId int) (items []*AiSummaryClassifyItems, err error) {
	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 = global.DEFAULT_DmSQL.Raw(sql, adminId).Find(&items).Error
	return
}

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

// GetAiSummaryClassifyMaxSort 获取ai纪要下最大的排序数
func GetAiSummaryClassifyMaxSort(parentId int) (sort int, err error) {
	sql := `SELECT  COALESCE(MAX(sort), 0) AS sort FROM ai_summary_classify WHERE parent_id=? `
	err = global.DEFAULT_DmSQL.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) {
	sql := `UPDATE ai_summary_classify SET classify_name=?, modify_time=NOW() WHERE ai_summary_classify_id=? `
	err = global.DEFAULT_DmSQL.Exec(sql, classifyName, classifyId).Error
	return
}

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

func GetAiSummaryInfoCountByClassifyId(classifyId int) (count int, err error) {
	var pars []interface{}
	var sql string
	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)
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
	return
}

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

// 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) {
	sql := `SELECT * FROM ai_summary_classify WHERE ai_summary_classify_id=? `
	err = global.DEFAULT_DmSQL.Raw(sql, classifyId).First(&item).Error
	return
}

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

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

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

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

func GetAiSummaryClassify(aiSummaryClassify int) (aiSummaryClassifyIds string, err error) {
	sql := `SELECT LISTAGG(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 = global.DEFAULT_DmSQL.Raw(sql, aiSummaryClassify, aiSummaryClassify, aiSummaryClassify).Scan(&aiSummaryClassifyIds).Error
	return
}

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

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

// delBYids
func DeleteAiSummaryClassifyByIds(ids []int) (err error) {
	sql := ` DELETE FROM ai_summary_classify WHERE ai_summary_classify_id in (` + utils.GetOrmInReplace(len(ids)) + `)`
	err = global.DEFAULT_DmSQL.Exec(sql, ids).Error
	return
}