some rfks, todo add network as a mod?

Author: erhant

compute/src/handlers/workflow.rs

@@ -161,7 +161,6 @@ impl ComputeHandler for WorkflowHandler {
         };
 
         // try publishing the result
-
         if let Err(publish_err) = node.publish(message) {
             let err_msg = format!("Could not publish result: {:?}", publish_err);
             log::error!("{}", err_msg);

compute/src/node.rs

@@ -156,8 +156,6 @@ impl DriaComputeNode {
                         log::debug!("{:?}", self.p2p.network_info().connection_counters());
                     }
 
-
-
                     let (peer_id, message_id, message) = event;
                     let topic = message.topic.clone();
                     let topic_str = topic.as_str();

compute/src/utils/available_nodes/mod.rs

@@ -11,6 +11,15 @@ use crate::DriaNetworkType;
 /// Number of seconds between refreshing the available nodes.
 const DEFAULT_REFRESH_INTERVAL_SECS: u64 = 25;
 
+impl DriaNetworkType {
+    /// Returns the URL for fetching available nodes w.r.t network type.
+    pub fn get_available_nodes_url(&self) -> &str {
+        match self {
+            DriaNetworkType::Community => "https://dkn.dria.co/available-nodes",
+            DriaNetworkType::Pro => "https://dkn.dria.co/sdk/available-nodes",
+        }
+    }
+}
 /// Available nodes within the hybrid P2P network.
 ///
 /// - Bootstrap: used for Kademlia DHT bootstrap.
@@ -98,11 +107,7 @@ impl AvailableNodes {
         }
 
         // make the request w.r.t network type
-        let url = match self.network_type {
-            DriaNetworkType::Community => "https://dkn.dria.co/available-nodes",
-            DriaNetworkType::Pro => "https://dkn.dria.co/sdk/available-nodes",
-        };
-        let response = reqwest::get(url).await?;
+        let response = reqwest::get(self.network_type.get_available_nodes_url()).await?;
         let response_body = response.json::<AvailableNodesApiResponse>().await?;
         self.bootstrap_nodes
             .extend(parse_vec(response_body.bootstraps).into_iter());