X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=src%2Fpersistence.rs;h=04c6b9a0c4076cab725b90b5a4ec769718d183a0;hb=74dd28d4e2c75a4dbb15b3c5045bddfeb2bd9c53;hp=e0c734d7a0907cce82401c8a2696625efcd72212;hpb=c5ad40eba4f970b0025cf590a1e2195a67cd541d;p=rapid-gossip-sync-server

diff --git a/src/persistence.rs b/src/persistence.rs
index e0c734d..04c6b9a 100644
--- a/src/persistence.rs
+++ b/src/persistence.rs
@@ -132,8 +132,58 @@ impl<L: Deref> GossipPersister<L> where L::Target: Logger {
 
 			let connections_cache_ref = Arc::clone(&connections_cache);
 			match gossip_message {
-				GossipMessage::NodeAnnouncement(_announcement, _seen_override) => {
+				GossipMessage::NodeAnnouncement(announcement, seen_override) => {
+					let public_key_hex = announcement.contents.node_id.to_string();
 
+					let mut announcement_signed = Vec::new();
+					announcement.write(&mut announcement_signed).unwrap();
+
+					let features = announcement.contents.features.encode();
+					let timestamp = announcement.contents.timestamp as i64;
+
+					let mut serialized_addresses = Vec::new();
+					announcement.contents.addresses.write(&mut serialized_addresses).unwrap();
+
+					let _task = self.tokio_runtime.spawn(async move {
+						if cfg!(test) && seen_override.is_some() {
+							tokio::time::timeout(POSTGRES_INSERT_TIMEOUT, client
+								.execute("INSERT INTO node_announcements (\
+								public_key, \
+								features, \
+								socket_addresses, \
+								timestamp, \
+								announcement_signed, \
+								seen \
+							) VALUES ($1, $2, $3, $4, $5, TO_TIMESTAMP($6))", &[
+									&public_key_hex,
+									&features,
+									&serialized_addresses,
+									&timestamp,
+									&announcement_signed,
+									&(seen_override.unwrap() as f64)
+								])).await.unwrap().unwrap();
+						} else {
+							tokio::time::timeout(POSTGRES_INSERT_TIMEOUT, client
+								.execute("INSERT INTO node_announcements (\
+								public_key, \
+								features, \
+								socket_addresses, \
+								timestamp, \
+								announcement_signed \
+							) VALUES ($1, $2, $3, $4, $5)", &[
+									&public_key_hex,
+									&features,
+									&serialized_addresses,
+									&timestamp,
+									&announcement_signed,
+								])).await.unwrap().unwrap();
+						}
+						let mut connections_set = connections_cache_ref.lock().await;
+						connections_set.push(client);
+						limiter_ref.add_permits(1);
+					});
+					#[cfg(test)]
+					tasks_spawned.push(_task);
 				},
 				GossipMessage::ChannelAnnouncement(announcement, seen_override) => {
 					let scid = announcement.contents.short_channel_id as i64;