Constructor for notifier config

This commit is contained in:
Jedrzej Stuczynski
2020-02-04 14:36:02 +00:00
parent ed9ee27340
commit b2f41de27b
2 changed files with 31 additions and 16 deletions
+7 -7
View File
@@ -248,14 +248,14 @@ impl MixNode {
let mut rt = Runtime::new()?;
// Spawn Tokio tasks as necessary for node functionality
let tmp_notifier_config = presence::NotifierConfig {
directory_server: self.config.get_directory_server(),
host: self.config.get_announce_address(),
pub_key: self.sphinx_keypair.public_key().to_base58_string(),
layer: self.config.get_layer(),
};
let notifier_config = presence::NotifierConfig::new(
self.config.get_directory_server(),
self.config.get_announce_address(),
self.sphinx_keypair.public_key().to_base58_string(),
self.config.get_layer(),
);
rt.spawn({
let presence_notifier = presence::Notifier::new(tmp_notifier_config);
let presence_notifier = presence::Notifier::new(notifier_config);
presence_notifier.run()
});
+24 -9
View File
@@ -1,4 +1,4 @@
use crate::node;
use crate::built_info;
use directory_client::presence::mixnodes::MixNodePresence;
use directory_client::requests::presence_mixnodes_post::PresenceMixNodesPoster;
use directory_client::DirectoryClient;
@@ -11,11 +11,26 @@ pub struct Notifier {
}
pub struct NotifierConfig {
// TEMPORARLY PUBLIC
pub directory_server: String,
pub host: String,
pub pub_key: String,
pub layer: u64,
directory_server: String,
announce_host: String,
pub_key_string: String,
layer: u64,
}
impl NotifierConfig {
pub fn new(
directory_server: String,
announce_host: String,
pub_key_string: String,
layer: u64,
) -> Self {
NotifierConfig {
directory_server,
announce_host,
pub_key_string,
layer,
}
}
}
impl Notifier {
@@ -25,11 +40,11 @@ impl Notifier {
};
let net_client = directory_client::Client::new(directory_client_cfg);
let presence = MixNodePresence {
host: config.host,
pub_key: config.pub_key,
host: config.announce_host,
pub_key: config.pub_key_string,
layer: config.layer,
last_seen: 0,
version: env!("CARGO_PKG_VERSION").to_string(),
version: built_info::PKG_VERSION.to_string(),
};
Notifier {
net_client,