db_impl.cc 134.1 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 12
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file. See the AUTHORS file for names of contributors.

#include "db/db_impl.h"

#include <algorithm>
13 14
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
15
#include <set>
16
#include <stdexcept>
17 18
#include <stdint.h>
#include <string>
19
#include <unordered_set>
20
#include <vector>
21

J
jorlow@chromium.org 已提交
22
#include "db/builder.h"
23
#include "db/db_iter.h"
K
kailiu 已提交
24
#include "db/dbformat.h"
J
jorlow@chromium.org 已提交
25 26 27 28
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
K
kailiu 已提交
29
#include "db/memtable_list.h"
30
#include "db/merge_context.h"
31
#include "db/merge_helper.h"
T
Tyler Harter 已提交
32
#include "db/prefix_filter_iterator.h"
J
jorlow@chromium.org 已提交
33
#include "db/table_cache.h"
K
kailiu 已提交
34
#include "db/table_properties_collector.h"
35
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
36 37
#include "db/version_set.h"
#include "db/write_batch_internal.h"
38
#include "port/port.h"
39 40 41 42 43 44
#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 已提交
45
#include "rocksdb/table.h"
J
jorlow@chromium.org 已提交
46
#include "table/block.h"
47
#include "table/block_based_table_factory.h"
J
jorlow@chromium.org 已提交
48 49
#include "table/merger.h"
#include "table/two_level_iterator.h"
K
Kai Liu 已提交
50
#include "util/autovector.h"
51 52
#include "util/auto_roll_logger.h"
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
53
#include "util/coding.h"
I
Igor Canadi 已提交
54
#include "util/hash_skiplist_rep.h"
J
jorlow@chromium.org 已提交
55 56
#include "util/logging.h"
#include "util/mutexlock.h"
57
#include "util/perf_context_imp.h"
58
#include "util/stop_watch.h"
J
jorlow@chromium.org 已提交
59

60
namespace rocksdb {
J
jorlow@chromium.org 已提交
61

K
kailiu 已提交
62
void DumpLeveldbBuildVersion(Logger * log);
63

64 65 66 67 68
// Information kept for every waiting writer
struct DBImpl::Writer {
  Status status;
  WriteBatch* batch;
  bool sync;
H
heyongqiang 已提交
69
  bool disableWAL;
70 71 72 73 74 75
  bool done;
  port::CondVar cv;

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

J
jorlow@chromium.org 已提交
76 77 78
struct DBImpl::CompactionState {
  Compaction* const compaction;

79 80 81 82 83
  // 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 已提交
84 85 86 87 88 89

  // Files produced by compaction
  struct Output {
    uint64_t number;
    uint64_t file_size;
    InternalKey smallest, largest;
90
    SequenceNumber smallest_seqno, largest_seqno;
J
jorlow@chromium.org 已提交
91 92
  };
  std::vector<Output> outputs;
93
  std::list<uint64_t> allocated_file_numbers;
J
jorlow@chromium.org 已提交
94 95

  // State kept for output being generated
96 97
  unique_ptr<WritableFile> outfile;
  unique_ptr<TableBuilder> builder;
J
jorlow@chromium.org 已提交
98 99 100 101 102 103 104 105 106

  uint64_t total_bytes;

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

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
107 108 109 110 111 112 113

  // Create a client visible context of this compaction
  CompactionFilter::Context GetFilterContext() {
    CompactionFilter::Context context;
    context.is_full_compaction = compaction->IsFullCompaction();
    return context;
  }
J
jorlow@chromium.org 已提交
114 115 116
};

// Fix user-supplied options to be reasonable
117
template <class T, class V>
J
jorlow@chromium.org 已提交
118
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
D
dgrogan@chromium.org 已提交
119 120
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
J
jorlow@chromium.org 已提交
121 122 123
}
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
124
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
125 126 127
                        const Options& src) {
  Options result = src;
  result.comparator = icmp;
128
  result.filter_policy = (src.filter_policy != nullptr) ? ipolicy : nullptr;
129
  ClipToRange(&result.max_open_files,            20,     1000000);
130 131
  ClipToRange(&result.write_buffer_size,         ((size_t)64)<<10,
                                                 ((size_t)64)<<30);
S
Sanjay Ghemawat 已提交
132
  ClipToRange(&result.block_size,                1<<10,  4<<20);
133

X
Xing Jin 已提交
134 135 136 137 138 139
  // if user sets arena_block_size, we trust user to use this value. Otherwise,
  // calculate a proper value from writer_buffer_size;
  if (result.arena_block_size <= 0) {
    result.arena_block_size = result.write_buffer_size / 10;
  }

140 141
  result.min_write_buffer_number_to_merge = std::min(
    result.min_write_buffer_number_to_merge, result.max_write_buffer_number-1);
142
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
143 144
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
145 146
    if (!s.ok()) {
      // No place suitable for logging
147
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
148 149
    }
  }
150
  if (result.block_cache == nullptr && !result.no_block_cache) {
151 152
    result.block_cache = NewLRUCache(8 << 20);
  }
153
  result.compression_per_level = src.compression_per_level;
154 155 156
  if (result.block_size_deviation < 0 || result.block_size_deviation > 100) {
    result.block_size_deviation = 0;
  }
157 158 159
  if (result.max_mem_compaction_level >= result.num_levels) {
    result.max_mem_compaction_level = result.num_levels - 1;
  }
J
Jim Paton 已提交
160 161 162
  if (result.soft_rate_limit > result.hard_rate_limit) {
    result.soft_rate_limit = result.hard_rate_limit;
  }
163 164
  if (result.compaction_filter) {
    Log(result.info_log, "Compaction filter specified, ignore factory");
165
  }
J
Jim Paton 已提交
166
  if (result.prefix_extractor) {
I
Igor Canadi 已提交
167
    // If a prefix extractor has been supplied and a HashSkipListRepFactory is
J
Jim Paton 已提交
168 169
    // being used, make sure that the latter uses the former as its transform
    // function.
I
Igor Canadi 已提交
170
    auto factory = dynamic_cast<HashSkipListRepFactory*>(
J
Jim Paton 已提交
171
      result.memtable_factory.get());
172
    if (factory &&
173
        factory->GetTransform() != result.prefix_extractor) {
J
Jim Paton 已提交
174 175 176 177
      Log(result.info_log, "A prefix hash representation factory was supplied "
          "whose prefix extractor does not match options.prefix_extractor. "
          "Falling back to skip list representation factory");
      result.memtable_factory = std::make_shared<SkipListFactory>();
178 179
    } else if (factory) {
      Log(result.info_log, "Prefix hash memtable rep is in use.");
J
Jim Paton 已提交
180 181
    }
  }
182 183 184 185 186

  if (result.wal_dir.empty()) {
    // Use dbname as default
    result.wal_dir = dbname;
  }
187

K
kailiu 已提交
188 189 190 191 192 193
  // -- Sanitize the table properties collector
  // All user defined properties collectors will be wrapped by
  // UserKeyTablePropertiesCollector since for them they only have the
  // knowledge of the user keys; internal keys are invisible to them.
  auto& collectors = result.table_properties_collectors;
  for (size_t i = 0; i < result.table_properties_collectors.size(); ++i) {
194 195
    assert(collectors[i]);
    collectors[i] =
K
kailiu 已提交
196
      std::make_shared<UserKeyTablePropertiesCollector>(collectors[i]);
197 198 199 200
  }

  // Add collector to collect internal key statistics
  collectors.push_back(
K
kailiu 已提交
201
      std::make_shared<InternalKeyPropertiesCollector>()
202 203
  );

J
jorlow@chromium.org 已提交
204 205 206
  return result;
}

S
Siying Dong 已提交
207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227
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;
  }
}

J
jorlow@chromium.org 已提交
228 229
DBImpl::DBImpl(const Options& options, const std::string& dbname)
    : env_(options.env),
H
heyongqiang 已提交
230
      dbname_(dbname),
J
jorlow@chromium.org 已提交
231
      internal_comparator_(options.comparator),
S
Sanjay Ghemawat 已提交
232 233
      options_(SanitizeOptions(
          dbname, &internal_comparator_, &internal_filter_policy_, options)),
H
heyongqiang 已提交
234
      internal_filter_policy_(options.filter_policy),
J
jorlow@chromium.org 已提交
235
      owns_info_log_(options_.info_log != options.info_log),
236
      db_lock_(nullptr),
H
Haobo Xu 已提交
237
      mutex_(options.use_adaptive_mutex),
238
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
239
      bg_cv_(&mutex_),
I
Igor Canadi 已提交
240
      mem_rep_factory_(options_.memtable_factory.get()),
X
Xing Jin 已提交
241 242
      mem_(new MemTable(internal_comparator_, mem_rep_factory_,
        NumberLevels(), options_)),
243
      logfile_number_(0),
I
Igor Canadi 已提交
244
      super_version_(nullptr),
245
      tmp_batch_(),
246
      bg_compaction_scheduled_(0),
247
      bg_manual_only_(0),
248
      bg_flush_scheduled_(0),
249
      bg_logstats_scheduled_(false),
250 251
      manual_compaction_(nullptr),
      logger_(nullptr),
252
      disable_delete_obsolete_files_(0),
I
Igor Canadi 已提交
253
      delete_obsolete_files_last_run_(options.env->NowMicros()),
254
      purge_wal_files_last_run_(0),
255
      last_stats_dump_time_microsec_(0),
256
      default_interval_to_delete_obsolete_WAL_(600),
M
Mark Callaghan 已提交
257 258 259
      stall_level0_slowdown_(0),
      stall_memtable_compaction_(0),
      stall_level0_num_files_(0),
J
Jim Paton 已提交
260 261 262
      stall_level0_slowdown_count_(0),
      stall_memtable_compaction_count_(0),
      stall_level0_num_files_count_(0),
263
      started_at_(options.env->NowMicros()),
264
      flush_on_destroy_(false),
265
      stats_(options.num_levels),
266
      delayed_writes_(0),
267 268 269
      storage_options_(options),
      bg_work_gate_closed_(false),
      refitting_level_(false) {
270
  mem_->Ref();
H
heyongqiang 已提交
271
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
272 273

  stall_leveln_slowdown_.resize(options.num_levels);
J
Jim Paton 已提交
274 275
  stall_leveln_slowdown_count_.resize(options.num_levels);
  for (int i = 0; i < options.num_levels; ++i) {
276
    stall_leveln_slowdown_[i] = 0;
J
Jim Paton 已提交
277 278
    stall_leveln_slowdown_count_[i] = 0;
  }
279

J
jorlow@chromium.org 已提交
280
  // Reserve ten files or so for other uses and give the rest to TableCache.
281
  const int table_cache_size = options_.max_open_files - 10;
282 283 284 285
  table_cache_.reset(new TableCache(dbname_, &options_,
                                    storage_options_, table_cache_size));
  versions_.reset(new VersionSet(dbname_, &options_, storage_options_,
                                 table_cache_.get(), &internal_comparator_));
286

K
kailiu 已提交
287
  DumpLeveldbBuildVersion(options_.info_log.get());
288
  options_.Dump(options_.info_log.get());
289

290
  char name[100];
K
kailiu 已提交
291 292
  Status s = env_->GetHostName(name, 100L);
  if (s.ok()) {
293 294 295 296 297 298
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
299

I
Igor Canadi 已提交
300
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
301 302 303
}

DBImpl::~DBImpl() {
K
Kai Liu 已提交
304
  autovector<MemTable*> to_delete;
305

J
jorlow@chromium.org 已提交
306
  // Wait for background work to finish
307
  if (flush_on_destroy_ && mem_->GetFirstSequenceNumber() != 0) {
308 309
    FlushMemTable(FlushOptions());
  }
310
  mutex_.Lock();
311
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
312 313 314
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
315
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
316
  }
I
Igor Canadi 已提交
317 318 319 320 321 322 323
  if (super_version_ != nullptr) {
    bool is_last_reference __attribute__((unused));
    is_last_reference = super_version_->Unref();
    assert(is_last_reference);
    super_version_->Cleanup();
    delete super_version_;
  }
J
jorlow@chromium.org 已提交
324 325
  mutex_.Unlock();

326
  if (db_lock_ != nullptr) {
J
jorlow@chromium.org 已提交
327 328 329
    env_->UnlockFile(db_lock_);
  }

330 331 332 333 334 335 336 337
  if (mem_ != nullptr) {
    delete mem_->Unref();
  }

  imm_.UnrefAll(&to_delete);
  for (MemTable* m: to_delete) {
    delete m;
  }
I
Igor Canadi 已提交
338
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
339 340
}

A
Abhishek Kona 已提交
341
// Do not flush and close database elegantly. Simulate a crash.
342 343 344 345 346 347
void DBImpl::TEST_Destroy_DBImpl() {
  // ensure that no new memtable flushes can occur
  flush_on_destroy_ = false;

  // wait till all background compactions are done.
  mutex_.Lock();
348 349 350
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
351 352
    bg_cv_.Wait();
  }
I
Igor Canadi 已提交
353 354 355 356 357 358 359
  if (super_version_ != nullptr) {
    bool is_last_reference __attribute__((unused));
    is_last_reference = super_version_->Unref();
    assert(is_last_reference);
    super_version_->Cleanup();
    delete super_version_;
  }
360 361

  // Prevent new compactions from occuring.
362
  bg_work_gate_closed_ = true;
363 364
  const int LargeNumber = 10000000;
  bg_compaction_scheduled_ += LargeNumber;
365

366
  mutex_.Unlock();
I
Igor Canadi 已提交
367
  LogFlush(options_.info_log);
368 369

  // force release the lock file.
370
  if (db_lock_ != nullptr) {
371 372
    env_->UnlockFile(db_lock_);
  }
373 374 375 376

  log_.reset();
  versions_.reset();
  table_cache_.reset();
377 378
}

A
Abhishek Kona 已提交
379 380 381
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
382

