package data_manage
import (
"eta_gn/eta_api/global"
"eta_gn/eta_api/utils"
"fmt"
"github.com/rdlucklib/rdluck_tools/paging"
"strconv"
"strings"
"time"
)
// ChartCollect 图表收藏
type ChartCollect struct {
ChartCollectId int `gorm:"primaryKey;autoIncrement;column:chart_collect_id;type:int(10) unsigned;not null"`
ChartCollectClassifyId int `gorm:"index:idx_classify_id;column:chart_collect_classify_id;type:int(10) unsigned;not null;default:0"` // 图表收藏分类ID
ChartInfoId int `gorm:"column:chart_info_id;type:int(10) unsigned;not null;default:0"` // 图表ID
SysUserId int `gorm:"column:sys_user_id;type:int(10) unsigned;not null;default:0"` // 创建人ID
SysRealName string `gorm:"column:sys_real_name;type:int(10) unsigned;not null;default:0"` // 创建人姓名
Sort int `gorm:"column:sort;type:int(10);default:0"` // 排序
CreateTime time.Time `gorm:"column:create_time;type:datetime"` // 创建时间
ModifyTime time.Time `gorm:"column:modify_time;type:datetime"` // 更新时间
}
func (m *ChartCollect) TableName() string {
return "chart_collect"
}
type ChartCollectCols struct {
PrimaryId string
ChartCollectClassifyId string
ChartInfoId string
SysUserId string
SysRealName string
Sort string
CreateTime string
ModifyTime string
}
func (m *ChartCollect) Cols() ChartCollectCols {
return ChartCollectCols{
PrimaryId: "chart_collect_id",
ChartCollectClassifyId: "chart_collect_classify_id",
ChartInfoId: "chart_info_id",
SysUserId: "sys_user_id",
SysRealName: "sys_real_name",
Sort: "sort",
CreateTime: "create_time",
ModifyTime: "modify_time",
}
}
func (m *ChartCollect) Create() (err error) {
err = global.DmSQL["data"].Create(m).Error
return
}
func (m *ChartCollect) CreateMulti(items []*ChartCollect) (err error) {
if len(items) == 0 {
return
}
err = global.DmSQL["data"].CreateInBatches(items, utils.MultiAddNum).Error
return
}
func (m *ChartCollect) Update(cols []string) (err error) {
err = global.DmSQL["data"].Select(cols).Updates(m).Error
return
}
func (m *ChartCollect) Remove() (err error) {
sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
err = global.DmSQL["data"].Exec(sql, m.ChartCollectId).Error
return
}
func (m *ChartCollect) MultiRemove(ids []int) (err error) {
if len(ids) == 0 {
return
}
sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.Cols().PrimaryId, utils.GetOrmInReplace(len(ids)))
err = global.DmSQL["data"].Exec(sql, ids).Error
return
}
func (m *ChartCollect) RemoveByCondition(condition string, pars []interface{}) (err error) {
if condition == "" {
return
}
sql := fmt.Sprintf(`DELETE FROM %s WHERE %s`, m.TableName(), condition)
err = global.DmSQL["data"].Exec(sql, pars...).Error
return
}
// RemoveAndCreateMulti
// @Description: 清除原有配置并新增收藏
// @receiver m
// @param delCondition
// @param delPars
// @param items
// @return err
func (m *ChartCollect) RemoveAndCreateMulti(delCondition string, delPars []interface{}, items []*ChartCollect) (err error) {
to := global.DmSQL["data"].Begin()
defer func() {
if err != nil {
_ = to.Rollback()
} else {
_ = to.Commit()
}
}()
// 先删除
{
if delCondition == "" {
return
}
sql := fmt.Sprintf(`DELETE FROM %s WHERE %s`, m.TableName(), delCondition)
err = to.Exec(sql, delPars...).Error
if err != nil {
return
}
}
if len(items) == 0 {
return
}
err = to.CreateInBatches(items, utils.MultiAddNum).Error
return
}
func (m *ChartCollect) GetItemById(id int) (item *ChartCollect, err error) {
sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.Cols().PrimaryId)
err = global.DmSQL["data"].Raw(sql, id).First(&item).Error
return
}
func (m *ChartCollect) GetItemByCondition(condition string, pars []interface{}, orderRule string) (item *ChartCollect, err error) {
order := ``
if orderRule != "" {
order = ` ORDER BY ` + orderRule
}
sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s %s LIMIT 1`, m.TableName(), condition, order)
err = global.DmSQL["data"].Raw(sql, pars...).First(&item).Error
return
}
func (m *ChartCollect) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
err = global.DmSQL["data"].Raw(sql, pars...).Scan(&count).Error
return
}
func (m *ChartCollect) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*ChartCollect, err error) {
fields := strings.Join(fieldArr, ",")
if len(fieldArr) == 0 {
fields = `*`
}
order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
if orderRule != "" {
order = ` ORDER BY ` + orderRule
}
sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
return
}
func (m *ChartCollect) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*ChartCollect, err error) {
fields := strings.Join(fieldArr, ",")
if len(fieldArr) == 0 {
fields = `*`
}
order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
if orderRule != "" {
order = ` ORDER BY ` + orderRule
}
sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s LIMIT ?,?`, fields, m.TableName(), condition, order)
pars = append(pars, startSize, pageSize)
err = global.DmSQL["data"].Raw(sql, pars...).Find(&items).Error
return
}
// GetCollectChartInfoByClassifyId 获取分类下收藏的图表信息
func GetCollectChartInfoByClassifyId(classifyId int) (items []*ChartInfo, err error) {
sql := `SELECT b.* FROM chart_collect AS a JOIN chart_info AS b ON a.chart_info_id = b.chart_info_id WHERE a.chart_collect_classify_id = ? ORDER BY a.sort ASC`
err = global.DmSQL["data"].Raw(sql, classifyId).Find(&items).Error
return
}
// ChartCollectReq 加入/取消收藏
type ChartCollectReq struct {
ClassifyId int `description:"分类ID"`
ClassifyIdList []int `description:"分类ID列表"`
ChartInfoId int `description:"图表ID"`
}
// GetCollectChartInfoCount 获取收藏的图表信息总数
func GetCollectChartInfoCount(condition string, pars []interface{}) (total int64, err error) {
sql := fmt.Sprintf(`SELECT COUNT(1) AS ct FROM (
SELECT b.chart_info_id FROM chart_collect AS a JOIN chart_info AS b ON a.chart_info_id = b.chart_info_id
WHERE 1=1 %s
GROUP BY b.chart_info_id
) AS sub`, condition)
err = global.DmSQL["data"].Raw(sql, pars...).Scan(&total).Error
return
}
// GetCollectChartInfoPageList 获取收藏的图表信息列表-分页
func GetCollectChartInfoPageList(condition string, pars []interface{}, startSize, pageSize int) (list []*CollectChartInfoQuery, err error) {
sql := fmt.Sprintf(`SELECT b."chart_info_id",
WM_CONCAT(DISTINCT a."chart_collect_classify_id") AS "collect_classify_id",
MAX(a.create_time) as collect_time,
MAX(b."chart_name") "chart_name",
MAX(b."chart_type") "chart_type",
MAX(b."source") "source",
MAX(b."chart_classify_id") "chart_classify_id",
MAX(b."create_time") "create_time",
MAX(b."unique_code") "unique_code",
MAX(b."chart_image") "chart_image",
MAX(b."modify_time") "modify_time",
MAX(a."sort") AS "sort"
FROM chart_collect AS a JOIN chart_info AS b ON a.chart_info_id = b.chart_info_id
WHERE 1=1 %s
GROUP BY b.chart_info_id
ORDER BY collect_time DESC LIMIT ?,?`, condition)
pars = append(pars, startSize, pageSize)
err = global.DmSQL["data"].Raw(sql, pars...).Scan(&list).Error
return
}
// GetUserAllCollectChartInfoIdList
// @Description: 获取用户所有收藏的图表id列表
// @author: Roc
// @datetime 2024-11-28 15:27:52
// @param userId int
// @return list []int
// @return err error
func GetUserAllCollectChartInfoIdList(userId int) (list []int, err error) {
sql := `SELECT chart_info_id FROM chart_collect WHERE 1=1 AND sys_user_id = ? GROUP BY chart_info_id`
err = global.DmSQL["data"].Raw(sql, userId).Scan(&list).Error
return
}
type CollectChartInfoQuery struct {
ChartInfo
CollectClassifyIdStr string `gorm:"column:collect_classify_id" description:"收藏分类ID"`
CollectTime time.Time `gorm:"column:collect_time" description:"收藏时间"`
}
// CollectChartInfoItem 收藏列表图表信息
type CollectChartInfoItem struct {
ChartInfoId int `description:"图表ID"`
ChartInfoType int `description:"图表类型：0-普通图表; 1-预测图表"`
ChartType int `description:"图表类型：1-基础图表; 2-计算图表"`
Source int `description:"来源ID"`
SourceName string `description:"来源名称"`
ChartName string `description:"图表名称"`
Frequency string `description:"频率"`
Unit string `description:"单位"`
UniqueCode string `description:"唯一编码"`
ChartImage string `description:"图表图片"`
ClassifyId int `description:"图表分类ID"`
CollectClassifyIdList []int `description:"收藏分类ID列表"`
CollectTime string `description:"收藏时间"`
CreateTime string `description:"创建时间"`
Sort int `description:"排序"`
}
func FormatChartInfo2CollectItem(origin *CollectChartInfoQuery) (item *CollectChartInfoItem) {
item = new(CollectChartInfoItem)
item.ChartInfoId = origin.ChartInfoId
item.ChartType = origin.ChartType
item.Source = origin.Source
item.ChartName = origin.ChartName
item.Unit = origin.Unit
item.UniqueCode = origin.UniqueCode
item.ChartImage = origin.ChartImage
item.ClassifyId = origin.ChartClassifyId
collectClassifyIdList := make([]int, 0)
if origin.CollectClassifyIdStr != `` {
collectClassifyIdStrList := strings.Split(origin.CollectClassifyIdStr, ",")
for _, v := range collectClassifyIdStrList {
collectClassifyId, tmpErr := strconv.Atoi(v)
if tmpErr == nil {
collectClassifyIdList = append(collectClassifyIdList, collectClassifyId)
}
}
}
item.CollectClassifyIdList = collectClassifyIdList
item.CollectTime = origin.CollectTime.Format(utils.FormatDateTime)
item.CreateTime = origin.CreateTime.Format(utils.FormatDateTime)
item.Sort = origin.Sort
return
}
// CollectChartInfoListResp 收藏图表分页列表相应
type CollectChartInfoListResp struct {
Paging *paging.PagingItem
List []*CollectChartInfoItem
}
// ChartCollectMoveReq 移动收藏
type ChartCollectMoveReq struct {
ChartInfoId int `description:"收藏的图表ID(分类下图表ID唯一)"`
PrevChartInfoId int `description:"前一个收藏的图表ID"`
NextChartInfoId int `description:"后一个收藏的图表ID"`
ClassifyId int `description:"当前分类ID"`
}
func GetChartCollectSort(adminId, classifyId, sort int) (item *ChartCollect, err error) {
sql := ` SELECT * FROM chart_collect WHERE 1=1 AND sys_user_id = ? AND chart_collect_classify_id = ? `
if sort == 1 {
sql += ` ORDER BY sort DESC, chart_collect_id ASC LIMIT 1 `
} else {
sql += ` ORDER BY sort ASC, chart_collect_id DESC LIMIT 1 `
}
err = global.DmSQL["data"].Raw(sql, adminId, classifyId).First(&item).Error
return
}
func GetChartCollectByChartInfoId(adminId, chartInfoId, classifyId int) (item *ChartCollect, err error) {
sql := `SELECT * FROM chart_collect WHERE sys_user_id = ? AND chart_info_id = ? AND chart_collect_classify_id=? `
err = global.DmSQL["data"].Raw(sql, adminId, chartInfoId, classifyId).First(&item).Error
return
}
func UpdateChartCollectSortByClassifyId(classifyId, nowSort int, prevMyChartClassifyMappingId int, updateSort string) (err error) {
sql := ` update chart_collect set sort = ` + updateSort + ` WHERE chart_collect_classify_id = ? `
if prevMyChartClassifyMappingId > 0 {
sql += ` AND ( sort > ? or ( chart_collect_id < ? and sort=? )) `
}
err = global.DmSQL["data"].Exec(sql, classifyId, nowSort, prevMyChartClassifyMappingId, nowSort).Error
return
}
func UpdateChartCollectMove(sort, adminId, chartInfoId, myChartClassifyId int) (err error) {
sql := ` UPDATE chart_collect SET sort = ?,modify_time=NOW() WHERE sys_user_id=? AND chart_info_id=? AND chart_collect_classify_id=? `
err = global.DmSQL["data"].Exec(sql, sort, adminId, chartInfoId, myChartClassifyId).Error
return
}
// ChartCollectItem 图表收藏信息
type ChartCollectItem struct {
ChartCollectId int `gorm:"primaryKey;autoIncrement;column:chart_collect_id;type:int(10) unsigned;not null"`
ChartCollectClassifyId int `gorm:"index:idx_classify_id;column:chart_collect_classify_id;type:int(10) unsigned;not null;default:0"` // 图表收藏分类ID
ChartInfoId int `gorm:"column:chart_info_id;type:int(10) unsigned;not null;default:0"` // 图表ID
SysUserId int `gorm:"column:sys_user_id;type:int(10) unsigned;not null;default:0"` // 创建人ID
SysRealName string `gorm:"column:sys_real_name;type:int(10) unsigned;not null;default:0"` // 创建人姓名
ClassifyName string `gorm:"column:classify_name;type:varchar(255);not null;default:''"` // 分类名称
Sort int `gorm:"column:sort;type:int(10);default:0"` // 排序
CreateTime time.Time `gorm:"column:create_time;type:datetime"` // 创建时间
ModifyTime time.Time `gorm:"column:modify_time;type:datetime"` // 更新时间
}
func (m *ChartCollect) GetItemsByUserIdAndChartInfoIdList(userId int, chartInfoIdList []int) (items []*ChartCollectItem, err error) {
if len(chartInfoIdList) <= 0 {
return
}
sql := `SELECT a.*,b.classify_name FROM chart_collect AS a
JOIN chart_collect_classify b on a.chart_collect_classify_id = b.chart_collect_classify_id
WHERE a.sys_user_id = ? AND a.chart_info_id in (?) ORDER BY a.chart_collect_id DESC `
err = global.DmSQL["data"].Raw(sql, userId, chartInfoIdList).Find(&items).Error
return
}