db_impl.cc 157.6 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"
39
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
40 41
#include "db/version_set.h"
#include "db/write_batch_internal.h"
42
#include "port/port.h"
I
Igor Canadi 已提交
43
#include "rocksdb/cache.h"
44
#include "port/likely.h"
45 46 47 48 49 50
#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 已提交
51
#include "rocksdb/table.h"
J
jorlow@chromium.org 已提交
52
#include "table/block.h"
53
#include "table/block_based_table_factory.h"
J
jorlow@chromium.org 已提交
54
#include "table/merger.h"
K
kailiu 已提交
55
#include "table/table_builder.h"
J
jorlow@chromium.org 已提交
56
#include "table/two_level_iterator.h"
57
#include "util/auto_roll_logger.h"
K
kailiu 已提交
58
#include "util/autovector.h"
59
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
60
#include "util/coding.h"
I
Igor Canadi 已提交
61
#include "util/hash_skiplist_rep.h"
62
#include "util/hash_linklist_rep.h"
J
jorlow@chromium.org 已提交
63
#include "util/logging.h"
H
Haobo Xu 已提交
64
#include "util/log_buffer.h"
J
jorlow@chromium.org 已提交
65
#include "util/mutexlock.h"
66
#include "util/perf_context_imp.h"
67
#include "util/stop_watch.h"
68
#include "util/sync_point.h"
J
jorlow@chromium.org 已提交
69

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

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

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

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

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

J
jorlow@chromium.org 已提交
88 89 90
struct DBImpl::CompactionState {
  Compaction* const compaction;

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

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

  // State kept for output being generated
108 109
  unique_ptr<WritableFile> outfile;
  unique_ptr<TableBuilder> builder;
J
jorlow@chromium.org 已提交
110 111 112 113 114 115 116 117 118

  uint64_t total_bytes;

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

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

120 121 122 123 124 125 126 127
  // 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;
  }

128
  // Create a client visible context of this compaction
D
Danny Guo 已提交
129 130
  CompactionFilterContext GetFilterContext() {
    CompactionFilterContext context;
131
    context.is_full_compaction = compaction->IsFullCompaction();
132
    context.is_manual_compaction = compaction->IsManualCompaction();
133 134
    return context;
  }
D
Danny Guo 已提交
135 136 137 138 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

  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 已提交
252 253
};

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

J
jorlow@chromium.org 已提交
263 264
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
265
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
266
                        const Options& src) {
267 268 269 270 271 272 273
  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;
274 275
  // result.max_open_files means an "infinite" open files.
  if (result.max_open_files != -1) {
276
    ClipToRange(&result.max_open_files, 20, 1000000);
277
  }
278

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

  if (result.wal_dir.empty()) {
    // Use dbname as default
    result.wal_dir = dbname;
  }
292
  if (result.wal_dir.back() == '/') {
I
Igor Canadi 已提交
293
    result.wal_dir = result.wal_dir.substr(0, result.wal_dir.size() - 1);
294
  }
295

J
jorlow@chromium.org 已提交
296 297 298
  return result;
}

S
Siying Dong 已提交
299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319
CompressionType GetCompressionType(const Options& options, int level,
                                   const bool enable_compression) {
  if (!enable_compression) {
    // disable compression
    return kNoCompression;
  }
  // If the use has specified a different compression level for each level,
  // then pick the compresison for that level.
  if (!options.compression_per_level.empty()) {
    const int n = options.compression_per_level.size() - 1;
    // It is possible for level_ to be -1; in that case, we use level
    // 0's compression.  This occurs mostly in backwards compatibility
    // situations when the builder doesn't know what level the file
    // belongs to.  Likewise, if level_ is beyond the end of the
    // specified compression levels, use the last value.
    return options.compression_per_level[std::max(0, std::min(level, n))];
  } else {
    return options.compression;
  }
}

320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341
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;

  if  (options.compaction_style == kCompactionStyleUniversal) {
    can_compress =
        (options.compaction_options_universal.compression_size_percent < 0);
  } else {
    // For leveled compress when min_level_to_compress == 0.
    can_compress = (GetCompressionType(options, 0, true) != kNoCompression);
  }

  if (can_compress) {
    return options.compression;
  } else {
    return kNoCompression;
  }
}

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

J
jorlow@chromium.org 已提交
374
  // Reserve ten files or so for other uses and give the rest to TableCache.
375 376
  // Give a large number for setting of "infinite" open files.
  const int table_cache_size =
377 378 379 380 381 382
      (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 已提交
383 384
  versions_.reset(
      new VersionSet(dbname_, &options_, storage_options_, table_cache_.get()));
385 386
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
387

K
kailiu 已提交
388
  DumpLeveldbBuildVersion(options_.info_log.get());
389
  options_.Dump(options_.info_log.get());
390

391
  char name[100];
K
kailiu 已提交
392 393
  Status s = env_->GetHostName(name, 100L);
  if (s.ok()) {
394 395 396 397 398 399
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
400

I
Igor Canadi 已提交
401
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
402 403 404
}

DBImpl::~DBImpl() {
405 406 407 408 409 410 411 412
  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();
413
        cfd->Unref();
414 415
      }
    }
416
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
417
  }
I
Igor Canadi 已提交
418 419

  // Wait for background work to finish
420
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
421 422 423
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
424
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
425
  }
426

I
Igor Canadi 已提交
427 428 429 430 431
  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();
432 433
  }

434 435 436 437 438 439 440 441 442 443 444 445 446 447 448
  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;
449 450 451
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
452 453 454
    }
  }

455
  // versions need to be destroyed before table_cache since it can hold
456 457
  // references to table_cache.
  versions_.reset();
458
  mutex_.Unlock();
I
Igor Canadi 已提交
459 460 461
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
462

I
Igor Canadi 已提交
463
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
464 465 466
}

Status DBImpl::NewDB() {
467
  VersionEdit new_db;
468
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
469 470 471 472
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

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

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

504
const Status DBImpl::CreateArchivalDirectory() {
505
  if (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0) {
506
    std::string archivalPath = ArchivalDirectory(options_.wal_dir);
507 508 509 510 511
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

512
void DBImpl::PrintStatistics() {
513
  auto dbstats = options_.statistics.get();
514 515
  if (dbstats) {
    Log(options_.info_log,
516 517
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
518 519 520
  }
}

521
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
522 523 524 525 526 527 528 529 530 531 532 533 534
  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;
535
    GetProperty("rocksdb.stats", &stats);
H
Haobo Xu 已提交
536
    Log(options_.info_log, "%s", stats.c_str());
537
    PrintStatistics();
538 539 540
  }
}

541
// Returns the list of live files in 'sst_live' and the list
K
kailiu 已提交
542
// of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
543 544 545 546 547 548 549
// 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 已提交
550 551
  mutex_.AssertHeld();

552
  // if deletion is disabled, do nothing
553
  if (disable_delete_obsolete_files_ > 0) {
554 555 556
    return;
  }

557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572
  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 已提交
573 574 575
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
576 577
  // store the current filenum, lognum, etc
  deletion_state.manifest_file_number = versions_->ManifestFileNumber();
578 579
  deletion_state.pending_manifest_file_number =
      versions_->PendingManifestFileNumber();
580
  deletion_state.log_number = versions_->MinLogNumber();
I
Igor Canadi 已提交
581 582
  deletion_state.prev_log_number = versions_->PrevLogNumber();

583 584 585 586 587 588 589 590
  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
I
Igor Canadi 已提交
591 592 593 594
  deletion_state.sst_live.assign(pending_outputs_.begin(),
                                 pending_outputs_.end());
  versions_->AddLiveFiles(&deletion_state.sst_live);

595
  if (doing_the_full_scan) {
K
kailiu 已提交
596 597 598 599 600
    // set of all files in the directory. We'll exclude files that are still
    // alive in the subsequent processings.
    env_->GetChildren(
        dbname_, &deletion_state.candidate_files
    ); // Ignore errors
601 602 603 604 605

    //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
K
kailiu 已提交
606 607
      deletion_state.candidate_files.insert(
        deletion_state.candidate_files.end(),
608 609 610
        log_files.begin(),
        log_files.end()
      );
611
    }
612
  }
613 614
}

D
Dhruba Borthakur 已提交
615
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
616
// belong to live files are posibly removed. Also, removes all the
617
// files in sst_delete_files and log_delete_files.
618
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
619
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
620 621
  // we'd better have sth to delete
  assert(state.HaveSomethingToDelete());
622

I
Igor Canadi 已提交
623 624 625 626
  // 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 已提交
627 628
    return;
  }
629

630 631
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
632 633
  std::unordered_set<uint64_t> sst_live(state.sst_live.begin(),
                                        state.sst_live.end());
I
Igor Canadi 已提交
634

K
kailiu 已提交
635 636 637 638 639 640 641
  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 = "";
642
  for (auto file : state.sst_delete_files) {
K
kailiu 已提交
643 644 645
    candidate_files.push_back(
        TableFileName(kDumbDbName, file->number).substr(1)
    );
646
    delete file;
I
Igor Canadi 已提交
647 648
  }

K
kailiu 已提交
649 650
  for (auto file_num : state.log_delete_files) {
    if (file_num > 0) {
651
      candidate_files.push_back(LogFileName(kDumbDbName, file_num).substr(1));
I
Igor Canadi 已提交
652 653
    }
  }
654

K
kailiu 已提交
655
  // dedup state.candidate_files so we don't try to delete the same
I
Igor Canadi 已提交
656
  // file twice
K
kailiu 已提交
657
  sort(candidate_files.begin(), candidate_files.end());
658 659
  candidate_files.erase(unique(candidate_files.begin(), candidate_files.end()),
                        candidate_files.end());
J
jorlow@chromium.org 已提交
660

661 662
  std::vector<std::string> old_info_log_files;

K
kailiu 已提交
663 664 665 666 667 668 669
  for (const auto& to_delete : candidate_files) {
    uint64_t number;
    FileType type;
    // Ignore file if we cannot recognize it.
    if (!ParseFileName(to_delete, &number, &type)) {
      continue;
    }
J
jorlow@chromium.org 已提交
670

K
kailiu 已提交
671 672 673 674 675 676 677 678
    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'
679
        // (can happen during manifest roll)
K
kailiu 已提交
680 681 682 683 684 685 686
        keep = (number >= state.manifest_file_number);
        break;
      case kTableFile:
        keep = (sst_live.find(number) != sst_live.end());
        break;
      case kTempFile:
        // Any temp files that are currently being written to must
687 688 689 690 691 692
        // 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
        keep = (sst_live.find(number) != sst_live.end()) ||
               (number == state.pending_manifest_file_number);
K
kailiu 已提交
693 694 695 696
        break;
      case kInfoLogFile:
        keep = true;
        if (number != 0) {
697
          old_info_log_files.push_back(to_delete);
J
jorlow@chromium.org 已提交
698
        }
K
kailiu 已提交
699 700 701 702 703 704 705 706 707 708 709 710 711 712 713
        break;
      case kCurrentFile:
      case kDBLockFile:
      case kIdentityFile:
      case kMetaDatabase:
        keep = true;
        break;
    }

    if (keep) {
      continue;
    }

    if (type == kTableFile) {
      // evict from cache
714
      TableCache::Evict(table_cache_.get(), number);
K
kailiu 已提交
715
    }
716

K
kailiu 已提交
717 718 719 720
    std::string fname = ((type == kLogFile) ? options_.wal_dir : dbname_) +
        "/" + to_delete;
    if (type == kLogFile &&
        (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0)) {
721
      auto archived_log_name = ArchivedLogFileName(options_.wal_dir, number);
722 723
      // The sync point below is used in (DBTest,TransactionLogIteratorRace)
      TEST_SYNC_POINT("DBImpl::PurgeObsoleteFiles:1");
724
      Status s = env_->RenameFile(fname, archived_log_name);
725 726
      // The sync point below is used in (DBTest,TransactionLogIteratorRace)
      TEST_SYNC_POINT("DBImpl::PurgeObsoleteFiles:2");
727 728 729
      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 已提交
730 731
    } else {
      Status s = env_->DeleteFile(fname);
732 733 734
      Log(options_.info_log, "Delete %s type=%d #%lu -- %s\n",
          fname.c_str(), type, (unsigned long)number,
          s.ToString().c_str());
J
jorlow@chromium.org 已提交
735 736
    }
  }
H
heyongqiang 已提交
737

738
  // Delete old info log files.
739
  size_t old_info_log_file_count = old_info_log_files.size();
K
Kai Liu 已提交
740 741
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
742
  if (old_info_log_file_count >= options_.keep_log_file_num &&
K
Kai Liu 已提交
743
      options_.db_log_dir.empty()) {
744 745
    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;
746
    for (unsigned int i = 0; i <= end; i++) {
747 748 749 750 751 752 753
      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 已提交
754 755
    }
  }
756
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
757
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
758 759 760 761 762
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
763
  FindObsoleteFiles(deletion_state, true);
764 765 766
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
767 768
}

I
Igor Canadi 已提交
769
#ifndef ROCKSDB_LITE
770 771 772 773 774 775 776 777
// 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
778
void DBImpl::PurgeObsoleteWALFiles() {
779 780 781 782 783 784
  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;
  }

785 786
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
787 788 789 790 791 792 793 794
  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_;
795

796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833
  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 已提交
834 835 836
          } else {
            MutexLock l(&read_first_record_cache_mutex_);
            read_first_record_cache_.erase(number);
837 838
          }
          continue;
839
        }
840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858
      }

      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 已提交
859 860 861
            } else {
              MutexLock l(&read_first_record_cache_mutex_);
              read_first_record_cache_.erase(number);
862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880
            }
          }
        }
      }
    }
  }

  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;
881
  GetSortedWalsOfType(archival_dir, archived_logs, kArchivedLogFile);
882 883 884 885 886 887 888 889 890 891 892 893 894 895

  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 已提交
896 897 898
    } else {
      MutexLock l(&read_first_record_cache_mutex_);
      read_first_record_cache_.erase(archived_logs[i]->LogNumber());
899 900
    }
  }
D
Dhruba Borthakur 已提交
901 902
}

903 904 905 906 907 908 909 910 911 912
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 已提交
913 914 915 916 917 918 919 920 921 922 923 924 925
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 已提交
926 927
      SequenceNumber sequence;
      Status s = ReadFirstRecord(log_type, number, &sequence);
I
Igor Canadi 已提交
928 929 930
      if (!s.ok()) {
        return s;
      }
I
Igor Canadi 已提交
931 932 933 934
      if (sequence == 0) {
        // empty file
        continue;
      }
