ruby memory profile

shijiang1130

1. include ObjectSpace
   
2. 
   
3. MEMORY_PROFILE_BAD_SIZE_METHOD = {FileTest => true, File => true, File::Stat => true}
   
4. 
   
5. class Object
   
6.   def memory_profile_size_of_object(seen={})
   
7.     return 0 if seen.has_key? object_id
   
8.     seen[object_id] = true
   
9.     count = 1
   
10.     if kind_of? Hash
    
11.       each_pair do |key,value|
    
12.         count += key.memory_profile_size_of_object(seen)
    
13.         count += value.memory_profile_size_of_object(seen)
    
14.       end
    
15.     elsif kind_of? Array
    
16.       count += size
    
17.       each do |element|
    
18.         count += element.memory_profile_size_of_object(seen)
    
19.       end
    
20.     end
    
21. 
    
22.     count += instance_variables.size
    
23.     instance_variables.each do |var|
    
24.       count += instance_variable_get(var.to_sym).memory_profile_size_of_object(seen)
    
25.     end
    
26. 
    
27.     count
    
28.   end
    
29. 
    
30.   def memory_profile_inspect(seen={},level=0)
    
31.     return object_id.to_s if seen.has_key? object_id
    
32.     seen[object_id] = true
    
33.     result = ' '*level
    
34.     if kind_of? Hash
    
35.       result += "{\n" + ' '*level
    
36.       each_pair do |key,value|
    
37.         result += key.memory_profile_inspect(seen,level+1) + "=>\n"
    
38.         result += value.memory_profile_inspect(seen,level+2) + ",\n" + ' '*level
    
39.       end
    
40.       result += "}\n" + ' '*level
    
41.     elsif kind_of? Array
    
42.       result += "[\n" + ' '*level
    
43.       each do |element|
    
44.         result += element.memory_profile_inspect(seen,level+1) + ",\n" + ' '*level
    
45.       end
    
46.       result += "]\n" + ' '*level
    
47.     elsif kind_of? String
    
48.       result += self
    
49.     elsif kind_of? Numeric
    
50.       result += self.to_s
    
51.     elsif kind_of? Class
    
52.       result += to_s
    
53.     else
    
54.       result += "---"+self.class.to_s + "---\n" + ' '*level
    
55.     end
    
56. 
    
57. 
    
58.     instance_variables.each do |var|
    
59.       result += var + "=" + instance_variable_get(var.to_sym).memory_profile_inspect(seen,level+1) + "\n" + ' '*level
    
60.     end
    
61. 
    
62.     result
    
63.   end
    
64. 
    
65. end
    
66. 
    
67. module MemoryProfile
    
68.   LOG_FILE = "/tmp/memory_profile.log"
    
69. 
    
70.   def MemoryProfile::report
    
71.     Dir.chdir "/tmp"
    
72.     ObjectSpace::garbage_collect
    
73.     sleep 10 # Give the GC thread a chance
    
74.     all = []
    
75.     ObjectSpace.each_object do |obj|
    
76.       next if obj.object_id == all.object_id
    
77.         
    
78.       all << obj
    
79.     end
    
80.    
    
81.     tally = Hash.new(0)
    
82.     max_obj = nil
    
83.     max_count = 0
    
84.     all.each do |obj|
    
85.       count = obj.memory_profile_size_of_object
    
86.       if max_count < count
    
87.         max_obj = obj
    
88.         max_count = count
    
89.       end
    
90.       
    
91.       tally[obj.class]+=count
    
92.     end
    
93.    
    
94.     open( LOG_FILE, 'a') do |outf|
    
95.       outf.puts '+'*70
    
96.       tally.keys.sort{|a,b|
    
97.         if tally[a] == tally[b]
    
98.           a.to_s <=> b.to_s
    
99.         else
    
100.           -1*(tally[a]<=>tally[b])
     
101.         end
     
102.       }.each do |klass|
     
103.         outf.puts "#{klass}\t#{tally[klass]}"
     
104.       end
     
105.       
     
106.       outf.puts '-'*70
     
107.       outf.puts "Max obj was #{max_obj.class} at #{max_count}"
     
108.       outf.puts "Maximum object is..."
     
109.       outf.puts max_obj.memory_profile_inspect
     
110.     end
     
111.   end
     
112. 
     
113.   def MemoryProfile::simple_count
     
114.     Dir.chdir "/tmp"
     
115.     ObjectSpace::garbage_collect
     
116.     sleep 10 # Give the GC thread a chance
     
117. 
     
118.     tally = Hash.new(0)
     
119.     ObjectSpace.each_object do |obj|
     
120.       next if obj.object_id == tally.object_id
     
121.       tally[obj.class]+=1
     
122.     end
     
123.    
     
124.     open( LOG_FILE, 'a') do |outf|
     
125.       outf.puts '='*70
     
126.       outf.puts "MemoryProfile report for #{$0}"
     
127.       outf.puts `cat /proc/#{Process.pid}/status`
     
128.       
     
129.       tally.keys.sort{|a,b|
     
130.         if tally[a] == tally[b]
     
131.           a.to_s <=> b.to_s
     
132.         else
     
133.           -1*(tally[a]<=>tally[b])
     
134.         end
     
135.       }.each do |klass|
     
136.         outf.puts "#{klass}\t#{tally[klass]}"
     
137.       end
     
138.     end
     
139.   end
     
140. end
     
141. 
     
142. if $0 == __FILE__ then
     
143.   File.unlink MemoryProfile::LOG_FILE if FileTest.exist? MemoryProfile::LOG_FILE
     
144. 
     
145.   at_exit{ system("cat #{MemoryProfile::LOG_FILE}")}
     
146. end
     
147. 
     
148. at_exit{
     
149.   MemoryProfile::simple_count;
     
150.   MemoryProfile::report;
     
151. }
     

复制代码

shijiang1130

http://rubyforge.org/snippet/detail.php?type=snippet&id=70

shijiang1130

http://rubyforge.org/snippet/detail.php?type=snippet&id=70

pitonas

相当好的memory profile