eta_api/models/data_manage/chart_framework.go

package data_manage

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

// ChartFramework 图库框架表
type ChartFramework struct {
	ChartFrameworkId int       `orm:"column(chart_framework_id);pk"`
	FrameworkCode    string    `description:"框架唯一编码"`
	FrameworkName    string    `description:"框架名称"`
	FrameworkImg     string    `description:"框架图片"`
	FrameworkContent string    `description:"框架内容"`
	IsPublic         int       `description:"是否公开：0-私有；1-公开"`
	PublicTime       time.Time `description:"公开时间"`
	Sort             int       `description:"排序"`
	AdminId          int       `description:"创建人ID"`
	AdminName        string    `description:"创建人姓名"`
	CreateTime       time.Time `description:"创建时间"`
	ModifyTime       time.Time `description:"更新时间"`
}

func (m *ChartFramework) TableName() string {
	return "chart_framework"
}

func (m *ChartFramework) PrimaryId() string {
	return ChartFrameworkColumns.ChartFrameworkId
}

var ChartFrameworkColumns = struct {
	ChartFrameworkId string
	FrameworkCode    string
	FrameworkName    string
	FrameworkImg     string
	FrameworkContent string
	IsPublic         string
	PublicTime       string
	Sort             string
	AdminId          string
	AdminName        string
	CreateTime       string
	ModifyTime       string
}{
	ChartFrameworkId: "chart_framework_id",
	FrameworkCode:    "framework_code",
	FrameworkName:    "framework_name",
	FrameworkImg:     "framework_img",
	FrameworkContent: "framework_content",
	IsPublic:         "is_public",
	PublicTime:       "public_time",
	Sort:             "sort",
	AdminId:          "admin_id",
	AdminName:        "admin_name",
	CreateTime:       "create_time",
	ModifyTime:       "modify_time",
}

func (m *ChartFramework) Create() (err error) {
	o := orm.NewOrmUsingDB("data")
	id, err := o.Insert(m)
	if err != nil {
		return
	}
	m.ChartFrameworkId = int(id)
	return
}

func (m *ChartFramework) CreateMulti(items []*ChartFramework) (err error) {
	if len(items) == 0 {
		return
	}
	o := orm.NewOrmUsingDB("data")
	_, err = o.InsertMulti(len(items), items)
	return
}

func (m *ChartFramework) Update(cols []string) (err error) {
	o := orm.NewOrmUsingDB("data")
	_, err = o.Update(m, cols...)
	return
}

func (m *ChartFramework) Del() (err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	_, err = o.Raw(sql, m.ChartFrameworkId).Exec()
	return
}

func (m *ChartFramework) GetItemById(id int) (item *ChartFramework, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	err = o.Raw(sql, id).QueryRow(&item)
	return
}

func (m *ChartFramework) GetItemByCondition(condition string, pars []interface{}) (item *ChartFramework, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s LIMIT 1`, m.TableName(), condition)
	err = o.Raw(sql, pars).QueryRow(&item)
	return
}

func (m *ChartFramework) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	o := orm.NewOrmUsingDB("data")
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = o.Raw(sql, pars).QueryRow(&count)
	return
}

func (m *ChartFramework) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*ChartFramework, err error) {
	o := orm.NewOrmUsingDB("data")
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := `ORDER BY create_time DESC`
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	_, err = o.Raw(sql, pars).QueryRows(&items)
	return
}

func (m *ChartFramework) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*ChartFramework, err error) {
	o := orm.NewOrmUsingDB("data")
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := `ORDER BY create_time DESC`
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
	_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
	return
}

func (m *ChartFramework) CreateFrameworkAndNodes(item *ChartFramework, nodes []*ChartFrameworkNode) (err error) {
	o := orm.NewOrmUsingDB("data")
	tx, e := o.Begin()
	if e != nil {
		err = fmt.Errorf("orm begin err: %s", e.Error())
		return
	}
	defer func() {
		if err != nil {
			_ = tx.Rollback()
			return
		}
		_ = tx.Commit()
	}()

	id, e := tx.Insert(item)
	if e != nil {
		err = fmt.Errorf("insert framework err: %s", e.Error())
		return
	}
	newId := int(id)
	item.ChartFrameworkId = newId

	if len(nodes) > 0 {
		for _, n := range nodes {
			n.ChartFrameworkId = newId
		}
		_, e = tx.InsertMulti(len(nodes), nodes)
		if e != nil {
			err = fmt.Errorf("insert multi nodes err: %s", e.Error())
			return
		}
	}
	return
}

func (m *ChartFramework) EditFrameworkAndNodes(item *ChartFramework, updateCols []string, nodes []*ChartFrameworkNode) (err error) {
	o := orm.NewOrmUsingDB("data")
	tx, e := o.Begin()
	if e != nil {
		err = fmt.Errorf("orm begin err: %s", e.Error())
		return
	}
	defer func() {
		if err != nil {
			_ = tx.Rollback()
			return
		}
		_ = tx.Commit()
	}()

	_, e = tx.Update(item, updateCols...)
	if e != nil {
		err = fmt.Errorf("framework update err: %s", e.Error())
		return
	}

	sql := `DELETE FROM chart_framework_node WHERE chart_framework_id = ?`
	_, e = tx.Raw(sql, item.ChartFrameworkId).Exec()
	if e != nil {
		err = fmt.Errorf("clear nodes err: %s", e.Error())
		return
	}
	if len(nodes) > 0 {
		_, e = tx.InsertMulti(len(nodes), nodes)
		if e != nil {
			err = fmt.Errorf("insert multi nodes err: %s", e.Error())
			return
		}
	}
	return
}

// ChartFrameworkAddReq 图库框架新增请求体
type ChartFrameworkAddReq struct {
	FrameworkName    string `description:"框架名称"`
	FrameworkImg     string `description:"框架图片"`
	FrameworkContent string `description:"框架内容"`
	Nodes            []int  `description:"框架关联的所有节点ID"`
}

// ChartFrameworkEditReq 图库框架编辑请求体
type ChartFrameworkEditReq struct {
	ChartFrameworkId int `description:"图库框架ID"`
	ChartFrameworkAddReq
}

// ChartFrameworkRemoveReq 图库框架编辑请求体
type ChartFrameworkRemoveReq struct {
	ChartFrameworkId int `description:"图库框架ID"`
}

// ChartFrameworkRenameReq 图库框架重命名请求体
type ChartFrameworkRenameReq struct {
	ChartFrameworkId int `description:"图库框架ID"`
	FrameworkName    int `description:"框架名称"`
}

// ChartFrameworkEditPublicReq 图库框架编辑公开请求体
type ChartFrameworkEditPublicReq struct {
	ChartFrameworkId int `description:"图库框架ID"`
	IsPublic         int `description:"0-隐藏公开; 1-公开"`
}

// ChartFrameworkMoveReq 图库框架移动排序请求体
type ChartFrameworkMoveReq struct {
	ChartFrameworkId     int `description:"图库框架ID"`
	PrevChartFrameworkId int `description:"上一个框架ID"`
	NextChartFrameworkId int `description:"下一个框架ID"`
}