J
jorlow@chromium.org 已提交
383
Status DBImpl::NewDB() {
384
  VersionEdit new_db(NumberLevels());
J
jorlow@chromium.org 已提交
385
  new_db.SetComparatorName(user_comparator()->Name());
386
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
387 388 389 390
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
391
  unique_ptr<WritableFile> file;
392
  Status s = env_->NewWritableFile(manifest, &file, storage_options_);
J
jorlow@chromium.org 已提交
393 394 395
  if (!s.ok()) {
    return s;
  }
396
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
397
  {
398
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415
    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 {
416
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
417 418 419 420
    *s = Status::OK();
  }
}

421
const Status DBImpl::CreateArchivalDirectory() {
422
  if (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0) {
423
    std::string archivalPath = ArchivalDirectory(options_.wal_dir);
424 425 426 427 428
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

429
void DBImpl::PrintStatistics() {
430
  auto dbstats = options_.statistics.get();
431 432
  if (dbstats) {
    Log(options_.info_log,
433 434
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
435 436 437
  }
}

438
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
439 440 441 442 443 444 445 446 447 448 449 450 451
  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;
452
    GetProperty("rocksdb.stats", &stats);
H
Haobo Xu 已提交
453
    Log(options_.info_log, "%s", stats.c_str());
454
    PrintStatistics();
455 456 457
  }
}

I
Igor Canadi 已提交
458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
// DBImpl::SuperVersion methods
DBImpl::SuperVersion::~SuperVersion() {
  for (auto td : to_delete) {
    delete td;
  }
}

DBImpl::SuperVersion* DBImpl::SuperVersion::Ref() {
  refs.fetch_add(1, std::memory_order_relaxed);
  return this;
}

bool DBImpl::SuperVersion::Unref() {
  assert(refs > 0);
  // fetch_sub returns the previous value of ref
  return refs.fetch_sub(1, std::memory_order_relaxed) == 1;
}

void DBImpl::SuperVersion::Cleanup() {
  assert(refs.load(std::memory_order_relaxed) == 0);
  imm.UnrefAll(&to_delete);
  MemTable* m = mem->Unref();
  if (m != nullptr) {
    to_delete.push_back(m);
  }
  current->Unref();
}

void DBImpl::SuperVersion::Init(MemTable* new_mem, const MemTableList& new_imm,
                                Version* new_current) {
  mem = new_mem;
  imm = new_imm;
  current = new_current;
  mem->Ref();
  imm.RefAll();
  current->Ref();
  refs.store(1, std::memory_order_relaxed);
}

497
// Returns the list of live files in 'sst_live' and the list
K
kailiu 已提交
498
// of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
499 500 501 502 503 504 505
// 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 已提交
506 507
  mutex_.AssertHeld();

508
  // if deletion is disabled, do nothing
509
  if (disable_delete_obsolete_files_ > 0) {
510 511 512
    return;
  }

513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528
  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 已提交
529 530 531
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
532 533 534 535 536
  // store the current filenum, lognum, etc
  deletion_state.manifest_file_number = versions_->ManifestFileNumber();
  deletion_state.log_number = versions_->LogNumber();
  deletion_state.prev_log_number = versions_->PrevLogNumber();

537 538 539 540 541 542 543 544
  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 已提交
545 546 547 548
  deletion_state.sst_live.assign(pending_outputs_.begin(),
                                 pending_outputs_.end());
  versions_->AddLiveFiles(&deletion_state.sst_live);

549
  if (doing_the_full_scan) {
K
kailiu 已提交
550 551 552 553 554
    // 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
555 556 557 558 559

    //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 已提交
560 561
      deletion_state.candidate_files.insert(
        deletion_state.candidate_files.end(),
562 563 564
        log_files.begin(),
        log_files.end()
      );
565
    }
566
  }
567 568
}

D
Dhruba Borthakur 已提交
569
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
570
// belong to live files are posibly removed. Also, removes all the
571
// files in sst_delete_files and log_delete_files.
572
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
573
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
574
  // check if there is anything to do
K
kailiu 已提交
575 576 577
  if (state.candidate_files.empty() &&
      state.sst_delete_files.empty() &&
      state.log_delete_files.empty()) {
578 579 580
    return;
  }

I
Igor Canadi 已提交
581 582 583 584
  // 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 已提交
585 586
    return;
  }
H
heyongqiang 已提交
587
  std::vector<std::string> old_log_files;
588

589 590
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
K
kailiu 已提交
591 592 593
  std::unordered_set<uint64_t> sst_live(
      state.sst_live.begin(), state.sst_live.end()
  );
I
Igor Canadi 已提交
594

K
kailiu 已提交
595 596 597 598 599 600 601
  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 = "";
602
  for (auto file : state.sst_delete_files) {
K
kailiu 已提交
603 604 605
    candidate_files.push_back(
        TableFileName(kDumbDbName, file->number).substr(1)
    );
606
    delete file;
I
Igor Canadi 已提交
607 608
  }

K
kailiu 已提交
609 610 611 612 613
  for (auto file_num : state.log_delete_files) {
    if (file_num > 0) {
      candidate_files.push_back(
          LogFileName(kDumbDbName, file_num).substr(1)
      );
I
Igor Canadi 已提交
614 615
    }
  }
616

K
kailiu 已提交
617
  // dedup state.candidate_files so we don't try to delete the same
I
Igor Canadi 已提交
618
  // file twice
K
kailiu 已提交
619 620 621 622 623
  sort(candidate_files.begin(), candidate_files.end());
  candidate_files.erase(
      unique(candidate_files.begin(), candidate_files.end()),
      candidate_files.end()
  );
J
jorlow@chromium.org 已提交
624

K
kailiu 已提交
625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
  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;
    }

    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'
        // (in case there is a race that allows other incarnations)
        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
        // be recorded in pending_outputs_, which is inserted into "live"
        keep = (sst_live.find(number) != sst_live.end());
        break;
      case kInfoLogFile:
        keep = true;
        if (number != 0) {
          old_log_files.push_back(to_delete);
J
jorlow@chromium.org 已提交
656
        }
K
kailiu 已提交
657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
        break;
      case kCurrentFile:
      case kDBLockFile:
      case kIdentityFile:
      case kMetaDatabase:
        keep = true;
        break;
    }

    if (keep) {
      continue;
    }

    if (type == kTableFile) {
      // evict from cache
      table_cache_->Evict(number);
    }
    std::string fname = ((type == kLogFile) ? options_.wal_dir : dbname_) +
        "/" + to_delete;
    Log(options_.info_log,
        "Delete type=%d #%lu",
        int(type),
        (unsigned long)number);

    if (type == kLogFile &&
        (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0)) {
      Status s = env_->RenameFile(fname,
          ArchivedLogFileName(options_.wal_dir, number));
      if (!s.ok()) {
K
Kai Liu 已提交
686
        Log(options_.info_log,
K
kailiu 已提交
687 688 689 690 691 692 693 694
            "RenameFile logfile #%lu FAILED -- %s\n",
            (unsigned long)number, s.ToString().c_str());
      }
    } else {
      Status s = env_->DeleteFile(fname);
      if (!s.ok()) {
        Log(options_.info_log, "Delete type=%d #%lu FAILED -- %s\n",
            int(type), (unsigned long)number, s.ToString().c_str());
J
jorlow@chromium.org 已提交
695 696 697
      }
    }
  }
H
heyongqiang 已提交
698

699
  // Delete old info log files.
K
Kai Liu 已提交
700 701 702 703 704
  size_t old_log_file_count = old_log_files.size();
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
  if (old_log_file_count >= options_.keep_log_file_num &&
      options_.db_log_dir.empty()) {
H
heyongqiang 已提交
705
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
706
    size_t end = old_log_file_count - options_.keep_log_file_num;
707
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
708
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
709 710
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
711 712 713
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
714
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
715
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
716 717 718 719 720
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
721
  FindObsoleteFiles(deletion_state, true);
D
Dhruba Borthakur 已提交
722
  PurgeObsoleteFiles(deletion_state);
723 724
}

725 726 727 728 729 730 731 732
// 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
733
void DBImpl::PurgeObsoleteWALFiles() {
734 735 736 737 738 739
  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;
  }

740 741
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
742 743 744 745 746 747 748 749
  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_;
750

751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790
  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;
          }
          continue;
791
        }
792 793 794 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 834 835 836 837 838 839 840 841 842 843 844
      }

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

  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;
  AppendSortedWalsOfType(archival_dir, archived_logs, kArchivedLogFile);

  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;
845 846
    }
  }
D
Dhruba Borthakur 已提交
847 848
}

849 850
// If externalTable is set, then apply recovered transactions
// to that table. This is used for readonly mode.
K
kailiu 已提交
851 852 853
Status DBImpl::Recover(
    VersionEdit* edit,
    MemTable* external_table,
H
heyongqiang 已提交
854
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
855 856
  mutex_.AssertHeld();

857
  assert(db_lock_ == nullptr);
858
  if (!external_table) {
859 860 861 862 863 864 865 866 867 868 869 870 871
    // We call CreateDirIfMissing() as the directory may already exist (if we
    // are reopening a DB), when this happens we don't want creating the
    // directory to cause an error. However, we need to check if creating the
    // directory fails or else we may get an obscure message about the lock
    // file not existing. One real-world example of this occurring is if
    // env->CreateDirIfMissing() doesn't create intermediate directories, e.g.
    // when dbname_ is "dir/db" but when "dir" doesn't exist.
    Status s = env_->CreateDirIfMissing(dbname_);
    if (!s.ok()) {
      return s;
    }

    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
872 873 874
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
875

876 877
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
878
        // TODO: add merge_operator name check
879 880 881 882 883 884 885
        s = NewDB();
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
886 887
      }
    } else {
888 889 890 891
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
892
    }
M
Mayank Agarwal 已提交
893 894 895 896 897 898 899
    // 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 已提交
900 901
  }

902
  Status s = versions_->Recover();
J
jorlow@chromium.org 已提交
903 904
  if (s.ok()) {
    SequenceNumber max_sequence(0);
905 906 907 908 909 910 911

    // 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
912
    // produced by an older version of rocksdb.
913 914 915
    const uint64_t min_log = versions_->LogNumber();
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
916
    s = env_->GetChildren(options_.wal_dir, &filenames);
917 918
    if (!s.ok()) {
      return s;
919
    }
K
kailiu 已提交
920

921 922
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
K
kailiu 已提交
923 924
      uint64_t number;
      FileType type;
925 926 927 928 929
      if (ParseFileName(filenames[i], &number, &type)
          && type == kLogFile
          && ((number >= min_log) || (number == prev_log))) {
        logs.push_back(number);
      }
J
jorlow@chromium.org 已提交
930
    }
931

H
heyongqiang 已提交
932 933 934 935 936 937
    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");
    }

938 939
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
K
kailiu 已提交
940 941
    for (const auto& log : logs) {
      s = RecoverLogFile(log, edit, &max_sequence, external_table);
942 943 944
      // 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 已提交
945
      versions_->MarkFileNumberUsed(log);
946 947
    }

J
jorlow@chromium.org 已提交
948
    if (s.ok()) {
949 950 951
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
      }
952
      SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
953
                     versions_->LastSequence());
J
jorlow@chromium.org 已提交
954 955 956 957 958 959 960 961
    }
  }

  return s;
}

Status DBImpl::RecoverLogFile(uint64_t log_number,
                              VersionEdit* edit,
962 963
                              SequenceNumber* max_sequence,
                              MemTable* external_table) {
J
jorlow@chromium.org 已提交
964 965
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
966
    Logger* info_log;
J
jorlow@chromium.org 已提交
967
    const char* fname;
968 969
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
970
    virtual void Corruption(size_t bytes, const Status& s) {
971
      Log(info_log, "%s%s: dropping %d bytes; %s",
972
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
973
          fname, static_cast<int>(bytes), s.ToString().c_str());
974
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
975 976 977 978 979 980
    }
  };

  mutex_.AssertHeld();

  // Open the log file
981
  std::string fname = LogFileName(options_.wal_dir, log_number);
982
  unique_ptr<SequentialFile> file;
983
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
984 985 986 987 988 989 990 991
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
992
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
993
  reporter.fname = fname.c_str();
994 995
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
996 997 998 999
  // 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).
1000
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1001
                     0/*initial_offset*/);
K
Kai Liu 已提交
1002 1003
  Log(options_.info_log, "Recovering log #%lu",
      (unsigned long) log_number);
J
jorlow@chromium.org 已提交
1004 1005 1006 1007 1008

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
1009
  MemTable* mem = nullptr;
1010 1011 1012
  if (external_table) {
    mem = external_table;
  }
1013
  while (reader.ReadRecord(&record, &scratch) && status.ok()) {
J
jorlow@chromium.org 已提交
1014 1015 1016 1017 1018 1019 1020
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

1021
    if (mem == nullptr) {
X
Xing Jin 已提交
1022 1023
      mem = new MemTable(internal_comparator_, mem_rep_factory_,
        NumberLevels(), options_);
1024
      mem->Ref();
J
jorlow@chromium.org 已提交
1025
    }
1026
    status = WriteBatchInternal::InsertInto(&batch, mem, &options_);
J
jorlow@chromium.org 已提交
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
    MaybeIgnoreError(&status);
    if (!status.ok()) {
      break;
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

1038 1039
    if (!external_table &&
        mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
1040
      status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
1041 1042 1043 1044 1045
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
        break;
      }
1046
      delete mem->Unref();
1047
      mem = nullptr;
J
jorlow@chromium.org 已提交
1048 1049 1050
    }
  }

1051
  if (status.ok() && mem != nullptr && !external_table) {
1052
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
1053 1054 1055 1056
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

1057 1058 1059
  if (mem != nullptr && !external_table) {
    delete mem->Unref();
  }
J
jorlow@chromium.org 已提交
1060 1061 1062
  return status;
}

1063
Status DBImpl::WriteLevel0TableForRecovery(MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1064
  mutex_.AssertHeld();
1065
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1066 1067 1068 1069
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
1070 1071 1072
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
K
Kai Liu 已提交
1073 1074
  Log(options_.info_log, "Level-0 table #%lu: started",
      (unsigned long) meta.number);
1075 1076 1077 1078

  Status s;
  {
    mutex_.Unlock();
1079 1080
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1081
                   user_comparator(), newest_snapshot,
1082
                   earliest_seqno_in_memtable, true);
I
Igor Canadi 已提交
1083
    LogFlush(options_.info_log);
1084 1085 1086
    mutex_.Lock();
  }

K
Kai Liu 已提交
1087 1088 1089
  Log(options_.info_log, "Level-0 table #%lu: %lu bytes %s",
      (unsigned long) meta.number,
      (unsigned long) meta.file_size,
J
jorlow@chromium.org 已提交
1090 1091
      s.ToString().c_str());
  delete iter;
1092

1093
  pending_outputs_.erase(meta.number);
1094 1095 1096 1097 1098 1099

  // 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,
1100 1101
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1102 1103
  }

1104 1105 1106
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
1107
  stats.files_out_levelnp1 = 1;
1108
  stats_[level].Add(stats);
1109
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
J
jorlow@chromium.org 已提交
1110 1111 1112
  return s;
}

1113

K
Kai Liu 已提交
1114
Status DBImpl::WriteLevel0Table(autovector<MemTable*>& mems, VersionEdit* edit,
1115
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
1116
  mutex_.AssertHeld();
1117 1118 1119 1120 1121
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
1122

1123 1124
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1125
    mems[0]->GetFirstSequenceNumber();
1126
  Version* base = versions_->current();
1127
  base->Ref();          // it is likely that we do not need this reference
1128 1129 1130
  Status s;
  {
    mutex_.Unlock();
K
Kai Liu 已提交
1131
    std::vector<Iterator*> memtables;
1132 1133 1134 1135
    for (MemTable* m : mems) {
      Log(options_.info_log,
          "Flushing memtable with log file: %lu\n",
          (unsigned long)m->GetLogNumber());
K
Kai Liu 已提交
1136
      memtables.push_back(m->NewIterator());
1137
    }
K
Kai Liu 已提交
1138 1139
    Iterator* iter = NewMergingIterator(
        env_, &internal_comparator_, &memtables[0], memtables.size());
1140 1141 1142
    Log(options_.info_log,
        "Level-0 flush table #%lu: started",
        (unsigned long)meta.number);
1143 1144 1145 1146 1147
    // We skip compression if universal compression is used and the size
    // threshold is set for compression.
    bool enable_compression = (options_.compaction_style
        != kCompactionStyleUniversal ||
        options_.compaction_options_universal.compression_size_percent < 0);
1148 1149
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1150
                   user_comparator(), newest_snapshot,
1151
                   earliest_seqno_in_memtable, enable_compression);
I
Igor Canadi 已提交
1152
    LogFlush(options_.info_log);
1153 1154 1155 1156 1157
    delete iter;
    Log(options_.info_log, "Level-0 flush table #%lu: %lu bytes %s",
        (unsigned long) meta.number,
        (unsigned long) meta.file_size,
        s.ToString().c_str());
1158 1159
    mutex_.Lock();
  }
