eta_api/models/data_stat/edb_info_stat_column.go

package data_stat

import (
	"eta_gn/eta_api/global"
)

// EdbInfoStatColumn 数据源统计表自定义列
//type EdbInfoStatColumn struct {
//	Id           int    `orm:"column(id);pk" description:"自增id"`
//	ColumnName   string `description:"字段名称"`
//	ColumnKey    string `description:"字段"`
//	CreateTime   string `description:" 创建时间"`
//	ModifyTime   string `description:"修改时间"`
//	Type         int    `description:"列表类型：1数据源明细表，2数据源统计表，3删除指标列表，4指标信息变更表"`
//	Sort         int    `description:"列排序"`
//	IsMust       int    `description:"是否必选列: 0非必选 1必须"`
//	IsSort       int    `description:"是否是允许排序的列，0否，1是"`
//	IsShow       int    `description:"是否展示：1展示，0不展示"`
//	ColumnNameEn string `description:"英文字段名称"`
//}

type EdbInfoStatColumn struct {
	Id           int    `gorm:"primaryKey;column:id;type:int(11) unsigned;not null"` // 自增id
	ColumnName   string `gorm:"column:column_name;type:varchar(100);not null"`       // 字段名称
	ColumnKey    string `gorm:"column:column_key;type:varchar(100);not null"`        // 字段
	CreateTime   string `gorm:"column:create_time;type:datetime"`                    // 创建时间
	ModifyTime   string `gorm:"column:modify_time;type:datetime"`                    // 修改时间
	Type         int    `gorm:"column:type;type:tinyint(4);not null;default:0"`      // 列表类型：1数据源明细表，2数据源统计表，3删除指标列表，4指标信息变更表
	Sort         int    `gorm:"column:sort;type:int(11);not null;default:0"`         // 列排序
	IsMust       int    `gorm:"column:is_must;type:tinyint(4);not null;default:1"`   // 是否必选列 1必须，0非必选
	IsSort       int    `gorm:"column:is_sort;type:tinyint(4);not null"`             // 是否是允许排序的列，0否，1是
	IsShow       int    `gorm:"column:is_show;type:tinyint(4);not null;default:1"`   // 是否展示：1展示，0不展示
	ColumnNameEn string `gorm:"column:column_name_en;type:varchar(100);not null"`    // 英文字段名称
}

type EdbInfoStatColumnListItem struct {
	Id           int    `description:"自增id"`
	ColumnName   string `description:"字段名称"`
	ColumnKey    string `description:"字段"`
	Type         int    `description:"列表类型：1数据源明细表，2数据源统计表，3删除指标列表，4指标信息变更表"`
	IsMust       int    `description:"是否必选列: 0非必选 1必须"`
	IsSort       int    `description:"是否是允许排序的列，0否，1是"`
	IsShow       int    `description:"是否展示：1展示，0不展示"`
	ColumnNameEn string `description:"英文字段名称"`
}

type StatColumnList struct {
	List []*EdbInfoStatColumnListItem
}

type EditStatColumnReq struct {
	List []*EdbInfoStatColumnListReq
}
type EdbInfoStatColumnListReq struct {
	Id     int `description:"自增id"`
	IsShow int `description:"是否展示：1展示，0不展示"`
}

// GetStatColumn 获取自定义列
func GetStatColumn(columnType int) (list []EdbInfoStatColumn, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := `SELECT * FROM edb_info_stat_column WHERE type=? ORDER BY sort asc, id asc `
	//_, err = o.Raw(sql, columnType).QueryRows(&list)
	err = global.DmSQL["data"].Raw(sql, columnType).Find(&list).Error
	return
}

// GetAllStatColumn 获取所有自定义列
func GetAllStatColumn() (list []EdbInfoStatColumn, err error) {
	//o := orm.NewOrmUsingDB("data")
	sql := `SELECT * FROM edb_info_stat_column ORDER BY sort asc, id asc  `
	//_, err = o.Raw(sql).QueryRows(&list)
	err = global.DmSQL["data"].Raw(sql).Find(&list).Error
	return
}

// UpdateStatColumn 批量更新自定义列
func UpdateStatColumn(list []*EdbInfoStatColumn) (err error) {
	//o := orm.NewOrmUsingDB("data")
	//p, err := o.Raw("UPDATE edb_info_stat_column SET is_show = ?, sort=? WHERE id = ?").Prepare()
	//if err != nil {
	//	return
	//}
	//defer func() {
	//	_ = p.Close() // 别忘记关闭 statement
	//}()
	//for _, v := range list {
	//	_, err = p.Exec(v.IsShow, v.Sort, v.Id)
	//	if err != nil {
	//		return
	//	}
	//}
	tx := global.DmSQL["data"].Begin()
	for _, v := range list {
		err = tx.Exec("UPDATE edb_info_stat_column SET is_show = ?, sort=? WHERE id = ?", v.IsShow, v.Sort, v.Id).Error
		if err != nil {
			tx.Rollback()
			return
		}
	}
	tx.Commit()
	return
}