db_impl.cc 169.8 KB
Newer Older
1 2 3 4 5
//  Copyright (c) 2013, Facebook, Inc.  All rights reserved.
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
//
J
jorlow@chromium.org 已提交
6 7 8 9 10 11
// 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"

I
Igor Canadi 已提交
12 13
#define __STDC_FORMAT_MACROS
#include <inttypes.h>
J
jorlow@chromium.org 已提交
14
#include <algorithm>
15 16
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
17
#include <set>
18
#include <stdexcept>
19 20
#include <stdint.h>
#include <string>
21
#include <unordered_set>
22
#include <unordered_map>
T
Tomislav Novak 已提交
23
#include <utility>
24
#include <vector>
25

J
jorlow@chromium.org 已提交
26
#include "db/builder.h"
27
#include "db/db_iter.h"
K
kailiu 已提交
28
#include "db/dbformat.h"
J
jorlow@chromium.org 已提交
29 30 31 32
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
K
kailiu 已提交
33
#include "db/memtable_list.h"
34
#include "db/merge_context.h"
35
#include "db/merge_helper.h"
J
jorlow@chromium.org 已提交
36
#include "db/table_cache.h"
K
kailiu 已提交
37
#include "db/table_properties_collector.h"
T
Tomislav Novak 已提交
38
#include "db/tailing_iter.h"
L
Lei Jin 已提交
39
#include "db/forward_iterator.h"
40
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
41 42
#include "db/version_set.h"
#include "db/write_batch_internal.h"
43
#include "port/port.h"
I
Igor Canadi 已提交
44
#include "rocksdb/cache.h"
45
#include "port/likely.h"
46 47 48 49 50 51
#include "rocksdb/compaction_filter.h"
#include "rocksdb/db.h"
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
#include "rocksdb/statistics.h"
#include "rocksdb/status.h"
S
Siying Dong 已提交
52
#include "rocksdb/table.h"
J
jorlow@chromium.org 已提交
53
#include "table/block.h"
54
#include "table/block_based_table_factory.h"
J
jorlow@chromium.org 已提交
55
#include "table/merger.h"
K
kailiu 已提交
56
#include "table/table_builder.h"
J
jorlow@chromium.org 已提交
57
#include "table/two_level_iterator.h"
58
#include "util/auto_roll_logger.h"
K
kailiu 已提交
59
#include "util/autovector.h"
60
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
61
#include "util/coding.h"
I
Igor Canadi 已提交
62
#include "util/hash_skiplist_rep.h"
63
#include "util/hash_linklist_rep.h"
J
jorlow@chromium.org 已提交
64
#include "util/logging.h"
H
Haobo Xu 已提交
65
#include "util/log_buffer.h"
J
jorlow@chromium.org 已提交
66
#include "util/mutexlock.h"
67
#include "util/perf_context_imp.h"
68
#include "util/stop_watch.h"
69
#include "util/sync_point.h"
J
jorlow@chromium.org 已提交
70

71
namespace rocksdb {
J
jorlow@chromium.org 已提交
72

73
const std::string kDefaultColumnFamilyName("default");
74

K
kailiu 已提交
75
void DumpLeveldbBuildVersion(Logger * log);
76

77 78 79 80 81
// Information kept for every waiting writer
struct DBImpl::Writer {
  Status status;
  WriteBatch* batch;
  bool sync;
H
heyongqiang 已提交
82
  bool disableWAL;
83
  bool in_batch_group;
84
  bool done;
85
  uint64_t timeout_hint_us;
86 87 88 89 90
  port::CondVar cv;

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

J
jorlow@chromium.org 已提交
91 92 93
struct DBImpl::CompactionState {
  Compaction* const compaction;

94 95 96 97 98
  // 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 已提交
99 100 101 102

  // Files produced by compaction
  struct Output {
    uint64_t number;
103
    uint32_t path_id;
J
jorlow@chromium.org 已提交
104 105
    uint64_t file_size;
    InternalKey smallest, largest;
106
    SequenceNumber smallest_seqno, largest_seqno;
J
jorlow@chromium.org 已提交
107 108
  };
  std::vector<Output> outputs;
109
  std::list<uint64_t> allocated_file_numbers;
J
jorlow@chromium.org 已提交
110 111

  // State kept for output being generated
112 113
  unique_ptr<WritableFile> outfile;
  unique_ptr<TableBuilder> builder;
J
jorlow@chromium.org 已提交
114 115 116 117 118 119 120 121 122

  uint64_t total_bytes;

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

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

124 125 126 127 128 129 130 131
  // Create a client visible context of this compaction
  CompactionFilter::Context GetFilterContextV1() {
    CompactionFilter::Context context;
    context.is_full_compaction = compaction->IsFullCompaction();
    context.is_manual_compaction = compaction->IsManualCompaction();
    return context;
  }

132
  // Create a client visible context of this compaction
D
Danny Guo 已提交
133 134
  CompactionFilterContext GetFilterContext() {
    CompactionFilterContext context;
135
    context.is_full_compaction = compaction->IsFullCompaction();
136
    context.is_manual_compaction = compaction->IsManualCompaction();
137 138
    return context;
  }
D
Danny Guo 已提交
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255

  std::vector<Slice> key_buf_;
  std::vector<Slice> existing_value_buf_;
  std::vector<std::string> key_str_buf_;
  std::vector<std::string> existing_value_str_buf_;
  // new_value_buf_ will only be appended if a value changes
  std::vector<std::string> new_value_buf_;
  // if values_changed_buf_[i] is true
  // new_value_buf_ will add a new entry with the changed value
  std::vector<bool> value_changed_buf_;
  // to_delete_buf_[i] is true iff key_buf_[i] is deleted
  std::vector<bool> to_delete_buf_;
  // buffer for the parsed internal keys, the string buffer is backed
  // by key_str_buf_
  std::vector<ParsedInternalKey> ikey_buf_;

  std::vector<Slice> other_key_buf_;
  std::vector<Slice> other_value_buf_;
  std::vector<std::string> other_key_str_buf_;
  std::vector<std::string> other_value_str_buf_;

  std::vector<Slice> combined_key_buf_;
  std::vector<Slice> combined_value_buf_;

  std::string cur_prefix_;

  // Buffers the kv-pair that will be run through compaction filter V2
  // in the future.
  void BufferKeyValueSlices(const Slice& key, const Slice& value) {
    key_str_buf_.emplace_back(key.ToString());
    existing_value_str_buf_.emplace_back(value.ToString());
    key_buf_.emplace_back(Slice(key_str_buf_.back()));
    existing_value_buf_.emplace_back(Slice(existing_value_str_buf_.back()));

    ParsedInternalKey ikey;
    ParseInternalKey(key_buf_.back(), &ikey);
    ikey_buf_.emplace_back(ikey);
  }

  // Buffers the kv-pair that will not be run through compaction filter V2
  // in the future.
  void BufferOtherKeyValueSlices(const Slice& key, const Slice& value) {
    other_key_str_buf_.emplace_back(key.ToString());
    other_value_str_buf_.emplace_back(value.ToString());
    other_key_buf_.emplace_back(Slice(other_key_str_buf_.back()));
    other_value_buf_.emplace_back(Slice(other_value_str_buf_.back()));
  }

  // Add a kv-pair to the combined buffer
  void AddToCombinedKeyValueSlices(const Slice& key, const Slice& value) {
    // The real strings are stored in the batch buffers
    combined_key_buf_.emplace_back(key);
    combined_value_buf_.emplace_back(value);
  }

  // Merging the two buffers
  void MergeKeyValueSliceBuffer(const InternalKeyComparator* comparator) {
    size_t i = 0;
    size_t j = 0;
    size_t total_size = key_buf_.size() + other_key_buf_.size();
    combined_key_buf_.reserve(total_size);
    combined_value_buf_.reserve(total_size);

    while (i + j < total_size) {
      int comp_res = 0;
      if (i < key_buf_.size() && j < other_key_buf_.size()) {
        comp_res = comparator->Compare(key_buf_[i], other_key_buf_[j]);
      } else if (i >= key_buf_.size() && j < other_key_buf_.size()) {
        comp_res = 1;
      } else if (j >= other_key_buf_.size() && i < key_buf_.size()) {
        comp_res = -1;
      }
      if (comp_res > 0) {
        AddToCombinedKeyValueSlices(other_key_buf_[j], other_value_buf_[j]);
        j++;
      } else if (comp_res < 0) {
        AddToCombinedKeyValueSlices(key_buf_[i], existing_value_buf_[i]);
        i++;
      }
    }
  }

  void CleanupBatchBuffer() {
    to_delete_buf_.clear();
    key_buf_.clear();
    existing_value_buf_.clear();
    key_str_buf_.clear();
    existing_value_str_buf_.clear();
    new_value_buf_.clear();
    value_changed_buf_.clear();
    ikey_buf_.clear();

    to_delete_buf_.shrink_to_fit();
    key_buf_.shrink_to_fit();
    existing_value_buf_.shrink_to_fit();
    key_str_buf_.shrink_to_fit();
    existing_value_str_buf_.shrink_to_fit();
    new_value_buf_.shrink_to_fit();
    value_changed_buf_.shrink_to_fit();
    ikey_buf_.shrink_to_fit();

    other_key_buf_.clear();
    other_value_buf_.clear();
    other_key_str_buf_.clear();
    other_value_str_buf_.clear();
    other_key_buf_.shrink_to_fit();
    other_value_buf_.shrink_to_fit();
    other_key_str_buf_.shrink_to_fit();
    other_value_str_buf_.shrink_to_fit();
  }

  void CleanupMergedBuffer() {
    combined_key_buf_.clear();
    combined_value_buf_.clear();
    combined_key_buf_.shrink_to_fit();
    combined_value_buf_.shrink_to_fit();
  }
J
jorlow@chromium.org 已提交
256 257
};

258
namespace {
J
jorlow@chromium.org 已提交
259
// Fix user-supplied options to be reasonable
260
template <class T, class V>
J
jorlow@chromium.org 已提交
261
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
D
dgrogan@chromium.org 已提交
262 263
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
J
jorlow@chromium.org 已提交
264
}
265 266
}  // anonymous namespace

J
jorlow@chromium.org 已提交
267 268
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
269
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
270
                        const Options& src) {
271 272 273 274 275 276 277
  auto db_options = SanitizeOptions(dbname, DBOptions(src));
  auto cf_options = SanitizeOptions(icmp, ipolicy, ColumnFamilyOptions(src));
  return Options(db_options, cf_options);
}

DBOptions SanitizeOptions(const std::string& dbname, const DBOptions& src) {
  DBOptions result = src;
278 279
  // result.max_open_files means an "infinite" open files.
  if (result.max_open_files != -1) {
280
    ClipToRange(&result.max_open_files, 20, 1000000);
281
  }
282

283
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
284 285
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
286 287
    if (!s.ok()) {
      // No place suitable for logging
288
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
289 290
    }
  }
291 292 293 294 295

  if (result.wal_dir.empty()) {
    // Use dbname as default
    result.wal_dir = dbname;
  }
296
  if (result.wal_dir.back() == '/') {
I
Igor Canadi 已提交
297
    result.wal_dir = result.wal_dir.substr(0, result.wal_dir.size() - 1);
298
  }
299

300 301 302 303
  if (result.db_paths.size() == 0) {
    result.db_paths.push_back(dbname);
  }

J
jorlow@chromium.org 已提交
304 305 306
  return result;
}

307
namespace {
308 309 310 311 312 313 314
CompressionType GetCompressionFlush(const Options& options) {
  // Compressing memtable flushes might not help unless the sequential load
  // optimization is used for leveled compaction. Otherwise the CPU and
  // latency overhead is not offset by saving much space.

  bool can_compress;

315
  if (options.compaction_style == kCompactionStyleUniversal) {
316 317 318 319
    can_compress =
        (options.compaction_options_universal.compression_size_percent < 0);
  } else {
    // For leveled compress when min_level_to_compress == 0.
320 321
    can_compress = options.compression_per_level.empty() ||
                   options.compression_per_level[0] != kNoCompression;
322 323 324 325 326 327 328 329
  }

  if (can_compress) {
    return options.compression;
  } else {
    return kNoCompression;
  }
}
330
}  // namespace
331

I
Igor Canadi 已提交
332
DBImpl::DBImpl(const DBOptions& options, const std::string& dbname)
J
jorlow@chromium.org 已提交
333
    : env_(options.env),
H
heyongqiang 已提交
334
      dbname_(dbname),
I
Igor Canadi 已提交
335
      options_(SanitizeOptions(dbname, options)),
336
      db_lock_(nullptr),
H
Haobo Xu 已提交
337
      mutex_(options.use_adaptive_mutex),
338
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
339
      bg_cv_(&mutex_),
340
      logfile_number_(0),
I
Igor Canadi 已提交
341
      log_empty_(true),
342
      default_cf_handle_(nullptr),
I
Igor Canadi 已提交
343 344
      total_log_size_(0),
      max_total_in_memory_state_(0),
345
      tmp_batch_(),
346
      bg_schedule_needed_(false),
347
      bg_compaction_scheduled_(0),
348
      bg_manual_only_(0),
349
      bg_flush_scheduled_(0),
350
      bg_logstats_scheduled_(false),
351 352
      manual_compaction_(nullptr),
      logger_(nullptr),
353
      disable_delete_obsolete_files_(0),
I
Igor Canadi 已提交
354
      delete_obsolete_files_last_run_(options.env->NowMicros()),
355
      purge_wal_files_last_run_(0),
356
      last_stats_dump_time_microsec_(0),
357
      default_interval_to_delete_obsolete_WAL_(600),
358
      flush_on_destroy_(false),
359
      delayed_writes_(0),
360 361
      storage_options_(options),
      bg_work_gate_closed_(false),
362 363
      refitting_level_(false),
      opened_successfully_(false) {
H
heyongqiang 已提交
364
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
365

J
jorlow@chromium.org 已提交
366
  // Reserve ten files or so for other uses and give the rest to TableCache.
367 368
  // Give a large number for setting of "infinite" open files.
  const int table_cache_size =
369 370 371 372 373 374
      (options_.max_open_files == -1) ? 4194304 : options_.max_open_files - 10;
  // Reserve ten files or so for other uses and give the rest to TableCache.
  table_cache_ =
      NewLRUCache(table_cache_size, options_.table_cache_numshardbits,
                  options_.table_cache_remove_scan_count_limit);

I
Igor Canadi 已提交
375 376
  versions_.reset(
      new VersionSet(dbname_, &options_, storage_options_, table_cache_.get()));
377 378
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
379

K
kailiu 已提交
380
  DumpLeveldbBuildVersion(options_.info_log.get());
381
  options_.Dump(options_.info_log.get());
382

383
  char name[100];
K
kailiu 已提交
384 385
  Status s = env_->GetHostName(name, 100L);
  if (s.ok()) {
386 387 388 389 390 391
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
392

I
Igor Canadi 已提交
393
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
394 395 396
}

DBImpl::~DBImpl() {
397 398 399 400 401 402 403 404
  mutex_.Lock();
  if (flush_on_destroy_) {
    for (auto cfd : *versions_->GetColumnFamilySet()) {
      if (cfd->mem()->GetFirstSequenceNumber() != 0) {
        cfd->Ref();
        mutex_.Unlock();
        FlushMemTable(cfd, FlushOptions());
        mutex_.Lock();
405
        cfd->Unref();
406 407
      }
    }
408
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
409
  }
I
Igor Canadi 已提交
410 411

  // Wait for background work to finish
412
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
413 414 415
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
416
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
417
  }
418

I
Igor Canadi 已提交
419 420 421 422 423
  if (default_cf_handle_ != nullptr) {
    // we need to delete handle outside of lock because it does its own locking
    mutex_.Unlock();
    delete default_cf_handle_;
    mutex_.Lock();
424 425
  }

426 427 428 429 430 431 432 433 434 435 436 437 438 439 440
  if (options_.allow_thread_local) {
    // Clean up obsolete files due to SuperVersion release.
    // (1) Need to delete to obsolete files before closing because RepairDB()
    // scans all existing files in the file system and builds manifest file.
    // Keeping obsolete files confuses the repair process.
    // (2) Need to check if we Open()/Recover() the DB successfully before
    // deleting because if VersionSet recover fails (may be due to corrupted
    // manifest file), it is not able to identify live files correctly. As a
    // result, all "live" files can get deleted by accident. However, corrupted
    // manifest is recoverable by RepairDB().
    if (opened_successfully_) {
      DeletionState deletion_state;
      FindObsoleteFiles(deletion_state, true);
      // manifest number starting from 2
      deletion_state.manifest_file_number = 1;
441 442 443
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
444 445 446
    }
  }

447
  // versions need to be destroyed before table_cache since it can hold
448 449
  // references to table_cache.
  versions_.reset();
450
  mutex_.Unlock();
I
Igor Canadi 已提交
451 452 453
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
454

I
Igor Canadi 已提交
455
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
456 457 458
}

Status DBImpl::NewDB() {
459
  VersionEdit new_db;
460
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
461 462 463 464
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
465
  unique_ptr<WritableFile> file;
I
Igor Canadi 已提交
466 467
  Status s = env_->NewWritableFile(
      manifest, &file, env_->OptimizeForManifestWrite(storage_options_));
J
jorlow@chromium.org 已提交
468 469 470
  if (!s.ok()) {
    return s;
  }
471
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
472
  {
473
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
474 475 476 477 478 479
    std::string record;
    new_db.EncodeTo(&record);
    s = log.AddRecord(record);
  }
  if (s.ok()) {
    // Make "CURRENT" file that points to the new manifest file.
480
    s = SetCurrentFile(env_, dbname_, 1, db_directory_.get());
J
jorlow@chromium.org 已提交
481 482 483 484 485 486 487 488 489 490
  } else {
    env_->DeleteFile(manifest);
  }
  return s;
}

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

