From 2210e3e60d450240b091011aa861f02c4126cd55 Mon Sep 17 00:00:00 2001
From: Thomas Haller <Thomas.Haller@lab10.coop>
Date: Wed, 4 Aug 2021 21:56:52 +0200
Subject: [PATCH] first working version that tracks hbbft messages as JSON.

---
 .../ethcore/src/engines/hbbft/hbbft_engine.rs |  2 +-
 .../engines/hbbft/hbbft_message_memorium.rs   | 61 +++++++++++++------
 2 files changed, 44 insertions(+), 19 deletions(-)

diff --git a/crates/ethcore/src/engines/hbbft/hbbft_engine.rs b/crates/ethcore/src/engines/hbbft/hbbft_engine.rs
index ddf0630998..c8562583e5 100644
--- a/crates/ethcore/src/engines/hbbft/hbbft_engine.rs
+++ b/crates/ethcore/src/engines/hbbft/hbbft_engine.rs
@@ -1,6 +1,6 @@
 use super::block_reward_hbbft::BlockRewardContract;
 use block::ExecutedBlock;
-use client::traits::{EngineClient, ForceUpdateSealing, TransactionRequest};
+use client::traits::{EngineClient, ForceUpdateSealing};
 use crypto::publickey::Signature;
 use engines::{
     default_system_or_code_call, signer::EngineSigner, Engine, EngineError, ForkChoice, Seal,
diff --git a/crates/ethcore/src/engines/hbbft/hbbft_message_memorium.rs b/crates/ethcore/src/engines/hbbft/hbbft_message_memorium.rs
index 353ff6bfde..3bdc720fc3 100644
--- a/crates/ethcore/src/engines/hbbft/hbbft_message_memorium.rs
+++ b/crates/ethcore/src/engines/hbbft/hbbft_message_memorium.rs
@@ -2,10 +2,16 @@ use hbbft::honey_badger::{self};
 
 // use threshold_crypto::{SignatureShare};
 use engines::hbbft::NodeId;
-use hbbft::honey_badger::Message;
+// use hbbft::honey_badger::Message;
 use serde::{Deserialize, Serialize};
 use serde_json::{json, Result, Value};
-use std::{borrow::Borrow, collections::BTreeMap};
+use std::{
+    borrow::Borrow,
+    collections::BTreeMap,
+    fs::{self, create_dir_all, File},
+    io::Write,
+    path::{Path, PathBuf},
+};
 
 pub type HbMessage = honey_badger::Message<NodeId>;
 
@@ -34,6 +40,8 @@ pub(crate) struct HbbftMessageMemorium {
     // HbMessage: message
     // */
     agreements: BTreeMap<u64, Vec<(NodeId, NodeId, HbMessage)>>,
+
+    message_tracking_id: u64,
 }
 
 impl HbbftMessageMemorium {
@@ -41,37 +49,54 @@ impl HbbftMessageMemorium {
         HbbftMessageMemorium {
             signature_shares: BTreeMap::new(),
             agreements: BTreeMap::new(),
+            message_tracking_id: 0,
         }
     }
 
-    pub fn on_message_string_received(&self, message: String, epoch: u64) {
+    pub fn on_message_string_received(&mut self, message_json: String, epoch: u64) {
+        self.message_tracking_id += 1;
+        let mut path_buf = PathBuf::from(format!(
+            "data/messages/{}/message_{}.json",
+            epoch, message_tracking_id
+        ));
+        if let Err(e) = create_dir_all(path_buf.as_path()) {
+            warn!("Error creating key directory: {:?}", e);
+            return;
+        };
+
+        path_buf.push(format!("{}", self.message_tracking_id));
+
+        let path = path_buf.as_path();
+        let mut file = match File::create(&path) {
+            Ok(file) => file,
+            Err(e) => {
+                warn!(target: "consensus", "Error creating hbbft memorial file: {:?}", e);
+                return;
+            }
+        };
 
-        // if (message.contains("")
+        if let Err(e) = file.write(message_json.as_bytes()) {
+            warn!(target: "consensus", "Error writing hbbft memorial file: {:?}", e);
+        }
     }
 
-    pub fn on_message_received(&self, message: &HbMessage) {
-        //performance: dispatcher pattern could improve performance a lot.
-        let message_string = format!("{:?}", message);
+    pub fn on_message_received(&mut self, message: &HbMessage) {
+        //performance: dispatcher pattern + multithreading could improve performance a lot.
 
         let epoch = message.epoch();
 
         match serde_json::to_string(message) {
-            Ok(jsonString) => {
-                debug!(target: "consensus", "{}", jsonString);
+            Ok(json_string) => {
+                debug!(target: "consensus", "{}", json_string);
+                self.on_message_string_received(json_string, epoch);
             }
             Err(e) => {
                 error!(target: "consensus", "could not create json.");
             }
         }
+    }
 
-        // how to figure out proposer ?
-
-        self.on_message_string_received(message_string, epoch);
-
-        //the details are imprisionated
-        // todo implementation.
-
-        // let Message(share) = message;
-        // let bytes = share.to_bytes();
+    pub fn free_epoch_memory(&mut self, epoch: u64) {
+        self.signature_shares.remove(&epoch);
     }
 }