package stl

import (
	"time"

	"eta/eta_api/global"
	"eta/eta_api/utils"
	"github.com/beego/beego/v2/client/orm"
)

type CalculateStlConfigMapping struct {
	Id                   int       `orm:"pk" description:"主键"`
	CalculateStlConfigId int       `description:"stl配置id"`
	EdbInfoId            int       `description:"edb信息id"`
	StlEdbType           int       `description:"stl指标类型: 1-Trend, 2-Seasonal, 3-Residual"`
	CreateTime           time.Time `description:"创建时间"`
	ModifyTime           time.Time `description:"修改时间"`
}

func (c *CalculateStlConfigMapping) Insert() (insertId int64, err error) {
	o := orm.NewOrmUsingDB("data")
	insertId, err = o.Insert(c)
	err = global.DbMap[utils.DbNameIndex].Create(c).Error
	return
}

// GetCalculateStlConfigMappingIdByEdbInfoId 获取配置文件id
func GetCalculateStlConfigMappingIdByEdbInfoId(edbInfoId int) (configId int, err error) {
	sql := `SELECT calculate_stl_config_id FROM calculate_stl_config_mapping WHERE edb_info_id=? LIMIT 1`
	err = global.DbMap[utils.DbNameIndex].Raw(sql, edbInfoId).Scan(&configId).Error
	return
}

// GetCalculateStlConfigMappingByConfigId 根据配置文件id获取配置文件映射信息
func GetCalculateStlConfigMappingByConfigId(configId int) (items []*CalculateStlConfigMapping, err error) {
	sql := `SELECT * FROM calculate_stl_config_mapping WHERE calculate_stl_config_id=?`
	err = global.DbMap[utils.DbNameIndex].Raw(sql).Find(&items).Error
	return
}