fix memory leak in ColorOutput

oilshell · Sep 25, 2022 · 484c5a9 · 484c5a9
1 parent 113b293
commit 484c5a9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/asdl/format.py b/asdl/format.py
@@ -348,6 +348,7 @@ def _PrintRecord(self, node, f, indent):
             s, num_chars = single_f.GetRaw()  # extra unpacking for mycpp
             f.WriteRaw((s, num_chars))
           else:
+            cast(mylib.BufWriter, single_f.f).reset()
             f.write('\n')
             self.PrintNode(val, f, indent+INDENT+INDENT)
 
@@ -376,6 +377,8 @@ def PrintNode(self, node, f, indent):
       s, num_chars = single_f.GetRaw()  # extra unpacking for mycpp
       f.WriteRaw((s, num_chars))
       return
+    else:
+      cast(mylib.BufWriter, single_f.f).reset()
 
     UP_node = node  # for mycpp
     tag = node.tag_()

diff --git a/mycpp/mylib.pyi b/mycpp/mylib.pyi
@@ -30,6 +30,7 @@ class Writer:
 class BufWriter(Writer):
   def write(self, s: str) -> None: ...
   def getvalue(self) -> str: ...
+  def reset(self) -> None: ...
 
 def Stdout() -> Writer: ...