I
Igor Canadi 已提交
935 936 937 938 939 940 941 942

      uint64_t size_bytes;
      s = env_->GetFileSize(LogFileName(path, number), &size_bytes);
      if (!s.ok()) {
        return s;
      }

      log_files.push_back(std::move(unique_ptr<LogFile>(
I
Igor Canadi 已提交
943
          new LogFileImpl(number, log_type, sequence, size_bytes))));
I
Igor Canadi 已提交
944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967
    }
  }
  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 已提交
968 969
  // end could be -ve.
  size_t start_index = std::max(static_cast<int64_t>(0), end);
I
Igor Canadi 已提交
970 971 972 973 974 975
  // 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 已提交
976 977 978 979 980 981 982 983 984 985 986 987 988
                               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 已提交
989 990
  if (type == kAliveLogFile) {
    std::string fname = LogFileName(options_.wal_dir, number);
I
Igor Canadi 已提交
991 992 993 994
    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 已提交
995
    }
I
Igor Canadi 已提交
996
  }
I
Igor Canadi 已提交
997

I
Igor Canadi 已提交
998
  if (type == kArchivedLogFile || !s.ok()) {
I
Igor Canadi 已提交
999 1000
    //  check if the file got moved to archive.
    std::string archived_file = ArchivedLogFileName(options_.wal_dir, number);
I
Igor Canadi 已提交
1001
    s = ReadFirstLine(archived_file, sequence);
I
Igor Canadi 已提交
1002
  }
I
Igor Canadi 已提交
1003 1004 1005 1006 1007 1008

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

I
Igor Canadi 已提交
1011 1012
// the function returns status.ok() and sequence == 0 if the file exists, but is
// empty
I
Igor Canadi 已提交
1013
Status DBImpl::ReadFirstLine(const std::string& fname,
I
Igor Canadi 已提交
1014
                             SequenceNumber* sequence) {
I
Igor Canadi 已提交
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
  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 已提交
1058 1059 1060
      WriteBatch batch;
      WriteBatchInternal::SetContents(&batch, record);
      *sequence = WriteBatchInternal::Sequence(&batch);
I
Igor Canadi 已提交
1061 1062 1063 1064
      return Status::OK();
    }
  }

I
Igor Canadi 已提交
1065 1066 1067
  // 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 已提交
1068 1069 1070 1071 1072
  return status;
}

#endif  // ROCKSDB_LITE

1073
Status DBImpl::Recover(
1074 1075
    const std::vector<ColumnFamilyDescriptor>& column_families, bool read_only,
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
1076 1077
  mutex_.AssertHeld();

1078
  bool is_new_db = false;
1079
  assert(db_lock_ == nullptr);
I
Igor Canadi 已提交
1080
  if (!read_only) {
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
    // 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;
    }

1093 1094 1095 1096 1097
    s = env_->NewDirectory(dbname_, &db_directory_);
    if (!s.ok()) {
      return s;
    }

1098
    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
1099 1100 1101
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
1102

1103 1104
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
1105
        // TODO: add merge_operator name check
1106
        s = NewDB();
1107
        is_new_db = true;
1108 1109 1110 1111 1112 1113
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
1114 1115
      }
    } else {
1116 1117 1118 1119
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
1120
    }
M
Mayank Agarwal 已提交
1121 1122 1123 1124 1125 1126 1127
    // 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 已提交
1128 1129
  }

1130
  Status s = versions_->Recover(column_families, read_only);
I
Igor Canadi 已提交
1131 1132 1133
  if (options_.paranoid_checks && s.ok()) {
    s = CheckConsistency();
  }
J
jorlow@chromium.org 已提交
1134 1135
  if (s.ok()) {
    SequenceNumber max_sequence(0);
1136 1137
    default_cf_handle_ = new ColumnFamilyHandleImpl(
        versions_->GetColumnFamilySet()->GetDefault(), this, &mutex_);
1138 1139 1140 1141 1142 1143 1144

    // 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
1145
    // produced by an older version of rocksdb.
1146
    const uint64_t min_log = versions_->MinLogNumber();
1147 1148
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
1149
    s = env_->GetChildren(options_.wal_dir, &filenames);
1150 1151
    if (!s.ok()) {
      return s;
1152
    }
K
kailiu 已提交
1153

1154 1155
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
K
kailiu 已提交
1156 1157
      uint64_t number;
      FileType type;
1158 1159 1160 1161 1162 1163 1164 1165 1166
      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);
        }
1167
      }
J
jorlow@chromium.org 已提交
1168
    }
1169

H
heyongqiang 已提交
1170 1171 1172 1173 1174 1175
    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");
    }

1176 1177
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
K
kailiu 已提交
1178
    for (const auto& log : logs) {
1179 1180 1181
      // 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 已提交
1182
      versions_->MarkFileNumberUsed(log);
K
Kai Liu 已提交
1183
      s = RecoverLogFile(log, &max_sequence, read_only);
1184
    }
I
Igor Canadi 已提交
1185 1186
    SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                   versions_->LastSequence());
J
jorlow@chromium.org 已提交
1187 1188 1189 1190 1191
  }

  return s;
}

I
Igor Canadi 已提交
1192 1193
Status DBImpl::RecoverLogFile(uint64_t log_number, SequenceNumber* max_sequence,
                              bool read_only) {
J
jorlow@chromium.org 已提交
1194 1195
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
1196
    Logger* info_log;
J
jorlow@chromium.org 已提交
1197
    const char* fname;
1198 1199
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
1200
    virtual void Corruption(size_t bytes, const Status& s) {
1201
      Log(info_log, "%s%s: dropping %d bytes; %s",
1202
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
1203
          fname, static_cast<int>(bytes), s.ToString().c_str());
1204
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
1205 1206 1207 1208 1209
    }
  };

  mutex_.AssertHeld();

1210
  std::unordered_map<int, VersionEdit> version_edits;
1211
  // no need to refcount because iteration is under mutex
1212 1213
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    VersionEdit edit;
1214 1215
    edit.SetColumnFamily(cfd->GetID());
    version_edits.insert({cfd->GetID(), edit});
1216
  }
I
Igor Canadi 已提交
1217

J
jorlow@chromium.org 已提交
1218
  // Open the log file
1219
  std::string fname = LogFileName(options_.wal_dir, log_number);
1220
  unique_ptr<SequentialFile> file;
1221
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
1222 1223 1224 1225 1226 1227 1228 1229
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
1230
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
1231
  reporter.fname = fname.c_str();
1232 1233
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
1234 1235 1236 1237
  // 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).
1238
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1239
                     0/*initial_offset*/);
K
Kai Liu 已提交
1240 1241
  Log(options_.info_log, "Recovering log #%lu",
      (unsigned long) log_number);
J
jorlow@chromium.org 已提交
1242 1243 1244 1245 1246

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
I
Igor Canadi 已提交
1247
  while (reader.ReadRecord(&record, &scratch)) {
J
jorlow@chromium.org 已提交
1248 1249 1250 1251 1252 1253 1254
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

1255
    status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
1256
        &batch, column_family_memtables_.get(), true, log_number);
1257

J
jorlow@chromium.org 已提交
1258 1259
    MaybeIgnoreError(&status);
    if (!status.ok()) {
I
Igor Canadi 已提交
1260
      return status;
J
jorlow@chromium.org 已提交
1261 1262 1263 1264 1265 1266 1267 1268
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

1269
    if (!read_only) {
1270 1271
      // no need to refcount since client still doesn't have access
      // to the DB and can not drop column families while we iterate
1272
      for (auto cfd : *versions_->GetColumnFamilySet()) {
1273
        if (cfd->mem()->ShouldFlush()) {
1274
          // If this asserts, it means that InsertInto failed in
1275 1276
          // filtering updates to already-flushed column families
          assert(cfd->GetLogNumber() <= log_number);
1277
          auto iter = version_edits.find(cfd->GetID());
1278 1279
          assert(iter != version_edits.end());
          VersionEdit* edit = &iter->second;
1280
          status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
1281 1282 1283 1284 1285 1286 1287 1288
          // 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 已提交
1289 1290 1291 1292
      }
    }
  }

1293 1294 1295 1296
  if (versions_->LastSequence() < *max_sequence) {
    versions_->SetLastSequence(*max_sequence);
  }

1297
  if (!read_only) {
1298 1299
    // no need to refcount since client still doesn't have access
    // to the DB and can not drop column families while we iterate
1300
    for (auto cfd : *versions_->GetColumnFamilySet()) {
1301
      auto iter = version_edits.find(cfd->GetID());
1302 1303 1304
      assert(iter != version_edits.end());
      VersionEdit* edit = &iter->second;

1305 1306 1307 1308
      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
1309
        // (in WriteBatch::InsertInto)
1310 1311 1312 1313 1314 1315 1316
        assert(cfd->mem()->GetFirstSequenceNumber() == 0);
        assert(edit->NumEntries() == 0);
        continue;
      }

      // flush the final memtable (if non-empty)
      if (cfd->mem()->GetFirstSequenceNumber() != 0) {
1317
        status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
1318
      }
1319 1320 1321 1322 1323
      // we still want to clear the memtable, even if the recovery failed
      cfd->CreateNewMemtable();
      if (!status.ok()) {
        return status;
      }
J
jorlow@chromium.org 已提交
1324

1325 1326 1327 1328 1329 1330 1331
      // 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);
1332 1333 1334
      // 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 已提交
1335
      // log number
1336
      versions_->MarkFileNumberUsed(log_number + 1);
1337 1338 1339 1340 1341
      status = versions_->LogAndApply(cfd, edit, &mutex_);
      if (!status.ok()) {
        return status;
      }
    }
1342
  }
I
Igor Canadi 已提交
1343

J
jorlow@chromium.org 已提交
1344 1345 1346
  return status;
}

1347 1348
Status DBImpl::WriteLevel0TableForRecovery(ColumnFamilyData* cfd, MemTable* mem,
                                           VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1349
  mutex_.AssertHeld();
1350
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1351 1352 1353
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
1354
  Iterator* iter = mem->NewIterator(ReadOptions(), true);
1355 1356 1357
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
I
Igor Canadi 已提交
1358 1359
  Log(options_.info_log, "[%s] Level-0 table #%lu: started",
      cfd->GetName().c_str(), (unsigned long)meta.number);
1360 1361 1362 1363

  Status s;
  {
    mutex_.Unlock();
I
Igor Canadi 已提交
1364
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1365
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1366
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1367
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1368
    LogFlush(options_.info_log);
1369 1370 1371
    mutex_.Lock();
  }

I
Igor Canadi 已提交
1372 1373 1374
  Log(options_.info_log, "[%s] Level-0 table #%lu: %lu bytes %s",
      cfd->GetName().c_str(), (unsigned long)meta.number,
      (unsigned long)meta.file_size, s.ToString().c_str());
J
jorlow@chromium.org 已提交
1375
  delete iter;
1376

1377
  pending_outputs_.erase(meta.number);
1378 1379 1380 1381 1382 1383

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

I
Igor Canadi 已提交
1388
  InternalStats::CompactionStats stats;
1389 1390
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
1391
  stats.files_out_levelnp1 = 1;
1392
  cfd->internal_stats()->AddCompactionStats(level, stats);
1393
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
J
jorlow@chromium.org 已提交
1394 1395 1396
  return s;
}

1397
Status DBImpl::WriteLevel0Table(ColumnFamilyData* cfd,
1398
                                autovector<MemTable*>& mems, VersionEdit* edit,
1399
                                uint64_t* filenumber, LogBuffer* log_buffer) {
J
jorlow@chromium.org 已提交
1400
  mutex_.AssertHeld();
1401 1402 1403 1404 1405
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
1406

1407 1408
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1409
    mems[0]->GetFirstSequenceNumber();
1410
  Version* base = cfd->current();
1411
  base->Ref();          // it is likely that we do not need this reference
1412 1413 1414
  Status s;
  {
    mutex_.Unlock();
H
Haobo Xu 已提交
1415
    log_buffer->FlushBufferToLog();
K
Kai Liu 已提交
1416
    std::vector<Iterator*> memtables;
1417
    for (MemTable* m : mems) {
I
Igor Canadi 已提交
1418 1419
      Log(options_.info_log, "[%s] Flushing memtable with next log file: %lu\n",
          cfd->GetName().c_str(), (unsigned long)m->GetNextLogNumber());
1420
      memtables.push_back(m->NewIterator(ReadOptions(), true));
1421
    }
I
Igor Canadi 已提交
1422
    Iterator* iter = NewMergingIterator(&cfd->internal_comparator(),
1423
                                        &memtables[0], memtables.size());
I
Igor Canadi 已提交
1424 1425
    Log(options_.info_log, "[%s] Level-0 flush table #%lu: started",
        cfd->GetName().c_str(), (unsigned long)meta.number);
1426

I
Igor Canadi 已提交
1427
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1428
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1429
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1430
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1431
    LogFlush(options_.info_log);
1432
    delete iter;
I
Igor Canadi 已提交
1433 1434 1435 1436 1437 1438 1439
    Log(options_.info_log, "[%s] Level-0 flush table #%lu: %lu bytes %s",
        cfd->GetName().c_str(), (unsigned long)meta.number,
        (unsigned long)meta.file_size, s.ToString().c_str());

    Version::LevelSummaryStorage tmp;
    Log(options_.info_log, "[%s] Level summary: %s\n", cfd->GetName().c_str(),
        cfd->current()->LevelSummary(&tmp));
1440 1441 1442
    if (!options_.disableDataSync) {
      db_directory_->Fsync();
    }
1443 1444
    mutex_.Lock();
  }
1445 1446
  base->Unref();

1447
  // re-acquire the most current version
1448
  base = cfd->current();
1449 1450 1451 1452 1453 1454

  // 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 已提交
1455
  // committed so that other threads can recognize this file as a
1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468
  // valid one.
  // pending_outputs_.erase(meta.number);

  // Note that if file_size is zero, the file has been deleted and
  // should not be added to the manifest.
  int level = 0;
  if (s.ok() && meta.file_size > 0) {
    const Slice min_user_key = meta.smallest.user_key();
    const Slice max_user_key = meta.largest.user_key();
    // if we have more than 1 background thread, then we cannot
    // insert files directly into higher levels because some other
    // threads could be concurrently producing compacted files for
    // that key range.
1469
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1470
        cfd->options()->compaction_style == kCompactionStyleLevel) {
1471 1472 1473
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
1474 1475
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1476 1477
  }

I
Igor Canadi 已提交
1478
  InternalStats::CompactionStats stats;
1479 1480
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
1481
  cfd->internal_stats()->AddCompactionStats(level, stats);
1482
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
1483 1484 1485
  return s;
}

