db_impl.cc 81.0 KB
Newer Older
J
jorlow@chromium.org 已提交
1 2 3 4 5 6 7
// 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 <algorithm>
8 9
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
10 11 12 13
#include <set>
#include <string>
#include <stdint.h>
#include <vector>
14

J
jorlow@chromium.org 已提交
15 16 17 18 19 20 21
#include "db/builder.h"
#include "db/db_iter.h"
#include "db/dbformat.h"
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
22
#include "db/memtablelist.h"
J
jorlow@chromium.org 已提交
23 24 25
#include "db/table_cache.h"
#include "db/version_set.h"
#include "db/write_batch_internal.h"
26
#include "db/transaction_log_iterator_impl.h"
27 28
#include "leveldb/db.h"
#include "leveldb/env.h"
29
#include "leveldb/statistics.h"
30 31 32
#include "leveldb/status.h"
#include "leveldb/table.h"
#include "leveldb/table_builder.h"
J
jorlow@chromium.org 已提交
33 34 35 36
#include "port/port.h"
#include "table/block.h"
#include "table/merger.h"
#include "table/two_level_iterator.h"
37 38
#include "util/auto_roll_logger.h"
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
39 40 41
#include "util/coding.h"
#include "util/logging.h"
#include "util/mutexlock.h"
42
#include "util/stop_watch.h"
J
jorlow@chromium.org 已提交
43 44 45

namespace leveldb {

46 47
void dumpLeveldbBuildVersion(Logger * log);

48 49 50 51 52
// Information kept for every waiting writer
struct DBImpl::Writer {
  Status status;
  WriteBatch* batch;
  bool sync;
H
heyongqiang 已提交
53
  bool disableWAL;
54 55 56 57 58 59
  bool done;
  port::CondVar cv;

  explicit Writer(port::Mutex* mu) : cv(mu) { }
};

J
jorlow@chromium.org 已提交
60 61 62
struct DBImpl::CompactionState {
  Compaction* const compaction;

63 64 65 66 67
  // If there were two snapshots with seq numbers s1 and
  // s2 and s1 < s2, and if we find two instances of a key k1 then lies
  // entirely within s1 and s2, then the earlier version of k1 can be safely
  // deleted because that version is not visible in any snapshot.
  std::vector<SequenceNumber> existing_snapshots;
J
jorlow@chromium.org 已提交
68 69 70 71 72 73 74 75

  // Files produced by compaction
  struct Output {
    uint64_t number;
    uint64_t file_size;
    InternalKey smallest, largest;
  };
  std::vector<Output> outputs;
76
  std::list<uint64_t> allocated_file_numbers;
J
jorlow@chromium.org 已提交
77 78

  // State kept for output being generated
79 80
  unique_ptr<WritableFile> outfile;
  unique_ptr<TableBuilder> builder;
J
jorlow@chromium.org 已提交
81 82 83 84 85 86 87 88 89 90 91

  uint64_t total_bytes;

  Output* current_output() { return &outputs[outputs.size()-1]; }

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
};

D
Dhruba Borthakur 已提交
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
struct DBImpl::DeletionState {

  // the set of all live files that cannot be deleted
  std::set<uint64_t> live;

  // a list of all siles that exists in the db directory
  std::vector<std::string> allfiles;

  // the current filenumber, lognumber and prevlognumber
  // that corresponds to the set of files in 'live'.
  uint64_t filenumber, lognumber, prevlognumber;

  // the list of all files to be evicted from the table cahce
  std::vector<uint64_t> files_to_evict;
};

J
jorlow@chromium.org 已提交
108 109 110
// Fix user-supplied options to be reasonable
template <class T,class V>
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
D
dgrogan@chromium.org 已提交
111 112
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
J
jorlow@chromium.org 已提交
113 114 115
}
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
116
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
117 118 119
                        const Options& src) {
  Options result = src;
  result.comparator = icmp;
120
  result.filter_policy = (src.filter_policy != nullptr) ? ipolicy : nullptr;
121
  ClipToRange(&result.max_open_files,            20,     1000000);
S
Sanjay Ghemawat 已提交
122 123
  ClipToRange(&result.write_buffer_size,         64<<10, 1<<30);
  ClipToRange(&result.block_size,                1<<10,  4<<20);
124
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
125 126
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
127 128
    if (!s.ok()) {
      // No place suitable for logging
129
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
130 131
    }
  }
132
  if (result.block_cache == nullptr && !result.no_block_cache) {
133 134
    result.block_cache = NewLRUCache(8 << 20);
  }
135
  result.compression_per_level = src.compression_per_level;
J
jorlow@chromium.org 已提交
136 137 138 139 140
  return result;
}

DBImpl::DBImpl(const Options& options, const std::string& dbname)
    : env_(options.env),
H
heyongqiang 已提交
141
      dbname_(dbname),
J
jorlow@chromium.org 已提交
142
      internal_comparator_(options.comparator),
S
Sanjay Ghemawat 已提交
143 144
      options_(SanitizeOptions(
          dbname, &internal_comparator_, &internal_filter_policy_, options)),
H
heyongqiang 已提交
145
      internal_filter_policy_(options.filter_policy),
J
jorlow@chromium.org 已提交
146
      owns_info_log_(options_.info_log != options.info_log),
147 148
      db_lock_(nullptr),
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
149
      bg_cv_(&mutex_),
150
      mem_(new MemTable(internal_comparator_, NumberLevels())),
151
      logfile_number_(0),
152
      tmp_batch_(),
153
      bg_compaction_scheduled_(0),
154
      bg_logstats_scheduled_(false),
155 156
      manual_compaction_(nullptr),
      logger_(nullptr),
157
      disable_delete_obsolete_files_(false),
158
      delete_obsolete_files_last_run_(0),
M
Mark Callaghan 已提交
159 160 161
      stall_level0_slowdown_(0),
      stall_memtable_compaction_(0),
      stall_level0_num_files_(0),
162
      started_at_(options.env->NowMicros()),
163
      flush_on_destroy_(false),
164
      stats_(options.num_levels),
165
      delayed_writes_(0),
