db_impl.cc 81.3 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_(new WriteBatch),
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
      delayed_writes_(0),
165 166
      last_flushed_sequence_(0),
      storage_options_(options) {
167

168
  mem_->Ref();
169

H
heyongqiang 已提交
170
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
171
  stats_ = new CompactionStats[options.num_levels];
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 189 190 191 192
#ifdef USE_SCRIBE
  logger_ = new ScribeLogger("localhost", 1456);
#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();
222
  delete tmp_batch_;
223
  delete[] stats_;
J
jorlow@chromium.org 已提交
224

225
  delete logger_;
J
jorlow@chromium.org 已提交
226 227
}

A
Abhishek Kona 已提交
228
// Do not flush and close database elegantly. Simulate a crash.
229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244
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.
245
  if (db_lock_ != nullptr) {
246 247
    env_->UnlockFile(db_lock_);
  }
248 249 250 251

  log_.reset();
  versions_.reset();
  table_cache_.reset();
252 253
}

A
Abhishek Kona 已提交
254 255 256
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
257

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

  const std::string manifest = DescriptorFileName(dbname_, 1);
266
  unique_ptr<WritableFile> file;
267
  Status s = env_->NewWritableFile(manifest, &file, storage_options_);
J
jorlow@chromium.org 已提交
268 269 270
  if (!s.ok()) {
    return s;
  }
271
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
272
  {
273
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290
    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 {
291
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
292 293 294 295
    *s = Status::OK();
  }
}

296 297
const Status DBImpl::CreateArchivalDirectory() {
  if (options_.WAL_ttl_seconds > 0) {
298
    std::string archivalPath = ArchivalDirectory(dbname_);
299 300 301 302 303
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

D
Dhruba Borthakur 已提交
304 305 306 307 308
// 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();

309 310 311 312 313
  // if deletion is disabled, do nothing
  if (disable_delete_obsolete_files_) {
    return;
  }

314 315 316 317 318
  // 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();
319
    if (delete_obsolete_files_last_run_ +
320 321 322 323 324 325
        options_.delete_obsolete_files_period_micros > now_micros) {
      return;
    }
    delete_obsolete_files_last_run_ = now_micros;
  }

J
jorlow@chromium.org 已提交
326
  // Make a set of all of the live files
D
Dhruba Borthakur 已提交
327 328 329 330 331
  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 已提交
332

D
Dhruba Borthakur 已提交
333 334 335 336 337 338 339 340 341 342 343
  // 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 已提交
344 345
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
346
  std::vector<std::string> old_log_files;
347

D
Dhruba Borthakur 已提交
348 349
  for (size_t i = 0; i < state.allfiles.size(); i++) {
    if (ParseFileName(state.allfiles[i], &number, &type)) {
J
jorlow@chromium.org 已提交
350 351 352
      bool keep = true;
      switch (type) {
        case kLogFile:
D
Dhruba Borthakur 已提交
353 354
          keep = ((number >= state.lognumber) ||
                  (number == state.prevlognumber));
J
jorlow@chromium.org 已提交
355 356 357 358
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
D
Dhruba Borthakur 已提交
359
          keep = (number >= state.filenumber);
J
jorlow@chromium.org 已提交
360 361
          break;
        case kTableFile:
D
Dhruba Borthakur 已提交
362
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
363 364 365 366
          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 已提交
367
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
368
          break;
H
heyongqiang 已提交
369 370 371
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
D
Dhruba Borthakur 已提交
372
            old_log_files.push_back(state.allfiles[i]);
H
heyongqiang 已提交
373 374
          }
          break;
J
jorlow@chromium.org 已提交
375 376
        case kCurrentFile:
        case kDBLockFile:
K
Kosie van der Merwe 已提交
377
        case kMetaDatabase:
J
jorlow@chromium.org 已提交
378 379 380 381 382 383
          keep = true;
          break;
      }

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

  // Delete old log files.
K
Kai Liu 已提交
411 412 413 414 415
  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 已提交
416
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
417
    size_t end = old_log_file_count - options_.keep_log_file_num;
418
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
419
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
420 421
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
422 423 424
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
J
jorlow@chromium.org 已提交
425 426
}

D
Dhruba Borthakur 已提交
427 428 429 430 431 432 433 434 435 436 437 438
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);
439 440 441 442 443 444
  PurgeObsoleteWALFiles();
}

void DBImpl::PurgeObsoleteWALFiles() {
  if (options_.WAL_ttl_seconds != ULONG_MAX && options_.WAL_ttl_seconds > 0) {
    std::vector<std::string> WALFiles;
445
    std::string archivalDir = ArchivalDirectory(dbname_);
446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469
    env_->GetChildren(archivalDir, &WALFiles);
    int64_t currentTime;
    const Status status = env_->GetCurrentTime(&currentTime);
    assert(status.ok());
    for (std::vector<std::string>::iterator it = WALFiles.begin();
         it != WALFiles.end();
         ++it) {

    uint64_t fileMTime;
    const std::string filePath = archivalDir + "/" + *it;
    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());
        }
      }
    } // Ignore errors.
    }
  }
D
Dhruba Borthakur 已提交
470 471
}

472 473 474
// 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 已提交
475
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
476 477
  mutex_.AssertHeld();

478
  assert(db_lock_ == nullptr);
479
  if (!external_table) {
480 481 482 483 484 485 486 487 488 489 490 491 492
    // 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_);
493 494 495
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
496

497 498 499 500 501 502 503 504 505
    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 已提交
506 507
      }
    } else {
508 509 510 511
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
512 513 514
    }
  }

