eta_api/models/edb_monitor/edb_monitor_classify.go

package edbmonitor

import (
	"eta/eta_api/utils"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

type EdbMonitorClassify struct {
	ClassifyId   int       `orm:"column(classify_id);pk"`
	ClassifyName string    `description:"分类名称"`
	Level        int       `description:"分类层级"`
	ParentId     int       `description:"父分类ID"`
	RootId       int       `description:"根分类ID"`
	Sort         int       `description:"排序"`
	CreateTime   time.Time `description:"创建时间"`
}

func (e *EdbMonitorClassify) Insert() (id int64, err error) {
	o := orm.NewOrmUsingDB("data")
	id, err = o.Insert(e)
	return
}

func (e *EdbMonitorClassify) Update(cols []string, classifyId []int) (err error) {
	tx, err := orm.NewOrmUsingDB("data").Begin()
	if err != nil {
		return
	}
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	_, err = tx.Update(e, cols...)
	if err != nil {
		return
	}
	if len(classifyId) == 0 {
		return
	}
	sql := `UPDATE edb_monitor_classify SET root_id = ? WHERE classify_id IN (` + utils.GetOrmInReplace(len(classifyId)) + `)`
	_, err = tx.Raw(sql, e.RootId, classifyId).Exec()
	if err != nil {
		return
	}
	return
}

func DeleteEdbMonitorClassifyById(id int) (err error) {
	o := orm.NewOrmUsingDB("data")
	_, err = o.Raw("DELETE FROM edb_monitor_classify WHERE classify_id =?", id).Exec()
	return
}

func GetEdbMonitorClassifyList() (items []*EdbMonitorClassify, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT * FROM edb_monitor_classify"
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

func GetEdbMonitorClassifyById(id int) (item *EdbMonitorClassify, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT * FROM edb_monitor_classify WHERE classify_id =?"
	err = o.Raw(sql, id).QueryRow(&item)
	return
}

func GetEdbMonitorClassifyMaxSortByParentId(parentId int) (sort int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT MAX(sort) FROM edb_monitor_classify WHERE parent_id =?"
	err = o.Raw(sql, parentId).QueryRow(&sort)
	return
}

func GetChildEdbMonitorClassifyIdById(id int) (classifyId []int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT classify_id FROM edb_monitor_classify WHERE parent_id =?"
	_, err = o.Raw(sql, id).QueryRows(&classifyId)
	return
}

func GetChildEdbMonitorClassifyById(id int) (items []*EdbMonitorClassify, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT * FROM edb_monitor_classify WHERE parent_id =?"
	_, err = o.Raw(sql, id).QueryRows(&items)
	return
}

func GetChildEdbMonitorClassifyByRootId(rootId int) (items []*EdbMonitorClassify, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := "SELECT * FROM edb_monitor_classify WHERE root_id =?"
	_, err = o.Raw(sql, rootId).QueryRows(&items)
	return
}

func GetEdbMonitorClassifyCountByIdList(ids []int) (count int, err error) {
	if len(ids) == 0 {
		return
	}
	o := orm.NewOrmUsingDB("data")
	sql := `SELECT COUNT(*) FROM edb_monitor_classify WHERE classify_id IN (` + utils.GetOrmInReplace(len(ids)) + `)`
	err = o.Raw(sql, ids).QueryRow(&count)
	return
}