fintek 0.1.1

Simple stock ticker
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use ::std::env;
use dotenv::dotenv;
use fintek::{check_tickers, metrics::MetricServer, Markets, StockMarket, Tickers};
use reqwest::Error;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter};
#[tokio::main]
async fn main() -> Result<(), Error> {
    let filter = EnvFilter::new(std::env::var("RUST_LOG").unwrap_or_else(|_| "info".into()));

    tracing_subscriber::registry()
        .with(filter)
        .with(
            tracing_subscriber::fmt::layer()
                .without_time()
                .with_thread_ids(true)
                .with_target(true)
                .json()
                .with_current_span(true)
                .with_span_list(false),
        )
        .init();

    tokio::spawn(async move {
        MetricServer::start(([127, 0, 0, 1], 9091).into()).await;
    });
    dotenv().ok();
    let api_key = env::var("API_KEY").expect("API_KEY must be set");

    let mut tickers = Tickers::init().await;

    loop {
        let night_time = fintek::should_sleep(Markets::Stock(StockMarket::NYSE), &api_key)
            .await
            .unwrap_or_default();

        tokio::time::sleep(tokio::time::Duration::from_secs(night_time)).await;

        if let Some(new) = check_tickers().await {
            tickers = new;
        }

        let num_tickers = tickers.get_tickers().len();
        let sleep_duration =
            fintek::calculate_sleep_duration(num_tickers, 8, 60, 800, (6.5 * 60. * 60.) as u64);

        if let Some(sleep_duration) = sleep_duration {
            for ticker in tickers.get_tickers() {
                let _ = fintek::call_api(ticker, &api_key).await.map_err(|e| {
                    tracing::error!(error = ?e, "Failed to call API");
                    e
                });
                tokio::time::sleep(tokio::time::Duration::from_secs(sleep_duration)).await;
            }
        }
    }
}

// {"price":"179.64000"}