eta_api/models/data_manage/base_from_hisugar.go

package data_manage

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

	"github.com/rdlucklib/rdluck_tools/paging"
	"gorm.io/gorm"
)

type BaseFromHisugarClassify struct {
	BaseFromHisugarClassifyId int    // 分类ID
	ClassifyName              string // 分类名称
	ParentID                  int    // 上级ID
	Level                     int    // 层级
	Sort                      int    // 排序
	CreateTime                string // 创建时间
	ModifyTime                string // 修改时间
}

type BaseFromHisugarIndex struct {
	BaseFromHisugarIndexId int    // 主键ID
	IndexCode              string // 指标编码
	IndexName              string // 指标名称
	ClassifyId             uint   // 分类ID
	Unit                   string // 单位
	Frequency              string // 频度
	Describe               string // 指标描述
	Sort                   int    // 排序
	CreateTime             string // 创建时间
	ModifyTime             string // 修改时间
}

func (b *BaseFromHisugarIndex) ConvertTimeStr() {
	b.CreateTime = utils.GormDateStrToDateTimeStr(b.CreateTime)
	b.ModifyTime = utils.GormDateStrToDateTimeStr(b.ModifyTime)
	return
}

type BaseFromHisugarData struct {
	BaseFromHisugarDataId  int    // 数据表ID
	BaseFromHisugarIndexId int    // 指标ID
	IndexCode              string // 指标编码
	DataTime               string
	Value                  string
	CreateTime             string
	ModifyTime             string
}

func (b *BaseFromHisugarData) AfterFind(db *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		b.CreateTime = utils.GormDateStrToDateTimeStr(b.CreateTime)
		b.ModifyTime = utils.GormDateStrToDateTimeStr(b.ModifyTime)
		b.DataTime = utils.GormDateStrToDateStr(b.DataTime)
	}
	return
}

func GetHisugarClassifyList() (list []*BaseFromHisugarClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := "SELECT * FROM base_from_hisugar_classify ORDER BY sort ASC"
	err = o.Raw(sql).Find(&list).Error
	return
}

type BaseFromHisugarIndexList struct {
	BaseFromHisugarIndexId int                    // 主键ID
	IndexCode              string                 // 指标编码
	IndexName              string                 // 指标名称
	ClassifyId             int                    // 分类ID
	Unit                   string                 // 单位
	Frequency              string                 // 频度
	Describe               string                 // 指标描述
	Sort                   int                    // 排序
	CreateTime             string                 // 创建时间
	ModifyTime             string                 // 修改时间
	EdbExist               int                    `description:"edb是否存在"`
	DataList               []*BaseFromHisugarData `gorm:"-"`
	Paging                 *paging.PagingItem     `description:"分页数据" gorm:"-"`
	EdbInfoId              int                    `description:"指标库主键id"`
}

func (b *BaseFromHisugarIndexList) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		b.ModifyTime = utils.GormDateStrToDateTimeStr(b.ModifyTime)
		b.CreateTime = utils.GormDateStrToDateTimeStr(b.CreateTime)
	}
	return
}


type BaseFromHisugarIndexListResp struct {
	List   []*BaseFromHisugarIndexView
	Paging *paging.PagingItem `description:"分页数据"`
}

func GetHisugarIndexById(indexId int) (item *BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1 base_from_hisugar_index_id = ? `
	sql += `ORDER BY base_from_hisugar_index_id ASC `
	err = o.Raw(sql, indexId).First(&item).Error
	if err != nil {
		return
	}
	item.ConvertTimeStr()
	return
}
func GetHisugarIndexByCode(indexCode string) (item *BaseFromHisugarIndexView, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT a.*,CASE WHEN e.edb_info_id IS NULL THEN 0 ELSE 1 END AS edb_exist,e.edb_info_id
	FROM base_from_hisugar_index as a 
LEFT JOIN edb_info AS e ON a.index_code=e.edb_code AND e.source=93 
	WHERE 1=1 and a.index_code = ? `
	sql += `ORDER BY a.base_from_hisugar_index_id ASC `
	err = o.Raw(sql, indexCode).First(&item).Error
	if err != nil {
		return
	}
	item.ConvertTimeStr()
	return
}

func GetHisugarIndexList(condition string, pars interface{}, startSize, pageSize int) (items []*BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}

	sql += `group BY index_code ASC order by create_time DESC LIMIT ?,? `
	err = o.Raw(sql).Find(&items).Error
	return
}

func GetHisugarIndexListCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count FROM base_from_hisugar_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

func GetHisugarDataListCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count FROM base_from_hisugar_data WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

func GetHisugarIndexData(condition string, pars []interface{}, startSize, pageSize int) (items []*BaseFromHisugarData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_data WHERE 1=1  `
	if condition != "" {
		sql += condition
	}

	sql += ` order by data_time DESC LIMIT ?,? `
	pars = append(pars, startSize, pageSize)
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// GetHisugarItemList 模糊查询泛糖科技数据库指标列表
func GetHisugarItemList(keyword string) (items []*BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := "SELECT * FROM base_from_hisugar_index WHERE CONCAT(index_name,index_code) LIKE ? "
	err = o.Raw(sql, utils.GetLikeKeyword(keyword)).Find(&items).Error
	return
}

// HisugarDataBatchListReq 泛糖科技指标批量列表
type HisugarDataBatchListReq struct {
	ClassifyId  int    `description:"分类id"`
	KeyWord     string `description:"关键字"`
	SelectedId  []int  `description:"已选指标id, 为true时表示反选"`
	IsSelectAll bool   `description:"是否查询全部, 默认false, true:全选, false:查询已选"`
}

// GetHisugarIndexByCondition 根据条件获取泛糖科技指标列表
func GetHisugarIndexByCondition(condition string, pars []interface{}) (items []*BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_hisugar_index_id ASC`
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// HisugarDataBatchAddCheckReq 泛糖科技指标批量添加校验
type HisugarDataBatchAddCheckReq struct {
	IndexCodes []string `description:"指标编码"`
}

// GetHisugarIndexAndEdbInfoByCondition 根据条件获取泛糖科技index和指标库的信息
func GetHisugarIndexAndEdbInfoByCondition(condition string, pars []interface{}) (items []*BaseFromHisugarIndexView, err error) {
	wherePars := make([]interface{}, 0)
	wherePars = append(wherePars, utils.DATA_SOURCE_HISUGAR)
	wherePars = append(wherePars, pars...)

	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT b.*, e.edb_info_id, e.unique_code, e.classify_id AS edb_classify_id FROM base_from_hisugar_index AS b LEFT JOIN edb_info AS e ON b.index_code=e.edb_code AND e.source=? WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC `

	err = o.Raw(sql, wherePars...).Find(&items).Error
	return
}

type BaseFromHisugarIndexView struct {
	BaseFromHisugarIndexId int    `orm:"pk" gorm:"primaryKey"`
	EdbInfoId              int    `description:"指标库id"`
	ClassifyId             int    `description:"指标分类id"`
	IndexCode              string `description:"指标编码"`
	IndexName              string `description:"指标名称"`
	UniqueCode             string `description:"唯一code"`
	Frequency              string `description:"频度"`
	Unit                   string `description:"单位"`
	StartDate              string `description:"开始日期"`
	EndDate                string `description:"结束日期"`
	Sort                   int    `description:"排序"`
	EdbExist               int    `description:"edb是否存在"`
	EdbClassifyId          int    `description:"edb分类id"`
	ModifyTime             string
	DataTime               string `description:"数据时间"`
	Value                  string `description:"值"`
}

func (b *BaseFromHisugarIndexView) AfterFind(db *gorm.DB) (err error) {
	b.ModifyTime = utils.GormDateStrToDateTimeStr(b.ModifyTime)
	b.DataTime = utils.GormDateStrToDateStr(b.DataTime)
	b.StartDate = utils.GormDateStrToDateStr(b.StartDate)
	b.EndDate = utils.GormDateStrToDateStr(b.EndDate)
	return
}

func (b *BaseFromHisugarIndexView) ConvertTimeStr() {
	b.ModifyTime = utils.GormDateStrToDateTimeStr(b.ModifyTime)
	b.DataTime = utils.GormDateStrToDateStr(b.DataTime)
	b.StartDate = utils.GormDateStrToDateStr(b.StartDate)
	b.EndDate = utils.GormDateStrToDateStr(b.EndDate)
	return
}

// ExportHisugarExcelReq 导出泛糖科技excel指标
type ExportHisugarExcelReq struct {
	KeyWord       string   `description:"关键字, 指标编码或指标ID"`
	IndexCode     []string `description:"指标编码，全选时，表示反选"`
	IsSelectedAll bool     `description:"是否全选：true：全选|false: 无"`
	ClassifyId    int      `description:"指标id"`
}

func (baseFromHisugarIndex *BaseFromHisugarIndex) AfterFind(db *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if baseFromHisugarIndex.ModifyTime != "" {
			baseFromHisugarIndex.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromHisugarIndex.ModifyTime)
		}
		if baseFromHisugarIndex.CreateTime != "" {
			baseFromHisugarIndex.CreateTime = utils.GormDateStrToDateTimeStr(baseFromHisugarIndex.CreateTime)
		}
	}
	return
}

// GetHisugarIndexByConditionAndFrequency 根据条件获取泛糖科技指标列表
func GetHisugarIndexByConditionAndFrequency(condition, frequency string, pars []interface{}) (items []*BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1 `

	if condition != "" {
		sql += condition
	}
	sql += ` AND frequency=?`
	sql += ` ORDER BY sort ASC, base_from_hisugar_index_id ASC`
	pars = append(pars, frequency)
	//err = o.Raw(sql, pars, frequency).Find(&items).Error
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

func GetHisugarDataMaxCount(classifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := `SELECT COALESCE(MAX(t.num), 0) AS count FROM (
				SELECT COUNT(1) AS num  FROM base_from_hisugar_index AS a
				INNER JOIN base_from_hisugar_data AS b ON a.index_code=b.index_code
				WHERE a.classify_id=?
				GROUP BY a.base_from_hisugar_index_id
			)AS t `
	err = o.Raw(sql, classifyId).Scan(&count).Error
	return
}

func GetHisugarIndexDataByCode(indexCode string) (items []*BaseFromHisugarData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT *  FROM base_from_hisugar_data WHERE index_code=? ORDER BY data_time DESC  `
	err = o.Raw(sql, indexCode).Find(&items).Error
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		for _, item := range items {
			item.DataTime = utils.GormDateStrToDateStr(item.DataTime)
		}
	}
	return
}

func GetHisugarFrequencyByCondition(condition string, pars []interface{}) (items []*string, err error) {
	sql := `SELECT DISTINCT frequency FROM base_from_hisugar_index WHERE 1=1 `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY FIELD(frequency,'日度','周度','月度','季度','半年','年度') `
	o := global.DbMap[utils.DbNameIndex]
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// GetHisugarIndexViewList 根据分类id获取泛糖科技指标列表
func GetHisugarIndexViewList(condition string, pars []interface{}) (items []*BaseFromHisugarIndexView, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT b.*, e.edb_info_id,
	CASE WHEN e.edb_info_id IS NULL THEN 0 ELSE 1 END AS edb_exist
	FROM base_from_hisugar_index AS b
	LEFT JOIN edb_info AS e ON b.index_code=e.edb_code AND e.source=93
	WHERE 1=1   `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY b.modify_time ASC `
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// GetHisugarIndexViewListCount 根据分类id获取泛糖科技指标列表
func GetHisugarIndexViewListCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count
	FROM base_from_hisugar_index AS b
	LEFT JOIN edb_info AS e ON b.index_code=e.edb_code AND e.source=93
	WHERE 1=1   `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY b.modify_time ASC `
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

// GetHisugarDataViewList 根据指标id获取泛糖科技指标列表
func GetHisugarDataViewList(indexIds []int) (items []*BaseFromHisugarData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_data WHERE base_from_hisugar_index_id IN (` + utils.GetOrmInReplace(len(indexIds)) + `) ORDER BY data_time desc  `
	err = o.Raw(sql, indexIds).Find(&items).Error
	return
}

// GetHisugarDataDataTimeByIndexId 根据指标id获取指标数据的日期列表
func GetHisugarDataDataTimeByIndexId(indexIdList []int) (items []string, err error) {
	if len(indexIdList) == 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	//sql := ` SELECT DISTINCT data_time FROM base_from_hisugar_data WHERE base_from_hisugar_index_id IN (` + utils.GetOrmInReplace(len(indexIdList)) + `) ORDER BY data_time DESC`
	sql := ` SELECT DISTINCT data_time FROM base_from_hisugar_data WHERE base_from_hisugar_index_id IN ? ORDER BY data_time DESC`
	err = o.Raw(sql, indexIdList).Find(&items).Error
	if err != nil {
		return
	}
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		for i, v := range items {
			items[i] = utils.GormDateStrToDateStr(v)
		}
	}
	return
}

type BaseFromHisugarClassifyItem struct {
	BaseFromHisugarClassifyId int                            `orm:"column(base_from_hisugar_classify_id);pk" gorm:"primaryKey"`
	ClassifyName              string                         `description:"分类名称"`
	ParentId                  int                            `description:"父级id"`
	Level                     int                            `description:"层级"`
	Sort                      int                            `description:"排序字段"`
	UniqueCode                string                         `description:"唯一code"`
	ModifyTime                time.Time                      `description:"修改时间"`
	CreateTime                time.Time                      `description:"创建时间"`
	ClassifyNameEn            string                         `description:"英文分类名称"`
	Children                  []*BaseFromHisugarClassifyItem `description:"子分类" gorm:"-"`
}

// 获取所有分类
func GetHisugarClassifyAll() (items []*BaseFromHisugarClassifyItem, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_classify ORDER BY sort ASC, base_from_hisugar_classify_id ASC`
	err = o.Raw(sql).Find(&items).Error
	return
}

func GetHisugarIndex(condition string, pars []interface{}) (items []*BaseFromHisugarIndexList, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY sort ASC, base_from_hisugar_index_id asc`
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

type HisugarIndexDataCountGroup struct {
	IndexCode string
	Count     int
}

func GetHisugarIndexDataCountGroup(indexCodes []string) (items []*HisugarIndexDataCountGroup, err error) {
	if len(indexCodes) <= 0 {
		return
	}
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count, index_code FROM base_from_hisugar_data WHERE index_code IN (` + utils.GetOrmInReplace(len(indexCodes)) + `) GROUP BY index_code`
	err = o.Raw(sql, indexCodes).Find(&items).Error
	return
}

func GetHisugarIndexDataV2(indexCode string, startSize, pageSize int) (items []*BaseFromHisugarData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT *  FROM base_from_hisugar_data WHERE index_code=? ORDER BY data_time DESC LIMIT ?,? `
	err = o.Raw(sql, indexCode, startSize, pageSize).Find(&items).Error
	return
}

// GetHisugarIndexInfoCount 查询指标信息总数
func GetHisugarIndexInfoCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count FROM base_from_hisugar_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

type BaseFromHisugarIndexPage struct {
	List   []*BaseFromHisugarIndex `description:"指标列表"`
	Paging *paging.PagingItem      `description:"分页数据"`
}

// GetHisugarIndexInfoPage 分页查询指标信息
func GetHisugarIndexInfoPage(condition string, pars []interface{}, size int, pageSize int) (items []*BaseFromHisugarIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_hisugar_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += ` ORDER BY base_from_hisugar_index_id asc LIMIT ?,?`
	pars = append(pars, size, pageSize)
	//err = o.Raw(sql, pars, size, pageSize).Find(&items).Error
	err = o.Raw(sql, pars...).Find(&items).Error
	return

}

// 获取所有分类
func GetHisugarClassifyById(classifyId int) (ClassifyIds string, err error) {
	o := global.DbMap[utils.DbNameIndex]
	//sql := fmt.Sprintf("SELECT %s AS classify_ids FROM base_from_hisugar_classify WHERE base_from_hisugar_classify_id=? OR parent_id=?", utils.GroupUnitFunc(utils.DbDriverName, "base_from_hisugar_classify_id", ",", false, "", "", false))

	sql := fmt.Sprintf("SELECT %s AS classify_ids FROM base_from_hisugar_classify WHERE base_from_hisugar_classify_id=? OR parent_id=?",
		utils.GenerateQuerySql(utils.GroupUnit, &utils.QueryParam{Column: "base_from_hisugar_classify_id",
			HasDistinct: false,
		}))
	//sql := ` SELECT GROUP_CONCAT(base_from_hisugar_classify_id) AS classify_ids FROM base_from_hisugar_classify WHERE base_from_hisugar_classify_id=? OR parent_id=?`
	//sql := ` SELECT AS classify_ids FROM base_from_hisugar_classify WHERE base_from_hisugar_classify_id=? OR parent_id=?`
	err = o.Raw(sql, classifyId, classifyId).Scan(&ClassifyIds).Error
	return
}