166 167
      last_flushed_sequence_(0),
      storage_options_(options) {
168

169
  mem_->Ref();
170

H
heyongqiang 已提交
171
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
172 173 174 175 176

  stall_leveln_slowdown_.resize(options.num_levels);
  for (int i = 0; i < options.num_levels; ++i)
    stall_leveln_slowdown_[i] = 0;

J
jorlow@chromium.org 已提交
177
  // Reserve ten files or so for other uses and give the rest to TableCache.
178
  const int table_cache_size = options_.max_open_files - 10;
179 180
  table_cache_.reset(new TableCache(dbname_, &options_,
                                    storage_options_, table_cache_size));
J
jorlow@chromium.org 已提交
181

182 183
  versions_.reset(new VersionSet(dbname_, &options_, storage_options_,
                                 table_cache_.get(), &internal_comparator_));
184

185 186
  dumpLeveldbBuildVersion(options_.info_log.get());
  options_.Dump(options_.info_log.get());
187

188
#ifdef USE_SCRIBE
189
  logger_.reset(new ScribeLogger("localhost", 1456));
190 191 192
#endif

  char name[100];
193
  Status st = env_->GetHostName(name, 100L);
194 195 196 197 198 199 200
  if(st.ok()) {
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
201

J
jorlow@chromium.org 已提交
202 203 204 205
}

DBImpl::~DBImpl() {
  // Wait for background work to finish
206 207 208
  if (flush_on_destroy_) {
    FlushMemTable(FlushOptions());
  }
209
  mutex_.Lock();
210
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
211
  while (bg_compaction_scheduled_ || bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
212
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
213 214 215
  }
  mutex_.Unlock();

216
  if (db_lock_ != nullptr) {
J
jorlow@chromium.org 已提交
217 218 219
    env_->UnlockFile(db_lock_);
  }

220
  if (mem_ != nullptr) mem_->Unref();
221
  imm_.UnrefAll();
J
jorlow@chromium.org 已提交
222 223
}

A
Abhishek Kona 已提交
224
// Do not flush and close database elegantly. Simulate a crash.
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240
void DBImpl::TEST_Destroy_DBImpl() {
  // ensure that no new memtable flushes can occur
  flush_on_destroy_ = false;

  // wait till all background compactions are done.
  mutex_.Lock();
  while (bg_compaction_scheduled_ || bg_logstats_scheduled_) {
    bg_cv_.Wait();
  }

  // Prevent new compactions from occuring.
  const int LargeNumber = 10000000;
  bg_compaction_scheduled_ += LargeNumber;
  mutex_.Unlock();

  // force release the lock file.
241
  if (db_lock_ != nullptr) {
242 243
    env_->UnlockFile(db_lock_);
  }
244 245 246 247

  log_.reset();
  versions_.reset();
  table_cache_.reset();
248 249
}

A
Abhishek Kona 已提交
250 251 252
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
253

J
jorlow@chromium.org 已提交
254
Status DBImpl::NewDB() {
255
  VersionEdit new_db(NumberLevels());
J
jorlow@chromium.org 已提交
256
  new_db.SetComparatorName(user_comparator()->Name());
257
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
258 259 260 261
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
262
  unique_ptr<WritableFile> file;
263
  Status s = env_->NewWritableFile(manifest, &file, storage_options_);
J
jorlow@chromium.org 已提交
264 265 266
  if (!s.ok()) {
    return s;
  }
267
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
268
  {
269
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
    std::string record;
    new_db.EncodeTo(&record);
    s = log.AddRecord(record);
  }
  if (s.ok()) {
    // Make "CURRENT" file that points to the new manifest file.
    s = SetCurrentFile(env_, dbname_, 1);
  } else {
    env_->DeleteFile(manifest);
  }
  return s;
}

void DBImpl::MaybeIgnoreError(Status* s) const {
  if (s->ok() || options_.paranoid_checks) {
    // No change needed
  } else {
287
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
288 289 290 291
    *s = Status::OK();
  }
}

292 293
const Status DBImpl::CreateArchivalDirectory() {
  if (options_.WAL_ttl_seconds > 0) {
294
    std::string archivalPath = ArchivalDirectory(dbname_);
295 296 297 298 299
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

D
Dhruba Borthakur 已提交
300 301 302 303 304
// Returns the list of live files in 'live' and the list
// of all files in the filesystem in 'allfiles'.
void DBImpl::FindObsoleteFiles(DeletionState& deletion_state) {
  mutex_.AssertHeld();

305 306 307 308 309
  // if deletion is disabled, do nothing
  if (disable_delete_obsolete_files_) {
    return;
  }

310 311 312 313 314
  // This method is costly when the number of files is large.
  // Do not allow it to trigger more often than once in
  // delete_obsolete_files_period_micros.
  if (options_.delete_obsolete_files_period_micros != 0) {
    const uint64_t now_micros = env_->NowMicros();
315
    if (delete_obsolete_files_last_run_ +
316 317 318 319 320 321
        options_.delete_obsolete_files_period_micros > now_micros) {
      return;
    }
    delete_obsolete_files_last_run_ = now_micros;
  }

J
jorlow@chromium.org 已提交
322
  // Make a set of all of the live files
D
Dhruba Borthakur 已提交
323 324 325 326 327
  deletion_state.live = pending_outputs_;
  versions_->AddLiveFiles(&deletion_state.live);

  // set of all files in the directory
  env_->GetChildren(dbname_, &deletion_state.allfiles); // Ignore errors
J
jorlow@chromium.org 已提交
328

D
Dhruba Borthakur 已提交
329 330 331 332 333 334 335 336 337 338 339
  // store the current filenum, lognum, etc
  deletion_state.filenumber = versions_->ManifestFileNumber();
  deletion_state.lognumber = versions_->LogNumber();
  deletion_state.prevlognumber = versions_->PrevLogNumber();
}

// Diffs the files listed in filenames and those that do not
// belong to live files are posibly removed. If the removed file
// is a sst file, then it returns the file number in files_to_evict.
// It is not necesary to hold the mutex when invoking this method.
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
J
jorlow@chromium.org 已提交
340 341
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
342
  std::vector<std::string> old_log_files;
343

D
Dhruba Borthakur 已提交
344 345
  for (size_t i = 0; i < state.allfiles.size(); i++) {
    if (ParseFileName(state.allfiles[i], &number, &type)) {
J
jorlow@chromium.org 已提交
346 347 348
      bool keep = true;
      switch (type) {
        case kLogFile:
D
Dhruba Borthakur 已提交
349 350
          keep = ((number >= state.lognumber) ||
                  (number == state.prevlognumber));
J
jorlow@chromium.org 已提交
351 352 353 354
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
D
Dhruba Borthakur 已提交
355
          keep = (number >= state.filenumber);
J
jorlow@chromium.org 已提交
356 357
          break;
        case kTableFile:
D
Dhruba Borthakur 已提交
358
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
359 360 361 362
          break;
        case kTempFile:
          // Any temp files that are currently being written to must
          // be recorded in pending_outputs_, which is inserted into "live"
D
Dhruba Borthakur 已提交
363
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
364
          break;
H
heyongqiang 已提交
365 366 367
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
D
Dhruba Borthakur 已提交
368
            old_log_files.push_back(state.allfiles[i]);
H
heyongqiang 已提交
369 370
          }
          break;
J
jorlow@chromium.org 已提交
371 372
        case kCurrentFile:
        case kDBLockFile:
K
Kosie van der Merwe 已提交
373
        case kMetaDatabase:
J
jorlow@chromium.org 已提交
374 375 376 377 378 379
          keep = true;
          break;
      }

      if (!keep) {
        if (type == kTableFile) {
D
Dhruba Borthakur 已提交
380 381
          // record the files to be evicted from the cache
          state.files_to_evict.push_back(number);
J
jorlow@chromium.org 已提交
382
        }
383
        Log(options_.info_log, "Delete type=%d #%lld\n",
J
jorlow@chromium.org 已提交
384 385
            int(type),
            static_cast<unsigned long long>(number));
386
        if (type == kLogFile && options_.WAL_ttl_seconds > 0) {
387 388
          Status st = env_->RenameFile(LogFileName(dbname_, number),
                                       ArchivedLogFileName(dbname_, number));
389 390
          if (!st.ok()) {
            Log(options_.info_log, "RenameFile type=%d #%lld FAILED\n",
H
heyongqiang 已提交
391 392
              int(type),
              static_cast<unsigned long long>(number));
393 394
          }
        } else {
395
          Status st = env_->DeleteFile(dbname_ + "/" + state.allfiles[i]);
396 397 398 399 400
          if(!st.ok()) {
            Log(options_.info_log, "Delete type=%d #%lld FAILED\n",
                int(type),
                static_cast<unsigned long long>(number));
          }
H
heyongqiang 已提交
401
        }
J
jorlow@chromium.org 已提交
402 403 404
      }
    }
  }
H
heyongqiang 已提交
405 406

  // Delete old log files.
K
Kai Liu 已提交
407 408 409 410 411
  size_t old_log_file_count = old_log_files.size();
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
  if (old_log_file_count >= options_.keep_log_file_num &&
      options_.db_log_dir.empty()) {
H
heyongqiang 已提交
412
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
413
    size_t end = old_log_file_count - options_.keep_log_file_num;
414
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
415
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
416 417
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
418 419 420
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
J
jorlow@chromium.org 已提交
421 422
}

D
Dhruba Borthakur 已提交
423 424 425 426 427 428 429 430 431 432 433 434
void DBImpl::EvictObsoleteFiles(DeletionState& state) {
  for (unsigned int i = 0; i < state.files_to_evict.size(); i++) {
    table_cache_->Evict(state.files_to_evict[i]);
  }
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
  FindObsoleteFiles(deletion_state);
  PurgeObsoleteFiles(deletion_state);
  EvictObsoleteFiles(deletion_state);
435 436 437 438 439 440
  PurgeObsoleteWALFiles();
}

void DBImpl::PurgeObsoleteWALFiles() {
  if (options_.WAL_ttl_seconds != ULONG_MAX && options_.WAL_ttl_seconds > 0) {
    std::vector<std::string> WALFiles;
441
    std::string archivalDir = ArchivalDirectory(dbname_);
442 443 444 445
    env_->GetChildren(archivalDir, &WALFiles);
    int64_t currentTime;
    const Status status = env_->GetCurrentTime(&currentTime);
    assert(status.ok());
446 447 448 449 450 451 452 453 454 455 456 457 458
    for (const auto& f : WALFiles) {
      uint64_t fileMTime;
      const std::string filePath = archivalDir + "/" + f;
      const Status s = env_->GetFileModificationTime(filePath, &fileMTime);
      if (s.ok()) {
        if (status.ok() &&
            (currentTime - fileMTime > options_.WAL_ttl_seconds)) {
          Status delStatus = env_->DeleteFile(filePath);
          if (!delStatus.ok()) {
            Log(options_.info_log,
                "Failed Deleting a WAL file Error : i%s",
                delStatus.ToString().c_str());
          }
459
        }
460
      } // Ignore errors.
461 462
    }
  }
D
Dhruba Borthakur 已提交
463 464
}

465 466 467
// If externalTable is set, then apply recovered transactions
// to that table. This is used for readonly mode.
Status DBImpl::Recover(VersionEdit* edit, MemTable* external_table,
H
heyongqiang 已提交
468
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
469 470
  mutex_.AssertHeld();

471
  assert(db_lock_ == nullptr);
472
  if (!external_table) {
473 474 475 476 477 478 479 480 481 482 483 484 485
    // We call CreateDirIfMissing() as the directory may already exist (if we
    // are reopening a DB), when this happens we don't want creating the
    // directory to cause an error. However, we need to check if creating the
    // directory fails or else we may get an obscure message about the lock
    // file not existing. One real-world example of this occurring is if
    // env->CreateDirIfMissing() doesn't create intermediate directories, e.g.
    // when dbname_ is "dir/db" but when "dir" doesn't exist.
    Status s = env_->CreateDirIfMissing(dbname_);
    if (!s.ok()) {
      return s;
    }

    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
486 487 488
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
489

490 491 492 493 494 495 496 497 498
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
        s = NewDB();
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
499 500
      }
    } else {
501 502 503 504
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
505 506 507
    }
  }

508
  Status s = versions_->Recover();
J
jorlow@chromium.org 已提交
509 510
  if (s.ok()) {
    SequenceNumber max_sequence(0);
511 512 513 514 515 516 517 518 519 520 521 522 523 524

    // Recover from all newer log files than the ones named in the
    // descriptor (new log files may have been added by the previous
    // incarnation without registering them in the descriptor).
    //
    // Note that PrevLogNumber() is no longer used, but we pay
    // attention to it in case we are recovering a database
    // produced by an older version of leveldb.
    const uint64_t min_log = versions_->LogNumber();
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
    s = env_->GetChildren(dbname_, &filenames);
    if (!s.ok()) {
      return s;
525
    }
526 527 528 529 530 531 532 533 534
    uint64_t number;
    FileType type;
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
      if (ParseFileName(filenames[i], &number, &type)
          && type == kLogFile
          && ((number >= min_log) || (number == prev_log))) {
        logs.push_back(number);
      }
J
jorlow@chromium.org 已提交
535
    }
536

H
heyongqiang 已提交
537 538 539 540 541 542
    if (logs.size() > 0 && error_if_log_file_exist) {
      return Status::Corruption(""
          "The db was opened in readonly mode with error_if_log_file_exist"
          "flag but a log file already exists");
    }

543 544 545
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
    for (size_t i = 0; i < logs.size(); i++) {
546
      s = RecoverLogFile(logs[i], edit, &max_sequence, external_table);
547 548 549 550
      // The previous incarnation may not have written any MANIFEST
      // records after allocating this log number.  So we manually
      // update the file number allocation counter in VersionSet.
      versions_->MarkFileNumberUsed(logs[i]);
551 552
    }

J
jorlow@chromium.org 已提交
553
    if (s.ok()) {
554 555
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
556 557 558
        last_flushed_sequence_ = max_sequence;
      } else {
        last_flushed_sequence_ = versions_->LastSequence();
559
      }
J
jorlow@chromium.org 已提交
560 561 562 563 564 565 566 567
    }
  }

  return s;
}

Status DBImpl::RecoverLogFile(uint64_t log_number,
                              VersionEdit* edit,
568 569
                              SequenceNumber* max_sequence,
                              MemTable* external_table) {
J
jorlow@chromium.org 已提交
570 571
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
572
    Logger* info_log;
J
jorlow@chromium.org 已提交
573
    const char* fname;
574
    Status* status;  // nullptr if options_.paranoid_checks==false
J
jorlow@chromium.org 已提交
575
    virtual void Corruption(size_t bytes, const Status& s) {
576
      Log(info_log, "%s%s: dropping %d bytes; %s",
577
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
578
          fname, static_cast<int>(bytes), s.ToString().c_str());
579
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
580 581 582 583 584 585 586
    }
  };

  mutex_.AssertHeld();

  // Open the log file
  std::string fname = LogFileName(dbname_, log_number);
587
  unique_ptr<SequentialFile> file;
588
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
589 590 591 592 593 594 595 596
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
597
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
598
  reporter.fname = fname.c_str();
599
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
J
jorlow@chromium.org 已提交
600 601 602 603
  // We intentially make log::Reader do checksumming even if
  // paranoid_checks==false so that corruptions cause entire commits
  // to be skipped instead of propagating bad information (like overly
  // large sequence numbers).
604
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
605
                     0/*initial_offset*/);