496
const Status DBImpl::CreateArchivalDirectory() {
497
  if (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0) {
498
    std::string archivalPath = ArchivalDirectory(options_.wal_dir);
499 500 501 502 503
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

504
void DBImpl::PrintStatistics() {
505
  auto dbstats = options_.statistics.get();
506 507
  if (dbstats) {
    Log(options_.info_log,
508 509
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
510 511 512
  }
}

513
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
514 515 516 517 518 519 520 521 522 523 524 525 526
  if (options_.stats_dump_period_sec == 0) return;

  const uint64_t now_micros = env_->NowMicros();

  if (last_stats_dump_time_microsec_ +
      options_.stats_dump_period_sec * 1000000
      <= now_micros) {
    // Multiple threads could race in here simultaneously.
    // However, the last one will update last_stats_dump_time_microsec_
    // atomically. We could see more than one dump during one dump
    // period in rare cases.
    last_stats_dump_time_microsec_ = now_micros;
    std::string stats;
527
    GetProperty("rocksdb.stats", &stats);
H
Haobo Xu 已提交
528
    Log(options_.info_log, "%s", stats.c_str());
529
    PrintStatistics();
530 531 532
  }
}

533
// Returns the list of live files in 'sst_live' and the list
K
kailiu 已提交
534
// of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
535 536 537 538 539 540 541
// no_full_scan = true -- never do the full scan using GetChildren()
// force = false -- don't force the full scan, except every
//  options_.delete_obsolete_files_period_micros
// force = true -- force the full scan
void DBImpl::FindObsoleteFiles(DeletionState& deletion_state,
                               bool force,
                               bool no_full_scan) {
D
Dhruba Borthakur 已提交
542 543
  mutex_.AssertHeld();

544
  // if deletion is disabled, do nothing
545
  if (disable_delete_obsolete_files_ > 0) {
546 547 548
    return;
  }

549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564
  bool doing_the_full_scan = false;

  // logic for figurint out if we're doing the full scan
  if (no_full_scan) {
    doing_the_full_scan = false;
  } else if (force || options_.delete_obsolete_files_period_micros == 0) {
    doing_the_full_scan = true;
  } else {
    const uint64_t now_micros = env_->NowMicros();
    if (delete_obsolete_files_last_run_ +
        options_.delete_obsolete_files_period_micros < now_micros) {
      doing_the_full_scan = true;
      delete_obsolete_files_last_run_ = now_micros;
    }
  }

I
Igor Canadi 已提交
565 566 567
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
568 569
  // store the current filenum, lognum, etc
  deletion_state.manifest_file_number = versions_->ManifestFileNumber();
570 571
  deletion_state.pending_manifest_file_number =
      versions_->PendingManifestFileNumber();
572
  deletion_state.log_number = versions_->MinLogNumber();
I
Igor Canadi 已提交
573 574
  deletion_state.prev_log_number = versions_->PrevLogNumber();

575 576 577 578 579 580 581 582
  if (!doing_the_full_scan && !deletion_state.HaveSomethingToDelete()) {
    // avoid filling up sst_live if we're sure that we
    // are not going to do the full scan and that we don't have
    // anything to delete at the moment
    return;
  }

  // don't delete live files
583 584 585 586 587
  for (auto pair : pending_outputs_) {
    deletion_state.sst_live.emplace_back(pair.first, pair.second, 0);
  }
  /*  deletion_state.sst_live.insert(pending_outputs_.begin(),
                                   pending_outputs_.end());*/
I
Igor Canadi 已提交
588 589
  versions_->AddLiveFiles(&deletion_state.sst_live);

590
  if (doing_the_full_scan) {
591 592 593 594 595 596 597 598 599
    for (uint32_t path_id = 0; path_id < options_.db_paths.size(); path_id++) {
      // set of all files in the directory. We'll exclude files that are still
      // alive in the subsequent processings.
      std::vector<std::string> files;
      env_->GetChildren(dbname_, &files);  // Ignore errors
      for (std::string file : files) {
        deletion_state.candidate_files.emplace_back(file, path_id);
      }
    }
600 601 602 603 604

    //Add log files in wal_dir
    if (options_.wal_dir != dbname_) {
      std::vector<std::string> log_files;
      env_->GetChildren(options_.wal_dir, &log_files); // Ignore errors
605 606 607
      for (std::string log_file : log_files) {
        deletion_state.candidate_files.emplace_back(log_file, 0);
      }
608
    }
609
  }
610 611
}

612 613 614 615 616 617 618 619 620 621 622 623 624
namespace {
bool CompareCandidateFile(const rocksdb::DBImpl::CandidateFileInfo& first,
                          const rocksdb::DBImpl::CandidateFileInfo& second) {
  if (first.file_name > second.file_name) {
    return true;
  } else if (first.file_name < second.file_name) {
    return false;
  } else {
    return (first.path_id > first.path_id);
  }
}
};  // namespace

D
Dhruba Borthakur 已提交
625
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
626
// belong to live files are posibly removed. Also, removes all the
627
// files in sst_delete_files and log_delete_files.
628
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
629
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
630 631
  // we'd better have sth to delete
  assert(state.HaveSomethingToDelete());
632

I
Igor Canadi 已提交
633 634 635 636
  // this checks if FindObsoleteFiles() was run before. If not, don't do
  // PurgeObsoleteFiles(). If FindObsoleteFiles() was run, we need to also
  // run PurgeObsoleteFiles(), even if disable_delete_obsolete_files_ is true
  if (state.manifest_file_number == 0) {
I
Igor Canadi 已提交
637 638
    return;
  }
639

640
  // Now, convert live list to an unordered map, WITHOUT mutex held;
641
  // set is slow.
642 643 644 645
  std::unordered_map<uint64_t, const FileDescriptor*> sst_live_map;
  for (FileDescriptor& fd : state.sst_live) {
    sst_live_map[fd.GetNumber()] = &fd;
  }
I
Igor Canadi 已提交
646

K
kailiu 已提交
647 648 649 650 651 652 653
  auto& candidate_files = state.candidate_files;
  candidate_files.reserve(
      candidate_files.size() +
      state.sst_delete_files.size() +
      state.log_delete_files.size());
  // We may ignore the dbname when generating the file names.
  const char* kDumbDbName = "";
654
  for (auto file : state.sst_delete_files) {
655 656 657
    candidate_files.emplace_back(
        MakeTableFileName(kDumbDbName, file->fd.GetNumber()),
        file->fd.GetPathId());
658
    delete file;
I
Igor Canadi 已提交
659 660
  }

K
kailiu 已提交
661 662
  for (auto file_num : state.log_delete_files) {
    if (file_num > 0) {
663 664
      candidate_files.emplace_back(LogFileName(kDumbDbName, file_num).substr(1),
                                   0);
I
Igor Canadi 已提交
665 666
    }
  }
667

K
kailiu 已提交
668
  // dedup state.candidate_files so we don't try to delete the same
I
Igor Canadi 已提交
669
  // file twice
670
  sort(candidate_files.begin(), candidate_files.end(), CompareCandidateFile);
671 672
  candidate_files.erase(unique(candidate_files.begin(), candidate_files.end()),
                        candidate_files.end());
J
jorlow@chromium.org 已提交
673

674 675
  std::vector<std::string> old_info_log_files;

676 677 678
  for (const auto& candidate_file : candidate_files) {
    std::string to_delete = candidate_file.file_name;
    uint32_t path_id = candidate_file.path_id;
K
kailiu 已提交
679 680 681 682 683 684
    uint64_t number;
    FileType type;
    // Ignore file if we cannot recognize it.
    if (!ParseFileName(to_delete, &number, &type)) {
      continue;
    }
J
jorlow@chromium.org 已提交
685

K
kailiu 已提交
686 687 688 689 690 691 692 693
    bool keep = true;
    switch (type) {
      case kLogFile:
        keep = ((number >= state.log_number) ||
                (number == state.prev_log_number));
        break;
      case kDescriptorFile:
        // Keep my manifest file, and any newer incarnations'
694
        // (can happen during manifest roll)
K
kailiu 已提交
695 696 697
        keep = (number >= state.manifest_file_number);
        break;
      case kTableFile:
698
        keep = (sst_live_map.find(number) != sst_live_map.end());
K
kailiu 已提交
699 700 701
        break;
      case kTempFile:
        // Any temp files that are currently being written to must
702 703 704 705
        // be recorded in pending_outputs_, which is inserted into "live".
        // Also, SetCurrentFile creates a temp file when writing out new
        // manifest, which is equal to state.pending_manifest_file_number. We
        // should not delete that file
706
        keep = (sst_live_map.find(number) != sst_live_map.end()) ||
707
               (number == state.pending_manifest_file_number);
K
kailiu 已提交
708 709 710 711
        break;
      case kInfoLogFile:
        keep = true;
        if (number != 0) {
712
          old_info_log_files.push_back(to_delete);
J
jorlow@chromium.org 已提交
713
        }
K
kailiu 已提交
714 715 716 717 718 719 720 721 722 723 724 725 726
        break;
      case kCurrentFile:
      case kDBLockFile:
      case kIdentityFile:
      case kMetaDatabase:
        keep = true;
        break;
    }

    if (keep) {
      continue;
    }

727
    std::string fname;
K
kailiu 已提交
728 729
    if (type == kTableFile) {
      // evict from cache
730
      TableCache::Evict(table_cache_.get(), number);
731 732 733 734
      fname = TableFileName(options_.db_paths, number, path_id);
    } else {
      fname =
          ((type == kLogFile) ? options_.wal_dir : dbname_) + "/" + to_delete;
K
kailiu 已提交
735
    }
736

K
kailiu 已提交
737 738
    if (type == kLogFile &&
        (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0)) {
739
      auto archived_log_name = ArchivedLogFileName(options_.wal_dir, number);
740 741
      // The sync point below is used in (DBTest,TransactionLogIteratorRace)
      TEST_SYNC_POINT("DBImpl::PurgeObsoleteFiles:1");
742
      Status s = env_->RenameFile(fname, archived_log_name);
743 744
      // The sync point below is used in (DBTest,TransactionLogIteratorRace)
      TEST_SYNC_POINT("DBImpl::PurgeObsoleteFiles:2");
745 746 747
      Log(options_.info_log,
          "Move log file %s to %s -- %s\n",
          fname.c_str(), archived_log_name.c_str(), s.ToString().c_str());
K
kailiu 已提交
748 749
    } else {
      Status s = env_->DeleteFile(fname);
750 751
      Log(options_.info_log, "Delete %s type=%d #%" PRIu64 " -- %s\n",
          fname.c_str(), type, number, s.ToString().c_str());
J
jorlow@chromium.org 已提交
752 753
    }
  }
H
heyongqiang 已提交
754

755
  // Delete old info log files.
756
  size_t old_info_log_file_count = old_info_log_files.size();
K
Kai Liu 已提交
757 758
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
759
  if (old_info_log_file_count >= options_.keep_log_file_num &&
K
Kai Liu 已提交
760
      options_.db_log_dir.empty()) {
761 762
    std::sort(old_info_log_files.begin(), old_info_log_files.end());
    size_t end = old_info_log_file_count - options_.keep_log_file_num;
763
    for (unsigned int i = 0; i <= end; i++) {
764 765 766 767 768 769 770
      std::string& to_delete = old_info_log_files.at(i);
      Log(options_.info_log, "Delete info log file %s\n", to_delete.c_str());
      Status s = env_->DeleteFile(dbname_ + "/" + to_delete);
      if (!s.ok()) {
        Log(options_.info_log, "Delete info log file %s FAILED -- %s\n",
            to_delete.c_str(), s.ToString().c_str());
      }
H
heyongqiang 已提交
771 772
    }
  }
773
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
774
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
775 776 777 778 779
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
780
  FindObsoleteFiles(deletion_state, true);
781 782 783
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
784 785
}

I
Igor Canadi 已提交
786
#ifndef ROCKSDB_LITE
787 788 789 790 791 792 793 794
// 1. Go through all archived files and
//    a. if ttl is enabled, delete outdated files
//    b. if archive size limit is enabled, delete empty files,
//        compute file number and size.
// 2. If size limit is enabled:
//    a. compute how many files should be deleted
//    b. get sorted non-empty archived logs
//    c. delete what should be deleted
795
void DBImpl::PurgeObsoleteWALFiles() {
796 797 798 799 800 801
  bool const ttl_enabled = options_.WAL_ttl_seconds > 0;
  bool const size_limit_enabled =  options_.WAL_size_limit_MB > 0;
  if (!ttl_enabled && !size_limit_enabled) {
    return;
  }

802 803
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
804 805 806 807 808 809 810 811
  if (!s.ok()) {
    Log(options_.info_log, "Can't get current time: %s", s.ToString().c_str());
    assert(false);
    return;
  }
  uint64_t const now_seconds = static_cast<uint64_t>(current_time);
  uint64_t const time_to_check = (ttl_enabled && !size_limit_enabled) ?
    options_.WAL_ttl_seconds / 2 : default_interval_to_delete_obsolete_WAL_;
812

813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
  if (purge_wal_files_last_run_ + time_to_check > now_seconds) {
    return;
  }

  purge_wal_files_last_run_ = now_seconds;

  std::string archival_dir = ArchivalDirectory(options_.wal_dir);
  std::vector<std::string> files;
  s = env_->GetChildren(archival_dir, &files);
  if (!s.ok()) {
    Log(options_.info_log, "Can't get archive files: %s", s.ToString().c_str());
    assert(false);
    return;
  }

  size_t log_files_num = 0;
  uint64_t log_file_size = 0;

  for (auto& f : files) {
    uint64_t number;
    FileType type;
    if (ParseFileName(f, &number, &type) && type == kLogFile) {
      std::string const file_path = archival_dir + "/" + f;
      if (ttl_enabled) {
        uint64_t file_m_time;
        Status const s = env_->GetFileModificationTime(file_path,
          &file_m_time);
        if (!s.ok()) {
          Log(options_.info_log, "Can't get file mod time: %s: %s",
              file_path.c_str(), s.ToString().c_str());
          continue;
        }
        if (now_seconds - file_m_time > options_.WAL_ttl_seconds) {
          Status const s = env_->DeleteFile(file_path);
          if (!s.ok()) {
            Log(options_.info_log, "Can't delete file: %s: %s",
                file_path.c_str(), s.ToString().c_str());
            continue;
I
Igor Canadi 已提交
851 852 853
          } else {
            MutexLock l(&read_first_record_cache_mutex_);
            read_first_record_cache_.erase(number);
854 855
          }
          continue;
856
        }
857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875
      }

      if (size_limit_enabled) {
        uint64_t file_size;
        Status const s = env_->GetFileSize(file_path, &file_size);
        if (!s.ok()) {
          Log(options_.info_log, "Can't get file size: %s: %s",
              file_path.c_str(), s.ToString().c_str());
          return;
        } else {
          if (file_size > 0) {
            log_file_size = std::max(log_file_size, file_size);
            ++log_files_num;
          } else {
            Status s = env_->DeleteFile(file_path);
            if (!s.ok()) {
              Log(options_.info_log, "Can't delete file: %s: %s",
                  file_path.c_str(), s.ToString().c_str());
              continue;
I
Igor Canadi 已提交
876 877 878
            } else {
              MutexLock l(&read_first_record_cache_mutex_);
              read_first_record_cache_.erase(number);
879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897
            }
          }
        }
      }
    }
  }

  if (0 == log_files_num || !size_limit_enabled) {
    return;
  }

  size_t const files_keep_num = options_.WAL_size_limit_MB *
    1024 * 1024 / log_file_size;
  if (log_files_num <= files_keep_num) {
    return;
  }

  size_t files_del_num = log_files_num - files_keep_num;
  VectorLogPtr archived_logs;
898
  GetSortedWalsOfType(archival_dir, archived_logs, kArchivedLogFile);
899 900 901 902 903 904 905 906 907 908 909 910 911 912

  if (files_del_num > archived_logs.size()) {
    Log(options_.info_log, "Trying to delete more archived log files than "
        "exist. Deleting all");
    files_del_num = archived_logs.size();
  }

  for (size_t i = 0; i < files_del_num; ++i) {
    std::string const file_path = archived_logs[i]->PathName();
    Status const s = DeleteFile(file_path);
    if (!s.ok()) {
      Log(options_.info_log, "Can't delete file: %s: %s",
          file_path.c_str(), s.ToString().c_str());
      continue;
I
Igor Canadi 已提交
913 914 915
    } else {
      MutexLock l(&read_first_record_cache_mutex_);
      read_first_record_cache_.erase(archived_logs[i]->LogNumber());
916 917
    }
  }
D
Dhruba Borthakur 已提交
918 919
}

920 921 922 923 924 925 926 927 928 929
namespace {
struct CompareLogByPointer {
  bool operator()(const unique_ptr<LogFile>& a, const unique_ptr<LogFile>& b) {
    LogFileImpl* a_impl = dynamic_cast<LogFileImpl*>(a.get());
    LogFileImpl* b_impl = dynamic_cast<LogFileImpl*>(b.get());
    return *a_impl < *b_impl;
  }
};
}

I
Igor Canadi 已提交
930 931 932 933 934 935 936 937 938 939 940 941 942
Status DBImpl::GetSortedWalsOfType(const std::string& path,
                                   VectorLogPtr& log_files,
                                   WalFileType log_type) {
  std::vector<std::string> all_files;
  const Status status = env_->GetChildren(path, &all_files);
  if (!status.ok()) {
    return status;
  }
  log_files.reserve(all_files.size());
  for (const auto& f : all_files) {
    uint64_t number;
    FileType type;
    if (ParseFileName(f, &number, &type) && type == kLogFile) {
I
Igor Canadi 已提交
943 944
      SequenceNumber sequence;
      Status s = ReadFirstRecord(log_type, number, &sequence);
I
Igor Canadi 已提交
945 946 947
      if (!s.ok()) {
        return s;
      }
I
Igor Canadi 已提交
948 949 950 951
      if (sequence == 0) {
        // empty file
        continue;
      }
I
Igor Canadi 已提交
952

953 954 955 956 957 958
      // Reproduce the race condition where a log file is moved
      // to archived dir, between these two sync points, used in
      // (DBTest,TransactionLogIteratorRace)
      TEST_SYNC_POINT("DBImpl::GetSortedWalsOfType:1");
      TEST_SYNC_POINT("DBImpl::GetSortedWalsOfType:2");

I
Igor Canadi 已提交
959 960
      uint64_t size_bytes;
      s = env_->GetFileSize(LogFileName(path, number), &size_bytes);
961 962 963 964 965
      // re-try in case the alive log file has been moved to archive.
      if (!s.ok() && log_type == kAliveLogFile &&
          env_->FileExists(ArchivedLogFileName(path, number))) {
        s = env_->GetFileSize(ArchivedLogFileName(path, number), &size_bytes);
      }
I
Igor Canadi 已提交
966 967 968 969 970
      if (!s.ok()) {
        return s;
      }

      log_files.push_back(std::move(unique_ptr<LogFile>(
I
Igor Canadi 已提交
971
          new LogFileImpl(number, log_type, sequence, size_bytes))));
I
Igor Canadi 已提交
972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995
    }
  }
  CompareLogByPointer compare_log_files;
  std::sort(log_files.begin(), log_files.end(), compare_log_files);
  return status;
}

Status DBImpl::RetainProbableWalFiles(VectorLogPtr& all_logs,
                                      const SequenceNumber target) {
  int64_t start = 0;  // signed to avoid overflow when target is < first file.
  int64_t end = static_cast<int64_t>(all_logs.size()) - 1;
  // Binary Search. avoid opening all files.
  while (end >= start) {
    int64_t mid = start + (end - start) / 2;  // Avoid overflow.
    SequenceNumber current_seq_num = all_logs.at(mid)->StartSequence();
    if (current_seq_num == target) {
      end = mid;
      break;
    } else if (current_seq_num < target) {
      start = mid + 1;
    } else {
      end = mid - 1;
    }
  }
I
Igor Canadi 已提交
996 997
  // end could be -ve.
  size_t start_index = std::max(static_cast<int64_t>(0), end);
I
Igor Canadi 已提交
998 999 1000 1001 1002 1003
  // The last wal file is always included
  all_logs.erase(all_logs.begin(), all_logs.begin() + start_index);
  return Status::OK();
}

Status DBImpl::ReadFirstRecord(const WalFileType type, const uint64_t number,
I
Igor Canadi 已提交
1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016
                               SequenceNumber* sequence) {
  if (type != kAliveLogFile && type != kArchivedLogFile) {
    return Status::NotSupported("File Type Not Known " + std::to_string(type));
  }
  {
    MutexLock l(&read_first_record_cache_mutex_);
    auto itr = read_first_record_cache_.find(number);
    if (itr != read_first_record_cache_.end()) {
      *sequence = itr->second;
      return Status::OK();
    }
  }
  Status s;
I
Igor Canadi 已提交
1017 1018
  if (type == kAliveLogFile) {
    std::string fname = LogFileName(options_.wal_dir, number);
I
Igor Canadi 已提交
1019 1020 1021 1022
    s = ReadFirstLine(fname, sequence);
    if (env_->FileExists(fname) && !s.ok()) {
      // return any error that is not caused by non-existing file
      return s;
I
Igor Canadi 已提交
1023
    }
I
Igor Canadi 已提交
1024
  }
I
Igor Canadi 已提交
1025

I
Igor Canadi 已提交
1026
  if (type == kArchivedLogFile || !s.ok()) {
I
Igor Canadi 已提交
1027 1028
    //  check if the file got moved to archive.
    std::string archived_file = ArchivedLogFileName(options_.wal_dir, number);
I
Igor Canadi 已提交
1029
    s = ReadFirstLine(archived_file, sequence);
I
Igor Canadi 已提交
1030
  }
I
Igor Canadi 已提交
1031 1032 1033 1034 1035 1036

  if (s.ok() && *sequence != 0) {
    MutexLock l(&read_first_record_cache_mutex_);
    read_first_record_cache_.insert({number, *sequence});
  }
  return s;
I
Igor Canadi 已提交
1037 1038
}

I
Igor Canadi 已提交
1039 1040
// the function returns status.ok() and sequence == 0 if the file exists, but is
// empty
I
Igor Canadi 已提交
1041
Status DBImpl::ReadFirstLine(const std::string& fname,
I
Igor Canadi 已提交
1042
                             SequenceNumber* sequence) {
I
Igor Canadi 已提交
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;

    Status* status;
    bool ignore_error;  // true if options_.paranoid_checks==false
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
          (this->ignore_error ? "(ignoring error) " : ""), fname,
          static_cast<int>(bytes), s.ToString().c_str());
      if (this->status->ok()) {
        // only keep the first error
        *this->status = s;
      }
    }
  };

  unique_ptr<SequentialFile> file;
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);

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

  LogReporter reporter;
  reporter.env = env_;
  reporter.info_log = options_.info_log.get();
  reporter.fname = fname.c_str();
  reporter.status = &status;
  reporter.ignore_error = !options_.paranoid_checks;
  log::Reader reader(std::move(file), &reporter, true /*checksum*/,
                     0 /*initial_offset*/);
  std::string scratch;
  Slice record;

  if (reader.ReadRecord(&record, &scratch) &&
      (status.ok() || !options_.paranoid_checks)) {
    if (record.size() < 12) {
      reporter.Corruption(record.size(),
                          Status::Corruption("log record too small"));
      // TODO read record's till the first no corrupt entry?
    } else {
I
Igor Canadi 已提交
1086 1087 1088
      WriteBatch batch;
      WriteBatchInternal::SetContents(&batch, record);
      *sequence = WriteBatchInternal::Sequence(&batch);
I
Igor Canadi 已提交
1089 1090 1091 1092
      return Status::OK();
    }
  }

I
Igor Canadi 已提交
1093 1094 1095
  // ReadRecord returns false on EOF, which means that the log file is empty. we
  // return status.ok() in that case and set sequence number to 0
  *sequence = 0;
I
Igor Canadi 已提交
1096 1097 1098 1099 1100
  return status;
}

#endif  // ROCKSDB_LITE

1101
Status DBImpl::Recover(
1102 1103
    const std::vector<ColumnFamilyDescriptor>& column_families, bool read_only,
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
1104 1105
  mutex_.AssertHeld();

1106
  bool is_new_db = false;
1107
  assert(db_lock_ == nullptr);
I
Igor Canadi 已提交
1108
  if (!read_only) {
1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120
    // 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;
    }

1121 1122 1123 1124 1125 1126 1127
    for (auto db_path : options_.db_paths) {
      s = env_->CreateDirIfMissing(db_path);
      if (!s.ok()) {
        return s;
      }
    }

1128 1129 1130 1131 1132
    s = env_->NewDirectory(dbname_, &db_directory_);
    if (!s.ok()) {
      return s;
    }

1133
    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
1134 1135 1136
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
1137

1138 1139
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
1140
        // TODO: add merge_operator name check
1141
        s = NewDB();
1142
        is_new_db = true;
1143 1144 1145 1146 1147 1148
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
1149 1150
      }
    } else {
1151 1152 1153 1154
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
1155
    }
M
Mayank Agarwal 已提交
1156 1157 1158 1159 1160 1161 1162
    // Check for the IDENTITY file and create it if not there
    if (!env_->FileExists(IdentityFileName(dbname_))) {
      s = SetIdentityFile(env_, dbname_);
      if (!s.ok()) {
        return s;
      }
    }
J
jorlow@chromium.org 已提交
1163 1164
  }

1165
  Status s = versions_->Recover(column_families, read_only);
I
Igor Canadi 已提交
1166 1167 1168
  if (options_.paranoid_checks && s.ok()) {
    s = CheckConsistency();
  }
J
jorlow@chromium.org 已提交
1169 1170
  if (s.ok()) {
    SequenceNumber max_sequence(0);
1171 1172
    default_cf_handle_ = new ColumnFamilyHandleImpl(
        versions_->GetColumnFamilySet()->GetDefault(), this, &mutex_);
1173 1174
    single_column_family_mode_ =
        versions_->GetColumnFamilySet()->NumberOfColumnFamilies() == 1;
1175 1176 1177 1178 1179 1180 1181

    // 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
1182
    // produced by an older version of rocksdb.
1183
    const uint64_t min_log = versions_->MinLogNumber();
1184 1185
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
1186
    s = env_->GetChildren(options_.wal_dir, &filenames);
1187 1188
    if (!s.ok()) {
      return s;
1189
    }
K
kailiu 已提交
1190

1191 1192
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
K
kailiu 已提交
1193 1194
      uint64_t number;
      FileType type;
1195 1196 1197 1198 1199 1200 1201 1202 1203
      if (ParseFileName(filenames[i], &number, &type) && type == kLogFile) {
        if (is_new_db) {
          return Status::Corruption(
              "While creating a new Db, wal_dir contains "
              "existing log file: ",
              filenames[i]);
        } else if ((number >= min_log) || (number == prev_log)) {
          logs.push_back(number);
        }
1204
      }
J
jorlow@chromium.org 已提交
1205
    }
1206

H
heyongqiang 已提交
1207 1208 1209 1210 1211 1212
    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");
    }

1213 1214
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
K
kailiu 已提交
1215
    for (const auto& log : logs) {
1216 1217 1218
      // 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.
K
kailiu 已提交
1219
      versions_->MarkFileNumberUsed(log);
K
Kai Liu 已提交
1220
      s = RecoverLogFile(log, &max_sequence, read_only);
1221
    }
I
Igor Canadi 已提交
1222 1223
    SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                   versions_->LastSequence());
J
jorlow@chromium.org 已提交
1224 1225
  }

I
Igor Canadi 已提交
1226 1227 1228 1229 1230
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    max_total_in_memory_state_ += cfd->options()->write_buffer_size *
                                  cfd->options()->max_write_buffer_number;
  }

J
jorlow@chromium.org 已提交
1231 1232 1233
  return s;
}

I
Igor Canadi 已提交
1234 1235
Status DBImpl::RecoverLogFile(uint64_t log_number, SequenceNumber* max_sequence,
                              bool read_only) {
J
jorlow@chromium.org 已提交
1236 1237
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
1238
    Logger* info_log;
J
jorlow@chromium.org 已提交
1239
    const char* fname;
1240 1241
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
1242
    virtual void Corruption(size_t bytes, const Status& s) {
1243
      Log(info_log, "%s%s: dropping %d bytes; %s",
1244
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
1245
          fname, static_cast<int>(bytes), s.ToString().c_str());
1246
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
1247 1248 1249 1250 1251
    }
  };

  mutex_.AssertHeld();

