From b0bb187f7591766256c78151ab81f540ea0b4f36 Mon Sep 17 00:00:00 2001
From: ModZero <modzero@modzero.xyz>
Date: Mon, 1 Jun 2020 19:30:56 +0200
Subject: [PATCH] Throttle the server a bit

---
 src/server.rs | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/server.rs b/src/server.rs
index 8a25e34..2bfa34b 100644
--- a/src/server.rs
+++ b/src/server.rs
@@ -1,11 +1,14 @@
-
-use std::sync::Arc;
+use std::{sync::Arc, time::Duration};
 
 use futures::{SinkExt, StreamExt};
 use serde_json::json;
 use tera::{Context, Tera};
-use tokio::sync::watch;
-use warp::{Filter, reject::Reject, Rejection, ws::{Message, WebSocket}};
+use tokio::{sync::watch, time::throttle};
+use warp::{
+    reject::Reject,
+    ws::{Message, WebSocket},
+    Filter, Rejection,
+};
 
 use crate::state::State;
 
@@ -54,23 +57,23 @@ struct WithTemplate {
 }
 
 async fn watch_state(ws: WebSocket, mut state_channel: watch::Receiver<State>) {
-    let (mut tx, mut rx) = ws.split();
+    let (mut tx, rx) = ws.split();
+    let mut rx = throttle(Duration::from_millis(1000 / 30), rx);
 
     while let Some(_) = rx.next().await {
         let state = state_channel.recv().await;
 
         let result = match state {
             Some(state) => tx.send(Message::text(json!(state).to_string())).await,
-            None => break
+            None => break,
         };
 
         if let Err(_) = result {
             break;
         }
-    };
+    }
 }
 
-
 #[derive(Debug)]
 struct SimulationUnavailable;
 
@@ -79,7 +82,7 @@ impl Reject for SimulationUnavailable {}
 async fn get_state(mut channel: watch::Receiver<State>) -> Result<State, Rejection> {
     match channel.recv().await {
         Some(state) => Ok(state),
-        None => Err(warp::reject::custom(SimulationUnavailable))
+        None => Err(warp::reject::custom(SimulationUnavailable)),
     }
 }