1486 1487
Status DBImpl::FlushMemTableToOutputFile(ColumnFamilyData* cfd,
                                         bool* madeProgress,
H
Haobo Xu 已提交
1488 1489
                                         DeletionState& deletion_state,
                                         LogBuffer* log_buffer) {
1490
  mutex_.AssertHeld();
1491
  assert(cfd->imm()->size() != 0);
1492
  assert(cfd->imm()->IsFlushPending());
1493 1494 1495

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
K
Kai Liu 已提交
1496
  autovector<MemTable*> mems;
1497
  cfd->imm()->PickMemtablesToFlush(&mems);
1498
  if (mems.empty()) {
I
Igor Canadi 已提交
1499 1500
    LogToBuffer(log_buffer, "[%s] Nothing in memtable to flush",
                cfd->GetName().c_str());
L
Lei Jin 已提交
1501
    return Status::OK();
1502 1503 1504
  }

  // record the logfile_number_ before we release the mutex
1505 1506 1507
  // 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.
1508
  MemTable* m = mems[0];
1509 1510
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1511 1512
  // SetLogNumber(log_num) indicates logs with number smaller than log_num
  // will no longer be picked up for recovery.
1513 1514
  edit->SetLogNumber(mems.back()->GetNextLogNumber());
  edit->SetColumnFamily(cfd->GetID());
1515

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

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

L
Lei Jin 已提交
1524
  if (!s.ok()) {
1525
    cfd->imm()->RollbackMemtableFlush(mems, file_number, &pending_outputs_);
1526 1527
  } else {
    // Replace immutable memtable with the generated Table
1528 1529 1530
    s = cfd->imm()->InstallMemtableFlushResults(
        cfd, mems, versions_.get(), &mutex_, options_.info_log.get(),
        file_number, pending_outputs_, &deletion_state.memtables_to_free,
1531
        db_directory_.get(), log_buffer);
1532
  }
J
jorlow@chromium.org 已提交
1533 1534

  if (s.ok()) {
1535
    InstallSuperVersion(cfd, deletion_state);
1536 1537 1538
    if (madeProgress) {
      *madeProgress = 1;
    }
1539

1540
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1541

1542
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1543
      // add to deletion state
1544 1545 1546 1547 1548
      while (alive_log_files_.size() &&
             *alive_log_files_.begin() < versions_->MinLogNumber()) {
        deletion_state.log_delete_files.push_back(*alive_log_files_.begin());
        alive_log_files_.pop_front();
      }
1549
    }
J
jorlow@chromium.org 已提交
1550
  }
1551 1552 1553 1554 1555 1556 1557

  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 已提交
1558 1559 1560
  return s;
}

1561
Status DBImpl::CompactRange(ColumnFamilyHandle* column_family,
1562 1563
                            const Slice* begin, const Slice* end,
                            bool reduce_level, int target_level) {
1564 1565
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
I
Igor Canadi 已提交
1566 1567

  Status s = FlushMemTable(cfd, FlushOptions());
L
Lei Jin 已提交
1568 1569 1570 1571 1572
  if (!s.ok()) {
    LogFlush(options_.info_log);
    return s;
  }

G
Gabor Cselle 已提交
1573 1574 1575
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
I
Igor Canadi 已提交
1576 1577
    Version* base = cfd->current();
    for (int level = 1; level < cfd->NumberLevels(); level++) {
G
Gabor Cselle 已提交
1578 1579 1580 1581 1582
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
1583 1584 1585
  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 已提交
1586
    if (cfd->options()->compaction_style == kCompactionStyleUniversal ||
1587
        level == max_level_with_files) {
I
Igor Canadi 已提交
1588
      s = RunManualCompaction(cfd, level, level, begin, end);
1589
    } else {
I
Igor Canadi 已提交
1590
      s = RunManualCompaction(cfd, level, level + 1, begin, end);
L
Lei Jin 已提交
1591 1592 1593 1594
    }
    if (!s.ok()) {
      LogFlush(options_.info_log);
      return s;
1595
    }
G
Gabor Cselle 已提交
1596
  }
1597 1598

  if (reduce_level) {
I
Igor Canadi 已提交
1599
    s = ReFitLevel(cfd, max_level_with_files, target_level);
1600
  }
I
Igor Canadi 已提交
1601
  LogFlush(options_.info_log);
L
Lei Jin 已提交
1602 1603

  return s;
1604 1605 1606
}

// return the same level if it cannot be moved
I
Igor Canadi 已提交
1607
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd, int level) {
1608
  mutex_.AssertHeld();
I
Igor Canadi 已提交
1609
  Version* current = cfd->current();
1610
  int minimum_level = level;
1611
  for (int i = level - 1; i > 0; --i) {
1612
    // stop if level i is not empty
1613
    if (current->NumLevelFiles(i) > 0) break;
1614
    // stop if level i is too small (cannot fit the level files)
I
Igor Canadi 已提交
1615
    if (cfd->compaction_picker()->MaxBytesForLevel(i) <
1616 1617 1618
        current->NumLevelBytes(level)) {
      break;
    }
1619 1620 1621 1622 1623 1624

    minimum_level = i;
  }
  return minimum_level;
}

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

I
Igor Canadi 已提交
1628
  SuperVersion* superversion_to_free = nullptr;
1629
  SuperVersion* new_superversion = new SuperVersion();
I
Igor Canadi 已提交
1630 1631

  mutex_.Lock();
1632 1633 1634

  // only allow one thread refitting
  if (refitting_level_) {
I
Igor Canadi 已提交
1635
    mutex_.Unlock();
1636
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
I
Igor Canadi 已提交
1637
    delete new_superversion;
L
Lei Jin 已提交
1638
    return Status::NotSupported("another thread is refitting");
1639 1640 1641 1642 1643
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1644
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1645
    Log(options_.info_log,
1646 1647
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1648 1649 1650 1651
    bg_cv_.Wait();
  }

  // move to a smaller level
1652 1653
  int to_level = target_level;
  if (target_level < 0) {
I
Igor Canadi 已提交
1654
    to_level = FindMinimumEmptyLevelFitting(cfd, level);
1655
  }
1656 1657 1658

  assert(to_level <= level);

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

1664
    VersionEdit edit;
I
Igor Canadi 已提交
1665 1666
    edit.SetColumnFamily(cfd->GetID());
    for (const auto& f : cfd->current()->files_[level]) {
1667
      edit.DeleteFile(level, f->number);
1668 1669
      edit.AddFile(to_level, f->number, f->file_size, f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1670
    }
I
Igor Canadi 已提交
1671 1672
    Log(options_.info_log, "[%s] Apply version edit:\n%s",
        cfd->GetName().c_str(), edit.DebugString().data());
1673

I
Igor Canadi 已提交
1674
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
1675
    superversion_to_free = cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
1676
    new_superversion = nullptr;
1677

I
Igor Canadi 已提交
1678 1679
    Log(options_.info_log, "[%s] LogAndApply: %s\n", cfd->GetName().c_str(),
        status.ToString().data());
1680 1681

    if (status.ok()) {
I
Igor Canadi 已提交
1682 1683
      Log(options_.info_log, "[%s] After refitting:\n%s",
          cfd->GetName().c_str(), cfd->current()->DebugString().data());
1684 1685 1686 1687 1688
    }
  }

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
1689 1690 1691 1692

  mutex_.Unlock();
  delete superversion_to_free;
  delete new_superversion;
L
Lei Jin 已提交
1693
  return status;
G
Gabor Cselle 已提交
1694 1695
}

1696 1697 1698
int DBImpl::NumberLevels(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->NumberLevels();
1699 1700
}

1701 1702 1703
int DBImpl::MaxMemCompactionLevel(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->max_mem_compaction_level;
1704 1705
}

1706 1707 1708
int DBImpl::Level0StopWriteTrigger(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->level0_stop_writes_trigger;
1709 1710
}

1711
Status DBImpl::Flush(const FlushOptions& options,
1712 1713 1714
                     ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return FlushMemTable(cfh->cfd(), options);
H
heyongqiang 已提交
1715 1716
}

1717
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1718 1719 1720
  return versions_->LastSequence();
}

I
Igor Canadi 已提交
1721 1722
Status DBImpl::RunManualCompaction(ColumnFamilyData* cfd, int input_level,
                                   int output_level, const Slice* begin,
L
Lei Jin 已提交
1723
                                   const Slice* end) {
1724
  assert(input_level >= 0);
1725

G
Gabor Cselle 已提交
1726 1727
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1728
  ManualCompaction manual;
I
Igor Canadi 已提交
1729
  manual.cfd = cfd;
1730 1731
  manual.input_level = input_level;
  manual.output_level = output_level;
G
Gabor Cselle 已提交
1732
  manual.done = false;
1733
  manual.in_progress = false;
1734 1735 1736
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
I
Igor Canadi 已提交
1737
      cfd->options()->compaction_style == kCompactionStyleUniversal) {
1738
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1739 1740 1741 1742
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1743
  if (end == nullptr ||
I
Igor Canadi 已提交
1744
      cfd->options()->compaction_style == kCompactionStyleUniversal) {
1745
    manual.end = nullptr;
G
Gabor Cselle 已提交
1746 1747 1748 1749 1750 1751
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1752

1753 1754 1755 1756 1757 1758
  // 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
1759
  // RunManualCompaction(), i.e. during that time no other compaction will
1760 1761 1762
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
1763
  // RunManualCompaction() from getting to the second while loop below.
1764 1765 1766 1767 1768 1769
  // 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 已提交
1770 1771 1772
        "[%s] Manual compaction waiting for all other scheduled background "
        "compactions to finish",
        cfd->GetName().c_str());
1773 1774
    bg_cv_.Wait();
  }
1775

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

1779 1780 1781 1782
  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 已提交
1783
      bg_cv_.Wait();
1784 1785 1786
    } else {
      manual_compaction_ = &manual;
      MaybeScheduleFlushOrCompaction();
G
Gabor Cselle 已提交
1787
    }
H
hans@chromium.org 已提交
1788
  }
1789

1790 1791 1792
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
L
Lei Jin 已提交
1793
  return manual.status;
J
jorlow@chromium.org 已提交
1794 1795
}

1796 1797
Status DBImpl::FlushMemTable(ColumnFamilyData* cfd,
                             const FlushOptions& options) {
1798 1799
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1800
  if (s.ok() && options.wait) {
1801
    // Wait until the compaction completes
1802
    s = WaitForFlushMemTable(cfd);
1803 1804
  }
  return s;
J
jorlow@chromium.org 已提交
1805 1806
}

1807
Status DBImpl::WaitForFlushMemTable(ColumnFamilyData* cfd) {
1808 1809 1810
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1811
  while (cfd->imm()->size() > 0 && bg_error_.ok()) {
1812 1813
    bg_cv_.Wait();
  }
1814
  if (!bg_error_.ok()) {
1815 1816 1817
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1818 1819
}

1820
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
1821
  mutex_.AssertHeld();
1822
  bg_schedule_needed_ = false;
1823 1824
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
J
jorlow@chromium.org 已提交
1825 1826 1827
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
  } else {
1828
    bool is_flush_pending = false;
1829
    // no need to refcount since we're under a mutex
1830
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1831
      if (cfd->imm()->IsFlushPending()) {
1832 1833 1834
        is_flush_pending = true;
      }
    }
1835
    if (is_flush_pending) {
1836
      // memtable flush needed
1837 1838 1839
      if (bg_flush_scheduled_ < options_.max_background_flushes) {
        bg_flush_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::HIGH);
1840
      } else if (options_.max_background_flushes > 0) {
1841
        bg_schedule_needed_ = true;
1842
      }
1843
    }
1844
    bool is_compaction_needed = false;
1845
    // no need to refcount since we're under a mutex
1846
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1847
      if (cfd->current()->NeedsCompaction()) {
1848 1849 1850 1851
        is_compaction_needed = true;
        break;
      }
    }
1852

1853 1854
    // Schedule BGWorkCompaction if there's a compaction pending (or a memtable
    // flush, but the HIGH pool is not enabled)
1855
    // Do it only if max_background_compactions hasn't been reached and, in case
1856
    // bg_manual_only_ > 0, if it's a manual compaction.
1857 1858
    if ((manual_compaction_ || is_compaction_needed ||
         (is_flush_pending && options_.max_background_flushes == 0)) &&
1859
        (!bg_manual_only_ || manual_compaction_)) {
1860 1861 1862 1863 1864 1865
      if (bg_compaction_scheduled_ < options_.max_background_compactions) {
        bg_compaction_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW);
      } else {
        bg_schedule_needed_ = true;
      }
1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
    }
  }
}

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

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

I
Igor Canadi 已提交
1878
Status DBImpl::BackgroundFlush(bool* madeProgress,
H
Haobo Xu 已提交
1879 1880
                               DeletionState& deletion_state,
                               LogBuffer* log_buffer) {
1881 1882 1883 1884 1885 1886
  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
1887
  for (auto cfd : *versions_->GetColumnFamilySet()) {
1888 1889 1890
    cfd->Ref();
    Status flush_status;
    while (flush_status.ok() && cfd->imm()->IsFlushPending()) {
1891 1892
      LogToBuffer(
          log_buffer,
1893
          "BackgroundCallFlush doing FlushMemTableToOutputFile with column "
I
Igor Canadi 已提交
1894 1895 1896
          "family [%s], flush slots available %d",
          cfd->GetName().c_str(),
          options_.max_background_flushes - bg_flush_scheduled_);
1897 1898
      flush_status = FlushMemTableToOutputFile(cfd, madeProgress,
                                               deletion_state, log_buffer);
1899 1900 1901
    }
    if (call_status.ok() && !flush_status.ok()) {
      call_status = flush_status;
1902
    }
1903
    cfd->Unref();
J
jorlow@chromium.org 已提交
1904
  }
1905
  versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
1906
  return call_status;
J
jorlow@chromium.org 已提交
1907 1908
}

1909
void DBImpl::BackgroundCallFlush() {
1910
  bool madeProgress = false;
K
Kai Liu 已提交
1911
  DeletionState deletion_state(true);
1912 1913
  assert(bg_flush_scheduled_);

1914
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, options_.info_log.get());
H
Haobo Xu 已提交
1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925
  {
    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.
1926 1927 1928
        uint64_t error_cnt = default_cf_handle_->cfd()
                                 ->internal_stats()
                                 ->BumpAndGetBackgroundErrorCount();
H
Haobo Xu 已提交
1929 1930
        bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
        mutex_.Unlock();
1931 1932 1933 1934
        Log(options_.info_log,
            "Waiting after background flush error: %s"
            "Accumulated background error counts: %" PRIu64,
            s.ToString().c_str(), error_cnt);
H
Haobo Xu 已提交
1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945
        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
1946
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
1947
      mutex_.Unlock();
1948 1949 1950 1951 1952
      // 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 已提交
1953
      log_buffer.FlushBufferToLog();
1954 1955 1956
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
1957 1958
      mutex_.Lock();
    }
