2013-10-16 23:59:46 +02:00
|
|
|
// 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.
|
|
|
|
//
|
2011-03-18 23:37:00 +01:00
|
|
|
// 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.
|
|
|
|
|
2013-08-23 17:38:13 +02:00
|
|
|
#include "rocksdb/env.h"
|
|
|
|
#include "rocksdb/options.h"
|
2011-03-18 23:37:00 +01:00
|
|
|
|
2013-10-04 06:49:15 +02:00
|
|
|
namespace rocksdb {
|
2011-03-18 23:37:00 +01:00
|
|
|
|
|
|
|
Env::~Env() {
|
|
|
|
}
|
|
|
|
|
|
|
|
SequentialFile::~SequentialFile() {
|
|
|
|
}
|
|
|
|
|
|
|
|
RandomAccessFile::~RandomAccessFile() {
|
|
|
|
}
|
|
|
|
|
|
|
|
WritableFile::~WritableFile() {
|
|
|
|
}
|
|
|
|
|
2011-07-21 04:40:18 +02:00
|
|
|
Logger::~Logger() {
|
|
|
|
}
|
|
|
|
|
2011-03-18 23:37:00 +01:00
|
|
|
FileLock::~FileLock() {
|
|
|
|
}
|
|
|
|
|
2013-11-07 20:31:56 +01:00
|
|
|
void LogFlush(Logger *info_log) {
|
|
|
|
if (info_log) {
|
|
|
|
info_log->Flush();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-07-21 04:40:18 +02:00
|
|
|
void Log(Logger* info_log, const char* format, ...) {
|
2013-01-20 11:07:13 +01:00
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-26 23:41:28 +01:00
|
|
|
void Log(const InfoLogLevel log_level, Logger* info_log, const char* format,
|
|
|
|
...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(log_level, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Debug(Logger* info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::DEBUG, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Info(Logger* info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::INFO, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Warn(Logger* info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::WARN, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
void Error(Logger* info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::ERROR, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
void Fatal(Logger* info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::FATAL, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-11-07 20:31:56 +01:00
|
|
|
void LogFlush(const shared_ptr<Logger>& info_log) {
|
|
|
|
if (info_log) {
|
|
|
|
info_log->Flush();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-02-26 23:41:28 +01:00
|
|
|
void Log(const InfoLogLevel log_level, const shared_ptr<Logger>& info_log,
|
|
|
|
const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(log_level, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Debug(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::DEBUG, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Info(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::INFO, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Warn(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::WARN, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Error(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::ERROR, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Fatal(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(InfoLogLevel::FATAL, format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-01-20 11:07:13 +01:00
|
|
|
void Log(const shared_ptr<Logger>& info_log, const char* format, ...) {
|
|
|
|
if (info_log) {
|
2011-07-21 04:40:18 +02:00
|
|
|
va_list ap;
|
|
|
|
va_start(ap, format);
|
|
|
|
info_log->Logv(format, ap);
|
|
|
|
va_end(ap);
|
|
|
|
}
|
2011-03-18 23:37:00 +01:00
|
|
|
}
|
|
|
|
|
2012-01-25 23:56:52 +01:00
|
|
|
static Status DoWriteStringToFile(Env* env, const Slice& data,
|
|
|
|
const std::string& fname,
|
|
|
|
bool should_sync) {
|
2013-01-20 11:07:13 +01:00
|
|
|
unique_ptr<WritableFile> file;
|
2013-06-08 00:35:17 +02:00
|
|
|
EnvOptions soptions;
|
2013-03-15 01:00:04 +01:00
|
|
|
Status s = env->NewWritableFile(fname, &file, soptions);
|
2011-03-18 23:37:00 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
s = file->Append(data);
|
2012-01-25 23:56:52 +01:00
|
|
|
if (s.ok() && should_sync) {
|
|
|
|
s = file->Sync();
|
|
|
|
}
|
2011-03-18 23:37:00 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
env->DeleteFile(fname);
|
|
|
|
}
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2012-01-25 23:56:52 +01:00
|
|
|
Status WriteStringToFile(Env* env, const Slice& data,
|
|
|
|
const std::string& fname) {
|
|
|
|
return DoWriteStringToFile(env, data, fname, false);
|
|
|
|
}
|
|
|
|
|
|
|
|
Status WriteStringToFileSync(Env* env, const Slice& data,
|
|
|
|
const std::string& fname) {
|
|
|
|
return DoWriteStringToFile(env, data, fname, true);
|
|
|
|
}
|
|
|
|
|
2011-03-18 23:37:00 +01:00
|
|
|
Status ReadFileToString(Env* env, const std::string& fname, std::string* data) {
|
2013-06-08 00:35:17 +02:00
|
|
|
EnvOptions soptions;
|
2011-03-18 23:37:00 +01:00
|
|
|
data->clear();
|
2013-01-20 11:07:13 +01:00
|
|
|
unique_ptr<SequentialFile> file;
|
2013-03-15 01:00:04 +01:00
|
|
|
Status s = env->NewSequentialFile(fname, &file, soptions);
|
2011-03-18 23:37:00 +01:00
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
static const int kBufferSize = 8192;
|
|
|
|
char* space = new char[kBufferSize];
|
|
|
|
while (true) {
|
|
|
|
Slice fragment;
|
|
|
|
s = file->Read(kBufferSize, &fragment, space);
|
|
|
|
if (!s.ok()) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
data->append(fragment.data(), fragment.size());
|
|
|
|
if (fragment.empty()) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
delete[] space;
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
EnvWrapper::~EnvWrapper() {
|
|
|
|
}
|
|
|
|
|
2013-06-08 00:35:17 +02:00
|
|
|
namespace { // anonymous namespace
|
|
|
|
|
|
|
|
void AssignEnvOptions(EnvOptions* env_options, const Options& options) {
|
|
|
|
env_options->use_os_buffer = options.allow_os_buffer;
|
|
|
|
env_options->use_mmap_reads = options.allow_mmap_reads;
|
|
|
|
env_options->use_mmap_writes = options.allow_mmap_writes;
|
|
|
|
env_options->set_fd_cloexec = options.is_fd_close_on_exec;
|
2013-06-14 07:49:46 +02:00
|
|
|
env_options->bytes_per_sync = options.bytes_per_sync;
|
2013-06-08 00:35:17 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2014-02-28 22:19:47 +01:00
|
|
|
EnvOptions EnvOptions::AdaptForLogWrite() const {
|
|
|
|
EnvOptions adapted = *this;
|
|
|
|
adapted.use_mmap_writes = false;
|
|
|
|
return adapted;
|
|
|
|
}
|
|
|
|
|
2013-06-08 00:35:17 +02:00
|
|
|
EnvOptions::EnvOptions(const Options& options) {
|
|
|
|
AssignEnvOptions(this, options);
|
|
|
|
}
|
|
|
|
|
|
|
|
EnvOptions::EnvOptions() {
|
|
|
|
Options options;
|
|
|
|
AssignEnvOptions(this, options);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-10-04 06:49:15 +02:00
|
|
|
} // namespace rocksdb
|