chaiwala 0.1.5

Service Layer for Kucoin Arbitrage, along with Continuous Deployment
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

use crate::event;
use axum::body::Body;
use axum::extract::{Query, TypedHeader};
use axum::http::Response;
use axum::Extension;
use serde::{Deserialize, Serialize};
use std::sync::Arc;
use tokio::sync::{broadcast, Mutex};

#[derive(Debug, Deserialize, Serialize)]
pub struct SetStatusParam {
    status: event::RuntimeStatus,
}

/// HTTP handler, returns plain text
pub async fn plain_hello_world(
    user_agent: Option<TypedHeader<headers::UserAgent>>,
) -> Response<Body> {
    log::info!("Connected: {}", user_agent.unwrap().as_str());
    Response::new(Body::from("Hello, World!".to_string()))
}

/// Set runtime status by query
pub async fn handler_set_status(
    Query(param): Query<SetStatusParam>,
    tx: Extension<Arc<Mutex<broadcast::Sender<event::RuntimeStatus>>>>,
) -> Response<Body> {
    let msg = format!(
        "Received REST command to set runtime status as [{:?}]",
        param.status
    );
    // Add your code to restart your service here.
    let tx = tx.lock().await;
    if tx.send(param.status).is_err() {
        log::warn!("problem publishing runtime status, check receiver side")
    }
    log::info!("{msg}");
    Response::new(Body::from(msg))
}