1160 1161
  base->Unref();

1162 1163 1164 1165 1166 1167 1168 1169
  // re-acquire the most current version
  base = versions_->current();

  // There could be multiple threads writing to its own level-0 file.
  // The pending_outputs cannot be cleared here, otherwise this newly
  // created file might not be considered as a live-file by another
  // compaction thread that is concurrently deleting obselete files.
  // The pending_outputs can be cleared only after the new version is
A
Abhishek Kona 已提交
1170
  // committed so that other threads can recognize this file as a
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183
  // 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.
1184
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1185
        options_.compaction_style == kCompactionStyleLevel) {
1186 1187 1188
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
1189 1190
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1191 1192 1193 1194 1195 1196
  }

  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
  stats_[level].Add(stats);
1197
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
1198 1199 1200
  return s;
}

I
Igor Canadi 已提交
1201 1202
Status DBImpl::FlushMemTableToOutputFile(bool* madeProgress,
                                         DeletionState& deletion_state) {
1203 1204 1205
  mutex_.AssertHeld();
  assert(imm_.size() != 0);

1206
  if (!imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
1207 1208
    Log(options_.info_log, "FlushMemTableToOutputFile already in progress");
    Status s = Status::IOError("FlushMemTableToOutputFile already in progress");
1209 1210 1211 1212 1213
    return s;
  }

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
K
Kai Liu 已提交
1214
  autovector<MemTable*> mems;
1215 1216
  imm_.PickMemtablesToFlush(&mems);
  if (mems.empty()) {
1217 1218 1219 1220 1221 1222
    Log(options_.info_log, "Nothing in memstore to flush");
    Status s = Status::IOError("Nothing in memstore to flush");
    return s;
  }

  // record the logfile_number_ before we release the mutex
1223 1224 1225
  // 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.
1226
  MemTable* m = mems[0];
1227 1228
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1229 1230 1231 1232 1233 1234 1235 1236 1237 1238
  // SetLogNumber(log_num) indicates logs with number smaller than log_num
  // will no longer be picked up for recovery.
  edit->SetLogNumber(
      mems.back()->GetNextLogNumber()
  );

  std::vector<uint64_t> logs_to_delete;
  for (auto mem : mems) {
    logs_to_delete.push_back(mem->GetLogNumber());
  }
1239

1240
  // This will release and re-acquire the mutex.
1241
  Status s = WriteLevel0Table(mems, edit, &file_number);
1242

1243
  if (s.ok() && shutting_down_.Acquire_Load()) {
1244 1245 1246
    s = Status::IOError(
      "Database shutdown started during memtable compaction"
    );
1247
  }
J
jorlow@chromium.org 已提交
1248

1249
  // Replace immutable memtable with the generated Table
1250
  s = imm_.InstallMemtableFlushResults(
1251
    mems, versions_.get(), s, &mutex_, options_.info_log.get(),
1252
    file_number, pending_outputs_, &deletion_state.memtables_to_free);
J
jorlow@chromium.org 已提交
1253 1254

  if (s.ok()) {
I
Igor Canadi 已提交
1255
    InstallSuperVersion(deletion_state);
1256 1257 1258
    if (madeProgress) {
      *madeProgress = 1;
    }
1259

1260
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1261

1262
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1263
      // add to deletion state
1264 1265 1266 1267
      deletion_state.log_delete_files.insert(
          deletion_state.log_delete_files.end(),
          logs_to_delete.begin(),
          logs_to_delete.end());
1268
    }
J
jorlow@chromium.org 已提交
1269 1270 1271 1272
  }
  return s;
}

1273
void DBImpl::CompactRange(const Slice* begin, const Slice* end,
1274
                          bool reduce_level, int target_level) {
G
Gabor Cselle 已提交
1275 1276 1277 1278
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->current();
1279
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
1280 1281 1282 1283 1284
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
1285
  TEST_FlushMemTable(); // TODO(sanjay): Skip if memtable does not overlap
G
Gabor Cselle 已提交
1286 1287 1288
  for (int level = 0; level < max_level_with_files; level++) {
    TEST_CompactRange(level, begin, end);
  }
1289 1290

  if (reduce_level) {
1291
    ReFitLevel(max_level_with_files, target_level);
1292
  }
I
Igor Canadi 已提交
1293
  LogFlush(options_.info_log);
1294 1295 1296 1297 1298 1299
}

// return the same level if it cannot be moved
int DBImpl::FindMinimumEmptyLevelFitting(int level) {
  mutex_.AssertHeld();
  int minimum_level = level;
1300
  for (int i = level - 1; i > 0; --i) {
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
    // stop if level i is not empty
    if (versions_->NumLevelFiles(i) > 0) break;

    // stop if level i is too small (cannot fit the level files)
    if (versions_->MaxBytesForLevel(i) < versions_->NumLevelBytes(level)) break;

    minimum_level = i;
  }
  return minimum_level;
}

1312
void DBImpl::ReFitLevel(int level, int target_level) {
1313 1314
  assert(level < NumberLevels());

I
Igor Canadi 已提交
1315
  SuperVersion* superversion_to_free = nullptr;
K
Kai Liu 已提交
1316
  SuperVersion* new_superversion = new SuperVersion();
I
Igor Canadi 已提交
1317 1318

  mutex_.Lock();
1319 1320 1321

  // only allow one thread refitting
  if (refitting_level_) {
I
Igor Canadi 已提交
1322
    mutex_.Unlock();
1323
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
I
Igor Canadi 已提交
1324
    delete new_superversion;
1325 1326 1327 1328 1329 1330
    return;
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1331
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1332
    Log(options_.info_log,
1333 1334
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1335 1336 1337 1338
    bg_cv_.Wait();
  }

  // move to a smaller level
1339 1340 1341 1342
  int to_level = target_level;
  if (target_level < 0) {
    to_level = FindMinimumEmptyLevelFitting(level);
  }
1343 1344 1345 1346 1347 1348 1349 1350 1351 1352

  assert(to_level <= level);

  if (to_level < level) {
    Log(options_.info_log, "Before refitting:\n%s",
        versions_->current()->DebugString().data());

    VersionEdit edit(NumberLevels());
    for (const auto& f : versions_->current()->files_[level]) {
      edit.DeleteFile(level, f->number);
1353 1354
      edit.AddFile(to_level, f->number, f->file_size, f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1355 1356 1357 1358 1359
    }
    Log(options_.info_log, "Apply version edit:\n%s",
        edit.DebugString().data());

    auto status = versions_->LogAndApply(&edit, &mutex_);
I
Igor Canadi 已提交
1360 1361
    superversion_to_free = InstallSuperVersion(new_superversion);
    new_superversion = nullptr;
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372

    Log(options_.info_log, "LogAndApply: %s\n", status.ToString().data());

    if (status.ok()) {
      Log(options_.info_log, "After refitting:\n%s",
          versions_->current()->DebugString().data());
    }
  }

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
1373 1374 1375 1376

  mutex_.Unlock();
  delete superversion_to_free;
  delete new_superversion;
G
Gabor Cselle 已提交
1377 1378
}

1379
int DBImpl::NumberLevels() {
1380
  return options_.num_levels;
1381 1382 1383
}

int DBImpl::MaxMemCompactionLevel() {
1384
  return options_.max_mem_compaction_level;
1385 1386 1387
}

int DBImpl::Level0StopWriteTrigger() {
1388
  return options_.level0_stop_writes_trigger;
1389 1390
}

H
heyongqiang 已提交
1391
Status DBImpl::Flush(const FlushOptions& options) {
K
kailiu 已提交
1392
  return FlushMemTable(options);
H
heyongqiang 已提交
1393 1394
}

1395
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1396 1397 1398
  return versions_->LastSequence();
}

1399
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
1400
                               unique_ptr<TransactionLogIterator>* iter) {
1401

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

1409 1410
  std::unique_ptr<VectorLogPtr> wal_files(new VectorLogPtr);
  Status s = GetSortedWalFiles(*wal_files);
1411 1412 1413 1414
  if (!s.ok()) {
    return s;
  }

1415
  s = RetainProbableWalFiles(*wal_files, seq);
1416 1417
  if (!s.ok()) {
    return s;
1418
  }
1419
  iter->reset(
1420
    new TransactionLogIteratorImpl(options_.wal_dir,
1421
                                   &options_,
1422
                                   storage_options_,
1423
                                   seq,
1424
                                   std::move(wal_files),
1425
                                   this));
1426
  return (*iter)->status();
1427 1428
}

1429 1430
Status DBImpl::RetainProbableWalFiles(VectorLogPtr& all_logs,
                                      const SequenceNumber target) {
1431
  long start = 0; // signed to avoid overflow when target is < first file.
1432
  long end = static_cast<long>(all_logs.size()) - 1;
1433
  // Binary Search. avoid opening all files.
1434 1435
  while (end >= start) {
    long mid = start + (end - start) / 2;  // Avoid overflow.
1436 1437
    SequenceNumber current_seq_num = all_logs.at(mid)->StartSequence();
    if (current_seq_num == target) {
1438
      end = mid;
1439
      break;
1440
    } else if (current_seq_num < target) {
1441
      start = mid + 1;
1442
    } else {
1443
      end = mid - 1;
1444 1445
    }
  }
1446 1447 1448
  size_t start_index = std::max(0l, end); // end could be -ve.
  // The last wal file is always included
  all_logs.erase(all_logs.begin(), all_logs.begin() + start_index);
1449 1450 1451
  return Status::OK();
}

1452 1453 1454
bool DBImpl::CheckWalFileExistsAndEmpty(const WalFileType type,
                                        const uint64_t number) {
  const std::string fname = (type == kAliveLogFile) ?
1455 1456
    LogFileName(options_.wal_dir, number) :
    ArchivedLogFileName(options_.wal_dir, number);
1457 1458
  uint64_t file_size;
  Status s = env_->GetFileSize(fname, &file_size);
1459
  return (s.ok() && (file_size == 0));
1460 1461
}

1462 1463
Status DBImpl::ReadFirstRecord(const WalFileType type, const uint64_t number,
                               WriteBatch* const result) {
1464

1465
  if (type == kAliveLogFile) {
1466
    std::string fname = LogFileName(options_.wal_dir, number);
1467 1468 1469
    Status status = ReadFirstLine(fname, result);
    if (!status.ok()) {
      //  check if the file got moved to archive.
1470 1471
      std::string archived_file =
        ArchivedLogFileName(options_.wal_dir, number);
1472
      Status s = ReadFirstLine(archived_file, result);
1473
      if (!s.ok()) {
1474
        return Status::IOError("Log File has been deleted: " + archived_file);
1475 1476 1477
      }
    }
    return Status::OK();
1478
  } else if (type == kArchivedLogFile) {
1479
    std::string fname = ArchivedLogFileName(options_.wal_dir, number);
1480 1481 1482
    Status status = ReadFirstLine(fname, result);
    return status;
  }
1483
  return Status::NotSupported("File Type Not Known: " + std::to_string(type));
1484 1485 1486 1487 1488 1489 1490 1491
}

Status DBImpl::ReadFirstLine(const std::string& fname,
                             WriteBatch* const batch) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;
1492
    Status* status;  // nullptr if options_.paranoid_checks==false
1493 1494
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
1495
          (this->status == nullptr ? "(ignoring error) " : ""),
1496
          fname, static_cast<int>(bytes), s.ToString().c_str());
1497
      if (this->status != nullptr && this->status->ok()) *this->status = s;
1498 1499 1500
    }
  };

1501
  unique_ptr<SequentialFile> file;
1502
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
1503 1504 1505 1506 1507 1508 1509 1510

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


  LogReporter reporter;
  reporter.env = env_;
1511
  reporter.info_log = options_.info_log.get();
1512
  reporter.fname = fname.c_str();
1513
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
1514
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1515 1516 1517
                     0/*initial_offset*/);
  std::string scratch;
  Slice record;
1518

1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531
  if (reader.ReadRecord(&record, &scratch) && status.ok()) {
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      return Status::IOError("Corruption noted");
      //  TODO read record's till the first no corrupt entry?
    }
    WriteBatchInternal::SetContents(batch, record);
    return Status::OK();
  }
  return Status::IOError("Error reading from file " + fname);
}

1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
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;
  }
};

Status DBImpl::AppendSortedWalsOfType(const std::string& path,
    VectorLogPtr& log_files, WalFileType log_type) {
  std::vector<std::string> all_files;
  const Status status = env_->GetChildren(path, &all_files);
1545 1546 1547
  if (!status.ok()) {
    return status;
  }
1548
  log_files.reserve(log_files.size() + all_files.size());
1549 1550 1551 1552 1553 1554
  VectorLogPtr::iterator pos_start;
  if (!log_files.empty()) {
    pos_start = log_files.end() - 1;
  } else {
    pos_start = log_files.begin();
  }
1555
  for (const auto& f : all_files) {
1556 1557
    uint64_t number;
    FileType type;
1558
    if (ParseFileName(f, &number, &type) && type == kLogFile){
1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576

      WriteBatch batch;
      Status s = ReadFirstRecord(log_type, number, &batch);
      if (!s.ok()) {
        if (CheckWalFileExistsAndEmpty(log_type, number)) {
          continue;
        }
        return s;
      }

      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>(new LogFileImpl(
        number, log_type, WriteBatchInternal::Sequence(&batch), size_bytes))));
1577 1578
    }
  }
1579
  CompareLogByPointer compare_log_files;
1580
  std::sort(pos_start, log_files.end(), compare_log_files);
1581 1582 1583
  return status;
}

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

G
Gabor Cselle 已提交
1587 1588
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1589 1590
  ManualCompaction manual;
  manual.level = level;
G
Gabor Cselle 已提交
1591
  manual.done = false;
1592
  manual.in_progress = false;
1593 1594 1595 1596
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
      options_.compaction_style == kCompactionStyleUniversal) {
1597
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1598 1599 1600 1601
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1602 1603
  if (end == nullptr ||
      options_.compaction_style == kCompactionStyleUniversal) {
1604
    manual.end = nullptr;
G
Gabor Cselle 已提交
1605 1606 1607 1608 1609 1610
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1611

1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630
  // 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
  // TEST_CompactRange(), i.e. during that time no other compaction will
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
  // TEST_CompactRange() from getting to the second while loop below.
  // 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,
        "Manual compaction waiting for all other scheduled background "
        "compactions to finish");
1631 1632
    bg_cv_.Wait();
  }
1633

1634 1635
  Log(options_.info_log, "Manual compaction starting");

1636 1637 1638 1639
  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 已提交
1640
      bg_cv_.Wait();
1641 1642 1643
    } else {
      manual_compaction_ = &manual;
      MaybeScheduleFlushOrCompaction();
G
Gabor Cselle 已提交
1644
    }
