implement solo mining mode for the handle_new_template

Author: lucasbalieiro

pool-apps/pool/src/lib/channel_manager/mining_message_handler.rs

@@ -163,6 +163,8 @@ impl HandleMiningMessagesFromClientAsync for ChannelManager {
             );
 
             downstream.downstream_data.super_safe_lock(|downstream_data| {
+                downstream_data.payout_mode = Some(payout_mode);
+
                 let nominal_hash_rate = msg.nominal_hash_rate;
                 let requested_max_target = Target::from_le_bytes(msg.max_target.inner_as_ref().try_into().unwrap());
                 let extranonce_prefix = channel_manager_data.extranonce_prefix_factory_standard.next_prefix_standard().map_err(PoolError::shutdown)?;
@@ -342,6 +344,8 @@ impl HandleMiningMessagesFromClientAsync for ChannelManager {
                             }
                         };
 
+                        downstream_data.payout_mode = Some(payout_mode.clone());
+
                         let channel_id = downstream_data
                             .channel_id_factory
                             .fetch_add(1, Ordering::SeqCst);

pool-apps/pool/src/lib/channel_manager/template_distribution_message_handler.rs

@@ -51,7 +51,13 @@ impl HandleTemplateDistributionMessagesFromServerAsync for ChannelManager {
                 }
 
                 let messages_: Vec<RouteMessageTo<'_>> = downstream.downstream_data.super_safe_lock(|data| {
-                    data.group_channel.on_new_template(msg.clone().into_static(), coinbase_output.clone()).map_err(|e| {
+                    let downstream_coinbase_outputs = if let Some(ref payout_mode) = data.payout_mode {
+                        payout_mode.coinbase_outputs(msg.coinbase_tx_value_remaining, &self.coinbase_reward_script)
+                    } else {
+                        coinbase_output.clone()
+                    };
+
+                    data.group_channel.on_new_template(msg.clone().into_static(), downstream_coinbase_outputs.clone()).map_err(|e| {
                         tracing::error!("Error while adding template to group channel");
                         PoolError::shutdown(e)
                     })?;
@@ -92,7 +98,7 @@ impl HandleTemplateDistributionMessagesFromServerAsync for ChannelManager {
                                 PoolError::shutdown(e)
                             })?;
                         } else {
-                            standard_channel.on_new_template(msg.clone().into_static(), coinbase_output.clone()).map_err(|e| {
+                            standard_channel.on_new_template(msg.clone().into_static(), downstream_coinbase_outputs.clone()).map_err(|e| {
                                 tracing::error!("Error while adding template to standard channel");
                                 PoolError::shutdown(e)
                             })?;

pool-apps/pool/src/lib/downstream/mod.rs

@@ -36,6 +36,7 @@ use crate::{
     error::{self, PoolError, PoolErrorKind, PoolResult},
     io_task::spawn_io_tasks,
     status::{handle_error, Status, StatusSender},
+    utils::PayoutMode,
 };
 
 mod common_message_handler;
@@ -58,6 +59,8 @@ pub struct DownstreamData {
     pub channel_id_factory: AtomicU32,
     /// Extensions that have been successfully negotiated with this client
     pub negotiated_extensions: Vec<u16>,
+    /// Payout mode derived from user_identity (None until channel is opened)
+    pub payout_mode: Option<PayoutMode>,
 }
 
 /// Communication layer for a downstream connection.
@@ -144,6 +147,7 @@ impl Downstream {
             group_channel,
             channel_id_factory,
             negotiated_extensions: vec![],
+            payout_mode: None,
         }));
 
         Downstream {

pool-apps/pool/src/lib/utils.rs

@@ -84,7 +84,7 @@ pub(crate) fn create_close_channel_msg(channel_id: ChannelId, msg: &str) -> Clos
 ///   `sri/donate/<percentage>/<payout_address>/<worker_name>` (percentage is 0-100, representing
 ///   pool's portion)
 /// - `FullDonation`: Full reward goes to the pool.
-#[derive(Debug)]
+#[derive(Debug, Clone)]
 pub enum PayoutMode {
     /// Solo mode: miner receives full block reward.
     Solo { script: CoinbaseRewardScript },