rocksdb/util/perf_context.cc

//  Copyright (c) 2013, Facebook, Inc.  All rights reserved.
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
//

#include <sstream>
#include "util/perf_context_imp.h"

namespace rocksdb {

// by default, enable counts only
#if defined(IOS_CROSS_COMPILE)
PerfLevel perf_level = kEnableCount;
#else
__thread PerfLevel perf_level = kEnableCount;
#endif

void SetPerfLevel(PerfLevel level) { perf_level = level; }

void PerfContext::Reset() {
  user_key_comparison_count = 0;
  block_cache_hit_count = 0;
  block_read_count = 0;
  block_read_byte = 0;
  block_read_time = 0;
  block_checksum_time = 0;
  block_decompress_time = 0;
  internal_key_skipped_count = 0;
  internal_delete_skipped_count = 0;
  write_wal_time = 0;

  get_snapshot_time = 0;
  get_from_memtable_time = 0;
  get_from_memtable_count = 0;
  get_post_process_time = 0;
  get_from_output_files_time = 0;
  seek_child_seek_time = 0;
  seek_child_seek_count = 0;
  seek_min_heap_time = 0;
  seek_internal_seek_time = 0;
  find_next_user_entry_time = 0;
  write_pre_and_post_process_time = 0;
  write_memtable_time = 0;
}

#define OUTPUT(counter) #counter << " = " << counter << ", "

std::string PerfContext::ToString() const {
  std::ostringstream ss;
  ss << OUTPUT(user_key_comparison_count)
     << OUTPUT(block_cache_hit_count)
     << OUTPUT(block_read_count)
     << OUTPUT(block_read_byte)
     << OUTPUT(block_read_time)
     << OUTPUT(block_checksum_time)
     << OUTPUT(block_decompress_time)
     << OUTPUT(internal_key_skipped_count)
     << OUTPUT(internal_delete_skipped_count)
     << OUTPUT(write_wal_time)
     << OUTPUT(get_snapshot_time)
     << OUTPUT(get_from_memtable_time)
     << OUTPUT(get_from_memtable_count)
     << OUTPUT(get_post_process_time)
     << OUTPUT(get_from_output_files_time)
     << OUTPUT(seek_child_seek_time)
     << OUTPUT(seek_child_seek_count)
     << OUTPUT(seek_min_heap_time)
     << OUTPUT(seek_internal_seek_time)
     << OUTPUT(find_next_user_entry_time)
     << OUTPUT(write_pre_and_post_process_time)
     << OUTPUT(write_memtable_time);
  return ss.str();
}

#if defined(IOS_CROSS_COMPILE)
PerfContext perf_context;
#else
__thread PerfContext perf_context;
#endif
}