H
hans@chromium.org 已提交
1645
  }
1646

1647 1648 1649
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
J
jorlow@chromium.org 已提交
1650 1651
}

H
heyongqiang 已提交
1652
Status DBImpl::FlushMemTable(const FlushOptions& options) {
1653 1654
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1655
  if (s.ok() && options.wait) {
1656
    // Wait until the compaction completes
1657
    s = WaitForFlushMemTable();
1658 1659
  }
  return s;
J
jorlow@chromium.org 已提交
1660 1661
}

1662
Status DBImpl::WaitForFlushMemTable() {
1663 1664 1665
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1666
  while (imm_.size() > 0 && bg_error_.ok()) {
1667 1668
    bg_cv_.Wait();
  }
1669
  if (imm_.size() != 0) {
1670 1671 1672
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1673 1674
}

1675
Status DBImpl::TEST_FlushMemTable() {
H
heyongqiang 已提交
1676 1677 1678
  return FlushMemTable(FlushOptions());
}

1679 1680
Status DBImpl::TEST_WaitForFlushMemTable() {
  return WaitForFlushMemTable();
1681 1682 1683
}

Status DBImpl::TEST_WaitForCompact() {
1684
  // Wait until the compaction completes
1685 1686 1687 1688 1689

  // TODO: a bug here. This function actually does not necessarily
  // wait for compact. It actually waits for scheduled compaction
  // OR flush to finish.

1690
  MutexLock l(&mutex_);
1691 1692
  while ((bg_compaction_scheduled_ || bg_flush_scheduled_) &&
         bg_error_.ok()) {
1693 1694 1695
    bg_cv_.Wait();
  }
  return bg_error_;
1696 1697
}

1698
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
1699
  mutex_.AssertHeld();
1700 1701
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
J
jorlow@chromium.org 已提交
1702 1703 1704
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
  } else {
1705 1706 1707 1708 1709 1710 1711 1712 1713
    bool is_flush_pending =
      imm_.IsFlushPending(options_.min_write_buffer_number_to_merge);
    if (is_flush_pending &&
        (bg_flush_scheduled_ < options_.max_background_flushes)) {
      // memtable flush needed
      bg_flush_scheduled_++;
      env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::HIGH);
    }

1714 1715 1716 1717
    // Schedule BGWorkCompaction if there's a compaction pending (or a memtable
    // flush, but the HIGH pool is not enabled). Do it only if
    // max_background_compactions hasn't been reached and, in case
    // bg_manual_only_ > 0, if it's a manual compaction.
1718 1719 1720
    if ((manual_compaction_ ||
         versions_->NeedsCompaction() ||
         (is_flush_pending && (options_.max_background_flushes <= 0))) &&
1721 1722 1723
        bg_compaction_scheduled_ < options_.max_background_compactions &&
        (!bg_manual_only_ || manual_compaction_)) {

1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737
      bg_compaction_scheduled_++;
      env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW);
    }
  }
}

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

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

I
Igor Canadi 已提交
1738 1739
Status DBImpl::BackgroundFlush(bool* madeProgress,
                               DeletionState& deletion_state) {
1740 1741 1742 1743
  Status stat;
  while (stat.ok() &&
         imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
    Log(options_.info_log,
1744
        "BackgroundCallFlush doing FlushMemTableToOutputFile, flush slots available %d",
1745
        options_.max_background_flushes - bg_flush_scheduled_);
I
Igor Canadi 已提交
1746
    stat = FlushMemTableToOutputFile(madeProgress, deletion_state);
J
jorlow@chromium.org 已提交
1747
  }
1748
  return stat;
J
jorlow@chromium.org 已提交
1749 1750
}

1751
void DBImpl::BackgroundCallFlush() {
1752
  bool madeProgress = false;
K
Kai Liu 已提交
1753
  DeletionState deletion_state(true);
1754 1755 1756
  assert(bg_flush_scheduled_);
  MutexLock l(&mutex_);

I
Igor Canadi 已提交
1757
  Status s;
1758
  if (!shutting_down_.Acquire_Load()) {
I
Igor Canadi 已提交
1759
    s = BackgroundFlush(&madeProgress, deletion_state);
1760 1761 1762 1763 1764 1765 1766 1767 1768
    if (!s.ok()) {
      // Wait a little bit before retrying background compaction in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed compactions for the duration of
      // the problem.
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      Log(options_.info_log, "Waiting after background flush error: %s",
          s.ToString().c_str());
      mutex_.Unlock();
I
Igor Canadi 已提交
1769
      LogFlush(options_.info_log);
1770 1771 1772 1773 1774
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
  }

I
Igor Canadi 已提交
1775 1776 1777
  // 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());
I
Igor Canadi 已提交
1778
  // delete unnecessary files if any, this is done outside the mutex
I
Igor Canadi 已提交
1779 1780 1781 1782 1783 1784
  if (deletion_state.HaveSomethingToDelete()) {
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
    mutex_.Lock();
  }

1785
  bg_flush_scheduled_--;
1786 1787 1788
  if (madeProgress) {
    MaybeScheduleFlushOrCompaction();
  }
1789
  bg_cv_.SignalAll();
J
jorlow@chromium.org 已提交
1790 1791
}

1792

1793 1794 1795 1796
void DBImpl::TEST_PurgeObsoleteteWAL() {
  PurgeObsoleteWALFiles();
}

1797
void DBImpl::BackgroundCallCompaction() {
1798
  bool madeProgress = false;
K
Kai Liu 已提交
1799
  DeletionState deletion_state(true);
H
Haobo Xu 已提交
1800 1801 1802

  MaybeDumpStats();

J
jorlow@chromium.org 已提交
1803
  MutexLock l(&mutex_);
1804
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1805
  assert(bg_compaction_scheduled_);
I
Igor Canadi 已提交
1806
  Status s;
H
hans@chromium.org 已提交
1807
  if (!shutting_down_.Acquire_Load()) {
I
Igor Canadi 已提交
1808
    s = BackgroundCompaction(&madeProgress, deletion_state);
1809 1810 1811 1812 1813 1814 1815 1816 1817
    if (!s.ok()) {
      // Wait a little bit before retrying background compaction in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed compactions for the duration of
      // the problem.
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      Log(options_.info_log, "Waiting after background compaction error: %s",
          s.ToString().c_str());
      mutex_.Unlock();
I
Igor Canadi 已提交
1818
      LogFlush(options_.info_log);
1819 1820 1821
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
J
jorlow@chromium.org 已提交
1822
  }
1823

I
Igor Canadi 已提交
1824 1825 1826 1827 1828
  // 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());
1829

I
Igor Canadi 已提交
1830
  // delete unnecessary files if any, this is done outside the mutex
I
Igor Canadi 已提交
1831
  if (deletion_state.HaveSomethingToDelete()) {
D
Dhruba Borthakur 已提交
1832 1833
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
1834
    mutex_.Lock();
D
Dhruba Borthakur 已提交
1835 1836
  }

1837
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1838

1839 1840
  MaybeScheduleLogDBDeployStats();

J
jorlow@chromium.org 已提交
1841
  // Previous compaction may have produced too many files in a level,
A
Abhishek Kona 已提交
1842
  // So reschedule another compaction if we made progress in the
1843 1844
  // last compaction.
  if (madeProgress) {
1845
    MaybeScheduleFlushOrCompaction();
1846
  }
H
hans@chromium.org 已提交
1847
  bg_cv_.SignalAll();
1848

J
jorlow@chromium.org 已提交
1849 1850
}

A
Abhishek Kona 已提交
1851
Status DBImpl::BackgroundCompaction(bool* madeProgress,
I
Igor Canadi 已提交
1852
                                    DeletionState& deletion_state) {
1853
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1854
  mutex_.AssertHeld();
1855

1856
  // TODO: remove memtable flush from formal compaction
1857
  while (imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
A
Abhishek Kona 已提交
1858
    Log(options_.info_log,
1859 1860
        "BackgroundCompaction doing FlushMemTableToOutputFile, compaction slots "
        "available %d",
1861
        options_.max_background_compactions - bg_compaction_scheduled_);
I
Igor Canadi 已提交
1862
    Status stat = FlushMemTableToOutputFile(madeProgress, deletion_state);
1863 1864 1865
    if (!stat.ok()) {
      return stat;
    }
1866 1867
  }

1868
  unique_ptr<Compaction> c;
1869
  bool is_manual = (manual_compaction_ != nullptr) &&
1870
                   (manual_compaction_->in_progress == false);
G
Gabor Cselle 已提交
1871
  InternalKey manual_end;
H
hans@chromium.org 已提交
1872
  if (is_manual) {
G
Gabor Cselle 已提交
1873
    ManualCompaction* m = manual_compaction_;
1874 1875
    assert(!m->in_progress);
    m->in_progress = true; // another thread cannot pick up the same work
1876 1877
    c.reset(versions_->CompactRange(m->level, m->begin, m->end));
    if (c) {
G
Gabor Cselle 已提交
1878
      manual_end = c->input(0, c->num_input_files(0) - 1)->largest;
1879 1880
    } else {
      m->done = true;
G
Gabor Cselle 已提交
1881 1882 1883
    }
    Log(options_.info_log,
        "Manual compaction at level-%d from %s .. %s; will stop at %s\n",
H
hans@chromium.org 已提交
1884
        m->level,
G
Gabor Cselle 已提交
1885 1886 1887
        (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
        (m->end ? m->end->DebugString().c_str() : "(end)"),
        (m->done ? "(end)" : manual_end.DebugString().c_str()));
1888
  } else if (!options_.disable_auto_compactions) {
1889
    c.reset(versions_->PickCompaction());
J
jorlow@chromium.org 已提交
1890 1891 1892
  }

  Status status;
1893
  if (!c) {
H
hans@chromium.org 已提交
1894
    // Nothing to do
1895
    Log(options_.info_log, "Compaction nothing to do");
H
hans@chromium.org 已提交
1896
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
1897
    // Move file to next level
1898
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
1899 1900 1901
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
1902 1903
                       f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
1904
    status = versions_->LogAndApply(c->edit(), &mutex_);
I
Igor Canadi 已提交
1905
    InstallSuperVersion(deletion_state);
H
hans@chromium.org 已提交
1906
    VersionSet::LevelSummaryStorage tmp;
1907
    Log(options_.info_log, "Moved #%lld to level-%d %lld bytes %s: %s\n",
J
jorlow@chromium.org 已提交
1908 1909 1910
        static_cast<unsigned long long>(f->number),
        c->level() + 1,
        static_cast<unsigned long long>(f->file_size),
H
hans@chromium.org 已提交
1911 1912
        status.ToString().c_str(),
        versions_->LevelSummary(&tmp));
1913
    versions_->ReleaseCompactionFiles(c.get(), status);
1914
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1915
  } else {
1916
    MaybeScheduleFlushOrCompaction(); // do more compaction work in parallel.
1917
    CompactionState* compact = new CompactionState(c.get());
I
Igor Canadi 已提交
1918
    status = DoCompactionWork(compact, deletion_state);
1919
    CleanupCompaction(compact, status);
1920
    versions_->ReleaseCompactionFiles(c.get(), status);
1921
    c->ReleaseInputs();
1922
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1923
  }
1924
  c.reset();
J
jorlow@chromium.org 已提交
1925 1926 1927 1928 1929 1930

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
1931
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1932 1933 1934 1935 1936
        "Compaction error: %s", status.ToString().c_str());
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
1937 1938

  if (is_manual) {
G
Gabor Cselle 已提交
1939
    ManualCompaction* m = manual_compaction_;
1940 1941 1942
    if (!status.ok()) {
      m->done = true;
    }
1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954
    // 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.
    if (options_.compaction_style == kCompactionStyleUniversal) {
      m->done = true;
    }
G
Gabor Cselle 已提交
1955 1956 1957 1958 1959 1960
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
      m->tmp_storage = manual_end;
      m->begin = &m->tmp_storage;
    }
1961
    m->in_progress = false; // not being processed anymore
1962
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1963
  }
1964
  return status;
J
jorlow@chromium.org 已提交
1965 1966
}

1967
void DBImpl::CleanupCompaction(CompactionState* compact, Status status) {
J
jorlow@chromium.org 已提交
1968
  mutex_.AssertHeld();
1969
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
1970 1971
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
1972
    compact->builder.reset();
J
jorlow@chromium.org 已提交
1973
  } else {
1974
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
1975
  }
D
dgrogan@chromium.org 已提交
1976
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1977 1978
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
1979 1980 1981 1982 1983 1984

    // If this file was inserted into the table cache then remove
    // them here because this compaction was not committed.
    if (!status.ok()) {
      table_cache_->Evict(out.number);
    }
J
jorlow@chromium.org 已提交
1985 1986 1987 1988
  }
  delete compact;
}

1989 1990 1991 1992 1993
// Allocate the file numbers for the output file. We allocate as
// many output file numbers as there are files in level+1.
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
1994 1995
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
1996
  int filesNeeded = compact->compaction->num_input_files(1);
1997
  for (int i = 0; i < filesNeeded; i++) {
1998 1999 2000 2001 2002 2003 2004 2005 2006
    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();
2007
  for (const auto file_number : compact->allocated_file_numbers) {
2008 2009 2010 2011 2012
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
2013
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
2014 2015
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
2016
  uint64_t file_number;
2017 2018 2019 2020 2021 2022 2023
  // 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 已提交
2024 2025 2026 2027 2028
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
2029 2030 2031 2032
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
2033
  out.smallest_seqno = out.largest_seqno = 0;
2034
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
2035 2036 2037

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

J
jorlow@chromium.org 已提交
2040
  if (s.ok()) {
2041 2042 2043
    // Over-estimate slightly so we don't end up just barely crossing
    // the threshold.
    compact->outfile->SetPreallocationBlockSize(
2044
      1.1 * versions_->MaxFileSizeForLevel(compact->compaction->output_level()));
2045

S
Siying Dong 已提交
2046 2047 2048 2049
    CompressionType compression_type = GetCompressionType(
        options_, compact->compaction->output_level(),
        compact->compaction->enable_compression());

S
Siying Dong 已提交
2050
    compact->builder.reset(
S
Siying Dong 已提交
2051
        GetTableBuilder(options_, compact->outfile.get(), compression_type));
J
jorlow@chromium.org 已提交
2052
  }
I
Igor Canadi 已提交
2053
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2054 2055 2056 2057 2058
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
2059
  assert(compact != nullptr);
2060
  assert(compact->outfile);
2061
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076

  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;
2077
  compact->builder.reset();
J
jorlow@chromium.org 已提交
2078 2079

  // Finish and check for file errors
2080
  if (s.ok() && !options_.disableDataSync) {
2081
    if (options_.use_fsync) {
2082
      StopWatch sw(env_, options_.statistics.get(),
2083
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2084 2085
      s = compact->outfile->Fsync();
    } else {
2086
      StopWatch sw(env_, options_.statistics.get(),
2087
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2088 2089
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
2090 2091 2092 2093
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
2094
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
2095 2096 2097

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
2098
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
2099
                                               storage_options_,
J
jorlow@chromium.org 已提交
2100 2101
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
2102 2103 2104
    s = iter->status();
    delete iter;
    if (s.ok()) {
2105
      Log(options_.info_log,
K
Kai Liu 已提交
2106 2107 2108 2109
          "Generated table #%lu: %lu keys, %lu bytes",
          (unsigned long) output_number,
          (unsigned long) current_entries,
          (unsigned long) current_bytes);
J
jorlow@chromium.org 已提交
2110 2111 2112 2113 2114 2115 2116 2117
    }
  }
  return s;
}


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
2118 2119 2120 2121 2122

  // 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.
2123
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
2124 2125 2126 2127 2128 2129 2130 2131
    Log(options_.info_log,  "Compaction %d@%d + %d@%d files aborted",
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1);
    return Status::IOError("Compaction input files inconsistent");
  }

2132
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
2133 2134 2135 2136 2137 2138 2139 2140 2141
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1,
      static_cast<long long>(compact->total_bytes));

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
  const int level = compact->compaction->level();
D
dgrogan@chromium.org 已提交
2142
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2143 2144
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
2145 2146
        (options_.compaction_style == kCompactionStyleUniversal) ?
          level : level + 1,
2147 2148
        out.number, out.file_size, out.smallest, out.largest,
        out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2149
  }
2150
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
2151 2152
}

