Sik/tradingbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify code

Browse Source
This commit is contained in:
Sik Yoon 2024-01-01 12:53:57 +09:00
parent 43b0423d30
commit ba4629f5e4
1 changed files with 103 additions and 106 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/value_estimation_team/indicators/rsi.rs
View File

@ -42,9 +42,8 @@ pub async fn rsi(
for element in filtered_symbols { for element in filtered_symbols {
let element_c = element.clone(); let element_c = element.clone();
let rsi_data_wrapper_arc_c = Arc::clone(&rsi_data_wrapper_arc); let rsi_data_wrapper_arc_c = Arc::clone(&rsi_data_wrapper_arc);
let symbol_search_result = input_rt_data.iter().position(|x| x.0 == *element_c.symbol); let search_result = input_rt_data.iter().position(|x| x.0 == *element_c.symbol);
match symbol_search_result { if search_result.is_some() {
Some(T) => {
let input_rt_data_c = input_rt_data.clone(); let input_rt_data_c = input_rt_data.clone();
task_vec.push(tokio::spawn(async move { task_vec.push(tokio::spawn(async move {
let mut rsi_data_vec: Vec<RsiData> = Vec::new(); let mut rsi_data_vec: Vec<RsiData> = Vec::new();
@ -58,12 +57,12 @@ pub async fn rsi(
let mut rsi: f64 = 0.0; let mut rsi: f64 = 0.0;
let mut last_close_time = 0; let mut last_close_time = 0;
if input_rt_data_c[T].1.len() < rsi_number + 1 { if input_rt_data_c[search_result.unwrap()].1.len() < rsi_number + 1 {
rsi_data.rsi_value = 0.0; rsi_data.rsi_value = 0.0;
rsi_data.close_time = 0; rsi_data.close_time = 0;
rsi_data_vec.push(rsi_data.clone()); rsi_data_vec.push(rsi_data.clone());
} else { } else {
read_data_vec = input_rt_data_c[T].1.clone(); read_data_vec = input_rt_data_c[search_result.unwrap()].1.clone();
if read_data_vec.len() >= (150 + rsi_number) as usize { if read_data_vec.len() >= (150 + rsi_number) as usize {
read_data_vec.reverse(); read_data_vec.reverse();
read_data_vec.truncate((150 + rsi_number) as usize); read_data_vec.truncate((150 + rsi_number) as usize);
@ -149,8 +148,6 @@ pub async fn rsi(
} }
})); }));
} }
None => {}
}
} }
try_join_all(task_vec).await?; try_join_all(task_vec).await?;
let a = rsi_data_wrapper_arc.lock().await.to_owned(); let a = rsi_data_wrapper_arc.lock().await.to_owned();