I
Igor Canadi 已提交
1959

H
Haobo Xu 已提交
1960
    bg_flush_scheduled_--;
1961 1962 1963 1964 1965
    // 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 已提交
1966 1967 1968
      MaybeScheduleFlushOrCompaction();
    }
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
1969 1970 1971 1972
    // 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.
1973
  }
J
jorlow@chromium.org 已提交
1974 1975
}

1976
void DBImpl::BackgroundCallCompaction() {
1977
  bool madeProgress = false;
K
Kai Liu 已提交
1978
  DeletionState deletion_state(true);
H
Haobo Xu 已提交
1979 1980

  MaybeDumpStats();
1981
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, options_.info_log.get());
1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
  {
    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.
1993 1994 1995
        uint64_t error_cnt = default_cf_handle_->cfd()
                                 ->internal_stats()
                                 ->BumpAndGetBackgroundErrorCount();
1996 1997
        bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
        mutex_.Unlock();
H
Haobo Xu 已提交
1998
        log_buffer.FlushBufferToLog();
1999 2000 2001 2002
        Log(options_.info_log,
            "Waiting after background compaction error: %s, "
            "Accumulated background error counts: %" PRIu64,
            s.ToString().c_str(), error_cnt);
2003 2004 2005 2006 2007
        LogFlush(options_.info_log);
        env_->SleepForMicroseconds(1000000);
        mutex_.Lock();
      }
    }
H
Haobo Xu 已提交
2008

2009 2010 2011 2012 2013 2014 2015
    // 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
2016
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
2017
      mutex_.Unlock();
2018 2019 2020 2021 2022
      // 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 已提交
2023
      log_buffer.FlushBufferToLog();
2024 2025 2026
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
2027 2028
      mutex_.Lock();
    }
D
Dhruba Borthakur 已提交
2029

2030
    bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
2031

2032
    MaybeScheduleLogDBDeployStats();
2033

2034 2035
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();

2036 2037 2038
    // Previous compaction may have produced too many files in a level,
    // So reschedule another compaction if we made progress in the
    // last compaction.
2039 2040 2041 2042 2043 2044
    //
    // 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_) {
2045 2046 2047
      MaybeScheduleFlushOrCompaction();
    }
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
2048 2049 2050 2051
    // 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.
2052
  }
J
jorlow@chromium.org 已提交
2053 2054
}

A
Abhishek Kona 已提交
2055
Status DBImpl::BackgroundCompaction(bool* madeProgress,
2056 2057
                                    DeletionState& deletion_state,
                                    LogBuffer* log_buffer) {
2058
  *madeProgress = false;
J
jorlow@chromium.org 已提交
2059
  mutex_.AssertHeld();
2060

2061 2062
  bool is_manual = (manual_compaction_ != nullptr) &&
                   (manual_compaction_->in_progress == false);
2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080

  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);
2081
      cfd->Unref();
2082 2083 2084 2085 2086 2087 2088
      if (!flush_stat.ok()) {
        if (is_manual) {
          manual_compaction_->status = flush_stat;
          manual_compaction_->done = true;
          manual_compaction_->in_progress = false;
          manual_compaction_ = nullptr;
        }
2089
        return flush_stat;
2090 2091 2092 2093 2094
      }
    }
  }

  unique_ptr<Compaction> c;
2095 2096
  InternalKey manual_end_storage;
  InternalKey* manual_end = &manual_end_storage;
H
hans@chromium.org 已提交
2097
  if (is_manual) {
G
Gabor Cselle 已提交
2098
    ManualCompaction* m = manual_compaction_;
2099
    assert(m->in_progress);
I
Igor Canadi 已提交
2100 2101
    c.reset(m->cfd->CompactRange(m->input_level, m->output_level, m->begin,
                                 m->end, &manual_end));
2102
    if (!c) {
2103
      m->done = true;
G
Gabor Cselle 已提交
2104
    }
I
Igor Canadi 已提交
2105 2106 2107 2108 2109 2110 2111 2112 2113
    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 已提交
2114
  } else {
2115
    // no need to refcount in iteration since it's always under a mutex
I
Igor Canadi 已提交
2116
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
2117
      if (!cfd->options()->disable_auto_compactions) {
2118
        c.reset(cfd->PickCompaction(log_buffer));
I
Igor Canadi 已提交
2119 2120 2121 2122 2123 2124
        if (c != nullptr) {
          // update statistics
          MeasureTime(options_.statistics.get(), NUM_FILES_IN_SINGLE_COMPACTION,
                      c->inputs(0)->size());
          break;
        }
I
Igor Canadi 已提交
2125 2126
      }
    }
J
jorlow@chromium.org 已提交
2127 2128 2129
  }

  Status status;
2130
  if (!c) {
H
hans@chromium.org 已提交
2131
    // Nothing to do
2132
    LogToBuffer(log_buffer, "Compaction nothing to do");
H
hans@chromium.org 已提交
2133
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
2134
    // Move file to next level
2135
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
2136 2137 2138
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
2139 2140
                       f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
I
Igor Canadi 已提交
2141
    status = versions_->LogAndApply(c->column_family_data(), c->edit(), &mutex_,
2142
                                    db_directory_.get());
I
Igor Canadi 已提交
2143
    InstallSuperVersion(c->column_family_data(), deletion_state);
2144

2145
    Version::LevelSummaryStorage tmp;
I
Igor Canadi 已提交
2146 2147 2148 2149 2150 2151
    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->number), c->level() + 1,
                static_cast<unsigned long long>(f->file_size),
                status.ToString().c_str(),
                c->input_version()->LevelSummary(&tmp));
I
Igor Canadi 已提交
2152
    c->ReleaseCompactionFiles(status);
2153
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2154
  } else {
2155
    MaybeScheduleFlushOrCompaction(); // do more compaction work in parallel.
2156
    CompactionState* compact = new CompactionState(c.get());
H
Haobo Xu 已提交
2157
    status = DoCompactionWork(compact, deletion_state, log_buffer);
2158
    CleanupCompaction(compact, status);
I
Igor Canadi 已提交
2159
    c->ReleaseCompactionFiles(status);
2160
    c->ReleaseInputs();
2161
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2162
  }
2163
  c.reset();
J
jorlow@chromium.org 已提交
2164 2165 2166 2167 2168 2169

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
2170
    Log(InfoLogLevel::WARN_LEVEL, options_.info_log, "Compaction error: %s",
2171
        status.ToString().c_str());
J
jorlow@chromium.org 已提交
2172 2173 2174 2175
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
2176 2177

  if (is_manual) {
G
Gabor Cselle 已提交
2178
    ManualCompaction* m = manual_compaction_;
2179
    if (!status.ok()) {
L
Lei Jin 已提交
2180
      m->status = status;
2181 2182
      m->done = true;
    }
2183 2184 2185 2186 2187 2188 2189 2190 2191
    // 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.
2192 2193 2194 2195 2196
    //
    // 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) {
2197 2198
      m->done = true;
    }
G
Gabor Cselle 已提交
2199 2200 2201
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
2202
      // Universal compaction should always compact the whole range
I
Igor Canadi 已提交
2203
      assert(m->cfd->options()->compaction_style != kCompactionStyleUniversal);
2204
      m->tmp_storage = *manual_end;
G
Gabor Cselle 已提交
2205 2206
      m->begin = &m->tmp_storage;
    }
2207
    m->in_progress = false; // not being processed anymore
2208
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
2209
  }
2210
  return status;
J
jorlow@chromium.org 已提交
2211 2212
}

2213
void DBImpl::CleanupCompaction(CompactionState* compact, Status status) {
J
jorlow@chromium.org 已提交
2214
  mutex_.AssertHeld();
2215
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
2216 2217
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
2218
    compact->builder.reset();
J
jorlow@chromium.org 已提交
2219
  } else {
2220
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
2221
  }
D
dgrogan@chromium.org 已提交
2222
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2223 2224
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
2225 2226 2227 2228

    // 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 已提交
2229
      TableCache::Evict(table_cache_.get(), out.number);
2230
    }
J
jorlow@chromium.org 已提交
2231 2232 2233 2234
  }
  delete compact;
}

2235
// Allocate the file numbers for the output file. We allocate as
2236
// many output file numbers as there are files in level+1 (at least one)
2237 2238 2239
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2240 2241
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
2242
  int filesNeeded = compact->compaction->num_input_files(1);
2243
  for (int i = 0; i < std::max(filesNeeded, 1); i++) {
2244 2245 2246 2247 2248 2249 2250 2251 2252
    uint64_t file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    compact->allocated_file_numbers.push_back(file_number);
  }
}

// Frees up unused file number.
void DBImpl::ReleaseCompactionUnusedFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2253
  for (const auto file_number : compact->allocated_file_numbers) {
2254 2255 2256 2257
    pending_outputs_.erase(file_number);
  }
}

J
jorlow@chromium.org 已提交
2258
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
2259 2260
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
2261
  uint64_t file_number;
2262 2263 2264 2265 2266 2267 2268
  // 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 已提交
2269 2270 2271 2272 2273
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
2274 2275 2276 2277
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
2278
  out.smallest_seqno = out.largest_seqno = 0;
2279
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
2280 2281 2282

  // Make the output file
  std::string fname = TableFileName(dbname_, file_number);
2283
  Status s = env_->NewWritableFile(fname, &compact->outfile, storage_options_);
2284

J
jorlow@chromium.org 已提交
2285
  if (s.ok()) {
2286 2287
    // Over-estimate slightly so we don't end up just barely crossing
    // the threshold.
I
Igor Canadi 已提交
2288
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2289
    compact->outfile->SetPreallocationBlockSize(
I
Igor Canadi 已提交
2290
        1.1 * cfd->compaction_picker()->MaxFileSizeForLevel(
2291
                  compact->compaction->output_level()));
2292

I
Igor Canadi 已提交
2293 2294 2295
    CompressionType compression_type =
        GetCompressionType(*cfd->options(), compact->compaction->output_level(),
                           compact->compaction->enable_compression());
S
Siying Dong 已提交
2296

2297
    compact->builder.reset(
I
Igor Canadi 已提交
2298
        NewTableBuilder(*cfd->options(), cfd->internal_comparator(),
2299
                        compact->outfile.get(), compression_type));
J
jorlow@chromium.org 已提交
2300
  }
I
Igor Canadi 已提交
2301
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2302 2303 2304 2305 2306
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
2307
  assert(compact != nullptr);
2308
  assert(compact->outfile);
2309
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324

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

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

  // Finish and check for file errors
2328
  if (s.ok() && !options_.disableDataSync) {
2329
    if (options_.use_fsync) {
2330
      StopWatch sw(env_, options_.statistics.get(),
2331
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2332 2333
      s = compact->outfile->Fsync();
    } else {
2334
      StopWatch sw(env_, options_.statistics.get(),
2335
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2336 2337
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
2338 2339 2340 2341
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
2342
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
2343 2344 2345

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
I
Igor Canadi 已提交
2346
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2347
    FileMetaData meta(output_number, current_bytes);
I
Igor Canadi 已提交
2348
    Iterator* iter = cfd->table_cache()->NewIterator(
2349
        ReadOptions(), storage_options_, cfd->internal_comparator(), meta);
J
jorlow@chromium.org 已提交
2350 2351 2352
    s = iter->status();
    delete iter;
    if (s.ok()) {
I
Igor Canadi 已提交
2353 2354 2355
      Log(options_.info_log, "[%s] Generated table #%lu: %lu keys, %lu bytes",
          cfd->GetName().c_str(), (unsigned long)output_number,
          (unsigned long)current_entries, (unsigned long)current_bytes);
J
jorlow@chromium.org 已提交
2356 2357 2358 2359 2360 2361
    }
  }
  return s;
}


2362 2363
Status DBImpl::InstallCompactionResults(CompactionState* compact,
                                        LogBuffer* log_buffer) {
J
jorlow@chromium.org 已提交
2364
  mutex_.AssertHeld();
2365 2366 2367 2368 2369

  // 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.
2370
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
I
Igor Canadi 已提交
2371 2372 2373 2374 2375
    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 已提交
2376
    return Status::Corruption("Compaction input files inconsistent");
2377 2378
  }

I
Igor Canadi 已提交
2379 2380
  LogToBuffer(log_buffer, "[%s] Compacted %d@%d + %d@%d files => %lld bytes",
              compact->compaction->column_family_data()->GetName().c_str(),
2381 2382 2383 2384 2385
              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 已提交
2386 2387 2388

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
D
dgrogan@chromium.org 已提交
2389
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2390 2391
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
2392 2393
        compact->compaction->output_level(), out.number, out.file_size,
        out.smallest, out.largest, out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2394
  }
I
Igor Canadi 已提交
2395 2396 2397
  return versions_->LogAndApply(compact->compaction->column_family_data(),
                                compact->compaction->edit(), &mutex_,
                                db_directory_.get());
J
jorlow@chromium.org 已提交
2398 2399
}

2400 2401 2402 2403 2404 2405 2406
// 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(
2407 2408
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
2409 2410 2411
  if (!IsSnapshotSupported()) {
    return 0;
  }
2412
  SequenceNumber prev __attribute__((unused)) = 0;
2413 2414 2415
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
2416
      *prev_snapshot = prev;
2417
      return cur;
2418
    }
2419 2420
    prev = cur; // assignment
    assert(prev);
2421 2422
  }
  Log(options_.info_log,
K
Kai Liu 已提交
2423 2424 2425
      "Looking for seqid %lu but maxseqid is %lu",
      (unsigned long)in,
      (unsigned long)snapshots[snapshots.size()-1]);
2426 2427 2428 2429
  assert(0);
  return 0;
}