2153 2154 2155 2156 2157 2158 2159 2160
//
// 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(
2161 2162
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
2163
  SequenceNumber prev __attribute__((unused)) = 0;
2164 2165 2166
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
2167
      *prev_snapshot = prev;
2168
      return cur;
2169
    }
2170 2171
    prev = cur; // assignment
    assert(prev);
2172 2173
  }
  Log(options_.info_log,
K
Kai Liu 已提交
2174 2175 2176
      "Looking for seqid %lu but maxseqid is %lu",
      (unsigned long)in,
      (unsigned long)snapshots[snapshots.size()-1]);
2177 2178 2179 2180
  assert(0);
  return 0;
}

I
Igor Canadi 已提交
2181 2182
Status DBImpl::DoCompactionWork(CompactionState* compact,
                                DeletionState& deletion_state) {
2183
  assert(compact);
2184
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions
A
Abhishek Kona 已提交
2185
  Log(options_.info_log,
2186
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
2187 2188 2189
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
2190
      compact->compaction->level() + 1,
2191
      compact->compaction->score(),
2192
      options_.max_background_compactions - bg_compaction_scheduled_);
2193 2194
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
2195
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
2196 2197

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
2198
  assert(compact->builder == nullptr);
2199
  assert(!compact->outfile);
2200 2201 2202

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
H
Haobo Xu 已提交
2203
  SequenceNumber latest_snapshot = 0;
2204 2205 2206 2207 2208
  snapshots_.getAll(compact->existing_snapshots);
  if (compact->existing_snapshots.size() == 0) {
    // optimize for fast path if there are no snapshots
    visible_at_tip = versions_->LastSequence();
    earliest_snapshot = visible_at_tip;
J
jorlow@chromium.org 已提交
2209
  } else {
H
Haobo Xu 已提交
2210
    latest_snapshot = compact->existing_snapshots.back();
2211 2212 2213 2214
    // Add the current seqno as the 'latest' virtual
    // snapshot to the end of this list.
    compact->existing_snapshots.push_back(versions_->LastSequence());
    earliest_snapshot = compact->existing_snapshots[0];
J
jorlow@chromium.org 已提交
2215 2216
  }

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

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

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

2226
  const uint64_t start_micros = env_->NowMicros();
2227
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
2228 2229 2230 2231 2232
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
2233 2234
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
2235
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2236
  std::string compaction_filter_value;
H
Haobo Xu 已提交
2237
  std::vector<char> delete_key; // for compaction filter
2238
  MergeHelper merge(user_comparator(), options_.merge_operator.get(),
2239 2240
                    options_.info_log.get(),
                    false /* internal key corruption is expected */);
2241 2242 2243
  auto compaction_filter = options_.compaction_filter;
  std::unique_ptr<CompactionFilter> compaction_filter_from_factory = nullptr;
  if (!compaction_filter) {
2244 2245 2246
    auto context = compact->GetFilterContext();
    compaction_filter_from_factory =
      options_.compaction_filter_factory->CreateCompactionFilter(context);
2247 2248
    compaction_filter = compaction_filter_from_factory.get();
  }
2249

J
jorlow@chromium.org 已提交
2250
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
2251
    // Prioritize immutable compaction work
2252
    // TODO: remove memtable flush from normal compaction work
2253
    if (imm_.imm_flush_needed.NoBarrier_Load() != nullptr) {
2254
      const uint64_t imm_start = env_->NowMicros();
I
Igor Canadi 已提交
2255
      LogFlush(options_.info_log);
2256
      mutex_.Lock();
2257
      if (imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
I
Igor Canadi 已提交
2258
        FlushMemTableToOutputFile(nullptr, deletion_state);
H
hans@chromium.org 已提交
2259
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
2260 2261 2262 2263 2264
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
2265
    Slice key = input->key();
2266
    Slice value = input->value();
H
Haobo Xu 已提交
2267

2268
    if (compact->compaction->ShouldStopBefore(key) &&
2269
        compact->builder != nullptr) {
2270
      status = FinishCompactionOutputFile(compact, input.get());
2271 2272 2273 2274 2275 2276
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2277
    bool drop = false;
2278
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2279 2280
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2281 2282
      // 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.
J
jorlow@chromium.org 已提交
2283 2284 2285
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2286
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2287 2288 2289 2290 2291 2292 2293 2294
    } else {
      if (!has_current_user_key ||
          user_comparator()->Compare(ikey.user_key,
                                     Slice(current_user_key)) != 0) {
        // First occurrence of this user key
        current_user_key.assign(ikey.user_key.data(), ikey.user_key.size());
        has_current_user_key = true;
        last_sequence_for_key = kMaxSequenceNumber;
2295
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2296 2297

        // apply the compaction filter to the first occurrence of the user key
2298
        if (compaction_filter &&
H
Haobo Xu 已提交
2299 2300 2301 2302 2303 2304 2305 2306 2307 2308
            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();
          bool to_delete =
2309
            compaction_filter->Filter(compact->compaction->level(),
S
Siying Dong 已提交
2310 2311 2312
                                               ikey.user_key, value,
                                               &compaction_filter_value,
                                               &value_changed);
H
Haobo Xu 已提交
2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324
          if (to_delete) {
            // make a copy of the original key
            delete_key.assign(key.data(), key.data() + key.size());
            // convert it to a delete
            UpdateInternalKey(&delete_key[0], delete_key.size(),
                              ikey.sequence, kTypeDeletion);
            // anchor the key again
            key = Slice(&delete_key[0], delete_key.size());
            // needed because ikey is backed by key
            ParseInternalKey(key, &ikey);
            // no value associated with delete
            value.clear();
2325
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2326 2327 2328 2329 2330
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }

J
jorlow@chromium.org 已提交
2331 2332
      }

2333 2334 2335
      // 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.
2336 2337 2338 2339 2340 2341
      SequenceNumber prev_snapshot = 0; // 0 means no previous snapshot
      SequenceNumber visible = visible_at_tip ?
        visible_at_tip :
        findEarliestVisibleSnapshot(ikey.sequence,
                                    compact->existing_snapshots,
                                    &prev_snapshot);
2342 2343 2344 2345 2346

      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 已提交
2347
        // Hidden by an newer entry for same user key
2348
        // TODO: why not > ?
2349
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2350
        drop = true;    // (A)
2351
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2352
      } else if (ikey.type == kTypeDeletion &&
2353
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
2354 2355 2356 2357 2358 2359 2360 2361 2362
                 compact->compaction->IsBaseLevelForKey(ikey.user_key)) {
        // For this user key:
        // (1) there is no data in higher levels
        // (2) data in lower levels will have larger sequence numbers
        // (3) data in layers that are being compacted here and have
        //     smaller sequence numbers will be dropped in the next
        //     few iterations of this loop (by rule (A) above).
        // Therefore this deletion marker is obsolete and can be dropped.
        drop = true;
2363
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2364 2365 2366 2367 2368 2369 2370
      } 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.
M
Mayank Agarwal 已提交
2371
        merge.MergeUntil(input.get(), prev_snapshot, bottommost_level,
2372
                         options_.statistics.get());
2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390
        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 已提交
2391 2392 2393
      }

      last_sequence_for_key = ikey.sequence;
2394
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
2395 2396
    }
#if 0
2397
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
2398
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
2399
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
2400
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
2401
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
2402
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
2403 2404
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
2405 2406 2407
#endif

    if (!drop) {
2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423
      // 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);
2424
      }
2425

2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436
      // 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.
2437 2438
        if (options_.compaction_style == kCompactionStyleLevel &&
            bottommost_level && ikey.sequence < earliest_snapshot &&
2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449
            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
2450

2451 2452 2453 2454 2455 2456 2457
        // Open output file if necessary
        if (compact->builder == nullptr) {
          status = OpenCompactionOutputFile(compact);
          if (!status.ok()) {
            break;
          }
        }
2458 2459

        SequenceNumber seqno = GetInternalKeySeqno(newkey);
2460 2461
        if (compact->builder->NumEntries() == 0) {
          compact->current_output()->smallest.DecodeFrom(newkey);
2462 2463 2464 2465
          compact->current_output()->smallest_seqno = seqno;
        } else {
          compact->current_output()->smallest_seqno =
            std::min(compact->current_output()->smallest_seqno, seqno);
2466 2467 2468
        }
        compact->current_output()->largest.DecodeFrom(newkey);
        compact->builder->Add(newkey, value);
2469 2470
        compact->current_output()->largest_seqno =
          std::max(compact->current_output()->largest_seqno, seqno);
2471 2472 2473 2474 2475 2476 2477 2478

        // Close output file if it is big enough
        if (compact->builder->FileSize() >=
            compact->compaction->MaxOutputFileSize()) {
          status = FinishCompactionOutputFile(compact, input.get());
          if (!status.ok()) {
            break;
          }
J
jorlow@chromium.org 已提交
2479 2480
        }

2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500
        // 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 已提交
2501 2502 2503 2504 2505
          break;
        }
      }
    }

2506
    // MergeUntil has moved input to the next entry
2507
    if (!current_entry_is_merging) {
2508 2509
      input->Next();
    }
J
jorlow@chromium.org 已提交
2510 2511 2512
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
2513
    status = Status::IOError("Database shutdown started during compaction");
J
jorlow@chromium.org 已提交
2514
  }
2515
  if (status.ok() && compact->builder != nullptr) {
2516
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
2517 2518 2519 2520
  }
  if (status.ok()) {
    status = input->status();
  }
2521
  input.reset();
J
jorlow@chromium.org 已提交
2522

2523 2524
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
2525 2526
  if (options_.statistics.get()) {
    options_.statistics.get()->measureTime(COMPACTION_TIME, stats.micros);
A
Abhishek Kona 已提交
2527
  }
M
Mark Callaghan 已提交
2528 2529
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
2530 2531

  int num_output_files = compact->outputs.size();
2532
  if (compact->builder != nullptr) {
P
Pascal Borreli 已提交
2533
    // An error occurred so ignore the last output.
2534 2535 2536 2537
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
2538

2539
  for (int i = 0; i < compact->compaction->num_input_files(0); i++) {
M
Mark Callaghan 已提交
2540
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;
2541 2542 2543
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(0, i)->file_size);
  }
M
Mark Callaghan 已提交
2544

2545
  for (int i = 0; i < compact->compaction->num_input_files(1); i++) {
M
Mark Callaghan 已提交
2546
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;
2547 2548 2549
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(1, i)->file_size);
  }
M
Mark Callaghan 已提交
2550

2551
  for (int i = 0; i < num_output_files; i++) {
2552
    stats.bytes_written += compact->outputs[i].file_size;
2553 2554
    RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
               compact->outputs[i].file_size);
2555 2556
  }

I
Igor Canadi 已提交
2557
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2558
  mutex_.Lock();
2559
  stats_[compact->compaction->output_level()].Add(stats);
J
jorlow@chromium.org 已提交
2560

2561 2562 2563 2564
  // 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 已提交
2565 2566
  if (status.ok()) {
    status = InstallCompactionResults(compact);
I
Igor Canadi 已提交
2567
    InstallSuperVersion(deletion_state);
J
jorlow@chromium.org 已提交
2568
  }
2569
  VersionSet::LevelSummaryStorage tmp;
2570
  Log(options_.info_log,
M
Mark Callaghan 已提交
2571
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
2572 2573
      "MB in(%.1f, %.1f) out(%.1f), read-write-amplify(%.1f) "
      "write-amplify(%.1f) %s\n",
M
Mark Callaghan 已提交
2574 2575 2576
      versions_->LevelSummary(&tmp),
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
          (double) stats.micros,
2577
      compact->compaction->output_level(),
M
Mark Callaghan 已提交
2578 2579 2580 2581
      stats.files_in_leveln, stats.files_in_levelnp1, stats.files_out_levelnp1,
      stats.bytes_readn / 1048576.0,
      stats.bytes_readnp1 / 1048576.0,
      stats.bytes_written / 1048576.0,
2582
      (stats.bytes_written + stats.bytes_readnp1 + stats.bytes_readn) /
2583
          (double) stats.bytes_readn,
2584
      stats.bytes_written / (double) stats.bytes_readn,
2585
      status.ToString().c_str());
M
Mark Callaghan 已提交
2586

J
jorlow@chromium.org 已提交
2587 2588 2589
  return status;
}

2590 2591 2592 2593
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
K
Kai Liu 已提交
2594
  autovector<MemTable*> mem; // includes both mem_ and imm_
I
Igor Canadi 已提交
2595
  DBImpl *db;
2596 2597 2598 2599
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
K
Kai Liu 已提交
2600
  DBImpl::DeletionState deletion_state;
2601
  state->mu->Lock();
K
Kai Liu 已提交
2602 2603
  auto mems_size = state->mem.size();
  for (size_t i = 0; i < mems_size; i++) {
2604 2605
    MemTable* m = state->mem[i]->Unref();
    if (m != nullptr) {
2606
      deletion_state.memtables_to_free.push_back(m);
2607
    }
2608
  }
2609 2610 2611 2612
  if (state->version->Unref()) {
    // fast path FindObsoleteFiles
    state->db->FindObsoleteFiles(deletion_state, false, true);
  }
2613
  state->mu->Unlock();
I
Igor Canadi 已提交
2614
  state->db->PurgeObsoleteFiles(deletion_state);
2615 2616
  delete state;
}
H
Hans Wennborg 已提交
2617
}  // namespace
2618

J
jorlow@chromium.org 已提交
2619 2620
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
2621
  IterState* cleanup = new IterState;
2622
  MemTable* mutable_mem;
K
Kai Liu 已提交
2623
  autovector<MemTable*> immutables;
2624
  Version* version;
J
jorlow@chromium.org 已提交
2625

2626
  // Collect together all needed child iterators for mem
2627 2628
  mutex_.Lock();
  *latest_snapshot = versions_->LastSequence();
2629
  mem_->Ref();
2630
  mutable_mem = mem_;
