update remove_endpoint

src/cluster.rs

@@ -108,14 +108,8 @@ impl ClusterMap {
         self.insert(None, endpoints);
     }
 
-    pub fn remove_endpoint(&self, endpoint: &Endpoint) -> bool {
-        for mut entry in self.0.iter_mut() {
-            if entry.value_mut().remove(endpoint) {
-                return true;
-            }
-        }
-
-        false
+    pub fn remove_endpoint(&self, needle: &Endpoint) -> bool {
+        self.remove_endpoint_if(|endpoint| endpoint.address == needle.address)
     }
 
     pub fn remove_endpoint_if(&self, closure: impl Fn(&Endpoint) -> bool) -> bool {

src/endpoint.rs

@@ -26,8 +26,7 @@ pub use self::{address::EndpointAddress, locality::Locality};
 pub type EndpointMetadata = crate::metadata::MetadataView<Metadata>;
 
 /// A destination endpoint with any associated metadata.
-/// For the purposes of comparison between endpoints, metadata is ignored.
-#[derive(Debug, Deserialize, Serialize, Clone, schemars::JsonSchema)]
+#[derive(Debug, Deserialize, Serialize, PartialEq, Clone, Eq, schemars::JsonSchema)]
 #[non_exhaustive]
 #[serde(deny_unknown_fields)]
 pub struct Endpoint {
@@ -37,14 +36,6 @@ pub struct Endpoint {
     pub metadata: EndpointMetadata,
 }
 
-impl PartialEq for Endpoint {
-    fn eq(&self, rhs: &Self) -> bool {
-        self.address == rhs.address
-    }
-}
-
-impl Eq for Endpoint {}
-
 impl Endpoint {
     /// Creates a new [`Endpoint`] with no metadata.
     pub fn new(address: EndpointAddress) -> Self {