44 lines
1.2 KiB
Rust
44 lines
1.2 KiB
Rust
use anyhow::{Error, Result};
|
|
use futures::{channel::mpsc::*, prelude::*, Stream};
|
|
|
|
use super::server::Handle;
|
|
use super::*;
|
|
|
|
pub async fn run_client(
|
|
handle: Handle,
|
|
socket: &mut (impl Stream<Item = Result<ClientMessage, Error>>
|
|
+ Sink<ServerMessage, Error = Error>
|
|
+ Send
|
|
+ Unpin),
|
|
) -> Result<()> {
|
|
let (sender, mut receiver) = channel(CHANNEL_BUFFER);
|
|
|
|
// register player
|
|
let player_id = handle.lock().await.add_player(sender, &handle).await?;
|
|
|
|
// main message loop
|
|
let result: Result<()> = async {
|
|
loop {
|
|
tokio::select! {
|
|
client_message = socket.next() => {
|
|
match client_message {
|
|
Some(msg) =>
|
|
handle.lock().await.process_message(player_id, msg?),
|
|
None => break Ok(()),
|
|
}
|
|
},
|
|
Some(server_message) = receiver.next() => {
|
|
socket.send(server_message).await?
|
|
},
|
|
else => break Ok(()),
|
|
}
|
|
}
|
|
}
|
|
.await;
|
|
|
|
// deregister player, whether normally or due to error
|
|
handle.lock().await.remove_player(player_id);
|
|
|
|
result
|
|
}
|