From 1e17e017cf3f04b02c29ac4f6703cedda0b97c1d Mon Sep 17 00:00:00 2001 From: Nikita Vilunov Date: Wed, 15 Feb 2023 21:49:52 +0100 Subject: [PATCH] make PlayerCommand mod-private --- src/core/player.rs | 8 +++++--- src/projections/irc/mod.rs | 6 ++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/core/player.rs b/src/core/player.rs index 468d6a6..7a4a984 100644 --- a/src/core/player.rs +++ b/src/core/player.rs @@ -89,8 +89,10 @@ impl PlayerConnection { .await; } - pub async fn send(&self, command: PlayerCommand) { - self.player_handle.send(command).await; + pub async fn get_rooms(&self) -> Result> { + let (promise, deferred) = oneshot(); + self.player_handle.send(PlayerCommand::GetRooms(promise)).await; + Ok(deferred.await?) } } @@ -153,7 +155,7 @@ impl PlayerHandle { } } -pub enum PlayerCommand { +enum PlayerCommand { /** Commands from connections */ AddConnection { sender: Sender, diff --git a/src/projections/irc/mod.rs b/src/projections/irc/mod.rs index 463314e..5f9b96c 100644 --- a/src/projections/irc/mod.rs +++ b/src/projections/irc/mod.rs @@ -7,7 +7,7 @@ use tokio::net::tcp::{ReadHalf, WriteHalf}; use tokio::net::{TcpListener, TcpStream}; use tokio::sync::oneshot::channel; -use crate::core::player::{PlayerCommand, PlayerConnection, PlayerId, PlayerRegistry, Updates}; +use crate::core::player::{PlayerConnection, PlayerId, PlayerRegistry, Updates}; use crate::core::room::{RoomId, RoomInfo, RoomRegistry}; use crate::prelude::*; use crate::protos::irc::client::{client_message, ClientMessage}; @@ -196,9 +196,7 @@ async fn handle_registered_socket<'a>( .write_async(writer) .await?; - let (promise, deferred) = oneshot(); - connection.send(PlayerCommand::GetRooms(promise)).await; - let rooms_list = deferred.await?; + let rooms_list = connection.get_rooms().await?; for room in &rooms_list { produce_on_join_cmd_messages( &config,