515
  Status s = versions_->Recover();
J
jorlow@chromium.org 已提交
516 517
  if (s.ok()) {
    SequenceNumber max_sequence(0);
518 519 520 521 522 523 524 525 526 527 528 529 530 531

    // 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;
532
    }
533 534 535 536 537 538 539 540 541
    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 已提交
542
    }
543

H
heyongqiang 已提交
544 545 546 547 548 549
    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");
    }

550 551 552
    // 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++) {
553
      s = RecoverLogFile(logs[i], edit, &max_sequence, external_table);
554 555 556 557
      // 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]);
558
    }
559 560 561
    // This could be the last_flushed_sequence as the next sequences will be
    // greater than this.
    last_flushed_sequence_ = max_sequence;
562

J
jorlow@chromium.org 已提交
563
    if (s.ok()) {
564 565 566
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
      }
J
jorlow@chromium.org 已提交
567 568 569 570 571 572 573 574
    }
  }

  return s;
}

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

  mutex_.AssertHeld();

  // Open the log file
  std::string fname = LogFileName(dbname_, log_number);
594
  unique_ptr<SequentialFile> file;
595
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
596 597 598 599 600 601 602 603
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
604
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
605
  reporter.fname = fname.c_str();
606
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
J
jorlow@chromium.org 已提交
607 608 609 610
  // 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).
611
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
612
                     0/*initial_offset*/);
613
  Log(options_.info_log, "Recovering log #%llu",
J
jorlow@chromium.org 已提交
614 615 616 617 618 619
      (unsigned long long) log_number);

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
620
  MemTable* mem = nullptr;
621 622 623
  if (external_table) {
    mem = external_table;
  }
J
jorlow@chromium.org 已提交
624 625 626 627 628 629 630 631 632
  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);

633
    if (mem == nullptr) {
634
      mem = new MemTable(internal_comparator_, NumberLevels());
635
      mem->Ref();
J
jorlow@chromium.org 已提交
636 637 638 639 640 641 642 643 644 645 646 647 648
    }
    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;
    }

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

662
  if (status.ok() && mem != nullptr && !external_table) {
663
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
664 665 666 667
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

668
  if (mem != nullptr && !external_table) mem->Unref();
J
jorlow@chromium.org 已提交
669 670 671
  return status;
}

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

  Status s;
  {
    mutex_.Unlock();
688 689
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
690 691
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
692 693 694
    mutex_.Lock();
  }

695
  Log(options_.info_log, "Level-0 table #%llu: %lld bytes %s",
J
jorlow@chromium.org 已提交
696 697 698 699
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;
700

701
  pending_outputs_.erase(meta.number);
702 703 704 705 706 707 708 709 710

  // 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);
  }

711 712 713
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
714
  stats.files_out_levelnp1 = 1;
715
  stats_[level].Add(stats);
J
jorlow@chromium.org 已提交
716 717 718
  return s;
}

719 720 721

Status DBImpl::WriteLevel0Table(MemTable* mem, VersionEdit* edit,
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
722
  mutex_.AssertHeld();
723 724 725 726 727 728
  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();
729 730 731
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
732 733
  Log(options_.info_log, "Level-0 flush table #%llu: started",
      (unsigned long long) meta.number);
J
jorlow@chromium.org 已提交
734

735 736
  Version* base = versions_->current();
  base->Ref();
737 738 739
  Status s;
  {
    mutex_.Unlock();
740 741
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
742 743
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
744 745
    mutex_.Lock();
  }
746 747
  base->Unref();

748 749 750 751 752 753 754 755 756 757 758 759 760 761
  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 已提交
762
  // committed so that other threads can recognize this file as a
763 764 765 766 767 768 769 770 771 772 773 774 775
  // 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.
776
    if (base != nullptr && options_.max_background_compactions <= 1) {
777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803
      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();
804
  if (m == nullptr) {
805 806 807 808 809 810 811 812 813 814 815 816
    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);

817 818 819
  if (s.ok() && shutting_down_.Acquire_Load()) {
    s = Status::IOError("Deleting DB during memtable compaction");
  }
J
jorlow@chromium.org 已提交
820

821
  // Replace immutable memtable with the generated Table
822 823 824
  s = imm_.InstallMemtableFlushResults(
    m, versions_.get(), s, &mutex_, options_.info_log.get(),
    file_number, pending_outputs_);
J
jorlow@chromium.org 已提交
825 826

  if (s.ok()) {
827 828 829
    if (madeProgress) {
      *madeProgress = 1;
    }
830
    MaybeScheduleLogDBDeployStats();
D
Dhruba Borthakur 已提交
831 832 833
    // 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 已提交
834 835 836 837
  }
  return s;
}

G
Gabor Cselle 已提交
838 839 840 841 842
void DBImpl::CompactRange(const Slice* begin, const Slice* end) {
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->current();
843
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
844 845 846 847 848 849 850 851 852 853 854
      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);
  }
}

855
int DBImpl::NumberLevels() {
856
  return options_.num_levels;
857 858 859
}

int DBImpl::MaxMemCompactionLevel() {
860
  return options_.max_mem_compaction_level;
861 862 863
}

int DBImpl::Level0StopWriteTrigger() {
864
  return options_.level0_stop_writes_trigger;
865 866
}

H
heyongqiang 已提交
867 868 869 870 871
Status DBImpl::Flush(const FlushOptions& options) {
  Status status = FlushMemTable(options);
  return status;
}