1252
  std::unordered_map<int, VersionEdit> version_edits;
1253
  // no need to refcount because iteration is under mutex
1254 1255
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    VersionEdit edit;
1256 1257
    edit.SetColumnFamily(cfd->GetID());
    version_edits.insert({cfd->GetID(), edit});
1258
  }
I
Igor Canadi 已提交
1259

J
jorlow@chromium.org 已提交
1260
  // Open the log file
1261
  std::string fname = LogFileName(options_.wal_dir, log_number);
1262
  unique_ptr<SequentialFile> file;
1263
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
1264 1265 1266 1267 1268 1269 1270 1271
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
1272
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
1273
  reporter.fname = fname.c_str();
1274 1275
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
1276 1277 1278 1279
  // 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).
1280
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1281
                     0/*initial_offset*/);
1282
  Log(options_.info_log, "Recovering log #%" PRIu64 "", log_number);
J
jorlow@chromium.org 已提交
1283 1284 1285 1286 1287

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
I
Igor Canadi 已提交
1288
  while (reader.ReadRecord(&record, &scratch)) {
J
jorlow@chromium.org 已提交
1289 1290 1291 1292 1293 1294 1295
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

1296
    status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
1297
        &batch, column_family_memtables_.get(), true, log_number);
1298

J
jorlow@chromium.org 已提交
1299 1300
    MaybeIgnoreError(&status);
    if (!status.ok()) {
I
Igor Canadi 已提交
1301
      return status;
J
jorlow@chromium.org 已提交
1302 1303 1304 1305 1306 1307 1308 1309
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

1310
    if (!read_only) {
1311 1312
      // no need to refcount since client still doesn't have access
      // to the DB and can not drop column families while we iterate
1313
      for (auto cfd : *versions_->GetColumnFamilySet()) {
1314
        if (cfd->mem()->ShouldFlush()) {
1315
          // If this asserts, it means that InsertInto failed in
1316 1317
          // filtering updates to already-flushed column families
          assert(cfd->GetLogNumber() <= log_number);
1318
          auto iter = version_edits.find(cfd->GetID());
1319 1320
          assert(iter != version_edits.end());
          VersionEdit* edit = &iter->second;
1321
          status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
1322 1323 1324 1325 1326 1327 1328 1329
          // we still want to clear the memtable, even if the recovery failed
          cfd->CreateNewMemtable();
          if (!status.ok()) {
            // Reflect errors immediately so that conditions like full
            // file-systems cause the DB::Open() to fail.
            return status;
          }
        }
J
jorlow@chromium.org 已提交
1330 1331 1332 1333
      }
    }
  }

1334 1335 1336 1337
  if (versions_->LastSequence() < *max_sequence) {
    versions_->SetLastSequence(*max_sequence);
  }

1338
  if (!read_only) {
1339 1340
    // no need to refcount since client still doesn't have access
    // to the DB and can not drop column families while we iterate
1341
    for (auto cfd : *versions_->GetColumnFamilySet()) {
1342
      auto iter = version_edits.find(cfd->GetID());
1343 1344 1345
      assert(iter != version_edits.end());
      VersionEdit* edit = &iter->second;

1346 1347 1348 1349
      if (cfd->GetLogNumber() > log_number) {
        // Column family cfd has already flushed the data
        // from log_number. Memtable has to be empty because
        // we filter the updates based on log_number
1350
        // (in WriteBatch::InsertInto)
1351 1352 1353 1354 1355 1356 1357
        assert(cfd->mem()->GetFirstSequenceNumber() == 0);
        assert(edit->NumEntries() == 0);
        continue;
      }

      // flush the final memtable (if non-empty)
      if (cfd->mem()->GetFirstSequenceNumber() != 0) {
1358
        status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
1359
      }
1360 1361 1362 1363 1364
      // we still want to clear the memtable, even if the recovery failed
      cfd->CreateNewMemtable();
      if (!status.ok()) {
        return status;
      }
J
jorlow@chromium.org 已提交
1365

1366 1367 1368 1369 1370 1371 1372
      // write MANIFEST with update
      // writing log number in the manifest means that any log file
      // with number strongly less than (log_number + 1) is already
      // recovered and should be ignored on next reincarnation.
      // Since we already recovered log_number, we want all logs
      // with numbers `<= log_number` (includes this one) to be ignored
      edit->SetLogNumber(log_number + 1);
1373 1374 1375
      // we must mark the next log number as used, even though it's
      // not actually used. that is because VersionSet assumes
      // VersionSet::next_file_number_ always to be strictly greater than any
I
Igor Canadi 已提交
1376
      // log number
1377
      versions_->MarkFileNumberUsed(log_number + 1);
1378 1379 1380 1381 1382
      status = versions_->LogAndApply(cfd, edit, &mutex_);
      if (!status.ok()) {
        return status;
      }
    }
1383
  }
I
Igor Canadi 已提交
1384

J
jorlow@chromium.org 已提交
1385 1386 1387
  return status;
}

1388 1389
Status DBImpl::WriteLevel0TableForRecovery(ColumnFamilyData* cfd, MemTable* mem,
                                           VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1390
  mutex_.AssertHeld();
1391
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1392
  FileMetaData meta;
1393 1394
  meta.fd = FileDescriptor(versions_->NewFileNumber(), 0, 0);
  pending_outputs_[meta.fd.GetNumber()] = 0;  // path 0 for level 0 file.
1395
  Iterator* iter = mem->NewIterator(ReadOptions(), true);
1396 1397 1398
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
1399 1400
  Log(options_.info_log, "[%s] Level-0 table #%" PRIu64 ": started",
      cfd->GetName().c_str(), meta.fd.GetNumber());
1401 1402 1403 1404

  Status s;
  {
    mutex_.Unlock();
I
Igor Canadi 已提交
1405
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1406
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1407
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1408
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1409
    LogFlush(options_.info_log);
1410 1411 1412
    mutex_.Lock();
  }

1413 1414 1415 1416
  Log(options_.info_log,
      "[%s] Level-0 table #%" PRIu64 ": %" PRIu64 " bytes %s",
      cfd->GetName().c_str(), meta.fd.GetNumber(), meta.fd.GetFileSize(),
      s.ToString().c_str());
J
jorlow@chromium.org 已提交
1417
  delete iter;
1418

1419
  pending_outputs_.erase(meta.fd.GetNumber());
1420 1421 1422 1423

  // Note that if file_size is zero, the file has been deleted and
  // should not be added to the manifest.
  int level = 0;
1424
  if (s.ok() && meta.fd.GetFileSize() > 0) {
1425 1426 1427
    edit->AddFile(level, meta.fd.GetNumber(), meta.fd.GetPathId(),
                  meta.fd.GetFileSize(), meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1428 1429
  }

I
Igor Canadi 已提交
1430
  InternalStats::CompactionStats stats;
1431
  stats.micros = env_->NowMicros() - start_micros;
1432
  stats.bytes_written = meta.fd.GetFileSize();
M
Mark Callaghan 已提交
1433
  stats.files_out_levelnp1 = 1;
1434
  cfd->internal_stats()->AddCompactionStats(level, stats);
1435 1436
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
             meta.fd.GetFileSize());
J
jorlow@chromium.org 已提交
1437 1438 1439
  return s;
}

1440
Status DBImpl::WriteLevel0Table(ColumnFamilyData* cfd,
1441
                                autovector<MemTable*>& mems, VersionEdit* edit,
1442
                                uint64_t* filenumber, LogBuffer* log_buffer) {
J
jorlow@chromium.org 已提交
1443
  mutex_.AssertHeld();
1444 1445
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
1446 1447

  meta.fd = FileDescriptor(versions_->NewFileNumber(), 0, 0);
1448
  *filenumber = meta.fd.GetNumber();
1449
  pending_outputs_[meta.fd.GetNumber()] = 0;  // path 0 for level 0 file.
1450

1451 1452
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1453
    mems[0]->GetFirstSequenceNumber();
1454
  Version* base = cfd->current();
1455
  base->Ref();          // it is likely that we do not need this reference
1456 1457 1458
  Status s;
  {
    mutex_.Unlock();
H
Haobo Xu 已提交
1459
    log_buffer->FlushBufferToLog();
K
Kai Liu 已提交
1460
    std::vector<Iterator*> memtables;
1461
    for (MemTable* m : mems) {
1462 1463 1464
      Log(options_.info_log,
          "[%s] Flushing memtable with next log file: %" PRIu64 "\n",
          cfd->GetName().c_str(), m->GetNextLogNumber());
1465
      memtables.push_back(m->NewIterator(ReadOptions(), true));
1466
    }
I
Igor Canadi 已提交
1467
    Iterator* iter = NewMergingIterator(&cfd->internal_comparator(),
1468
                                        &memtables[0], memtables.size());
1469 1470
    Log(options_.info_log, "[%s] Level-0 flush table #%" PRIu64 ": started",
        cfd->GetName().c_str(), meta.fd.GetNumber());
1471

I
Igor Canadi 已提交
1472
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1473
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1474
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1475
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1476
    LogFlush(options_.info_log);
1477
    delete iter;
1478 1479 1480 1481
    Log(options_.info_log,
        "[%s] Level-0 flush table #%" PRIu64 ": %" PRIu64 " bytes %s",
        cfd->GetName().c_str(), meta.fd.GetFileSize(), meta.fd.GetFileSize(),
        s.ToString().c_str());
I
Igor Canadi 已提交
1482

1483 1484 1485
    if (!options_.disableDataSync) {
      db_directory_->Fsync();
    }
1486 1487
    mutex_.Lock();
  }
1488 1489
  base->Unref();

1490
  // re-acquire the most current version
1491
  base = cfd->current();
1492 1493 1494 1495 1496 1497

  // 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 已提交
1498
  // committed so that other threads can recognize this file as a
1499 1500 1501 1502 1503 1504
  // 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;
1505
  if (s.ok() && meta.fd.GetFileSize() > 0) {
1506 1507 1508 1509 1510 1511
    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.
1512
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1513
        cfd->options()->compaction_style == kCompactionStyleLevel) {
1514 1515
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
1516 1517 1518
    edit->AddFile(level, meta.fd.GetNumber(), meta.fd.GetPathId(),
                  meta.fd.GetFileSize(), meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1519 1520
  }

I
Igor Canadi 已提交
1521
  InternalStats::CompactionStats stats;
1522
  stats.micros = env_->NowMicros() - start_micros;
1523
  stats.bytes_written = meta.fd.GetFileSize();
1524
  cfd->internal_stats()->AddCompactionStats(level, stats);
1525 1526
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
             meta.fd.GetFileSize());
1527 1528 1529
  return s;
}

1530 1531
Status DBImpl::FlushMemTableToOutputFile(ColumnFamilyData* cfd,
                                         bool* madeProgress,
H
Haobo Xu 已提交
1532 1533
                                         DeletionState& deletion_state,
                                         LogBuffer* log_buffer) {
1534
  mutex_.AssertHeld();
1535
  assert(cfd->imm()->size() != 0);
1536
  assert(cfd->imm()->IsFlushPending());
1537 1538 1539

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
K
Kai Liu 已提交
1540
  autovector<MemTable*> mems;
1541
  cfd->imm()->PickMemtablesToFlush(&mems);
1542
  if (mems.empty()) {
I
Igor Canadi 已提交
1543 1544
    LogToBuffer(log_buffer, "[%s] Nothing in memtable to flush",
                cfd->GetName().c_str());
L
Lei Jin 已提交
1545
    return Status::OK();
1546 1547 1548
  }

  // record the logfile_number_ before we release the mutex
1549 1550 1551
  // entries mems are (implicitly) sorted in ascending order by their created
  // time. We will use the first memtable's `edit` to keep the meta info for
  // this flush.
1552
  MemTable* m = mems[0];
1553 1554
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1555 1556
  // SetLogNumber(log_num) indicates logs with number smaller than log_num
  // will no longer be picked up for recovery.
1557 1558
  edit->SetLogNumber(mems.back()->GetNextLogNumber());
  edit->SetColumnFamily(cfd->GetID());
1559

1560
  // This will release and re-acquire the mutex.
1561
  Status s = WriteLevel0Table(cfd, mems, edit, &file_number, log_buffer);
1562

1563
  if (s.ok() && shutting_down_.Acquire_Load() && cfd->IsDropped()) {
L
Lei Jin 已提交
1564
    s = Status::ShutdownInProgress(
I
Igor Canadi 已提交
1565
        "Database shutdown or Column family drop during flush");
1566
  }
J
jorlow@chromium.org 已提交
1567

L
Lei Jin 已提交
1568
  if (!s.ok()) {
1569
    cfd->imm()->RollbackMemtableFlush(mems, file_number, &pending_outputs_);
1570 1571
  } else {
    // Replace immutable memtable with the generated Table
1572 1573
    s = cfd->imm()->InstallMemtableFlushResults(
        cfd, mems, versions_.get(), &mutex_, options_.info_log.get(),
1574
        file_number, &pending_outputs_, &deletion_state.memtables_to_free,
1575
        db_directory_.get(), log_buffer);
1576
  }
J
jorlow@chromium.org 已提交
1577 1578

  if (s.ok()) {
1579
    InstallSuperVersion(cfd, deletion_state);
1580 1581 1582
    if (madeProgress) {
      *madeProgress = 1;
    }
I
Igor Canadi 已提交
1583 1584 1585
    Version::LevelSummaryStorage tmp;
    LogToBuffer(log_buffer, "[%s] Level summary: %s\n", cfd->GetName().c_str(),
                cfd->current()->LevelSummary(&tmp));
1586

1587
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1588

1589
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1590
      // add to deletion state
1591
      while (alive_log_files_.size() &&
I
Igor Canadi 已提交
1592 1593 1594 1595
             alive_log_files_.begin()->number < versions_->MinLogNumber()) {
        const auto& earliest = *alive_log_files_.begin();
        deletion_state.log_delete_files.push_back(earliest.number);
        total_log_size_ -= earliest.size;
1596 1597
        alive_log_files_.pop_front();
      }
1598
    }
J
jorlow@chromium.org 已提交
1599
  }
1600 1601 1602 1603 1604 1605 1606

  if (!s.ok() && !s.IsShutdownInProgress() && options_.paranoid_checks &&
      bg_error_.ok()) {
    // if a bad error happened (not ShutdownInProgress) and paranoid_checks is
    // true, mark DB read-only
    bg_error_ = s;
  }
J
jorlow@chromium.org 已提交
1607 1608 1609
  return s;
}

1610
Status DBImpl::CompactRange(ColumnFamilyHandle* column_family,
1611 1612
                            const Slice* begin, const Slice* end,
                            bool reduce_level, int target_level) {
1613 1614
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
I
Igor Canadi 已提交
1615 1616

  Status s = FlushMemTable(cfd, FlushOptions());
L
Lei Jin 已提交
1617 1618 1619 1620 1621
  if (!s.ok()) {
    LogFlush(options_.info_log);
    return s;
  }

I
Igor Canadi 已提交
1622
  int max_level_with_files = 0;
G
Gabor Cselle 已提交
1623 1624
  {
    MutexLock l(&mutex_);
I
Igor Canadi 已提交
1625 1626
    Version* base = cfd->current();
    for (int level = 1; level < cfd->NumberLevels(); level++) {
G
Gabor Cselle 已提交
1627 1628 1629 1630 1631
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
1632 1633 1634
  for (int level = 0; level <= max_level_with_files; level++) {
    // in case the compaction is unversal or if we're compacting the
    // bottom-most level, the output level will be the same as input one
I
Igor Canadi 已提交
1635
    if (cfd->options()->compaction_style == kCompactionStyleUniversal ||
I
Igor Canadi 已提交
1636
        cfd->options()->compaction_style == kCompactionStyleFIFO ||
1637
        level == max_level_with_files) {
I
Igor Canadi 已提交
1638
      s = RunManualCompaction(cfd, level, level, begin, end);
1639
    } else {
I
Igor Canadi 已提交
1640
      s = RunManualCompaction(cfd, level, level + 1, begin, end);
L
Lei Jin 已提交
1641 1642 1643 1644
    }
    if (!s.ok()) {
      LogFlush(options_.info_log);
      return s;
1645
    }
G
Gabor Cselle 已提交
1646
  }
1647 1648

  if (reduce_level) {
I
Igor Canadi 已提交
1649
    s = ReFitLevel(cfd, max_level_with_files, target_level);
1650
  }
I
Igor Canadi 已提交
1651
  LogFlush(options_.info_log);
L
Lei Jin 已提交
1652 1653

  return s;
1654 1655 1656
}

// return the same level if it cannot be moved
I
Igor Canadi 已提交
1657
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd, int level) {
1658
  mutex_.AssertHeld();
I
Igor Canadi 已提交
1659
  Version* current = cfd->current();
1660
  int minimum_level = level;
1661
  for (int i = level - 1; i > 0; --i) {
1662
    // stop if level i is not empty
1663
    if (current->NumLevelFiles(i) > 0) break;
1664
    // stop if level i is too small (cannot fit the level files)
I
Igor Canadi 已提交
1665
    if (cfd->compaction_picker()->MaxBytesForLevel(i) <
1666 1667 1668
        current->NumLevelBytes(level)) {
      break;
    }
1669 1670 1671 1672 1673 1674

    minimum_level = i;
  }
  return minimum_level;
}

I
Igor Canadi 已提交
1675 1676
Status DBImpl::ReFitLevel(ColumnFamilyData* cfd, int level, int target_level) {
  assert(level < cfd->NumberLevels());
1677

I
Igor Canadi 已提交
1678
  SuperVersion* superversion_to_free = nullptr;
1679
  SuperVersion* new_superversion = new SuperVersion();
I
Igor Canadi 已提交
1680 1681

  mutex_.Lock();
1682 1683 1684

  // only allow one thread refitting
  if (refitting_level_) {
I
Igor Canadi 已提交
1685
    mutex_.Unlock();
1686
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
I
Igor Canadi 已提交
1687
    delete new_superversion;
L
Lei Jin 已提交
1688
    return Status::NotSupported("another thread is refitting");
1689 1690 1691 1692 1693
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1694
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1695
    Log(options_.info_log,
1696 1697
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1698 1699 1700 1701
    bg_cv_.Wait();
  }

  // move to a smaller level
1702 1703
  int to_level = target_level;
  if (target_level < 0) {
I
Igor Canadi 已提交
1704
    to_level = FindMinimumEmptyLevelFitting(cfd, level);
1705
  }
1706 1707 1708

  assert(to_level <= level);

L
Lei Jin 已提交
1709
  Status status;
1710
  if (to_level < level) {
I
Igor Canadi 已提交
1711
    Log(options_.info_log, "[%s] Before refitting:\n%s", cfd->GetName().c_str(),
I
Igor Canadi 已提交
1712
        cfd->current()->DebugString().data());
1713

1714
    VersionEdit edit;
I
Igor Canadi 已提交
1715 1716
    edit.SetColumnFamily(cfd->GetID());
    for (const auto& f : cfd->current()->files_[level]) {
1717
      edit.DeleteFile(level, f->fd.GetNumber());
1718 1719 1720
      edit.AddFile(to_level, f->fd.GetNumber(), f->fd.GetPathId(),
                   f->fd.GetFileSize(), f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1721
    }
I
Igor Canadi 已提交
1722 1723
    Log(options_.info_log, "[%s] Apply version edit:\n%s",
        cfd->GetName().c_str(), edit.DebugString().data());
1724

I
Igor Canadi 已提交
1725
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
1726
    superversion_to_free = cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
1727
    new_superversion = nullptr;
1728

I
Igor Canadi 已提交
1729 1730
    Log(options_.info_log, "[%s] LogAndApply: %s\n", cfd->GetName().c_str(),
        status.ToString().data());
1731 1732

    if (status.ok()) {
I
Igor Canadi 已提交
1733 1734
      Log(options_.info_log, "[%s] After refitting:\n%s",
          cfd->GetName().c_str(), cfd->current()->DebugString().data());
1735 1736 1737 1738 1739
    }
  }

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
1740 1741 1742 1743

  mutex_.Unlock();
  delete superversion_to_free;
  delete new_superversion;
L
Lei Jin 已提交
1744
  return status;
G
Gabor Cselle 已提交
1745 1746
}

1747 1748 1749
int DBImpl::NumberLevels(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->NumberLevels();
1750 1751
}

1752 1753 1754
int DBImpl::MaxMemCompactionLevel(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->max_mem_compaction_level;
1755 1756
}

1757 1758 1759
int DBImpl::Level0StopWriteTrigger(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->level0_stop_writes_trigger;
1760 1761
}

1762
Status DBImpl::Flush(const FlushOptions& options,
1763 1764 1765
                     ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return FlushMemTable(cfh->cfd(), options);
H
heyongqiang 已提交
1766 1767
}

1768
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1769 1770 1771
  return versions_->LastSequence();
}

I
Igor Canadi 已提交
1772 1773
Status DBImpl::RunManualCompaction(ColumnFamilyData* cfd, int input_level,
                                   int output_level, const Slice* begin,
L
Lei Jin 已提交
1774
                                   const Slice* end) {
1775
  assert(input_level >= 0);
1776

G
Gabor Cselle 已提交
1777 1778
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1779
  ManualCompaction manual;
I
Igor Canadi 已提交
1780
  manual.cfd = cfd;
1781 1782
  manual.input_level = input_level;
  manual.output_level = output_level;
G
Gabor Cselle 已提交
1783
  manual.done = false;
1784
  manual.in_progress = false;
1785 1786 1787
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
I
Igor Canadi 已提交
1788 1789
      cfd->options()->compaction_style == kCompactionStyleUniversal ||
      cfd->options()->compaction_style == kCompactionStyleFIFO) {
1790
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1791 1792 1793 1794
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1795
  if (end == nullptr ||
I
Igor Canadi 已提交
1796 1797
      cfd->options()->compaction_style == kCompactionStyleUniversal ||
      cfd->options()->compaction_style == kCompactionStyleFIFO) {
1798
    manual.end = nullptr;
G
Gabor Cselle 已提交
1799 1800 1801 1802 1803 1804
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1805

1806 1807 1808 1809 1810 1811
  // When a manual compaction arrives, temporarily disable scheduling of
  // non-manual compactions and wait until the number of scheduled compaction
  // jobs drops to zero. This is needed to ensure that this manual compaction
  // can compact any range of keys/files.
  //
  // bg_manual_only_ is non-zero when at least one thread is inside
1812
  // RunManualCompaction(), i.e. during that time no other compaction will
1813 1814 1815
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
1816
  // RunManualCompaction() from getting to the second while loop below.
1817 1818 1819 1820 1821 1822
  // However, only one of them will actually schedule compaction, while
  // others will wait on a condition variable until it completes.

  ++bg_manual_only_;
  while (bg_compaction_scheduled_ > 0) {
    Log(options_.info_log,
I
Igor Canadi 已提交
1823 1824 1825
        "[%s] Manual compaction waiting for all other scheduled background "
        "compactions to finish",
        cfd->GetName().c_str());
1826 1827
    bg_cv_.Wait();
  }
1828

I
Igor Canadi 已提交
1829 1830
  Log(options_.info_log, "[%s] Manual compaction starting",
      cfd->GetName().c_str());
1831

1832 1833 1834 1835
  while (!manual.done && !shutting_down_.Acquire_Load() && bg_error_.ok()) {
    assert(bg_manual_only_ > 0);
    if (manual_compaction_ != nullptr) {
      // Running either this or some other manual compaction
G
Gabor Cselle 已提交
1836
      bg_cv_.Wait();
1837 1838 1839
    } else {
      manual_compaction_ = &manual;
      MaybeScheduleFlushOrCompaction();
G
Gabor Cselle 已提交
1840
    }
H
hans@chromium.org 已提交
1841
  }
1842

1843 1844 1845
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
L
Lei Jin 已提交
1846
  return manual.status;
J
jorlow@chromium.org 已提交
1847 1848
}

1849 1850
Status DBImpl::FlushMemTable(ColumnFamilyData* cfd,
                             const FlushOptions& options) {
1851 1852
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1853
  if (s.ok() && options.wait) {
1854
    // Wait until the compaction completes
1855
    s = WaitForFlushMemTable(cfd);
1856 1857
  }
  return s;
J
jorlow@chromium.org 已提交
1858 1859
}

1860
Status DBImpl::WaitForFlushMemTable(ColumnFamilyData* cfd) {
1861 1862 1863
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1864
  while (cfd->imm()->size() > 0 && bg_error_.ok()) {
1865 1866
    bg_cv_.Wait();
  }
1867
  if (!bg_error_.ok()) {
1868 1869 1870
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1871 1872
}

1873
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
1874
  mutex_.AssertHeld();
1875
  bg_schedule_needed_ = false;
1876 1877
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
J
jorlow@chromium.org 已提交
1878 1879 1880
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
  } else {
1881
    bool is_flush_pending = false;
1882
    // no need to refcount since we're under a mutex
1883
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1884
      if (cfd->imm()->IsFlushPending()) {
1885 1886 1887
        is_flush_pending = true;
      }
    }
1888
    if (is_flush_pending) {
1889
      // memtable flush needed
1890 1891 1892
      if (bg_flush_scheduled_ < options_.max_background_flushes) {
        bg_flush_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::HIGH);
1893
      } else if (options_.max_background_flushes > 0) {
1894
        bg_schedule_needed_ = true;
1895
      }
1896
    }
1897
    bool is_compaction_needed = false;
1898
    // no need to refcount since we're under a mutex
1899
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1900
      if (cfd->current()->NeedsCompaction()) {
1901 1902 1903 1904
        is_compaction_needed = true;
        break;
      }
    }
1905

1906 1907
    // Schedule BGWorkCompaction if there's a compaction pending (or a memtable
    // flush, but the HIGH pool is not enabled)
1908
    // Do it only if max_background_compactions hasn't been reached and, in case
1909
    // bg_manual_only_ > 0, if it's a manual compaction.
1910 1911
    if ((manual_compaction_ || is_compaction_needed ||
         (is_flush_pending && options_.max_background_flushes == 0)) &&
1912
        (!bg_manual_only_ || manual_compaction_)) {
1913 1914 1915 1916 1917 1918
      if (bg_compaction_scheduled_ < options_.max_background_compactions) {
        bg_compaction_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW);
      } else {
        bg_schedule_needed_ = true;
      }
1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930
    }
  }
}

