// Copyright (c) 2011-present, Facebook, Inc. All rights reserved. // This source code is licensed under both the GPLv2 (found in the // COPYING file in the root directory) and Apache 2.0 License // (found in the LICENSE.Apache file in the root directory). // // Copyright (c) 2011 The LevelDB Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. See the AUTHORS file for names of contributors. #include "db/malloc_stats.h" #ifndef ROCKSDB_LITE #include #include namespace rocksdb { #ifdef ROCKSDB_JEMALLOC #ifdef __FreeBSD__ #include #else #include "jemalloc/jemalloc.h" #ifdef JEMALLOC_NO_RENAME #define malloc_stats_print je_malloc_stats_print #endif #endif typedef struct { char* cur; char* end; } MallocStatus; static void GetJemallocStatus(void* mstat_arg, const char* status) { MallocStatus* mstat = reinterpret_cast(mstat_arg); size_t status_len = status ? strlen(status) : 0; size_t buf_size = (size_t)(mstat->end - mstat->cur); if (!status_len || status_len > buf_size) { return; } snprintf(mstat->cur, buf_size, "%s", status); mstat->cur += status_len; } #endif // ROCKSDB_JEMALLOC #ifdef ROCKSDB_JEMALLOC void DumpMallocStats(std::string* stats) { MallocStatus mstat; const unsigned int kMallocStatusLen = 1000000; std::unique_ptr buf{new char[kMallocStatusLen + 1]}; mstat.cur = buf.get(); mstat.end = buf.get() + kMallocStatusLen; malloc_stats_print(GetJemallocStatus, &mstat, ""); stats->append(buf.get()); } #else void DumpMallocStats(std::string*) {} #endif // ROCKSDB_JEMALLOC } #endif // !ROCKSDB_LITE