Iterate in the correct order, add error to attribute

CosmWasm · Apr 24, 2024 · 9ea997f · 9ea997f
1 parent 6f7ec50
commit 9ea997f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/derive/src/lib.rs b/packages/derive/src/lib.rs
@@ -120,17 +120,17 @@ pub fn entry_point(
 }
 
 fn expand_attributes(func: &mut ItemFn) -> syn::Result<TokenStream> {
-    let mut attributes = std::mem::take(&mut func.attrs);
+    let attributes = std::mem::take(&mut func.attrs);
     let mut stream = TokenStream::new();
-    while let Some(attribute) = attributes.pop() {
+    for attribute in attributes {
         if !attribute.path().is_ident("set_contract_state_version") {
             func.attrs.push(attribute);
             continue;
         }
 
         if func.sig.ident != "migrate" {
             return Err(syn::Error::new_spanned(
-                &func.sig.ident,
+                &attribute,
                 "you only want to add this macro to your migrate function",
             ));
         }