From dfb682c027ce19d893693636d814a0be4a023eea Mon Sep 17 00:00:00 2001
From: "zhangkun9038@dingtalk.com" <zhangkun9038@dingtalk.com>
Date: Wed, 12 Feb 2025 03:56:15 +0800
Subject: [PATCH] plate

---
 plate.go | 82 +++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 72 insertions(+), 10 deletions(-)

diff --git a/plate.go b/plate.go
index f280b88..db146ad 100644
--- a/plate.go
+++ b/plate.go
@@ -2,6 +2,7 @@ package core
 
 import (
 	"encoding/json"
+	"fmt"
 	"time"
 )
 
@@ -20,23 +21,84 @@ func (pl *Plate) Init(instId string) {
 }
 
 // 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"
-	_, err := cr.RedisLocalCli.Exists().Result()
-	if err == nil {
-		str, _ := cr.RedisLocalCli.Get(plateName).Result()
-		json.Unmarshal([]byte(str), &pl)
-	} else {
-		pl.Init(instId)
-		prs := cr.Cfg.Config.Get("candleDimentions").MustArray()
-		for _, v := range prs {
-			pl.MakeCoaster(cr, v.(string))
+
+	// 尝试从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"