Add C profiling.

Rakefile

@@ -20,14 +20,18 @@ task :exceptions do
   end
 end
 
-task :valgrind do
- exec("ruby #{File.dirname(__FILE__)}/test/profile/valgrind.rb")
-end
-
 task :benchmark do
  exec("ruby #{File.dirname(__FILE__)}/test/profile/benchmark.rb")
 end
 
-task :profile do
- exec("ruby #{File.dirname(__FILE__)}/test/profile/profile.rb")
+task :rb_profile do
+ exec("ruby #{File.dirname(__FILE__)}/test/profile/rb_profiler.rb")
+end
+
+task :c_profile do
+ exec("ruby #{File.dirname(__FILE__)}/test/profile/c_profiler.rb")
+end
+
+task :valgrind do
+ exec("ruby #{File.dirname(__FILE__)}/test/profile/valgrind.rb")
 end

test/profile/c_profiler.rb

@@ -0,0 +1,14 @@
+require "#{File.dirname(__FILE__)}/../setup"
+
+$LOAD_PATH << "#{File.dirname(__FILE__)}/../../lib/"
+require 'memcached'
+
+profile = "/tmp/memcached_#{Memcached::VERSION}_c"
+
+system("env CPUPROFILE_FREQUENCY=500 CPUPROFILE=#{profile}.out DYLD_INSERT_LIBRARIES=/opt/local/lib/libprofiler.dylib ruby -r#{File.dirname(__FILE__)}/exercise -e \"Worker.new('mixed', 200000).work\"")
+
+ruby = `which ruby`.chomp
+
+system("pprof --nodefraction=0.0000001 --text #{ruby} #{profile}.out")
+system("pprof --nodefraction=0.0000001 --pdf #{ruby} #{profile}.out > #{profile}.pdf")
+system("open #{profile}.pdf")

test/profile/exercise.rb

@@ -8,24 +8,26 @@
 GC.copy_on_write_friendly = true if GC.respond_to?("copy_on_write_friendly=")
 
 class Worker
-  def initialize(method_name = 'mixed', iterations = 10000, ignore_memory = false)
-    @method = method_name
-    @i = iterations.to_i
-    @ignore_memory = ignore_memory
+  def initialize(method_name, iterations, with_memory = 'false')
+    @method = method_name || 'mixed'
+    @i = (iterations || 10000).to_i
+    @with_memory = with_memory
 
     puts "*** Running #{@method.inspect} test for #{@i} loops. ***"
 
-    @key1 = "key1-"*8
-    @key2 = "key2-"*8
+    @key1 = "key1--------------------------------"
+    @key2 = "key2--------------------------------"
+    @key3 = "key3--------------------------------"
 
     @value = []
     @marshalled = Marshal.dump(@value)
 
     @opts = [
-      ["#{UNIX_SOCKET_NAME}0", "#{UNIX_SOCKET_NAME}1"],
+      ['localhost:43042', 'localhost:43043', 'localhost:43044'],
       {
-        :buffer_requests => false,
-        :no_block => false,
+        :buffer_requests => true,
+        :no_block => true,
+        :noreply => true,
         :namespace => "namespace"
       }
     ]
@@ -81,7 +83,14 @@ def work
       when "mixed"
         @i.times do
           @cache.set @key1, @value
+          @cache.set @key2, @value
           @cache.get @key1
+          @cache.get @key3
+          @cache.get [@key1, @key2, @key3]
+          @cache.prepend @key1, @marshalled
+          @cache.prepend @key2, @marshalled
+          @cache.delete @key1
+          @cache.delete @key2
         end
       when "stats"
         @i.times do
@@ -111,7 +120,7 @@ def work
         raise "No such method"
     end
 
-    unless @ignore_memory
+    if @with_memory == "true"
       puts "*** Garbage collect. ***"
       10.times do
         GC.start

test/profile/rb_profiler.rb

@@ -0,0 +1,21 @@
+
+require "#{File.dirname(__FILE__)}/../setup"
+
+$LOAD_PATH << "#{File.dirname(__FILE__)}/../../lib/"
+
+ENV['CPUPROFILE_FREQUENCY'] = '500'
+require 'memcached'
+require 'rubygems'
+require 'perftools'
+require "#{HERE}/profile/exercise"
+
+profile = "/tmp/memcached_#{Memcached::VERSION}_rb"
+worker = Worker.new('mixed', 200000)
+
+PerfTools::CpuProfiler.start("#{profile}.out") do
+  worker.work
+end
+
+system("pprof.rb --nodefraction=0.0000001 --text #{profile}.out")
+system("pprof.rb --nodefraction=0.0000001 --edgefraction=0.0000001 --pdf #{profile}.out > #{profile}.pdf")
+system("open #{profile}.pdf")

test/profile/valgrind.rb

@@ -1 +1,3 @@
-exec("valgrind  --tool=memcheck --leak-check=full --show-reachable=no --num-callers=15 --track-fds=yes --workaround-gcc296-bugs=yes --max-stackframe=7304328 --dsymutil=yes --track-origins=yes ruby -r#{File.dirname(__FILE__)}/test/profile/exercise.rb -e \"Worker.new(ENV['METHOD'], ENV['LOOPS'], ENV['IGNORE_MEMORY']).work\"")
+require "#{File.dirname(__FILE__)}/../setup"
+
+exec("valgrind  --tool=memcheck --leak-check=full --show-reachable=no --num-callers=15 --track-fds=yes --workaround-gcc296-bugs=yes --max-stackframe=7304328 --dsymutil=yes --track-origins=yes ruby -r#{File.dirname(__FILE__)}/exercise.rb -e \"Worker.new('mixed', 10000, 'true').work\"")