2631 2632 2633
  // Collect together all needed child iterators for imm_
  imm_.GetMemTables(&immutables);
  for (unsigned int i = 0; i < immutables.size(); i++) {
2634 2635 2636 2637 2638 2639 2640
    immutables[i]->Ref();
  }
  // Collect iterators for files in L0 - Ln
  versions_->current()->Ref();
  version = versions_->current();
  mutex_.Unlock();

K
Kai Liu 已提交
2641 2642
  std::vector<Iterator*> memtables;
  memtables.push_back(mutable_mem->NewIterator(options));
2643 2644
  cleanup->mem.push_back(mutable_mem);
  for (MemTable* m : immutables) {
K
Kai Liu 已提交
2645
    memtables.push_back(m->NewIterator(options));
2646
    cleanup->mem.push_back(m);
2647
  }
K
Kai Liu 已提交
2648 2649 2650 2651
  version->AddIterators(options, storage_options_, &memtables);
  Iterator* internal_iter = NewMergingIterator(
      env_, &internal_comparator_, memtables.data(), memtables.size()
  );
2652
  cleanup->version = version;
2653
  cleanup->mu = &mutex_;
I
Igor Canadi 已提交
2654
  cleanup->db = this;
2655
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2656 2657 2658 2659 2660 2661 2662 2663 2664

  return internal_iter;
}

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

J
jorlow@chromium.org 已提交
2665
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
2666 2667 2668 2669
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
2670 2671 2672
Status DBImpl::Get(const ReadOptions& options,
                   const Slice& key,
                   std::string* value) {
2673 2674 2675
  return GetImpl(options, key, value);
}

I
Igor Canadi 已提交
2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713
// DeletionState gets created and destructed outside of the lock -- we
// use this convinently to:
// * malloc one SuperVersion() outside of the lock -- new_superversion
// * delete one SuperVersion() outside of the lock -- superversion_to_free
//
// 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
void DBImpl::InstallSuperVersion(DeletionState& deletion_state) {
  // if new_superversion == nullptr, it means somebody already used it
  SuperVersion* new_superversion =
    (deletion_state.new_superversion != nullptr) ?
    deletion_state.new_superversion : new SuperVersion();
  SuperVersion* old_superversion = InstallSuperVersion(new_superversion);
  deletion_state.new_superversion = nullptr;
  if (deletion_state.superversion_to_free != nullptr) {
    // somebody already put it there
    delete old_superversion;
  } else {
    deletion_state.superversion_to_free = old_superversion;
  }
}

DBImpl::SuperVersion* DBImpl::InstallSuperVersion(
    SuperVersion* new_superversion) {
  mutex_.AssertHeld();
  new_superversion->Init(mem_, imm_, versions_->current());
  SuperVersion* old_superversion = super_version_;
  super_version_ = new_superversion;
  if (old_superversion != nullptr && old_superversion->Unref()) {
    old_superversion->Cleanup();
    return old_superversion; // will let caller delete outside of mutex
  }
  return nullptr;
}

2714 2715 2716
Status DBImpl::GetImpl(const ReadOptions& options,
                       const Slice& key,
                       std::string* value,
2717
                       bool* value_found) {
2718
  Status s;
2719

2720
  StopWatch sw(env_, options_.statistics.get(), DB_GET, false);
2721 2722
  StopWatchNano snapshot_timer(env_, false);
  StartPerfTimer(&snapshot_timer);
2723
  SequenceNumber snapshot;
K
kailiu 已提交
2724

2725
  if (options.snapshot != nullptr) {
2726 2727 2728
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2729
  }
2730

I
Igor Canadi 已提交
2731 2732 2733
  // This can be replaced by using atomics and spinlock instead of big mutex
  mutex_.Lock();
  SuperVersion* get_version = super_version_->Ref();
2734
  mutex_.Unlock();
I
Igor Canadi 已提交
2735

2736
  bool have_stat_update = false;
2737
  Version::GetStats stats;
2738

2739
  // Prepare to store a list of merge operations if merge occurs.
2740
  MergeContext merge_context;
2741

2742
  // First look in the memtable, then in the immutable memtable (if any).
2743
  // s is both in/out. When in, s could either be OK or MergeInProgress.
2744
  // merge_operands will contain the sequence of merges in the latter case.
2745
  LookupKey lkey(key, snapshot);
2746
  BumpPerfTime(&perf_context.get_snapshot_time, &snapshot_timer);
I
Igor Canadi 已提交
2747
  if (get_version->mem->Get(lkey, value, &s, merge_context, options_)) {
2748
    // Done
2749
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
2750
  } else if (get_version->imm.Get(lkey, value, &s, merge_context, options_)) {
2751
    // Done
2752
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
2753
  } else {
2754 2755 2756
    StopWatchNano from_files_timer(env_, false);
    StartPerfTimer(&from_files_timer);

I
Igor Canadi 已提交
2757 2758
    get_version->current->Get(options, lkey, value, &s, &merge_context, &stats,
                              options_, value_found);
2759
    have_stat_update = true;
2760
    BumpPerfTime(&perf_context.get_from_output_files_time, &from_files_timer);
2761
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
2762
  }
2763 2764 2765

  StopWatchNano post_process_timer(env_, false);
  StartPerfTimer(&post_process_timer);
2766

I
Igor Canadi 已提交
2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787
  bool delete_get_version = false;
  if (!options_.disable_seek_compaction && have_stat_update) {
    mutex_.Lock();
    if (get_version->current->UpdateStats(stats)) {
      MaybeScheduleFlushOrCompaction();
    }
    if (get_version->Unref()) {
      get_version->Cleanup();
      delete_get_version = true;
    }
    mutex_.Unlock();
  } else {
    if (get_version->Unref()) {
      mutex_.Lock();
      get_version->Cleanup();
      mutex_.Unlock();
      delete_get_version = true;
    }
  }
  if (delete_get_version) {
    delete get_version;
2788
  }
2789

2790
  // Note, tickers are atomic now - no lock protection needed any more.
2791

2792 2793
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
K
Kai Liu 已提交
2794
  BumpPerfTime(&perf_context.get_post_process_time, &post_process_timer);
2795
  return s;
J
jorlow@chromium.org 已提交
2796 2797
}

2798 2799 2800
std::vector<Status> DBImpl::MultiGet(const ReadOptions& options,
                                     const std::vector<Slice>& keys,
                                     std::vector<std::string>* values) {
2801
  StopWatch sw(env_, options_.statistics.get(), DB_MULTIGET, false);
2802 2803
  StopWatchNano snapshot_timer(env_, false);
  StartPerfTimer(&snapshot_timer);
K
Kai Liu 已提交
2804

2805
  SequenceNumber snapshot;
K
Kai Liu 已提交
2806
  autovector<MemTable*> to_delete;
2807

2808
  mutex_.Lock();
2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }

  MemTable* mem = mem_;
  MemTableList imm = imm_;
  Version* current = versions_->current();
  mem->Ref();
  imm.RefAll();
  current->Ref();

  // Unlock while reading from files and memtables

  mutex_.Unlock();
  bool have_stat_update = false;
  Version::GetStats stats;

2828 2829
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
2830

2831 2832 2833 2834 2835 2836 2837
  // Note: this always resizes the values array
  int numKeys = keys.size();
  std::vector<Status> statList(numKeys);
  values->resize(numKeys);

  // Keep track of bytes that we read for statistics-recording later
  uint64_t bytesRead = 0;
2838
  BumpPerfTime(&perf_context.get_snapshot_time, &snapshot_timer);
2839 2840 2841 2842

  // 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.
2843 2844
  // merge_operands will contain the sequence of merges in the latter case.
  for (int i=0; i<numKeys; ++i) {
2845
    merge_context.Clear();
2846 2847 2848 2849
    Status& s = statList[i];
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
2850
    if (mem->Get(lkey, value, &s, merge_context, options_)) {
2851
      // Done
2852
    } else if (imm.Get(lkey, value, &s, merge_context, options_)) {
2853 2854
      // Done
    } else {
2855
      current->Get(options, lkey, value, &s, &merge_context, &stats, options_);
2856 2857 2858 2859 2860 2861 2862 2863 2864
      have_stat_update = true;
    }

    if (s.ok()) {
      bytesRead += value->size();
    }
  }

  // Post processing (decrement reference counts and record statistics)
2865 2866
  StopWatchNano post_process_timer(env_, false);
  StartPerfTimer(&post_process_timer);
2867 2868 2869
  mutex_.Lock();
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
2870
    MaybeScheduleFlushOrCompaction();
2871
  }
2872 2873
  MemTable* m = mem->Unref();
  imm.UnrefAll(&to_delete);
2874
  current->Unref();
2875 2876
  mutex_.Unlock();

2877 2878 2879 2880
  // free up all obsolete memtables outside the mutex
  delete m;
  for (MemTable* v: to_delete) delete v;

2881 2882 2883
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_CALLS);
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_KEYS_READ, numKeys);
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_BYTES_READ, bytesRead);
2884
  BumpPerfTime(&perf_context.get_post_process_time, &post_process_timer);
2885 2886 2887 2888

  return statList;
}

2889 2890 2891 2892 2893
bool DBImpl::KeyMayExist(const ReadOptions& options,
                         const Slice& key,
                         std::string* value,
                         bool* value_found) {
  if (value_found != nullptr) {
K
Kai Liu 已提交
2894 2895
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
2896
  }
2897 2898
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
K
Kai Liu 已提交
2899 2900 2901 2902 2903 2904
  auto s = GetImpl(roptions, key, value, value_found);

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

J
jorlow@chromium.org 已提交
2907 2908
Iterator* DBImpl::NewIterator(const ReadOptions& options) {
  SequenceNumber latest_snapshot;
T
Tyler Harter 已提交
2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921
  Iterator* iter = NewInternalIterator(options, &latest_snapshot);
  iter = NewDBIterator(
             &dbname_, env_, options_, user_comparator(), iter,
             (options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
              : latest_snapshot));
  if (options.prefix) {
    // use extra wrapper to exclude any keys from the results which
    // don't begin with the prefix
    iter = new PrefixFilterIterator(iter, *options.prefix,
                                    options_.prefix_extractor);
  }
  return iter;
J
jorlow@chromium.org 已提交
2922 2923 2924 2925
}

const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
2926
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
2927 2928 2929 2930
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
2931
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
2932 2933 2934 2935 2936 2937 2938
}

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

2939 2940 2941 2942 2943 2944 2945 2946 2947
Status DBImpl::Merge(const WriteOptions& o, const Slice& key,
                     const Slice& val) {
  if (!options_.merge_operator) {
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
    return DB::Merge(o, key, val);
  }
}

J
jorlow@chromium.org 已提交
2948 2949 2950 2951
Status DBImpl::Delete(const WriteOptions& options, const Slice& key) {
  return DB::Delete(options, key);
}

2952
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
2953 2954
  StopWatchNano pre_post_process_timer(env_, false);
  StartPerfTimer(&pre_post_process_timer);
2955 2956 2957
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
2958
  w.disableWAL = options.disableWAL;
2959
  w.done = false;
2960

2961
  StopWatch sw(env_, options_.statistics.get(), DB_WRITE, false);
I
Igor Canadi 已提交
2962
  mutex_.Lock();
2963 2964 2965 2966
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
2967 2968 2969 2970 2971

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

2972
  if (w.done) {
I
Igor Canadi 已提交
2973
    mutex_.Unlock();
2974
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
2975
    return w.status;
2976 2977
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
2978 2979 2980
  }

  // May temporarily unlock and wait.
I
Igor Canadi 已提交
2981 2982
  SuperVersion* superversion_to_free = nullptr;
  Status status = MakeRoomForWrite(my_batch == nullptr, &superversion_to_free);
D
dgrogan@chromium.org 已提交
2983
  uint64_t last_sequence = versions_->LastSequence();
2984
  Writer* last_writer = &w;
2985
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
2986 2987 2988 2989 2990
    // TODO: BuildBatchGroup physically concatenate/copy all write batches into
    // a new one. Mem copy is done with the lock held. Ideally, we only need
    // the lock to obtain the last_writer and the references to all batches.
    // Creation (copy) of the merged batch could have been done outside of the
    // lock protected region.
2991
    WriteBatch* updates = BuildBatchGroup(&last_writer);
2992

2993 2994 2995 2996
    // Add to log and apply to memtable.  We can release the lock
    // during this phase since &w is currently responsible for logging
    // and protects against concurrent loggers and concurrent writes
    // into mem_.
2997
    {
2998
      mutex_.Unlock();
2999 3000 3001 3002 3003
      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
3004 3005 3006
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3007 3008
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3009 3010
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3011
      }
3012 3013
      BumpPerfTime(&perf_context.write_pre_and_post_process_time,
                   &pre_post_process_timer);
3014 3015

      if (!options.disableWAL) {
3016 3017
        StopWatchNano timer(env_);
        StartPerfTimer(&timer);
3018 3019 3020 3021
        Slice log_entry = WriteBatchInternal::Contents(updates);
        status = log_->AddRecord(log_entry);
        RecordTick(options_.statistics.get(), WAL_FILE_SYNCED, 1);
        RecordTick(options_.statistics.get(), WAL_FILE_BYTES, log_entry.size());
H
heyongqiang 已提交
3022
        if (status.ok() && options.sync) {
3023
          if (options_.use_fsync) {
3024
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3025
            status = log_->file()->Fsync();
3026
          } else {
3027
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3028
            status = log_->file()->Sync();
3029
          }
H
heyongqiang 已提交
3030
        }
3031
        BumpPerfTime(&perf_context.write_wal_time, &timer);
3032 3033
      }
      if (status.ok()) {
3034 3035
        StopWatchNano write_memtable_timer(env_, false);
        StartPerfTimer(&write_memtable_timer);
3036 3037
        status = WriteBatchInternal::InsertInto(updates, mem_, &options_, this,
                                                options_.filter_deletes);
3038
        BumpPerfTime(&perf_context.write_memtable_time, &write_memtable_timer);
3039 3040 3041 3042 3043 3044 3045
        if (!status.ok()) {
          // Panic for in-memory corruptions
          // 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.
          throw std::runtime_error("In memory WriteBatch corruption!");
        }
3046 3047
        SetTickerCount(options_.statistics.get(),
                       SEQUENCE_NUMBER, last_sequence);
3048
      }
3049
      StartPerfTimer(&pre_post_process_timer);
3050 3051
      mutex_.Lock();
      if (status.ok()) {
3052
        versions_->SetLastSequence(last_sequence);
3053
      }
J
jorlow@chromium.org 已提交
3054
    }
3055
    if (updates == &tmp_batch_) tmp_batch_.Clear();
J
jorlow@chromium.org 已提交
3056
  }
I
Igor Canadi 已提交
3057 3058 3059
  if (options_.paranoid_checks && !status.ok() && bg_error_.ok()) {
    bg_error_ = status; // stop compaction & fail any further writes
  }
3060

3061 3062 3063 3064 3065 3066 3067
  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
3068
    }
3069 3070
    if (ready == last_writer) break;
  }
3071

3072 3073 3074
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
3075
  }
I
Igor Canadi 已提交
3076 3077
  mutex_.Unlock();
  delete superversion_to_free;
3078 3079
  BumpPerfTime(&perf_context.write_pre_and_post_process_time,
               &pre_post_process_timer);
J
jorlow@chromium.org 已提交
3080 3081 3082
  return status;
}

