add: is_active to ServerInfor

niclabs · Sep 21, 2024 · 5c7c4ff · 5c7c4ff
1 parent 85d58af
commit 5c7c4ff
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/async_resolver/server_info.rs b/src/async_resolver/server_info.rs
@@ -22,6 +22,7 @@ pub struct ServerInfo {
     udp_connection: ClientUDPConnection,
     //TCP connection
     tcp_connection: ClientTCPConnection,
+    is_active: bool,
 }
 
 impl ServerInfo {
@@ -36,6 +37,7 @@ impl ServerInfo {
             algorithm,
             udp_connection,
             tcp_connection,
+            is_active: true,
         }
     }
 
@@ -51,6 +53,7 @@ impl ServerInfo {
             algorithm,
             udp_connection,
             tcp_connection,
+            is_active: true,
         }
     }
 
@@ -68,6 +71,7 @@ impl ServerInfo {
             algorithm,
             udp_connection,
             tcp_connection,
+            is_active: true,
         }
     }
 
@@ -154,6 +158,14 @@ impl ServerInfo {
         self.tcp_connection = tcp_connection;
     }
 
+    pub fn is_active(&self) -> bool {
+        self.is_active
+    }
+
+    pub fn set_active(&mut self, is_active: bool) {
+        self.is_active = is_active;
+    }
+
 }
 
 #[cfg(test)]