Merge pull request #448 from Caparow/scala2-feature-clean-thread-loca…

…ls-causing-oom Removing ThreadLocals on finish to clean up CompileTimeStack.
softwaremill · Jan 14, 2023 · b7ee4aa · b7ee4aa
2 parents 490d4bf + a4eb9bd
commit b7ee4aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/scala/magnolia1/magnolia.scala b/core/src/main/scala/magnolia1/magnolia.scala
@@ -903,8 +903,8 @@ private[magnolia1] object CompileTimeState {
       val depth = c.enclosingMacros.count(m => workSet(m.macroApplication.symbol))
       try fn(stack.asInstanceOf[Stack[c.type]], depth)
       finally if (depth <= 1) {
-        stack.clear()
-        workSet.clear()
+        threadLocalStack.remove()
+        threadLocalWorkSet.remove()
       }
     }
   }