void DBImpl::BGWorkFlush(void* db) {
  reinterpret_cast<DBImpl*>(db)->BackgroundCallFlush();
}

void DBImpl::BGWorkCompaction(void* db) {
  reinterpret_cast<DBImpl*>(db)->BackgroundCallCompaction();
}

I
Igor Canadi 已提交
1931
Status DBImpl::BackgroundFlush(bool* madeProgress,
H
Haobo Xu 已提交
1932 1933
                               DeletionState& deletion_state,
                               LogBuffer* log_buffer) {
1934 1935 1936 1937 1938 1939
  mutex_.AssertHeld();
  // call_status is failure if at least one flush was a failure. even if
  // flushing one column family reports a failure, we will continue flushing
  // other column families. however, call_status will be a failure in that case.
  Status call_status;
  // refcounting in iteration
1940
  for (auto cfd : *versions_->GetColumnFamilySet()) {
1941 1942 1943
    cfd->Ref();
    Status flush_status;
    while (flush_status.ok() && cfd->imm()->IsFlushPending()) {
1944 1945
      LogToBuffer(
          log_buffer,
1946
          "BackgroundCallFlush doing FlushMemTableToOutputFile with column "
I
Igor Canadi 已提交
1947 1948 1949
          "family [%s], flush slots available %d",
          cfd->GetName().c_str(),
          options_.max_background_flushes - bg_flush_scheduled_);
1950 1951
      flush_status = FlushMemTableToOutputFile(cfd, madeProgress,
                                               deletion_state, log_buffer);
1952 1953 1954
    }
    if (call_status.ok() && !flush_status.ok()) {
      call_status = flush_status;
1955
    }
1956
    cfd->Unref();
J
jorlow@chromium.org 已提交
1957
  }
1958
  versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
1959
  return call_status;
J
jorlow@chromium.org 已提交
1960 1961
}

1962
void DBImpl::BackgroundCallFlush() {
1963
  bool madeProgress = false;
K
Kai Liu 已提交
1964
  DeletionState deletion_state(true);
1965 1966
  assert(bg_flush_scheduled_);

1967
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, options_.info_log.get());
H
Haobo Xu 已提交
1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978
  {
    MutexLock l(&mutex_);

    Status s;
    if (!shutting_down_.Acquire_Load()) {
      s = BackgroundFlush(&madeProgress, deletion_state, &log_buffer);
      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.
1979 1980 1981
        uint64_t error_cnt = default_cf_handle_->cfd()
                                 ->internal_stats()
                                 ->BumpAndGetBackgroundErrorCount();
H
Haobo Xu 已提交
1982 1983
        bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
        mutex_.Unlock();
1984 1985 1986 1987
        Log(options_.info_log,
            "Waiting after background flush error: %s"
            "Accumulated background error counts: %" PRIu64,
            s.ToString().c_str(), error_cnt);
H
Haobo Xu 已提交
1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998
        log_buffer.FlushBufferToLog();
        LogFlush(options_.info_log);
        env_->SleepForMicroseconds(1000000);
        mutex_.Lock();
      }
    }

    // If !s.ok(), this means that Flush failed. In that case, we want
    // to delete all obsolete files and we force FindObsoleteFiles()
    FindObsoleteFiles(deletion_state, !s.ok());
    // delete unnecessary files if any, this is done outside the mutex
1999
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
2000
      mutex_.Unlock();
2001 2002 2003 2004 2005
      // Have to flush the info logs before bg_flush_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
2006
      log_buffer.FlushBufferToLog();
2007 2008 2009
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
2010 2011
      mutex_.Lock();
    }
I
Igor Canadi 已提交
2012

H
Haobo Xu 已提交
2013
    bg_flush_scheduled_--;
2014 2015 2016 2017 2018
    // Any time the mutex is released After finding the work to do, another
    // thread might execute MaybeScheduleFlushOrCompaction(). It is possible
    // that there is a pending job but it is not scheduled because of the
    // max thread limit.
    if (madeProgress || bg_schedule_needed_) {
H
Haobo Xu 已提交
2019 2020 2021
      MaybeScheduleFlushOrCompaction();
    }
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
2022 2023 2024 2025
    // IMPORTANT: there should be no code after calling SignalAll. This call may
    // signal the DB destructor that it's OK to proceed with destruction. In
    // that case, all DB variables will be dealloacated and referencing them
    // will cause trouble.
2026
  }
J
jorlow@chromium.org 已提交
2027 2028
}

2029
void DBImpl::BackgroundCallCompaction() {
2030
  bool madeProgress = false;
K
Kai Liu 已提交
2031
  DeletionState deletion_state(true);
H
Haobo Xu 已提交
2032 2033

  MaybeDumpStats();
2034
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, options_.info_log.get());
2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045
  {
    MutexLock l(&mutex_);
    assert(bg_compaction_scheduled_);
    Status s;
    if (!shutting_down_.Acquire_Load()) {
      s = BackgroundCompaction(&madeProgress, deletion_state, &log_buffer);
      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.
2046 2047 2048
        uint64_t error_cnt = default_cf_handle_->cfd()
                                 ->internal_stats()
                                 ->BumpAndGetBackgroundErrorCount();
2049 2050
        bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
        mutex_.Unlock();
H
Haobo Xu 已提交
2051
        log_buffer.FlushBufferToLog();
2052 2053 2054 2055
        Log(options_.info_log,
            "Waiting after background compaction error: %s, "
            "Accumulated background error counts: %" PRIu64,
            s.ToString().c_str(), error_cnt);
2056 2057 2058 2059 2060
        LogFlush(options_.info_log);
        env_->SleepForMicroseconds(1000000);
        mutex_.Lock();
      }
    }
H
Haobo Xu 已提交
2061

2062 2063 2064 2065 2066 2067 2068
    // If !s.ok(), this means that Compaction failed. In that case, we want
    // to delete all obsolete files we might have created and we force
    // FindObsoleteFiles(). This is because deletion_state does not catch
    // all created files if compaction failed.
    FindObsoleteFiles(deletion_state, !s.ok());

    // delete unnecessary files if any, this is done outside the mutex
2069
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
2070
      mutex_.Unlock();
2071 2072 2073 2074 2075
      // Have to flush the info logs before bg_compaction_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
2076
      log_buffer.FlushBufferToLog();
2077 2078 2079
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
2080 2081
      mutex_.Lock();
    }
D
Dhruba Borthakur 已提交
2082

2083
    bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
2084

2085
    MaybeScheduleLogDBDeployStats();
2086

2087 2088
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();

2089 2090 2091
    // Previous compaction may have produced too many files in a level,
    // So reschedule another compaction if we made progress in the
    // last compaction.
2092 2093 2094 2095 2096 2097
    //
    // Also, any time the mutex is released After finding the work to do,
    // another thread might execute MaybeScheduleFlushOrCompaction(). It is
    // possible  that there is a pending job but it is not scheduled because of
    // the max thread limit.
    if (madeProgress || bg_schedule_needed_) {
2098 2099
      MaybeScheduleFlushOrCompaction();
    }
2100 2101 2102 2103 2104 2105 2106 2107 2108
    if (madeProgress || bg_compaction_scheduled_ == 0 || bg_manual_only_ > 0) {
      // signal if
      // * madeProgress -- need to wakeup MakeRoomForWrite
      // * bg_compaction_scheduled_ == 0 -- need to wakeup ~DBImpl
      // * bg_manual_only_ > 0 -- need to wakeup RunManualCompaction
      // If none of this is true, there is no need to signal since nobody is
      // waiting for it
      bg_cv_.SignalAll();
    }
I
Igor Canadi 已提交
2109 2110 2111 2112
    // IMPORTANT: there should be no code after calling SignalAll. This call may
    // signal the DB destructor that it's OK to proceed with destruction. In
    // that case, all DB variables will be dealloacated and referencing them
    // will cause trouble.
2113
  }
J
jorlow@chromium.org 已提交
2114 2115
}

A
Abhishek Kona 已提交
2116
Status DBImpl::BackgroundCompaction(bool* madeProgress,
2117 2118
                                    DeletionState& deletion_state,
                                    LogBuffer* log_buffer) {
2119
  *madeProgress = false;
J
jorlow@chromium.org 已提交
2120
  mutex_.AssertHeld();
2121

2122 2123
  bool is_manual = (manual_compaction_ != nullptr) &&
                   (manual_compaction_->in_progress == false);
2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141

  if (is_manual) {
    // another thread cannot pick up the same work
    manual_compaction_->in_progress = true;
  }

  // FLUSH preempts compaction
  Status flush_stat;
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    while (cfd->imm()->IsFlushPending()) {
      LogToBuffer(
          log_buffer,
          "BackgroundCompaction doing FlushMemTableToOutputFile, "
          "compaction slots available %d",
          options_.max_background_compactions - bg_compaction_scheduled_);
      cfd->Ref();
      flush_stat = FlushMemTableToOutputFile(cfd, madeProgress, deletion_state,
                                             log_buffer);
2142
      cfd->Unref();
2143 2144 2145 2146 2147 2148 2149
      if (!flush_stat.ok()) {
        if (is_manual) {
          manual_compaction_->status = flush_stat;
          manual_compaction_->done = true;
          manual_compaction_->in_progress = false;
          manual_compaction_ = nullptr;
        }
2150
        return flush_stat;
2151 2152 2153 2154 2155
      }
    }
  }

  unique_ptr<Compaction> c;
2156 2157
  InternalKey manual_end_storage;
  InternalKey* manual_end = &manual_end_storage;
H
hans@chromium.org 已提交
2158
  if (is_manual) {
G
Gabor Cselle 已提交
2159
    ManualCompaction* m = manual_compaction_;
2160
    assert(m->in_progress);
I
Igor Canadi 已提交
2161 2162
    c.reset(m->cfd->CompactRange(m->input_level, m->output_level, m->begin,
                                 m->end, &manual_end));
2163
    if (!c) {
2164
      m->done = true;
G
Gabor Cselle 已提交
2165
    }
I
Igor Canadi 已提交
2166 2167 2168 2169 2170 2171 2172 2173 2174
    LogToBuffer(log_buffer,
                "[%s] Manual compaction from level-%d to level-%d from %s .. "
                "%s; will stop at %s\n",
                m->cfd->GetName().c_str(), m->input_level, m->output_level,
                (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
                (m->end ? m->end->DebugString().c_str() : "(end)"),
                ((m->done || manual_end == nullptr)
                     ? "(end)"
                     : manual_end->DebugString().c_str()));
I
Igor Canadi 已提交
2175
  } else {
2176
    // no need to refcount in iteration since it's always under a mutex
I
Igor Canadi 已提交
2177
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
2178
      if (!cfd->options()->disable_auto_compactions) {
2179
        c.reset(cfd->PickCompaction(log_buffer));
I
Igor Canadi 已提交
2180 2181 2182 2183 2184 2185
        if (c != nullptr) {
          // update statistics
          MeasureTime(options_.statistics.get(), NUM_FILES_IN_SINGLE_COMPACTION,
                      c->inputs(0)->size());
          break;
        }
I
Igor Canadi 已提交
2186 2187
      }
    }
J
jorlow@chromium.org 已提交
2188 2189 2190
  }

  Status status;
2191
  if (!c) {
H
hans@chromium.org 已提交
2192
    // Nothing to do
2193
    LogToBuffer(log_buffer, "Compaction nothing to do");
I
Igor Canadi 已提交
2194 2195 2196 2197 2198 2199 2200 2201
  } else if (c->IsDeletionCompaction()) {
    // TODO(icanadi) Do we want to honor snapshots here? i.e. not delete old
    // file if there is alive snapshot pointing to it
    assert(c->num_input_files(1) == 0);
    assert(c->level() == 0);
    assert(c->column_family_data()->options()->compaction_style ==
           kCompactionStyleFIFO);
    for (const auto& f : *c->inputs(0)) {
2202
      c->edit()->DeleteFile(c->level(), f->fd.GetNumber());
I
Igor Canadi 已提交
2203 2204 2205 2206 2207 2208 2209 2210 2211
    }
    status = versions_->LogAndApply(c->column_family_data(), c->edit(), &mutex_,
                                    db_directory_.get());
    InstallSuperVersion(c->column_family_data(), deletion_state);
    LogToBuffer(log_buffer, "[%s] Deleted %d files\n",
                c->column_family_data()->GetName().c_str(),
                c->num_input_files(0));
    c->ReleaseCompactionFiles(status);
    *madeProgress = true;
H
hans@chromium.org 已提交
2212
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
2213
    // Move file to next level
2214
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
2215
    FileMetaData* f = c->input(0, 0);
2216
    c->edit()->DeleteFile(c->level(), f->fd.GetNumber());
2217 2218 2219
    c->edit()->AddFile(c->level() + 1, f->fd.GetNumber(), f->fd.GetPathId(),
                       f->fd.GetFileSize(), f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
I
Igor Canadi 已提交
2220
    status = versions_->LogAndApply(c->column_family_data(), c->edit(), &mutex_,
2221
                                    db_directory_.get());
I
Igor Canadi 已提交
2222
    InstallSuperVersion(c->column_family_data(), deletion_state);
2223

2224
    Version::LevelSummaryStorage tmp;
2225 2226 2227 2228 2229 2230
    LogToBuffer(
        log_buffer, "[%s] Moved #%lld to level-%d %lld bytes %s: %s\n",
        c->column_family_data()->GetName().c_str(),
        static_cast<unsigned long long>(f->fd.GetNumber()), c->level() + 1,
        static_cast<unsigned long long>(f->fd.GetFileSize()),
        status.ToString().c_str(), c->input_version()->LevelSummary(&tmp));
I
Igor Canadi 已提交
2231
    c->ReleaseCompactionFiles(status);
2232
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2233
  } else {
2234
    MaybeScheduleFlushOrCompaction(); // do more compaction work in parallel.
2235
    CompactionState* compact = new CompactionState(c.get());
H
Haobo Xu 已提交
2236
    status = DoCompactionWork(compact, deletion_state, log_buffer);
2237
    CleanupCompaction(compact, status);
I
Igor Canadi 已提交
2238
    c->ReleaseCompactionFiles(status);
2239
    c->ReleaseInputs();
2240
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2241
  }
2242
  c.reset();
J
jorlow@chromium.org 已提交
2243 2244 2245 2246 2247 2248

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
2249
    Log(InfoLogLevel::WARN_LEVEL, options_.info_log, "Compaction error: %s",
2250
        status.ToString().c_str());
J
jorlow@chromium.org 已提交
2251 2252 2253 2254
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
2255 2256

  if (is_manual) {
G
Gabor Cselle 已提交
2257
    ManualCompaction* m = manual_compaction_;
2258
    if (!status.ok()) {
L
Lei Jin 已提交
2259
      m->status = status;
2260 2261
      m->done = true;
    }
2262 2263 2264 2265 2266 2267 2268 2269 2270
    // For universal compaction:
    //   Because universal compaction always happens at level 0, so one
    //   compaction will pick up all overlapped files. No files will be
    //   filtered out due to size limit and left for a successive compaction.
    //   So we can safely conclude the current compaction.
    //
    //   Also note that, if we don't stop here, then the current compaction
    //   writes a new file back to level 0, which will be used in successive
    //   compaction. Hence the manual compaction will never finish.
2271 2272 2273 2274 2275
    //
    // Stop the compaction if manual_end points to nullptr -- this means
    // that we compacted the whole range. manual_end should always point
    // to nullptr in case of universal compaction
    if (manual_end == nullptr) {
2276 2277
      m->done = true;
    }
G
Gabor Cselle 已提交
2278 2279 2280
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
I
Igor Canadi 已提交
2281
      // Universal and FIFO compactions should always compact the whole range
I
Igor Canadi 已提交
2282
      assert(m->cfd->options()->compaction_style != kCompactionStyleUniversal);
I
Igor Canadi 已提交
2283
      assert(m->cfd->options()->compaction_style != kCompactionStyleFIFO);
2284
      m->tmp_storage = *manual_end;
G
Gabor Cselle 已提交
2285 2286
      m->begin = &m->tmp_storage;
    }
2287
    m->in_progress = false; // not being processed anymore
2288
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
2289
  }
2290
  return status;
J
jorlow@chromium.org 已提交
2291 2292
}

2293
void DBImpl::CleanupCompaction(CompactionState* compact, Status status) {
J
jorlow@chromium.org 已提交
2294
  mutex_.AssertHeld();
2295
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
2296 2297
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
2298
    compact->builder.reset();
J
jorlow@chromium.org 已提交
2299
  } else {
2300
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
2301
  }
D
dgrogan@chromium.org 已提交
2302
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2303 2304
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
2305 2306 2307 2308

    // If this file was inserted into the table cache then remove
    // them here because this compaction was not committed.
    if (!status.ok()) {
I
Igor Canadi 已提交
2309
      TableCache::Evict(table_cache_.get(), out.number);
2310
    }
J
jorlow@chromium.org 已提交
2311 2312 2313 2314
  }
  delete compact;
}

2315
// Allocate the file numbers for the output file. We allocate as
2316
// many output file numbers as there are files in level+1 (at least one)
2317 2318 2319
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2320 2321
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
2322
  int filesNeeded = compact->compaction->num_input_files(1);
2323
  for (int i = 0; i < std::max(filesNeeded, 1); i++) {
2324
    uint64_t file_number = versions_->NewFileNumber();
2325
    pending_outputs_[file_number] = compact->compaction->GetOutputPathId();
2326 2327 2328 2329 2330 2331 2332
    compact->allocated_file_numbers.push_back(file_number);
  }
}

// Frees up unused file number.
void DBImpl::ReleaseCompactionUnusedFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2333
  for (const auto file_number : compact->allocated_file_numbers) {
2334 2335 2336 2337
    pending_outputs_.erase(file_number);
  }
}

J
jorlow@chromium.org 已提交
2338
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
2339 2340
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
2341
  uint64_t file_number;
2342 2343 2344 2345 2346 2347 2348
  // 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 已提交
2349 2350
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
2351
    pending_outputs_[file_number] = compact->compaction->GetOutputPathId();
J
jorlow@chromium.org 已提交
2352 2353
    mutex_.Unlock();
  }
2354 2355
  CompactionState::Output out;
  out.number = file_number;
2356
  out.path_id = compact->compaction->GetOutputPathId();
2357 2358
  out.smallest.Clear();
  out.largest.Clear();
2359
  out.smallest_seqno = out.largest_seqno = 0;
2360
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
2361 2362

  // Make the output file
2363 2364
  std::string fname = TableFileName(options_.db_paths, file_number,
                                    compact->compaction->GetOutputPathId());
2365
  Status s = env_->NewWritableFile(fname, &compact->outfile, storage_options_);
2366

J
jorlow@chromium.org 已提交
2367
  if (s.ok()) {
2368
    compact->outfile->SetPreallocationBlockSize(
2369
        compact->compaction->OutputFilePreallocationSize());
2370

2371
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2372 2373 2374
    compact->builder.reset(NewTableBuilder(
        *cfd->options(), cfd->internal_comparator(), compact->outfile.get(),
        compact->compaction->OutputCompressionType()));
J
jorlow@chromium.org 已提交
2375
  }
I
Igor Canadi 已提交
2376
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2377 2378 2379 2380 2381
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
2382
  assert(compact != nullptr);
2383
  assert(compact->outfile);
2384
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
2385 2386

  const uint64_t output_number = compact->current_output()->number;
2387
  const uint32_t output_path_id = compact->current_output()->path_id;
J
jorlow@chromium.org 已提交
2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400
  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;
2401
  compact->builder.reset();
J
jorlow@chromium.org 已提交
2402 2403

  // Finish and check for file errors