872 873 874 875
SequenceNumber DBImpl::GetLatestSequenceNumber() {
  return versions_->LastSequence();
}

876
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
877
                               unique_ptr<TransactionLogIterator>* iter) {
878 879 880 881 882 883 884 885 886 887 888 889 890

  //  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.
891 892 893 894 895 896
  std::string archivedir = ArchivalDirectory(dbname_);
  if (env_->FileExists(archivedir)) {
    s = ListAllWALFiles(archivedir, &walFiles, kArchivedLogFile);
    if (!s.ok()) {
      return s;
    }
897 898 899 900 901 902 903 904
  }

  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>();
905 906 907
  s = FindProbableWALFiles(&walFiles, probableWALFiles, seq);
  if (!s.ok()) {
    return s;
908
  }
909
  iter->reset(
910 911
    new TransactionLogIteratorImpl(dbname_,
                                   &options_,
912
                                   storage_options_,
913 914 915 916
                                   seq,
                                   probableWALFiles,
                                   &last_flushed_sequence_));
  iter->get()->Next();
917 918 919 920 921 922
  return Status::OK();
}

Status DBImpl::FindProbableWALFiles(std::vector<LogFile>* const allLogs,
                                    std::vector<LogFile>* const result,
                                    const SequenceNumber target) {
923 924
  assert(allLogs != nullptr);
  assert(result != nullptr);
925 926

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

964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981
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;
}

982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009
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;
1010
    Status* status;  // nullptr if options_.paranoid_checks==false
1011 1012
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
1013
          (this->status == nullptr ? "(ignoring error) " : ""),
1014
          fname, static_cast<int>(bytes), s.ToString().c_str());
1015
      if (this->status != nullptr && this->status->ok()) *this->status = s;
1016 1017 1018
    }
  };

1019
  unique_ptr<SequentialFile> file;
1020
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
1021 1022 1023 1024 1025 1026 1027 1028

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


  LogReporter reporter;
  reporter.env = env_;
1029
  reporter.info_log = options_.info_log.get();
1030
  reporter.fname = fname.c_str();
1031
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
1032
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051
                     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) {
1052
  assert(logFiles != nullptr);
1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
  std::vector<std::string> allFiles;
  const Status status = env_->GetChildren(path, &allFiles);
  if (!status.ok()) {
    return status;
  }
  for(std::vector<std::string>::iterator it = allFiles.begin();
      it != allFiles.end();
      ++it) {
    uint64_t number;
    FileType type;
    if (ParseFileName(*it, &number, &type) && type == kLogFile){
      logFiles->push_back(LogFile(number, logType));
    }
  }
  return status;
}

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

G
Gabor Cselle 已提交
1073 1074
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1075 1076
  ManualCompaction manual;
  manual.level = level;
G
Gabor Cselle 已提交
1077
  manual.done = false;
1078
  manual.in_progress = false;
1079 1080
  if (begin == nullptr) {
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1081 1082 1083 1084
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1085 1086
  if (end == nullptr) {
    manual.end = nullptr;
G
Gabor Cselle 已提交
1087 1088 1089 1090 1091 1092
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1093

A
Abhishek Kona 已提交
1094 1095 1096 1097
  // 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
1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
  // 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 已提交
1111
  while (!manual.done) {
1112
    while (manual_compaction_ != nullptr) {
G
Gabor Cselle 已提交
1113 1114 1115
      bg_cv_.Wait();
    }
    manual_compaction_ = &manual;
1116 1117 1118
    if (bg_compaction_scheduled_ == LargeNumber) {
      bg_compaction_scheduled_ = newvalue;
    }
G
Gabor Cselle 已提交
1119 1120 1121 1122
    MaybeScheduleCompaction();
    while (manual_compaction_ == &manual) {
      bg_cv_.Wait();
    }
H
hans@chromium.org 已提交
1123
  }
1124 1125 1126 1127 1128 1129 1130 1131 1132
  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 已提交
1133 1134
}

H
heyongqiang 已提交
1135
Status DBImpl::FlushMemTable(const FlushOptions& options) {
1136 1137
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1138
  if (s.ok() && options.wait) {
1139
    // Wait until the compaction completes
H
heyongqiang 已提交
1140
    s = WaitForCompactMemTable();
1141 1142
  }
  return s;
J
jorlow@chromium.org 已提交
1143 1144
}

H
heyongqiang 已提交
1145
Status DBImpl::WaitForCompactMemTable() {
1146 1147 1148
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1149
  while (imm_.size() > 0 && bg_error_.ok()) {
1150 1151
    bg_cv_.Wait();
  }
1152
  if (imm_.size() != 0) {
1153 1154 1155
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1156 1157 1158 1159 1160 1161
}

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

1162
Status DBImpl::TEST_WaitForCompactMemTable() {
1163
  return WaitForCompactMemTable();
1164 1165 1166
}

Status DBImpl::TEST_WaitForCompact() {
1167 1168 1169 1170 1171 1172
  // Wait until the compaction completes
  MutexLock l(&mutex_);
  while (bg_compaction_scheduled_ && bg_error_.ok()) {
    bg_cv_.Wait();
  }
  return bg_error_;
1173 1174
}

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

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

void DBImpl::BackgroundCall() {
1196
  bool madeProgress = false;
D
Dhruba Borthakur 已提交
1197
  DeletionState deletion_state;
J
jorlow@chromium.org 已提交
1198
  MutexLock l(&mutex_);
1199
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1200
  assert(bg_compaction_scheduled_);
H
hans@chromium.org 已提交
1201
  if (!shutting_down_.Acquire_Load()) {
1202
    Status s = BackgroundCompaction(&madeProgress, deletion_state);
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
    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 已提交
1215
  }
1216

D
Dhruba Borthakur 已提交
1217 1218 1219 1220
  // delete unnecessary files if any, this is done outside the mutex
  if (!deletion_state.live.empty()) {
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
D
Dhruba Borthakur 已提交
1221
    EvictObsoleteFiles(deletion_state);
1222
    mutex_.Lock();
D
Dhruba Borthakur 已提交
1223 1224
  }

1225
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1226

1227 1228
  MaybeScheduleLogDBDeployStats();

J
jorlow@chromium.org 已提交
1229
  // Previous compaction may have produced too many files in a level,
A
Abhishek Kona 已提交
1230
  // So reschedule another compaction if we made progress in the
1231 1232 1233 1234
  // last compaction.
  if (madeProgress) {
    MaybeScheduleCompaction();
  }
H
hans@chromium.org 已提交
1235
  bg_cv_.SignalAll();
J
jorlow@chromium.org 已提交
1236 1237
}

A
Abhishek Kona 已提交
1238
Status DBImpl::BackgroundCompaction(bool* madeProgress,
1239
  DeletionState& deletion_state) {
1240
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1241
  mutex_.AssertHeld();
1242

1243
  while (imm_.IsFlushPending()) {
A
Abhishek Kona 已提交
1244
    Log(options_.info_log,
1245 1246 1247 1248 1249 1250
        "BackgroundCompaction doing CompactMemTable, compaction slots available %d",
        options_.max_background_compactions - bg_compaction_scheduled_);
    Status stat = CompactMemTable(madeProgress);
    if (!stat.ok()) {
      return stat;
    }
1251 1252
  }

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

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

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
1315
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1316 1317 1318 1319 1320
        "Compaction error: %s", status.ToString().c_str());
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
1321 1322

  if (is_manual) {
G
Gabor Cselle 已提交
1323
    ManualCompaction* m = manual_compaction_;
1324 1325 1326
    if (!status.ok()) {
      m->done = true;
    }
G
Gabor Cselle 已提交
1327 1328 1329 1330 1331 1332
    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;
    }
1333
    m->in_progress = false; // not being processed anymore
1334
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1335
  }
1336
  return status;
J
jorlow@chromium.org 已提交
1337 1338 1339 1340
}

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

