Let try_get_attribute() use either &str or &[u8]

tafia · Mar 27, 2021 · b24cb9a · b24cb9a
1 parent 7266bbc
commit b24cb9a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/events/mod.rs b/src/events/mod.rs
@@ -315,10 +315,13 @@ impl<'a> BytesStart<'a> {
     }
 
     /// Try to get an attribute
-    pub fn try_get_attribute(&'a self, attr_name: &[u8]) -> Result<Option<Attribute<'a>>> {
+    pub fn try_get_attribute<N: AsRef<[u8]> + Sized>(
+        &'a self,
+        attr_name: N,
+    ) -> Result<Option<Attribute<'a>>> {
         for a in self.attributes() {
             let a = a?;
-            if a.key == attr_name {
+            if a.key == attr_name.as_ref() {
                 return Ok(Some(a));
             }
         }