Merge pull request #90 from casperisfine/dump-fork

Dump heap from a fork
tmm1 · Feb 1, 2023 · aa3e716 · aa3e716
2 parents 326800f + 3e03fd3
commit aa3e716
Showing 1 changed file with 26 additions and 2 deletions.
diff --git a/lib/rbtrace/cli.rb b/lib/rbtrace/cli.rb
@@ -503,7 +503,19 @@ def self.run
           temp.unlink
         end
 
-        tracer.eval("file = File.open('#{filename}', 'w'); ObjectSpace.dump_all(output: file); file.close")
+        tracer.eval(<<-RUBY)
+          Thread.new do
+            Thread.current.name = '__RBTrace__'
+            pid = ::Process.fork do
+              file = File.open('#{filename}.tmp', 'w')
+              ObjectSpace.dump_all(output: file)
+              file.close
+              File.rename('#{filename}.tmp', '#{filename}')
+              exit!(0)
+            end
+            Process.waitpid(pid)
+          end
+        RUBY
         puts "Heapdump being written to #{filename}"
 
       elsif opts[:shapesdump_given]
@@ -517,7 +529,19 @@ def self.run
           temp.unlink
         end
 
-        tracer.eval("file = File.open('#{filename}', 'w'); ObjectSpace.dump_shapes(output: file); file.close")
+        tracer.eval(<<-RUBY)
+          Thread.new do
+            Thread.current.name = '__RBTrace__'
+            pid = ::Process.fork do
+              file = File.open('#{filename}.tmp', 'w')
+              ObjectSpace.dump_shapes(output: file)
+              file.close
+              File.rename('#{filename}.tmp', '#{filename}')
+              exit!(0)
+            end
+            Process.waitpid(pid)
+          end
+        RUBY
         puts "Shapes dump being written to #{filename}"
 
       elsif opts[:eval_given]