I
Igor Canadi 已提交
2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448
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 已提交
2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460
Status DBImpl::ProcessKeyValueCompaction(
    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 已提交
2461
  Status status;
D
Danny Guo 已提交
2462
  std::string compaction_filter_value;
J
jorlow@chromium.org 已提交
2463
  ParsedInternalKey ikey;
2464
  IterKey current_user_key;
J
jorlow@chromium.org 已提交
2465
  bool has_current_user_key = false;
2466
  IterKey delete_key;
2467 2468
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
2469
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
2470
  ColumnFamilyData* cfd = compact->compaction->column_family_data();
I
Igor Canadi 已提交
2471 2472
  MergeHelper merge(
      cfd->user_comparator(), cfd->options()->merge_operator.get(),
2473 2474
      options_.info_log.get(), cfd->options()->min_partial_merge_operands,
      false /* internal key corruption is expected */);
I
Igor Canadi 已提交
2475
  auto compaction_filter = cfd->options()->compaction_filter;
2476 2477
  std::unique_ptr<CompactionFilter> compaction_filter_from_factory = nullptr;
  if (!compaction_filter) {
2478
    auto context = compact->GetFilterContextV1();
2479
    compaction_filter_from_factory =
I
Igor Canadi 已提交
2480 2481
        cfd->options()->compaction_filter_factory->CreateCompactionFilter(
            context);
2482 2483
    compaction_filter = compaction_filter_from_factory.get();
  }
2484

2485 2486 2487 2488 2489 2490
  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 已提交
2491
    imm_micros += CallFlushDuringCompaction(cfd, deletion_state, log_buffer);
2492

D
Danny Guo 已提交
2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513
    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 已提交
2514

2515
    if (compact->compaction->ShouldStopBefore(key) &&
2516
        compact->builder != nullptr) {
D
Danny Guo 已提交
2517
      status = FinishCompactionOutputFile(compact, input);
2518 2519 2520 2521 2522 2523
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2524
    bool drop = false;
2525
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2526 2527
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2528 2529
      // 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.
2530
      current_user_key.Clear();
J
jorlow@chromium.org 已提交
2531 2532
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2533
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2534 2535
    } else {
      if (!has_current_user_key ||
2536
          cfd->user_comparator()->Compare(ikey.user_key,
2537
                                          current_user_key.GetKey()) != 0) {
J
jorlow@chromium.org 已提交
2538
        // First occurrence of this user key
2539
        current_user_key.SetUserKey(ikey.user_key);
J
jorlow@chromium.org 已提交
2540 2541
        has_current_user_key = true;
        last_sequence_for_key = kMaxSequenceNumber;
2542
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2543
        // apply the compaction filter to the first occurrence of the user key
D
Danny Guo 已提交
2544
        if (compaction_filter && !is_compaction_v2 &&
H
Haobo Xu 已提交
2545 2546 2547 2548 2549 2550 2551 2552 2553
            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 已提交
2554 2555 2556
          bool to_delete = compaction_filter->Filter(
              compact->compaction->level(), ikey.user_key, value,
              &compaction_filter_value, &value_changed);
H
Haobo Xu 已提交
2557
          if (to_delete) {
2558 2559 2560
            // make a copy of the original key and convert it to a delete
            delete_key.SetInternalKey(ExtractUserKey(key), ikey.sequence,
                                      kTypeDeletion);
H
Haobo Xu 已提交
2561
            // anchor the key again
2562
            key = delete_key.GetKey();
H
Haobo Xu 已提交
2563 2564 2565 2566
            // needed because ikey is backed by key
            ParseInternalKey(key, &ikey);
            // no value associated with delete
            value.clear();
2567
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2568 2569 2570 2571
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }
J
jorlow@chromium.org 已提交
2572 2573
      }

2574 2575 2576
      // 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.
2577 2578 2579 2580
      SequenceNumber prev_snapshot = 0; // 0 means no previous snapshot
      SequenceNumber visible = visible_at_tip ?
        visible_at_tip :
        findEarliestVisibleSnapshot(ikey.sequence,
D
Danny Guo 已提交
2581 2582
            compact->existing_snapshots,
            &prev_snapshot);
2583 2584 2585 2586 2587

      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 已提交
2588
        // Hidden by an newer entry for same user key
2589
        // TODO: why not > ?
2590
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2591
        drop = true;    // (A)
2592
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2593
      } else if (ikey.type == kTypeDeletion &&
D
Danny Guo 已提交
2594 2595
          ikey.sequence <= earliest_snapshot &&
          compact->compaction->IsBaseLevelForKey(ikey.user_key)) {
J
jorlow@chromium.org 已提交
2596 2597 2598 2599 2600 2601 2602 2603
        // 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;
2604
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2605 2606 2607 2608 2609 2610 2611
      } 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 已提交
2612 2613 2614 2615 2616 2617
        int steps = 0;
        merge.MergeUntil(input, prev_snapshot, bottommost_level,
            options_.statistics.get(), &steps);
        // Skip the Merge ops
        combined_idx = combined_idx - 1 + steps;

2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635
        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 已提交
2636 2637 2638
      }

      last_sequence_for_key = ikey.sequence;
2639
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
2640 2641 2642
    }

    if (!drop) {
2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658
      // 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);
2659
      }
2660

2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671
      // 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.
2672
        if (bottommost_level && ikey.sequence < earliest_snapshot &&
2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683
            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
2684

2685 2686 2687 2688 2689 2690 2691
        // Open output file if necessary
        if (compact->builder == nullptr) {
          status = OpenCompactionOutputFile(compact);
          if (!status.ok()) {
            break;
          }
        }
2692 2693

        SequenceNumber seqno = GetInternalKeySeqno(newkey);
2694 2695
        if (compact->builder->NumEntries() == 0) {
          compact->current_output()->smallest.DecodeFrom(newkey);
2696 2697 2698 2699
          compact->current_output()->smallest_seqno = seqno;
        } else {
          compact->current_output()->smallest_seqno =
            std::min(compact->current_output()->smallest_seqno, seqno);
2700 2701 2702
        }
        compact->current_output()->largest.DecodeFrom(newkey);
        compact->builder->Add(newkey, value);
2703 2704
        compact->current_output()->largest_seqno =
          std::max(compact->current_output()->largest_seqno, seqno);
2705 2706 2707 2708

        // Close output file if it is big enough
        if (compact->builder->FileSize() >=
            compact->compaction->MaxOutputFileSize()) {
D
Danny Guo 已提交
2709
          status = FinishCompactionOutputFile(compact, input);
2710 2711 2712
          if (!status.ok()) {
            break;
          }
J
jorlow@chromium.org 已提交
2713 2714
        }

2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734
        // 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 已提交
2735 2736 2737 2738 2739
          break;
        }
      }
    }

2740
    // MergeUntil has moved input to the next entry
2741
    if (!current_entry_is_merging) {
2742 2743
      input->Next();
    }
J
jorlow@chromium.org 已提交
2744 2745
  }

D
Danny Guo 已提交
2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
  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();
2806
  bool prefix_initialized = false;
D
Danny Guo 已提交
2807 2808

  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions
2809
  ColumnFamilyData* cfd = compact->compaction->column_family_data();
2810 2811
  LogToBuffer(
      log_buffer,
I
Igor Canadi 已提交
2812 2813
      "[%s] Compacting %d@%d + %d@%d files, score %.2f slots available %d",
      cfd->GetName().c_str(), compact->compaction->num_input_files(0),
2814 2815
      compact->compaction->level(), compact->compaction->num_input_files(1),
      compact->compaction->output_level(), compact->compaction->score(),
D
Danny Guo 已提交
2816 2817 2818
      options_.max_background_compactions - bg_compaction_scheduled_);
  char scratch[2345];
  compact->compaction->Summary(scratch, sizeof(scratch));
I
Igor Canadi 已提交
2819 2820
  LogToBuffer(log_buffer, "[%s] Compaction start summary: %s\n",
              cfd->GetName().c_str(), scratch);
D
Danny Guo 已提交
2821

2822
  assert(cfd->current()->NumLevelFiles(compact->compaction->level()) > 0);
D
Danny Guo 已提交
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
  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);

  // Release mutex while we're actually doing the compaction work
  mutex_.Unlock();
2850
  log_buffer->FlushBufferToLog();
D
Danny Guo 已提交
2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864

  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 =
2865 2866
      cfd->options()->compaction_filter_factory_v2->CreateCompactionFilterV2(
          context);
D
Danny Guo 已提交
2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877
  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) {
2878 2879
    while (backup_input->Valid() && !shutting_down_.Acquire_Load() &&
           !cfd->IsDropped()) {
I
Igor Canadi 已提交
2880 2881 2882 2883 2884 2885
      // 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 已提交
2886 2887 2888 2889
      Slice key = backup_input->key();
      Slice value = backup_input->value();

      const SliceTransform* transformer =
2890
          cfd->options()->compaction_filter_factory_v2->GetPrefixExtractor();
2891
      const auto key_prefix = transformer->Transform(key);
2892
      if (!prefix_initialized) {
2893
        compact->cur_prefix_ = key_prefix.ToString();
2894
        prefix_initialized = true;
D
Danny Guo 已提交
2895 2896 2897
      }
      if (!ParseInternalKey(key, &ikey)) {
        // log error
I
Igor Canadi 已提交
2898 2899
        Log(options_.info_log, "[%s] Failed to parse key: %s",
            cfd->GetName().c_str(), key.ToString().c_str());
D
Danny Guo 已提交
2900 2901 2902
        continue;
      } else {
        // If the prefix remains the same, keep buffering
2903
        if (key_prefix.compare(Slice(compact->cur_prefix_)) == 0) {
D
Danny Guo 已提交
2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923
          // 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);
          }
2924
          compact->cur_prefix_ = key_prefix.ToString();
D
Danny Guo 已提交
2925 2926 2927 2928
        }
      }

      // Merge this batch of data (values + ineligible keys)
2929
      compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965

      // Done buffering for the current prefix. Spit it out to disk
      // Now just iterate through all the kv-pairs
      status = ProcessKeyValueCompaction(
          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);
        }
2966
        compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987

        status = ProcessKeyValueCompaction(
            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);
    }
2988
    compact->MergeKeyValueSliceBuffer(&cfd->internal_comparator());
D
Danny Guo 已提交
2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015
    status = ProcessKeyValueCompaction(
        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(
      visible_at_tip,
      earliest_snapshot,
      latest_snapshot,
      deletion_state,
      bottommost_level,
      imm_micros,
      input.get(),
      compact,
      false,
      log_buffer);
  }

3016
  if (status.ok() && (shutting_down_.Acquire_Load() || cfd->IsDropped())) {
L
Lei Jin 已提交
3017
    status = Status::ShutdownInProgress(
3018
        "Database shutdown or Column family drop during compaction");
J
jorlow@chromium.org 已提交
3019
  }
3020
  if (status.ok() && compact->builder != nullptr) {
3021
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
3022 3023 3024 3025
  }
  if (status.ok()) {
    status = input->status();
  }
3026
  input.reset();
J
jorlow@chromium.org 已提交
3027

3028 3029 3030
  if (!options_.disableDataSync) {
    db_directory_->Fsync();
  }
I
Igor Canadi 已提交
3031 3032

  InternalStats::CompactionStats stats;
3033
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
I
Igor Canadi 已提交
3034
  MeasureTime(options_.statistics.get(), COMPACTION_TIME, stats.micros);
M
Mark Callaghan 已提交
3035 3036
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
3037 3038

  int num_output_files = compact->outputs.size();
3039
  if (compact->builder != nullptr) {
P
Pascal Borreli 已提交
3040
    // An error occurred so ignore the last output.
3041 3042 3043 3044
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
3045

3046
  for (int i = 0; i < compact->compaction->num_input_files(0); i++) {
M
Mark Callaghan 已提交
3047
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;
3048 3049 3050
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(0, i)->file_size);
  }
M
Mark Callaghan 已提交
3051

3052
  for (int i = 0; i < compact->compaction->num_input_files(1); i++) {
M
Mark Callaghan 已提交
3053
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;
3054 3055 3056
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(1, i)->file_size);
  }
M
Mark Callaghan 已提交
3057

3058
  for (int i = 0; i < num_output_files; i++) {
3059
    stats.bytes_written += compact->outputs[i].file_size;
3060 3061
    RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
               compact->outputs[i].file_size);
3062 3063
  }

I
Igor Canadi 已提交
3064
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
3065
  mutex_.Lock();
3066 3067
  cfd->internal_stats()->AddCompactionStats(compact->compaction->output_level(),
                                            stats);
J
jorlow@chromium.org 已提交
3068

3069 3070 3071 3072
  // 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 已提交
3073
  if (status.ok()) {
3074
    status = InstallCompactionResults(compact, log_buffer);
I
Igor Canadi 已提交
3075
    InstallSuperVersion(cfd, deletion_state);
J
jorlow@chromium.org 已提交
3076
  }
3077
  Version::LevelSummaryStorage tmp;
3078 3079
  LogToBuffer(
      log_buffer,
I
Igor Canadi 已提交
3080
      "[%s] compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
3081 3082
      "MB in(%.1f, %.1f) out(%.1f), read-write-amplify(%.1f) "
      "write-amplify(%.1f) %s\n",
I
Igor Canadi 已提交
3083
      cfd->GetName().c_str(), cfd->current()->LevelSummary(&tmp),
M
Mark Callaghan 已提交
3084
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
3085 3086 3087 3088
          (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 已提交
3089
      stats.bytes_written / 1048576.0,
3090
      (stats.bytes_written + stats.bytes_readnp1 + stats.bytes_readn) /
3091 3092
          (double)stats.bytes_readn,
      stats.bytes_written / (double)stats.bytes_readn,
3093
      status.ToString().c_str());
M
Mark Callaghan 已提交
3094

J
jorlow@chromium.org 已提交
3095 3096 3097
  return status;
}

3098 3099
namespace {
struct IterState {
3100 3101
  IterState(DBImpl* db, port::Mutex* mu, SuperVersion* super_version)
      : db(db), mu(mu), super_version(super_version) {}
3102 3103

  DBImpl* db;
3104
  port::Mutex* mu;
3105
  SuperVersion* super_version;
3106 3107 3108 3109
};

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

3111
  if (state->super_version->Unref()) {
3112 3113
    DBImpl::DeletionState deletion_state;

3114 3115 3116 3117 3118 3119
    state->mu->Lock();
    state->super_version->Cleanup();
    state->db->FindObsoleteFiles(deletion_state, false, true);
    state->mu->Unlock();

    delete state->super_version;
3120 3121 3122
    if (deletion_state.HaveSomethingToDelete()) {
      state->db->PurgeObsoleteFiles(deletion_state);
    }
I
Igor Canadi 已提交
3123
  }
T
Tomislav Novak 已提交
3124

3125 3126
  delete state;
}
H
Hans Wennborg 已提交
3127
}  // namespace
3128

J
jorlow@chromium.org 已提交
3129
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
3130 3131
                                      ColumnFamilyData* cfd,
                                      SuperVersion* super_version) {
I
Igor Canadi 已提交
3132
  std::vector<Iterator*> iterator_list;
3133 3134
  // Collect iterator for mutable mem
  iterator_list.push_back(super_version->mem->NewIterator(options));
T
Tomislav Novak 已提交
3135
  // Collect all needed child iterators for immutable memtables
3136
  super_version->imm->AddIterators(options, &iterator_list);
T
Tomislav Novak 已提交
3137
  // Collect iterators for files in L0 - Ln
3138 3139
  super_version->current->AddIterators(options, storage_options_,
                                       &iterator_list);
I
Igor Canadi 已提交
3140 3141
  Iterator* internal_iter = NewMergingIterator(
      &cfd->internal_comparator(), &iterator_list[0], iterator_list.size());
3142 3143

  IterState* cleanup = new IterState(this, &mutex_, super_version);
3144
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
3145 3146 3147 3148

  return internal_iter;
}