606
  Log(options_.info_log, "Recovering log #%llu",
J
jorlow@chromium.org 已提交
607 608 609 610 611 612
      (unsigned long long) log_number);

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
613
  MemTable* mem = nullptr;
614 615 616
  if (external_table) {
    mem = external_table;
  }
J
jorlow@chromium.org 已提交
617 618 619 620 621 622 623 624 625
  while (reader.ReadRecord(&record, &scratch) &&
         status.ok()) {
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

626
    if (mem == nullptr) {
627
      mem = new MemTable(internal_comparator_, NumberLevels());
628
      mem->Ref();
J
jorlow@chromium.org 已提交
629 630 631 632 633 634 635 636 637 638 639 640 641
    }
    status = WriteBatchInternal::InsertInto(&batch, mem);
    MaybeIgnoreError(&status);
    if (!status.ok()) {
      break;
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

642 643
    if (!external_table &&
        mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
644
      status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
645 646 647 648 649
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
        break;
      }
650
      mem->Unref();
651
      mem = nullptr;
J
jorlow@chromium.org 已提交
652 653 654
    }
  }

655
  if (status.ok() && mem != nullptr && !external_table) {
656
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
657 658 659 660
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

661
  if (mem != nullptr && !external_table) mem->Unref();
J
jorlow@chromium.org 已提交
662 663 664
  return status;
}

665
Status DBImpl::WriteLevel0TableForRecovery(MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
666
  mutex_.AssertHeld();
667
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
668 669 670 671
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
672 673 674
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
675
  Log(options_.info_log, "Level-0 table #%llu: started",
J
jorlow@chromium.org 已提交
676
      (unsigned long long) meta.number);
677 678 679 680

  Status s;
  {
    mutex_.Unlock();
681 682
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
683 684
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
685 686 687
    mutex_.Lock();
  }

688
  Log(options_.info_log, "Level-0 table #%llu: %lld bytes %s",
J
jorlow@chromium.org 已提交
689 690 691 692
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;
693

694
  pending_outputs_.erase(meta.number);
695 696 697 698 699 700 701 702 703

  // Note that if file_size is zero, the file has been deleted and
  // should not be added to the manifest.
  int level = 0;
  if (s.ok() && meta.file_size > 0) {
    edit->AddFile(level, meta.number, meta.file_size,
                  meta.smallest, meta.largest);
  }

704 705 706
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
707
  stats.files_out_levelnp1 = 1;
708
  stats_[level].Add(stats);
J
jorlow@chromium.org 已提交
709 710 711
  return s;
}

712 713 714

Status DBImpl::WriteLevel0Table(MemTable* mem, VersionEdit* edit,
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
715
  mutex_.AssertHeld();
716 717 718 719 720 721
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
722 723 724
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
725 726
  Log(options_.info_log, "Level-0 flush table #%llu: started",
      (unsigned long long) meta.number);
J
jorlow@chromium.org 已提交
727

728 729
  Version* base = versions_->current();
  base->Ref();
730 731 732
  Status s;
  {
    mutex_.Unlock();
733 734
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
735 736
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
737 738
    mutex_.Lock();
  }
739 740
  base->Unref();

741 742 743 744 745 746 747 748 749 750 751 752 753 754
  Log(options_.info_log, "Level-0 flush table #%llu: %lld bytes %s",
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;

  // re-acquire the most current version
  base = versions_->current();

  // There could be multiple threads writing to its own level-0 file.
  // The pending_outputs cannot be cleared here, otherwise this newly
  // created file might not be considered as a live-file by another
  // compaction thread that is concurrently deleting obselete files.
  // The pending_outputs can be cleared only after the new version is
A
Abhishek Kona 已提交
755
  // committed so that other threads can recognize this file as a
756 757 758 759 760 761 762 763 764 765 766 767 768
  // valid one.
  // pending_outputs_.erase(meta.number);

  // Note that if file_size is zero, the file has been deleted and
  // should not be added to the manifest.
  int level = 0;
  if (s.ok() && meta.file_size > 0) {
    const Slice min_user_key = meta.smallest.user_key();
    const Slice max_user_key = meta.largest.user_key();
    // if we have more than 1 background thread, then we cannot
    // insert files directly into higher levels because some other
    // threads could be concurrently producing compacted files for
    // that key range.
769
    if (base != nullptr && options_.max_background_compactions <= 1) {
770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
                  meta.smallest, meta.largest);
  }

  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
  stats_[level].Add(stats);
  return s;
}

Status DBImpl::CompactMemTable(bool* madeProgress) {
  mutex_.AssertHeld();
  assert(imm_.size() != 0);

  if (!imm_.IsFlushPending()) {
    Log(options_.info_log, "Memcompaction already in progress");
    Status s = Status::IOError("Memcompaction already in progress");
    return s;
  }

  // Save the contents of the earliest memtable as a new Table
  // This will release and re-acquire the mutex.
  uint64_t file_number;
  MemTable* m = imm_.PickMemtableToFlush();
797
  if (m == nullptr) {
798 799 800 801 802 803 804 805 806 807 808 809
    Log(options_.info_log, "Nothing in memstore to flush");
    Status s = Status::IOError("Nothing in memstore to flush");
    return s;
  }

  // record the logfile_number_ before we release the mutex
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
  edit->SetLogNumber(logfile_number_);  // Earlier logs no longer needed

  Status s = WriteLevel0Table(m, edit, &file_number);

810 811 812
  if (s.ok() && shutting_down_.Acquire_Load()) {
    s = Status::IOError("Deleting DB during memtable compaction");
  }
J
jorlow@chromium.org 已提交
813

814
  // Replace immutable memtable with the generated Table
815 816 817
  s = imm_.InstallMemtableFlushResults(
    m, versions_.get(), s, &mutex_, options_.info_log.get(),
    file_number, pending_outputs_);
J
jorlow@chromium.org 已提交
818 819

  if (s.ok()) {
820 821 822
    if (madeProgress) {
      *madeProgress = 1;
    }
823
    MaybeScheduleLogDBDeployStats();
D
Dhruba Borthakur 已提交
824 825 826
    // we could have deleted obsolete files here, but it is not
    // absolutely necessary because it could be also done as part
    // of other background compaction
J
jorlow@chromium.org 已提交
827 828 829 830
  }
  return s;
}

G
Gabor Cselle 已提交
831 832 833 834 835
void DBImpl::CompactRange(const Slice* begin, const Slice* end) {
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->current();
836
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
837 838 839 840 841 842 843 844 845 846 847
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
  TEST_CompactMemTable(); // TODO(sanjay): Skip if memtable does not overlap
  for (int level = 0; level < max_level_with_files; level++) {
    TEST_CompactRange(level, begin, end);
  }
}

848
int DBImpl::NumberLevels() {
849
  return options_.num_levels;
850 851 852
}

int DBImpl::MaxMemCompactionLevel() {
853
  return options_.max_mem_compaction_level;
854 855 856
}

int DBImpl::Level0StopWriteTrigger() {
857
  return options_.level0_stop_writes_trigger;
858 859
}

H
heyongqiang 已提交
860 861 862 863 864
Status DBImpl::Flush(const FlushOptions& options) {
  Status status = FlushMemTable(options);
  return status;
}

865 866 867 868
SequenceNumber DBImpl::GetLatestSequenceNumber() {
  return versions_->LastSequence();
}

869
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
870
                               unique_ptr<TransactionLogIterator>* iter) {
871 872 873 874 875 876 877 878 879 880 881 882 883

  //  Get All Log Files.
  //  Sort Files
  //  Get the first entry from each file.
  //  Do binary search and open files and find the seq number.

  std::vector<LogFile> walFiles;
  // list wal files in main db dir.
  Status s = ListAllWALFiles(dbname_, &walFiles, kAliveLogFile);
  if (!s.ok()) {
    return s;
  }
  //  list wal files in archive dir.
884 885 886 887 888 889
  std::string archivedir = ArchivalDirectory(dbname_);
  if (env_->FileExists(archivedir)) {
    s = ListAllWALFiles(archivedir, &walFiles, kArchivedLogFile);
    if (!s.ok()) {
      return s;
    }
890 891 892 893 894 895 896 897
  }

  if (walFiles.empty()) {
    return Status::IOError(" NO WAL Files present in the db");
  }
  //  std::shared_ptr would have been useful here.

  std::vector<LogFile>* probableWALFiles = new std::vector<LogFile>();
898 899 900
  s = FindProbableWALFiles(&walFiles, probableWALFiles, seq);
  if (!s.ok()) {
    return s;
901
  }
902
  iter->reset(
903 904
    new TransactionLogIteratorImpl(dbname_,
                                   &options_,
905
                                   storage_options_,
906 907 908 909
                                   seq,
                                   probableWALFiles,
                                   &last_flushed_sequence_));
  iter->get()->Next();
910
  return iter->get()->status();
911 912 913 914 915
}

Status DBImpl::FindProbableWALFiles(std::vector<LogFile>* const allLogs,
                                    std::vector<LogFile>* const result,
                                    const SequenceNumber target) {
916 917
  assert(allLogs != nullptr);
  assert(result != nullptr);
918 919

  std::sort(allLogs->begin(), allLogs->end());
920 921
  long start = 0; // signed to avoid overflow when target is < first file.
  long end = static_cast<long>(allLogs->size()) - 1;
922
  // Binary Search. avoid opening all files.
923 924
  while (end >= start) {
    long mid = start + (end - start) / 2;  // Avoid overflow.
925 926 927
    WriteBatch batch;
    Status s = ReadFirstRecord(allLogs->at(mid), &batch);
    if (!s.ok()) {
928 929
      if (CheckFileExistsAndEmpty(allLogs->at(mid))) {
        allLogs->erase(allLogs->begin() + mid);
930
        --end;
931 932
        continue;
      }
933 934 935 936 937 938
      return s;
    }
    SequenceNumber currentSeqNum = WriteBatchInternal::Sequence(&batch);
    if (currentSeqNum == target) {
      start = mid;
      end = mid;
939
      break;
940
    } else if (currentSeqNum < target) {
941
      start = mid + 1;
942
    } else {
943
      end = mid - 1;
944 945
    }
  }
946 947
  size_t startIndex = std::max(0l, end); // end could be -ve.
  for( size_t i = startIndex; i < allLogs->size(); ++i) {
948 949
    result->push_back(allLogs->at(i));
  }
950 951 952 953
  if (result->empty()) {
    return Status::IOError(
        "No probable files. Check if the db contains log files");
  }
954 955 956
  return Status::OK();
}

957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974
bool DBImpl::CheckFileExistsAndEmpty(const LogFile& file) {
  if (file.type == kAliveLogFile) {
    const std::string fname = LogFileName(dbname_, file.logNumber);
    uint64_t file_size;
    Status s = env_->GetFileSize(fname, &file_size);
    if (s.ok() && file_size == 0) {
      return true;
    }
  }
  const std::string fname = ArchivedLogFileName(dbname_, file.logNumber);
  uint64_t file_size;
  Status s = env_->GetFileSize(fname, &file_size);
  if (s.ok() && file_size == 0) {
    return true;
  }
  return false;
}

975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002
Status DBImpl::ReadFirstRecord(const LogFile& file, WriteBatch* const result) {

  if (file.type == kAliveLogFile) {
    std::string fname = LogFileName(dbname_, file.logNumber);
    Status status = ReadFirstLine(fname, result);
    if (!status.ok()) {
      //  check if the file got moved to archive.
      std::string archivedFile = ArchivedLogFileName(dbname_, file.logNumber);
      Status s = ReadFirstLine(archivedFile, result);
      if (!s.ok()) {
        return Status::IOError("Log File Has been deleted");
      }
    }
    return Status::OK();
  } else if (file.type == kArchivedLogFile) {
    std::string fname = ArchivedLogFileName(dbname_, file.logNumber);
    Status status = ReadFirstLine(fname, result);
    return status;
  }
  return Status::NotSupported("File Type Not Known");
}

Status DBImpl::ReadFirstLine(const std::string& fname,
                             WriteBatch* const batch) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;
1003
    Status* status;  // nullptr if options_.paranoid_checks==false
1004 1005
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
1006
          (this->status == nullptr ? "(ignoring error) " : ""),
1007
          fname, static_cast<int>(bytes), s.ToString().c_str());
