core/src/chat/serv_conn.rs

55 lines
1.6 KiB
Rust
Raw Normal View History

2021-02-27 21:58:55 +00:00
use super::serv_handler::to_server_sender;
use crate::chat::types::{PackedMessage, RP, SP};
2021-01-06 09:41:33 +00:00
use async_std::task;
2021-01-06 04:18:46 +00:00
use serde_derive::Deserialize;
use std::env;
2021-01-06 09:41:33 +00:00
use tokio::{
2021-02-27 21:58:55 +00:00
io::{self, AsyncReadExt},
net::{tcp::OwnedReadHalf, TcpStream},
2021-01-06 09:41:33 +00:00
};
2021-01-06 04:18:46 +00:00
2021-01-06 09:41:33 +00:00
#[tokio::main]
2021-01-09 14:03:00 +00:00
pub async fn listen_server(client_sender: SP, server_receiver: RP) -> io::Result<()> {
task::block_on(connect_to_server(client_sender, server_receiver))
2021-01-06 04:18:46 +00:00
}
2021-01-09 14:03:00 +00:00
async fn connect_to_server(client_sender: SP, server_receiver: RP) -> io::Result<()> {
2021-01-06 04:18:46 +00:00
let addr = env::args()
.nth(1)
.unwrap_or_else(|| "127.0.0.1:9481".to_string());
2021-01-06 09:41:33 +00:00
let stream = TcpStream::connect(&addr).await.expect("weeror here");
2021-02-27 21:58:55 +00:00
let (receiver, sender) = stream.into_split();
2021-01-09 14:03:00 +00:00
let t = task::spawn(server_responce_getter(receiver, client_sender));
to_server_sender(sender, server_receiver).await?;
match t.await {
Ok(_) => Ok(()),
Err(e) => Err(e),
2021-01-09 14:03:00 +00:00
}
}
async fn server_responce_getter(mut receiver: OwnedReadHalf, client_sender: SP) -> io::Result<()> {
loop {
2021-01-06 09:41:33 +00:00
let mut buffer = [0; 512];
match receiver.read(&mut buffer).await {
2021-01-09 14:03:00 +00:00
Ok(_) => {
2021-01-06 09:41:33 +00:00
let received = String::from_utf8_lossy(&buffer[..]);
2021-01-09 14:03:00 +00:00
println!("received {}", received);
2021-01-06 09:41:33 +00:00
client_sender
.send(PackedMessage {
message: received.to_string(),
})
.unwrap();
}
Err(e) => println!("Error: {} ", e),
}
}
2021-01-06 04:18:46 +00:00
}
#[derive(Deserialize, Debug)]
struct FrontMsg {
2021-01-09 14:03:00 +00:00
user_id: u32,
receiver_id: u32,
2021-01-06 04:18:46 +00:00
message: String,
time: String,
}