FEATURE: memory stats route for diagnostics in admin

app/controllers/admin/diagnostics_controller.rb

@@ -0,0 +1,14 @@
+class Admin::DiagnosticsController < Admin::AdminController
+  layout false
+  skip_before_filter :check_xhr
+
+  def memory_stats
+
+    stats = GC.stat.map{|k,v| "#{k}: #{v}"}
+    counts = ObjectSpace.count_objects.map{|k,v| "#{k}: #{v}"}
+
+    render text: "GC STATS:\n#{stats.join("\n")} \n\nObjects:\n#{counts.join("\n")}",
+      content_type: Mime::TEXT
+
+  end
+end

config/routes.rb

@@ -119,6 +119,8 @@ Discourse::Application.routes.draw do
         put "readonly" => "backups#readonly"
       end
     end
+    
+    get "memory_stats"=> "diagnostics#memory_stats", constraints: AdminConstraint.new
 
   end # admin namespace
 

script/bench.rb

@@ -7,6 +7,8 @@ require "optparse"
 @result_file = nil
 @iterations = 500
 @best_of = 1
+@mem_stats = false
+
 opts = OptionParser.new do |o|
   o.banner = "Usage: ruby bench.rb [options]"
 
@@ -22,6 +24,9 @@ opts = OptionParser.new do |o|
   o.on("-b", "--best_of [NUM]", "Number of times to run the bench taking best as result") do |i|
     @best_of = i.to_i
   end
+  o.on("-m", "--memory_stats") do
+    @mem_stats = true
+  end
 end
 opts.parse!
 
@@ -212,6 +217,11 @@ begin
 
   puts results.to_yaml
 
+  if @mem_stats
+    puts
+    puts open("http://127.0.0.1:#{@port}/admin/memory_stats#{append}").read
+  end
+
   if @result_file
     File.open(@result_file,"wb") do |f|
       f.write(results)