1008
      if (this->status != nullptr && this->status->ok()) *this->status = s;
1009 1010 1011
    }
  };

1012
  unique_ptr<SequentialFile> file;
1013
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
1014 1015 1016 1017 1018 1019 1020 1021

  if (!status.ok()) {
    return status;
  }


  LogReporter reporter;
  reporter.env = env_;
1022
  reporter.info_log = options_.info_log.get();
1023
  reporter.fname = fname.c_str();
1024
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
1025
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044
                     0/*initial_offset*/);
  std::string scratch;
  Slice record;
  if (reader.ReadRecord(&record, &scratch) && status.ok()) {
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      return Status::IOError("Corruption noted");
      //  TODO read record's till the first no corrupt entry?
    }
    WriteBatchInternal::SetContents(batch, record);
    return Status::OK();
  }
  return Status::IOError("Error reading from file " + fname);
}

Status DBImpl::ListAllWALFiles(const std::string& path,
                               std::vector<LogFile>* const logFiles,
                               WalFileType logType) {
1045
  assert(logFiles != nullptr);
1046 1047 1048 1049 1050
  std::vector<std::string> allFiles;
  const Status status = env_->GetChildren(path, &allFiles);
  if (!status.ok()) {
    return status;
  }
1051
  for (const auto& f : allFiles) {
1052 1053
    uint64_t number;
    FileType type;
1054
    if (ParseFileName(f, &number, &type) && type == kLogFile){
1055 1056 1057 1058 1059 1060
      logFiles->push_back(LogFile(number, logType));
    }
  }
  return status;
}

G
Gabor Cselle 已提交
1061
void DBImpl::TEST_CompactRange(int level, const Slice* begin,const Slice* end) {
1062 1063
  assert(level >= 0);

G
Gabor Cselle 已提交
1064 1065
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1066 1067
  ManualCompaction manual;
  manual.level = level;
G
Gabor Cselle 已提交
1068
  manual.done = false;
1069
  manual.in_progress = false;
1070 1071
  if (begin == nullptr) {
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1072 1073 1074 1075
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1076 1077
  if (end == nullptr) {
    manual.end = nullptr;
G
Gabor Cselle 已提交
1078 1079 1080 1081 1082 1083
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1084

A
Abhishek Kona 已提交
1085 1086 1087 1088
  // When a manual compaction arrives, temporarily throttle down
  // the number of background compaction threads to 1. This is
  // needed to ensure that this manual compaction can compact
  // any range of keys/files. We artificialy increase
1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
  // bg_compaction_scheduled_ by a large number, this causes
  // the system to have a single background thread. Now,
  // this manual compaction can progress without stomping
  // on any other concurrent compactions.
  const int LargeNumber = 10000000;
  const int newvalue = options_.max_background_compactions-1;
  bg_compaction_scheduled_ += LargeNumber;
  while (bg_compaction_scheduled_ > LargeNumber) {
    Log(options_.info_log, "Manual compaction request waiting for background threads to fall below 1");
    bg_cv_.Wait();
  }
  Log(options_.info_log, "Manual compaction starting");

G
Gabor Cselle 已提交
1102
  while (!manual.done) {
1103
    while (manual_compaction_ != nullptr) {
G
Gabor Cselle 已提交
1104 1105 1106
      bg_cv_.Wait();
    }
    manual_compaction_ = &manual;
1107 1108 1109
    if (bg_compaction_scheduled_ == LargeNumber) {
      bg_compaction_scheduled_ = newvalue;
    }
G
Gabor Cselle 已提交
1110 1111 1112 1113
    MaybeScheduleCompaction();
    while (manual_compaction_ == &manual) {
      bg_cv_.Wait();
    }
H
hans@chromium.org 已提交
1114
  }
1115 1116 1117 1118 1119 1120 1121 1122 1123
  assert(!manual.in_progress);

  // wait till there are no background threads scheduled
  bg_compaction_scheduled_ += LargeNumber;
  while (bg_compaction_scheduled_ > LargeNumber + newvalue) {
    Log(options_.info_log, "Manual compaction resetting background threads");
    bg_cv_.Wait();
  }
  bg_compaction_scheduled_ = 0;
J
jorlow@chromium.org 已提交
1124 1125
}

H
heyongqiang 已提交
1126
Status DBImpl::FlushMemTable(const FlushOptions& options) {
1127 1128
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1129
  if (s.ok() && options.wait) {
1130
    // Wait until the compaction completes
H
heyongqiang 已提交
1131
    s = WaitForCompactMemTable();
1132 1133
  }
  return s;
J
jorlow@chromium.org 已提交
1134 1135
}

H
heyongqiang 已提交
1136
Status DBImpl::WaitForCompactMemTable() {
1137 1138 1139
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1140
  while (imm_.size() > 0 && bg_error_.ok()) {
1141 1142
    bg_cv_.Wait();
  }
1143
  if (imm_.size() != 0) {
1144 1145 1146
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1147 1148 1149 1150 1151 1152
}

Status DBImpl::TEST_CompactMemTable() {
  return FlushMemTable(FlushOptions());
}

1153
Status DBImpl::TEST_WaitForCompactMemTable() {
1154
  return WaitForCompactMemTable();
1155 1156 1157
}

Status DBImpl::TEST_WaitForCompact() {
1158 1159 1160 1161 1162 1163
  // Wait until the compaction completes
  MutexLock l(&mutex_);
  while (bg_compaction_scheduled_ && bg_error_.ok()) {
    bg_cv_.Wait();
  }
  return bg_error_;
1164 1165
}

J
jorlow@chromium.org 已提交
1166 1167
void DBImpl::MaybeScheduleCompaction() {
  mutex_.AssertHeld();
1168
  if (bg_compaction_scheduled_ >= options_.max_background_compactions) {
J
jorlow@chromium.org 已提交
1169 1170 1171
    // Already scheduled
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
1172
  } else if (!imm_.IsFlushPending() &&
1173
             manual_compaction_ == nullptr &&
H
hans@chromium.org 已提交
1174
             !versions_->NeedsCompaction()) {
J
jorlow@chromium.org 已提交
1175 1176
    // No work to be done
  } else {
1177
    bg_compaction_scheduled_++;
J
jorlow@chromium.org 已提交
1178 1179 1180 1181 1182 1183 1184 1185 1186
    env_->Schedule(&DBImpl::BGWork, this);
  }
}

void DBImpl::BGWork(void* db) {
  reinterpret_cast<DBImpl*>(db)->BackgroundCall();
}

void DBImpl::BackgroundCall() {
1187
  bool madeProgress = false;
D
Dhruba Borthakur 已提交
1188
  DeletionState deletion_state;
J
jorlow@chromium.org 已提交
1189
  MutexLock l(&mutex_);
1190
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1191
  assert(bg_compaction_scheduled_);
H
hans@chromium.org 已提交
1192
  if (!shutting_down_.Acquire_Load()) {
1193
    Status s = BackgroundCompaction(&madeProgress, deletion_state);
1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205
    if (!s.ok()) {
      // Wait a little bit before retrying background compaction in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed compactions for the duration of
      // the problem.
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      Log(options_.info_log, "Waiting after background compaction error: %s",
          s.ToString().c_str());
      mutex_.Unlock();
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
J
jorlow@chromium.org 已提交
1206
  }
1207

D
Dhruba Borthakur 已提交
1208 1209 1210 1211
  // delete unnecessary files if any, this is done outside the mutex
  if (!deletion_state.live.empty()) {
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
D
Dhruba Borthakur 已提交
1212
    EvictObsoleteFiles(deletion_state);
1213
    mutex_.Lock();
D
Dhruba Borthakur 已提交
1214 1215
  }

1216
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1217

1218 1219
  MaybeScheduleLogDBDeployStats();

J
jorlow@chromium.org 已提交
1220
  // Previous compaction may have produced too many files in a level,
A
Abhishek Kona 已提交
1221
  // So reschedule another compaction if we made progress in the
1222 1223 1224 1225
  // last compaction.
  if (madeProgress) {
    MaybeScheduleCompaction();
  }
H
hans@chromium.org 已提交
1226
  bg_cv_.SignalAll();
J
jorlow@chromium.org 已提交
1227 1228
}

A
Abhishek Kona 已提交
1229
Status DBImpl::BackgroundCompaction(bool* madeProgress,
1230
  DeletionState& deletion_state) {
1231
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1232
  mutex_.AssertHeld();
1233

1234
  while (imm_.IsFlushPending()) {
A
Abhishek Kona 已提交
1235
    Log(options_.info_log,
1236 1237 1238 1239 1240 1241
        "BackgroundCompaction doing CompactMemTable, compaction slots available %d",
        options_.max_background_compactions - bg_compaction_scheduled_);
    Status stat = CompactMemTable(madeProgress);
    if (!stat.ok()) {
      return stat;
    }
1242 1243
  }

1244
  unique_ptr<Compaction> c;
1245
  bool is_manual = (manual_compaction_ != nullptr) &&
1246
                   (manual_compaction_->in_progress == false);
G
Gabor Cselle 已提交
1247
  InternalKey manual_end;
H
hans@chromium.org 已提交
1248
  if (is_manual) {
G
Gabor Cselle 已提交
1249
    ManualCompaction* m = manual_compaction_;
1250 1251
    assert(!m->in_progress);
    m->in_progress = true; // another thread cannot pick up the same work
1252 1253
    c.reset(versions_->CompactRange(m->level, m->begin, m->end));
    if (c) {
G
Gabor Cselle 已提交
1254
      manual_end = c->input(0, c->num_input_files(0) - 1)->largest;
1255 1256
    } else {
      m->done = true;
G
Gabor Cselle 已提交
1257 1258 1259
    }
    Log(options_.info_log,
        "Manual compaction at level-%d from %s .. %s; will stop at %s\n",
H
hans@chromium.org 已提交
1260
        m->level,
G
Gabor Cselle 已提交
1261 1262 1263
        (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
        (m->end ? m->end->DebugString().c_str() : "(end)"),
        (m->done ? "(end)" : manual_end.DebugString().c_str()));
1264
  } else if (!options_.disable_auto_compactions) {
1265
    c.reset(versions_->PickCompaction());
J
jorlow@chromium.org 已提交
1266 1267 1268
  }

  Status status;
1269
  if (!c) {
H
hans@chromium.org 已提交
1270
    // Nothing to do
1271
    Log(options_.info_log, "Compaction nothing to do");
H
hans@chromium.org 已提交
1272
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
1273
    // Move file to next level
1274
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
1275 1276 1277 1278
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
                       f->smallest, f->largest);
1279
    status = versions_->LogAndApply(c->edit(), &mutex_);
H
hans@chromium.org 已提交
1280
    VersionSet::LevelSummaryStorage tmp;
1281
    Log(options_.info_log, "Moved #%lld to level-%d %lld bytes %s: %s\n",
J
jorlow@chromium.org 已提交
1282 1283 1284
        static_cast<unsigned long long>(f->number),
        c->level() + 1,
        static_cast<unsigned long long>(f->file_size),
H
hans@chromium.org 已提交
1285 1286
        status.ToString().c_str(),
        versions_->LevelSummary(&tmp));
1287
    versions_->ReleaseCompactionFiles(c.get(), status);
1288
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1289
  } else {
1290
    MaybeScheduleCompaction(); // do more compaction work in parallel.
1291
    CompactionState* compact = new CompactionState(c.get());
J
jorlow@chromium.org 已提交
1292 1293
    status = DoCompactionWork(compact);
    CleanupCompaction(compact);
1294
    versions_->ReleaseCompactionFiles(c.get(), status);
1295
    c->ReleaseInputs();
D
Dhruba Borthakur 已提交
1296
    FindObsoleteFiles(deletion_state);
1297
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1298
  }
1299
  c.reset();
J
jorlow@chromium.org 已提交
1300 1301 1302 1303 1304 1305

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
1306
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1307 1308 1309 1310 1311
        "Compaction error: %s", status.ToString().c_str());
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
1312 1313

  if (is_manual) {
G
Gabor Cselle 已提交
1314
    ManualCompaction* m = manual_compaction_;
1315 1316 1317
    if (!status.ok()) {
      m->done = true;
    }
G
Gabor Cselle 已提交
1318 1319 1320 1321 1322 1323
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
      m->tmp_storage = manual_end;
      m->begin = &m->tmp_storage;
    }
1324
    m->in_progress = false; // not being processed anymore
1325
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1326
  }
1327
  return status;
J
jorlow@chromium.org 已提交
1328 1329 1330 1331
}

void DBImpl::CleanupCompaction(CompactionState* compact) {
  mutex_.AssertHeld();
1332
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
1333 1334
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
1335
    compact->builder.reset();
J
jorlow@chromium.org 已提交
1336
  } else {
1337
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
1338
  }
D
dgrogan@chromium.org 已提交
1339
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1340 1341 1342 1343 1344 1345
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
  }
  delete compact;
}