1355 1356 1357 1358 1359
// 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();
1360 1361
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
1362
  int filesNeeded = compact->compaction->num_input_files(1);
1363
  for (int i = 0; i < filesNeeded; i++) {
1364 1365 1366 1367 1368 1369 1370 1371 1372
    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();
A
Abhishek Kona 已提交
1373
  for (std::list<uint64_t>::iterator it =
1374 1375 1376 1377 1378 1379 1380 1381
       compact->allocated_file_numbers.begin();
       it != compact->allocated_file_numbers.end(); ++it) {
    uint64_t file_number = *it;
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
1382
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
1383 1384
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
1385
  uint64_t file_number;
1386 1387 1388 1389 1390 1391 1392
  // 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 已提交
1393 1394 1395 1396 1397
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
1398 1399 1400 1401 1402
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
1403 1404 1405

  // Make the output file
  std::string fname = TableFileName(dbname_, file_number);
1406
  Status s = env_->NewWritableFile(fname, &compact->outfile, storage_options_);
1407 1408 1409 1410 1411 1412

  // 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 已提交
1413
  if (s.ok()) {
1414 1415
    compact->builder.reset(new TableBuilder(options_, compact->outfile.get(),
                                            compact->compaction->level() + 1));
J
jorlow@chromium.org 已提交
1416 1417 1418 1419 1420 1421
  }
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
1422
  assert(compact != nullptr);
1423
  assert(compact->outfile);
1424
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439

  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;
1440
  compact->builder.reset();
J
jorlow@chromium.org 已提交
1441 1442

  // Finish and check for file errors
1443
  if (s.ok() && !options_.disableDataSync) {
1444 1445 1446 1447 1448
    if (options_.use_fsync) {
      s = compact->outfile->Fsync();
    } else {
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
1449 1450 1451 1452
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
1453
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
1454 1455 1456

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
1457
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
1458
                                               storage_options_,
J
jorlow@chromium.org 已提交
1459 1460
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
1461 1462 1463
    s = iter->status();
    delete iter;
    if (s.ok()) {
1464
      Log(options_.info_log,
J
jorlow@chromium.org 已提交
1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476
          "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();
1477 1478 1479 1480 1481

  // 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.
1482
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
1483 1484 1485 1486 1487 1488 1489 1490
    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");
  }

1491
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
1492 1493 1494 1495 1496 1497 1498 1499 1500
      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 已提交
1501
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1502 1503 1504 1505 1506
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
        level + 1,
        out.number, out.file_size, out.smallest, out.largest);
  }
1507
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
1508 1509
}

1510 1511 1512 1513 1514 1515 1516 1517 1518
//
// 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) {
1519
  SequenceNumber prev __attribute__((unused)) = 0;
1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534
  for (std::vector<SequenceNumber>::iterator it = snapshots.begin();
       it < snapshots.end(); it++) {
    assert (prev <= *it);
    if (*it >= in) {
      return *it;
    }
    assert(prev = *it); // assignment
  }
  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 已提交
1535
Status DBImpl::DoCompactionWork(CompactionState* compact) {
1536 1537
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions

A
Abhishek Kona 已提交
1538
  Log(options_.info_log,
1539
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
1540 1541 1542
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
1543
      compact->compaction->level() + 1,
1544
      compact->compaction->score(),
1545
      options_.max_background_compactions - bg_compaction_scheduled_);
1546 1547
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
1548
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
1549 1550

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
1551
  assert(compact->builder == nullptr);
1552
  assert(!compact->outfile);
1553 1554 1555 1556 1557 1558 1559 1560

  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 已提交
1561
  } else {
1562 1563 1564 1565
    // 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 已提交
1566 1567
  }

1568 1569 1570 1571 1572 1573 1574 1575 1576 1577
  // 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;
    }
  }

