added missing harvesting

swiftlang · Aug 6, 2024 · 1bdf205 · 1bdf205
1 parent ca32bff
commit 1bdf205
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/Sources/SwiftSyntaxMacroExpansion/MacroSystem.swift b/Sources/SwiftSyntaxMacroExpansion/MacroSystem.swift
@@ -1310,6 +1310,11 @@ extension MacroApplication {
   /// }
   /// ```
   func expandCodeBlockItem(node: CodeBlockItemSyntax) -> MacroExpansionResult<CodeBlockItemListSyntax> {
+    let macroCount = potentiallyAcyclicMacros.count
+    defer {
+      harvestAcyclicMacros(oldCount: macroCount)
+    }
+
     return expandFreestandingMacro(node.item.asProtocol(FreestandingMacroExpansionSyntax.self)) { macro, node in
       return try expandFreestandingCodeItemList(
         definition: macro,
@@ -1329,6 +1334,11 @@ extension MacroApplication {
   /// }
   /// ```
   func expandMemberDecl(node: MemberBlockItemSyntax) -> MacroExpansionResult<MemberBlockItemListSyntax> {
+    let macroCount = potentiallyAcyclicMacros.count
+    defer {
+      harvestAcyclicMacros(oldCount: macroCount)
+    }
+
     return expandFreestandingMacro(node.decl.as(MacroExpansionDeclSyntax.self)) { macro, node in
       return try expandFreestandingMemberDeclList(
         definition: macro,
@@ -1347,6 +1357,11 @@ extension MacroApplication {
   /// let a = #foo
   /// ```
   func expandExpr(node: Syntax) -> MacroExpansionResult<ExprSyntax> {
+    let macroCount = potentiallyAcyclicMacros.count
+    defer {
+      harvestAcyclicMacros(oldCount: macroCount)
+    }
+
     return expandFreestandingMacro(node.as(MacroExpansionExprSyntax.self)) { macro, node in
       return try expandFreestandingExpr(
         definition: macro,