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

feat: ✨ blockchain and block transfer

Browse source
This commit is contained in:
Ahmet Kaan GÜMÜŞ 2024-05-25 20:07:22 +03:00
parent 35c1207cff
commit e7d10c46f8
6 changed files with 98 additions and 62 deletions
Download patch file Download diff file Expand all files Collapse all files

54
src/client_network.rs
View file

@ -2,7 +2,7 @@ use futures_util::{stream::SplitStream, StreamExt};
use tokio::net::TcpStream;
use tokio_tungstenite::{connect_async, MaybeTlsStream, WebSocketStream};
use crate::ClientConfig;
use crate::{block::Block, blockchain::BlockChain, ClientConfig};
pub async fn start_network(client_config: ClientConfig) {
let ws_stream = match connect_async(format!(
@ -19,38 +19,64 @@ pub async fn start_network(client_config: ClientConfig) {
}
async fn sync(ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>) {
let mut ws_stream = match receive_blockchain(ws_stream_receiver).await {
Some(ws_stream) => ws_stream,
let (mut ws_stream, mut blockchain) = match receive_blockchain(ws_stream_receiver).await {
Some((ws_stream, blockchain)) => (ws_stream, blockchain),
None => return,
};
loop {
ws_stream = match receive_block(ws_stream).await {
Some(ws_stream) => ws_stream,
let block: Block;
(ws_stream, block) = match receive_block(ws_stream).await {
Some((ws_stream, block)) => (ws_stream, block),
None => return,
}
};
blockchain.add_block(block);
}
}
async fn receive_blockchain(mut ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>) -> Option<SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>> {
async fn receive_blockchain(
mut ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>,
) -> Option<(
SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>,
BlockChain,
)> {
match ws_stream_receiver.next().await {
Some(message) => match message {
Ok(message) => {
println!("{}", message);
Some(ws_stream_receiver)
},
if let tokio_tungstenite::tungstenite::Message::Text(message) = message {
let blockchain: BlockChain = match serde_json::from_str(&message[..]) {
Ok(blockchain) => blockchain,
Err(_) => return None,
};
Some((ws_stream_receiver, blockchain))
} else {
return None;
}
}
Err(_) => return None,
},
None => return None,
}
}
async fn receive_block(mut ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>) -> Option<SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>> {
async fn receive_block(
mut ws_stream_receiver: SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>,
) -> Option<(
SplitStream<WebSocketStream<MaybeTlsStream<TcpStream>>>,
Block,
)> {
match ws_stream_receiver.next().await {
Some(message) => match message {
Ok(message) => {
println!("{}", message);
Some(ws_stream_receiver)
},
if let tokio_tungstenite::tungstenite::Message::Text(message) = message {
let block: Block = match serde_json::from_str(&message[..]) {
Ok(block) => block,
Err(_) => return None,
};
Some((ws_stream_receiver, block))
} else {
return None;
}
}
Err(_) => return None,
},
None => return None,