package data_manage import ( "eta_gn/eta_api/global" "eta_gn/eta_api/utils" "fmt" "time" ) type ChartClassify struct { ChartClassifyId int `orm:"column(chart_classify_id);pk" gorm:"primaryKey" ` ChartClassifyName 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:"层级"` UniqueCode string `description:"唯一编码"` Sort int `description:"排序字段,越小越靠前,默认值:10"` Source int `description:"1:ETA图库;2:商品价格曲线"` IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"` ChartClassifyNameEn string `description:"英文分类名称"` RootId int `description:"顶级ID"` ChartClassifyNamePath string `description:"分类名称的完整路径,格式为:父级分类名称|当前分类名称"` ChartClassifyIdPath string `description:"分类的完整路径,格式为:父级ID,当前ID"` } func AddChartClassify(item *ChartClassify) (err error) { err = global.DmSQL["data"].Create(item).Error if err != nil { return } return } type AddChartClassifyReq struct { ChartClassifyName string `description:"分类名称"` ParentId int `description:"父级id,第一级传0"` Level int `description:"层级,第一级传0,其余传上一级的层级"` } // GetChartClassifyCount // @Description: 根据图表分类名称获取对应的数量 // @author: Roc // @datetime 2024-04-16 18:13:40 // @param chartClassifyName string // @param parentId int // @param source int // @return count int // @return err error func GetChartClassifyCount(chartClassifyName string, parentId, source int) (count int, err error) { o := global.DmSQL["data"] sql := `SELECT COUNT(1) AS count FROM chart_classify WHERE parent_id=? AND source = ? AND chart_classify_name=? ` err = o.Raw(sql, parentId, source, chartClassifyName).Scan(&count).Error return } // GetChartClassifyEnCount // @Description: 根据图表分类英文名称获取对应的数量 // @author: Roc // @datetime 2024-04-16 18:13:24 // @param chartClassifyNameEn string // @param parentId int // @param source int // @return count int // @return err error func GetChartClassifyEnCount(chartClassifyNameEn string, parentId, source int) (count int, err error) { o := global.DmSQL["data"] sql := `SELECT COUNT(1) AS count FROM chart_classify WHERE parent_id=? AND source = ? AND chart_classify_name_en = ? ` err = o.Raw(sql, parentId, source, chartClassifyNameEn).Scan(&count).Error return } type EditChartClassifyReq struct { ChartClassifyName string `description:"分类名称"` ChartClassifyId int `description:"分类id"` } func GetChartClassifyById(classifyId int) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE chart_classify_id=? ` err = o.Raw(sql, classifyId).First(&item).Error return } func GetChartClassifyCountById(classifyId int) (count int, err error) { o := global.DmSQL["data"] sql := `SELECT count(1) AS count FROM chart_classify WHERE chart_classify_id=? ` err = o.Raw(sql, classifyId).Scan(&count).Error return } type DeleteChartClassifyReq struct { ChartClassifyId int `description:"分类id"` ChartInfoId int `description:"指标id"` } // GetChartClassifyByParentId // @param source int 1:ETA图库;2:商品价格曲线;3:相关性图表 func GetChartClassifyByParentId(parentId, source, sysUserId int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE parent_id=? AND source = ? ` if sysUserId > 0 { sql += fmt.Sprintf(` AND sys_user_id = %d `, sysUserId) } sql += ` order by sort asc,chart_classify_id asc ` err = o.Raw(sql, parentId, source).Scan(&items).Error return } // GetChartClassifyAll // @param source int 1:ETA图库;2:商品价格曲线;3:相关性图表 func GetChartClassifyAll(source, sysUserId int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE parent_id<>0 AND source = ?` if sysUserId > 0 { sql += fmt.Sprintf(` AND sys_user_id = %d `, sysUserId) } sql += ` order by sort asc,chart_classify_id asc ` err = o.Raw(sql, source).Scan(&items).Error return } // GetAllChartClassify // @Description: 获取所有类型的图表分类 // @author: Roc // @datetime 2024-12-09 10:11:46 // @return items []*ChartClassifyItems // @return err error func GetAllChartClassify() (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify order by sort asc,chart_classify_id asc` err = o.Raw(sql).Scan(&items).Error return } // GetChartClassifyByParentId // @param source int 1:ETA图库;2:商品价格曲线;3:相关性图表 func GetAllChartClassifyByParentId(parentId int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE parent_id=? order by sort asc,chart_classify_id asc` err = o.Raw(sql, parentId).Scan(&items).Error return } type ChartClassifyItems struct { ChartClassifyId int `description:"分类id"` ChartInfoId int `description:"指标id"` ChartClassifyName string ChartClassifyNameEn string ParentId int Level int `description:"层级"` Sort int `description:"排序字段,越小越靠前,默认值:10"` UniqueCode string `description:"唯一编码"` Source int `description:"来源id"` SourceName string `description:"来源名称"` SysUserId int `description:"创建人id"` SysUserRealName string `description:"创建人姓名"` DateType int `description:"日期类型:1:00年至今,2:10年至今,3:15年至今,4:年初至今,5:自定义时间"` StartDate string `description:"自定义开始日期"` EndDate string `description:"自定义结束日期"` ChartType int `description:"生成样式:1:曲线图,2:季节性图"` Calendar string `description:"公历/农历"` SeasonStartDate string `description:"季节性图开始日期"` SeasonEndDate string `description:"季节性图开始日期"` Children []*ChartClassifyItems `gorm:"-"` Button ChartClassifyItemsButton `gorm:"-" description:"按钮权限"` IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"` HaveOperaAuth bool `description:"是否有数据权限,默认:false"` ChartClassifyIdPath string `description:"分类的完整路径,格式为:父级ID,当前ID"` } // ChartClassifyItemsButton 操作按钮 type ChartClassifyItemsButton struct { AddButton bool `description:"是否可添加"` OpButton bool `description:"是否可编辑"` DeleteButton bool `description:"是否可删除"` MoveButton bool `description:"是否可移动"` } type ChartClassifyListResp struct { AllNodes []*ChartClassifyItems Language string `description:"指标的展示语言,CN:中文,EN:英文"` CanOpClassify bool `description:"是否允许操作分类"` } type ChartClassifyDeleteCheckResp struct { DeleteStatus int `description:"检测状态:0:默认值,如果为0,继续走其他校验,1:该分类下关联图表不可删除,2:确认删除当前目录及包含的子目录吗"` TipsMsg string `description:"提示信息"` } type ChartClassifyDeleteCheckReq struct { ChartClassifyId int `description:"分类id"` ChartInfoId int `description:"指标id"` } func GetChartClassifyByCondition(condition string, pars []interface{}) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE 1=1 ` if condition != "" { sql += condition } err = o.Raw(sql, pars...).First(&item).Error return } // MoveChartClassifyReq 移动图表分类请求参数 type MoveChartClassifyReq struct { ClassifyId int `description:"分类id"` ParentClassifyId int `description:"父级分类id"` PrevClassifyId int `description:"上一个兄弟节点分类id"` NextClassifyId int `description:"下一个兄弟节点分类id"` ChartInfoId int `description:"图表ID, 如果图表ID有值,则移动对象为图表,否则认为移动对象分类"` PrevChartInfoId int `description:"上一个图表ID"` NextChartInfoId int `description:"下一个图表ID"` } // GetFirstChartClassifyByParentId 获取当前父级图表分类下的排序第一条的数据 func GetFirstChartClassifyByParentId(parentId int) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE parent_id=? order by sort asc,chart_classify_id asc limit 1` err = o.Raw(sql, parentId).First(&item).Error return } // GetFirstChartClassifyByParentIdAndSource 获取当前父级图表分类下的排序第一条的数据 func GetFirstChartClassifyByParentIdAndSource(parentId, source int) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE parent_id=? and source = ? order by sort asc,chart_classify_id asc limit 1` err = o.Raw(sql, parentId, source).First(&item).Error return } // UpdateChartClassifySortByParentId 根据图表父类id更新排序 func UpdateChartClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string) (err error) { o := global.DmSQL["data"] sql := ` update chart_classify set sort = ` + updateSort + ` WHERE parent_id=? and sort > ? ` if classifyId > 0 { sql += ` or ( chart_classify_id > ` + fmt.Sprint(classifyId) + ` and sort= ` + fmt.Sprint(nowSort) + `)` } err = o.Exec(sql, parentId, nowSort).Error return } // Update 更新图表分类基础信息 func (chartClassify *ChartClassify) Update(cols []string) (err error) { err = global.DmSQL["data"].Select(cols).Updates(chartClassify).Error return } // GetChartClassifyMaxSort 获取图表分类下最大的排序数 func GetChartClassifyMaxSort(parentId, source int) (sort int, err error) { o := global.DmSQL["data"] sql := `SELECT COALESCE(MAX(sort),0) AS sort FROM chart_classify WHERE parent_id=? AND source = ? ` err = o.Raw(sql, parentId, source).Scan(&sort).Error return } type ChartClassifyView struct { ChartClassifyId int `orm:"column(chart_classify_id);pk" gorm:"primaryKey" ` ChartClassifyName string `description:"分类名称"` ParentId int `description:"父级id"` } func GetChartClassifyViewById(classifyId int) (item *ChartClassifyView, err error) { o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE chart_classify_id=? ` err = o.Raw(sql, classifyId).First(&item).Error return } func GetChartClassifyByLevel(level, source int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE level=? AND source = ? order by sort asc,chart_classify_id asc` err = o.Raw(sql, level, source).Scan(&items).Error return } func EditChartClassifySysUser(classifyId, sysUserId int, chartClassifyName string) (err error) { o := global.DmSQL["data"] sql := `UPDATE chart_classify SET sys_user_id=?,sys_user_real_name=?,modify_time=NOW() WHERE parent_id=? AND level=3 ` err = o.Exec(sql, sysUserId, chartClassifyName, classifyId).Error return } func GetAllChartClassifyItemsBySource(source int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE source = ? order by sort ASC,chart_classify_id ASC` err = o.Raw(sql, source).Scan(&items).Error return } // GetCrossVarietyChartClassifyBySysUserId // @Description: 根据创建人获取跨品种分析的分类 // @author: Roc // @datetime 2023-11-24 14:05:43 // @param sysUserId int // @return item *ChartClassify // @return err error func GetCrossVarietyChartClassifyBySysUserId(sysUserId int) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE source = ? AND sys_user_id=?` err = o.Raw(sql, utils.CHART_SOURCE_CROSS_HEDGING, sysUserId).First(&item).Error return } // GetChartClassifyByParentIdAndName 根据父级ID和名称获取分类 func GetChartClassifyByParentIdAndName(source, parentId int, classifyName string, classifyId int) (item *ChartClassify, err error) { o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE source = ? AND parent_id = ? AND chart_classify_name = ? AND chart_classify_id <> ? LIMIT 1` err = o.Raw(sql, source, parentId, classifyName, classifyId).First(&item).Error return } // GetChartClassifyByIdList // @Description: 根据分类ID列表获取分类 // @author: Roc // @datetime 2024-04-01 13:21:48 // @param classifyId []int // @return items []*ChartClassify // @return err error func GetChartClassifyByIdList(classifyIdList []int) (items []*ChartClassify, err error) { num := len(classifyIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE chart_classify_id in (` + utils.GetOrmInReplace(num) + `) ` err = o.Raw(sql, classifyIdList).Scan(&items).Error return } // GetChartClassifyItemsByIdList // @Description: 根据分类ID列表获取分类 // @author: Roc // @datetime 2024-04-01 13:21:48 // @param classifyId []int // @return items []*ChartClassify // @return err error func GetChartClassifyItemsByIdList(classifyIdList []int) (items []*ChartClassifyItems, err error) { num := len(classifyIdList) if num <= 0 { return } o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE chart_classify_id in (` + utils.GetOrmInReplace(num) + `) ` err = o.Raw(sql, classifyIdList).Scan(&items).Error return } // GetChartClassifyBySourceAndIsJoinPermission // @Description: 根据分类类型和是否涉密获取分类列表 // @author: Roc // @datetime 2024-03-29 10:30:21 // @param source int // @param isJoinPermission int // @return items []*ChartClassify // @return err error func GetChartClassifyBySourceAndIsJoinPermission(source, isJoinPermission int) (items []*ChartClassify, err error) { o := global.DmSQL["data"] sql := `SELECT * FROM chart_classify WHERE source = ? AND is_join_permission = ? ` err = o.Raw(sql, source, isJoinPermission).Scan(&items).Error return } type ChartClassifyIdItems struct { ChartClassifyId int `description:"分类id"` ChartClassifyName string `description:"分类名称"` UniqueCode string `description:"唯一编码"` ParentId int `description:"父级分类id"` Level int `description:"层级"` RootId int `description:"顶级分类id"` IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0" json:"-"` HaveOperaAuth bool `description:"是否有该数据权限,默认:false"` } func GetChartClassifyByRootIdLevel(rootId, source int, orderStr string) (items []*ChartClassifyIdItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE root_id = ? AND source = ? ` if orderStr != "" { sql += orderStr } else { sql += ` order by level desc, sort asc, chart_classify_id asc` } err = o.Raw(sql, rootId, source).Scan(&items).Error return } // UpdateChartClassifySortByParentIdAndSource 根据图表父类id更新排序 func UpdateChartClassifySortByParentIdAndSource(parentId, classifyId, nowSort int, updateSort string, source int) (err error) { o := global.DmSQL["data"] sql := ` update chart_classify set sort = ` + updateSort + ` WHERE parent_id=? and sort > ? AND source = ? ` if classifyId > 0 { sql += ` or ( chart_classify_id > ` + fmt.Sprint(classifyId) + ` and sort= ` + fmt.Sprint(nowSort) + `)` } err = o.Exec(sql, parentId, nowSort, source).Error return } // GetChartClassifyAllBySource 根据来源获取所有分类 func GetChartClassifyAllBySource(source int) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE source = ? ORDER BY parent_id ASC, sort ASC, chart_classify_id ASC` err = o.Raw(sql, source).Scan(&items).Error return } // delByIds func DeleteChartClassifyByIds(ids []int) (err error) { o := global.DmSQL["data"] sql := `DELETE FROM chart_classify WHERE chart_classify_id IN (` + utils.GetOrmInReplace(len(ids)) + `)` err = o.Exec(sql, ids).Error return } // UpdateChartClassify // @Description: 更新图表分类的基础信息 // @author: Roc // @datetime 2024-12-09 10:29:51 // @param classifyId int // @param classifyNamePath string // @param classifyIdPath string // @return err error func UpdateChartClassify(classifyId int, classifyNamePath, classifyIdPath string) (err error) { sql := `UPDATE chart_classify SET chart_classify_name_path = ?,chart_classify_id_path = ? WHERE chart_classify_id = ?` err = global.DmSQL["data"].Exec(sql, classifyNamePath, classifyIdPath, classifyId).Error return } // UpdateClassifyIdByChartInfoId // @Description: 根据图表id更新所属分类id // @author: Roc // @datetime 2024-11-29 17:54:44 // @param edbInfoId int // @param edbClassifyId int // @return err error func UpdateClassifyIdByChartInfoId(chartInfoId, chartClassifyId int) (err error) { sql := ` UPDATE chart_info SET chart_classify_id = ? WHERE chart_info_id=? ` err = global.DmSQL["data"].Exec(sql, chartClassifyId, chartInfoId).Error return } // DelChartClassifyByIdList // @Description: 根据id删除图表分类 // @author: Roc // @datetime 2024-12-02 09:27:26 // @param idList []int // @return err error func DelChartClassifyByIdList(idList []int) (err error) { num := len(idList) if num <= 0 { return } sql := `DELETE FROM chart_classify WHERE chart_classify_id IN (?) ` err = global.DmSQL["data"].Exec(sql, idList).Error return } // UpdateClassifyNameAndNamePath // @Description: 更新分类名称和完整名称路径 // @author: Roc // @receiver chartClassify // @datetime 2024-12-09 14:13:05 // @param cols []string // @param oldClassifyNamePath string // @param newClassifyNamePath string // @return err error func (chartClassify *ChartClassify) UpdateClassifyNameAndNamePath(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(chartClassify).Error if err != nil { return } // 更改子分类的完整的路径 if oldClassifyNamePath != `` && newClassifyNamePath != `` { sql := `UPDATE chart_classify SET chart_classify_name_path = REPLACE(chart_classify_name_path,?,?) WHERE chart_classify_id_path LIKE ?` err = tx.Exec(sql, oldClassifyNamePath, newClassifyNamePath, fmt.Sprint(chartClassify.ChartClassifyIdPath+`,%`)).Error if err != nil { return } } return } // UpdateChartClassifyNameAndNamePathByOldClassifyIdPath // @Description: 根据源分类id全路径更新分类名称和子分类的全路径 // @author: Roc // @receiver edbClassify // @datetime 2024-11-26 14:39:09 // @param cols []string // @param oldClassifyNamePath string // @param newClassifyNamePath string // @return err error func UpdateChartClassifyNameAndNamePathByOldClassifyIdPath(oldClassifyIdPath, newClassifyIdPath, oldClassifyNamePath, newClassifyNamePath string) (err error) { // 更改子分类的完整的路径 if oldClassifyIdPath != `` && newClassifyNamePath != `` && newClassifyIdPath != `` { sql := `UPDATE chart_classify SET chart_classify_id_path = REPLACE(chart_classify_id_path,?,?) ,chart_classify_name_path = REPLACE(chart_classify_name_path,?,?) WHERE chart_classify_id_path LIKE ?` err = global.DmSQL["data"].Exec(sql, oldClassifyIdPath, newClassifyIdPath, oldClassifyNamePath, newClassifyNamePath, fmt.Sprint(oldClassifyIdPath+`,%`)).Error if err != nil { return } } return } func GetChildChartClassifyListByParentId(classifyIdPath, orderStr string) (items []*ChartClassifyItems, err error) { o := global.DmSQL["data"] sql := ` SELECT * FROM chart_classify WHERE chart_classify_id_path like ? ` if orderStr != "" { sql += orderStr } else { sql += ` order by level desc, sort asc, chart_classify_id asc` } err = o.Raw(sql, classifyIdPath).Scan(&items).Error return } // GetChartInfoMaxPublicSortByClassifyId // @Description: 获取分类下公共图库的最大的排序数 // @param classifyId // @return sort // @return err func GetChartInfoMaxPublicSortByClassifyId(classifyId int) (sort int, err error) { sql := `SELECT COALESCE(Max(public_sort), 0) AS sort FROM chart_info WHERE public_status = 3 AND chart_public_classify_id = ? ` err = global.DmSQL["data"].Raw(sql, classifyId).Scan(&sort).Error return } type ChartClassifyItemList []*ChartClassifyItems func (m ChartClassifyItemList) Len() int { return len(m) } func (m ChartClassifyItemList) Less(i, j int) bool { return m[i].Sort < m[j].Sort } func (m ChartClassifyItemList) Swap(i, j int) { m[i], m[j] = m[j], m[i] } func (m *ChartClassifyItems) ToUserShareChartClassifyItem() (item *UserShareChartClassifyItem) { return &UserShareChartClassifyItem{ ChartClassifyId: m.ChartClassifyId, ChartInfoId: m.ChartInfoId, ChartClassifyName: m.ChartClassifyName, ChartClassifyNameEn: m.ChartClassifyNameEn, ParentId: m.ParentId, Level: m.Level, Sort: m.Sort, UniqueCode: m.UniqueCode, Source: m.Source, SourceName: m.SourceName, SysUserId: m.SysUserId, SysUserRealName: m.SysUserRealName, DateType: m.DateType, StartDate: m.StartDate, EndDate: m.EndDate, ChartType: m.ChartType, Calendar: m.Calendar, SeasonStartDate: m.SeasonStartDate, SeasonEndDate: m.SeasonEndDate, Children: m.ToUserShareChartClassifyItemByList(m.Children), Button: ChartClassifyItemsButton{}, IsJoinPermission: m.IsJoinPermission, HaveOperaAuth: true, ChartClassifyIdPath: m.ChartClassifyIdPath, } } func (m *ChartClassifyItems) ToUserShareChartClassifyItemByList(list []*ChartClassifyItems) (items []*UserShareChartClassifyItem) { items = make([]*UserShareChartClassifyItem, 0) if list == nil { return } for _, v := range list { items = append(items, v.ToUserShareChartClassifyItem()) } return }