3083
// REQUIRES: Writer list must be non-empty
3084
// REQUIRES: First writer must have a non-nullptr batch
3085 3086 3087 3088
WriteBatch* DBImpl::BuildBatchGroup(Writer** last_writer) {
  assert(!writers_.empty());
  Writer* first = writers_.front();
  WriteBatch* result = first->batch;
3089
  assert(result != nullptr);
3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110

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

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

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

H
heyongqiang 已提交
3111 3112 3113 3114 3115 3116
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

3117
    if (w->batch != nullptr) {
3118 3119 3120 3121 3122 3123 3124 3125 3126
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

      // Append to *reuslt
      if (result == first->batch) {
        // Switch to temporary batch instead of disturbing caller's batch
3127
        result = &tmp_batch_;
3128 3129 3130 3131 3132 3133 3134 3135 3136 3137
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

3138 3139 3140
// 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 已提交
3141 3142 3143 3144
// if n < bottom, return 0;
// if n >= top, return 1000;
// otherwise, let r = (n - bottom) /
//                    (top - bottom)
3145 3146 3147 3148
//  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.
J
Jim Paton 已提交
3149
uint64_t DBImpl::SlowdownAmount(int n, int top, int bottom) {
3150
  uint64_t delay;
J
Jim Paton 已提交
3151
  if (n >= top) {
3152 3153
    delay = 1000;
  }
J
Jim Paton 已提交
3154
  else if (n < bottom) {
3155 3156 3157 3158
    delay = 0;
  }
  else {
    // If we are here, we know that:
J
Jim Paton 已提交
3159
    //   level0_start_slowdown <= n < level0_slowdown
3160 3161
    // since the previous two conditions are false.
    float how_much =
J
Jim Paton 已提交
3162 3163
      (float) (n - bottom) /
              (top - bottom);
3164 3165 3166 3167 3168 3169
    delay = how_much * how_much * 1000;
  }
  assert(delay <= 1000);
  return delay;
}

3170
// REQUIRES: mutex_ is held
3171
// REQUIRES: this thread is currently at the front of the writer queue
I
Igor Canadi 已提交
3172 3173
Status DBImpl::MakeRoomForWrite(bool force,
                                SuperVersion** superversion_to_free) {
3174
  mutex_.AssertHeld();
3175
  assert(!writers_.empty());
3176
  bool allow_delay = !force;
J
Jim Paton 已提交
3177 3178
  bool allow_hard_rate_limit_delay = !force;
  bool allow_soft_rate_limit_delay = !force;
3179
  uint64_t rate_limit_delay_millis = 0;
3180
  Status s;
3181
  double score;
I
Igor Canadi 已提交
3182
  *superversion_to_free = nullptr;
3183

3184 3185 3186 3187 3188
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
3189 3190
    } else if (
        allow_delay &&
3191
        versions_->NumLevelFiles(0) >=
3192
          options_.level0_slowdown_writes_trigger) {
3193 3194 3195
      // 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
3196
      // individual write by 0-1ms to reduce latency variance.  Also,
3197 3198 3199
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
      mutex_.Unlock();
3200
      uint64_t delayed;
J
Jim Paton 已提交
3201
      {
3202
        StopWatch sw(env_, options_.statistics.get(), STALL_L0_SLOWDOWN_COUNT);
J
Jim Paton 已提交
3203 3204 3205 3206 3207
        env_->SleepForMicroseconds(
          SlowdownAmount(versions_->NumLevelFiles(0),
                         options_.level0_slowdown_writes_trigger,
                         options_.level0_stop_writes_trigger)
        );
3208
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3209
      }
3210
      RecordTick(options_.statistics.get(), STALL_L0_SLOWDOWN_MICROS, delayed);
3211
      stall_level0_slowdown_ += delayed;
J
Jim Paton 已提交
3212
      stall_level0_slowdown_count_++;
3213 3214
      allow_delay = false;  // Do not delay a single write more than once
      mutex_.Lock();
3215
      delayed_writes_++;
3216 3217 3218
    } else if (!force &&
               (mem_->ApproximateMemoryUsage() <= options_.write_buffer_size)) {
      // There is room in current memtable
3219 3220 3221
      if (allow_delay) {
        DelayLoggingAndReset();
      }
3222
      break;
3223
    } else if (imm_.size() == options_.max_write_buffer_number - 1) {
3224
      // We have filled up the current memtable, but the previous
3225 3226
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
3227
      Log(options_.info_log, "wait for memtable compaction...\n");
3228
      uint64_t stall;
J
Jim Paton 已提交
3229
      {
3230
        StopWatch sw(env_, options_.statistics.get(),
J
Jim Paton 已提交
3231 3232
          STALL_MEMTABLE_COMPACTION_COUNT);
        bg_cv_.Wait();
3233
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3234
      }
3235 3236
      RecordTick(options_.statistics.get(),
                 STALL_MEMTABLE_COMPACTION_MICROS, stall);
3237
      stall_memtable_compaction_ += stall;
J
Jim Paton 已提交
3238
      stall_memtable_compaction_count_++;
3239
    } else if (versions_->NumLevelFiles(0) >=
3240
               options_.level0_stop_writes_trigger) {
3241
      // There are too many level-0 files.
3242 3243
      DelayLoggingAndReset();
      Log(options_.info_log, "wait for fewer level0 files...\n");
3244
      uint64_t stall;
J
Jim Paton 已提交
3245
      {
3246 3247
        StopWatch sw(env_, options_.statistics.get(),
                     STALL_L0_NUM_FILES_COUNT);
J
Jim Paton 已提交
3248
        bg_cv_.Wait();
3249
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3250
      }
3251
      RecordTick(options_.statistics.get(), STALL_L0_NUM_FILES_MICROS, stall);
3252
      stall_level0_num_files_ += stall;
J
Jim Paton 已提交
3253
      stall_level0_num_files_count_++;
3254
    } else if (
J
Jim Paton 已提交
3255 3256 3257
        allow_hard_rate_limit_delay &&
        options_.hard_rate_limit > 1.0 &&
        (score = versions_->MaxCompactionScore()) > options_.hard_rate_limit) {
3258
      // Delay a write when the compaction score for any level is too large.
3259
      int max_level = versions_->MaxCompactionScoreLevel();
3260
      mutex_.Unlock();
3261
      uint64_t delayed;
J
Jim Paton 已提交
3262
      {
3263 3264
        StopWatch sw(env_, options_.statistics.get(),
                     HARD_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3265
        env_->SleepForMicroseconds(1000);
3266
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3267
      }
3268
      stall_leveln_slowdown_[max_level] += delayed;
J
Jim Paton 已提交
3269
      stall_leveln_slowdown_count_[max_level]++;
3270
      // Make sure the following value doesn't round to zero.
3271 3272
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
3273 3274
      RecordTick(options_.statistics.get(),
                 RATE_LIMIT_DELAY_MILLIS, rate_limit);
J
Jim Paton 已提交
3275 3276 3277 3278
      if (options_.rate_limit_delay_max_milliseconds > 0 &&
          rate_limit_delay_millis >=
          (unsigned)options_.rate_limit_delay_max_milliseconds) {
        allow_hard_rate_limit_delay = false;
3279
      }
3280
      mutex_.Lock();
J
Jim Paton 已提交
3281 3282 3283 3284 3285 3286 3287
    } else if (
        allow_soft_rate_limit_delay &&
        options_.soft_rate_limit > 0.0 &&
        (score = versions_->MaxCompactionScore()) > options_.soft_rate_limit) {
      // Delay a write when the compaction score for any level is too large.
      // TODO: add statistics
      mutex_.Unlock();
J
Jim Paton 已提交
3288
      {
3289 3290
        StopWatch sw(env_, options_.statistics.get(),
                     SOFT_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3291 3292 3293 3294 3295 3296 3297
        env_->SleepForMicroseconds(SlowdownAmount(
          score,
          options_.soft_rate_limit,
          options_.hard_rate_limit)
        );
        rate_limit_delay_millis += sw.ElapsedMicros();
      }
J
Jim Paton 已提交
3298 3299
      allow_soft_rate_limit_delay = false;
      mutex_.Lock();
3300

3301
    } else {
3302
      unique_ptr<WritableFile> lfile;
K
kailiu 已提交
3303
      MemTable* new_mem = nullptr;
3304 3305 3306

      // Attempt to switch to a new memtable and trigger compaction of old.
      // Do this without holding the dbmutex lock.
3307 3308
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
I
Igor Canadi 已提交
3309
      SuperVersion* new_superversion = nullptr;
3310 3311 3312 3313 3314 3315
      mutex_.Unlock();
      {
        EnvOptions soptions(storage_options_);
        soptions.use_mmap_writes = false;
        DelayLoggingAndReset();
        s = env_->NewWritableFile(
3316
            LogFileName(options_.wal_dir, new_log_number),
3317 3318 3319
            &lfile,
            soptions
          );
3320 3321 3322 3323
        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 * options_.write_buffer_size);
K
kailiu 已提交
3324
          new_mem = new MemTable(
3325
            internal_comparator_, mem_rep_factory_, NumberLevels(), options_);
K
Kai Liu 已提交
3326
          new_superversion = new SuperVersion();
3327 3328 3329
        }
      }
      mutex_.Lock();
3330
      if (!s.ok()) {
H
heyongqiang 已提交
3331
        // Avoid chewing through file number space in a tight loop.
3332
        versions_->ReuseFileNumber(new_log_number);
K
kailiu 已提交
3333
        assert (!new_mem);
3334 3335
        break;
      }
3336
      logfile_number_ = new_log_number;
3337
      log_.reset(new log::Writer(std::move(lfile)));
3338
      mem_->SetNextLogNumber(logfile_number_);
3339
      imm_.Add(mem_);
3340 3341 3342
      if (force) {
        imm_.FlushRequested();
      }
K
kailiu 已提交
3343
      mem_ = new_mem;
3344
      mem_->Ref();
3345
      Log(options_.info_log,
K
Kai Liu 已提交
3346 3347
          "New memtable created with log file: #%lu\n",
          (unsigned long)logfile_number_);
3348
      mem_->SetLogNumber(logfile_number_);
3349
      force = false;   // Do not force another compaction if have room
3350
      MaybeScheduleFlushOrCompaction();
I
Igor Canadi 已提交
3351
      *superversion_to_free = InstallSuperVersion(new_superversion);
3352 3353 3354 3355 3356
    }
  }
  return s;
}

I
Igor Canadi 已提交
3357 3358 3359 3360
const std::string& DBImpl::GetName() const {
  return dbname_;
}

3361 3362 3363 3364
Env* DBImpl::GetEnv() const {
  return env_;
}

I
Igor Canadi 已提交
3365 3366 3367 3368
const Options& DBImpl::GetOptions() const {
  return options_;
}

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

J
jorlow@chromium.org 已提交
3372 3373
  MutexLock l(&mutex_);
  Slice in = property;
3374
  Slice prefix("rocksdb.");
J
jorlow@chromium.org 已提交
3375 3376 3377 3378 3379 3380 3381
  if (!in.starts_with(prefix)) return false;
  in.remove_prefix(prefix.size());

  if (in.starts_with("num-files-at-level")) {
    in.remove_prefix(strlen("num-files-at-level"));
    uint64_t level;
    bool ok = ConsumeDecimalNumber(&in, &level) && in.empty();
3382
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
3383 3384
      return false;
    } else {
3385
      char buf[100];
D
dgrogan@chromium.org 已提交
3386 3387
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
3388
      *value = buf;
J
jorlow@chromium.org 已提交
3389 3390
      return true;
    }
3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407
  } else if (in == "levelstats") {
    char buf[1000];
    snprintf(buf, sizeof(buf),
             "Level Files Size(MB)\n"
             "--------------------\n");
    value->append(buf);

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

3408
  } else if (in == "stats") {
M
Mark Callaghan 已提交
3409
    char buf[1000];
3410 3411 3412 3413 3414 3415 3416

    uint64_t wal_bytes = 0;
    uint64_t wal_synced = 0;
    uint64_t user_bytes_written = 0;
    uint64_t write_other = 0;
    uint64_t write_self = 0;
    uint64_t write_with_wal = 0;
3417 3418
    uint64_t total_bytes_written = 0;
    uint64_t total_bytes_read = 0;
M
Mark Callaghan 已提交
3419
    uint64_t micros_up = env_->NowMicros() - started_at_;
3420 3421
    // Add "+1" to make sure seconds_up is > 0 and avoid NaN later
    double seconds_up = (micros_up + 1) / 1000000.0;
3422
    uint64_t total_slowdown = 0;
J
Jim Paton 已提交
3423
    uint64_t total_slowdown_count = 0;
3424 3425 3426 3427
    uint64_t interval_bytes_written = 0;
    uint64_t interval_bytes_read = 0;
    uint64_t interval_bytes_new = 0;
    double   interval_seconds_up = 0;
M
Mark Callaghan 已提交
3428

3429 3430 3431 3432 3433 3434 3435 3436 3437 3438
    Statistics* s = options_.statistics.get();
    if (s) {
      wal_bytes = s->getTickerCount(WAL_FILE_BYTES);
      wal_synced = s->getTickerCount(WAL_FILE_SYNCED);
      user_bytes_written = s->getTickerCount(BYTES_WRITTEN);
      write_other = s->getTickerCount(WRITE_DONE_BY_OTHER);
      write_self = s->getTickerCount(WRITE_DONE_BY_SELF);
      write_with_wal = s->getTickerCount(WRITE_WITH_WAL);
    }

M
Mark Callaghan 已提交
3439
    // Pardon the long line but I think it is easier to read this way.
3440 3441
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
3442
             "Level  Files Size(MB) Score Time(sec)  Read(MB) Write(MB)    Rn(MB)  Rnp1(MB)  Wnew(MB) RW-Amplify Read(MB/s) Write(MB/s)      Rn     Rnp1     Wnp1     NewW    Count  Ln-stall Stall-cnt\n"
J
Jim Paton 已提交
3443
             "--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
3444 3445
             );
    value->append(buf);
