package request

import (
	"database/sql"
)

// HandleTaskRecordFailReq
// @Description: 任务处理失败
type HandleTaskRecordFailReq struct {
	IndexTaskRecordId int    `description:"子任务id"`
	FailMsg           string `description:"失败原因"`
}

// HandleTaskRecordSuccessReq
// @Description: 任务处理成功
type HandleTaskRecordSuccessReq struct {
	IndexTaskRecordId int                      `gorm:"primaryKey;column:ai_predict_model_index_config_id" description:"子任务记录id"`
	Data              AiPredictModelImportData `description:"导入的指标"`
}

// AiPredictModelIndex AI预测模型标的
type AiPredictModelIndex struct {
	AiPredictModelIndexId       int     `orm:"column(ai_predict_model_index_id);pk" gorm:"primaryKey"`
	IndexName                   string  `description:"标的名称"`
	IndexCode                   string  `description:"自生成的指标编码"`
	ClassifyId                  int     `description:"分类ID"`
	ModelFramework              string  `description:"模型框架"`
	PredictDate                 string  `description:"预测日期"`
	PredictValue                float64 `description:"预测值"`
	PredictFrequency            string  `description:"预测频度"`
	DirectionAccuracy           string  `description:"方向准确度"`
	AbsoluteDeviation           string  `description:"绝对偏差"`
	ExtraConfig                 string  `description:"模型参数"`
	Sort                        int     `description:"排序"`
	SysUserId                   int     `description:"创建人ID"`
	SysUserRealName             string  `description:"创建人姓名"`
	LeftMin                     string  `description:"图表左侧最小值"`
	LeftMax                     string  `description:"图表左侧最大值"`
	AiPredictModelIndexConfigId int     `gorm:"column:ai_predict_model_index_config_id" description:"标的当前的配置id"`
	ScriptPath                  string  `gorm:"column:script_path" description:"脚本的路径"`
	TrainStatus                 string  `gorm:"column:train_status" description:"训练状态，枚举值：待训练，训练中，训练成功，训练失败"`
	RunStatus                   string  `gorm:"column:run_status" description:"运行状态，枚举值：待运行，运行中，运行成功，运行失败"`
}

// AiPredictModelData AI预测模型标的数据
type AiPredictModelData struct {
	DataTime      string          `description:"数据日期"`
	Value         sql.NullFloat64 `description:"实际值"`
	PredictValue  sql.NullFloat64 `description:"预测值"`
	Direction     string          `description:"方向"`
	DeviationRate string          `description:"偏差率"`
	DataTimestamp int64           `description:"数据日期时间戳"`
	Source        int             `description:"来源：1-月度预测（默认）；2-日度预测"`
}

type AiPredictModelImportData struct {
	Index     *AiPredictModelIndex
	Data      []*AiPredictModelData
	TrainData TrainData
}

// TrainData
// @Description: 训练结果
type TrainData struct {
	TrainMse float64
	TrainR2  float64
	TestMse  float64
	TestR2   float64
}

type AiPredictModelIndexExtraConfig struct {
	MonthlyChart MonthlyChartConfig
	DailyChart   DailyChartConfig
}
type MonthlyChartConfig struct {
	LeftMin string `description:"图表左侧最小值"`
	LeftMax string `description:"图表左侧最大值"`
	Unit    string `description:"单位"`
}

type DailyChartConfig struct {
	LeftMin           string `description:"图表左侧最小值"`
	LeftMax           string `description:"图表左侧最大值"`
	Unit              string `description:"单位"`
	PredictLegendName string `description:"预测图例的名称（通常为Predicted）"`
}