123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664 |
- package data_manage
- import (
- "eta/eta_api/utils"
- "fmt"
- "github.com/beego/beego/v2/client/orm"
- "strconv"
- "time"
- )
- type EdbClassify struct {
- ClassifyId int `orm:"column(classify_id);pk"`
- ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
- ClassifyName string `description:"分类名称"`
- ParentId int `description:"父级id"`
- RootId 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"`
- LastModifyUserId int `description:"最后修改人ID"`
- LastModifyUserRealName string `description:"最后修改人姓名"`
- IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
- ClassifyNameEn string `description:"英文分类名称"`
- }
- func AddEdbClassify(item *EdbClassify) (lastId int64, err error) {
- o := orm.NewOrmUsingDB("data")
- lastId, err = o.Insert(item)
- return
- }
- type AddEdbClassifyReq struct {
- ClassifyName string `description:"分类名称"`
- ParentId int `description:"父级id,第一级传0"`
- Level int `description:"层级,第一级传0,其余传上一级的层级"`
- ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
- }
- // GetEdbClassifyCount
- // @Description: 根据父级id获取同级中文名称的数量
- // @author: Roc
- // @datetime 2024-04-17 14:12:22
- // @param classifyName string
- // @param parentId int
- // @param classifyType uint8
- // @return count int
- // @return err error
- func GetEdbClassifyCount(classifyName string, parentId int, classifyType uint8) (count int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT COUNT(1) AS count FROM edb_classify WHERE parent_id=? AND classify_name=? AND classify_type = ? `
- err = o.Raw(sql, parentId, classifyName, classifyType).QueryRow(&count)
- return
- }
- // GetEdbClassifyEnCount
- // @Description: 根据父级id获取同级英文名称的数量
- // @author: Roc
- // @datetime 2024-04-17 14:12:09
- // @param classifyNameEn string
- // @param parentId int
- // @param classifyType uint8
- // @return count int
- // @return err error
- func GetEdbClassifyEnCount(classifyNameEn string, parentId int, classifyType uint8) (count int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT COUNT(1) AS count FROM edb_classify WHERE parent_id=? AND classify_name_en = ? AND classify_type = ? `
- err = o.Raw(sql, parentId, classifyNameEn, classifyType).QueryRow(&count)
- return
- }
- type EditEdbClassifyReq struct {
- ClassifyName string `description:"分类名称"`
- ClassifyId int `description:"分类名称"`
- }
- func GetEdbClassifyById(classifyId int) (item *EdbClassify, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT * FROM edb_classify WHERE classify_id=? `
- err = o.Raw(sql, classifyId).QueryRow(&item)
- return
- }
- func GetEdbClassifyCountById(classifyId int) (count int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT count(1) AS count FROM edb_classify WHERE classify_id=? `
- err = o.Raw(sql, classifyId).QueryRow(&count)
- return
- }
- // EditEdbClassify
- // @Description: 修改指标分类的中文名称
- // @author: Roc
- // @datetime 2024-04-17 14:14:48
- // @param classifyId int
- // @param classifyName string
- // @param userId int
- // @param userRealName string
- // @return err error
- func EditEdbClassify(classifyId int, classifyName string, userId int, userRealName string) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `UPDATE edb_classify SET classify_name=?,modify_time=NOW(), last_modify_user_id = ?, last_modify_user_real_name = ? WHERE classify_id=? `
- _, err = o.Raw(sql, classifyName, userId, userRealName, classifyId).Exec()
- return
- }
- // EditEdbClassifyEn
- // @Description: 修改指标分类的英文名称
- // @author: Roc
- // @datetime 2024-04-17 14:14:36
- // @param classifyId int
- // @param classifyNameEn string
- // @param userId int
- // @param userRealName string
- // @return err error
- func EditEdbClassifyEn(classifyId int, classifyNameEn string, userId int, userRealName string) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `UPDATE edb_classify SET classify_name_en = ?,modify_time=NOW(), last_modify_user_id = ?, last_modify_user_real_name = ? WHERE classify_id=? `
- _, err = o.Raw(sql, classifyNameEn, userId, userRealName, classifyId).Exec()
- return
- }
- type DeleteEdbClassifyReq struct {
- ClassifyId int `description:"分类id"`
- EdbInfoId int `description:"指标id"`
- }
- func GetEdbInfoCountByClassifyId(classifyId int) (count int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT COUNT(1) AS count FROM edb_info AS a
- WHERE a.classify_id IN(
- SELECT t.classify_id FROM
- (
- SELECT rd.*
- FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
- (SELECT @pid := ?) pd
- WHERE FIND_IN_SET(parent_id, @pid) > 0
- AND @pid := CONCAT(@pid, ',', classify_id)
- UNION SELECT * FROM edb_classify WHERE classify_id = @pid
- )AS t
- ) `
- err = o.Raw(sql, classifyId).QueryRow(&count)
- return
- }
- func DeleteEdbClassify(classifyId int) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` DELETE FROM edb_classify
- WHERE classify_id IN(
- SELECT t.classify_id FROM
- (
- SELECT rd.*
- FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
- (SELECT @pid := ?) pd
- WHERE FIND_IN_SET(parent_id, @pid) > 0
- AND @pid := CONCAT(@pid, ',', classify_id)
- UNION SELECT * FROM edb_classify WHERE classify_id = @pid
- )AS t
- ) `
- _, err = o.Raw(sql, classifyId).Exec()
- return
- }
- func GetEdbClassifyByParentId(parentId int, classifyType int8) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE parent_id=? AND classify_type = ? order by sort asc,classify_id asc`
- _, err = o.Raw(sql, parentId, classifyType).QueryRows(&items)
- return
- }
- func GetEdbClassifyByRootIdLevel(rootId int, classifyType uint8, orderStr string) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE root_id=? AND classify_type = ? `
- if orderStr != "" {
- sql += orderStr
- } else {
- sql += ` order by level desc, sort asc, classify_id asc`
- }
- _, err = o.Raw(sql, rootId, classifyType).QueryRows(&items)
- return
- }
- func GetEdbClassifyByRootIdLevelOrder(rootId int, classifyType uint8) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE root_id=? AND classify_type = ? order by level desc, sort asc, classify_id asc`
- _, err = o.Raw(sql, rootId, classifyType).QueryRows(&items)
- return
- }
- // GetEdbClassifyAll 获取普通指标的分类列表
- func GetEdbClassifyAll() (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE parent_id<>0 AND classify_type = 0 order by sort asc,classify_id asc`
- _, err = o.Raw(sql).QueryRows(&items)
- return
- }
- // GetAllEdbClassify 获取所有的普通指标的分类列表(包含第一级)
- func GetAllEdbClassify() (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE classify_type = 0 order by sort asc,classify_id asc`
- _, err = o.Raw(sql).QueryRows(&items)
- return
- }
- // GetNormalEdbClassifyAll 获取普通指标的分类列表
- func GetNormalEdbClassifyAll() (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE classify_type = 0 order by sort asc,classify_id asc`
- _, err = o.Raw(sql).QueryRows(&items)
- return
- }
- // GetPredictEdbClassifyAll 获取预测指标的分类列表
- func GetPredictEdbClassifyAll() (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE classify_type = 1 order by sort asc,classify_id asc`
- _, err = o.Raw(sql).QueryRows(&items)
- return
- }
- // GetAllEdbClassifyByType 根据类型获取分类列表
- func GetAllEdbClassifyByType(classifyType int) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE classify_type = ? order by sort ASC,classify_id ASC`
- _, err = o.Raw(sql, classifyType).QueryRows(&items)
- return
- }
- type EdbClassifyItems struct {
- ClassifyId int `description:"分类id"`
- EdbInfoId int `description:"指标id"`
- ClassifyName string
- ClassifyNameEn string
- ParentId int
- RootId int `description:"顶级id"`
- Level int `description:"层级"`
- Sort int `description:"排序字段,越小越靠前,默认值:10"`
- UniqueCode string `description:"唯一编码"`
- Source int `description:"来源id"`
- SourceName string `description:"来源名称"`
- SysUserId int `description:"创建人id"`
- SysUserRealName string `description:"创建人姓名"`
- StartDate string
- EdbCode string
- EdbType int `description:"指标类型:1:基础指标,2:计算指标"`
- Children []*EdbClassifyItems
- Button EdbClassifyItemsButton `description:"操作权限"`
- IsJoinPermission int `description:"是否加入权限管控,0:不加入;1:加入;默认:0"`
- HaveOperaAuth bool `description:"是否有数据权限"`
- }
- type EdbClassifyIdItems struct {
- ClassifyId int `description:"分类id"`
- ClassifyName string
- ClassifyNameEn string
- 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"`
- }
- type EdbClassifyItemList []*EdbClassifyItems
- func (m EdbClassifyItemList) Len() int {
- return len(m)
- }
- func (m EdbClassifyItemList) Less(i, j int) bool {
- return m[i].Sort < m[j].Sort
- }
- func (m EdbClassifyItemList) Swap(i, j int) {
- m[i], m[j] = m[j], m[i]
- }
- // EdbClassifyItemsButton 操作按钮
- type EdbClassifyItemsButton struct {
- AddButton bool `description:"是否可添加"`
- OpButton bool `description:"是否可编辑"`
- DeleteButton bool `description:"是否可删除"`
- MoveButton bool `description:"是否可移动"`
- ShowEdbRelation bool `description:"是否展示关联指标"`
- ShowChartRelation bool `description:"是否展示关联图表"`
- InsertNewDataButton bool `description:"是否可插入最新值"`
- }
- type EdbClassifyListResp struct {
- AllNodes EdbClassifyItemList
- CanOpClassify bool `description:"是否允许操作分类"`
- Language string `description:"指标的展示语言,CN:中文,EN:英文"`
- }
- type ClassifyDeleteCheckResp struct {
- DeleteStatus int `description:"检测状态:0:默认值,如果为0,继续走其他校验,1:若目录关联指标不可删除,2:确认删除当前目录及包含的子目录吗,3:当前指标已用作画图,不可删除"`
- TipsMsg string `description:"提示信息"`
- TableList []*ExcelBaseInfo `description:"关联的表格"`
- }
- type ExcelBaseInfo struct {
- ExcelInfoId int `orm:"column(excel_info_id);pk"`
- Source int `description:"表格来源,1:excel插件的表格,2:自定义表格,3:混合表格,4:自定义分析,默认:1"`
- ExcelType int `description:"表格类型,1:指标列,2:日期列,默认:1"`
- ExcelName string `description:"表格名称"`
- UniqueCode string `description:"表格唯一编码"`
- ExcelClassifyId int `description:"表格分类id"`
- //SysUserId int `description:"操作人id"`
- //SysUserRealName string `description:"操作人真实姓名"`
- }
- type ClassifyDeleteCheckReq struct {
- ClassifyId int `description:"分类id"`
- EdbInfoId int `description:"指标id"`
- }
- func GetClassifyCountByClassifyId(classifyId int) (count int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT COUNT(1) AS count FROM (
- SELECT rd.*
- FROM (SELECT * FROM edb_classify WHERE parent_id IS NOT NULL) rd,
- (SELECT @pid := ?) pd
- WHERE FIND_IN_SET(parent_id, @pid) > 0
- AND @pid := CONCAT(@pid, ',', classify_id)
- UNION SELECT * FROM edb_classify WHERE classify_id = @pid
- )AS t
- WHERE t.classify_id<>? `
- err = o.Raw(sql, classifyId, classifyId).QueryRow(&count)
- return
- }
- type EdbClassifySimplify struct {
- ClassifyId int `description:"分类id"`
- ClassifyName string `description:"分类名称"`
- ParentId int
- }
- // 递归获取所有目录
- func GetClassifyAllByClassifyId(classifyId int) (resultList []*EdbClassifySimplify, err error) {
- fmt.Println("185")
- o := orm.NewOrmUsingDB("data")
- for {
- sql := ` SELECT * FROM edb_classify AS a
- WHERE a.classify_id=? order by sort asc,classify_id asc`
- item := new(EdbClassifySimplify)
- err = o.Raw(sql, classifyId).QueryRow(&item)
- if err != nil {
- break
- }
- if item.ClassifyId > 0 {
- resultList = append(resultList, item)
- classifyId = item.ParentId
- } else {
- break
- }
- }
- return resultList, err
- }
- // MoveEdbClassifyReq 移动分类请求参数
- type MoveEdbClassifyReq struct {
- ClassifyId int `description:"分类id"`
- ParentClassifyId int `description:"父级分类id"`
- PrevClassifyId int `description:"上一个兄弟节点分类id"`
- NextClassifyId int `description:"下一个兄弟节点分类id"`
- EdbInfoId int `description:"指标ID, 如果指标ID有值,则移动对象为指标,否则认为移动对象为分类"`
- PrevEdbInfoId int `description:"上一个指标ID"`
- NextEdbInfoId int `description:"下一个指标ID"`
- }
- // GetFirstEdbClassifyByParentId 获取当前父级分类下,且排序数相同 的排序第一条的数据
- func GetFirstEdbClassifyByParentId(parentId int) (item *EdbClassify, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE parent_id=? order by sort asc,classify_id asc limit 1`
- err = o.Raw(sql, parentId).QueryRow(&item)
- return
- }
- // UpdateEdbClassifySortByParentId 根据父类id更新排序
- func UpdateEdbClassifySortByParentId(parentId, classifyId, nowSort int, updateSort string, classifyType uint8) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` update edb_classify set sort = ` + updateSort + ` WHERE parent_id=? AND sort > ? AND classify_type = ? `
- if classifyId > 0 {
- sql += ` or ( classify_id > ` + fmt.Sprint(classifyId) + ` and sort = ` + fmt.Sprint(nowSort) + `)`
- }
- _, err = o.Raw(sql, parentId, nowSort, classifyType).Exec()
- return
- }
- // Update 更新分类基础信息
- func (edbClassify *EdbClassify) Update(cols []string) (err error) {
- o := orm.NewOrmUsingDB("data")
- _, err = o.Update(edbClassify, cols...)
- return
- }
- // GetEdbClassifyMaxSort 获取分类下最大的排序数
- func GetEdbClassifyMaxSort(parentId int, classifyType uint8) (sort int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT Max(sort) AS sort FROM edb_classify WHERE parent_id=? AND classify_type=? `
- err = o.Raw(sql, parentId, classifyType).QueryRow(&sort)
- return
- }
- // GetEdbInfoMaxSortByClassifyId 获取分类下指标的最大的排序数
- func GetEdbInfoMaxSortByClassifyId(classifyId int) (sort int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT Max(sort) AS sort FROM edb_info WHERE classify_id=? `
- err = o.Raw(sql, classifyId).QueryRow(&sort)
- return
- }
- type ClassifyEdbInfoListResp struct {
- EdbInfoList []*EdbClassifyItems
- }
- // GetEdbClassifyAllV2 获取普通指标的分类列表
- func GetEdbClassifyAllV2(classifyType int8) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE parent_id <>0 AND classify_type = ? order by sort asc,classify_id asc`
- _, err = o.Raw(sql, classifyType).QueryRows(&items)
- return
- }
- func GetEdbClassifyByParentIdTwo(classifyType int8) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE level=2 AND classify_type = ? order by sort asc,classify_id asc`
- _, err = o.Raw(sql, classifyType).QueryRows(&items)
- return
- }
- type AddEdbClassifyReqV2 struct {
- ClassifyName string `description:"分类名称"`
- ParentId int `description:"父级id,第一级传0"`
- Level int `description:"层级,第一级传0,其余传上一级的层级"`
- ClassifyType uint8 `description:"分类类型,0:普通指标分类,1:预测指标分类"`
- }
- // GetEdbClassifyMaxSortV2 获取分类下最大的排序数
- func GetEdbClassifyMaxSortV2(parentId int, classifyType uint8) (sort int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT Max(sort) AS sort FROM edb_classify WHERE parent_id=? AND parent_id_two=? AND classify_type=? `
- err = o.Raw(sql, parentId, classifyType).QueryRow(&sort)
- return
- }
- func GetEdbClassifyByLevel(level, source int) (items []*EdbClassifyItems, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` SELECT * FROM edb_classify WHERE level=? AND classify_type = ? `
- _, err = o.Raw(sql, level, source).QueryRows(&items)
- return
- }
- func FIXEdbClassifyId(newId, oldId int) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := ` UPDATE edb_info SET classify_id=? WHERE classify_id=? `
- _, err = o.Raw(sql, newId, oldId).Exec()
- return
- }
- func FixEdbClassify() {
- //新增三级分类
- list, err := GetEdbClassifyByLevel(2, 0)
- if err != nil {
- return
- }
- for _, v := range list {
- timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- classifyInfo := &EdbClassify{
- //ClassifyId: 0,
- ClassifyType: 0,
- ClassifyName: v.ClassifyName,
- ParentId: v.ClassifyId,
- HasData: 0,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- SysUserRealName: "admin",
- Level: 3,
- UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
- Sort: v.Sort,
- }
- if utils.RunMode == "debug" {
- classifyInfo.SysUserId = 11
- } else {
- classifyInfo.SysUserId = 163
- }
- id, err := AddEdbClassify(classifyInfo)
- if err != nil {
- return
- }
- err = FIXEdbClassifyId(int(id), v.ClassifyId)
- if err != nil {
- return
- }
- }
- }
- func FixPredictEdbClassify() {
- //新增三级分类
- list, err := GetEdbClassifyByLevel(2, 1)
- if err != nil {
- return
- }
- for _, v := range list {
- timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
- classifyInfo := &EdbClassify{
- //ClassifyId: 0,
- ClassifyType: 1,
- ClassifyName: v.ClassifyName,
- ParentId: v.ClassifyId,
- HasData: 0,
- CreateTime: time.Now(),
- ModifyTime: time.Now(),
- SysUserRealName: "admin",
- Level: 3,
- UniqueCode: utils.MD5(utils.DATA_PREFIX + "_" + timestamp),
- Sort: v.Sort,
- }
- if utils.RunMode == "debug" {
- classifyInfo.SysUserId = 11
- } else {
- classifyInfo.SysUserId = 163
- }
- id, err := AddEdbClassify(classifyInfo)
- if err != nil {
- return
- }
- err = FIXEdbClassifyId(int(id), v.ClassifyId)
- if err != nil {
- return
- }
- }
- }
- func GetEdbClassify(classifyId int) (classifyIds string, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT GROUP_CONCAT(t.classify_id) AS classify_ids FROM (
- SELECT a.classify_id FROM edb_classify AS a
- WHERE a.classify_id=?
- UNION ALL
- SELECT a.classify_id FROM edb_classify AS a
- WHERE a.parent_id=? UNION ALL
- SELECT
- classify_id
- FROM
- edb_classify
- WHERE
- parent_id IN ( SELECT classify_id FROM edb_classify WHERE parent_id = ? )
- )AS t`
- err = o.Raw(sql, classifyId, classifyId, classifyId).QueryRow(&classifyIds)
- return
- }
- func FixEdbClassifySysUser() {
- list, err := GetEdbClassifyByLevel(2, 0)
- if err != nil {
- return
- }
- for _, v := range list {
- err := EditEdbClassifySysUser(v.ClassifyId, v.SysUserId, v.SysUserRealName)
- if err != nil {
- return
- }
- fmt.Println("ClassifyId:", v.ClassifyId)
- }
- fmt.Println("FixEdbClassifySysUser end")
- }
- func EditEdbClassifySysUser(classifyId, sysUserId int, chartClassifyName string) (err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `UPDATE edb_classify SET sys_user_id=?,sys_user_real_name=?,modify_time=NOW() WHERE parent_id=? AND level=3 `
- _, err = o.Raw(sql, sysUserId, chartClassifyName, classifyId).Exec()
- return
- }
- func FixPredictEdbClassifySysUser() {
- list, err := GetEdbClassifyByLevel(2, 1)
- if err != nil {
- return
- }
- for _, v := range list {
- err := EditEdbClassifySysUser(v.ClassifyId, v.SysUserId, v.SysUserRealName)
- if err != nil {
- return
- }
- fmt.Println("ClassifyId:", v.ClassifyId)
- }
- fmt.Println("EditChartClassifySysUser end")
- }
- func UpdateEdbClassifyChildByParentClassifyId(classifyIds []int, rootId int, levelStep int) (err error) {
- o := orm.NewOrmUsingDB("data")
- var pars []interface{}
- pars = append(pars, rootId, levelStep)
- pars = append(pars, classifyIds)
- // 更新相关联的二级分类的parentId,和classify_name_second
- sql := `update edb_classify
- SET root_id = ?, level = level+?
- where classify_id IN (` + utils.GetOrmInReplace(len(classifyIds)) + `)`
- _, err = o.Raw(sql, pars).Exec()
- if err != nil {
- return
- }
- return
- }
- // GetEdbClassifyByParentIdAndName 根据父级ID和名称获取分类
- func GetEdbClassifyByParentIdAndName(parentId int, classifyName string, classifyId int) (item *EdbClassify, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT * FROM edb_classify WHERE parent_id = ? AND classify_name = ? AND classify_id <> ? LIMIT 1`
- err = o.Raw(sql, parentId, classifyName, classifyId).QueryRow(&item)
- return
- }
- // GetEdbClassifyByIdList
- // @Description: 根据分类id列表获取分类列表
- // @author: Roc
- // @datetime 2024-03-29 10:30:21
- // @param classifyId int
- // @return item *EdbClassify
- // @return err error
- func GetEdbClassifyByIdList(classifyIdList []int) (item []*EdbClassify, err error) {
- num := len(classifyIdList)
- if num <= 0 {
- return
- }
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT * FROM edb_classify WHERE classify_id in (` + utils.GetOrmInReplace(num) + `) `
- _, err = o.Raw(sql, classifyIdList).QueryRows(&item)
- return
- }
- // GetEdbClassifyByClassifyTypeAndIsJoinPermission
- // @Description: 根据分类类型和是否涉密获取分类列表
- // @author: Roc
- // @datetime 2024-03-29 10:30:21
- // @param classifyType int
- // @param isJoinPermission int
- // @return items []*EdbClassify
- // @return err error
- func GetEdbClassifyByClassifyTypeAndIsJoinPermission(classifyType, isJoinPermission int) (items []*EdbClassify, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT * FROM edb_classify WHERE classify_type = ? AND is_join_permission = ? `
- _, err = o.Raw(sql, classifyType, isJoinPermission).QueryRows(&items)
- return
- }
- // GetEdbClassifyRootIdsByClassifyIds 获取普通指标的顶级分类列表
- func GetEdbClassifyRootIdsByClassifyIds(classifyIds []int) (items []int, err error) {
- o := orm.NewOrmUsingDB("data")
- sql := `SELECT distinct root_id FROM edb_classify WHERE classify_type=0 and classify_id in (` + utils.GetOrmInReplace(len(classifyIds)) + `)`
- _, err = o.Raw(sql, classifyIds).QueryRows(&items)
- return
- }
|