|
|
@@ -1,13 +1,13 @@
|
|
|
1
|
+use std::fs::create_dir_all;
|
|
1
|
2
|
use std::net::SocketAddr;
|
|
2
|
3
|
use std::sync::{Arc, Mutex};
|
|
3
|
|
-use std::fs::create_dir_all;
|
|
4
|
4
|
|
|
5
|
5
|
use async_tungstenite::accept_async;
|
|
|
6
|
+use futures::sink::SinkExt;
|
|
|
7
|
+use futures_tokio_compat::Compat;
|
|
6
|
8
|
use log::*;
|
|
7
|
9
|
use tokio::net::{TcpListener, TcpStream};
|
|
8
|
10
|
use tokio::stream::StreamExt;
|
|
9
|
|
-use futures_tokio_compat::Compat;
|
|
10
|
|
-use futures::sink::SinkExt;
|
|
11
|
11
|
|
|
12
|
12
|
use twfss::Database;
|
|
13
|
13
|
|
|
|
@@ -65,7 +65,9 @@ async fn main() {
|
|
65
|
65
|
// TODO: Robust error handling.
|
|
66
|
66
|
|
|
67
|
67
|
async fn accept_connection(peer: SocketAddr, stream: TcpStream) {
|
|
68
|
|
- let mut ws_stream = accept_async(Compat::new(stream)).await.expect("Failed to accept");
|
|
|
68
|
+ let mut ws_stream = accept_async(Compat::new(stream))
|
|
|
69
|
+ .await
|
|
|
70
|
+ .expect("Failed to accept");
|
|
69
|
71
|
|
|
70
|
72
|
info!("New WebSocket connection from {}.", peer);
|
|
71
|
73
|
|