eta_api/models/data_manage/base_from_rzd_index.go

// Package data_manage
// @Author gmy 2024/8/7 9:38:00
package data_manage

import (
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"fmt"
	"gorm.io/gorm"

	"github.com/beego/beego/v2/client/orm"
	"github.com/rdlucklib/rdluck_tools/paging"
)

type BaseFromRzdIndex struct {
	BaseFromRzdIndexId    int    `orm:"column(base_from_rzd_index_id);pk" gorm:"primaryKey"`
	CreateTime            string `orm:"column(create_time)"`
	ModifyTime            string `orm:"column(modify_time)"`
	BaseFromRzdClassifyId int    `orm:"column(base_from_rzd_classify_id)"`
	IndexCode             string `orm:"column(index_code)"`
	IndexName             string `orm:"column(index_name)"`
	Frequency             string `orm:"column(frequency)"`
	Unit                  string `orm:"column(unit)"`
}

type BaseFromRzdIndexAndData struct {
	BaseFromRzdIndexId    int     `orm:"column(base_from_rzd_index_id);pk" gorm:"primaryKey"`
	CreateTime            string  `orm:"column(create_time)"`
	ModifyTime            string  `orm:"column(modify_time)"`
	BaseFromRzdClassifyId int     `orm:"column(base_from_rzd_classify_id)"`
	IndexCode             string  `orm:"column(index_code)"`
	IndexName             string  `orm:"column(index_name)"`
	Frequency             string  `orm:"column(frequency)"`
	Unit                  string  `orm:"column(unit)"`
	ModifyTimeMax         string  `json:"modify_time_max" description:"最后修改时间"`
	Value                 float64 `orm:"column(value)" description:"数据值"`
}

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

type BaseFromRzdIndexList struct {
	BaseFromRzdIndexId    int                `gorm:"column:base_from_rzd_index_id;primaryKey"`
	CreateTime            string             `orm:"column(create_time)"`
	ModifyTime            string             `orm:"column(modify_time)"`
	BaseFromRzdClassifyId int                `orm:"column(base_from_rzd_classify_id)"`
	IndexCode             string             `orm:"column(index_code)"`
	IndexName             string             `orm:"column(index_name)"`
	Frequency             string             `orm:"column(frequency)"`
	Unit                  string             `orm:"column(unit)"`
	DataList              []*BaseFromRzdData `gorm:"-"`
	Paging                *paging.PagingItem `description:"分页数据" gorm:"-"`
	EdbInfoId             int                `description:"指标库主键id"`
}

func (baseFromRzdIndexList *BaseFromRzdIndexList) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if baseFromRzdIndexList.ModifyTime != "" {
			baseFromRzdIndexList.ModifyTime = utils.GormDateStrToDateTimeStr(baseFromRzdIndexList.ModifyTime)
		}
		if baseFromRzdIndexList.CreateTime != "" {
			baseFromRzdIndexList.CreateTime = utils.GormDateStrToDateTimeStr(baseFromRzdIndexList.CreateTime)
		}

	}
	return
}

// RzdNameCheckResult 校验名称是否存在
type RzdNameCheckResult struct {
	IndexCode string `from:"EdbCode" description:"edb编码"`
	IndexName string `from:"EdbName" description:"edb名称"`
	Exist     bool
}

// RzdIndexCheckData 校验编码是否存在
type RzdIndexCheckData struct {
	IndexCode  string `orm:"column(index_code)" description:"指标编码"`
	IndexName  string `orm:"column(index_name)" description:"指标名称"`
	Frequency  string `orm:"column(frequency)" description:"频度"`
	Unit       string `orm:"column(unit)" description:"单位"`
	EdbInfoId  int    `json:"edb_info_id" description:"指标库主键id"`
	UniqueCode string `json:"unique_code" description:"指标库唯一编码"`
	ClassifyId int    `json:"classify_id" description:"分类id"`
}

// BaseFromRzdIndexBatchAddCheckReq 校验编码是否存在请求参数
type BaseFromRzdIndexBatchAddCheckReq struct {
	IndexCodes     []string `form:"IndexCodes" description:"指标编码列表"`
	ClassifyIdList []int    `description:"分类id"`
	FrequencyList  []string `description:"频度"`
	SearchParams   string   `description:"搜索参数 指标编码/指标名称"`
	IsCheckAll     bool     `description:"是否全选"`
}