3446
    for (int level = 0; level < NumberLevels(); level++) {
3447 3448
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
3449 3450 3451 3452 3453 3454
        int64_t bytes_read = stats_[level].bytes_readn +
                             stats_[level].bytes_readnp1;
        int64_t bytes_new = stats_[level].bytes_written -
                            stats_[level].bytes_readnp1;
        double amplify = (stats_[level].bytes_readn == 0)
            ? 0.0
3455 3456 3457
            : (stats_[level].bytes_written +
               stats_[level].bytes_readnp1 +
               stats_[level].bytes_readn) /
M
Mark Callaghan 已提交
3458 3459
                (double) stats_[level].bytes_readn;

3460 3461 3462
        total_bytes_read += bytes_read;
        total_bytes_written += stats_[level].bytes_written;

3463 3464
        snprintf(
            buf, sizeof(buf),
3465
            "%3d %8d %8.0f %5.1f %9.0f %9.0f %9.0f %9.0f %9.0f %9.0f %10.1f %9.1f %11.1f %8d %8d %8d %8d %8d %9.1f %9lu\n",
3466 3467 3468
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
3469
            versions_->NumLevelBytes(level) /
3470
                versions_->MaxBytesForLevel(level),
3471
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
3472 3473 3474 3475 3476 3477
            bytes_read / 1048576.0,
            stats_[level].bytes_written / 1048576.0,
            stats_[level].bytes_readn / 1048576.0,
            stats_[level].bytes_readnp1 / 1048576.0,
            bytes_new / 1048576.0,
            amplify,
3478 3479
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
3480
            (stats_[level].bytes_written / 1048576.0) /
3481
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
3482 3483 3484 3485
            stats_[level].files_in_leveln,
            stats_[level].files_in_levelnp1,
            stats_[level].files_out_levelnp1,
            stats_[level].files_out_levelnp1 - stats_[level].files_in_levelnp1,
3486
            stats_[level].count,
J
Jim Paton 已提交
3487 3488
            stall_leveln_slowdown_[level] / 1000000.0,
            (unsigned long) stall_leveln_slowdown_count_[level]);
3489
        total_slowdown += stall_leveln_slowdown_[level];
J
Jim Paton 已提交
3490
        total_slowdown_count += stall_leveln_slowdown_count_[level];
3491 3492 3493
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
3494

3495 3496 3497 3498
    interval_bytes_new = user_bytes_written - last_stats_.ingest_bytes_;
    interval_bytes_read = total_bytes_read - last_stats_.compaction_bytes_read_;
    interval_bytes_written =
        total_bytes_written - last_stats_.compaction_bytes_written_;
3499 3500 3501 3502 3503 3504
    interval_seconds_up = seconds_up - last_stats_.seconds_up_;

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

3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522
    snprintf(buf, sizeof(buf),
             "Writes cumulative: %llu total, %llu batches, "
             "%.1f per batch, %.2f ingest GB\n",
             (unsigned long long) (write_other + write_self),
             (unsigned long long) write_self,
             (write_other + write_self) / (double) (write_self + 1),
             user_bytes_written / (1048576.0 * 1024));
    value->append(buf);

    snprintf(buf, sizeof(buf),
             "WAL cumulative: %llu WAL writes, %llu WAL syncs, "
             "%.2f writes per sync, %.2f GB written\n",
             (unsigned long long) write_with_wal,
             (unsigned long long ) wal_synced,
             write_with_wal / (double) (wal_synced + 1),
             wal_bytes / (1048576.0 * 1024));
    value->append(buf);

M
Mark Callaghan 已提交
3523
    snprintf(buf, sizeof(buf),
3524 3525
             "Compaction IO cumulative (GB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
3526
             user_bytes_written / (1048576.0 * 1024),
3527 3528 3529 3530 3531 3532 3533 3534
             total_bytes_read / (1048576.0 * 1024),
             total_bytes_written / (1048576.0 * 1024),
             (total_bytes_read + total_bytes_written) / (1048576.0 * 1024));
    value->append(buf);

    snprintf(buf, sizeof(buf),
             "Compaction IO cumulative (MB/sec): "
             "%.1f new, %.1f read, %.1f write, %.1f read+write\n",
3535
             user_bytes_written / 1048576.0 / seconds_up,
3536 3537 3538 3539 3540 3541 3542 3543
             total_bytes_read / 1048576.0 / seconds_up,
             total_bytes_written / 1048576.0 / seconds_up,
             (total_bytes_read + total_bytes_written) / 1048576.0 / seconds_up);
    value->append(buf);

    // +1 to avoid divide by 0 and NaN
    snprintf(buf, sizeof(buf),
             "Amplification cumulative: %.1f write, %.1f compaction\n",
3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575
             (double) (total_bytes_written + wal_bytes)
                 / (user_bytes_written + 1),
             (double) (total_bytes_written + total_bytes_read + wal_bytes)
                 / (user_bytes_written + 1));
    value->append(buf);

    uint64_t interval_write_other = write_other - last_stats_.write_other_;
    uint64_t interval_write_self = write_self - last_stats_.write_self_;

    snprintf(buf, sizeof(buf),
             "Writes interval: %llu total, %llu batches, "
             "%.1f per batch, %.1f ingest MB\n",
             (unsigned long long) (interval_write_other + interval_write_self),
             (unsigned long long) interval_write_self,
             (double) (interval_write_other + interval_write_self)
                 / (interval_write_self + 1),
             (user_bytes_written - last_stats_.ingest_bytes_) /  1048576.0);
    value->append(buf);

    uint64_t interval_write_with_wal =
        write_with_wal - last_stats_.write_with_wal_;

    uint64_t interval_wal_synced = wal_synced - last_stats_.wal_synced_;
    uint64_t interval_wal_bytes = wal_bytes - last_stats_.wal_bytes_;

    snprintf(buf, sizeof(buf),
             "WAL interval: %llu WAL writes, %llu WAL syncs, "
             "%.2f writes per sync, %.2f MB written\n",
             (unsigned long long) interval_write_with_wal,
             (unsigned long long ) interval_wal_synced,
             interval_write_with_wal / (double) (interval_wal_synced + 1),
             interval_wal_bytes / (1048576.0 * 1024));
M
Mark Callaghan 已提交
3576 3577
    value->append(buf);

3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599
    snprintf(buf, sizeof(buf),
             "Compaction IO interval (MB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
             interval_bytes_new / 1048576.0,
             interval_bytes_read/ 1048576.0,
             interval_bytes_written / 1048576.0,
             (interval_bytes_read + interval_bytes_written) / 1048576.0);
    value->append(buf);

    snprintf(buf, sizeof(buf),
             "Compaction IO interval (MB/sec): "
             "%.1f new, %.1f read, %.1f write, %.1f read+write\n",
             interval_bytes_new / 1048576.0 / interval_seconds_up,
             interval_bytes_read / 1048576.0 / interval_seconds_up,
             interval_bytes_written / 1048576.0 / interval_seconds_up,
             (interval_bytes_read + interval_bytes_written)
                 / 1048576.0 / interval_seconds_up);
    value->append(buf);

    // +1 to avoid divide by 0 and NaN
    snprintf(buf, sizeof(buf),
             "Amplification interval: %.1f write, %.1f compaction\n",
3600 3601 3602 3603
             (double) (interval_bytes_written + wal_bytes)
                 / (interval_bytes_new + 1),
             (double) (interval_bytes_written + interval_bytes_read + wal_bytes)
                 / (interval_bytes_new + 1));
M
Mark Callaghan 已提交
3604 3605 3606 3607
    value->append(buf);

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
3608
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
3609 3610
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
3611
            stall_memtable_compaction_ / 1000000.0,
3612
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
3613 3614
    value->append(buf);

J
Jim Paton 已提交
3615 3616 3617 3618 3619 3620 3621 3622 3623
    snprintf(buf, sizeof(buf),
            "Stalls(count): %lu level0_slowdown, %lu level0_numfiles, "
            "%lu memtable_compaction, %lu leveln_slowdown\n",
            (unsigned long) stall_level0_slowdown_count_,
            (unsigned long) stall_level0_num_files_count_,
            (unsigned long) stall_memtable_compaction_count_,
            (unsigned long) total_slowdown_count);
    value->append(buf);

3624 3625 3626
    last_stats_.compaction_bytes_read_ = total_bytes_read;
    last_stats_.compaction_bytes_written_ = total_bytes_written;
    last_stats_.ingest_bytes_ = user_bytes_written;
3627
    last_stats_.seconds_up_ = seconds_up;
3628 3629 3630 3631 3632
    last_stats_.wal_bytes_ = wal_bytes;
    last_stats_.wal_synced_ = wal_synced;
    last_stats_.write_with_wal_ = write_with_wal;
    last_stats_.write_other_ = write_other;
    last_stats_.write_self_ = write_self;
3633

3634
    return true;
G
Gabor Cselle 已提交
3635 3636 3637
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
3638 3639 3640
  } else if (in == "num-immutable-mem-table") {
    *value = std::to_string(imm_.size());
    return true;
J
jorlow@chromium.org 已提交
3641
  }
3642

J
jorlow@chromium.org 已提交
3643 3644 3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671
  return false;
}

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

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

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

3672 3673 3674 3675 3676 3677 3678
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

3679 3680 3681
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
3682 3683 3684 3685
  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());
3686 3687 3688
    return Status::InvalidArgument("Invalid file name");
  }

3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700 3701 3702
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
      Log(options_.info_log, "DeleteFile %s failed.\n", name.c_str());
      return Status::NotSupported("Delete only supported for archived logs");
    }
    status = env_->DeleteFile(options_.wal_dir + "/" + name.c_str());
    if (!status.ok()) {
      Log(options_.info_log, "DeleteFile %s failed.\n", name.c_str());
    }
    return status;
  }

3703 3704 3705 3706
  int level;
  FileMetaData metadata;
  int maxlevel = NumberLevels();
  VersionEdit edit(maxlevel);
K
Kai Liu 已提交
3707
  DeletionState deletion_state(true);
D
Dhruba Borthakur 已提交
3708 3709 3710 3711
  {
    MutexLock l(&mutex_);
    status = versions_->GetMetadataForFile(number, &level, &metadata);
    if (!status.ok()) {
3712 3713
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
3714 3715 3716
      return Status::InvalidArgument("File not found");
    }
    assert((level > 0) && (level < maxlevel));
3717

D
Dhruba Borthakur 已提交
3718 3719
    // If the file is being compacted no need to delete.
    if (metadata.being_compacted) {
3720
      Log(options_.info_log,
3721
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
3722
      return Status::OK();
3723 3724
    }

D
Dhruba Borthakur 已提交
3725 3726 3727 3728 3729 3730
    // 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.
    for (int i = level + 1; i < maxlevel; i++) {
      if (versions_->NumLevelFiles(i) != 0) {
        Log(options_.info_log,
3731
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
3732 3733 3734 3735 3736
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
    status = versions_->LogAndApply(&edit, &mutex_);
I
Igor Canadi 已提交
3737 3738 3739
    if (status.ok()) {
      InstallSuperVersion(deletion_state);
    }
I
Igor Canadi 已提交
3740
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
3741
  } // lock released here
I
Igor Canadi 已提交
3742
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
3743 3744
  // remove files outside the db-lock
  PurgeObsoleteFiles(deletion_state);
3745 3746 3747
  return status;
}

3748
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData> *metadata) {
3749 3750 3751 3752
  MutexLock l(&mutex_);
  return versions_->GetLiveFilesMetaData(metadata);
}

3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779
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 已提交
3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
Status DB::Put(const WriteOptions& opt, const Slice& key, const Slice& value) {
  WriteBatch batch;
  batch.Put(key, value);
  return Write(opt, &batch);
}

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

3794 3795 3796 3797 3798 3799 3800
Status DB::Merge(const WriteOptions& opt, const Slice& key,
                 const Slice& value) {
  WriteBatch batch;
  batch.Merge(key, value);
  return Write(opt, &batch);
}

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

J
Jim Paton 已提交
3803
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
3804
  *dbptr = nullptr;
H
Haobo Xu 已提交
3805
  EnvOptions soptions;
J
jorlow@chromium.org 已提交
3806

3807
  if (options.block_cache != nullptr && options.no_block_cache) {
3808
    return Status::InvalidArgument(
3809
        "no_block_cache is true while block_cache is not nullptr");
3810
  }
3811

J
jorlow@chromium.org 已提交
3812
  DBImpl* impl = new DBImpl(options, dbname);
3813 3814 3815 3816 3817 3818 3819
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
3820 3821 3822 3823
  if (!s.ok()) {
    delete impl;
    return s;
  }
K
kailiu 已提交
3824
  impl->mutex_.Lock();  // DBImpl::Recover() requires lock being held
3825
  VersionEdit edit(impl->NumberLevels());
3826
  s = impl->Recover(&edit); // Handles create_if_missing, error_if_exists
J
jorlow@chromium.org 已提交
3827
  if (s.ok()) {
3828
    uint64_t new_log_number = impl->versions_->NewFileNumber();
3829
    unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
3830
    soptions.use_mmap_writes = false;
3831 3832 3833 3834 3835
    s = options.env->NewWritableFile(
      LogFileName(impl->options_.wal_dir, new_log_number),
      &lfile,
      soptions
    );
J
jorlow@chromium.org 已提交
3836
    if (s.ok()) {
3837
      lfile->SetPreallocationBlockSize(1.1 * options.write_buffer_size);
3838
      edit.SetLogNumber(new_log_number);
3839
      impl->logfile_number_ = new_log_number;
3840
      impl->log_.reset(new log::Writer(std::move(lfile)));
3841
      s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
J
jorlow@chromium.org 已提交
3842 3843
    }
    if (s.ok()) {
I
Igor Canadi 已提交
3844
      delete impl->InstallSuperVersion(new DBImpl::SuperVersion());
3845
      impl->mem_->SetLogNumber(impl->logfile_number_);
J
jorlow@chromium.org 已提交
3846
      impl->DeleteObsoleteFiles();
3847
      impl->MaybeScheduleFlushOrCompaction();
3848
      impl->MaybeScheduleLogDBDeployStats();
J
jorlow@chromium.org 已提交
3849 3850 3851
    }
  }
  impl->mutex_.Unlock();
3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864

  if (options.compaction_style == kCompactionStyleUniversal) {
    int num_files;
    for (int i = 1; i < impl->NumberLevels(); i++) {
      num_files = impl->versions_->NumLevelFiles(i);
      if (num_files > 0) {
        s = Status::InvalidArgument("Not all files are at level 0. Cannot "
          "open with universal compaction style.");
        break;
      }
    }
  }

J
jorlow@chromium.org 已提交
3865 3866 3867 3868 3869 3870 3871 3872
  if (s.ok()) {
    *dbptr = impl;
  } else {
    delete impl;
  }
  return s;
}

3873 3874 3875
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
3876
Status DestroyDB(const std::string& dbname, const Options& options) {
3877 3878 3879 3880 3881
  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 已提交
3882
  std::vector<std::string> filenames;
3883 3884
  std::vector<std::string> archiveFiles;

3885
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
3886 3887
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
3888 3889 3890 3891 3892 3893 3894

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

J
jorlow@chromium.org 已提交
3896 3897 3898 3899 3900
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
3901 3902
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
3903 3904 3905
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
3906
    for (size_t i = 0; i < filenames.size(); i++) {
3907
      if (ParseFileName(filenames[i], &number, &type) &&
3908
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
3909 3910 3911
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
3912 3913
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
3914 3915 3916
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
3917 3918 3919 3920 3921
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
3922

3923
    env->GetChildren(archivedir, &archiveFiles);
3924 3925
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
3926 3927
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
3928
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
3929 3930 3931 3932 3933
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
3934
    // ignore case where no archival directory is present.
3935
    env->DeleteDir(archivedir);
3936

J
jorlow@chromium.org 已提交
3937
    env->UnlockFile(lock);  // Ignore error since state is already gone
3938
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
3939
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
3940
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
3941 3942 3943 3944
  }
  return result;
}

3945 3946
//
// A global method that can dump out the build version
K
kailiu 已提交
3947
void DumpLeveldbBuildVersion(Logger * log) {
3948
  Log(log, "Git sha %s", rocksdb_build_git_sha);
3949
  Log(log, "Compile time %s %s",
3950
      rocksdb_build_compile_time, rocksdb_build_compile_date);
3951 3952
}

3953
}  // namespace rocksdb