use impl Trait to avoid some allocations

1 files changed, 3 insertions, 5 deletions

diff --git a/src/power.rs b/src/power.rs
index 6bb90ca..6ae762c 100644
--- a/src/power.rs
+++ b/src/power.rs
@@ -99,21 +99,19 @@ impl PowerInfo {
     }
 
     pub fn has_batteries(&self) -> bool {
-        !self.batteries().is_empty()
+        self.batteries().count() > 0
     }
 
-    fn batteries(&self) -> Vec<&PowerSupplyInfo> {
+    fn batteries(&self) -> impl Iterator<Item=&PowerSupplyInfo> {
         self.power_supplies
             .iter()
             .filter(|p| p.ty == PowerSupplyType::Battery)
-            .collect()
     }
 
-    fn mains(&self) -> Vec<&PowerSupplyInfo> {
+    fn mains(&self) -> impl Iterator<Item=&PowerSupplyInfo> {
         self.power_supplies
             .iter()
             .filter(|p| p.ty == PowerSupplyType::AC)
-            .collect()
     }
 }