2404
  if (s.ok() && !options_.disableDataSync) {
2405
    if (options_.use_fsync) {
2406
      StopWatch sw(env_, options_.statistics.get(),
2407
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2408 2409
      s = compact->outfile->Fsync();
    } else {
2410
      StopWatch sw(env_, options_.statistics.get(),
2411
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2412 2413
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
2414 2415 2416 2417
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
2418
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
2419 2420 2421

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
I
Igor Canadi 已提交
2422
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2423
    FileDescriptor fd(output_number, output_path_id, current_bytes);
I
Igor Canadi 已提交
2424
    Iterator* iter = cfd->table_cache()->NewIterator(
2425
        ReadOptions(), storage_options_, cfd->internal_comparator(), fd);
J
jorlow@chromium.org 已提交
2426 2427 2428
    s = iter->status();
    delete iter;
    if (s.ok()) {
2429 2430 2431 2432
      Log(options_.info_log, "[%s] Generated table #%" PRIu64 ": %" PRIu64
                             " keys, %" PRIu64 " bytes",
          cfd->GetName().c_str(), output_number, current_entries,
          current_bytes);
J
jorlow@chromium.org 已提交
2433 2434 2435 2436 2437 2438
    }
  }
  return s;
}


2439 2440
Status DBImpl::InstallCompactionResults(CompactionState* compact,
                                        LogBuffer* log_buffer) {
J
jorlow@chromium.org 已提交
2441
  mutex_.AssertHeld();
2442 2443 2444 2445 2446

  // 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.
2447
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
I
Igor Canadi 已提交
2448 2449 2450 2451 2452
    Log(options_.info_log, "[%s] Compaction %d@%d + %d@%d files aborted",
        compact->compaction->column_family_data()->GetName().c_str(),
        compact->compaction->num_input_files(0), compact->compaction->level(),
        compact->compaction->num_input_files(1),
        compact->compaction->output_level());
L
Lei Jin 已提交
2453
    return Status::Corruption("Compaction input files inconsistent");
2454 2455
  }

I
Igor Canadi 已提交
2456 2457
  LogToBuffer(log_buffer, "[%s] Compacted %d@%d + %d@%d files => %lld bytes",
              compact->compaction->column_family_data()->GetName().c_str(),
2458 2459 2460 2461 2462
              compact->compaction->num_input_files(0),
              compact->compaction->level(),
              compact->compaction->num_input_files(1),
              compact->compaction->output_level(),
              static_cast<long long>(compact->total_bytes));
J
jorlow@chromium.org 已提交
2463 2464 2465

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
D
dgrogan@chromium.org 已提交
2466
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2467
    const CompactionState::Output& out = compact->outputs[i];
2468 2469 2470 2471
    compact->compaction->edit()->AddFile(compact->compaction->output_level(),
                                         out.number, out.path_id, out.file_size,
                                         out.smallest, out.largest,
                                         out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2472
  }
I
Igor Canadi 已提交
2473 2474 2475
  return versions_->LogAndApply(compact->compaction->column_family_data(),
                                compact->compaction->edit(), &mutex_,
                                db_directory_.get());
J
jorlow@chromium.org 已提交
2476 2477
}

2478 2479 2480 2481 2482 2483 2484
// 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(
2485 2486
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
2487
  SequenceNumber prev __attribute__((unused)) = 0;
2488 2489 2490
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
2491
      *prev_snapshot = prev;
2492
      return cur;
2493
    }
2494 2495
    prev = cur; // assignment
    assert(prev);
2496 2497
  }
  Log(options_.info_log,
2498 2499
      "Looking for seqid %" PRIu64 " but maxseqid is %" PRIu64 "", in,
      snapshots[snapshots.size() - 1]);
2500 2501 2502 2503
  assert(0);
  return 0;
}

I
Igor Canadi 已提交
2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522
uint64_t DBImpl::CallFlushDuringCompaction(ColumnFamilyData* cfd,
                                           DeletionState& deletion_state,
                                           LogBuffer* log_buffer) {
  if (cfd->imm()->imm_flush_needed.NoBarrier_Load() != nullptr) {
    const uint64_t imm_start = env_->NowMicros();
    mutex_.Lock();
    if (cfd->imm()->IsFlushPending()) {
      cfd->Ref();
      FlushMemTableToOutputFile(cfd, nullptr, deletion_state, log_buffer);
      cfd->Unref();
      bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
    }
    mutex_.Unlock();
    log_buffer->FlushBufferToLog();
    return env_->NowMicros() - imm_start;
  }
  return 0;
}

D
Danny Guo 已提交
2523
Status DBImpl::ProcessKeyValueCompaction(
2524
    bool is_snapshot_supported,
D
Danny Guo 已提交
2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535
    SequenceNumber visible_at_tip,
    SequenceNumber earliest_snapshot,
    SequenceNumber latest_snapshot,
    DeletionState& deletion_state,
    bool bottommost_level,
    int64_t& imm_micros,
    Iterator* input,
    CompactionState* compact,
    bool is_compaction_v2,
    LogBuffer* log_buffer) {
  size_t combined_idx = 0;
J
jorlow@chromium.org 已提交
2536
  Status status;
D
Danny Guo 已提交
2537
  std::string compaction_filter_value;
J
jorlow@chromium.org 已提交
2538
  ParsedInternalKey ikey;
2539
  IterKey current_user_key;
J
jorlow@chromium.org 已提交
2540
  bool has_current_user_key = false;
2541
  IterKey delete_key;
2542 2543
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
2544
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
2545
  ColumnFamilyData* cfd = compact->compaction->column_family_data();
I
Igor Canadi 已提交
2546 2547
  MergeHelper merge(
      cfd->user_comparator(), cfd->options()->merge_operator.get(),
2548 2549
      options_.info_log.get(), cfd->options()->min_partial_merge_operands,
      false /* internal key corruption is expected */);
I
Igor Canadi 已提交
2550
  auto compaction_filter = cfd->options()->compaction_filter;
2551 2552
  std::unique_ptr<CompactionFilter> compaction_filter_from_factory = nullptr;
  if (!compaction_filter) {
2553
    auto context = compact->GetFilterContextV1();
2554
    compaction_filter_from_factory =
I
Igor Canadi 已提交
2555 2556
        cfd->options()->compaction_filter_factory->CreateCompactionFilter(
            context);
2557 2558
    compaction_filter = compaction_filter_from_factory.get();
  }
2559

2560 2561 2562 2563 2564 2565
  while (input->Valid() && !shutting_down_.Acquire_Load() &&
         !cfd->IsDropped()) {
    // FLUSH preempts compaction
    // TODO(icanadi) this currently only checks if flush is necessary on
    // compacting column family. we should also check if flush is necessary on
    // other column families, too
I
Igor Canadi 已提交
2566
    imm_micros += CallFlushDuringCompaction(cfd, deletion_state, log_buffer);
2567

D
Danny Guo 已提交
2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588
    Slice key;
    Slice value;
    // If is_compaction_v2 is on, kv-pairs are reset to the prefix batch.
    // This prefix batch should contain results after calling
    // compaction_filter_v2.
    //
    // If is_compaction_v2 is off, this function will go through all the
    // kv-pairs in input.
    if (!is_compaction_v2) {
      key = input->key();
      value = input->value();
    } else {
      if (combined_idx >= compact->combined_key_buf_.size()) {
        break;
      }
      assert(combined_idx < compact->combined_key_buf_.size());
      key = compact->combined_key_buf_[combined_idx];
      value = compact->combined_value_buf_[combined_idx];

      ++combined_idx;
    }
H
Haobo Xu 已提交
2589

2590
    if (compact->compaction->ShouldStopBefore(key) &&
2591
        compact->builder != nullptr) {
D
Danny Guo 已提交
2592
      status = FinishCompactionOutputFile(compact, input);
2593 2594 2595 2596 2597 2598
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2599
    bool drop = false;
2600
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2601 2602
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2603 2604
      // TODO: error key stays in db forever? Figure out the intention/rationale
      // v10 error v8 : we cannot hide v8 even though it's pretty obvious.
2605
      current_user_key.Clear();
J
jorlow@chromium.org 已提交
2606 2607
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2608
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2609 2610
    } else {
      if (!has_current_user_key ||
2611
          cfd->user_comparator()->Compare(ikey.user_key,
2612
                                          current_user_key.GetKey()) != 0) {
J
jorlow@chromium.org 已提交
2613
        // First occurrence of this user key
L
Lei Jin 已提交
2614
        current_user_key.SetKey(ikey.user_key);
J
jorlow@chromium.org 已提交
2615 2616
        has_current_user_key = true;
        last_sequence_for_key = kMaxSequenceNumber;
2617
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2618
        // apply the compaction filter to the first occurrence of the user key
D
Danny Guo 已提交
2619
        if (compaction_filter && !is_compaction_v2 &&
H
Haobo Xu 已提交
2620 2621 2622 2623 2624 2625 2626 2627 2628
            ikey.type == kTypeValue &&
            (visible_at_tip || ikey.sequence > latest_snapshot)) {
          // If the user has specified a compaction filter and the sequence
          // number is greater than any external snapshot, then invoke the
          // filter.
          // If the return value of the compaction filter is true, replace
          // the entry with a delete marker.
          bool value_changed = false;
          compaction_filter_value.clear();
I
Igor Canadi 已提交
2629 2630 2631
          bool to_delete = compaction_filter->Filter(
              compact->compaction->level(), ikey.user_key, value,
              &compaction_filter_value, &value_changed);
H
Haobo Xu 已提交
2632
          if (to_delete) {
2633 2634 2635
            // make a copy of the original key and convert it to a delete
            delete_key.SetInternalKey(ExtractUserKey(key), ikey.sequence,
                                      kTypeDeletion);
H
Haobo Xu 已提交
2636
            // anchor the key again
2637
            key = delete_key.GetKey();
H
Haobo Xu 已提交
2638 2639 2640 2641
            // needed because ikey is backed by key
            ParseInternalKey(key, &ikey);
            // no value associated with delete
            value.clear();
2642
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2643 2644 2645 2646
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }
J
jorlow@chromium.org 已提交
2647 2648
      }

2649 2650 2651
      // 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.
2652
      SequenceNumber prev_snapshot = 0; // 0 means no previous snapshot
2653 2654 2655 2656
      SequenceNumber visible = visible_at_tip ? visible_at_tip :
        is_snapshot_supported ?  findEarliestVisibleSnapshot(ikey.sequence,
                                  compact->existing_snapshots, &prev_snapshot)
                              : 0;
2657 2658 2659 2660 2661

      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 已提交
2662
        // Hidden by an newer entry for same user key
2663
        // TODO: why not > ?
2664
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2665
        drop = true;    // (A)
2666
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2667
      } else if (ikey.type == kTypeDeletion &&
D
Danny Guo 已提交
2668 2669
          ikey.sequence <= earliest_snapshot &&
          compact->compaction->IsBaseLevelForKey(ikey.user_key)) {
J
jorlow@chromium.org 已提交
2670 2671 2672 2673 2674 2675 2676 2677
        // 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;
2678
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2679 2680 2681 2682 2683 2684 2685
      } else if (ikey.type == kTypeMerge) {
        // We know the merge type entry is not hidden, otherwise we would
        // have hit (A)
        // We encapsulate the merge related state machine in a different
        // object to minimize change to the existing flow. Turn out this
        // logic could also be nicely re-used for memtable flush purge
        // optimization in BuildTable.
D
Danny Guo 已提交
2686 2687 2688 2689 2690 2691
        int steps = 0;
        merge.MergeUntil(input, prev_snapshot, bottommost_level,
            options_.statistics.get(), &steps);
        // Skip the Merge ops
        combined_idx = combined_idx - 1 + steps;

2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709
        current_entry_is_merging = true;
        if (merge.IsSuccess()) {
          // Successfully found Put/Delete/(end-of-key-range) while merging
          // Get the merge result
          key = merge.key();
          ParseInternalKey(key, &ikey);
          value = merge.value();
        } else {
          // Did not find a Put/Delete/(end-of-key-range) while merging
          // We now have some stack of merge operands to write out.
          // NOTE: key,value, and ikey are now referring to old entries.
          //       These will be correctly set below.
          assert(!merge.keys().empty());
          assert(merge.keys().size() == merge.values().size());

          // Hack to make sure last_sequence_for_key is correct
          ParseInternalKey(merge.keys().front(), &ikey);
        }
J
jorlow@chromium.org 已提交
2710 2711 2712
      }

      last_sequence_for_key = ikey.sequence;
2713
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
2714 2715 2716
    }

    if (!drop) {
2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732
      // We may write a single key (e.g.: for Put/Delete or successful merge).
      // Or we may instead have to write a sequence/list of keys.
      // We have to write a sequence iff we have an unsuccessful merge
      bool has_merge_list = current_entry_is_merging && !merge.IsSuccess();
      const std::deque<std::string>* keys = nullptr;
      const std::deque<std::string>* values = nullptr;
      std::deque<std::string>::const_reverse_iterator key_iter;
      std::deque<std::string>::const_reverse_iterator value_iter;
      if (has_merge_list) {
        keys = &merge.keys();
        values = &merge.values();
        key_iter = keys->rbegin();    // The back (*rbegin()) is the first key
        value_iter = values->rbegin();

        key = Slice(*key_iter);
        value = Slice(*value_iter);
2733
      }
2734

2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745
      // If we have a list of keys to write, traverse the list.
      // If we have a single key to write, simply write that key.
      while (true) {
        // Invariant: key,value,ikey will always be the next entry to write
        char* kptr = (char*)key.data();
        std::string kstr;

        // 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.
2746
        if (bottommost_level && ikey.sequence < earliest_snapshot &&
2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757
            ikey.type != kTypeMerge) {
          assert(ikey.type != kTypeDeletion);
          // 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);
        }

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

2759 2760 2761 2762 2763 2764 2765
        // Open output file if necessary
        if (compact->builder == nullptr) {
          status = OpenCompactionOutputFile(compact);
          if (!status.ok()) {
            break;
          }
        }
2766 2767

        SequenceNumber seqno = GetInternalKeySeqno(newkey);
2768 2769
        if (compact->builder->NumEntries() == 0) {
          compact->current_output()->smallest.DecodeFrom(newkey);
2770 2771 2772 2773
          compact->current_output()->smallest_seqno = seqno;
        } else {
          compact->current_output()->smallest_seqno =
            std::min(compact->current_output()->smallest_seqno, seqno);
2774 2775 2776
        }
        compact->current_output()->largest.DecodeFrom(newkey);
        compact->builder->Add(newkey, value);
2777 2778
        compact->current_output()->largest_seqno =
          std::max(compact->current_output()->largest_seqno, seqno);
2779 2780 2781 2782

        // Close output file if it is big enough
        if (compact->builder->FileSize() >=
            compact->compaction->MaxOutputFileSize()) {
D
Danny Guo 已提交
2783
          status = FinishCompactionOutputFile(compact, input);
2784 2785 2786
          if (!status.ok()) {
            break;
          }
J
jorlow@chromium.org 已提交
2787 2788
        }

2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808
        // If we have a list of entries, move to next element
        // If we only had one entry, then break the loop.
        if (has_merge_list) {
          ++key_iter;
          ++value_iter;

          // If at end of list
          if (key_iter == keys->rend() || value_iter == values->rend()) {
            // Sanity Check: if one ends, then both end
            assert(key_iter == keys->rend() && value_iter == values->rend());
            break;
          }

          // Otherwise not at end of list. Update key, value, and ikey.
          key = Slice(*key_iter);
          value = Slice(*value_iter);
          ParseInternalKey(key, &ikey);

        } else{
          // Only had one item to begin with (Put/Delete)
J
jorlow@chromium.org 已提交
2809 2810 2811 2812 2813
          break;
        }
      }
    }

2814
    // MergeUntil has moved input to the next entry
2815
    if (!current_entry_is_merging) {
2816 2817
      input->Next();
    }
J
jorlow@chromium.org 已提交
2818 2819
  }

D
Danny Guo 已提交
2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879
  return status;
}

void DBImpl::CallCompactionFilterV2(CompactionState* compact,
  CompactionFilterV2* compaction_filter_v2) {
  if (compact == nullptr || compaction_filter_v2 == nullptr) {
    return;
  }

  std::vector<Slice> user_key_buf;
  for (const auto& key : compact->ikey_buf_) {
    user_key_buf.emplace_back(key.user_key);
  }

  // If the user has specified a compaction filter and the sequence
  // number is greater than any external snapshot, then invoke the
  // filter.
  // If the return value of the compaction filter is true, replace
  // the entry with a delete marker.
  compact->to_delete_buf_ = compaction_filter_v2->Filter(
      compact->compaction->level(),
      user_key_buf, compact->existing_value_buf_,
      &compact->new_value_buf_,
      &compact->value_changed_buf_);

  // new_value_buf_.size() <= to_delete__buf_.size(). "=" iff all
  // kv-pairs in this compaction run needs to be deleted.
  assert(compact->to_delete_buf_.size() ==
      compact->key_buf_.size());
  assert(compact->to_delete_buf_.size() ==
      compact->existing_value_buf_.size());
  assert(compact->to_delete_buf_.size() ==
      compact->value_changed_buf_.size());

  int new_value_idx = 0;
  for (unsigned int i = 0; i < compact->to_delete_buf_.size(); ++i) {
    if (compact->to_delete_buf_[i]) {
      // update the string buffer directly
      // the Slice buffer points to the updated buffer
      UpdateInternalKey(&compact->key_str_buf_[i][0],
          compact->key_str_buf_[i].size(),
          compact->ikey_buf_[i].sequence,
          kTypeDeletion);

      // no value associated with delete
      compact->existing_value_buf_[i].clear();
      RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
    } else if (compact->value_changed_buf_[i]) {
      compact->existing_value_buf_[i] =
        Slice(compact->new_value_buf_[new_value_idx++]);
    }
  }  // for
}

Status DBImpl::DoCompactionWork(CompactionState* compact,
                                DeletionState& deletion_state,
                                LogBuffer* log_buffer) {
  assert(compact);
  compact->CleanupBatchBuffer();
  compact->CleanupMergedBuffer();
2880
  bool prefix_initialized = false;
D
Danny Guo 已提交
2881 2882

  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions
2883
  ColumnFamilyData* cfd = compact->compaction->column_family_data();
2884 2885
  LogToBuffer(
      log_buffer,
I
Igor Canadi 已提交
2886 2887
      "[%s] Compacting %d@%d + %d@%d files, score %.2f slots available %d",
      cfd->GetName().c_str(), compact->compaction->num_input_files(0),
2888 2889
      compact->compaction->level(), compact->compaction->num_input_files(1),
      compact->compaction->output_level(), compact->compaction->score(),
D
Danny Guo 已提交
2890 2891 2892
      options_.max_background_compactions - bg_compaction_scheduled_);
  char scratch[2345];
  compact->compaction->Summary(scratch, sizeof(scratch));
I
Igor Canadi 已提交
2893 2894
  LogToBuffer(log_buffer, "[%s] Compaction start summary: %s\n",
              cfd->GetName().c_str(), scratch);
D
Danny Guo 已提交
2895

2896
  assert(cfd->current()->NumLevelFiles(compact->compaction->level()) > 0);
D
Danny Guo 已提交
2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921
  assert(compact->builder == nullptr);
  assert(!compact->outfile);

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
  SequenceNumber latest_snapshot = 0;
  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;
  } else {
    latest_snapshot = compact->existing_snapshots.back();
    // 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];
  }

  // Is this compaction producing files at the bottommost level?
  bool bottommost_level = compact->compaction->BottomMostLevel();

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

2922
  bool is_snapshot_supported = IsSnapshotSupported();
D
Danny Guo 已提交
2923 2924
  // Release mutex while we're actually doing the compaction work
  mutex_.Unlock();
2925
  log_buffer->FlushBufferToLog();
D
Danny Guo 已提交
2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939

  const uint64_t start_micros = env_->NowMicros();
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
  input->SeekToFirst();
  shared_ptr<Iterator> backup_input(
      versions_->MakeInputIterator(compact->compaction));
  backup_input->SeekToFirst();

  Status status;
  ParsedInternalKey ikey;
  std::unique_ptr<CompactionFilterV2> compaction_filter_from_factory_v2
    = nullptr;
  auto context = compact->GetFilterContext();
  compaction_filter_from_factory_v2 =
2940 2941
      cfd->options()->compaction_filter_factory_v2->CreateCompactionFilterV2(
          context);
D
Danny Guo 已提交
2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952
  auto compaction_filter_v2 =
    compaction_filter_from_factory_v2.get();

  // temp_backup_input always point to the start of the current buffer
  // temp_backup_input = backup_input;
  // iterate through input,
  // 1) buffer ineligible keys and value keys into 2 separate buffers;
  // 2) send value_buffer to compaction filter and alternate the values;
  // 3) merge value_buffer with ineligible_value_buffer;
  // 4) run the modified "compaction" using the old for loop.
  if (compaction_filter_v2) {
2953 2954
    while (backup_input->Valid() && !shutting_down_.Acquire_Load() &&
           !cfd->IsDropped()) {
I
Igor Canadi 已提交
2955 2956 2957 2958 2959 2960
      // FLUSH preempts compaction
      // TODO(icanadi) this currently only checks if flush is necessary on
      // compacting column family. we should also check if flush is necessary on
      // other column families, too
      imm_micros += CallFlushDuringCompaction(cfd, deletion_state, log_buffer);

D
Danny Guo 已提交
2961 2962 2963 2964
      Slice key = backup_input->key();
      Slice value = backup_input->value();

      const SliceTransform* transformer =
2965
          cfd->options()->compaction_filter_factory_v2->GetPrefixExtractor();
2966
      const auto key_prefix = transformer->Transform(key);
2967
      if (!prefix_initialized) {
2968
        compact->cur_prefix_ = key_prefix.ToString();
2969
        prefix_initialized = true;
D
Danny Guo 已提交
2970 2971 2972
      }
      if (!ParseInternalKey(key, &ikey)) {
        // log error
I
Igor Canadi 已提交
2973 2974
        Log(options_.info_log, "[%s] Failed to parse key: %s",
            cfd->GetName().c_str(), key.ToString().c_str());
D
Danny Guo 已提交
2975 2976 2977
        continue;
      } else {
        // If the prefix remains the same, keep buffering
2978
        if (key_prefix.compare(Slice(compact->cur_prefix_)) == 0) {
D
Danny Guo 已提交
2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998
          // Apply the compaction filter V2 to all the kv pairs sharing
          // the same prefix
          if (ikey.type == kTypeValue &&
              (visible_at_tip || ikey.sequence > latest_snapshot)) {
            // Buffer all keys sharing the same prefix for CompactionFilterV2
            // Iterate through keys to check prefix
            compact->BufferKeyValueSlices(key, value);
          } else {
            // buffer ineligible keys
            compact->BufferOtherKeyValueSlices(key, value);
          }
          backup_input->Next();
          continue;
          // finish changing values for eligible keys
        } else {
          // Now prefix changes, this batch is done.
          // Call compaction filter on the buffered values to change the value
          if (compact->key_buf_.size() > 0) {
            CallCompactionFilterV2(compact, compaction_filter_v2);
          }
2999
          compact->cur_prefix_ = key_prefix.ToString();
D
Danny Guo 已提交
3000 3001 3002 3003
        }
      }

      // Merge this batch of data (values + ineligible keys)
3004
      compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
3005 3006 3007 3008

      // Done buffering for the current prefix. Spit it out to disk
      // Now just iterate through all the kv-pairs
      status = ProcessKeyValueCompaction(
3009
          is_snapshot_supported,
D
Danny Guo 已提交
3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041
          visible_at_tip,
          earliest_snapshot,
          latest_snapshot,
          deletion_state,
          bottommost_level,
          imm_micros,
          input.get(),
          compact,
          true,
          log_buffer);

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

      // After writing the kv-pairs, we can safely remove the reference
      // to the string buffer and clean them up
      compact->CleanupBatchBuffer();
      compact->CleanupMergedBuffer();
      // Buffer the key that triggers the mismatch in prefix
      if (ikey.type == kTypeValue &&
        (visible_at_tip || ikey.sequence > latest_snapshot)) {
        compact->BufferKeyValueSlices(key, value);
      } else {
        compact->BufferOtherKeyValueSlices(key, value);
      }
      backup_input->Next();
      if (!backup_input->Valid()) {
        // If this is the single last value, we need to merge it.
        if (compact->key_buf_.size() > 0) {
          CallCompactionFilterV2(compact, compaction_filter_v2);
        }
3042
        compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
3043 3044

        status = ProcessKeyValueCompaction(
3045
            is_snapshot_supported,
D
Danny Guo 已提交
3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064
            visible_at_tip,
            earliest_snapshot,
            latest_snapshot,
            deletion_state,
            bottommost_level,
            imm_micros,
            input.get(),
            compact,
            true,
            log_buffer);

        compact->CleanupBatchBuffer();
        compact->CleanupMergedBuffer();
      }
    }  // done processing all prefix batches
    // finish the last batch
    if (compact->key_buf_.size() > 0) {
      CallCompactionFilterV2(compact, compaction_filter_v2);
    }
3065
    compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
3066
    status = ProcessKeyValueCompaction(
3067
        is_snapshot_supported,
D
Danny Guo 已提交
3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081
        visible_at_tip,
        earliest_snapshot,
        latest_snapshot,
        deletion_state,
        bottommost_level,
        imm_micros,
        input.get(),
        compact,
        true,
        log_buffer);
  }  // checking for compaction filter v2

  if (!compaction_filter_v2) {
    status = ProcessKeyValueCompaction(
3082
      is_snapshot_supported,
D
Danny Guo 已提交
3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094
      visible_at_tip,
      earliest_snapshot,
      latest_snapshot,
      deletion_state,
      bottommost_level,
      imm_micros,
      input.get(),
      compact,
      false,
      log_buffer);
  }