3149 3150 3151 3152
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

J
jorlow@chromium.org 已提交
3153
Status DBImpl::Get(const ReadOptions& options,
3154
                   ColumnFamilyHandle* column_family, const Slice& key,
J
jorlow@chromium.org 已提交
3155
                   std::string* value) {
3156
  return GetImpl(options, column_family, key, value);
3157 3158
}

I
Igor Canadi 已提交
3159 3160 3161
// DeletionState gets created and destructed outside of the lock -- we
// use this convinently to:
// * malloc one SuperVersion() outside of the lock -- new_superversion
3162
// * delete SuperVersion()s outside of the lock -- superversions_to_free
I
Igor Canadi 已提交
3163 3164 3165 3166 3167 3168
//
// 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
3169 3170
void DBImpl::InstallSuperVersion(ColumnFamilyData* cfd,
                                 DeletionState& deletion_state) {
3171
  mutex_.AssertHeld();
I
Igor Canadi 已提交
3172 3173 3174 3175
  // if new_superversion == nullptr, it means somebody already used it
  SuperVersion* new_superversion =
    (deletion_state.new_superversion != nullptr) ?
    deletion_state.new_superversion : new SuperVersion();
3176 3177
  SuperVersion* old_superversion =
      cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
3178
  deletion_state.new_superversion = nullptr;
3179
  deletion_state.superversions_to_free.push_back(old_superversion);
I
Igor Canadi 已提交
3180 3181
}

3182
Status DBImpl::GetImpl(const ReadOptions& options,
3183 3184
                       ColumnFamilyHandle* column_family, const Slice& key,
                       std::string* value, bool* value_found) {
3185
  StopWatch sw(env_, options_.statistics.get(), DB_GET, false);
L
Lei Jin 已提交
3186
  PERF_TIMER_AUTO(get_snapshot_time);
3187

3188 3189 3190
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

3191
  SequenceNumber snapshot;
3192
  if (options.snapshot != nullptr) {
3193 3194 3195
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
3196
  }
3197

3198 3199
  // Acquire SuperVersion
  SuperVersion* sv = nullptr;
3200
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
3201
  if (LIKELY(options_.allow_thread_local)) {
3202
    sv = cfd->GetThreadLocalSuperVersion(&mutex_);
3203 3204
  } else {
    mutex_.Lock();
3205
    sv = cfd->GetSuperVersion()->Ref();
3206 3207
    mutex_.Unlock();
  }
I
Igor Canadi 已提交
3208

3209
  bool have_stat_update = false;
3210
  Version::GetStats stats;
3211

3212
  // Prepare to store a list of merge operations if merge occurs.
3213
  MergeContext merge_context;
3214

3215
  Status s;
3216
  // First look in the memtable, then in the immutable memtable (if any).
3217
  // s is both in/out. When in, s could either be OK or MergeInProgress.
3218
  // merge_operands will contain the sequence of merges in the latter case.
3219
  LookupKey lkey(key, snapshot);
L
Lei Jin 已提交
3220
  PERF_TIMER_STOP(get_snapshot_time);
I
Igor Canadi 已提交
3221
  if (sv->mem->Get(lkey, value, &s, merge_context, *cfd->options())) {
3222
    // Done
3223
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
3224
  } else if (sv->imm->Get(lkey, value, &s, merge_context, *cfd->options())) {
3225
    // Done
3226
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
3227
  } else {
L
Lei Jin 已提交
3228
    PERF_TIMER_START(get_from_output_files_time);
3229

3230
    sv->current->Get(options, lkey, value, &s, &merge_context, &stats,
3231
                     value_found);
3232
    have_stat_update = true;
L
Lei Jin 已提交
3233
    PERF_TIMER_STOP(get_from_output_files_time);
3234
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
3235
  }
3236

L
Lei Jin 已提交
3237
  PERF_TIMER_START(get_post_process_time);
3238

I
Igor Canadi 已提交
3239
  if (!cfd->options()->disable_seek_compaction && have_stat_update) {
I
Igor Canadi 已提交
3240
    mutex_.Lock();
3241
    if (sv->current->UpdateStats(stats)) {
I
Igor Canadi 已提交
3242 3243 3244
      MaybeScheduleFlushOrCompaction();
    }
    mutex_.Unlock();
3245 3246
  }

3247
  bool unref_sv = true;
3248
  if (LIKELY(options_.allow_thread_local)) {
3249
    unref_sv = !cfd->ReturnThreadLocalSuperVersion(sv);
3250 3251 3252 3253
  }

  if (unref_sv) {
    // Release SuperVersion
3254
    if (sv->Unref()) {
I
Igor Canadi 已提交
3255
      mutex_.Lock();
3256
      sv->Cleanup();
I
Igor Canadi 已提交
3257
      mutex_.Unlock();
3258
      delete sv;
3259
      RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_CLEANUPS);
I
Igor Canadi 已提交
3260
    }
3261
    RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_RELEASES);
3262
  }
3263

3264 3265
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
L
Lei Jin 已提交
3266
  PERF_TIMER_STOP(get_post_process_time);
3267
  return s;
J
jorlow@chromium.org 已提交
3268 3269
}

3270 3271
std::vector<Status> DBImpl::MultiGet(
    const ReadOptions& options,
3272
    const std::vector<ColumnFamilyHandle*>& column_family,
3273
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
3274

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

3278
  SequenceNumber snapshot;
3279

3280
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
3281
    ColumnFamilyData* cfd;
3282 3283 3284 3285 3286 3287 3288
    SuperVersion* super_version;
    Version::GetStats stats;
    bool have_stat_update = false;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
3289 3290 3291 3292 3293 3294
    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});
3295 3296 3297
    }
  }

3298
  mutex_.Lock();
3299 3300 3301 3302 3303
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }
3304
  for (auto mgd_iter : multiget_cf_data) {
3305 3306
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
3307
  }
3308
  mutex_.Unlock();
3309

3310 3311
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
3312

3313
  // Note: this always resizes the values array
3314 3315 3316
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
3317 3318

  // Keep track of bytes that we read for statistics-recording later
3319
  uint64_t bytes_read = 0;
L
Lei Jin 已提交
3320
  PERF_TIMER_STOP(get_snapshot_time);
3321 3322 3323 3324

  // 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.
3325
  // merge_operands will contain the sequence of merges in the latter case.
3326
  for (size_t i = 0; i < num_keys; ++i) {
3327
    merge_context.Clear();
3328
    Status& s = stat_list[i];
3329 3330 3331
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
3332 3333
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
3334 3335 3336
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
I
Igor Canadi 已提交
3337 3338
    auto cfd = mgd->cfd;
    if (super_version->mem->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3339
                                *cfd->options())) {
3340
      // Done
3341
    } else if (super_version->imm->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3342
                                       *cfd->options())) {
3343 3344
      // Done
    } else {
3345
      super_version->current->Get(options, lkey, value, &s, &merge_context,
3346
                                  &mgd->stats);
3347
      mgd->have_stat_update = true;
3348 3349 3350
    }

    if (s.ok()) {
3351
      bytes_read += value->size();
3352 3353 3354 3355
    }
  }

  // Post processing (decrement reference counts and record statistics)
L
Lei Jin 已提交
3356
  PERF_TIMER_START(get_post_process_time);
3357 3358 3359 3360 3361 3362
  autovector<SuperVersion*> superversions_to_delete;

  bool schedule_flush_or_compaction = false;
  mutex_.Lock();
  for (auto mgd_iter : multiget_cf_data) {
    auto mgd = mgd_iter.second;
I
Igor Canadi 已提交
3363 3364
    auto cfd = mgd->cfd;
    if (!cfd->options()->disable_seek_compaction && mgd->have_stat_update) {
3365 3366 3367
      if (mgd->super_version->current->UpdateStats(mgd->stats)) {
        schedule_flush_or_compaction = true;
      }
3368
    }
3369 3370 3371
    if (mgd->super_version->Unref()) {
      mgd->super_version->Cleanup();
      superversions_to_delete.push_back(mgd->super_version);
3372 3373
    }
  }
3374 3375 3376 3377 3378 3379 3380 3381 3382 3383
  if (schedule_flush_or_compaction) {
    MaybeScheduleFlushOrCompaction();
  }
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
3384
  }
3385

3386
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_CALLS);
3387 3388
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_KEYS_READ, num_keys);
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_BYTES_READ, bytes_read);
L
Lei Jin 已提交
3389
  PERF_TIMER_STOP(get_post_process_time);
3390

3391
  return stat_list;
3392 3393
}

3394
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& options,
3395
                                  const std::string& column_family_name,
3396
                                  ColumnFamilyHandle** handle) {
I
Igor Canadi 已提交
3397 3398 3399
  *handle = nullptr;
  MutexLock l(&mutex_);

I
Igor Canadi 已提交
3400 3401
  if (versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name) !=
      nullptr) {
I
Igor Canadi 已提交
3402 3403
    return Status::InvalidArgument("Column family already exists");
  }
3404
  VersionEdit edit;
3405
  edit.AddColumnFamily(column_family_name);
3406 3407
  uint32_t new_id = versions_->GetColumnFamilySet()->GetNextColumnFamilyID();
  edit.SetColumnFamily(new_id);
I
Igor Canadi 已提交
3408
  edit.SetLogNumber(logfile_number_);
I
Igor Canadi 已提交
3409
  edit.SetComparatorName(options.comparator->Name());
3410

I
Igor Canadi 已提交
3411 3412
  // LogAndApply will both write the creation in MANIFEST and create
  // ColumnFamilyData object
3413 3414
  Status s = versions_->LogAndApply(nullptr, &edit, &mutex_,
                                    db_directory_.get(), false, &options);
3415
  if (s.ok()) {
3416 3417 3418
    auto cfd =
        versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
    assert(cfd != nullptr);
3419
    delete cfd->InstallSuperVersion(new SuperVersion(), &mutex_);
3420
    *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
I
Igor Canadi 已提交
3421
    Log(options_.info_log, "Created column family [%s] (ID %u)",
3422 3423
        column_family_name.c_str(), (unsigned)cfd->GetID());
  } else {
I
Igor Canadi 已提交
3424
    Log(options_.info_log, "Creating column family [%s] FAILED -- %s",
3425 3426
        column_family_name.c_str(), s.ToString().c_str());
  }
3427
  return s;
3428 3429
}

3430 3431 3432 3433
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
3434 3435
    return Status::InvalidArgument("Can't drop default column family");
  }
3436

I
Igor Canadi 已提交
3437 3438
  VersionEdit edit;
  edit.DropColumnFamily();
3439 3440
  edit.SetColumnFamily(cfd->GetID());

3441
  Status s;
3442 3443 3444 3445 3446 3447 3448 3449
  {
    MutexLock l(&mutex_);
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
      s = versions_->LogAndApply(cfd, &edit, &mutex_);
    }
3450
  }
3451

3452
  if (s.ok()) {
I
Igor Canadi 已提交
3453
    assert(cfd->IsDropped());
3454
    Log(options_.info_log, "Dropped column family with id %u\n", cfd->GetID());
3455
    // Flush the memtables. This will make all WAL files referencing dropped
I
Igor Canadi 已提交
3456
    // column family to be obsolete. They will be deleted once user deletes
3457 3458
    // column family handle
    Write(WriteOptions(), nullptr);  // ignore error
3459 3460 3461 3462 3463
  } else {
    Log(options_.info_log, "Dropping column family with id %u FAILED -- %s\n",
        cfd->GetID(), s.ToString().c_str());
  }

3464
  return s;
3465 3466
}

3467
bool DBImpl::KeyMayExist(const ReadOptions& options,
3468 3469
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
3470
  if (value_found != nullptr) {
K
Kai Liu 已提交
3471 3472
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
3473
  }
3474 3475
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
3476
  auto s = GetImpl(roptions, column_family, key, value, value_found);
K
Kai Liu 已提交
3477 3478 3479 3480 3481

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

3484
Iterator* DBImpl::NewIterator(const ReadOptions& options,
3485 3486 3487
                              ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
T
Tomislav Novak 已提交
3488

3489
  Iterator* iter;
T
Tomislav Novak 已提交
3490
  if (options.tailing) {
I
Igor Canadi 已提交
3491 3492 3493 3494
#ifdef ROCKSDB_LITE
    // not supported in lite version
    return nullptr;
#else
L
Lei Jin 已提交
3495
    iter = new TailingIterator(env_, this, options, cfd);
I
Igor Canadi 已提交
3496
#endif
T
Tomislav Novak 已提交
3497
  } else {
3498 3499 3500 3501 3502
    SequenceNumber latest_snapshot = versions_->LastSequence();
    SuperVersion* sv = nullptr;
    sv = cfd->GetReferencedSuperVersion(&mutex_);

    iter = NewInternalIterator(options, cfd, sv);
T
Tomislav Novak 已提交
3503

I
Igor Canadi 已提交
3504 3505 3506 3507
    auto snapshot =
        options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
            : latest_snapshot;
L
Lei Jin 已提交
3508
    iter = NewDBIterator(env_, *cfd->options(),
I
Igor Canadi 已提交
3509
                         cfd->user_comparator(), iter, snapshot);
T
Tomislav Novak 已提交
3510 3511
  }

T
Tyler Harter 已提交
3512
  return iter;
J
jorlow@chromium.org 已提交
3513 3514
}

3515 3516
Status DBImpl::NewIterators(
    const ReadOptions& options,
I
Igor Canadi 已提交
3517
    const std::vector<ColumnFamilyHandle*>& column_families,
3518
    std::vector<Iterator*>* iterators) {
I
Igor Canadi 已提交
3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535
  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 已提交
3536 3537 3538 3539
#ifdef ROCKSDB_LITE
    return Status::InvalidArgument(
        "Tailing interator not supported in RocksDB lite");
#else
I
Igor Canadi 已提交
3540 3541
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
L
Lei Jin 已提交
3542
      iterators->push_back(new TailingIterator(env_, this, options, cfd));
I
Igor Canadi 已提交
3543
    }
I
Igor Canadi 已提交
3544
#endif
I
Igor Canadi 已提交
3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555
  } 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 已提交
