chore: remove allocation from GetAttribute

riok · Jul 11, 2023 · 8ca1d91 · 8ca1d91
1 parent ae1ada4
commit 8ca1d91
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs b/src/Riok.Mapperly/Descriptors/SymbolAccessor.cs
@@ -22,8 +22,13 @@ internal IEnumerable<AttributeData> GetAttributes<T>(ISymbol symbol)
         where T : Attribute
     {
         var attributeSymbol = _types.Get<T>();
-        return GetAttributesCore(symbol)
-            .Where(x => SymbolEqualityComparer.Default.Equals(x.AttributeClass?.ConstructedFrom ?? x.AttributeClass, attributeSymbol));
+        foreach (var attr in GetAttributesCore(symbol))
+        {
+            if (SymbolEqualityComparer.Default.Equals(attr.AttributeClass?.ConstructedFrom ?? attr.AttributeClass, attributeSymbol))
+            {
+                yield return attr;
+            }
+        }
     }
 
     internal bool HasAttribute<T>(ISymbol symbol)