1578 1579 1580
  // Allocate the output file numbers before we release the lock
  AllocateCompactionOutputFileNumbers(compact);

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

1584
  const uint64_t start_micros = env_->NowMicros();
1585
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
1586 1587 1588 1589 1590
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
1591 1592
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
1593
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1594
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
1595
    // Prioritize immutable compaction work
1596
    if (imm_.imm_flush_needed.NoBarrier_Load() != nullptr) {
1597 1598
      const uint64_t imm_start = env_->NowMicros();
      mutex_.Lock();
1599
      if (imm_.IsFlushPending()) {
1600
        CompactMemTable();
H
hans@chromium.org 已提交
1601
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
1602 1603 1604 1605 1606
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
1607
    Slice key = input->key();
1608
    Slice value = input->value();
1609
    Slice* compaction_filter_value = nullptr;
1610
    if (compact->compaction->ShouldStopBefore(key) &&
1611
        compact->builder != nullptr) {
1612
      status = FinishCompactionOutputFile(compact, input.get());
1613 1614 1615 1616 1617 1618
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
1619 1620 1621 1622 1623 1624
    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;
1625
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1626 1627 1628 1629 1630 1631 1632 1633
    } 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;
1634
        visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1635 1636
      }

1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647
      // 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 已提交
1648
        // Hidden by an newer entry for same user key
1649
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
1650
        drop = true;    // (A)
1651
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
1652
      } else if (ikey.type == kTypeDeletion &&
1653
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
1654 1655 1656 1657 1658 1659 1660 1661 1662
                 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;
1663
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_OBSOLETE);
1664
      } else if (options_.CompactionFilter != nullptr &&
1665
                 ikey.type != kTypeDeletion &&
1666
                 ikey.sequence < earliest_snapshot) {
1667 1668
        // If the user has specified a compaction filter, then invoke
        // it. If this key is not visible via any snapshot and the
1669 1670
        // return value of the compaction filter is true and then
        // drop this key from the output.
1671 1672
        drop = options_.CompactionFilter(options_.compaction_filter_args,
                         compact->compaction->level(),
1673 1674
                         ikey.user_key, value, &compaction_filter_value);

1675 1676 1677
        if (drop) {
          RecordTick(options_.statistics, COMPACTION_KEY_DROP_USER);
        }
1678
        // If the application wants to change the value, then do so here.
1679
        if (compaction_filter_value != nullptr) {
1680 1681 1682
          value = *compaction_filter_value;
          delete compaction_filter_value;
        }
J
jorlow@chromium.org 已提交
1683 1684 1685
      }

      last_sequence_for_key = ikey.sequence;
1686
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
1687 1688
    }
#if 0
1689
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1690
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
1691
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
1692
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
1693
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
1694
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
1695 1696
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
1697 1698 1699
#endif

    if (!drop) {
1700

1701 1702 1703
      char* kptr = (char*)key.data();
      std::string kstr;

1704 1705 1706 1707 1708 1709
      // 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);
1710 1711 1712 1713 1714
        // 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);
1715
      }
1716

1717 1718 1719
      Slice newkey(kptr, key.size());
      assert((key.clear(), 1)); // we do not need 'key' anymore

J
jorlow@chromium.org 已提交
1720
      // Open output file if necessary
1721
      if (compact->builder == nullptr) {
J
jorlow@chromium.org 已提交
1722 1723 1724 1725 1726 1727
        status = OpenCompactionOutputFile(compact);
        if (!status.ok()) {
          break;
        }
      }
      if (compact->builder->NumEntries() == 0) {
1728
        compact->current_output()->smallest.DecodeFrom(newkey);
J
jorlow@chromium.org 已提交
1729
      }
1730 1731
      compact->current_output()->largest.DecodeFrom(newkey);
      compact->builder->Add(newkey, value);
J
jorlow@chromium.org 已提交
1732 1733 1734 1735

      // Close output file if it is big enough
      if (compact->builder->FileSize() >=
          compact->compaction->MaxOutputFileSize()) {
1736
        status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748
        if (!status.ok()) {
          break;
        }
      }
    }

    input->Next();
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
    status = Status::IOError("Deleting DB during compaction");
  }
1749
  if (status.ok() && compact->builder != nullptr) {
1750
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
1751 1752 1753 1754
  }
  if (status.ok()) {
    status = input->status();
  }
1755
  input.reset();
J
jorlow@chromium.org 已提交
1756

1757 1758
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
A
Abhishek Kona 已提交
1759 1760 1761
  if (options_.statistics) {
    options_.statistics->measureTime(COMPACTION_TIME, stats.micros);
  }
M
Mark Callaghan 已提交
1762 1763
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
1764 1765

  int num_output_files = compact->outputs.size();
