ps2.live/agg-population
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

background refresh

Browse source
This commit is contained in:
41666 2023-06-09 07:52:19 -04:00
parent 30266b294d
commit 315a05d59a
2 changed files with 24 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/handlers.rs
View file

@ -9,7 +9,7 @@ use axum::{
use tokio::task::JoinSet; use tokio::task::JoinSet;
pub async fn get_one_world(State(db): State<sled::Db>, Path(world): Path<i32>) -> Json<Response> { pub async fn get_one_world(State(db): State<sled::Db>, Path(world): Path<i32>) -> Json<Response> {
Json(get_world(db, world).await) Json(get_world(db, world, false).await)
} }
pub async fn get_all_worlds(State(db): State<sled::Db>) -> Json<Vec<Response>> { pub async fn get_all_worlds(State(db): State<sled::Db>) -> Json<Vec<Response>> {
@ -17,7 +17,7 @@ pub async fn get_all_worlds(State(db): State<sled::Db>) -> Json<Vec<Response>> {
let mut worlds = vec![Response::default(); 8]; let mut worlds = vec![Response::default(); 8];
for world in vec![1, 10, 13, 17, 19, 40, 1000, 2000] { for world in vec![1, 10, 13, 17, 19, 40, 1000, 2000] {
set.spawn(get_world(db.clone(), world)); set.spawn(get_world(db.clone(), world, false));
} }
let mut i = 0; let mut i = 0;
@ -29,9 +29,11 @@ pub async fn get_all_worlds(State(db): State<sled::Db>) -> Json<Vec<Response>> {
Json(worlds) Json(worlds)
} }
async fn get_world(db: sled::Db, world: i32) -> Response { pub async fn get_world(db: sled::Db, world: i32, skip_cache: bool) -> Response {
if let Ok(data) = world_from_cache(db.clone(), world) { if !skip_cache {
return data; if let Ok(data) = world_from_cache(db.clone(), world) {
return data;
}
} }
let mut response = Response::default(); let mut response = Response::default();
@ -81,7 +83,7 @@ fn world_from_cache(db: sled::Db, world: i32) -> Result<Response, ()> {
match bincode::deserialize::<Response>(&value) { match bincode::deserialize::<Response>(&value) {
Ok(response) => { Ok(response) => {
if response.cached_at + chrono::Duration::minutes(3) < chrono::Utc::now() { if response.cached_at + chrono::Duration::minutes(5) < chrono::Utc::now() {
return Err(()); return Err(());
} }
Ok(response) Ok(response)

18
src/main.rs
View file

@ -1,6 +1,7 @@
use crate::handlers::{get_all_worlds, get_one_world}; use crate::handlers::{get_all_worlds, get_one_world, get_world};
use axum::{response::Html, routing::get, Router}; use axum::{response::Html, routing::get, Router};
use std::net::SocketAddr; use std::net::SocketAddr;
use tokio::task::JoinSet;
use tower_http::trace::TraceLayer; use tower_http::trace::TraceLayer;
mod handlers; mod handlers;
@ -21,7 +22,20 @@ async fn main() {
.route("/population/all", get(get_all_worlds)) .route("/population/all", get(get_all_worlds))
.route("/population/:world", get(get_one_world)) .route("/population/:world", get(get_one_world))
.layer(TraceLayer::new_for_http()) .layer(TraceLayer::new_for_http())
.with_state(db); .with_state(db.clone());
tokio::spawn(async move {
loop {
let mut set = JoinSet::new();
for world in vec![1, 10, 13, 17, 19, 40, 1000, 2000] {
set.spawn(get_world(db.clone(), world, true));
}
while let Some(_) = set.join_next().await {}
tokio::time::sleep(tokio::time::Duration::from_secs(60 * 3)).await;
}
});
let addr = SocketAddr::from(([0, 0, 0, 0], 3000)); let addr = SocketAddr::from(([0, 0, 0, 0], 3000));
tracing::debug!("listening on {}", addr); tracing::debug!("listening on {}", addr);