59 lines
1.6 KiB
Rust
59 lines
1.6 KiB
Rust
use anyhow::{Error, Result};
|
|
use futures::{channel::mpsc::*, prelude::*, Stream};
|
|
|
|
use super::server::Handle;
|
|
use super::*;
|
|
|
|
pub struct ClientAgent {
|
|
server: Handle,
|
|
}
|
|
|
|
impl ClientAgent {
|
|
pub fn new(server: Handle) -> ClientAgent {
|
|
ClientAgent { server }
|
|
}
|
|
|
|
pub async fn run(
|
|
&mut self,
|
|
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 = self
|
|
.server
|
|
.lock()
|
|
.await
|
|
.add_player(sender, &self.server)
|
|
.await?;
|
|
|
|
// main message loop
|
|
let result: Result<()> = async {
|
|
loop {
|
|
tokio::select! {
|
|
client_message = socket.next() => {
|
|
match client_message {
|
|
Some(msg) =>
|
|
self.server.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
|
|
self.server.lock().await.remove_player(player_id);
|
|
|
|
result
|
|
}
|
|
}
|