1346 1347 1348 1349 1350
// Allocate the file numbers for the output file. We allocate as
// many output file numbers as there are files in level+1.
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
1351 1352
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
1353
  int filesNeeded = compact->compaction->num_input_files(1);
1354
  for (int i = 0; i < filesNeeded; i++) {
1355 1356 1357 1358 1359 1360 1361 1362 1363
    uint64_t file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    compact->allocated_file_numbers.push_back(file_number);
  }
}

// Frees up unused file number.
void DBImpl::ReleaseCompactionUnusedFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
1364
  for (const auto file_number : compact->allocated_file_numbers) {
1365 1366 1367 1368 1369
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
1370
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
1371 1372
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
1373
  uint64_t file_number;
1374 1375 1376 1377 1378 1379 1380
  // If we have not yet exhausted the pre-allocated file numbers,
  // then use the one from the front. Otherwise, we have to acquire
  // the heavyweight lock and allocate a new file number.
  if (!compact->allocated_file_numbers.empty()) {
    file_number = compact->allocated_file_numbers.front();
    compact->allocated_file_numbers.pop_front();
  } else {
J
jorlow@chromium.org 已提交
1381 1382 1383 1384 1385
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
1386 1387 1388 1389 1390
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
1391 1392 1393

  // Make the output file
  std::string fname = TableFileName(dbname_, file_number);
1394
  Status s = env_->NewWritableFile(fname, &compact->outfile, storage_options_);
1395 1396 1397 1398 1399 1400

  // Over-estimate slightly so we don't end up just barely crossing
  // the threshold.
  compact->outfile->SetPreallocationBlockSize(
    1.1 * versions_->MaxFileSizeForLevel(compact->compaction->level() + 1));

J
jorlow@chromium.org 已提交
1401
  if (s.ok()) {
1402 1403
    compact->builder.reset(new TableBuilder(options_, compact->outfile.get(),
                                            compact->compaction->level() + 1));
J
jorlow@chromium.org 已提交
1404 1405 1406 1407 1408 1409
  }
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
1410
  assert(compact != nullptr);
1411
  assert(compact->outfile);
1412
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427

  const uint64_t output_number = compact->current_output()->number;
  assert(output_number != 0);

  // Check for iterator errors
  Status s = input->status();
  const uint64_t current_entries = compact->builder->NumEntries();
  if (s.ok()) {
    s = compact->builder->Finish();
  } else {
    compact->builder->Abandon();
  }
  const uint64_t current_bytes = compact->builder->FileSize();
  compact->current_output()->file_size = current_bytes;
  compact->total_bytes += current_bytes;
1428
  compact->builder.reset();
J
jorlow@chromium.org 已提交
1429 1430

  // Finish and check for file errors
1431
  if (s.ok() && !options_.disableDataSync) {
1432 1433 1434 1435 1436
    if (options_.use_fsync) {
      s = compact->outfile->Fsync();
    } else {
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
1437 1438 1439 1440
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
1441
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
1442 1443 1444

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
1445
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
1446
                                               storage_options_,
J
jorlow@chromium.org 已提交
1447 1448
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
1449 1450 1451
    s = iter->status();
    delete iter;
    if (s.ok()) {
1452
      Log(options_.info_log,
J
jorlow@chromium.org 已提交
1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464
          "Generated table #%llu: %lld keys, %lld bytes",
          (unsigned long long) output_number,
          (unsigned long long) current_entries,
          (unsigned long long) current_bytes);
    }
  }
  return s;
}


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
1465 1466 1467 1468 1469

  // paranoia: verify that the files that we started with
  // still exist in the current version and in the same original level.
  // This ensures that a concurrent compaction did not erroneously
  // pick the same files to compact.
1470
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
1471 1472 1473 1474 1475 1476 1477 1478
    Log(options_.info_log,  "Compaction %d@%d + %d@%d files aborted",
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1);
    return Status::IOError("Compaction input files inconsistent");
  }

1479
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
1480 1481 1482 1483 1484 1485 1486 1487 1488
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1,
      static_cast<long long>(compact->total_bytes));

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
  const int level = compact->compaction->level();
D
dgrogan@chromium.org 已提交
1489
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1490 1491 1492 1493 1494
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
        level + 1,
        out.number, out.file_size, out.smallest, out.largest);
  }
1495
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
1496 1497
}

1498 1499 1500 1501 1502 1503 1504 1505 1506
//
// Given a sequence number, return the sequence number of the
// earliest snapshot that this sequence number is visible in.
// The snapshots themselves are arranged in ascending order of
// sequence numbers.
// Employ a sequential search because the total number of
// snapshots are typically small.
inline SequenceNumber DBImpl::findEarliestVisibleSnapshot(
  SequenceNumber in, std::vector<SequenceNumber>& snapshots) {
1507
  SequenceNumber prev __attribute__((unused)) = 0;
1508 1509 1510 1511
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
      return cur;
1512
    }
1513 1514
    prev = cur; // assignment
    assert(prev);
1515 1516 1517 1518 1519 1520 1521 1522
  }
  Log(options_.info_log,
      "Looking for seqid %ld but maxseqid is %ld", in,
      snapshots[snapshots.size()-1]);
  assert(0);
  return 0;
}

