nodio-cli 1.0.7 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nodio-cli",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Nodio distributed storage network",
   "main": "src/server/index.js",
   "type": "commonjs",

package/src/server/config.js

@@ -5,7 +5,7 @@ function getServerConfig() {
   return {
     port: Number(process.env.PORT || process.env.NODIO_SERVER_PORT || 4000),
     mongoUri: process.env.NODIO_MONGO_URI || 'mongodb://127.0.0.1:27017/nodio',
-    heartbeatIntervalMs: Number(process.env.NODIO_HEARTBEAT_INTERVAL_MS || 30000),
+    heartbeatIntervalMs: Number(process.env.NODIO_HEARTBEAT_INTERVAL_MS || 10000),
     offlineAfterMisses: Number(process.env.NODIO_OFFLINE_AFTER_MISSES || 3),
     minReplicas: Number(process.env.NODIO_MIN_REPLICAS || 5),
     emergencyReplicaFloor: Number(process.env.NODIO_EMERGENCY_REPLICA_FLOOR || 2)

package/src/server/models.js

@@ -9,7 +9,9 @@ const nodeSchema = new mongoose.Schema(
     capacityBytes: { type: Number, required: true, min: 1 },
     freeBytes: { type: Number, required: true, min: 0 },
     status: { type: String, enum: ['online', 'offline'], default: 'online', index: true },
-    lastHeartbeatAt: { type: Date, default: Date.now, index: true }
+    lastHeartbeatAt: { type: Date, default: Date.now, index: true },
+    pendingRelayAlert: { type: Boolean, default: false },
+    pendingRelayAlertAt: { type: Date, default: null }
   },
   { timestamps: true }
 );

package/src/server/routes.js

@@ -271,6 +271,23 @@ function buildRoutes(config) {
     }
   });
 
+  router.post('/nodes/:nodeId/alert-relay-pending', async (req, res, next) => {
+    try {
+      const { nodeId } = req.params;
+      const node = await NodeModel.findOne({ nodeId });
+      if (!node) {
+        return res.status(404).json({ error: 'node not found' });
+      }
+      // Set pending relay flag so donor knows to check urgently
+      node.pendingRelayAlert = true;
+      node.pendingRelayAlertAt = new Date();
+      await node.save();
+      res.json({ ok: true, message: 'relay pending alert sent' });
+    } catch (error) {
+      next(error);
+    }
+  });
+
   router.post('/replication-tasks/:taskId/complete', async (req, res, next) => {
     try {
       const { taskId } = req.params;

package/src/user/commands.js

@@ -31,7 +31,7 @@ function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 
-async function relayStoreShard(api, { shardId, fileId, nodeIds, dataBuffer, timeoutMs = 120000, pollMs = 1500 }) {
+async function relayStoreShard(api, { shardId, fileId, nodeIds, dataBuffer, timeoutMs = 90000, pollMs = 500 }) {
   const opId = uuidv4();
   await api.post('/relay/shards/store', {
     opId,
@@ -41,6 +41,16 @@ async function relayStoreShard(api, { shardId, fileId, nodeIds, dataBuffer, time
     dataBase64: dataBuffer.toString('base64')
   });
 
+  // Alert server that relay tasks are pending so donors check urgently
+  try {
+    for (const nodeId of nodeIds) {
+      // eslint-disable-next-line no-await-in-loop
+      await api.post(`/nodes/${nodeId}/alert-relay-pending`);
+    }
+  } catch (alertError) {
+    console.warn('relay alert failed:', alertError.message);
+  }
+
   const deadline = Date.now() + timeoutMs;
   while (Date.now() < deadline) {
     const response = await api.get(`/relay/shards/store/${opId}`);
@@ -51,7 +61,7 @@ async function relayStoreShard(api, { shardId, fileId, nodeIds, dataBuffer, time
         failed: payload.failed || []
       };
     }
-    // Wait for donor heartbeats to pull relay tasks.
+    // Polls every 500ms; with 10s heartbeat, donor should respond within 10s+network latency
     // eslint-disable-next-line no-await-in-loop
     await sleep(pollMs);
   }
@@ -59,7 +69,7 @@ async function relayStoreShard(api, { shardId, fileId, nodeIds, dataBuffer, time
   throw new Error(`relay store timed out for shard ${shardId}`);
 }
 
-async function relayFetchShard(api, { shardId, nodeIds, timeoutMs = 120000, pollMs = 1500 }) {
+async function relayFetchShard(api, { shardId, nodeIds, timeoutMs = 90000, pollMs = 500 }) {
   const opId = uuidv4();
   await api.post('/relay/shards/fetch', {
     opId,
@@ -67,6 +77,16 @@ async function relayFetchShard(api, { shardId, nodeIds, timeoutMs = 120000, poll
     nodeIds
   });
 
+  // Alert server that relay tasks are pending so donors check urgently
+  try {
+    for (const nodeId of nodeIds) {
+      // eslint-disable-next-line no-await-in-loop
+      await api.post(`/nodes/${nodeId}/alert-relay-pending`);
+    }
+  } catch (alertError) {
+    console.warn('relay alert failed:', alertError.message);
+  }
+
   const deadline = Date.now() + timeoutMs;
   while (Date.now() < deadline) {
     const response = await api.get(`/relay/shards/fetch/${opId}`);