3556
      iter = NewDBIterator(env_, *cfd->options(),
I
Igor Canadi 已提交
3557 3558 3559 3560 3561 3562
                           cfd->user_comparator(), iter, snapshot);
      iterators->push_back(iter);
    }
  }

  return Status::OK();
3563 3564
}

3565 3566 3567 3568 3569 3570 3571 3572 3573
bool DBImpl::IsSnapshotSupported() const {
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    if (!cfd->mem()->IsSnapshotSupported()) {
      return false;
    }
  }
  return true;
}

J
jorlow@chromium.org 已提交
3574
const Snapshot* DBImpl::GetSnapshot() {
3575 3576
  // returns null if the underlying memtable does not support snapshot.
  if (!IsSnapshotSupported()) return nullptr;
J
jorlow@chromium.org 已提交
3577
  MutexLock l(&mutex_);
3578
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
3579 3580 3581 3582
}

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

// Convenience methods
3587 3588
Status DBImpl::Put(const WriteOptions& o, ColumnFamilyHandle* column_family,
                   const Slice& key, const Slice& val) {
3589
  return DB::Put(o, column_family, key, val);
J
jorlow@chromium.org 已提交
3590 3591
}

3592 3593 3594 3595
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) {
3596 3597
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
3598
    return DB::Merge(o, column_family, key, val);
3599 3600 3601
  }
}

3602
Status DBImpl::Delete(const WriteOptions& options,
3603
                      ColumnFamilyHandle* column_family, const Slice& key) {
3604
  return DB::Delete(options, column_family, key);
J
jorlow@chromium.org 已提交
3605 3606
}

3607
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
L
Lei Jin 已提交
3608
  PERF_TIMER_AUTO(write_pre_and_post_process_time);
3609 3610 3611
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
3612
  w.disableWAL = options.disableWAL;
3613
  w.done = false;
3614

3615
  StopWatch sw(env_, options_.statistics.get(), DB_WRITE, false);
I
Igor Canadi 已提交
3616
  mutex_.Lock();
3617 3618 3619 3620
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
3621 3622 3623 3624 3625

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

3626
  if (w.done) {
I
Igor Canadi 已提交
3627
    mutex_.Unlock();
3628
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
3629
    return w.status;
3630 3631
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
3632 3633
  }

3634
  Status status;
3635
  // refcounting cfd in iteration
3636
  bool dead_cfd = false;
3637 3638
  autovector<SuperVersion*> superversions_to_free;
  autovector<log::Writer*> logs_to_free;
3639
  for (auto cfd : *versions_->GetColumnFamilySet()) {
3640
    cfd->Ref();
3641
    // May temporarily unlock and wait.
3642 3643
    status = MakeRoomForWrite(cfd, my_batch == nullptr, &superversions_to_free,
                              &logs_to_free);
3644
    if (cfd->Unref()) {
3645
      dead_cfd = true;
3646
    }
3647 3648 3649 3650
    if (!status.ok()) {
      break;
    }
  }
3651 3652
  if (dead_cfd) {
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
3653
  }
3654

D
dgrogan@chromium.org 已提交
3655
  uint64_t last_sequence = versions_->LastSequence();
3656
  Writer* last_writer = &w;
3657
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
3658 3659
    autovector<WriteBatch*> write_batch_group;
    BuildBatchGroup(&last_writer, &write_batch_group);
3660

3661 3662 3663
    // 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
3664
    // into memtables
3665
    {
3666
      mutex_.Unlock();
3667 3668 3669 3670 3671 3672 3673 3674 3675 3676
      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]);
        }
      }

3677 3678 3679 3680 3681
      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
3682 3683 3684
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3685 3686
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3687 3688
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3689
      }
L
Lei Jin 已提交
3690
      PERF_TIMER_STOP(write_pre_and_post_process_time);
3691 3692

      if (!options.disableWAL) {
L
Lei Jin 已提交
3693
        PERF_TIMER_START(write_wal_time);
3694 3695
        Slice log_entry = WriteBatchInternal::Contents(updates);
        status = log_->AddRecord(log_entry);
3696
        log_empty_ = false;
3697 3698
        RecordTick(options_.statistics.get(), WAL_FILE_SYNCED, 1);
        RecordTick(options_.statistics.get(), WAL_FILE_BYTES, log_entry.size());
H
heyongqiang 已提交
3699
        if (status.ok() && options.sync) {
3700
          if (options_.use_fsync) {
3701
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3702
            status = log_->file()->Fsync();
3703
          } else {
3704
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3705
            status = log_->file()->Sync();
3706
          }
H
heyongqiang 已提交
3707
        }
L
Lei Jin 已提交
3708
        PERF_TIMER_STOP(write_wal_time);
3709 3710
      }
      if (status.ok()) {
L
Lei Jin 已提交
3711
        PERF_TIMER_START(write_memtable_time);
3712
        status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
3713
            updates, column_family_memtables_.get(), false, 0, this, false);
L
Lei Jin 已提交
3714
        PERF_TIMER_STOP(write_memtable_time);
3715

3716
        if (!status.ok()) {
3717 3718 3719
          // Iteration failed (either in-memory writebatch corruption (very
          // bad), or the client specified invalid column family). Return
          // failure.
3720 3721 3722
          // 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.
3723
          return status;
3724
        }
I
Igor Canadi 已提交
3725 3726
        SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                       last_sequence);
3727
      }
L
Lei Jin 已提交
3728
      PERF_TIMER_START(write_pre_and_post_process_time);
3729
      if (updates == &tmp_batch_) tmp_batch_.Clear();
3730 3731
      mutex_.Lock();
      if (status.ok()) {
3732
        versions_->SetLastSequence(last_sequence);
3733
      }
J
jorlow@chromium.org 已提交
3734 3735
    }
  }
I
Igor Canadi 已提交
3736 3737 3738
  if (options_.paranoid_checks && !status.ok() && bg_error_.ok()) {
    bg_error_ = status; // stop compaction & fail any further writes
  }
3739

3740 3741 3742 3743 3744 3745 3746
  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
3747
    }
3748 3749
    if (ready == last_writer) break;
  }
3750

3751 3752 3753
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
3754
  }
I
Igor Canadi 已提交
3755
  mutex_.Unlock();
3756 3757 3758 3759 3760 3761 3762 3763

  for (auto& sv : superversions_to_free) {
    delete sv;
  }
  for (auto& log : logs_to_free) {
    delete log;
  }

L
Lei Jin 已提交
3764
  PERF_TIMER_STOP(write_pre_and_post_process_time);
J
jorlow@chromium.org 已提交
3765 3766 3767
  return status;
}

3768
// REQUIRES: Writer list must be non-empty
3769
// REQUIRES: First writer must have a non-nullptr batch
3770 3771
void DBImpl::BuildBatchGroup(Writer** last_writer,
                             autovector<WriteBatch*>* write_batch_group) {
3772 3773
  assert(!writers_.empty());
  Writer* first = writers_.front();
3774
  assert(first->batch != nullptr);
3775 3776

  size_t size = WriteBatchInternal::ByteSize(first->batch);
3777
  write_batch_group->push_back(first->batch);
3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796

  // 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 已提交
3797 3798 3799 3800 3801 3802
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

3803
    if (w->batch != nullptr) {
3804 3805 3806 3807 3808 3809
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

3810
      write_batch_group->push_back(w->batch);
3811 3812 3813 3814 3815
    }
    *last_writer = w;
  }
}

3816 3817 3818
// 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 已提交
3819 3820 3821 3822
// if n < bottom, return 0;
// if n >= top, return 1000;
// otherwise, let r = (n - bottom) /
//                    (top - bottom)
3823 3824 3825 3826
//  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 已提交
3827
uint64_t DBImpl::SlowdownAmount(int n, double bottom, double top) {
3828
  uint64_t delay;
J
Jim Paton 已提交
3829
  if (n >= top) {
3830 3831
    delay = 1000;
  }
J
Jim Paton 已提交
3832
  else if (n < bottom) {
3833 3834 3835 3836
    delay = 0;
  }
  else {
    // If we are here, we know that:
J
Jim Paton 已提交
3837
    //   level0_start_slowdown <= n < level0_slowdown
3838
    // since the previous two conditions are false.
M
Mark Callaghan 已提交
3839 3840
    double how_much =
      (double) (n - bottom) /
J
Jim Paton 已提交
3841
              (top - bottom);
M
Mark Callaghan 已提交
3842
    delay = std::max(how_much * how_much * 1000, 100.0);
3843 3844 3845 3846 3847
  }
  assert(delay <= 1000);
  return delay;
}

3848
// REQUIRES: mutex_ is held
3849
// REQUIRES: this thread is currently at the front of the writer queue
3850 3851 3852 3853
Status DBImpl::MakeRoomForWrite(
    ColumnFamilyData* cfd, bool force,
    autovector<SuperVersion*>* superversions_to_free,
    autovector<log::Writer*>* logs_to_free) {
3854
  mutex_.AssertHeld();
3855
  assert(!writers_.empty());
3856
  bool allow_delay = !force;
J
Jim Paton 已提交
3857 3858
  bool allow_hard_rate_limit_delay = !force;
  bool allow_soft_rate_limit_delay = !force;
3859
  uint64_t rate_limit_delay_millis = 0;
3860
  Status s;
3861
  double score;
3862

3863 3864 3865 3866 3867
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
3868
    } else if (allow_delay && cfd->NeedSlowdownForNumLevel0Files()) {
3869 3870 3871
      // 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
3872
      // individual write by 0-1ms to reduce latency variance.  Also,
3873 3874
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
3875
      uint64_t slowdown =
3876 3877 3878
          SlowdownAmount(cfd->current()->NumLevelFiles(0),
                         cfd->options()->level0_slowdown_writes_trigger,
                         cfd->options()->level0_stop_writes_trigger);
3879
      mutex_.Unlock();
3880
      uint64_t delayed;
J
Jim Paton 已提交
3881
      {
3882
        StopWatch sw(env_, options_.statistics.get(), STALL_L0_SLOWDOWN_COUNT);
3883
        env_->SleepForMicroseconds(slowdown);
3884
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3885
      }
3886
      RecordTick(options_.statistics.get(), STALL_L0_SLOWDOWN_MICROS, delayed);
3887 3888
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_SLOWDOWN,
                                              delayed);
3889 3890
      allow_delay = false;  // Do not delay a single write more than once
      mutex_.Lock();
3891
      delayed_writes_++;
3892
    } else if (!force && !cfd->mem()->ShouldFlush()) {
3893
      // There is room in current memtable
3894 3895 3896
      if (allow_delay) {
        DelayLoggingAndReset();
      }
3897
      break;
3898 3899
    } else if (cfd->imm()->size() ==
               cfd->options()->max_write_buffer_number - 1) {
3900
      // We have filled up the current memtable, but the previous
3901
      // ones are still being flushed, so we wait.
3902
      DelayLoggingAndReset();
I
Igor Canadi 已提交
3903 3904
      Log(options_.info_log, "[%s] wait for memtable flush...\n",
          cfd->GetName().c_str());
3905
      MaybeScheduleFlushOrCompaction();
3906
      uint64_t stall;
J
Jim Paton 已提交
3907
      {
3908
        StopWatch sw(env_, options_.statistics.get(),
3909
                     STALL_MEMTABLE_COMPACTION_COUNT);
J
Jim Paton 已提交
3910
        bg_cv_.Wait();
3911
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3912
      }
3913 3914
      RecordTick(options_.statistics.get(),
                 STALL_MEMTABLE_COMPACTION_MICROS, stall);
3915 3916
      cfd->internal_stats()->RecordWriteStall(
          InternalStats::MEMTABLE_COMPACTION, stall);
3917 3918
    } else if (cfd->current()->NumLevelFiles(0) >=
               cfd->options()->level0_stop_writes_trigger) {
3919
      // There are too many level-0 files.
3920
      DelayLoggingAndReset();
I
Igor Canadi 已提交
3921 3922
      Log(options_.info_log, "[%s] wait for fewer level0 files...\n",
          cfd->GetName().c_str());
3923
      uint64_t stall;
J
Jim Paton 已提交
3924
      {
3925 3926
        StopWatch sw(env_, options_.statistics.get(),
                     STALL_L0_NUM_FILES_COUNT);
J
Jim Paton 已提交
3927
        bg_cv_.Wait();
3928
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3929
      }
3930
      RecordTick(options_.statistics.get(), STALL_L0_NUM_FILES_MICROS, stall);
3931 3932
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_NUM_FILES,
                                              stall);
I
Igor Canadi 已提交
3933 3934
    } else if (allow_hard_rate_limit_delay &&
               cfd->options()->hard_rate_limit > 1.0 &&
3935 3936
               (score = cfd->current()->MaxCompactionScore()) >
                   cfd->options()->hard_rate_limit) {
3937
      // Delay a write when the compaction score for any level is too large.
3938
      int max_level = cfd->current()->MaxCompactionScoreLevel();
3939
      mutex_.Unlock();
3940
      uint64_t delayed;
J
Jim Paton 已提交
3941
      {
3942 3943
        StopWatch sw(env_, options_.statistics.get(),
                     HARD_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3944
        env_->SleepForMicroseconds(1000);
3945
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3946
      }
3947
      cfd->internal_stats()->RecordLevelNSlowdown(max_level, delayed);
3948
      // Make sure the following value doesn't round to zero.
3949 3950
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
3951 3952
      RecordTick(options_.statistics.get(),
                 RATE_LIMIT_DELAY_MILLIS, rate_limit);
3953
      if (cfd->options()->rate_limit_delay_max_milliseconds > 0 &&
J
Jim Paton 已提交
3954
          rate_limit_delay_millis >=
3955
              (unsigned)cfd->options()->rate_limit_delay_max_milliseconds) {
J
Jim Paton 已提交
3956
        allow_hard_rate_limit_delay = false;
3957
      }
3958
      mutex_.Lock();
3959 3960 3961 3962
    } else if (allow_soft_rate_limit_delay &&
               cfd->options()->soft_rate_limit > 0.0 &&
               (score = cfd->current()->MaxCompactionScore()) >
                   cfd->options()->soft_rate_limit) {
J
Jim Paton 已提交
3963 3964 3965
      // Delay a write when the compaction score for any level is too large.
      // TODO: add statistics
      mutex_.Unlock();
J
Jim Paton 已提交
3966
      {
3967 3968
        StopWatch sw(env_, options_.statistics.get(),
                     SOFT_RATE_LIMIT_DELAY_COUNT);
3969 3970 3971
        env_->SleepForMicroseconds(
            SlowdownAmount(score, cfd->options()->soft_rate_limit,
                           cfd->options()->hard_rate_limit));
J
Jim Paton 已提交
3972 3973
        rate_limit_delay_millis += sw.ElapsedMicros();
      }
J
Jim Paton 已提交
3974 3975
      allow_soft_rate_limit_delay = false;
      mutex_.Lock();
3976

3977
    } else {
3978
      unique_ptr<WritableFile> lfile;
S
sdong 已提交
3979
      log::Writer* new_log = nullptr;
K
kailiu 已提交
3980
      MemTable* new_mem = nullptr;
3981

3982
      // Attempt to switch to a new memtable and trigger flush of old.
3983
      // Do this without holding the dbmutex lock.
3984
      assert(versions_->PrevLogNumber() == 0);
I
Igor Canadi 已提交
3985 3986 3987
      bool creating_new_log = !log_empty_;
      uint64_t new_log_number =
          creating_new_log ? versions_->NewFileNumber() : logfile_number_;
I
Igor Canadi 已提交
3988
      SuperVersion* new_superversion = nullptr;
3989 3990 3991
      mutex_.Unlock();
      {
        DelayLoggingAndReset();
I
Igor Canadi 已提交
3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004
        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));
          }
        }