1766
  if (compact->builder != nullptr) {
1767 1768 1769 1770 1771
    // 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 已提交
1772 1773 1774 1775 1776 1777 1778

  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;

1779
  for (int i = 0; i < num_output_files; i++) {
1780 1781 1782
    stats.bytes_written += compact->outputs[i].file_size;
  }

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

1786 1787 1788 1789
  // 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 已提交
1790 1791 1792
  if (status.ok()) {
    status = InstallCompactionResults(compact);
  }
1793
  VersionSet::LevelSummaryStorage tmp;
1794
  Log(options_.info_log,
M
Mark Callaghan 已提交
1795
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
1796
      "MB in(%.1f, %.1f) out(%.1f), amplify(%.1f) %s\n",
M
Mark Callaghan 已提交
1797 1798 1799 1800 1801 1802 1803 1804 1805
      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) /
1806 1807
          (double) stats.bytes_readn,
      status.ToString().c_str());
M
Mark Callaghan 已提交
1808

J
jorlow@chromium.org 已提交
1809 1810 1811
  return status;
}

1812 1813 1814 1815
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
1816
  std::vector<MemTable*> mem; // includes both mem_ and imm_
1817 1818 1819 1820 1821
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
  state->mu->Lock();
1822 1823 1824
  for (unsigned int i = 0; i < state->mem.size(); i++) {
    state->mem[i]->Unref();
  }
1825 1826 1827 1828
  state->version->Unref();
  state->mu->Unlock();
  delete state;
}
H
Hans Wennborg 已提交
1829
}  // namespace
1830

J
jorlow@chromium.org 已提交
1831 1832
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
1833
  IterState* cleanup = new IterState;
J
jorlow@chromium.org 已提交
1834
  mutex_.Lock();
1835
  *latest_snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1836

1837
  // Collect together all needed child iterators for mem
J
jorlow@chromium.org 已提交
1838
  std::vector<Iterator*> list;
1839
  mem_->Ref();
1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850
  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);
1851
  }
1852 1853

  // Collect iterators for files in L0 - Ln
1854
  versions_->current()->AddIterators(options, storage_options_, &list);
J
jorlow@chromium.org 已提交
1855 1856 1857
  Iterator* internal_iter =
      NewMergingIterator(&internal_comparator_, &list[0], list.size());
  versions_->current()->Ref();
1858 1859 1860

  cleanup->mu = &mutex_;
  cleanup->version = versions_->current();
1861
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
1862 1863 1864 1865 1866 1867 1868 1869 1870 1871

  mutex_.Unlock();
  return internal_iter;
}

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

J
jorlow@chromium.org 已提交
1872
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
1873 1874 1875 1876
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
1877 1878 1879
Status DBImpl::Get(const ReadOptions& options,
                   const Slice& key,
                   std::string* value) {
1880
  Status s;
1881

1882 1883 1884
  std::unique_ptr<StopWatch> sw = stats::StartStopWatch(env_,
                                                        options_.statistics,
                                                        DB_GET);
1885
  SequenceNumber snapshot;
1886
  MutexLock l(&mutex_);
1887
  if (options.snapshot != nullptr) {
1888 1889 1890
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1891
  }
1892

1893
  MemTable* mem = mem_;
1894
  MemTableList imm = imm_;
1895
  Version* current = versions_->current();
1896
  mem->Ref();
1897
  imm.RefAll();
1898
  current->Ref();
1899

1900 1901 1902
  // Unlock while reading from files and memtables

  mutex_.Unlock();
1903
  bool have_stat_update = false;
1904
  Version::GetStats stats;
1905 1906 1907 1908 1909 1910 1911 1912 1913
  // 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;
1914
  }
1915
  mutex_.Lock();
1916

1917 1918
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
1919 1920
    MaybeScheduleCompaction();
  }
1921
  mem->Unref();
1922
  imm.UnrefAll();
1923
  current->Unref();
1924
  RecordTick(options_.statistics, NUMBER_KEYS_READ);
1925
  RecordTick(options_.statistics, BYTES_READ, value->size());
1926
  return s;
J
jorlow@chromium.org 已提交
1927 1928 1929 1930 1931
}

Iterator* DBImpl::NewIterator(const ReadOptions& options) {
  SequenceNumber latest_snapshot;
  Iterator* internal_iter = NewInternalIterator(options, &latest_snapshot);
1932 1933
  return NewDBIterator(
      &dbname_, env_, user_comparator(), internal_iter,
1934
      (options.snapshot != nullptr
1935 1936
       ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
       : latest_snapshot));
J
jorlow@chromium.org 已提交
1937 1938 1939 1940
}

const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
1941
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
1942 1943 1944 1945
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
1946
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957
}

// 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);
}

1958 1959 1960 1961
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
1962
  w.disableWAL = options.disableWAL;
1963
  w.done = false;
1964

1965 1966 1967
  std::unique_ptr<StopWatch> sw = stats::StartStopWatch(env_,
                                                        options_.statistics,
                                                        DB_WRITE);
1968
  MutexLock l(&mutex_);
1969 1970 1971 1972 1973 1974 1975 1976 1977
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
  if (w.done) {
    return w.status;
  }

  // May temporarily unlock and wait.
1978
  Status status = MakeRoomForWrite(my_batch == nullptr);
D
dgrogan@chromium.org 已提交
1979
  uint64_t last_sequence = versions_->LastSequence();
1980
  Writer* last_writer = &w;
1981
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
1982
    WriteBatch* updates = BuildBatchGroup(&last_writer);
1983 1984
    const SequenceNumber current_sequence = last_sequence + 1;
    WriteBatchInternal::SetSequence(updates, current_sequence);
