isolate protobuf-only and http-only tracers

nuclearsandwich · Oct 21, 2011 · 36f2d15 · 36f2d15
1 parent f781947
commit 36f2d15
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/lib/rpm_contrib/instrumentation/riak_client.rb b/lib/rpm_contrib/instrumentation/riak_client.rb
@@ -12,21 +12,12 @@
   executes do
     backend_tracers = proc do
       add_method_tracer :ping, 'Database/Riak/ping'
-      add_method_tracer :stats, 'Database/Riak/stats'
-      add_method_tracer :server_info, 'Database/Riak/server_info'
-
-      add_method_tracer :get_client_id, 'Database/Riak/get_client_id'
-      add_method_tracer :set_client_id, 'Database/Riak/set_client_id'
 
       add_method_tracer :list_buckets, 'Database/Riak/list_buckets'
       add_method_tracer :get_bucket_props, 'Database/Riak/get_bucket_props'
       add_method_tracer :set_bucket_props, 'Database/Riak/set_bucket_props'
 
       add_method_tracer :mapred, 'Database/Riak/mapred'
-      add_method_tracer :link_walk, 'Database/Riak/link_walk'
-      add_method_tracer :get_index, 'Database/Riak/get_index'
-      add_method_tracer :search, 'Database/Riak/search'
-      add_method_tracer :update_search_index, 'Database/riak/update_search_index'
 
       add_method_tracer :list_keys, 'Database/Riak/list_keys'
       add_method_tracer :fetch_object, 'Database/Riak/fetch_object'
@@ -35,8 +26,19 @@
       add_method_tracer :delete_object, 'Database/Riak/delete_object'
     end
 
-    ::Riak::Client::ProtobuffsBackend.class_eval &backend_tracers
     ::Riak::Client::BeefcakeProtobuffsBackend.class_eval &backend_tracers
+    ::Riak::Client::BeefcakeProtobuffsBackend.class_eval do
+      add_method_tracer :server_info, 'Database/Riak/server_info'
+      add_method_tracer :get_client_id, 'Database/Riak/get_client_id'
+      add_method_tracer :set_client_id, 'Database/Riak/set_client_id'
+    end
     ::Riak::Client::HTTPBackend.class_eval &backend_tracers
+    ::Riak::Client::HTTPBackend.class_eval do
+      add_method_tracer :stats, 'Database/Riak/stats'
+      add_method_tracer :link_walk, 'Database/Riak/link_walk'
+      add_method_tracer :get_index, 'Database/Riak/get_index'
+      add_method_tracer :search, 'Database/Riak/search'
+      add_method_tracer :update_search_index, 'Database/Riak/update_search_index'
+    end
   end
 end