4005
        if (s.ok()) {
4006
          new_mem = new MemTable(cfd->internal_comparator(), *cfd->options());
4007
          new_superversion = new SuperVersion();
4008 4009 4010
        }
      }
      mutex_.Lock();
4011
      if (!s.ok()) {
I
Igor Canadi 已提交
4012 4013
        // how do we fail if we're not creating new log?
        assert(creating_new_log);
H
heyongqiang 已提交
4014
        // Avoid chewing through file number space in a tight loop.
4015
        versions_->ReuseFileNumber(new_log_number);
4016
        assert(!new_mem);
4017
        assert(!new_log);
4018 4019
        break;
      }
I
Igor Canadi 已提交
4020 4021 4022
      if (creating_new_log) {
        logfile_number_ = new_log_number;
        assert(new_log != nullptr);
4023
        logs_to_free->push_back(log_.release());
I
Igor Canadi 已提交
4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037
        log_.reset(new_log);
        log_empty_ = true;
        alive_log_files_.push_back(logfile_number_);
        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_);
          }
        }
      }
4038 4039
      cfd->mem()->SetNextLogNumber(logfile_number_);
      cfd->imm()->Add(cfd->mem());
4040
      if (force) {
4041
        cfd->imm()->FlushRequested();
4042
      }
4043 4044
      new_mem->Ref();
      cfd->SetMemtable(new_mem);
I
Igor Canadi 已提交
4045 4046
      Log(options_.info_log, "[%s] New memtable created with log file: #%lu\n",
          cfd->GetName().c_str(), (unsigned long)logfile_number_);
4047
      force = false;  // Do not force another compaction if have room
4048
      MaybeScheduleFlushOrCompaction();
4049 4050
      superversions_to_free->push_back(
          cfd->InstallSuperVersion(new_superversion, &mutex_));
4051 4052 4053 4054 4055
    }
  }
  return s;
}

I
Igor Canadi 已提交
4056
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
4057 4058 4059 4060 4061
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

4062 4063
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
4064
  auto version = cfd->current();
4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}
I
Igor Canadi 已提交
4077
#endif  // ROCKSDB_LITE
4078

I
Igor Canadi 已提交
4079 4080 4081 4082
const std::string& DBImpl::GetName() const {
  return dbname_;
}

4083 4084 4085 4086
Env* DBImpl::GetEnv() const {
  return env_;
}

4087 4088
const Options& DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
I
Igor Canadi 已提交
4089
  return *cfh->cfd()->options();
I
Igor Canadi 已提交
4090 4091
}

4092
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
4093
                         const Slice& property, std::string* value) {
4094
  value->clear();
4095 4096
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
4097
  DBPropertyType property_type = GetPropertyType(property);
J
jorlow@chromium.org 已提交
4098
  MutexLock l(&mutex_);
4099 4100
  return cfd->internal_stats()->GetProperty(property_type, property, value,
                                            cfd);
J
jorlow@chromium.org 已提交
4101 4102
}

4103
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
4104
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
4105 4106
  // TODO(opt): better implementation
  Version* v;
4107 4108
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
J
jorlow@chromium.org 已提交
4109 4110
  {
    MutexLock l(&mutex_);
4111
    v = cfd->current();
4112
    v->Ref();
J
jorlow@chromium.org 已提交
4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129
  }

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

4130 4131 4132 4133 4134 4135 4136
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

I
Igor Canadi 已提交
4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164
#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();
}

4165 4166 4167
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
4168 4169 4170 4171
  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());
4172 4173 4174
    return Status::InvalidArgument("Invalid file name");
  }

4175 4176 4177 4178
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
4179 4180
      Log(options_.info_log, "DeleteFile %s failed - not archived log.\n",
          name.c_str());
4181 4182 4183 4184
      return Status::NotSupported("Delete only supported for archived logs");
    }
    status = env_->DeleteFile(options_.wal_dir + "/" + name.c_str());
    if (!status.ok()) {
4185 4186
      Log(options_.info_log, "DeleteFile %s failed -- %s.\n",
          name.c_str(), status.ToString().c_str());
4187 4188 4189 4190
    }
    return status;
  }

4191
  int level;
I
Igor Canadi 已提交
4192
  FileMetaData* metadata;
4193
  ColumnFamilyData* cfd;
4194
  VersionEdit edit;
K
Kai Liu 已提交
4195
  DeletionState deletion_state(true);
D
Dhruba Borthakur 已提交
4196 4197
  {
    MutexLock l(&mutex_);
4198
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
4199
    if (!status.ok()) {
4200 4201
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
4202 4203
      return Status::InvalidArgument("File not found");
    }
I
Igor Canadi 已提交
4204
    assert((level > 0) && (level < cfd->NumberLevels()));
4205

D
Dhruba Borthakur 已提交
4206
    // If the file is being compacted no need to delete.
4207
    if (metadata->being_compacted) {
4208
      Log(options_.info_log,
4209
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
4210
      return Status::OK();
4211 4212
    }

D
Dhruba Borthakur 已提交
4213 4214 4215
    // 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 已提交
4216
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
4217
      if (cfd->current()->NumLevelFiles(i) != 0) {
D
Dhruba Borthakur 已提交
4218
        Log(options_.info_log,
4219
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
4220 4221 4222 4223
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
4224
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
I
Igor Canadi 已提交
4225
    if (status.ok()) {
4226
      InstallSuperVersion(cfd, deletion_state);
I
Igor Canadi 已提交
4227
    }
I
Igor Canadi 已提交
4228
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
4229
  } // lock released here
I
Igor Canadi 已提交
4230
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
4231
  // remove files outside the db-lock
4232 4233 4234
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
4235 4236 4237 4238 4239 4240
  {
    MutexLock l(&mutex_);
    // schedule flush if file deletion means we freed the space for flushes to
    // continue
    MaybeScheduleFlushOrCompaction();
  }
4241 4242 4243
  return status;
}

4244
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
4245
  MutexLock l(&mutex_);
4246
  versions_->GetLiveFilesMetaData(metadata);
4247
}
I
Igor Canadi 已提交
4248
#endif  // ROCKSDB_LITE
4249

I
Igor Canadi 已提交
4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274 4275 4276
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

  std::string corruption_messages;
  for (const auto& md : metadata) {
    std::string file_path = dbname_ + md.name;
    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) {
      corruption_messages += "Sst file size mismatch: " + md.name +
                             ". 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);
  }
}

4277 4278 4279 4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299 4300 4301 4302 4303
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 已提交
4304 4305
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
4306
Status DB::Put(const WriteOptions& opt, ColumnFamilyHandle* column_family,
4307
               const Slice& key, const Slice& value) {
4308 4309 4310 4311
  // 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);
4312
  batch.Put(column_family, key, value);
J
jorlow@chromium.org 已提交
4313 4314 4315
  return Write(opt, &batch);
}

4316 4317
Status DB::Delete(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                  const Slice& key) {
J
jorlow@chromium.org 已提交
4318
  WriteBatch batch;
4319
  batch.Delete(column_family, key);
J
jorlow@chromium.org 已提交
4320 4321 4322
  return Write(opt, &batch);
}

4323 4324
Status DB::Merge(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                 const Slice& key, const Slice& value) {
4325
  WriteBatch batch;
4326
  batch.Merge(column_family, key, value);
4327 4328 4329
  return Write(opt, &batch);
}

4330 4331
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
4332
                              const std::string& column_family_name,
4333
                              ColumnFamilyHandle** handle) {
4334
  return Status::NotSupported("");
4335
}
4336
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
4337
  return Status::NotSupported("");
4338 4339
}

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

J
Jim Paton 已提交
4342
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
4343 4344 4345 4346
  DBOptions db_options(options);
  ColumnFamilyOptions cf_options(options);
  std::vector<ColumnFamilyDescriptor> column_families;
  column_families.push_back(
4347
      ColumnFamilyDescriptor(kDefaultColumnFamilyName, cf_options));
4348 4349 4350 4351 4352 4353 4354 4355 4356
  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;
4357 4358
}

4359 4360
Status DB::Open(const DBOptions& db_options, const std::string& dbname,
                const std::vector<ColumnFamilyDescriptor>& column_families,
4361
                std::vector<ColumnFamilyHandle*>* handles, DB** dbptr) {
4362
  *dbptr = nullptr;
4363
  handles->clear();
J
jorlow@chromium.org 已提交
4364

I
Igor Canadi 已提交
4365 4366 4367 4368 4369 4370 4371 4372
  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");
    }
4373
  }
4374

I
Igor Canadi 已提交
4375
  DBImpl* impl = new DBImpl(db_options, dbname);
4376 4377 4378 4379 4380 4381 4382
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
4383 4384 4385 4386
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
4387
  impl->mutex_.Lock();
4388 4389
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
4390
  if (s.ok()) {
4391
    uint64_t new_log_number = impl->versions_->NewFileNumber();
4392
    unique_ptr<WritableFile> lfile;
4393
    EnvOptions soptions(db_options);
4394
    s = impl->options_.env->NewWritableFile(
I
Igor Canadi 已提交
4395
        LogFileName(impl->options_.wal_dir, new_log_number), &lfile,
I
Igor Canadi 已提交
4396
        impl->options_.env->OptimizeForLogWrite(soptions));
J
jorlow@chromium.org 已提交
4397
    if (s.ok()) {
4398
      lfile->SetPreallocationBlockSize(1.1 * max_write_buffer_size);
4399
      impl->logfile_number_ = new_log_number;
4400
      impl->log_.reset(new log::Writer(std::move(lfile)));
I
Igor Canadi 已提交
4401

4402 4403
      // set column family handles
      for (auto cf : column_families) {
I
Igor Canadi 已提交
4404 4405 4406
        auto cfd =
            impl->versions_->GetColumnFamilySet()->GetColumnFamily(cf.name);
        if (cfd == nullptr) {
I
Igor Canadi 已提交
4407 4408 4409
          s = Status::InvalidArgument("Column family not found: ", cf.name);
          break;
        }
4410 4411
        handles->push_back(
            new ColumnFamilyHandleImpl(cfd, impl, &impl->mutex_));
4412
      }
I
Igor Canadi 已提交
4413 4414
    }
    if (s.ok()) {
4415
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
4416
        delete cfd->InstallSuperVersion(new SuperVersion(), &impl->mutex_);
4417
      }
4418
      impl->alive_log_files_.push_back(impl->logfile_number_);
J
jorlow@chromium.org 已提交
4419
      impl->DeleteObsoleteFiles();
4420
      impl->MaybeScheduleFlushOrCompaction();
4421
      impl->MaybeScheduleLogDBDeployStats();
4422
      s = impl->db_directory_->Fsync();
J
jorlow@chromium.org 已提交
4423 4424
    }
  }
4425

I
Igor Canadi 已提交
4426 4427 4428 4429 4430 4431 4432 4433 4434 4435 4436 4437 4438
  if (s.ok()) {
    for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
      if (cfd->options()->compaction_style == kCompactionStyleUniversal) {
        Version* current = cfd->current();
        for (int i = 1; i < current->NumberLevels(); ++i) {
          int num_files = current->NumLevelFiles(i);
          if (num_files > 0) {
            s = Status::InvalidArgument("Not all files are at level 0. Cannot "
                "open with universal compaction style.");
            break;
          }
        }
      }
4439 4440 4441 4442 4443 4444
      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 已提交
4445
      if (!s.ok()) {
4446 4447 4448 4449 4450
        break;
      }
    }
  }

4451 4452
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4453
  if (s.ok()) {
4454
    impl->opened_successfully_ = true;
J
jorlow@chromium.org 已提交
4455 4456
    *dbptr = impl;
  } else {
4457 4458 4459
    for (auto h : *handles) {
      delete h;
    }
4460
    handles->clear();
J
jorlow@chromium.org 已提交
4461 4462 4463 4464 4465
    delete impl;
  }
  return s;
}

4466 4467 4468
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4469
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4470 4471
}

4472 4473 4474
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4475
Status DestroyDB(const std::string& dbname, const Options& options) {
4476 4477 4478 4479 4480
  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 已提交
4481
  std::vector<std::string> filenames;
4482 4483
  std::vector<std::string> archiveFiles;

4484
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
4485 4486
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
4487 4488 4489 4490 4491 4492 4493

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

J
jorlow@chromium.org 已提交
4495 4496 4497 4498 4499
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
4500 4501
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4502 4503 4504
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
4505
    for (size_t i = 0; i < filenames.size(); i++) {
4506
      if (ParseFileName(filenames[i], &number, &type) &&
4507
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4508 4509 4510
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
4511 4512
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
4513 4514 4515
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
4516 4517 4518 4519 4520
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4521

4522
    env->GetChildren(archivedir, &archiveFiles);
4523 4524
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4525 4526
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4527
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4528 4529 4530 4531 4532
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4533
    // ignore case where no archival directory is present.
4534
    env->DeleteDir(archivedir);
4535

J
jorlow@chromium.org 已提交
4536
    env->UnlockFile(lock);  // Ignore error since state is already gone
4537
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4538
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4539
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4540 4541 4542 4543
  }
  return result;
}

4544 4545
//
// A global method that can dump out the build version
K
kailiu 已提交
4546
void DumpLeveldbBuildVersion(Logger * log) {
I
Igor Canadi 已提交
4547 4548
#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
4549
  Log(log, "Git sha %s", rocksdb_build_git_sha);
4550
  Log(log, "Compile time %s %s",
4551
      rocksdb_build_compile_time, rocksdb_build_compile_date);
I
Igor Canadi 已提交
4552
#endif
4553 4554
}

4555
}  // namespace rocksdb