GlcoinCore/electrs
1
0
Fork 0
mirror of https://github.com/romanz/electrs.git synced 2026-08-17 13:06:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Notify only about new blocks from P2P layer

Browse source
This commit is contained in:
Roman Zeyde 2018-08-15 17:05:12 +03:00
parent 818148dc0e
commit 604f7680df
No known key found for this signature in database
GPG key ID: 87CAE5FA46917CBB
2 changed files with 12 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/listener.rs
View file

@ -9,7 +9,7 @@ use electrs::notify;
fn main() {
let _ = Config::from_args();
let rx = notify::run().into_receiver();
for msg in rx.iter() {
info!("{:?}", msg)
for blockhash in rx.iter() {
info!("{:?}", blockhash.be_hex_string())
}
}

16
src/notify.rs
View file

@ -1,10 +1,12 @@
use bitcoin::network::constants::Network;
use bitcoin::network::message::NetworkMessage;
use bitcoin::network::message_blockdata::InvType;
use bitcoin::network::socket::Socket;
use bitcoin::util::hash::Sha256dHash;
use util;
pub fn run() -> util::Channel<NetworkMessage> {
pub fn run() -> util::Channel<Sha256dHash> {
let chan = util::Channel::new();
let tx = chan.sender();
@ -32,17 +34,19 @@ pub fn run() -> util::Channel<NetworkMessage> {
break;
}
};
debug!("recv {:?}", msg);
match msg {
NetworkMessage::Alert(_) => continue, // deprecated
NetworkMessage::Version(_) => outgoing.push(NetworkMessage::Verack),
NetworkMessage::Ping(nonce) => outgoing.push(NetworkMessage::Pong(nonce)),
NetworkMessage::Inv(ref inventory) => {
inventory
.iter()
.filter(|inv| inv.inv_type == InvType::Block)
.for_each(|inv| tx.send(inv.hash).expect("failed to send message"));
}
_ => (),
};
debug!("recv {:?}", msg);
if tx.send(msg).is_err() {
warn!("failed to connect to node");
return;
}
}
});