func init() {
	orm.RegisterModel(new(BaseFromRzdIndex))
}

// GetRzdIndexByClassifyIds 根据分类id获取指标信息
func GetRzdIndexByClassifyIds(classifyIds []int) (items []*BaseFromRzdIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]

	sql := fmt.Sprintf(`SELECT * FROM base_from_rzd_index WHERE base_from_rzd_classify_id IN (`+utils.GetOrmInReplace(len(classifyIds))) + `)`
	err = o.Raw(sql, classifyIds).Find(&items).Error
	if err != nil {
		return nil, err
	}

	return items, nil
}

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

func GetRzdIndexNotExistEdbInfoCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT count(1) FROM base_from_rzd_index WHERE index_code not in (select edb_code from edb_info) `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

func GetRzdIndexNotExistEdbInfoPage(condition string, pars []interface{}) (items []*BaseFromRzdIndexAndData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_rzd_index WHERE index_code not in (select edb_code from edb_info) `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

// GetRzdIndexFrequency 获取指标频度
func GetRzdIndexFrequency(classifyIdList []int) (items []*string, err error) {
	sql := `SELECT DISTINCT frequency 
        FROM base_from_rzd_index 
        WHERE frequency is not null`

	// 如果 classifyId > 0，则添加该条件
	if len(classifyIdList) > 0 {
		sql += ` AND base_from_rzd_classify_id in (` + utils.GetOrmInReplace(len(classifyIdList)) + `)`
	}

	sql += ` ORDER BY FIELD(frequency, '日度', '周度', '月度', '季度', '半年度', '年度')`

	o := global.DbMap[utils.DbNameIndex]
	if len(classifyIdList) > 0 {
		err = o.Raw(sql, classifyIdList).Find(&items).Error
	} else {
		err = o.Raw(sql).Find(&items).Error
	}

	return items, err
}

// GetRzdIndexByCodeAndClassify 根据指标编码和分类查询 indexCode非必传
func GetRzdIndexByCodeAndClassify(indexCode string, classifyIdList []int, frequency *string) (items []*BaseFromRzdIndex, err error) {
	o := global.DbMap[utils.DbNameIndex]

	// SQL 查询语句
	sql := `SELECT a.index_code, a.index_name, a.frequency, a.unit, MAX(b.modify_time) AS modify_time
			FROM base_from_rzd_index AS a
			INNER JOIN base_from_rzd_data AS b ON a.base_from_rzd_index_id = b.base_from_rzd_index_id
			WHERE 1=1`

	var params []interface{}

	if len(classifyIdList) > 0 {
		sql += ` AND a.base_from_rzd_classify_id in (` + utils.GetOrmInReplace(len(classifyIdList)) + `)`
		for _, id := range classifyIdList {
			params = append(params, id)
		}
	}

	// 如果 indexCode 不为空，增加过滤条件
	if indexCode != "" {
		sql += ` AND a.index_code = ?`
		params = append(params, indexCode)
	}

	if frequency != nil {
		sql += ` AND a.frequency = ?`
		params = append(params, *frequency)
	}

	sql += ` GROUP BY a.index_code, a.index_name, a.frequency, a.unit`

	err = o.Raw(sql, params...).Find(&items).Error
	if err != nil {
		return nil, err
	}
	return
}

// GetRzdIndexInfoCount 分页查询指标信息行数
func GetRzdIndexInfoCount(condition string, pars []interface{}) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT count(1) FROM base_from_rzd_index WHERE index_code not in (select edb_code from edb_info) `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Scan(&count).Error
	return
}

// GetRzdIndexInfoPage 分页查询指标信息
func GetRzdIndexInfoPage(condition string, pars []interface{}) (items []*BaseFromRzdIndexAndData, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_rzd_index WHERE index_code not in (select edb_code from edb_info) `
	if condition != "" {
		sql += condition
	}
	err = o.Raw(sql, pars...).Find(&items).Error
	return

}