mirror of https://github.com/facebook/rocksdb.git
92 lines
2.4 KiB
C++
92 lines
2.4 KiB
C++
// 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/db_impl.h"
|
|
#include <string>
|
|
#include <stdint.h>
|
|
#include "db/version_set.h"
|
|
#include "leveldb/db.h"
|
|
#include "leveldb/env.h"
|
|
#include "port/port.h"
|
|
#include "util/mutexlock.h"
|
|
|
|
namespace leveldb {
|
|
|
|
void DBImpl::MaybeScheduleLogDBDeployStats() {
|
|
|
|
// There is a lock in the actual logger.
|
|
if (!logger_ || options_.db_stats_log_interval < 0
|
|
|| host_name_.empty()) {
|
|
return;
|
|
}
|
|
|
|
if(bg_logstats_scheduled_ || shutting_down_.Acquire_Load()) {
|
|
// Already scheduled
|
|
} else {
|
|
int64_t current_ts = 0;
|
|
Status st = env_->GetCurrentTime(¤t_ts);
|
|
if (!st.ok()) {
|
|
return;
|
|
}
|
|
if ((current_ts - last_log_ts) < options_.db_stats_log_interval) {
|
|
return;
|
|
}
|
|
last_log_ts = current_ts;
|
|
bg_logstats_scheduled_ = true;
|
|
env_->Schedule(&DBImpl::BGLogDBDeployStats, this);
|
|
}
|
|
}
|
|
|
|
void DBImpl::BGLogDBDeployStats(void* db) {
|
|
DBImpl* db_inst = reinterpret_cast<DBImpl*>(db);
|
|
db_inst->LogDBDeployStats();
|
|
}
|
|
|
|
void DBImpl::LogDBDeployStats() {
|
|
mutex_.Lock();
|
|
|
|
if (shutting_down_.Acquire_Load()) {
|
|
bg_logstats_scheduled_ = false;
|
|
bg_cv_.SignalAll();
|
|
mutex_.Unlock();
|
|
return;
|
|
}
|
|
|
|
mutex_.Unlock();
|
|
|
|
std::string version_info;
|
|
version_info += boost::lexical_cast<std::string>(kMajorVersion);
|
|
version_info += ".";
|
|
version_info += boost::lexical_cast<std::string>(kMinorVersion);
|
|
std::string data_dir;
|
|
env_->GetAbsolutePath(dbname_, &data_dir);
|
|
|
|
uint64_t file_total_size = 0;
|
|
uint32_t file_total_num = 0;
|
|
for (int i = 0; i < versions_->NumberLevels(); i++) {
|
|
file_total_num += versions_->NumLevelFiles(i);
|
|
file_total_size += versions_->NumLevelBytes(i);
|
|
}
|
|
|
|
VersionSet::LevelSummaryStorage scratch;
|
|
const char* file_num_summary = versions_->LevelSummary(&scratch);
|
|
std::string file_num_per_level(file_num_summary);
|
|
const char* file_size_summary = versions_->LevelDataSizeSummary(
|
|
&scratch);
|
|
std::string data_size_per_level(file_num_summary);
|
|
int64_t unix_ts;
|
|
env_->GetCurrentTime(&unix_ts);
|
|
|
|
logger_->Log_Deploy_Stats(version_info, host_name_,
|
|
data_dir, file_total_size, file_total_num, file_num_per_level,
|
|
data_size_per_level, unix_ts);
|
|
|
|
mutex_.Lock();
|
|
bg_logstats_scheduled_ = false;
|
|
bg_cv_.SignalAll();
|
|
mutex_.Unlock();
|
|
}
|
|
|
|
}
|