Tahinli/rust-blockchain
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: ✨ send mined block for client

Browse source
This commit is contained in:
Ahmet Kaan GÜMÜŞ 2024-05-30 01:37:18 +03:00
parent 92b748ab6a
commit 2f4051d751
3 changed files with 44 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/client_network.rs
View file

@ -1,6 +1,9 @@
use futures_util::{stream::SplitStream, StreamExt};
use futures_util::{
stream::{SplitSink, SplitStream},
SinkExt, StreamExt,
};
use tokio::net::TcpStream;
use tokio_tungstenite::{connect_async, MaybeTlsStream, WebSocketStream};
use tokio_tungstenite::{connect_async, tungstenite::Message, MaybeTlsStream, WebSocketStream};
use crate::{block::Block, blockchain::BlockChain, ClientConfig};
@ -14,11 +17,14 @@ pub async fn start_network(client_config: ClientConfig) {
Ok(ws_stream) => ws_stream,
Err(_) => return,
};
let (_, ws_receiver) = ws_stream.0.split();
sync(ws_receiver).await;
let (ws_stream_sender, ws_stream_receiver) = ws_stream.0.split();
sync(ws_stream_sender, ws_stream_receiver).await;
}
async fn sync(ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>) {
async fn sync(
mut ws_stream_sender: SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>,
ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>,
) {
let (mut ws_stream_receiver, mut blockchain) =
match receive_blockchain(ws_stream_receiver).await {
Some((ws_stream_receiver, blockchain)) => (ws_stream_receiver, blockchain),
@ -30,7 +36,11 @@ async fn sync(ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<Tcp
Some((ws_stream_receiver, block)) => (ws_stream_receiver, block),
None => return,
};
blockchain.add_block(block);
let block = blockchain.add_block(block);
ws_stream_sender = match send_block(ws_stream_sender, block).await {
Some(ws_stream_sender) => ws_stream_sender,
None => return,
}
}
}
@ -83,3 +93,17 @@ async fn receive_block(
None => return None,
}
}
async fn send_block(
mut ws_stream_sender: SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>,
block: Block,
) -> Option<SplitSink<WebSocketStream<MaybeTlsStream<TcpStream>>, Message>> {
let block_data = serde_json::json!(block).to_string();
match ws_stream_sender.send(block_data.into()).await {
Ok(_) => match ws_stream_sender.flush().await {
Ok(_) => Some(ws_stream_sender),
Err(_) => None,
},
Err(_) => None,
}
}