package data_manage

import (
	"database/sql"
	"fmt"
	"github.com/beego/beego/v2/client/orm"
	"strings"
	"time"
)

const (
	ModelDataSourceMonthly = 1 // 月度预测数据
	ModelDataSourceDaily   = 2 // 日度预测数据
)

// AiPredictModelData AI预测模型标的数据
type AiPredictModelData struct {
	AiPredictModelDataId  int             `orm:"column(ai_predict_model_data_id);pk" gorm:"primaryKey"`
	AiPredictModelIndexId int             `description:"标的ID"`
	IndexCode             string          `description:"标的编码"`
	DataTime              time.Time       `description:"数据日期"`
	Value                 sql.NullFloat64 `description:"实际值"`
	PredictValue          sql.NullFloat64 `description:"预测值"`
	Direction             string          `description:"方向"`
	DeviationRate         string          `description:"偏差率"`
	CreateTime            time.Time       `description:"创建时间"`
	ModifyTime            time.Time       `description:"修改时间"`
	DataTimestamp         int64           `description:"数据日期时间戳"`
	Source                int             `description:"来源：1-月度预测（默认）；2-日度预测"`
}

func (m *AiPredictModelData) TableName() string {
	return "ai_predict_model_data"
}

type AiPredictModelDataCols struct {
	PrimaryId             string
	AiPredictModelIndexId string
	IndexCode             string
	DataTime              string
	Value                 string
	PredictValue          string
	Direction             string
	DeviationRate         string
	CreateTime            string
	ModifyTime            string
	DataTimestamp         string
	Source                string
}

func (m *AiPredictModelData) Cols() AiPredictModelDataCols {
	return AiPredictModelDataCols{
		PrimaryId:             "ai_predict_model_data_id",
		AiPredictModelIndexId: "ai_predict_model_index_id",
		IndexCode:             "index_code",
		DataTime:              "data_time",
		Value:                 "value",
		PredictValue:          "predict_value",
		Direction:             "direction",
		DeviationRate:         "deviation_rate",
		CreateTime:            "create_time",
		ModifyTime:            "modify_time",
		DataTimestamp:         "data_timestamp",
		Source:                "source",
	}
}

func (m *AiPredictModelData) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*AiPredictModelData, err error) {
	o := orm.NewOrmUsingDB("data")
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := fmt.Sprintf(`ORDER BY %s DESC`, m.Cols().CreateTime)
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sqlRun := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	_, err = o.Raw(sqlRun, pars...).QueryRows(&items)
	return
}