第八版

chanlun/src/structure/fractal_obj.rs

@@ -25,14 +25,15 @@
 use crate::kline::chan_kline::缠论K线;
 use crate::types::分型结构;
 use crate::types::相对方向;
-use std::rc::Rc;
+use std::sync::atomic::Ordering;
+use std::sync::Arc;
 
 /// 分型 — 由三根缠K构成（可能缺左或右）
 #[derive(Debug, Clone)]
 pub struct 分型 {
-    pub 左: Option<Rc<缠论K线>>,
-    pub 中: Rc<缠论K线>,
-    pub 右: Option<Rc<缠论K线>>,
+    pub 左: Option<Arc<缠论K线>>,
+    pub 中: Arc<缠论K线>,
+    pub 右: Option<Arc<缠论K线>>,
     pub 结构: 分型结构,
     pub 时间戳: i64,
     pub 分型特征值: f64,
@@ -40,11 +41,11 @@ pub struct 分型 {
 
 impl 分型 {
     pub fn new(
-        左: Option<Rc<缠论K线>>, 中: Rc<缠论K线>, 右: Option<Rc<缠论K线>>
+        左: Option<Arc<缠论K线>>, 中: Arc<缠论K线>, 右: Option<Arc<缠论K线>>
     ) -> Self {
-        let 结构 = 中.分型.unwrap_or(分型结构::散);
-        let 时间戳 = 中.时间戳;
-        let 分型特征值 = 中.分型特征值;
+        let 结构 = 中.分型.read().unwrap().unwrap_or(分型结构::散);
+        let 时间戳 = 中.时间戳.load(Ordering::Relaxed);
+        let 分型特征值 = 中.分型特征值.get();
         Self {
             左,
             中,
@@ -60,9 +61,9 @@ impl 分型 {
         let 左 = self.左.as_ref()?;
         let 右 = self.右.as_ref()?;
         Some((
-            相对方向::分析(左.高, 左.低, self.中.高, self.中.低),
-            相对方向::分析(self.中.高, self.中.低, 右.高, 右.低),
-            相对方向::分析(左.高, 左.低, 右.高, 右.低),
+            相对方向::分析(左.高.get(), 左.低.get(), self.中.高.get(), self.中.低.get()),
+            相对方向::分析(self.中.高.get(), self.中.低.get(), 右.高.get(), 右.低.get()),
+            相对方向::分析(左.高.get(), 左.低.get(), 右.高.get(), 右.低.get()),
         ))
     }
 
@@ -97,17 +98,19 @@ impl 分型 {
 
         if let (Some(ref 左), Some(ref 右)) = (&self.左, &self.右) {
             if self.结构 == 分型结构::底 {
-                if 右.标的K线.收盘价 > 左.标的K线.高 {
+                if 右.标的K线.read().unwrap().收盘价 > 左.标的K线.read().unwrap().高 {
                     return "强";
-                } else if 右.标的K线.收盘价 > self.中.标的K线.高 {
+                } else if 右.标的K线.read().unwrap().收盘价 > self.中.标的K线.read().unwrap().高
+                {
                     return "中";
                 } else {
                     return "弱";
                 }
             } else if self.结构 == 分型结构::顶 {
-                if 右.标的K线.收盘价 < 左.标的K线.低 {
+                if 右.标的K线.read().unwrap().收盘价 < 左.标的K线.read().unwrap().低 {
                     return "强";
-                } else if 右.标的K线.收盘价 < self.中.标的K线.低 {
+                } else if 右.标的K线.read().unwrap().收盘价 < self.中.标的K线.read().unwrap().低
+                {
                     return "中";
                 } else {
                     return "弱";
@@ -121,15 +124,21 @@ impl 分型 {
     pub fn 与MACD柱子分型匹配(&self) -> bool {
         if let (Some(ref 左), Some(ref 右)) = (&self.左, &self.右) {
             if self.结构 == 分型结构::底 {
+                let 左_k = 左.标的K线.read().unwrap();
+                let 中_k = self.中.标的K线.read().unwrap();
+                let 右_k = 右.标的K线.read().unwrap();
                 if let (Some(ref 左macd), Some(ref 中macd), Some(ref 右macd)) =
-                    (&左.标的K线.macd, &self.中.标的K线.macd, &右.标的K线.macd)
+                    (&左_k.macd, &中_k.macd, &右_k.macd)
                 {
                     return 左macd.MACD柱 > 中macd.MACD柱 && 中macd.MACD柱 < 右macd.MACD柱;
                 }
             }
             if self.结构 == 分型结构::顶 {
+                let 左_k = 左.标的K线.read().unwrap();
+                let 中_k = self.中.标的K线.read().unwrap();
+                let 右_k = 右.标的K线.read().unwrap();
                 if let (Some(ref 左macd), Some(ref 中macd), Some(ref 右macd)) =
-                    (&左.标的K线.macd, &self.中.标的K线.macd, &右.标的K线.macd)
+                    (&左_k.macd, &中_k.macd, &右_k.macd)
                 {
                     return 左macd.MACD柱 < 中macd.MACD柱 && 中macd.MACD柱 > 右macd.MACD柱;
                 }
@@ -139,35 +148,36 @@ impl 分型 {
     }
 
     /// 判断两个分型是否匹配
-    pub fn 判断分型(左: &Rc<分型>, 右: &Rc<分型>, 模式: &str) -> bool {
+    pub fn 判断分型(左: &Arc<分型>, 右: &Arc<分型>, 模式: &str) -> bool {
         match 模式 {
-            "中" => Rc::as_ptr(左) == Rc::as_ptr(右),
+            "中" => Arc::as_ptr(左) == Arc::as_ptr(右),
             _ => false,
         }
     }
 
     /// 从缠K序列中获取以指定缠K为中元素的分型
     pub fn 从缠K序列中获取分型(
-        K线序列: &[Rc<缠论K线>], 中: &Rc<缠论K线>
+        K线序列: &[Arc<缠论K线>],
+        中: &Arc<缠论K线>,
     ) -> Option<Self> {
         let idx = K线序列
             .iter()
-            .position(|k| Rc::as_ptr(k) == Rc::as_ptr(中))?;
+            .position(|k| Arc::as_ptr(k) == Arc::as_ptr(中))?;
         let 左 = if idx > 0 {
-            Some(Rc::clone(&K线序列[idx - 1]))
+            Some(Arc::clone(&K线序列[idx - 1]))
         } else {
             None
         };
         let 右 = if idx + 1 < K线序列.len() {
-            Some(Rc::clone(&K线序列[idx + 1]))
+            Some(Arc::clone(&K线序列[idx + 1]))
         } else {
             None
         };
-        Some(Self::new(左, Rc::clone(中), 右))
+        Some(Self::new(左, Arc::clone(中), 右))
     }
 
     /// 向分型序列中添加新分型
-    pub fn 向序列中添加(分型序列: &mut Vec<Rc<分型>>, 当前分型: Rc<分型>) {
+    pub fn 向序列中添加(分型序列: &mut Vec<Arc<分型>>, 当前分型: Arc<分型>) {
         if 分型序列.is_empty() {
             if 当前分型.结构 != 分型结构::顶 && 当前分型.结构 != 分型结构::底
             {
@@ -188,10 +198,10 @@ impl 分型 {
 
 impl crate::types::fractal::有高低 for 分型 {
     fn 高(&self) -> f64 {
-        self.中.高
+        self.中.高.get()
     }
     fn 低(&self) -> f64 {
-        self.中.低
+        self.中.低.get()
     }
 }
 
@@ -200,7 +210,11 @@ impl std::fmt::Display for 分型 {
         write!(
             f,
             "{}<{}, {}, None: {}, None: {}>",
-            self.中.分型.unwrap_or(crate::types::分型结构::散),
+            self.中
+                .分型
+                .read()
+                .unwrap()
+                .unwrap_or(crate::types::分型结构::散),
             self.时间戳,
             crate::utils::format_f64_g(self.分型特征值),
             if self.左.is_none() { "True" } else { "False" },