J
jorlow@chromium.org 已提交
1523
Status DBImpl::DoCompactionWork(CompactionState* compact) {
1524 1525
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions

A
Abhishek Kona 已提交
1526
  Log(options_.info_log,
1527
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
1528 1529 1530
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
1531
      compact->compaction->level() + 1,
1532
      compact->compaction->score(),
1533
      options_.max_background_compactions - bg_compaction_scheduled_);
1534 1535
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
1536
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
1537 1538

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
1539
  assert(compact->builder == nullptr);
1540
  assert(!compact->outfile);
1541 1542 1543 1544 1545 1546 1547 1548

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
  snapshots_.getAll(compact->existing_snapshots);
  if (compact->existing_snapshots.size() == 0) {
    // optimize for fast path if there are no snapshots
    visible_at_tip = versions_->LastSequence();
    earliest_snapshot = visible_at_tip;
J
jorlow@chromium.org 已提交
1549
  } else {
1550 1551 1552 1553
    // Add the current seqno as the 'latest' virtual
    // snapshot to the end of this list.
    compact->existing_snapshots.push_back(versions_->LastSequence());
    earliest_snapshot = compact->existing_snapshots[0];
J
jorlow@chromium.org 已提交
1554 1555
  }

1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
  // Is this compaction producing files at the bottommost level?
  bool bottommost_level = true;
  for (int i = compact->compaction->level() + 2;
       i < versions_->NumberLevels(); i++) {
    if (versions_->NumLevelFiles(i) > 0) {
      bottommost_level = false;
      break;
    }
  }

1566 1567 1568
  // Allocate the output file numbers before we release the lock
  AllocateCompactionOutputFileNumbers(compact);

J
jorlow@chromium.org 已提交
1569 1570 1571
  // Release mutex while we're actually doing the compaction work
  mutex_.Unlock();

1572
  const uint64_t start_micros = env_->NowMicros();
1573
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
1574 1575 1576 1577 1578
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
1579 1580
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
1581
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1582
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
1583
    // Prioritize immutable compaction work
1584
    if (imm_.imm_flush_needed.NoBarrier_Load() != nullptr) {
1585 1586
      const uint64_t imm_start = env_->NowMicros();
      mutex_.Lock();
1587
      if (imm_.IsFlushPending()) {
1588
        CompactMemTable();
H
hans@chromium.org 已提交
1589
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
1590 1591 1592 1593 1594
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
1595
    Slice key = input->key();
1596
    Slice value = input->value();
1597
    Slice* compaction_filter_value = nullptr;
1598
    if (compact->compaction->ShouldStopBefore(key) &&
1599
        compact->builder != nullptr) {
1600
      status = FinishCompactionOutputFile(compact, input.get());
1601 1602 1603 1604 1605 1606
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
1607 1608 1609 1610 1611 1612
    bool drop = false;
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
1613
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1614 1615 1616 1617 1618 1619 1620 1621
    } else {
      if (!has_current_user_key ||
          user_comparator()->Compare(ikey.user_key,
                                     Slice(current_user_key)) != 0) {
        // First occurrence of this user key
        current_user_key.assign(ikey.user_key.data(), ikey.user_key.size());
        has_current_user_key = true;
        last_sequence_for_key = kMaxSequenceNumber;
1622
        visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1623 1624
      }

1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635
      // If there are no snapshots, then this kv affect visibility at tip.
      // Otherwise, search though all existing snapshots to find
      // the earlist snapshot that is affected by this kv.
      SequenceNumber visible = visible_at_tip ? visible_at_tip :
                                 findEarliestVisibleSnapshot(ikey.sequence,
                                 compact->existing_snapshots);

      if (visible_in_snapshot == visible) {
        // If the earliest snapshot is which this key is visible in
        // is the same as the visibily of a previous instance of the
        // same key, then this kv is not visible in any snapshot.
J
jorlow@chromium.org 已提交
1636
        // Hidden by an newer entry for same user key
1637
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
1638
        drop = true;    // (A)
1639
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
1640
      } else if (ikey.type == kTypeDeletion &&
1641
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
1642 1643 1644 1645 1646 1647 1648 1649 1650
                 compact->compaction->IsBaseLevelForKey(ikey.user_key)) {
        // For this user key:
        // (1) there is no data in higher levels
        // (2) data in lower levels will have larger sequence numbers
        // (3) data in layers that are being compacted here and have
        //     smaller sequence numbers will be dropped in the next
        //     few iterations of this loop (by rule (A) above).
        // Therefore this deletion marker is obsolete and can be dropped.
        drop = true;
1651
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_OBSOLETE);
1652
      } else if (options_.CompactionFilter != nullptr &&
1653
                 ikey.type != kTypeDeletion &&
1654
                 ikey.sequence < earliest_snapshot) {
1655 1656
        // If the user has specified a compaction filter, then invoke
        // it. If this key is not visible via any snapshot and the
1657 1658
        // return value of the compaction filter is true and then
        // drop this key from the output.
1659 1660
        drop = options_.CompactionFilter(options_.compaction_filter_args,
                         compact->compaction->level(),
1661 1662
                         ikey.user_key, value, &compaction_filter_value);

1663 1664 1665
        if (drop) {
          RecordTick(options_.statistics, COMPACTION_KEY_DROP_USER);
        }
1666
        // If the application wants to change the value, then do so here.
1667
        if (compaction_filter_value != nullptr) {
1668 1669 1670
          value = *compaction_filter_value;
          delete compaction_filter_value;
        }
J
jorlow@chromium.org 已提交
1671 1672 1673
      }

      last_sequence_for_key = ikey.sequence;
1674
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
1675 1676
    }
#if 0
1677
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1678
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
1679
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
1680
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
1681
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
1682
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
1683 1684
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
1685 1686 1687
#endif

    if (!drop) {
1688

1689 1690 1691
      char* kptr = (char*)key.data();
      std::string kstr;

1692 1693 1694 1695 1696 1697
      // Zeroing out the sequence number leads to better compression.
      // If this is the bottommost level (no files in lower levels)
      // and the earliest snapshot is larger than this seqno
      // then we can squash the seqno to zero.
      if (bottommost_level && ikey.sequence < earliest_snapshot) {
        assert(ikey.type != kTypeDeletion);
1698 1699 1700 1701 1702
        // make a copy because updating in place would cause problems
        // with the priority queue that is managing the input key iterator
        kstr.assign(key.data(), key.size());
        kptr = (char *)kstr.c_str();
        UpdateInternalKey(kptr, key.size(), (uint64_t)0, ikey.type);
1703
      }
1704

1705 1706 1707
      Slice newkey(kptr, key.size());
      assert((key.clear(), 1)); // we do not need 'key' anymore

J
jorlow@chromium.org 已提交
1708
      // Open output file if necessary
1709
      if (compact->builder == nullptr) {
J
jorlow@chromium.org 已提交
1710 1711 1712 1713 1714 1715
        status = OpenCompactionOutputFile(compact);
        if (!status.ok()) {
          break;
        }
      }
      if (compact->builder->NumEntries() == 0) {
1716
        compact->current_output()->smallest.DecodeFrom(newkey);
J
jorlow@chromium.org 已提交
1717
      }
1718 1719
      compact->current_output()->largest.DecodeFrom(newkey);
      compact->builder->Add(newkey, value);
J
jorlow@chromium.org 已提交
1720 1721 1722 1723

      // Close output file if it is big enough
      if (compact->builder->FileSize() >=
          compact->compaction->MaxOutputFileSize()) {
1724
        status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736
        if (!status.ok()) {
          break;
        }
      }
    }

    input->Next();
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
    status = Status::IOError("Deleting DB during compaction");
  }
1737
  if (status.ok() && compact->builder != nullptr) {
1738
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
1739 1740 1741 1742
  }
  if (status.ok()) {
    status = input->status();
  }
1743
  input.reset();
J
jorlow@chromium.org 已提交
1744

1745 1746
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
A
Abhishek Kona 已提交
1747 1748 1749
  if (options_.statistics) {
    options_.statistics->measureTime(COMPACTION_TIME, stats.micros);
  }
M
Mark Callaghan 已提交
1750 1751
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
1752 1753

  int num_output_files = compact->outputs.size();
1754
  if (compact->builder != nullptr) {
1755 1756 1757 1758 1759
    // An error occured so ignore the last output.
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
1760 1761 1762 1763 1764 1765 1766

  for (int i = 0; i < compact->compaction->num_input_files(0); i++)
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;

  for (int i = 0; i < compact->compaction->num_input_files(1); i++)
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;

1767
  for (int i = 0; i < num_output_files; i++) {
1768 1769 1770
    stats.bytes_written += compact->outputs[i].file_size;
  }

J
jorlow@chromium.org 已提交
1771
  mutex_.Lock();
1772
  stats_[compact->compaction->level() + 1].Add(stats);
J
jorlow@chromium.org 已提交
1773

1774 1775 1776 1777
  // if there were any unused file number (mostly in case of
  // compaction error), free up the entry from pending_putputs
  ReleaseCompactionUnusedFileNumbers(compact);

J
jorlow@chromium.org 已提交
1778 1779 1780
  if (status.ok()) {
    status = InstallCompactionResults(compact);
  }
1781
  VersionSet::LevelSummaryStorage tmp;
1782
  Log(options_.info_log,
M
Mark Callaghan 已提交
1783
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
1784
      "MB in(%.1f, %.1f) out(%.1f), amplify(%.1f) %s\n",
M
Mark Callaghan 已提交
1785 1786 1787 1788 1789 1790 1791 1792 1793
      versions_->LevelSummary(&tmp),
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
          (double) stats.micros,
      compact->compaction->level() + 1,
      stats.files_in_leveln, stats.files_in_levelnp1, stats.files_out_levelnp1,
      stats.bytes_readn / 1048576.0,
      stats.bytes_readnp1 / 1048576.0,
      stats.bytes_written / 1048576.0,
      (stats.bytes_written + stats.bytes_readnp1) /
1794 1795
          (double) stats.bytes_readn,
      status.ToString().c_str());
M
Mark Callaghan 已提交
1796

J
jorlow@chromium.org 已提交
1797 1798 1799
  return status;
}

1800 1801 1802 1803
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
1804
  std::vector<MemTable*> mem; // includes both mem_ and imm_
1805 1806 1807 1808 1809
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
  state->mu->Lock();
1810 1811 1812
  for (unsigned int i = 0; i < state->mem.size(); i++) {
    state->mem[i]->Unref();
  }
1813 1814 1815 1816
  state->version->Unref();
  state->mu->Unlock();
  delete state;
}
H
Hans Wennborg 已提交
1817
}  // namespace
1818

J
jorlow@chromium.org 已提交
1819 1820
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
1821
  IterState* cleanup = new IterState;
J
jorlow@chromium.org 已提交
1822
  mutex_.Lock();
1823
  *latest_snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1824

1825
  // Collect together all needed child iterators for mem
J
jorlow@chromium.org 已提交
1826
  std::vector<Iterator*> list;
1827
  mem_->Ref();
1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838
  list.push_back(mem_->NewIterator());
  cleanup->mem.push_back(mem_);

  // Collect together all needed child iterators for imm_
  std::vector<MemTable*> immutables;
  imm_.GetMemTables(&immutables);
  for (unsigned int i = 0; i < immutables.size(); i++) {
    MemTable* m = immutables[i];
    m->Ref();
    list.push_back(m->NewIterator());
    cleanup->mem.push_back(m);
1839
  }
1840 1841

  // Collect iterators for files in L0 - Ln
1842
  versions_->current()->AddIterators(options, storage_options_, &list);
J
jorlow@chromium.org 已提交
1843 1844 1845
  Iterator* internal_iter =
      NewMergingIterator(&internal_comparator_, &list[0], list.size());
  versions_->current()->Ref();
1846 1847 1848

  cleanup->mu = &mutex_;
  cleanup->version = versions_->current();
1849
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
1850 1851 1852 1853 1854 1855 1856 1857 1858 1859

  mutex_.Unlock();
  return internal_iter;
}

Iterator* DBImpl::TEST_NewInternalIterator() {
  SequenceNumber ignored;
  return NewInternalIterator(ReadOptions(), &ignored);
}