3095
  if (status.ok() && (shutting_down_.Acquire_Load() || cfd->IsDropped())) {
L
Lei Jin 已提交
3096
    status = Status::ShutdownInProgress(
3097
        "Database shutdown or Column family drop during compaction");
J
jorlow@chromium.org 已提交
3098
  }
3099
  if (status.ok() && compact->builder != nullptr) {
3100
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
3101 3102 3103 3104
  }
  if (status.ok()) {
    status = input->status();
  }
3105
  input.reset();
J
jorlow@chromium.org 已提交
3106

3107 3108 3109
  if (!options_.disableDataSync) {
    db_directory_->Fsync();
  }
I
Igor Canadi 已提交
3110 3111

  InternalStats::CompactionStats stats;
3112
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
I
Igor Canadi 已提交
3113
  MeasureTime(options_.statistics.get(), COMPACTION_TIME, stats.micros);
M
Mark Callaghan 已提交
3114 3115
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
3116 3117

  int num_output_files = compact->outputs.size();
3118
  if (compact->builder != nullptr) {
P
Pascal Borreli 已提交
3119
    // An error occurred so ignore the last output.
3120 3121 3122 3123
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
3124

3125
  for (int i = 0; i < compact->compaction->num_input_files(0); i++) {
3126
    stats.bytes_readn += compact->compaction->input(0, i)->fd.GetFileSize();
3127
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
3128
               compact->compaction->input(0, i)->fd.GetFileSize());
3129
  }
M
Mark Callaghan 已提交
3130

3131
  for (int i = 0; i < compact->compaction->num_input_files(1); i++) {
3132
    stats.bytes_readnp1 += compact->compaction->input(1, i)->fd.GetFileSize();
3133
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
3134
               compact->compaction->input(1, i)->fd.GetFileSize());
3135
  }
M
Mark Callaghan 已提交
3136

3137
  for (int i = 0; i < num_output_files; i++) {
3138
    stats.bytes_written += compact->outputs[i].file_size;
3139 3140
    RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
               compact->outputs[i].file_size);
3141 3142
  }

I
Igor Canadi 已提交
3143
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
3144
  mutex_.Lock();
3145 3146
  cfd->internal_stats()->AddCompactionStats(compact->compaction->output_level(),
                                            stats);
J
jorlow@chromium.org 已提交
3147

3148 3149 3150 3151
  // 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 已提交
3152
  if (status.ok()) {
3153
    status = InstallCompactionResults(compact, log_buffer);
I
Igor Canadi 已提交
3154
    InstallSuperVersion(cfd, deletion_state);
J
jorlow@chromium.org 已提交
3155
  }
3156
  Version::LevelSummaryStorage tmp;
3157 3158
  LogToBuffer(
      log_buffer,
I
Igor Canadi 已提交
3159
      "[%s] compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
3160 3161
      "MB in(%.1f, %.1f) out(%.1f), read-write-amplify(%.1f) "
      "write-amplify(%.1f) %s\n",
I
Igor Canadi 已提交
3162
      cfd->GetName().c_str(), cfd->current()->LevelSummary(&tmp),
M
Mark Callaghan 已提交
3163
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
3164 3165 3166 3167
          (double)stats.micros,
      compact->compaction->output_level(), stats.files_in_leveln,
      stats.files_in_levelnp1, stats.files_out_levelnp1,
      stats.bytes_readn / 1048576.0, stats.bytes_readnp1 / 1048576.0,
M
Mark Callaghan 已提交
3168
      stats.bytes_written / 1048576.0,
3169
      (stats.bytes_written + stats.bytes_readnp1 + stats.bytes_readn) /
3170 3171
          (double)stats.bytes_readn,
      stats.bytes_written / (double)stats.bytes_readn,
3172
      status.ToString().c_str());
M
Mark Callaghan 已提交
3173

J
jorlow@chromium.org 已提交
3174 3175 3176
  return status;
}

3177 3178
namespace {
struct IterState {
3179 3180
  IterState(DBImpl* db, port::Mutex* mu, SuperVersion* super_version)
      : db(db), mu(mu), super_version(super_version) {}
3181 3182

  DBImpl* db;
3183
  port::Mutex* mu;
3184
  SuperVersion* super_version;
3185 3186 3187 3188
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
3189

3190
  if (state->super_version->Unref()) {
3191 3192
    DBImpl::DeletionState deletion_state;

3193 3194 3195 3196 3197 3198
    state->mu->Lock();
    state->super_version->Cleanup();
    state->db->FindObsoleteFiles(deletion_state, false, true);
    state->mu->Unlock();

    delete state->super_version;
3199 3200 3201
    if (deletion_state.HaveSomethingToDelete()) {
      state->db->PurgeObsoleteFiles(deletion_state);
    }
I
Igor Canadi 已提交
3202
  }
T
Tomislav Novak 已提交
3203

3204 3205
  delete state;
}
H
Hans Wennborg 已提交
3206
}  // namespace
3207

J
jorlow@chromium.org 已提交
3208
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
3209
                                      ColumnFamilyData* cfd,
3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237
                                      SuperVersion* super_version,
                                      Arena* arena) {
  Iterator* internal_iter;
  if (arena != nullptr) {
    // Need to create internal iterator from the arena.
    MergeIteratorBuilder merge_iter_builder(&cfd->internal_comparator(), arena);
    // Collect iterator for mutable mem
    merge_iter_builder.AddIterator(
        super_version->mem->NewIterator(options, false, arena));
    // Collect all needed child iterators for immutable memtables
    super_version->imm->AddIterators(options, &merge_iter_builder);
    // Collect iterators for files in L0 - Ln
    super_version->current->AddIterators(options, storage_options_,
                                         &merge_iter_builder);
    internal_iter = merge_iter_builder.Finish();
  } else {
    // Need to create internal iterator using malloc.
    std::vector<Iterator*> iterator_list;
    // Collect iterator for mutable mem
    iterator_list.push_back(super_version->mem->NewIterator(options));
    // Collect all needed child iterators for immutable memtables
    super_version->imm->AddIterators(options, &iterator_list);
    // Collect iterators for files in L0 - Ln
    super_version->current->AddIterators(options, storage_options_,
                                         &iterator_list);
    internal_iter = NewMergingIterator(&cfd->internal_comparator(),
                                       &iterator_list[0], iterator_list.size());
  }
3238
  IterState* cleanup = new IterState(this, &mutex_, super_version);
3239
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
3240 3241 3242 3243

  return internal_iter;
}

3244 3245 3246 3247
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

J
jorlow@chromium.org 已提交
3248
Status DBImpl::Get(const ReadOptions& options,
3249
                   ColumnFamilyHandle* column_family, const Slice& key,
J
jorlow@chromium.org 已提交
3250
                   std::string* value) {
3251
  return GetImpl(options, column_family, key, value);
3252 3253
}

I
Igor Canadi 已提交
3254 3255 3256
// DeletionState gets created and destructed outside of the lock -- we
// use this convinently to:
// * malloc one SuperVersion() outside of the lock -- new_superversion
3257
// * delete SuperVersion()s outside of the lock -- superversions_to_free
I
Igor Canadi 已提交
3258 3259 3260 3261 3262 3263
//
// However, if InstallSuperVersion() gets called twice with the same,
// deletion_state, we can't reuse the SuperVersion() that got malloced because
// first call already used it. In that rare case, we take a hit and create a
// new SuperVersion() inside of the mutex. We do similar thing
// for superversion_to_free
3264 3265
void DBImpl::InstallSuperVersion(ColumnFamilyData* cfd,
                                 DeletionState& deletion_state) {
3266
  mutex_.AssertHeld();
I
Igor Canadi 已提交
3267 3268 3269 3270
  // if new_superversion == nullptr, it means somebody already used it
  SuperVersion* new_superversion =
    (deletion_state.new_superversion != nullptr) ?
    deletion_state.new_superversion : new SuperVersion();
3271 3272
  SuperVersion* old_superversion =
      cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
3273
  deletion_state.new_superversion = nullptr;
3274
  deletion_state.superversions_to_free.push_back(old_superversion);
I
Igor Canadi 已提交
3275 3276
}

3277
Status DBImpl::GetImpl(const ReadOptions& options,
3278 3279
                       ColumnFamilyHandle* column_family, const Slice& key,
                       std::string* value, bool* value_found) {
3280
  StopWatch sw(env_, options_.statistics.get(), DB_GET, false);
L
Lei Jin 已提交
3281
  PERF_TIMER_AUTO(get_snapshot_time);
3282

3283 3284 3285
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

3286
  SequenceNumber snapshot;
3287
  if (options.snapshot != nullptr) {
3288 3289 3290
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
3291
  }
3292

3293 3294
  // Acquire SuperVersion
  SuperVersion* sv = nullptr;
3295
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
3296
  if (LIKELY(options_.allow_thread_local)) {
3297
    sv = cfd->GetThreadLocalSuperVersion(&mutex_);
3298 3299
  } else {
    mutex_.Lock();
3300
    sv = cfd->GetSuperVersion()->Ref();
3301 3302
    mutex_.Unlock();
  }
I
Igor Canadi 已提交
3303

3304
  // Prepare to store a list of merge operations if merge occurs.
3305
  MergeContext merge_context;
3306

3307
  Status s;
3308
  // First look in the memtable, then in the immutable memtable (if any).
3309
  // s is both in/out. When in, s could either be OK or MergeInProgress.
3310
  // merge_operands will contain the sequence of merges in the latter case.
3311
  LookupKey lkey(key, snapshot);
L
Lei Jin 已提交
3312
  PERF_TIMER_STOP(get_snapshot_time);
I
Igor Canadi 已提交
3313
  if (sv->mem->Get(lkey, value, &s, merge_context, *cfd->options())) {
3314
    // Done
3315
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
3316
  } else if (sv->imm->Get(lkey, value, &s, merge_context, *cfd->options())) {
3317
    // Done
3318
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
3319
  } else {
L
Lei Jin 已提交
3320
    PERF_TIMER_START(get_from_output_files_time);
3321

I
Igor Canadi 已提交
3322
    sv->current->Get(options, lkey, value, &s, &merge_context, value_found);
L
Lei Jin 已提交
3323
    PERF_TIMER_STOP(get_from_output_files_time);
3324
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
3325
  }
3326

L
Lei Jin 已提交
3327
  PERF_TIMER_START(get_post_process_time);
3328

3329
  bool unref_sv = true;
3330
  if (LIKELY(options_.allow_thread_local)) {
3331
    unref_sv = !cfd->ReturnThreadLocalSuperVersion(sv);
3332 3333 3334 3335
  }

  if (unref_sv) {
    // Release SuperVersion
3336
    if (sv->Unref()) {
I
Igor Canadi 已提交
3337
      mutex_.Lock();
3338
      sv->Cleanup();
I
Igor Canadi 已提交
3339
      mutex_.Unlock();
3340
      delete sv;
3341
      RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_CLEANUPS);
I
Igor Canadi 已提交
3342
    }
3343
    RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_RELEASES);
3344
  }
3345

3346 3347
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
L
Lei Jin 已提交
3348
  PERF_TIMER_STOP(get_post_process_time);
3349
  return s;
J
jorlow@chromium.org 已提交
3350 3351
}

3352 3353
std::vector<Status> DBImpl::MultiGet(
    const ReadOptions& options,
3354
    const std::vector<ColumnFamilyHandle*>& column_family,
3355
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
3356

3357
  StopWatch sw(env_, options_.statistics.get(), DB_MULTIGET, false);
L
Lei Jin 已提交
3358
  PERF_TIMER_AUTO(get_snapshot_time);
K
Kai Liu 已提交
3359

3360
  SequenceNumber snapshot;
3361

3362
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
3363
    ColumnFamilyData* cfd;
3364 3365 3366 3367 3368
    SuperVersion* super_version;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
3369 3370 3371 3372 3373 3374
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(cf);
    auto cfd = cfh->cfd();
    if (multiget_cf_data.find(cfd->GetID()) == multiget_cf_data.end()) {
      auto mgcfd = new MultiGetColumnFamilyData();
      mgcfd->cfd = cfd;
      multiget_cf_data.insert({cfd->GetID(), mgcfd});
3375 3376 3377
    }
  }

3378
  mutex_.Lock();
3379 3380 3381 3382 3383
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }
3384
  for (auto mgd_iter : multiget_cf_data) {
3385 3386
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
3387
  }
3388
  mutex_.Unlock();
3389

3390 3391
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
3392

3393
  // Note: this always resizes the values array
3394 3395 3396
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
3397 3398

  // Keep track of bytes that we read for statistics-recording later
3399
  uint64_t bytes_read = 0;
L
Lei Jin 已提交
3400
  PERF_TIMER_STOP(get_snapshot_time);
3401 3402 3403 3404

  // For each of the given keys, apply the entire "get" process as follows:
  // First look in the memtable, then in the immutable memtable (if any).
  // s is both in/out. When in, s could either be OK or MergeInProgress.
3405
  // merge_operands will contain the sequence of merges in the latter case.
3406
  for (size_t i = 0; i < num_keys; ++i) {
3407
    merge_context.Clear();
3408
    Status& s = stat_list[i];
3409 3410 3411
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
3412 3413
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
3414 3415 3416
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
I
Igor Canadi 已提交
3417 3418
    auto cfd = mgd->cfd;
    if (super_version->mem->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3419
                                *cfd->options())) {
3420
      // Done
3421
    } else if (super_version->imm->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3422
                                       *cfd->options())) {
3423 3424
      // Done
    } else {
I
Igor Canadi 已提交
3425
      super_version->current->Get(options, lkey, value, &s, &merge_context);
3426 3427 3428
    }

    if (s.ok()) {
3429
      bytes_read += value->size();
3430 3431 3432 3433
    }
  }

  // Post processing (decrement reference counts and record statistics)
L
Lei Jin 已提交
3434
  PERF_TIMER_START(get_post_process_time);
3435 3436
  autovector<SuperVersion*> superversions_to_delete;

I
Igor Canadi 已提交
3437
  // TODO(icanadi) do we need lock here or just around Cleanup()?
3438 3439 3440 3441 3442 3443
  mutex_.Lock();
  for (auto mgd_iter : multiget_cf_data) {
    auto mgd = mgd_iter.second;
    if (mgd->super_version->Unref()) {
      mgd->super_version->Cleanup();
      superversions_to_delete.push_back(mgd->super_version);
3444 3445
    }
  }
3446 3447 3448 3449 3450 3451 3452
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
3453
  }
3454

3455
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_CALLS);
3456 3457
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_KEYS_READ, num_keys);
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_BYTES_READ, bytes_read);
L
Lei Jin 已提交
3458
  PERF_TIMER_STOP(get_post_process_time);
3459

3460
  return stat_list;
3461 3462
}

3463
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& options,
3464
                                  const std::string& column_family_name,
3465
                                  ColumnFamilyHandle** handle) {
I
Igor Canadi 已提交
3466 3467 3468
  *handle = nullptr;
  MutexLock l(&mutex_);

I
Igor Canadi 已提交
3469 3470
  if (versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name) !=
      nullptr) {
I
Igor Canadi 已提交
3471 3472
    return Status::InvalidArgument("Column family already exists");
  }
3473
  VersionEdit edit;
3474
  edit.AddColumnFamily(column_family_name);
3475 3476
  uint32_t new_id = versions_->GetColumnFamilySet()->GetNextColumnFamilyID();
  edit.SetColumnFamily(new_id);
I
Igor Canadi 已提交
3477
  edit.SetLogNumber(logfile_number_);
I
Igor Canadi 已提交
3478
  edit.SetComparatorName(options.comparator->Name());
3479

I
Igor Canadi 已提交
3480 3481
  // LogAndApply will both write the creation in MANIFEST and create
  // ColumnFamilyData object
3482 3483
  Status s = versions_->LogAndApply(nullptr, &edit, &mutex_,
                                    db_directory_.get(), false, &options);
3484
  if (s.ok()) {
3485
    single_column_family_mode_ = false;
3486 3487 3488
    auto cfd =
        versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
    assert(cfd != nullptr);
3489
    delete cfd->InstallSuperVersion(new SuperVersion(), &mutex_);
3490
    *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
I
Igor Canadi 已提交
3491
    Log(options_.info_log, "Created column family [%s] (ID %u)",
3492
        column_family_name.c_str(), (unsigned)cfd->GetID());
I
Igor Canadi 已提交
3493 3494
    max_total_in_memory_state_ += cfd->options()->write_buffer_size *
                                  cfd->options()->max_write_buffer_number;
3495
  } else {
I
Igor Canadi 已提交
3496
    Log(options_.info_log, "Creating column family [%s] FAILED -- %s",
3497 3498
        column_family_name.c_str(), s.ToString().c_str());
  }
3499
  return s;
3500 3501
}

3502 3503 3504 3505
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
3506 3507
    return Status::InvalidArgument("Can't drop default column family");
  }
3508

I
Igor Canadi 已提交
3509 3510
  VersionEdit edit;
  edit.DropColumnFamily();
3511 3512
  edit.SetColumnFamily(cfd->GetID());

3513
  Status s;
3514 3515 3516 3517 3518 3519 3520 3521
  {
    MutexLock l(&mutex_);
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
      s = versions_->LogAndApply(cfd, &edit, &mutex_);
    }
3522
  }
3523

3524
  if (s.ok()) {
I
Igor Canadi 已提交
3525
    assert(cfd->IsDropped());
I
Igor Canadi 已提交
3526 3527
    max_total_in_memory_state_ -= cfd->options()->write_buffer_size *
                                  cfd->options()->max_write_buffer_number;
3528
    Log(options_.info_log, "Dropped column family with id %u\n", cfd->GetID());
3529
    // Flush the memtables. This will make all WAL files referencing dropped
I
Igor Canadi 已提交
3530
    // column family to be obsolete. They will be deleted once user deletes
3531 3532
    // column family handle
    Write(WriteOptions(), nullptr);  // ignore error
3533 3534 3535 3536 3537
  } else {
    Log(options_.info_log, "Dropping column family with id %u FAILED -- %s\n",
        cfd->GetID(), s.ToString().c_str());
  }

3538
  return s;
3539 3540
}

3541
bool DBImpl::KeyMayExist(const ReadOptions& options,
3542 3543
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
3544
  if (value_found != nullptr) {
K
Kai Liu 已提交
3545 3546
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
3547
  }
3548 3549
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
3550
  auto s = GetImpl(roptions, column_family, key, value, value_found);
K
Kai Liu 已提交
3551 3552 3553 3554 3555

  // If options.block_cache != nullptr and the index block of the table didn't
  // not present in block_cache, the return value will be Status::Incomplete.
  // In this case, key may still exist in the table.
  return s.ok() || s.IsIncomplete();
3556 3557
}

3558
Iterator* DBImpl::NewIterator(const ReadOptions& options,
3559 3560 3561
                              ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
T
Tomislav Novak 已提交
3562 3563

  if (options.tailing) {
I
Igor Canadi 已提交
3564 3565 3566 3567
#ifdef ROCKSDB_LITE
    // not supported in lite version
    return nullptr;
#else
L
Lei Jin 已提交
3568
    // TODO(ljin): remove tailing iterator
I
Igor Canadi 已提交
3569
    auto iter = new ForwardIterator(this, options, cfd);
3570 3571 3572
    return NewDBIterator(env_, *cfd->options(), cfd->user_comparator(), iter,
                         kMaxSequenceNumber);
// return new TailingIterator(env_, this, options, cfd);
I
Igor Canadi 已提交
3573
#endif
T
Tomislav Novak 已提交
3574
  } else {
3575 3576 3577 3578
    SequenceNumber latest_snapshot = versions_->LastSequence();
    SuperVersion* sv = nullptr;
    sv = cfd->GetReferencedSuperVersion(&mutex_);

I
Igor Canadi 已提交
3579 3580 3581 3582
    auto snapshot =
        options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
            : latest_snapshot;
T
Tomislav Novak 已提交
3583

3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633
    // Try to generate a DB iterator tree in continuous memory area to be
    // cache friendly. Here is an example of result:
    // +-------------------------------+
    // |                               |
    // | ArenaWrappedDBIter            |
    // |  +                            |
    // |  +---> Inner Iterator   ------------+
    // |  |                            |     |
    // |  |    +-- -- -- -- -- -- -- --+     |
    // |  +--- | Arena                 |     |
    // |       |                       |     |
    // |          Allocated Memory:    |     |
    // |       |   +-------------------+     |
    // |       |   | DBIter            | <---+
    // |           |  +                |
    // |       |   |  +-> iter_  ------------+
    // |       |   |                   |     |
    // |       |   +-------------------+     |
    // |       |   | MergingIterator   | <---+
    // |           |  +                |
    // |       |   |  +->child iter1  ------------+
    // |       |   |  |                |          |
    // |           |  +->child iter2  ----------+ |
    // |       |   |  |                |        | |
    // |       |   |  +->child iter3  --------+ | |
    // |           |                   |      | | |
    // |       |   +-------------------+      | | |
    // |       |   | Iterator1         | <--------+
    // |       |   +-------------------+      | |
    // |       |   | Iterator2         | <------+
    // |       |   +-------------------+      |
    // |       |   | Iterator3         | <----+
    // |       |   +-------------------+
    // |       |                       |
    // +-------+-----------------------+
    //
    // ArenaWrappedDBIter inlines an arena area where all the iterartor in the
    // the iterator tree is allocated in the order of being accessed when
    // querying.
    // Laying out the iterators in the order of being accessed makes it more
    // likely that any iterator pointer is close to the iterator it points to so
    // that they are likely to be in the same cache line and/or page.
    ArenaWrappedDBIter* db_iter = NewArenaWrappedDbIterator(
        env_, *cfd->options(), cfd->user_comparator(), snapshot);
    Iterator* internal_iter =
        NewInternalIterator(options, cfd, sv, db_iter->GetArena());
    db_iter->SetIterUnderDBIter(internal_iter);

    return db_iter;
  }
J
jorlow@chromium.org 已提交
3634 3635
}