1985 1986 1987 1988
    int my_batch_count = WriteBatchInternal::Count(updates);
    last_sequence += my_batch_count;
    // Record statistics
    RecordTick(options_.statistics, NUMBER_KEYS_WRITTEN, my_batch_count);
1989 1990 1991
    RecordTick(options_.statistics,
               BYTES_WRITTEN,
               WriteBatchInternal::ByteSize(updates));
1992 1993 1994 1995
    // 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_.
1996 1997
    {
      mutex_.Unlock();
1998 1999 2000 2001
      if (options.disableWAL) {
        flush_on_destroy_ = true;
      }

H
heyongqiang 已提交
2002 2003 2004
      if (!options.disableWAL) {
        status = log_->AddRecord(WriteBatchInternal::Contents(updates));
        if (status.ok() && options.sync) {
2005
          if (options_.use_fsync) {
2006
            status = log_->file()->Fsync();
2007
          } else {
2008
            status = log_->file()->Sync();
2009
          }
H
heyongqiang 已提交
2010
        }
2011 2012 2013 2014 2015
      }
      if (status.ok()) {
        status = WriteBatchInternal::InsertInto(updates, mem_);
      }
      mutex_.Lock();
J
jorlow@chromium.org 已提交
2016
    }
2017
    last_flushed_sequence_ = current_sequence;
2018
    if (updates == tmp_batch_) tmp_batch_->Clear();
2019 2020

    versions_->SetLastSequence(last_sequence);
J
jorlow@chromium.org 已提交
2021
  }
2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037

  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 已提交
2038 2039 2040
  return status;
}

2041
// REQUIRES: Writer list must be non-empty
2042
// REQUIRES: First writer must have a non-nullptr batch
2043 2044 2045 2046
WriteBatch* DBImpl::BuildBatchGroup(Writer** last_writer) {
  assert(!writers_.empty());
  Writer* first = writers_.front();
  WriteBatch* result = first->batch;
2047
  assert(result != nullptr);
2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068

  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 已提交
2069 2070 2071 2072 2073 2074
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

2075
    if (w->batch != nullptr) {
2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095
      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
        result = tmp_batch_;
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

2096
// REQUIRES: mutex_ is held
2097
// REQUIRES: this thread is currently at the front of the writer queue
2098 2099
Status DBImpl::MakeRoomForWrite(bool force) {
  mutex_.AssertHeld();
2100
  assert(!writers_.empty());
2101
  bool allow_delay = !force;
2102 2103
  bool allow_rate_limit_delay = !force;
  uint64_t rate_limit_delay_millis = 0;
2104
  Status s;
2105
  double score;
2106

2107 2108 2109 2110 2111
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
2112 2113
    } else if (
        allow_delay &&
2114
        versions_->NumLevelFiles(0) >=
2115
          options_.level0_slowdown_writes_trigger) {
2116 2117 2118 2119 2120 2121 2122
      // 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();
2123
      uint64_t t1 = env_->NowMicros();
2124
      env_->SleepForMicroseconds(1000);
2125 2126
      uint64_t delayed = env_->NowMicros() - t1;
      stall_level0_slowdown_ += delayed;
2127
      allow_delay = false;  // Do not delay a single write more than once
2128 2129
      //Log(options_.info_log,
      //    "delaying write %llu usecs for level0_slowdown_writes_trigger\n",
2130
      //     (long long unsigned int)delayed);
2131
      mutex_.Lock();
2132
      delayed_writes_++;
2133 2134 2135
    } else if (!force &&
               (mem_->ApproximateMemoryUsage() <= options_.write_buffer_size)) {
      // There is room in current memtable
2136 2137 2138
      if (allow_delay) {
        DelayLoggingAndReset();
      }
2139
      break;
2140
    } else if (imm_.size() == options_.max_write_buffer_number - 1) {
2141
      // We have filled up the current memtable, but the previous
2142 2143
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
2144
      Log(options_.info_log, "wait for memtable compaction...\n");
M
Mark Callaghan 已提交
2145
      uint64_t t1 = env_->NowMicros();
H
hans@chromium.org 已提交
2146
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2147
      stall_memtable_compaction_ += env_->NowMicros() - t1;
2148
    } else if (versions_->NumLevelFiles(0) >=
2149
               options_.level0_stop_writes_trigger) {
2150
      // There are too many level-0 files.
2151
      DelayLoggingAndReset();
M
Mark Callaghan 已提交
2152
      uint64_t t1 = env_->NowMicros();
2153
      Log(options_.info_log, "wait for fewer level0 files...\n");
H
hans@chromium.org 已提交
2154
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2155
      stall_level0_num_files_ += env_->NowMicros() - t1;
2156
    } else if (
2157
        allow_rate_limit_delay &&
2158 2159 2160
        options_.rate_limit > 1.0 &&
        (score = versions_->MaxCompactionScore()) > options_.rate_limit) {
      // Delay a write when the compaction score for any level is too large.
2161
      int max_level = versions_->MaxCompactionScoreLevel();
2162
      mutex_.Unlock();
2163
      uint64_t t1 = env_->NowMicros();
2164
      env_->SleepForMicroseconds(1000);
2165
      uint64_t delayed = env_->NowMicros() - t1;
2166 2167 2168
      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);
2169 2170
      if (rate_limit_delay_millis >=
          (unsigned)options_.rate_limit_delay_milliseconds) {
2171 2172 2173 2174 2175
        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);
2176
      mutex_.Lock();
2177 2178
    } else {
      // Attempt to switch to a new memtable and trigger compaction of old
2179
      DelayLoggingAndReset();
2180 2181
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
2182
      unique_ptr<WritableFile> lfile;
2183 2184 2185 2186 2187 2188 2189
      StorageOptions soptions(storage_options_);
      soptions.DisableMmapWrites();
      s = env_->NewWritableFile(
            LogFileName(dbname_, new_log_number),
            &lfile,
            soptions
          );
2190
      if (!s.ok()) {
H
heyongqiang 已提交
2191
        // Avoid chewing through file number space in a tight loop.
2192
        versions_->ReuseFileNumber(new_log_number);
2193 2194
        break;
      }
2195 2196 2197
      // 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);
2198
      logfile_number_ = new_log_number;
2199
      log_.reset(new log::Writer(std::move(lfile)));
2200 2201
      imm_.Add(mem_);
      mem_ = new MemTable(internal_comparator_, NumberLevels());
2202
      mem_->Ref();
2203 2204 2205 2206 2207 2208 2209 2210 2211 2212
      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 已提交
2213 2214 2215 2216 2217 2218 2219 2220 2221 2222
  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();
2223
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
2224 2225
      return false;
    } else {
2226
      char buf[100];
D
dgrogan@chromium.org 已提交
2227 2228
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
2229
      *value = buf;
J
jorlow@chromium.org 已提交
2230 2231
      return true;
    }
2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248
  } 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;