J
jorlow@chromium.org 已提交
1860
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
1861 1862 1863 1864
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
1865 1866 1867
Status DBImpl::Get(const ReadOptions& options,
                   const Slice& key,
                   std::string* value) {
1868
  Status s;
1869

1870 1871 1872
  std::unique_ptr<StopWatch> sw = stats::StartStopWatch(env_,
                                                        options_.statistics,
                                                        DB_GET);
1873
  SequenceNumber snapshot;
1874
  MutexLock l(&mutex_);
1875
  if (options.snapshot != nullptr) {
1876 1877 1878
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1879
  }
1880

1881
  MemTable* mem = mem_;
1882
  MemTableList imm = imm_;
1883
  Version* current = versions_->current();
1884
  mem->Ref();
1885
  imm.RefAll();
1886
  current->Ref();
1887

1888 1889 1890
  // Unlock while reading from files and memtables

  mutex_.Unlock();
1891
  bool have_stat_update = false;
1892
  Version::GetStats stats;
1893 1894 1895 1896 1897 1898 1899 1900 1901
  // First look in the memtable, then in the immutable memtable (if any).
  LookupKey lkey(key, snapshot);
  if (mem->Get(lkey, value, &s)) {
    // Done
  } else if (imm.Get(lkey, value, &s)) {
    // Done
  } else {
    s = current->Get(options, lkey, value, &stats);
    have_stat_update = true;
1902
  }
1903
  mutex_.Lock();
1904

1905 1906
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
1907 1908
    MaybeScheduleCompaction();
  }
1909
  mem->Unref();
1910
  imm.UnrefAll();
1911
  current->Unref();
1912
  RecordTick(options_.statistics, NUMBER_KEYS_READ);
1913
  RecordTick(options_.statistics, BYTES_READ, value->size());
1914
  return s;
J
jorlow@chromium.org 已提交
1915 1916 1917 1918 1919
}

Iterator* DBImpl::NewIterator(const ReadOptions& options) {
  SequenceNumber latest_snapshot;
  Iterator* internal_iter = NewInternalIterator(options, &latest_snapshot);
1920 1921
  return NewDBIterator(
      &dbname_, env_, user_comparator(), internal_iter,
1922
      (options.snapshot != nullptr
1923 1924
       ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
       : latest_snapshot));
J
jorlow@chromium.org 已提交
1925 1926 1927 1928
}

const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
1929
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
1930 1931 1932 1933
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
1934
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945
}

// Convenience methods
Status DBImpl::Put(const WriteOptions& o, const Slice& key, const Slice& val) {
  return DB::Put(o, key, val);
}

Status DBImpl::Delete(const WriteOptions& options, const Slice& key) {
  return DB::Delete(options, key);
}

1946 1947 1948 1949
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
1950
  w.disableWAL = options.disableWAL;
1951
  w.done = false;
1952

1953 1954 1955
  std::unique_ptr<StopWatch> sw = stats::StartStopWatch(env_,
                                                        options_.statistics,
                                                        DB_WRITE);
1956
  MutexLock l(&mutex_);
1957 1958 1959 1960 1961 1962 1963 1964 1965
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
  if (w.done) {
    return w.status;
  }

  // May temporarily unlock and wait.
1966
  Status status = MakeRoomForWrite(my_batch == nullptr);
D
dgrogan@chromium.org 已提交
1967
  uint64_t last_sequence = versions_->LastSequence();
1968
  Writer* last_writer = &w;
1969
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
1970
    WriteBatch* updates = BuildBatchGroup(&last_writer);
1971 1972
    const SequenceNumber current_sequence = last_sequence + 1;
    WriteBatchInternal::SetSequence(updates, current_sequence);
1973 1974 1975 1976
    int my_batch_count = WriteBatchInternal::Count(updates);
    last_sequence += my_batch_count;
    // Record statistics
    RecordTick(options_.statistics, NUMBER_KEYS_WRITTEN, my_batch_count);
1977 1978 1979
    RecordTick(options_.statistics,
               BYTES_WRITTEN,
               WriteBatchInternal::ByteSize(updates));
1980 1981 1982 1983
    // Add to log and apply to memtable.  We can release the lock
    // during this phase since &w is currently responsible for logging
    // and protects against concurrent loggers and concurrent writes
    // into mem_.
1984 1985
    {
      mutex_.Unlock();
1986 1987 1988 1989
      if (options.disableWAL) {
        flush_on_destroy_ = true;
      }

H
heyongqiang 已提交
1990 1991 1992
      if (!options.disableWAL) {
        status = log_->AddRecord(WriteBatchInternal::Contents(updates));
        if (status.ok() && options.sync) {
1993
          if (options_.use_fsync) {
1994
            status = log_->file()->Fsync();
1995
          } else {
1996
            status = log_->file()->Sync();
1997
          }
H
heyongqiang 已提交
1998
        }
1999 2000 2001 2002 2003
      }
      if (status.ok()) {
        status = WriteBatchInternal::InsertInto(updates, mem_);
      }
      mutex_.Lock();
J
jorlow@chromium.org 已提交
2004
    }
2005
    last_flushed_sequence_ = current_sequence;
2006
    if (updates == &tmp_batch_) tmp_batch_.Clear();
2007 2008

    versions_->SetLastSequence(last_sequence);
J
jorlow@chromium.org 已提交
2009
  }
2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025

  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
    }
    if (ready == last_writer) break;
  }

  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
  }
J
jorlow@chromium.org 已提交
2026 2027 2028
  return status;
}

