Stream respan without collecting Vec

dtolnay · Jun 4, 2020 · 379c4d4 · 379c4d4
1 parent 925f164
commit 379c4d4
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/respan.rs b/src/respan.rs
@@ -12,9 +12,11 @@ where
 }
 
 fn respan_tokens(tokens: TokenStream, span: Span) -> TokenStream {
-    let mut tokens = tokens.into_iter().collect::<Vec<_>>();
-    for token in tokens.iter_mut() {
-        token.set_span(span);
-    }
-    tokens.into_iter().collect()
+    tokens
+        .into_iter()
+        .map(|mut token| {
+            token.set_span(span);
+            token
+        })
+        .collect()
 }