3636 3637
Status DBImpl::NewIterators(
    const ReadOptions& options,
I
Igor Canadi 已提交
3638
    const std::vector<ColumnFamilyHandle*>& column_families,
3639
    std::vector<Iterator*>* iterators) {
I
Igor Canadi 已提交
3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656
  iterators->clear();
  iterators->reserve(column_families.size());
  SequenceNumber latest_snapshot = 0;
  std::vector<SuperVersion*> super_versions;
  super_versions.reserve(column_families.size());

  if (!options.tailing) {
    mutex_.Lock();
    latest_snapshot = versions_->LastSequence();
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      super_versions.push_back(cfd->GetSuperVersion()->Ref());
    }
    mutex_.Unlock();
  }

  if (options.tailing) {
I
Igor Canadi 已提交
3657 3658 3659 3660
#ifdef ROCKSDB_LITE
    return Status::InvalidArgument(
        "Tailing interator not supported in RocksDB lite");
#else
I
Igor Canadi 已提交
3661 3662
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
L
Lei Jin 已提交
3663
      iterators->push_back(new TailingIterator(env_, this, options, cfd));
I
Igor Canadi 已提交
3664
    }
I
Igor Canadi 已提交
3665
#endif
I
Igor Canadi 已提交
3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676
  } else {
    for (size_t i = 0; i < column_families.size(); ++i) {
      auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_families[i]);
      auto cfd = cfh->cfd();

      auto snapshot =
          options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
              : latest_snapshot;

      auto iter = NewInternalIterator(options, cfd, super_versions[i]);
L
Lei Jin 已提交
3677
      iter = NewDBIterator(env_, *cfd->options(),
I
Igor Canadi 已提交
3678 3679 3680 3681 3682 3683
                           cfd->user_comparator(), iter, snapshot);
      iterators->push_back(iter);
    }
  }

  return Status::OK();
3684 3685
}

3686 3687 3688 3689 3690 3691 3692 3693 3694
bool DBImpl::IsSnapshotSupported() const {
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    if (!cfd->mem()->IsSnapshotSupported()) {
      return false;
    }
  }
  return true;
}

J
jorlow@chromium.org 已提交
3695
const Snapshot* DBImpl::GetSnapshot() {
3696
  MutexLock l(&mutex_);
3697 3698
  // returns null if the underlying memtable does not support snapshot.
  if (!IsSnapshotSupported()) return nullptr;
3699
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
3700 3701 3702 3703
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
3704
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
3705 3706 3707
}

// Convenience methods
3708 3709
Status DBImpl::Put(const WriteOptions& o, ColumnFamilyHandle* column_family,
                   const Slice& key, const Slice& val) {
3710
  return DB::Put(o, column_family, key, val);
J
jorlow@chromium.org 已提交
3711 3712
}

3713 3714 3715 3716
Status DBImpl::Merge(const WriteOptions& o, ColumnFamilyHandle* column_family,
                     const Slice& key, const Slice& val) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  if (!cfh->cfd()->options()->merge_operator) {
3717 3718
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
3719
    return DB::Merge(o, column_family, key, val);
3720 3721 3722
  }
}

3723
Status DBImpl::Delete(const WriteOptions& options,
3724
                      ColumnFamilyHandle* column_family, const Slice& key) {
3725
  return DB::Delete(options, column_family, key);
J
jorlow@chromium.org 已提交
3726 3727
}

3728
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
L
Lei Jin 已提交
3729
  PERF_TIMER_AUTO(write_pre_and_post_process_time);
3730 3731 3732
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
3733
  w.disableWAL = options.disableWAL;
3734 3735 3736 3737 3738 3739 3740 3741 3742 3743
  w.in_batch_group = false;
  w.done = false;
  w.timeout_hint_us = options.timeout_hint_us;

  uint64_t expiration_time = 0;
  if (w.timeout_hint_us == 0) {
    w.timeout_hint_us = kNoTimeOut;
  } else {
    expiration_time = env_->NowMicros() + w.timeout_hint_us;
  }
3744
  w.done = false;
3745

I
Igor Canadi 已提交
3746
  mutex_.Lock();
3747 3748 3749 3750 3751
  // the following code block pushes the current writer "w" into the writer
  // queue "writers_" and wait until one of the following conditions met:
  // 1. the job of "w" has been done by some other writers.
  // 2. "w" becomes the first writer in "writers_"
  // 3. "w" timed-out.
3752
  writers_.push_back(&w);
3753 3754

  bool timed_out = false;
3755
  while (!w.done && &w != writers_.front()) {
3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768
    if (expiration_time == 0) {
      w.cv.Wait();
    } else if (w.cv.TimedWait(expiration_time)) {
      if (w.in_batch_group) {
        // then it means the front writer is currently doing the
        // write on behalf of this "timed-out" writer.  Then it
        // should wait until the write completes.
        expiration_time = 0;
      } else {
        timed_out = true;
        break;
      }
    }
3769
  }
3770 3771 3772 3773 3774

  if (!options.disableWAL) {
    RecordTick(options_.statistics.get(), WRITE_WITH_WAL, 1);
  }

3775
  if (w.done) {
I
Igor Canadi 已提交
3776
    mutex_.Unlock();
3777
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
3778
    return w.status;
3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797
  } else if (timed_out) {
    bool found = false;
    for (auto iter = writers_.begin(); iter != writers_.end(); iter++) {
      if (*iter == &w) {
        writers_.erase(iter);
        found = true;
        break;
      }
    }
    assert(found);
    // writers_.front() might still be in cond_wait without a time-out.
    // As a result, we need to signal it to wake it up.  Otherwise no
    // one else will wake him up, and RocksDB will hang.
    if (!writers_.empty()) {
      writers_.front()->cv.Signal();
    }
    mutex_.Unlock();
    RecordTick(options_.statistics.get(), WRITE_TIMEDOUT, 1);
    return Status::TimedOut();
3798 3799
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
3800 3801
  }

3802 3803 3804 3805
  // Once reaches this point, the current writer "w" will try to do its write
  // job.  It may also pick up some of the remaining writers in the "writers_"
  // when it finds suitable, and finish them in the same write batch.
  // This is how a write job could be done by the other writer.
3806 3807 3808
  assert(!single_column_family_mode_ ||
         versions_->GetColumnFamilySet()->NumberOfColumnFamilies() == 1);

I
Igor Canadi 已提交
3809
  uint64_t flush_column_family_if_log_file = 0;
I
Igor Canadi 已提交
3810
  uint64_t max_total_wal_size = (options_.max_total_wal_size == 0)
3811
                                    ? 4 * max_total_in_memory_state_
I
Igor Canadi 已提交
3812
                                    : options_.max_total_wal_size;
3813
  if (UNLIKELY(!single_column_family_mode_) &&
3814
      alive_log_files_.begin()->getting_flushed == false &&
I
Igor Canadi 已提交
3815 3816 3817 3818
      total_log_size_ > max_total_wal_size) {
    flush_column_family_if_log_file = alive_log_files_.begin()->number;
    alive_log_files_.begin()->getting_flushed = true;
    Log(options_.info_log,
3819 3820 3821
        "Flushing all column families with data in WAL number %" PRIu64
        ". Total log size is %" PRIu64 " while max_total_wal_size is %" PRIu64,
        flush_column_family_if_log_file, total_log_size_, max_total_wal_size);
I
Igor Canadi 已提交
3822 3823
  }

3824
  Status status;
3825 3826
  autovector<SuperVersion*> superversions_to_free;
  autovector<log::Writer*> logs_to_free;
3827 3828 3829

  if (LIKELY(single_column_family_mode_)) {
    // fast path
3830 3831 3832 3833
    status = MakeRoomForWrite(
        default_cf_handle_->cfd(), my_batch == nullptr,
        &superversions_to_free, &logs_to_free,
        expiration_time);
3834 3835 3836 3837 3838 3839 3840 3841 3842 3843
  } else {
    // refcounting cfd in iteration
    bool dead_cfd = false;
    for (auto cfd : *versions_->GetColumnFamilySet()) {
      cfd->Ref();
      bool force_flush =
          my_batch == nullptr ||
          (flush_column_family_if_log_file != 0 &&
           cfd->GetLogNumber() <= flush_column_family_if_log_file);
      // May temporarily unlock and wait.
3844 3845 3846
      status = MakeRoomForWrite(
          cfd, force_flush, &superversions_to_free, &logs_to_free,
          expiration_time);
3847 3848 3849 3850 3851 3852
      if (cfd->Unref()) {
        dead_cfd = true;
      }
      if (!status.ok()) {
        break;
      }
3853
    }
3854 3855
    if (dead_cfd) {
      versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
3856 3857
    }
  }
3858

D
dgrogan@chromium.org 已提交
3859
  uint64_t last_sequence = versions_->LastSequence();
3860
  Writer* last_writer = &w;
3861
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
3862 3863
    autovector<WriteBatch*> write_batch_group;
    BuildBatchGroup(&last_writer, &write_batch_group);
3864

3865 3866 3867
    // 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
3868
    // into memtables
3869
    {
3870
      mutex_.Unlock();
3871 3872 3873 3874 3875 3876 3877 3878 3879 3880
      WriteBatch* updates = nullptr;
      if (write_batch_group.size() == 1) {
        updates = write_batch_group[0];
      } else {
        updates = &tmp_batch_;
        for (size_t i = 0; i < write_batch_group.size(); ++i) {
          WriteBatchInternal::Append(updates, write_batch_group[i]);
        }
      }

3881 3882 3883 3884 3885
      const SequenceNumber current_sequence = last_sequence + 1;
      WriteBatchInternal::SetSequence(updates, current_sequence);
      int my_batch_count = WriteBatchInternal::Count(updates);
      last_sequence += my_batch_count;
      // Record statistics
3886 3887 3888
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3889 3890
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3891 3892
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3893
      }
L
Lei Jin 已提交
3894
      PERF_TIMER_STOP(write_pre_and_post_process_time);
3895 3896

      if (!options.disableWAL) {
L
Lei Jin 已提交
3897
        PERF_TIMER_START(write_wal_time);
3898 3899
        Slice log_entry = WriteBatchInternal::Contents(updates);
        status = log_->AddRecord(log_entry);
I
Igor Canadi 已提交
3900 3901
        total_log_size_ += log_entry.size();
        alive_log_files_.back().AddSize(log_entry.size());
3902
        log_empty_ = false;
3903 3904
        RecordTick(options_.statistics.get(), WAL_FILE_SYNCED, 1);
        RecordTick(options_.statistics.get(), WAL_FILE_BYTES, log_entry.size());
H
heyongqiang 已提交
3905
        if (status.ok() && options.sync) {
3906
          if (options_.use_fsync) {
3907
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3908
            status = log_->file()->Fsync();
3909
          } else {
3910
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3911
            status = log_->file()->Sync();
3912
          }
H
heyongqiang 已提交
3913
        }
L
Lei Jin 已提交
3914
        PERF_TIMER_STOP(write_wal_time);
3915 3916
      }
      if (status.ok()) {
L
Lei Jin 已提交
3917
        PERF_TIMER_START(write_memtable_time);
Y
Yueh-Hsuan Chiang 已提交
3918

3919
        status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
3920
            updates, column_family_memtables_.get(), false, 0, this, false);
Y
Yueh-Hsuan Chiang 已提交
3921 3922 3923 3924 3925 3926 3927 3928
        // A non-OK status here indicates iteration failure (either in-memory
        // writebatch corruption (very bad), or the client specified invalid
        // column family).  This will later on trigger bg_error_.
        //
        // Note that existing logic was not sound. Any partial failure writing
        // into the memtable would result in a state that some write ops might
        // have succeeded in memtable but Status reports error for all writes.

L
Lei Jin 已提交
3929
        PERF_TIMER_STOP(write_memtable_time);
3930

I
Igor Canadi 已提交
3931 3932
        SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                       last_sequence);
3933
      }
L
Lei Jin 已提交
3934
      PERF_TIMER_START(write_pre_and_post_process_time);
3935
      if (updates == &tmp_batch_) tmp_batch_.Clear();
3936 3937
      mutex_.Lock();
      if (status.ok()) {
3938
        versions_->SetLastSequence(last_sequence);
3939
      }
J
jorlow@chromium.org 已提交
3940 3941
    }
  }
3942 3943
  if (options_.paranoid_checks && !status.ok() &&
      !status.IsTimedOut() && bg_error_.ok()) {
I
Igor Canadi 已提交
3944 3945
    bg_error_ = status; // stop compaction & fail any further writes
  }
3946

3947 3948 3949
  // Pop out the current writer and all writers being pushed before the
  // current writer from the writer queue.
  while (!writers_.empty()) {
3950 3951 3952 3953 3954 3955
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
3956
    }
3957 3958
    if (ready == last_writer) break;
  }
3959

3960 3961 3962
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
3963
  }
I
Igor Canadi 已提交
3964
  mutex_.Unlock();
3965

3966 3967 3968 3969
  if (status.IsTimedOut()) {
    RecordTick(options_.statistics.get(), WRITE_TIMEDOUT, 1);
  }

3970 3971 3972 3973 3974 3975 3976
  for (auto& sv : superversions_to_free) {
    delete sv;
  }
  for (auto& log : logs_to_free) {
    delete log;
  }

L
Lei Jin 已提交
3977
  PERF_TIMER_STOP(write_pre_and_post_process_time);
J
jorlow@chromium.org 已提交
3978 3979 3980
  return status;
}

3981 3982 3983
// This function will be called only when the first writer succeeds.
// All writers in the to-be-built batch group will be processed.
//
3984
// REQUIRES: Writer list must be non-empty
3985
// REQUIRES: First writer must have a non-nullptr batch
3986 3987
void DBImpl::BuildBatchGroup(Writer** last_writer,
                             autovector<WriteBatch*>* write_batch_group) {
3988 3989
  assert(!writers_.empty());
  Writer* first = writers_.front();
3990
  assert(first->batch != nullptr);
3991 3992

  size_t size = WriteBatchInternal::ByteSize(first->batch);
3993
  write_batch_group->push_back(first->batch);
3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011 4012

  // 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 已提交
4013 4014 4015 4016 4017 4018
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

4019 4020 4021 4022 4023 4024
    if (w->timeout_hint_us < first->timeout_hint_us) {
      // Do not include those writes with shorter timeout.  Otherwise, we might
      // execute a write that should instead be aborted because of timeout.
      break;
    }

4025
    if (w->batch != nullptr) {
4026 4027 4028 4029 4030 4031
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

4032
      write_batch_group->push_back(w->batch);
4033
    }
4034
    w->in_batch_group = true;
4035 4036 4037 4038
    *last_writer = w;
  }
}

4039 4040 4041
// This function computes the amount of time in microseconds by which a write
// should be delayed based on the number of level-0 files according to the
// following formula:
J
Jim Paton 已提交
4042 4043 4044 4045
// if n < bottom, return 0;
// if n >= top, return 1000;
// otherwise, let r = (n - bottom) /
//                    (top - bottom)
4046 4047 4048 4049
//  and return r^2 * 1000.
// The goal of this formula is to gradually increase the rate at which writes
// are slowed. We also tried linear delay (r * 1000), but it seemed to do
// slightly worse. There is no other particular reason for choosing quadratic.
M
Mark Callaghan 已提交
4050
uint64_t DBImpl::SlowdownAmount(int n, double bottom, double top) {
4051
  uint64_t delay;
J
Jim Paton 已提交
4052
  if (n >= top) {
4053 4054
    delay = 1000;
  }
J
Jim Paton 已提交
4055
  else if (n < bottom) {
4056 4057 4058 4059
    delay = 0;
  }
  else {
    // If we are here, we know that:
J
Jim Paton 已提交
4060
    //   level0_start_slowdown <= n < level0_slowdown
4061
    // since the previous two conditions are false.
M
Mark Callaghan 已提交
4062 4063
    double how_much =
      (double) (n - bottom) /
J
Jim Paton 已提交
4064
              (top - bottom);
M
Mark Callaghan 已提交
4065
    delay = std::max(how_much * how_much * 1000, 100.0);
4066 4067 4068 4069 4070
  }
  assert(delay <= 1000);
  return delay;
}

4071
// REQUIRES: mutex_ is held
4072
// REQUIRES: this thread is currently at the front of the writer queue
4073 4074 4075
Status DBImpl::MakeRoomForWrite(
    ColumnFamilyData* cfd, bool force,
    autovector<SuperVersion*>* superversions_to_free,
4076 4077
    autovector<log::Writer*>* logs_to_free,
    uint64_t expiration_time) {
4078
  mutex_.AssertHeld();
4079
  assert(!writers_.empty());
4080
  bool allow_delay = !force;
J
Jim Paton 已提交
4081 4082
  bool allow_hard_rate_limit_delay = !force;
  bool allow_soft_rate_limit_delay = !force;
4083
  uint64_t rate_limit_delay_millis = 0;
4084
  Status s;
4085
  double score;
4086 4087 4088 4089
  // Once we schedule background work, we shouldn't schedule it again, since it
  // might generate a tight feedback loop, constantly scheduling more background
  // work, even if additional background work is not needed
  bool schedule_background_work = true;
4090
  bool has_timeout = (expiration_time > 0);
4091

4092 4093 4094 4095 4096
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
4097 4098 4099
    } else if (has_timeout && env_->NowMicros() > expiration_time) {
      s = Status::TimedOut();
      break;
4100
    } else if (allow_delay && cfd->NeedSlowdownForNumLevel0Files()) {
4101 4102 4103
      // 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
4104
      // individual write by 0-1ms to reduce latency variance.  Also,
4105 4106
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
4107
      uint64_t slowdown =
4108 4109 4110
          SlowdownAmount(cfd->current()->NumLevelFiles(0),
                         cfd->options()->level0_slowdown_writes_trigger,
                         cfd->options()->level0_stop_writes_trigger);
4111
      mutex_.Unlock();
4112
      uint64_t delayed;
J
Jim Paton 已提交
4113
      {
4114
        StopWatch sw(env_, options_.statistics.get(), STALL_L0_SLOWDOWN_COUNT);
4115
        env_->SleepForMicroseconds(slowdown);
4116
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
4117
      }
4118
      RecordTick(options_.statistics.get(), STALL_L0_SLOWDOWN_MICROS, delayed);
4119 4120
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_SLOWDOWN,
                                              delayed);
4121 4122
      allow_delay = false;  // Do not delay a single write more than once
      mutex_.Lock();
4123
      delayed_writes_++;
4124
    } else if (!force && !cfd->mem()->ShouldFlush()) {
4125
      // There is room in current memtable
4126 4127 4128
      if (allow_delay) {
        DelayLoggingAndReset();
      }
4129
      break;
4130
    } else if (cfd->NeedWaitForNumMemtables()) {
4131
      // We have filled up the current memtable, but the previous
4132
      // ones are still being flushed, so we wait.
4133
      DelayLoggingAndReset();
I
Igor Canadi 已提交
4134 4135
      Log(options_.info_log, "[%s] wait for memtable flush...\n",
          cfd->GetName().c_str());
4136 4137 4138 4139
      if (schedule_background_work) {
        MaybeScheduleFlushOrCompaction();
        schedule_background_work = false;
      }
4140
      uint64_t stall;
J
Jim Paton 已提交
4141
      {
4142
        StopWatch sw(env_, options_.statistics.get(),
4143
                     STALL_MEMTABLE_COMPACTION_COUNT);
4144 4145 4146 4147 4148
        if (!has_timeout) {
          bg_cv_.Wait();
        } else {
          bg_cv_.TimedWait(expiration_time);
        }
4149
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
4150
      }
4151 4152
      RecordTick(options_.statistics.get(),
                 STALL_MEMTABLE_COMPACTION_MICROS, stall);
4153 4154
      cfd->internal_stats()->RecordWriteStall(
          InternalStats::MEMTABLE_COMPACTION, stall);
4155
    } else if (cfd->NeedWaitForNumLevel0Files()) {
4156
      DelayLoggingAndReset();
I
Igor Canadi 已提交
4157 4158
      Log(options_.info_log, "[%s] wait for fewer level0 files...\n",
          cfd->GetName().c_str());
4159
      uint64_t stall;
J
Jim Paton 已提交
4160
      {
4161 4162
        StopWatch sw(env_, options_.statistics.get(),
                     STALL_L0_NUM_FILES_COUNT);
4163 4164 4165 4166 4167
        if (!has_timeout) {
          bg_cv_.Wait();
        } else {
          bg_cv_.TimedWait(expiration_time);
        }
4168
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
4169
      }
4170 4171
      RecordTick(options_.statistics.get(),
                 STALL_L0_NUM_FILES_MICROS, stall);
4172 4173
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_NUM_FILES,
                                              stall);
