Add node id to farmer and harvester stats

felixbrucker · Sep 23, 2023 · 30dcc6e · 30dcc6e
1 parent cb48604
commit 30dcc6e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/service/db/satellite.js b/lib/service/db/satellite.js
@@ -41,6 +41,7 @@ const satelliteSchema = new mongoose.Schema({
           }],
           default: undefined,
         },
+        nodeId: String,
       },
       lastUpdate: Date,
     },
@@ -60,6 +61,7 @@ const satelliteSchema = new mongoose.Schema({
         totalRawPlotCapacityInGib: String,
         totalEffectivePlotCapacityInGib: String,
         farmerConnectionsCount: Number,
+        nodeId: String,
       },
       lastUpdate: Date,
     },

diff --git a/lib/service/update-query-factory.js b/lib/service/update-query-factory.js
@@ -32,6 +32,9 @@ class UpdateQueryFactory {
     if (stats.farmingInfos !== undefined) {
       updateQuery['services.farmer.stats.farmingInfos'] = stats.farmingInfos
     }
+    if (stats.nodeId !== undefined) {
+      updateQuery['services.farmer.stats.nodeId'] = stats.nodeId
+    }
 
     return updateQuery
   }
@@ -118,6 +121,9 @@ class UpdateQueryFactory {
     if (stats.farmerConnectionsCount !== undefined) {
       updateQuery['services.harvester.stats.farmerConnectionsCount'] = stats.farmerConnectionsCount
     }
+    if (stats.nodeId !== undefined) {
+      updateQuery['services.harvester.stats.nodeId'] = stats.nodeId
+    }
 
     return updateQuery
   }