From cdd4612ceebda20f6abb135ddb8c4ea00888d1c3 Mon Sep 17 00:00:00 2001
From: Daniel Sudzilouski <dsudzilouski@olin.edu>
Date: Sun, 15 Dec 2024 22:39:05 -0500
Subject: [PATCH] re-announce bug with timestamp refreshing (#51)

* re-announce bug with timestamp refreshing

* Update info.rs
---
 src/info.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/info.rs b/src/info.rs
index 392926c..a7b406c 100644
--- a/src/info.rs
+++ b/src/info.rs
@@ -72,9 +72,10 @@ impl RouterHandler for InfoRouter {
         for block in reader_writers.iter_mut().enumerate() {
             if let Some(writer) = &mut block.1.writer {
                 if writer.announce_id == req.shard_id {
-                    // update ip/port
+                    // update ip/port and refresh timestamp
                     writer.ip = req.ip;
                     writer.port = req.port;
+                    writer.timestamp = SystemTime::now();
                     // already announced
                     return AnnounceShardResponse {
                         writer_number: block.0 as u16,
@@ -83,9 +84,10 @@ impl RouterHandler for InfoRouter {
             }
             for reader in block.1.readers.iter_mut() {
                 if reader.announce_id == req.shard_id {
-                    // update ip/port
+                    // update ip/port and refresh timestamp
                     reader.ip = req.ip;
                     reader.port = req.port;
+                    reader.timestamp = SystemTime::now();
                     // already announced
                     return AnnounceShardResponse {
                         writer_number: block.0 as u16,