package data_source

import (
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"github.com/rdlucklib/rdluck_tools/paging"
	"gorm.io/gorm"
	"time"
)

type BaseFromIcpiIndex struct {
	BaseFromIcpiIndexId    int       `orm:"column(base_from_icpi_index_id);pk" gorm:"primaryKey"`
	BaseFromIcpiClassifyId int       `description:"分类id"`
	IndexCode              string    `description:"指标编码"`
	IndexName              string    `description:"指标名称"`
	Frequency              string    `description:"频度"`
	StartDate              time.Time `description:"开始日期"`
	EndDate                time.Time `description:"结束日期"`
	CreateTime             time.Time `description:"创建时间"`
	ModifyTime             time.Time `description:"修改时间"`
	Unit                   string    `description:"单位"`
	LatestValue            float64   `description:"最新值"`
}

type BaseFromIcpiData struct {
	BaseFromIcpiDataId  int       `orm:"column(base_from_icpi_data_id);pk" gorm:"primaryKey"`
	BaseFromIcpiIndexId int       `description:"指标id"`
	IndexCode           string    `description:"指标编码"`
	DataTime            string    `description:"日期"`
	Value               string    `description:"值"`
	CreateTime          time.Time `description:"创建时间"`
	ModifyTime          time.Time `description:"修改时间"`
}

type BaseFromIcpiClassify struct {
	BaseFromIcpiClassifyId int    `orm:"column(base_from_icpi_classify_id);pk" gorm:"primaryKey"`
	ClassifyName           string `description:"分类名称"`
	ClassifyNameEn         string `description:"英文名称"`
	ParentId               int    `description:"上级id"`
	CreateTime             string `description:"创建时间"`
	ModifyTime             string `description:"修改时间"`
}

func (obj *BaseFromIcpiIndex) GetBaseFromIcpiClassifyAll() (list []*BaseFromIcpiClassify, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := `SELECT * FROM base_from_icpi_classify `
	err = o.Raw(sql).Find(&list).Error
	return
}

type BaseFromIcpiDataView struct {
	BaseFromIcpiDataId  int       `orm:"column(base_from_icpi_data_id);pk" gorm:"primaryKey"`
	BaseFromIcpiIndexId int       `description:"指标id"`
	IndexCode           string    `description:"指标编码"`
	DataTime            string    `description:"日期"`
	Value               string    `description:"值"`
	CreateTime          time.Time `description:"创建时间"`
	ModifyTime          time.Time `description:"修改时间"`
}

func (obj *BaseFromIcpiDataView) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if obj.DataTime != "" {
			obj.DataTime = utils.GormDateStrToDateStr(obj.DataTime)
		}
	}
	return
}

type BaseFromIcpiIndexView struct {
	BaseFromIcpiIndexId    int                     `description:"指标id"`
	BaseFromIcpiClassifyId int                     `description:"分类id"`
	IndexCode              string                  `description:"指标编码"`
	IndexName              string                  `description:"指标名称"`
	Frequency              string                  `description:"频度"`
	Unit                   string                  `description:"单位"`
	StartDate              string                  `description:"开始日期"`
	EndDate                string                  `description:"结束日期"`
	CreateTime             string                  `description:"创建时间"`
	ModifyTime             string                  `description:"修改时间"`
	DataList               []*BaseFromIcpiDataView `gorm:"-"`
	Paging                 *paging.PagingItem      `description:"分页数据" gorm:"-"`
}

func (obj *BaseFromIcpiIndexView) AfterFind(tx *gorm.DB) (err error) {
	if utils.NeedDateOrTimeFormat(utils.DbDriverName) {
		if obj.CreateTime != "" {
			obj.CreateTime = utils.GormDateStrToDateTimeStr(obj.CreateTime)
		}
		if obj.ModifyTime != "" {
			obj.ModifyTime = utils.GormDateStrToDateTimeStr(obj.ModifyTime)
		}
		if obj.StartDate != "" {
			obj.StartDate = utils.GormDateStrToDateStr(obj.StartDate)
		}
		if obj.EndDate != "" {
			obj.EndDate = utils.GormDateStrToDateStr(obj.EndDate)
		}
	}
	return
}
func (obj *BaseFromIcpiIndex) GetIcpiIndex(condition string, pars []interface{}) (items []*BaseFromIcpiIndexView, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM base_from_icpi_index WHERE 1=1  `
	if condition != "" {
		sql += condition
	}
	sql += `ORDER BY base_from_icpi_index_id ASC `
	err = o.Raw(sql, pars...).Find(&items).Error
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiIndexDataCount(indexCode string) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT COUNT(1) AS count  FROM base_from_icpi_data WHERE index_code=? `
	err = o.Raw(sql, indexCode).Scan(&count).Error
	return
}

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

func (obj *BaseFromIcpiIndex) GetIcpiDataMaxCount(classifyId int) (count int, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := `SELECT MAX(t.num) AS count FROM (
				SELECT COUNT(1) AS num  FROM base_from_icpi_index AS a
				INNER JOIN base_from_icpi_data AS b ON a.base_from_icpi_index_id=b.base_from_icpi_index_id
				WHERE a.base_from_icpi_classify_id=?
				GROUP BY a.base_from_icpi_index_id
			)AS t `
	err = o.Raw(sql, classifyId).Scan(&count).Error
	return
}

func (obj *BaseFromIcpiIndex) GetIcpiIndexDataByCode(indexCode string) (items []*BaseFromIcpiDataView, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT *  FROM base_from_icpi_data WHERE index_code=? ORDER BY data_time DESC  `
	err = o.Raw(sql, indexCode).Find(&items).Error
	return
}