class Riak::Client
  include Instrumentable

  client_payload = {protocol: :protocol, client_id: :client_id}

  instrument_method :buckets, 'riak.list_buckets', client_payload
  instrument_method :list_buckets, 'riak.list_buckets', client_payload
  instrument_method :list_keys, 'riak.list_keys', client_payload
  instrument_method :set_bucket_props, 'riak.set_bucket_props', client_payload
  instrument_method :get_bucket_props, 'riak.get_bucket_props', client_payload
  instrument_method :clear_bucket_props, 'riak.clear_bucket_props', client_payload
  instrument_method :get_index, 'riak.get_index', client_payload
  instrument_method :store_object, 'riak.store_object', client_payload
  instrument_method :get_object, 'riak.get_object', client_payload
  instrument_method :reload_object, 'riak.reload_object', client_payload
  instrument_method :delete_object, 'riak.delete_object', client_payload
  instrument_method :store_file, 'riak.store_file', client_payload
  instrument_method :get_file, 'riak.get_file', client_payload
  instrument_method :delete_file, 'riak.delete_file', client_payload
  instrument_method :file_exists?, 'riak.file_exists', client_payload
  instrument_method :file_exist?, 'riak.file_exists', client_payload
  instrument_method :link_walk, 'riak.link_walk', client_payload
  instrument_method :mapred, 'riak.map_reduce', client_payload
  instrument_method :ping, 'riak.ping', client_payload
end