core/plate.go

package core

import (
	"encoding/json"
	"fmt"
	"time"
)

type Plate struct {
	InstID     string             `json:"instId,string"`
	Scale      float64            `json:"scale,number"`
	Count      int                `json:"count,number"`
	CoasterMap map[string]Coaster `json:"coasterMap"`
}

func (pl *Plate) Init(instId string) {
	pl.InstID = instId
	pl.Count = 24
	pl.Scale = float64(0.005)
	pl.CoasterMap = make(map[string]Coaster)
}

// TODO 从redis里读出来已经存储的plate，如果不存在就创建一个新的
// LoadPlate 加载或创建指定Instrument的Plate
// 1. 尝试从Redis加载已存储的Plate
// 2. 如果Redis中不存在，则初始化一个新的Plate
// 3. 根据配置创建所有需要的Coaster
// 参数:
//   - cr: 核心上下文对象
//   - instId: Instrument ID
//
// 返回值:
//   - *Plate: 加载或新建的Plate对象
//   - error: 操作过程中发生的错误
func LoadPlate(cr *Core, instId string) (*Plate, error) {
	// 初始化一个空的Plate对象
	pl := Plate{}

	// 构造Redis中存储Plate数据的key，格式为"instId|plate"
	plateName := instId + "|plate"

	// 尝试从Redis获取Plate数据
	str, err := cr.RedisLocalCli.Get(plateName).Result()

	// 如果Redis中存在数据且没有错误
	if err == nil && str != "" {
		// 将JSON字符串反序列化为Plate对象
		if err := json.Unmarshal([]byte(str), &pl); err != nil {
			// 反序列化失败时初始化一个新的Plate
			// logrus.Warnf("failed to unmarshal plate data from redis, init new plate: %v", err)
			pl.Init(instId)
		}
		// 返回从Redis加载的Plate对象
		return &pl, nil
	}

	// Redis不可用或数据不存在时，初始化一个新的Plate
	pl.Init(instId)

	// 从配置中获取candleDimentions配置项
	prs := cr.Cfg.Config.Get("candleDimentions")

	// 检查配置项是否存在
	if prs == nil || prs.Interface() == nil {
		return nil, fmt.Errorf("candleDimentions config not found")
	}

	// 将配置项转换为数组
	periods := prs.MustArray()

	// 遍历所有周期配置
	for _, v := range periods {
		// 将interface{}类型转换为string
		period, ok := v.(string)
		// 检查周期字符串是否有效
		if !ok || period == "" {
			continue // 跳过无效的周期配置
		}
		// 为每个周期创建Coaster
		if err := pl.MakeCoaster(cr, period); err != nil {
			// 如果创建失败，返回错误
			return nil, fmt.Errorf("failed to create coaster for period %s: %v", period, err)
		}
	}

	// 将新创建的Plate保存到Redis（可选操作）
	if err := savePlateToRedis(cr, plateName, &pl); err != nil {
		// 保存失败时记录警告日志
		// logrus.Warnf("failed to save plate to redis: %v", err)
	}

	// 返回创建好的Plate对象
	return &pl, nil
}
func savePlateToRedis(cr *Core, key string, pl *Plate) error {
	data, err := json.Marshal(pl)
	if err != nil {
		return err
	}
	return cr.RedisLocalCli.Set(key, data, 0).Err()
}
func (pl *Plate) SetToKey(cr *Core) error {
	js, _ := json.Marshal(*pl)
	plateName := pl.InstID + "|plate"
	_, err := cr.RedisLocalCli.Set(plateName, string(js), 0).Result()
	return err
}

func (pl *Plate) MakeCoaster(cr *Core, period string) error {
	lastTime := time.Now()
	setName := "candle" + period + "|" + pl.InstID + "|sortedSet"
	cdl, err := cr.GetRangeCandleSortedSet(setName, pl.Count, lastTime)
	if err != nil {
		return err
	}
	err = cdl.RecursiveBubbleS(len(cdl.List), "asc")
	setName7 := "ma7|" + setName
	setName30 := "ma30|" + setName
	mxl7, err := cr.GetRangeMaXSortedSet(setName7, pl.Count, lastTime)
	if err != nil {
		return err
	}
	err = mxl7.RecursiveBubbleS(len(mxl7.List), "asc")
	if err != nil {
		return err
	}
	mxl30, err := cr.GetRangeMaXSortedSet(setName30, pl.Count, lastTime)
	if err != nil {
		return err
	}
	err = mxl30.RecursiveBubbleS(len(mxl30.List), "asc")
	coaster := Coaster{
		InstID:     pl.InstID,
		Period:     period,
		Count:      pl.Count,
		Scale:      pl.Scale,
		CandleList: *cdl,
		Ma7List:    *mxl7,
		Ma30List:   *mxl30,
	}
	pl.CoasterMap["period"+period] = coaster
	return err
}