andr3/unwritten-application
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

simple clean up

Browse Source
This commit is contained in:
Andre Henriques 2024-12-08 16:23:30 +00:00
parent 36b60cd298
commit f12ed610d7
1 changed files with 20 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/main.rs
View File

@ -1,12 +1,10 @@
use std::sync::Mutex;
use axum::{
response::IntoResponse,
routing::{delete, get, put},
Extension, Router,
};
use polars::prelude::*;
use tokio_postgres::NoTls;
use std::sync::Mutex;
// Generetes a bunch of random data in the db
async fn generate_random_data(
@ -198,17 +196,19 @@ async fn get_data(
async fn accounts(Extension(state): Extension<Arc<State>>) -> impl IntoResponse {
let accounts = state.accounts_dataframe.lock().unwrap();
if accounts.is_some() {
if accounts.is_none() {
return "Injest Data First".into();
}
let df = accounts.as_ref().unwrap().clone();
let to_send = df.select(vec![col("uuid"), col("name")]).collect().unwrap();
return format!("{}", to_send);
}
return "Injest Data First".into();
}
async fn transactions(Extension(state): Extension<Arc<State>>) -> impl IntoResponse {
let transactions = state.transactions_dataframe.lock().unwrap();
if transactions.is_some() {
if transactions.is_none() {
return "Injest Data First".into();
}
let df = transactions.as_ref().unwrap().clone();
let to_send = df
.select(vec![
@ -221,8 +221,6 @@ async fn transactions(Extension(state): Extension<Arc<State>>) -> impl IntoRespo
.unwrap();
return format!("{}", to_send);
}
return "Injest Data First".into();
}
struct State {
accounts_dataframe: Arc<Mutex<Option<LazyFrame>>>,
@ -246,7 +244,7 @@ async fn main() {
}
let psql_conn = psql_conn.unwrap();
let (client, connection) = tokio_postgres::connect(psql_conn.as_str(), NoTls)
let (client, connection) = tokio_postgres::connect(psql_conn.as_str(), tokio_postgres::NoTls)
.await
.unwrap();