2249
  } else if (in == "stats") {
M
Mark Callaghan 已提交
2250 2251 2252 2253
    char buf[1000];
    uint64_t total_bytes = 0;
    uint64_t micros_up = env_->NowMicros() - started_at_;
    double seconds_up = micros_up / 1000000.0;
2254
    uint64_t total_slowdown = 0;
M
Mark Callaghan 已提交
2255 2256

    // Pardon the long line but I think it is easier to read this way.
2257 2258
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
2259 2260
             "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"
2261 2262
             );
    value->append(buf);
2263
    for (int level = 0; level < NumberLevels(); level++) {
2264 2265
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
2266 2267 2268 2269 2270 2271 2272 2273 2274 2275
        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;
2276 2277
        snprintf(
            buf, sizeof(buf),
2278
            "%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",
2279 2280 2281 2282
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
2283 2284 2285 2286 2287 2288
            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,
2289 2290
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
2291
            (stats_[level].bytes_written / 1048576.0) /
2292
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
2293 2294 2295 2296
            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,
2297 2298 2299
            stats_[level].count,
            stall_leveln_slowdown_[level] / 1000000.0);
        total_slowdown += stall_leveln_slowdown_[level];
2300 2301 2302
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
2303 2304

    snprintf(buf, sizeof(buf),
M
Mark Callaghan 已提交
2305
             "Amplification: %.1f rate, %.2f GB in, %.2f GB out, %.2f MB/sec in, %.2f MB/sec out\n",
M
Mark Callaghan 已提交
2306 2307
             (double) total_bytes / stats_[0].bytes_written,
             stats_[0].bytes_written / (1048576.0 * 1024),
M
Mark Callaghan 已提交
2308 2309 2310
             total_bytes / (1048576.0 * 1024),
             stats_[0].bytes_written / 1048576.0 / seconds_up,
             total_bytes / 1048576.0 / seconds_up);
M
Mark Callaghan 已提交
2311 2312 2313 2314 2315 2316 2317
    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, "
2318
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
2319 2320
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
2321
            stall_memtable_compaction_ / 1000000.0,
2322
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
2323 2324
    value->append(buf);

2325
    return true;
G
Gabor Cselle 已提交
2326 2327 2328
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
J
jorlow@chromium.org 已提交
2329
  }
2330

J
jorlow@chromium.org 已提交
2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359
  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();
  }
}

2360 2361 2362 2363 2364 2365 2366
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 已提交
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384
// 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) {
2385
  *dbptr = nullptr;
2386
  StorageOptions soptions;
J
jorlow@chromium.org 已提交
2387

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

2429 2430 2431
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2432 2433 2434
Status DestroyDB(const std::string& dbname, const Options& options) {
  Env* env = options.env;
  std::vector<std::string> filenames;
2435 2436
  std::vector<std::string> archiveFiles;

J
jorlow@chromium.org 已提交
2437 2438
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2439 2440
  env->GetChildren(ArchivalDirectory(dbname), &archiveFiles);

J
jorlow@chromium.org 已提交
2441 2442 2443 2444 2445
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
2446 2447
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2448 2449 2450
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
2451
    for (size_t i = 0; i < filenames.size(); i++) {
2452
      if (ParseFileName(filenames[i], &number, &type) &&
2453
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
2454 2455 2456 2457 2458 2459
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
2460 2461 2462 2463 2464
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476

    // 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;
        }
      }
    }
2477 2478
    // ignore case where no archival directory is present.
    env->DeleteDir(ArchivalDirectory(dbname));
2479

J
jorlow@chromium.org 已提交
2480
    env->UnlockFile(lock);  // Ignore error since state is already gone
2481
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2482 2483 2484 2485 2486
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
  }
  return result;
}

2487 2488
//
// A global method that can dump out the build version
2489 2490
void dumpLeveldbBuildVersion(Logger * log) {
  Log(log, "Git sha %s", leveldb_build_git_sha);
2491 2492
  Log(log, "Compile time %s %s",
      leveldb_build_compile_time, leveldb_build_compile_date);
2493 2494
}

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