0e08452 node: Ensure we don't prune our own routes — heartwood

Radicle Heartwood Protocol & Stack

node: Ensure we don't prune our own routes

cloudhead committed 1 year ago

commit 0e08452323bd771d722502a3695fc3c29a157d69
parent 52554af4f12d9cb8b46793dbccbcf2451a5995da

2 files changed +21 -7

modified radicle-node/src/service.rs

@@ -2377,9 +2377,11 @@ where

         }
         let delta = count - self.config.limits.routing_max_size;
         let nid = self.node_id();
         self.db.routing_mut().prune(
             (*now - self.config.limits.routing_max_age).into(),
             Some(delta),
             &nid,
         )?;
         Ok(())
     }

modified radicle/src/node/routing.rs

@@ -64,7 +64,12 @@ pub trait Store {

     /// Get the total number of routing entries.
     fn len(&self) -> Result<usize, Error>;
     /// Prune entries older than the given timestamp.
     fn prune(&mut self, oldest: Timestamp, limit: Option<usize>) -> Result<usize, Error>;
     fn prune(
         &mut self,
         oldest: Timestamp,
         limit: Option<usize>,
         ignore: &NodeId,
     ) -> Result<usize, Error>;
     /// Count the number of routes for a specific repo RID.
     fn count(&self, id: &RepoId) -> Result<usize, Error>;
 }

@@ -210,18 +215,25 @@ impl Store for Database {

         Ok(count)
     }
     fn prune(&mut self, oldest: Timestamp, limit: Option<usize>) -> Result<usize, Error> {
     fn prune(
         &mut self,
         oldest: Timestamp,
         limit: Option<usize>,
         ignore: &NodeId,
     ) -> Result<usize, Error> {
         let limit: i64 = limit
             .unwrap_or(i64::MAX as usize)
             .try_into()
             .map_err(|_| Error::UnitOverflow)?;
         let mut stmt = self.db.prepare(
             "DELETE FROM routing WHERE rowid IN
             (SELECT rowid FROM routing WHERE timestamp < ? LIMIT ?)",
             "DELETE FROM routing
              WHERE node <> ?1 AND rowid IN
              (SELECT rowid FROM routing WHERE timestamp < ?2 ORDER BY timestamp LIMIT ?3)",
         )?;
         stmt.bind((1, &oldest))?;
         stmt.bind((2, limit))?;
         stmt.bind((1, ignore))?;
         stmt.bind((2, &oldest))?;
         stmt.bind((3, limit))?;
         stmt.next()?;
         Ok(self.db.change_count())

@@ -479,7 +491,7 @@ mod test {

                 .unwrap();
         }
         let pruned = db.prune(now.into(), None).unwrap();
         let pruned = db.prune(now.into(), None, &arbitrary::gen(1)).unwrap();
         assert_eq!(pruned, ids.len() * nodes.len());
         for id in &ids {