4174
    } else if (allow_hard_rate_limit_delay && cfd->ExceedsHardRateLimit()) {
4175
      // Delay a write when the compaction score for any level is too large.
4176
      int max_level = cfd->current()->MaxCompactionScoreLevel();
4177
      score = cfd->current()->MaxCompactionScore();
4178
      mutex_.Unlock();
4179
      uint64_t delayed;
J
Jim Paton 已提交
4180
      {
4181 4182
        StopWatch sw(env_, options_.statistics.get(),
                     HARD_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
4183
        env_->SleepForMicroseconds(1000);
4184
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
4185
      }
4186
      cfd->internal_stats()->RecordLevelNSlowdown(max_level, delayed);
4187
      // Make sure the following value doesn't round to zero.
4188 4189
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
4190 4191
      RecordTick(options_.statistics.get(),
                 RATE_LIMIT_DELAY_MILLIS, rate_limit);
4192
      if (cfd->options()->rate_limit_delay_max_milliseconds > 0 &&
J
Jim Paton 已提交
4193
          rate_limit_delay_millis >=
4194
              (unsigned)cfd->options()->rate_limit_delay_max_milliseconds) {
J
Jim Paton 已提交
4195
        allow_hard_rate_limit_delay = false;
4196
      }
4197
      mutex_.Lock();
4198 4199
    } else if (allow_soft_rate_limit_delay && cfd->ExceedsSoftRateLimit()) {
      score = cfd->current()->MaxCompactionScore();
J
Jim Paton 已提交
4200 4201
      // Delay a write when the compaction score for any level is too large.
      // TODO: add statistics
4202 4203
      uint64_t slowdown = SlowdownAmount(score, cfd->options()->soft_rate_limit,
                                         cfd->options()->hard_rate_limit);
J
Jim Paton 已提交
4204
      mutex_.Unlock();
J
Jim Paton 已提交
4205
      {
4206 4207
        StopWatch sw(env_, options_.statistics.get(),
                     SOFT_RATE_LIMIT_DELAY_COUNT);
4208 4209 4210
        env_->SleepForMicroseconds(slowdown);
        slowdown = sw.ElapsedMicros();
        rate_limit_delay_millis += slowdown;
J
Jim Paton 已提交
4211
      }
J
Jim Paton 已提交
4212 4213
      allow_soft_rate_limit_delay = false;
      mutex_.Lock();
4214
    } else {
4215
      unique_ptr<WritableFile> lfile;
S
sdong 已提交
4216
      log::Writer* new_log = nullptr;
K
kailiu 已提交
4217
      MemTable* new_mem = nullptr;
4218

4219
      // Attempt to switch to a new memtable and trigger flush of old.
4220
      // Do this without holding the dbmutex lock.
4221
      assert(versions_->PrevLogNumber() == 0);
I
Igor Canadi 已提交
4222 4223 4224
      bool creating_new_log = !log_empty_;
      uint64_t new_log_number =
          creating_new_log ? versions_->NewFileNumber() : logfile_number_;
I
Igor Canadi 已提交
4225
      SuperVersion* new_superversion = nullptr;
4226 4227 4228
      mutex_.Unlock();
      {
        DelayLoggingAndReset();
I
Igor Canadi 已提交
4229 4230 4231 4232 4233 4234 4235 4236 4237 4238 4239 4240 4241
        if (creating_new_log) {
          s = env_->NewWritableFile(
              LogFileName(options_.wal_dir, new_log_number), &lfile,
              env_->OptimizeForLogWrite(storage_options_));
          if (s.ok()) {
            // Our final size should be less than write_buffer_size
            // (compression, etc) but err on the side of caution.
            lfile->SetPreallocationBlockSize(1.1 *
                                             cfd->options()->write_buffer_size);
            new_log = new log::Writer(std::move(lfile));
          }
        }

4242
        if (s.ok()) {
4243
          new_mem = new MemTable(cfd->internal_comparator(), *cfd->options());
4244
          new_superversion = new SuperVersion();
4245 4246 4247
        }
      }
      mutex_.Lock();
4248
      if (!s.ok()) {
I
Igor Canadi 已提交
4249 4250
        // how do we fail if we're not creating new log?
        assert(creating_new_log);
H
heyongqiang 已提交
4251
        // Avoid chewing through file number space in a tight loop.
4252
        versions_->ReuseLogFileNumber(new_log_number);
4253
        assert(!new_mem);
4254
        assert(!new_log);
4255 4256
        break;
      }
I
Igor Canadi 已提交
4257 4258 4259
      if (creating_new_log) {
        logfile_number_ = new_log_number;
        assert(new_log != nullptr);
4260
        logs_to_free->push_back(log_.release());
I
Igor Canadi 已提交
4261 4262
        log_.reset(new_log);
        log_empty_ = true;
I
Igor Canadi 已提交
4263
        alive_log_files_.push_back(LogFileNumberSize(logfile_number_));
I
Igor Canadi 已提交
4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274
        for (auto cfd : *versions_->GetColumnFamilySet()) {
          // all this is just optimization to delete logs that
          // are no longer needed -- if CF is empty, that means it
          // doesn't need that particular log to stay alive, so we just
          // advance the log number. no need to persist this in the manifest
          if (cfd->mem()->GetFirstSequenceNumber() == 0 &&
              cfd->imm()->size() == 0) {
            cfd->SetLogNumber(logfile_number_);
          }
        }
      }
4275 4276
      cfd->mem()->SetNextLogNumber(logfile_number_);
      cfd->imm()->Add(cfd->mem());
4277
      if (force) {
4278
        cfd->imm()->FlushRequested();
4279
      }
4280 4281
      new_mem->Ref();
      cfd->SetMemtable(new_mem);
4282 4283 4284
      Log(options_.info_log,
          "[%s] New memtable created with log file: #%" PRIu64 "\n",
          cfd->GetName().c_str(), logfile_number_);
4285
      force = false;  // Do not force another compaction if have room
4286
      MaybeScheduleFlushOrCompaction();
4287 4288
      superversions_to_free->push_back(
          cfd->InstallSuperVersion(new_superversion, &mutex_));
4289 4290 4291 4292 4293
    }
  }
  return s;
}

I
Igor Canadi 已提交
4294
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
4295 4296 4297 4298 4299
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

4300 4301
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
4302
  auto version = cfd->current();
4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

  // Decrement the ref count
  mutex_.Lock();
  version->Unref();
  mutex_.Unlock();

  return s;
}
I
Igor Canadi 已提交
4315
#endif  // ROCKSDB_LITE
4316

I
Igor Canadi 已提交
4317 4318 4319 4320
const std::string& DBImpl::GetName() const {
  return dbname_;
}

4321 4322 4323 4324
Env* DBImpl::GetEnv() const {
  return env_;
}

4325 4326
const Options& DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
I
Igor Canadi 已提交
4327
  return *cfh->cfd()->options();
I
Igor Canadi 已提交
4328 4329
}

4330
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
4331
                         const Slice& property, std::string* value) {
4332
  value->clear();
4333 4334
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
4335
  DBPropertyType property_type = GetPropertyType(property);
J
jorlow@chromium.org 已提交
4336
  MutexLock l(&mutex_);
4337 4338
  return cfd->internal_stats()->GetProperty(property_type, property, value,
                                            cfd);
J
jorlow@chromium.org 已提交
4339 4340
}

4341
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
4342
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
4343 4344
  // TODO(opt): better implementation
  Version* v;
4345 4346
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
J
jorlow@chromium.org 已提交
4347 4348
  {
    MutexLock l(&mutex_);
4349
    v = cfd->current();
4350
    v->Ref();
J
jorlow@chromium.org 已提交
4351 4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367
  }

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

4368 4369 4370 4371 4372 4373 4374
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

I
Igor Canadi 已提交
4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389 4390 4391 4392 4393 4394 4395 4396 4397 4398 4399 4400 4401 4402
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {

  RecordTick(options_.statistics.get(), GET_UPDATES_SINCE_CALLS);
  if (seq > versions_->LastSequence()) {
    return Status::NotFound("Requested sequence not yet written in the db");
  }
  //  Get all sorted Wal Files.
  //  Do binary search and open files and find the seq number.

  std::unique_ptr<VectorLogPtr> wal_files(new VectorLogPtr);
  Status s = GetSortedWalFiles(*wal_files);
  if (!s.ok()) {
    return s;
  }

  s = RetainProbableWalFiles(*wal_files, seq);
  if (!s.ok()) {
    return s;
  }
  iter->reset(new TransactionLogIteratorImpl(options_.wal_dir, &options_,
                                             read_options, storage_options_,
                                             seq, std::move(wal_files), this));
  return (*iter)->status();
}

4403 4404 4405
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
4406 4407 4408 4409
  WalFileType log_type;
  if (!ParseFileName(name, &number, &type, &log_type) ||
      (type != kTableFile && type != kLogFile)) {
    Log(options_.info_log, "DeleteFile %s failed.\n", name.c_str());
4410 4411 4412
    return Status::InvalidArgument("Invalid file name");
  }

4413 4414 4415 4416
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
4417 4418
      Log(options_.info_log, "DeleteFile %s failed - not archived log.\n",
          name.c_str());
4419 4420 4421 4422
      return Status::NotSupported("Delete only supported for archived logs");
    }
    status = env_->DeleteFile(options_.wal_dir + "/" + name.c_str());
    if (!status.ok()) {
4423 4424
      Log(options_.info_log, "DeleteFile %s failed -- %s.\n",
          name.c_str(), status.ToString().c_str());
4425 4426 4427 4428
    }
    return status;
  }

4429
  int level;
I
Igor Canadi 已提交
4430
  FileMetaData* metadata;
4431
  ColumnFamilyData* cfd;
4432
  VersionEdit edit;
K
Kai Liu 已提交
4433
  DeletionState deletion_state(true);
D
Dhruba Borthakur 已提交
4434 4435
  {
    MutexLock l(&mutex_);
4436
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
4437
    if (!status.ok()) {
4438 4439
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
4440 4441
      return Status::InvalidArgument("File not found");
    }
I
Igor Canadi 已提交
4442
    assert((level > 0) && (level < cfd->NumberLevels()));
4443

D
Dhruba Borthakur 已提交
4444
    // If the file is being compacted no need to delete.
4445
    if (metadata->being_compacted) {
4446
      Log(options_.info_log,
4447
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
4448
      return Status::OK();
4449 4450
    }

D
Dhruba Borthakur 已提交
4451 4452 4453
    // Only the files in the last level can be deleted externally.
    // This is to make sure that any deletion tombstones are not
    // lost. Check that the level passed is the last level.
I
Igor Canadi 已提交
4454
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
4455
      if (cfd->current()->NumLevelFiles(i) != 0) {
D
Dhruba Borthakur 已提交
4456
        Log(options_.info_log,
4457
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
4458 4459 4460 4461
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
4462
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
I
Igor Canadi 已提交
4463
    if (status.ok()) {
4464
      InstallSuperVersion(cfd, deletion_state);
I
Igor Canadi 已提交
4465
    }
I
Igor Canadi 已提交
4466
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
4467
  } // lock released here
I
Igor Canadi 已提交
4468
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
4469
  // remove files outside the db-lock
4470 4471 4472
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
4473 4474 4475 4476 4477 4478
  {
    MutexLock l(&mutex_);
    // schedule flush if file deletion means we freed the space for flushes to
    // continue
    MaybeScheduleFlushOrCompaction();
  }
4479 4480 4481
  return status;
}

4482
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
4483
  MutexLock l(&mutex_);
4484
  versions_->GetLiveFilesMetaData(metadata);
4485
}
I
Igor Canadi 已提交
4486
#endif  // ROCKSDB_LITE
4487

I
Igor Canadi 已提交
4488 4489 4490 4491 4492 4493 4494
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

  std::string corruption_messages;
  for (const auto& md : metadata) {
4495 4496
    std::string file_path = md.db_path + "/" + md.name;

I
Igor Canadi 已提交
4497 4498 4499 4500 4501 4502
    uint64_t fsize = 0;
    Status s = env_->GetFileSize(file_path, &fsize);
    if (!s.ok()) {
      corruption_messages +=
          "Can't access " + md.name + ": " + s.ToString() + "\n";
    } else if (fsize != md.size) {
4503
      corruption_messages += "Sst file size mismatch: " + file_path +
I
Igor Canadi 已提交
4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515
                             ". Size recorded in manifest " +
                             std::to_string(md.size) + ", actual size " +
                             std::to_string(fsize) + "\n";
    }
  }
  if (corruption_messages.size() == 0) {
    return Status::OK();
  } else {
    return Status::Corruption(corruption_messages);
  }
}

4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529 4530 4531 4532 4533 4534 4535 4536 4537 4538 4539 4540 4541 4542
Status DBImpl::GetDbIdentity(std::string& identity) {
  std::string idfilename = IdentityFileName(dbname_);
  unique_ptr<SequentialFile> idfile;
  const EnvOptions soptions;
  Status s = env_->NewSequentialFile(idfilename, &idfile, soptions);
  if (!s.ok()) {
    return s;
  }
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
  char buffer[file_size];
  Slice id;
  s = idfile->Read(file_size, &id, buffer);
  if (!s.ok()) {
    return s;
  }
  identity.assign(id.ToString());
  // If last character is '\n' remove it from identity
  if (identity.size() > 0 && identity.back() == '\n') {
    identity.pop_back();
  }
  return s;
}

J
jorlow@chromium.org 已提交
4543 4544
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
4545
Status DB::Put(const WriteOptions& opt, ColumnFamilyHandle* column_family,
4546
               const Slice& key, const Slice& value) {
4547 4548 4549 4550
  // Pre-allocate size of write batch conservatively.
  // 8 bytes are taken by header, 4 bytes for count, 1 byte for type,
  // and we allocate 11 extra bytes for key length, as well as value length.
  WriteBatch batch(key.size() + value.size() + 24);
4551
  batch.Put(column_family, key, value);
J
jorlow@chromium.org 已提交
4552 4553 4554
  return Write(opt, &batch);
}

4555 4556
Status DB::Delete(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                  const Slice& key) {
J
jorlow@chromium.org 已提交
4557
  WriteBatch batch;
4558
  batch.Delete(column_family, key);
J
jorlow@chromium.org 已提交
4559 4560 4561
  return Write(opt, &batch);
}

4562 4563
Status DB::Merge(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                 const Slice& key, const Slice& value) {
4564
  WriteBatch batch;
4565
  batch.Merge(column_family, key, value);
4566 4567 4568
  return Write(opt, &batch);
}

4569 4570
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
4571
                              const std::string& column_family_name,
4572
                              ColumnFamilyHandle** handle) {
4573
  return Status::NotSupported("");
4574
}
4575
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
4576
  return Status::NotSupported("");
4577 4578
}

J
jorlow@chromium.org 已提交
4579 4580
DB::~DB() { }

J
Jim Paton 已提交
4581
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
4582 4583 4584 4585
  DBOptions db_options(options);
  ColumnFamilyOptions cf_options(options);
  std::vector<ColumnFamilyDescriptor> column_families;
  column_families.push_back(
4586
      ColumnFamilyDescriptor(kDefaultColumnFamilyName, cf_options));
4587 4588 4589 4590 4591 4592 4593 4594 4595
  std::vector<ColumnFamilyHandle*> handles;
  Status s = DB::Open(db_options, dbname, column_families, &handles, dbptr);
  if (s.ok()) {
    assert(handles.size() == 1);
    // i can delete the handle since DBImpl is always holding a reference to
    // default column family
    delete handles[0];
  }
  return s;
4596 4597
}

4598 4599
Status DB::Open(const DBOptions& db_options, const std::string& dbname,
                const std::vector<ColumnFamilyDescriptor>& column_families,
4600
                std::vector<ColumnFamilyHandle*>* handles, DB** dbptr) {
4601 4602 4603 4604 4605
  if (db_options.db_paths.size() > 1) {
    return Status::NotSupported(
        "More than one DB paths are not supported yet. ");
  }

4606
  *dbptr = nullptr;
4607
  handles->clear();
J
jorlow@chromium.org 已提交
4608

I
Igor Canadi 已提交
4609 4610 4611 4612 4613 4614 4615 4616
  size_t max_write_buffer_size = 0;
  for (auto cf : column_families) {
    max_write_buffer_size =
        std::max(max_write_buffer_size, cf.options.write_buffer_size);
    if (cf.options.block_cache != nullptr && cf.options.no_block_cache) {
      return Status::InvalidArgument(
          "no_block_cache is true while block_cache is not nullptr");
    }
4617
  }
4618

I
Igor Canadi 已提交
4619
  DBImpl* impl = new DBImpl(db_options, dbname);
4620
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
4621 4622 4623 4624 4625 4626 4627 4628 4629
  if (s.ok()) {
    for (auto path : impl->options_.db_paths) {
      s = impl->env_->CreateDirIfMissing(path);
      if (!s.ok()) {
        break;
      }
    }
  }

4630 4631 4632 4633 4634 4635
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
4636 4637 4638 4639
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
4640
  impl->mutex_.Lock();
4641 4642
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
4643
  if (s.ok()) {
4644
    uint64_t new_log_number = impl->versions_->NewFileNumber();
4645
    unique_ptr<WritableFile> lfile;
4646
    EnvOptions soptions(db_options);
4647
    s = impl->options_.env->NewWritableFile(
I
Igor Canadi 已提交
4648
        LogFileName(impl->options_.wal_dir, new_log_number), &lfile,
I
Igor Canadi 已提交
4649
        impl->options_.env->OptimizeForLogWrite(soptions));
J
jorlow@chromium.org 已提交
4650
    if (s.ok()) {
4651
      lfile->SetPreallocationBlockSize(1.1 * max_write_buffer_size);
4652
      impl->logfile_number_ = new_log_number;
4653
      impl->log_.reset(new log::Writer(std::move(lfile)));
I
Igor Canadi 已提交
4654

4655 4656
      // set column family handles
      for (auto cf : column_families) {
I
Igor Canadi 已提交
4657 4658
        auto cfd =
            impl->versions_->GetColumnFamilySet()->GetColumnFamily(cf.name);
I
Igor Canadi 已提交
4659 4660 4661 4662 4663 4664 4665 4666 4667 4668 4669 4670 4671 4672 4673 4674 4675 4676 4677
        if (cfd != nullptr) {
          handles->push_back(
              new ColumnFamilyHandleImpl(cfd, impl, &impl->mutex_));
        } else {
          if (db_options.create_missing_column_families) {
            // missing column family, create it
            ColumnFamilyHandle* handle;
            impl->mutex_.Unlock();
            s = impl->CreateColumnFamily(cf.options, cf.name, &handle);
            impl->mutex_.Lock();
            if (s.ok()) {
              handles->push_back(handle);
            } else {
              break;
            }
          } else {
            s = Status::InvalidArgument("Column family not found: ", cf.name);
            break;
          }
I
Igor Canadi 已提交
4678
        }
4679
      }
I
Igor Canadi 已提交
4680 4681
    }
    if (s.ok()) {
4682
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
4683
        delete cfd->InstallSuperVersion(new SuperVersion(), &impl->mutex_);
4684
      }
I
Igor Canadi 已提交
4685 4686
      impl->alive_log_files_.push_back(
          DBImpl::LogFileNumberSize(impl->logfile_number_));
J
jorlow@chromium.org 已提交
4687
      impl->DeleteObsoleteFiles();
4688
      impl->MaybeScheduleFlushOrCompaction();
4689
      impl->MaybeScheduleLogDBDeployStats();
4690
      s = impl->db_directory_->Fsync();
J
jorlow@chromium.org 已提交
4691 4692
    }
  }
4693

I
Igor Canadi 已提交
4694 4695
  if (s.ok()) {
    for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
4696 4697
      if (cfd->options()->compaction_style == kCompactionStyleUniversal ||
          cfd->options()->compaction_style == kCompactionStyleFIFO) {
I
Igor Canadi 已提交
4698 4699 4700 4701
        Version* current = cfd->current();
        for (int i = 1; i < current->NumberLevels(); ++i) {
          int num_files = current->NumLevelFiles(i);
          if (num_files > 0) {
I
Igor Canadi 已提交
4702 4703 4704
            s = Status::InvalidArgument(
                "Not all files are at level 0. Cannot "
                "open with universal or FIFO compaction style.");
I
Igor Canadi 已提交
4705 4706 4707 4708
            break;
          }
        }
      }
4709 4710 4711 4712 4713 4714
      if (cfd->options()->merge_operator != nullptr &&
          !cfd->mem()->IsMergeOperatorSupported()) {
        s = Status::InvalidArgument(
            "The memtable of column family %s does not support merge operator "
            "its options.merge_operator is non-null", cfd->GetName().c_str());
      }
I
Igor Canadi 已提交
4715
      if (!s.ok()) {
4716 4717 4718 4719 4720
        break;
      }
    }
  }

4721 4722
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4723
  if (s.ok()) {
4724
    impl->opened_successfully_ = true;
J
jorlow@chromium.org 已提交
4725 4726
    *dbptr = impl;
  } else {
4727 4728 4729
    for (auto h : *handles) {
      delete h;
    }
4730
    handles->clear();
J
jorlow@chromium.org 已提交
4731 4732 4733 4734 4735
    delete impl;
  }
  return s;
}

4736 4737 4738
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4739
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4740 4741
}

4742 4743 4744
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4745
Status DestroyDB(const std::string& dbname, const Options& options) {
4746 4747 4748 4749 4750
  const InternalKeyComparator comparator(options.comparator);
  const InternalFilterPolicy filter_policy(options.filter_policy);
  const Options& soptions(SanitizeOptions(
    dbname, &comparator, &filter_policy, options));
  Env* env = soptions.env;
J
jorlow@chromium.org 已提交
4751
  std::vector<std::string> filenames;
4752 4753
  std::vector<std::string> archiveFiles;

4754
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
4755 4756
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
4757 4758 4759 4760 4761 4762 4763

  if (dbname != soptions.wal_dir) {
    std::vector<std::string> logfilenames;
    env->GetChildren(soptions.wal_dir, &logfilenames);
    filenames.insert(filenames.end(), logfilenames.begin(), logfilenames.end());
    archivedir = ArchivalDirectory(soptions.wal_dir);
  }
4764

J
jorlow@chromium.org 已提交
4765 4766 4767 4768 4769
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
4770 4771
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4772 4773 4774
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
4775
    for (size_t i = 0; i < filenames.size(); i++) {
4776
      if (ParseFileName(filenames[i], &number, &type) &&
4777
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4778 4779 4780
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
4781 4782
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
4783 4784 4785
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
4786 4787 4788 4789 4790
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4791

4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806
    for (auto db_path : options.db_paths) {
      env->GetChildren(db_path, &filenames);
      uint64_t number;
      FileType type;
      for (size_t i = 0; i < filenames.size(); i++) {
        if (ParseFileName(filenames[i], &number, &type) &&
            type == kTableFile) {  // Lock file will be deleted at end
          Status del = env->DeleteFile(db_path + "/" + filenames[i]);
          if (result.ok() && !del.ok()) {
            result = del;
          }
        }
      }
    }

4807
    env->GetChildren(archivedir, &archiveFiles);
4808 4809
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4810 4811
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4812
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4813 4814 4815 4816 4817
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4818
    // ignore case where no archival directory is present.
4819
    env->DeleteDir(archivedir);
4820

J
jorlow@chromium.org 已提交
4821
    env->UnlockFile(lock);  // Ignore error since state is already gone
4822
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4823
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4824
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4825 4826 4827 4828
  }
  return result;
}

4829 4830
//
// A global method that can dump out the build version
K
kailiu 已提交
4831
void DumpLeveldbBuildVersion(Logger * log) {
I
Igor Canadi 已提交
4832 4833
#if !defined(IOS_CROSS_COMPILE)
  // if we compile with Xcode, we don't run build_detect_vesion, so we don't generate util/build_version.cc
4834
  Log(log, "Git sha %s", rocksdb_build_git_sha);
4835
  Log(log, "Compile time %s %s",
4836
      rocksdb_build_compile_time, rocksdb_build_compile_date);
I
Igor Canadi 已提交
4837
#endif
4838 4839
}

4840
}  // namespace rocksdb