2029
// REQUIRES: Writer list must be non-empty
2030
// REQUIRES: First writer must have a non-nullptr batch
2031 2032 2033 2034
WriteBatch* DBImpl::BuildBatchGroup(Writer** last_writer) {
  assert(!writers_.empty());
  Writer* first = writers_.front();
  WriteBatch* result = first->batch;
2035
  assert(result != nullptr);
2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056

  size_t size = WriteBatchInternal::ByteSize(first->batch);

  // Allow the group to grow up to a maximum size, but if the
  // original write is small, limit the growth so we do not slow
  // down the small write too much.
  size_t max_size = 1 << 20;
  if (size <= (128<<10)) {
    max_size = size + (128<<10);
  }

  *last_writer = first;
  std::deque<Writer*>::iterator iter = writers_.begin();
  ++iter;  // Advance past "first"
  for (; iter != writers_.end(); ++iter) {
    Writer* w = *iter;
    if (w->sync && !first->sync) {
      // Do not include a sync write into a batch handled by a non-sync write.
      break;
    }

H
heyongqiang 已提交
2057 2058 2059 2060 2061 2062
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

2063
    if (w->batch != nullptr) {
2064 2065 2066 2067 2068 2069 2070 2071 2072
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

      // Append to *reuslt
      if (result == first->batch) {
        // Switch to temporary batch instead of disturbing caller's batch
2073
        result = &tmp_batch_;
2074 2075 2076 2077 2078 2079 2080 2081 2082 2083
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

2084
// REQUIRES: mutex_ is held
2085
// REQUIRES: this thread is currently at the front of the writer queue
2086 2087
Status DBImpl::MakeRoomForWrite(bool force) {
  mutex_.AssertHeld();
2088
  assert(!writers_.empty());
2089
  bool allow_delay = !force;
2090 2091
  bool allow_rate_limit_delay = !force;
  uint64_t rate_limit_delay_millis = 0;
2092
  Status s;
2093
  double score;
2094

2095 2096 2097 2098 2099
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
2100 2101
    } else if (
        allow_delay &&
2102
        versions_->NumLevelFiles(0) >=
2103
          options_.level0_slowdown_writes_trigger) {
2104 2105 2106 2107 2108 2109 2110
      // We are getting close to hitting a hard limit on the number of
      // L0 files.  Rather than delaying a single write by several
      // seconds when we hit the hard limit, start delaying each
      // individual write by 1ms to reduce latency variance.  Also,
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
      mutex_.Unlock();
2111
      uint64_t t1 = env_->NowMicros();
2112
      env_->SleepForMicroseconds(1000);
2113 2114
      uint64_t delayed = env_->NowMicros() - t1;
      stall_level0_slowdown_ += delayed;
2115
      allow_delay = false;  // Do not delay a single write more than once
2116 2117
      //Log(options_.info_log,
      //    "delaying write %llu usecs for level0_slowdown_writes_trigger\n",
2118
      //     (long long unsigned int)delayed);
2119
      mutex_.Lock();
2120
      delayed_writes_++;
2121 2122 2123
    } else if (!force &&
               (mem_->ApproximateMemoryUsage() <= options_.write_buffer_size)) {
      // There is room in current memtable
2124 2125 2126
      if (allow_delay) {
        DelayLoggingAndReset();
      }
2127
      break;
2128
    } else if (imm_.size() == options_.max_write_buffer_number - 1) {
2129
      // We have filled up the current memtable, but the previous
2130 2131
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
2132
      Log(options_.info_log, "wait for memtable compaction...\n");
M
Mark Callaghan 已提交
2133
      uint64_t t1 = env_->NowMicros();
H
hans@chromium.org 已提交
2134
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2135
      stall_memtable_compaction_ += env_->NowMicros() - t1;
2136
    } else if (versions_->NumLevelFiles(0) >=
2137
               options_.level0_stop_writes_trigger) {
2138
      // There are too many level-0 files.
2139
      DelayLoggingAndReset();
M
Mark Callaghan 已提交
2140
      uint64_t t1 = env_->NowMicros();
2141
      Log(options_.info_log, "wait for fewer level0 files...\n");
H
hans@chromium.org 已提交
2142
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2143
      stall_level0_num_files_ += env_->NowMicros() - t1;
2144
    } else if (
2145
        allow_rate_limit_delay &&
2146 2147 2148
        options_.rate_limit > 1.0 &&
        (score = versions_->MaxCompactionScore()) > options_.rate_limit) {
      // Delay a write when the compaction score for any level is too large.
2149
      int max_level = versions_->MaxCompactionScoreLevel();
2150
      mutex_.Unlock();
2151
      uint64_t t1 = env_->NowMicros();
2152
      env_->SleepForMicroseconds(1000);
2153
      uint64_t delayed = env_->NowMicros() - t1;
2154 2155 2156
      stall_leveln_slowdown_[max_level] += delayed;
      // Make sure the following value doesn't round to zero.
      rate_limit_delay_millis += std::max((delayed / 1000), (uint64_t) 1);
2157 2158
      if (rate_limit_delay_millis >=
          (unsigned)options_.rate_limit_delay_milliseconds) {
2159 2160 2161 2162 2163
        allow_rate_limit_delay = false;
      }
      // Log(options_.info_log,
      //    "delaying write %llu usecs for rate limits with max score %.2f\n",
      //    (long long unsigned int)delayed, score);
2164
      mutex_.Lock();
2165 2166
    } else {
      // Attempt to switch to a new memtable and trigger compaction of old
2167
      DelayLoggingAndReset();
2168 2169
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
2170
      unique_ptr<WritableFile> lfile;
2171 2172 2173 2174 2175 2176 2177
      StorageOptions soptions(storage_options_);
      soptions.DisableMmapWrites();
      s = env_->NewWritableFile(
            LogFileName(dbname_, new_log_number),
            &lfile,
            soptions
          );
2178
      if (!s.ok()) {
H
heyongqiang 已提交
2179
        // Avoid chewing through file number space in a tight loop.
2180
        versions_->ReuseFileNumber(new_log_number);
2181 2182
        break;
      }
2183 2184 2185
      // Our final size should be less than write_buffer_size
      // (compression, etc) but err on the side of caution.
      lfile->SetPreallocationBlockSize(1.1 * options_.write_buffer_size);
2186
      logfile_number_ = new_log_number;
2187
      log_.reset(new log::Writer(std::move(lfile)));
2188 2189
      imm_.Add(mem_);
      mem_ = new MemTable(internal_comparator_, NumberLevels());
2190
      mem_->Ref();
2191 2192 2193 2194 2195 2196 2197 2198 2199 2200
      force = false;   // Do not force another compaction if have room
      MaybeScheduleCompaction();
    }
  }
  return s;
}

bool DBImpl::GetProperty(const Slice& property, std::string* value) {
  value->clear();

J
jorlow@chromium.org 已提交
2201 2202 2203 2204 2205 2206 2207 2208 2209 2210
  MutexLock l(&mutex_);
  Slice in = property;
  Slice prefix("leveldb.");
  if (!in.starts_with(prefix)) return false;
  in.remove_prefix(prefix.size());

  if (in.starts_with("num-files-at-level")) {
    in.remove_prefix(strlen("num-files-at-level"));
    uint64_t level;
    bool ok = ConsumeDecimalNumber(&in, &level) && in.empty();
2211
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
2212 2213
      return false;
    } else {
2214
      char buf[100];
D
dgrogan@chromium.org 已提交
2215 2216
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
2217
      *value = buf;
J
jorlow@chromium.org 已提交
2218 2219
      return true;
    }
2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236
  } else if (in == "levelstats") {
    char buf[1000];
    snprintf(buf, sizeof(buf),
             "Level Files Size(MB)\n"
             "--------------------\n");
    value->append(buf);

    for (int level = 0; level < NumberLevels(); level++) {
      snprintf(buf, sizeof(buf),
               "%3d %8d %8.0f\n",
               level,
               versions_->NumLevelFiles(level),
               versions_->NumLevelBytes(level) / 1048576.0);
      value->append(buf);
    }
    return true;

2237
  } else if (in == "stats") {
M
Mark Callaghan 已提交
2238 2239 2240 2241
    char buf[1000];
    uint64_t total_bytes = 0;
    uint64_t micros_up = env_->NowMicros() - started_at_;
    double seconds_up = micros_up / 1000000.0;
2242
    uint64_t total_slowdown = 0;
M
Mark Callaghan 已提交
2243 2244

    // Pardon the long line but I think it is easier to read this way.
2245 2246
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
2247 2248
             "Level  Files Size(MB) Time(sec)  Read(MB) Write(MB)    Rn(MB)  Rnp1(MB)  Wnew(MB) Amplify Read(MB/s) Write(MB/s)      Rn     Rnp1     Wnp1     NewW    Count  Ln-stall\n"
             "----------------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
2249 2250
             );
    value->append(buf);
2251
    for (int level = 0; level < NumberLevels(); level++) {
2252 2253
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
2254 2255 2256 2257 2258 2259 2260 2261 2262 2263
        int64_t bytes_read = stats_[level].bytes_readn +
                             stats_[level].bytes_readnp1;
        int64_t bytes_new = stats_[level].bytes_written -
                            stats_[level].bytes_readnp1;
        double amplify = (stats_[level].bytes_readn == 0)
            ? 0.0
            : (stats_[level].bytes_written + stats_[level].bytes_readnp1) /
                (double) stats_[level].bytes_readn;

        total_bytes += bytes_read + stats_[level].bytes_written;
2264 2265
        snprintf(
            buf, sizeof(buf),
2266
            "%3d %8d %8.0f %9.0f %9.0f %9.0f %9.0f %9.0f %9.0f %7.1f %9.1f %11.1f %8d %8d %8d %8d %8d %9.1f\n",
2267 2268 2269 2270
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
2271 2272 2273 2274 2275 2276
            bytes_read / 1048576.0,
            stats_[level].bytes_written / 1048576.0,
            stats_[level].bytes_readn / 1048576.0,
            stats_[level].bytes_readnp1 / 1048576.0,
            bytes_new / 1048576.0,
            amplify,
2277 2278
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
2279
            (stats_[level].bytes_written / 1048576.0) /
2280
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
2281 2282 2283 2284
            stats_[level].files_in_leveln,
            stats_[level].files_in_levelnp1,
            stats_[level].files_out_levelnp1,
            stats_[level].files_out_levelnp1 - stats_[level].files_in_levelnp1,
2285 2286 2287
            stats_[level].count,
            stall_leveln_slowdown_[level] / 1000000.0);
        total_slowdown += stall_leveln_slowdown_[level];
2288 2289 2290
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
2291 2292

    snprintf(buf, sizeof(buf),
M
Mark Callaghan 已提交
2293
             "Amplification: %.1f rate, %.2f GB in, %.2f GB out, %.2f MB/sec in, %.2f MB/sec out\n",
M
Mark Callaghan 已提交
2294 2295
             (double) total_bytes / stats_[0].bytes_written,
             stats_[0].bytes_written / (1048576.0 * 1024),
M
Mark Callaghan 已提交
2296 2297 2298
             total_bytes / (1048576.0 * 1024),
             stats_[0].bytes_written / 1048576.0 / seconds_up,
             total_bytes / 1048576.0 / seconds_up);
M
Mark Callaghan 已提交
2299 2300 2301 2302 2303 2304 2305
    value->append(buf);

    snprintf(buf, sizeof(buf), "Uptime(secs): %.1f\n", seconds_up);
    value->append(buf);

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
2306
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
2307 2308
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
2309
            stall_memtable_compaction_ / 1000000.0,
2310
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
2311 2312
    value->append(buf);

2313
    return true;
G
Gabor Cselle 已提交
2314 2315 2316
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
J
jorlow@chromium.org 已提交
2317
  }
2318

J
jorlow@chromium.org 已提交
2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347
  return false;
}

void DBImpl::GetApproximateSizes(
    const Range* range, int n,
    uint64_t* sizes) {
  // TODO(opt): better implementation
  Version* v;
  {
    MutexLock l(&mutex_);
    versions_->current()->Ref();
    v = versions_->current();
  }

  for (int i = 0; i < n; i++) {
    // Convert user_key into a corresponding internal key.
    InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
    uint64_t start = versions_->ApproximateOffsetOf(v, k1);
    uint64_t limit = versions_->ApproximateOffsetOf(v, k2);
    sizes[i] = (limit >= start ? limit - start : 0);
  }

  {
    MutexLock l(&mutex_);
    v->Unref();
  }
}

2348 2349 2350 2351 2352 2353 2354
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

J
jorlow@chromium.org 已提交
2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
Status DB::Put(const WriteOptions& opt, const Slice& key, const Slice& value) {
  WriteBatch batch;
  batch.Put(key, value);
  return Write(opt, &batch);
}

Status DB::Delete(const WriteOptions& opt, const Slice& key) {
  WriteBatch batch;
  batch.Delete(key);
  return Write(opt, &batch);
}

DB::~DB() { }

Status DB::Open(const Options& options, const std::string& dbname,
                DB** dbptr) {
2373
  *dbptr = nullptr;
2374
  StorageOptions soptions;
J
jorlow@chromium.org 已提交
2375

2376
  if (options.block_cache != nullptr && options.no_block_cache) {
2377
    return Status::InvalidArgument(
2378
        "no_block_cache is true while block_cache is not nullptr");
2379
  }
J
jorlow@chromium.org 已提交
2380
  DBImpl* impl = new DBImpl(options, dbname);
2381 2382 2383 2384 2385
  Status s = impl->CreateArchivalDirectory();
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
2386
  impl->mutex_.Lock();
2387
  VersionEdit edit(impl->NumberLevels());
2388
  s = impl->Recover(&edit); // Handles create_if_missing, error_if_exists
J
jorlow@chromium.org 已提交
2389
  if (s.ok()) {
2390
    uint64_t new_log_number = impl->versions_->NewFileNumber();
2391
    unique_ptr<WritableFile> lfile;
2392
    soptions.DisableMmapWrites();
2393
    s = options.env->NewWritableFile(LogFileName(dbname, new_log_number),
2394
                                     &lfile, soptions);
J
jorlow@chromium.org 已提交
2395
    if (s.ok()) {
2396
      lfile->SetPreallocationBlockSize(1.1 * options.write_buffer_size);
2397
      edit.SetLogNumber(new_log_number);
2398
      impl->logfile_number_ = new_log_number;
2399
      impl->log_.reset(new log::Writer(std::move(lfile)));
2400
      s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
J
jorlow@chromium.org 已提交
2401 2402 2403
    }
    if (s.ok()) {
      impl->DeleteObsoleteFiles();
2404
      impl->MaybeScheduleCompaction();
2405
      impl->MaybeScheduleLogDBDeployStats();
J
jorlow@chromium.org 已提交
2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416
    }
  }
  impl->mutex_.Unlock();
  if (s.ok()) {
    *dbptr = impl;
  } else {
    delete impl;
  }
  return s;
}

2417 2418 2419
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2420 2421 2422
Status DestroyDB(const std::string& dbname, const Options& options) {
  Env* env = options.env;
  std::vector<std::string> filenames;
2423 2424
  std::vector<std::string> archiveFiles;

J
jorlow@chromium.org 已提交
2425 2426
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2427 2428
  env->GetChildren(ArchivalDirectory(dbname), &archiveFiles);

J
jorlow@chromium.org 已提交
2429 2430 2431 2432 2433
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
2434 2435
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2436 2437 2438
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
2439
    for (size_t i = 0; i < filenames.size(); i++) {
2440
      if (ParseFileName(filenames[i], &number, &type) &&
2441
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
2442 2443 2444 2445 2446 2447
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
2448 2449 2450 2451 2452
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464

    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
      ParseFileName(archiveFiles[i], &number, &type);
      if (type == kLogFile) {
        Status del = env->DeleteFile(ArchivalDirectory(dbname) + "/" +
                                     archiveFiles[i]);
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2465 2466
    // ignore case where no archival directory is present.
    env->DeleteDir(ArchivalDirectory(dbname));
2467

J
jorlow@chromium.org 已提交
2468
    env->UnlockFile(lock);  // Ignore error since state is already gone
2469
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2470 2471 2472 2473 2474
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
  }
  return result;
}

2475 2476
//
// A global method that can dump out the build version
2477 2478
void dumpLeveldbBuildVersion(Logger * log) {
  Log(log, "Git sha %s", leveldb_build_git_sha);
2479 2480
  Log(log, "Compile time %s %s",
      leveldb_build_compile_time, leveldb_build_compile_date);
2481 2482
}

H
Hans Wennborg 已提交
2483
}  // namespace leveldb