diff options
author | pommicket <pommicket@gmail.com> | 2024-08-09 16:55:37 -0400 |
---|---|---|
committer | pommicket <pommicket@gmail.com> | 2024-08-09 16:55:37 -0400 |
commit | c8450482fec7bbe40f3ebfd7c38facbde0323d5d (patch) | |
tree | 8820d05d33c2472dd6e9438274369ce2e42976b2 /server/src | |
parent | cd991e8b24db1215e6a935ab0b305fb90ed44aa4 (diff) |
update TODO, cargo fmt
Diffstat (limited to 'server/src')
-rw-r--r-- | server/src/main.rs | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/server/src/main.rs b/server/src/main.rs index 059ce44..d6d85d3 100644 --- a/server/src/main.rs +++ b/server/src/main.rs @@ -1,11 +1,11 @@ use anyhow::anyhow; use futures_util::{SinkExt, StreamExt}; use rand::Rng; +use std::io::prelude::*; use std::net::SocketAddr; use std::sync::LazyLock; use tokio::io::AsyncWriteExt; use tungstenite::protocol::Message; -use std::io::prelude::*; const PUZZLE_ID_CHARSET: &[u8] = b"23456789abcdefghijkmnopqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ"; const PUZZLE_ID_LEN: usize = 6; @@ -56,8 +56,8 @@ async fn handle_connection( let mut ws = tokio_tungstenite::accept_async_with_config( conn, Some(tungstenite::protocol::WebSocketConfig { - max_message_size: Some(4096), - max_frame_size: Some(4096), + max_message_size: Some(65536), + max_frame_size: Some(65536), ..Default::default() }), ) @@ -250,14 +250,17 @@ async fn handle_connection( ws.send(Message::Binary(data)).await?; } else if text == "randomFeaturedWikimedia" { let choice = rand::thread_rng().gen_range(0..server.wikimedia_featured.len()); - ws.send(Message::Text(format!("wikimediaImage {}", server.wikimedia_featured[choice]))).await?; + ws.send(Message::Text(format!( + "wikimediaImage {}", + server.wikimedia_featured[choice] + ))) + .await?; } } } Ok(()) } - fn read_to_lines(path: &str) -> std::io::Result<Vec<String>> { let file = std::fs::File::open(path)?; let reader = std::io::BufReader::new(file); @@ -282,7 +285,8 @@ async fn main() { } }); static SERVER_VALUE: LazyLock<Server> = LazyLock::new(|| { - let wikimedia_featured = read_to_lines("featuredpictures.txt").expect("Couldn't read featuredpictures.txt"); + let wikimedia_featured = + read_to_lines("featuredpictures.txt").expect("Couldn't read featuredpictures.txt"); let db = sled::open("database.sled").expect("error opening database"); let puzzles = db.open_tree("PUZZLES").expect("error opening puzzles tree"); let pieces = db.open_tree("PIECES").expect("error opening pieces tree"); |