db_impl.cc 136.3 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 23
#include "db/builder.h"
#include "db/dbformat.h"
24
#include "db/db_iter.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"
29
#include "db/memtablelist.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
#include "rocksdb/compaction_filter.h"
#include "rocksdb/db.h"
41
#include "rocksdb/column_family.h"
42 43 44 45
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
#include "rocksdb/statistics.h"
#include "rocksdb/status.h"
S
Siying Dong 已提交
46 47
#include "rocksdb/table.h"
#include "port/port.h"
J
jorlow@chromium.org 已提交
48
#include "table/block.h"
49
#include "table/block_based_table_factory.h"
J
jorlow@chromium.org 已提交
50 51
#include "table/merger.h"
#include "table/two_level_iterator.h"
52 53
#include "util/auto_roll_logger.h"
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
54
#include "util/coding.h"
I
Igor Canadi 已提交
55
#include "util/hash_skiplist_rep.h"
J
jorlow@chromium.org 已提交
56 57
#include "util/logging.h"
#include "util/mutexlock.h"
58
#include "util/perf_context_imp.h"
59
#include "util/stop_watch.h"
J
jorlow@chromium.org 已提交
60

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

63 64
const Slice& default_column_family_name("default");

65 66
void dumpLeveldbBuildVersion(Logger * log);

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

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

J
jorlow@chromium.org 已提交
79 80 81
struct DBImpl::CompactionState {
  Compaction* const compaction;

82 83 84 85 86
  // 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 已提交
87 88 89 90 91 92

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

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

  uint64_t total_bytes;

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

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
110 111 112 113 114 115 116

  // 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 已提交
117 118 119
};

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

X
Xing Jin 已提交
137 138 139 140 141 142
  // 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;
  }

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

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

K
kailiu 已提交
191 192 193 194 195 196
  // -- 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) {
197 198
    assert(collectors[i]);
    collectors[i] =
K
kailiu 已提交
199
      std::make_shared<UserKeyTablePropertiesCollector>(collectors[i]);
200 201 202 203
  }

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

J
jorlow@chromium.org 已提交
207 208 209
  return result;
}

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

274
  mem_->Ref();
275

H
heyongqiang 已提交
276
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
277 278

  stall_leveln_slowdown_.resize(options.num_levels);
J
Jim Paton 已提交
279 280
  stall_leveln_slowdown_count_.resize(options.num_levels);
  for (int i = 0; i < options.num_levels; ++i) {
281
    stall_leveln_slowdown_[i] = 0;
J
Jim Paton 已提交
282 283
    stall_leveln_slowdown_count_[i] = 0;
  }
284

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

290 291
  versions_.reset(new VersionSet(dbname_, &options_, storage_options_,
                                 table_cache_.get(), &internal_comparator_));
292

293 294
  dumpLeveldbBuildVersion(options_.info_log.get());
  options_.Dump(options_.info_log.get());
295

296
  char name[100];
297
  Status st = env_->GetHostName(name, 100L);
298
  if (st.ok()) {
299 300 301 302 303 304
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
305

I
Igor Canadi 已提交
306
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
307 308 309
}

DBImpl::~DBImpl() {
310 311 312
  std::vector<MemTable*> to_delete;
  to_delete.reserve(options_.max_write_buffer_number);

J
jorlow@chromium.org 已提交
313
  // Wait for background work to finish
314
  if (flush_on_destroy_ && mem_->GetFirstSequenceNumber() != 0) {
315 316
    FlushMemTable(FlushOptions());
  }
317
  mutex_.Lock();
318
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
319 320 321
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
322
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
323
  }
I
Igor Canadi 已提交
324 325 326 327 328 329 330
  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 已提交
331 332
  mutex_.Unlock();

333
  if (db_lock_ != nullptr) {
J
jorlow@chromium.org 已提交
334 335 336
    env_->UnlockFile(db_lock_);
  }

337 338 339 340 341 342 343 344
  if (mem_ != nullptr) {
    delete mem_->Unref();
  }

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

A
Abhishek Kona 已提交
348
// Do not flush and close database elegantly. Simulate a crash.
349 350 351 352 353 354
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();
355 356 357
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
358 359
    bg_cv_.Wait();
  }
I
Igor Canadi 已提交
360 361 362 363 364 365 366
  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_;
  }
367 368

  // Prevent new compactions from occuring.
369
  bg_work_gate_closed_ = true;
370 371
  const int LargeNumber = 10000000;
  bg_compaction_scheduled_ += LargeNumber;
372

373
  mutex_.Unlock();
I
Igor Canadi 已提交
374
  LogFlush(options_.info_log);
375 376

  // force release the lock file.
377
  if (db_lock_ != nullptr) {
378 379
    env_->UnlockFile(db_lock_);
  }
380 381 382 383

  log_.reset();
  versions_.reset();
  table_cache_.reset();
384 385
}

A
Abhishek Kona 已提交
386 387 388
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
389

J
jorlow@chromium.org 已提交
390
Status DBImpl::NewDB() {
391
  VersionEdit new_db(NumberLevels());
J
jorlow@chromium.org 已提交
392
  new_db.SetComparatorName(user_comparator()->Name());
393
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
394 395 396 397
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

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

428
const Status DBImpl::CreateArchivalDirectory() {
429
  if (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0) {
430
    std::string archivalPath = ArchivalDirectory(options_.wal_dir);
431 432 433 434 435
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

436
void DBImpl::PrintStatistics() {
437
  auto dbstats = options_.statistics.get();
438 439
  if (dbstats) {
    Log(options_.info_log,
440 441
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
442 443 444
  }
}

445
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
446 447 448 449 450 451 452 453 454 455 456 457 458
  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;
459
    GetProperty("rocksdb.stats", &stats);
H
Haobo Xu 已提交
460
    Log(options_.info_log, "%s", stats.c_str());
461
    PrintStatistics();
462 463 464
  }
}

I
Igor Canadi 已提交
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 497 498 499 500 501 502 503 504 505 506 507
// DBImpl::SuperVersion methods
DBImpl::SuperVersion::SuperVersion(const int num_memtables) {
  to_delete.resize(num_memtables);
}

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

508 509
// Returns the list of live files in 'sst_live' and the list
// of all files in the filesystem in 'all_files'.
I
Igor Canadi 已提交
510 511 512 513 514 515 516
// 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 已提交
517 518
  mutex_.AssertHeld();

519
  // if deletion is disabled, do nothing
520
  if (disable_delete_obsolete_files_ > 0) {
521 522 523
    return;
  }

524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
  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 已提交
540 541 542
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
543 544 545 546 547
  // 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();

548 549 550 551 552 553 554 555
  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 已提交
556 557 558 559
  deletion_state.sst_live.assign(pending_outputs_.begin(),
                                 pending_outputs_.end());
  versions_->AddLiveFiles(&deletion_state.sst_live);

560 561 562 563 564 565 566 567 568 569 570 571 572
  if (doing_the_full_scan) {
    // set of all files in the directory
    env_->GetChildren(dbname_, &deletion_state.all_files); // Ignore errors

    //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
      deletion_state.all_files.insert(
        deletion_state.all_files.end(),
        log_files.begin(),
        log_files.end()
      );
573
    }
574
  }
575 576
}

D
Dhruba Borthakur 已提交
577
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
578
// belong to live files are posibly removed. Also, removes all the
579
// files in sst_delete_files and log_delete_files.
580
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
581
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
582 583 584 585 586 587 588 589

  // check if there is anything to do
  if (!state.all_files.size() &&
      !state.sst_delete_files.size() &&
      !state.log_delete_files.size()) {
    return;
  }

I
Igor Canadi 已提交
590 591 592 593
  // 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 已提交
594 595 596
    return;
  }

J
jorlow@chromium.org 已提交
597 598
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
599
  std::vector<std::string> old_log_files;
600

601 602
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
603 604
  std::unordered_set<uint64_t> live_set(state.sst_live.begin(),
                                        state.sst_live.end());
I
Igor Canadi 已提交
605

606 607 608
  state.all_files.reserve(state.all_files.size() +
      state.sst_delete_files.size());
  for (auto file : state.sst_delete_files) {
I
Igor Canadi 已提交
609
    state.all_files.push_back(TableFileName("", file->number).substr(1));
610
    delete file;
I
Igor Canadi 已提交
611 612
  }

613 614 615
  state.all_files.reserve(state.all_files.size() +
      state.log_delete_files.size());
  for (auto filenum : state.log_delete_files) {
I
Igor Canadi 已提交
616
    if (filenum > 0) {
I
Igor Canadi 已提交
617
      state.all_files.push_back(LogFileName("", filenum).substr(1));
I
Igor Canadi 已提交
618 619
    }
  }
620

I
Igor Canadi 已提交
621 622 623 624 625 626
  // dedup state.all_files so we don't try to delete the same
  // file twice
  sort(state.all_files.begin(), state.all_files.end());
  auto unique_end = unique(state.all_files.begin(), state.all_files.end());

  for (size_t i = 0; state.all_files.begin() + i < unique_end; i++) {
627
    if (ParseFileName(state.all_files[i], &number, &type)) {
J
jorlow@chromium.org 已提交
628 629 630
      bool keep = true;
      switch (type) {
        case kLogFile:
I
Igor Canadi 已提交
631 632
          keep = ((number >= state.log_number) ||
                  (number == state.prev_log_number));
J
jorlow@chromium.org 已提交
633 634 635 636
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
I
Igor Canadi 已提交
637
          keep = (number >= state.manifest_file_number);
J
jorlow@chromium.org 已提交
638 639
          break;
        case kTableFile:
640
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
641 642 643 644
          break;
        case kTempFile:
          // Any temp files that are currently being written to must
          // be recorded in pending_outputs_, which is inserted into "live"
645
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
646
          break;
H
heyongqiang 已提交
647 648 649
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
650
            old_log_files.push_back(state.all_files[i]);
H
heyongqiang 已提交
651 652
          }
          break;
J
jorlow@chromium.org 已提交
653 654
        case kCurrentFile:
        case kDBLockFile:
M
Mayank Agarwal 已提交
655
        case kIdentityFile:
K
Kosie van der Merwe 已提交
656
        case kMetaDatabase:
J
jorlow@chromium.org 已提交
657 658 659 660 661 662
          keep = true;
          break;
      }

      if (!keep) {
        if (type == kTableFile) {
I
Igor Canadi 已提交
663 664
          // evict from cache
          table_cache_->Evict(number);
J
jorlow@chromium.org 已提交
665
        }
I
Igor Canadi 已提交
666 667
        std::string fname = ((type == kLogFile) ? options_.wal_dir : dbname_) +
            "/" + state.all_files[i];
K
Kai Liu 已提交
668 669 670 671
        Log(options_.info_log,
            "Delete type=%d #%lu",
            int(type),
            (unsigned long)number);
672

I
Igor Canadi 已提交
673 674 675
        Status st;
        if (type == kLogFile && (options_.WAL_ttl_seconds > 0 ||
              options_.WAL_size_limit_MB > 0)) {
I
Igor Canadi 已提交
676 677
            st = env_->RenameFile(fname,
                ArchivedLogFileName(options_.wal_dir, number));
I
Igor Canadi 已提交
678
            if (!st.ok()) {
K
Kai Liu 已提交
679
              Log(options_.info_log,
I
Igor Canadi 已提交
680 681
                  "RenameFile logfile #%lu FAILED -- %s\n",
                  (unsigned long)number, st.ToString().c_str());
I
Igor Canadi 已提交
682
            }
683
        } else {
I
Igor Canadi 已提交
684
          st = env_->DeleteFile(fname);
685
          if (!st.ok()) {
I
Igor Canadi 已提交
686 687
            Log(options_.info_log, "Delete type=%d #%lu FAILED -- %s\n",
                int(type), (unsigned long)number, st.ToString().c_str());
688
          }
H
heyongqiang 已提交
689
        }
J
jorlow@chromium.org 已提交
690 691 692
      }
    }
  }
H
heyongqiang 已提交
693

694
  // Delete old info log files.
K
Kai Liu 已提交
695 696 697 698 699
  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 已提交
700
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
701
    size_t end = old_log_file_count - options_.keep_log_file_num;
702
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
703
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
704 705
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
706 707 708
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
709
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
710
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
711 712 713 714 715
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
716
  FindObsoleteFiles(deletion_state, true);
D
Dhruba Borthakur 已提交
717
  PurgeObsoleteFiles(deletion_state);
718 719
}

720 721 722 723 724 725 726 727
// 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
728
void DBImpl::PurgeObsoleteWALFiles() {
729 730 731 732 733 734
  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;
  }

735 736
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
737 738 739 740 741 742 743 744
  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_;
745

746 747 748 749 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
  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;
786
        }
787 788 789 790 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
      }

      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;
840 841
    }
  }
D
Dhruba Borthakur 已提交
842 843
}

844 845 846
// If externalTable is set, then apply recovered transactions
// to that table. This is used for readonly mode.
Status DBImpl::Recover(VersionEdit* edit, MemTable* external_table,
H
heyongqiang 已提交
847
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
848 849
  mutex_.AssertHeld();

850
  assert(db_lock_ == nullptr);
851
  if (!external_table) {
852 853 854 855 856 857 858 859 860 861 862 863 864
    // 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_);
865 866 867
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
868

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

895
  Status s = versions_->Recover();
J
jorlow@chromium.org 已提交
896 897
  if (s.ok()) {
    SequenceNumber max_sequence(0);
898 899 900 901 902 903 904

    // 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
905
    // produced by an older version of rocksdb.
906 907 908
    const uint64_t min_log = versions_->LogNumber();
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
909
    s = env_->GetChildren(options_.wal_dir, &filenames);
910 911
    if (!s.ok()) {
      return s;
912
    }
913 914 915 916 917 918 919 920 921
    uint64_t number;
    FileType type;
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
      if (ParseFileName(filenames[i], &number, &type)
          && type == kLogFile
          && ((number >= min_log) || (number == prev_log))) {
        logs.push_back(number);
      }
J
jorlow@chromium.org 已提交
922
    }
923

H
heyongqiang 已提交
924 925 926 927 928 929
    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");
    }

930 931 932
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
    for (size_t i = 0; i < logs.size(); i++) {
933
      s = RecoverLogFile(logs[i], edit, &max_sequence, external_table);
934 935 936 937
      // The previous incarnation may not have written any MANIFEST
      // records after allocating this log number.  So we manually
      // update the file number allocation counter in VersionSet.
      versions_->MarkFileNumberUsed(logs[i]);
938 939
    }

J
jorlow@chromium.org 已提交
940
    if (s.ok()) {
941 942 943
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
      }
944
      SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
945
                     versions_->LastSequence());
J
jorlow@chromium.org 已提交
946 947 948 949 950 951 952 953
    }
  }

  return s;
}

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

  mutex_.AssertHeld();

  // Open the log file
973
  std::string fname = LogFileName(options_.wal_dir, log_number);
974
  unique_ptr<SequentialFile> file;
975
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
976 977 978 979 980 981 982 983
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

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

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
1001
  MemTable* mem = nullptr;
1002 1003 1004
  if (external_table) {
    mem = external_table;
  }
1005
  while (reader.ReadRecord(&record, &scratch) && status.ok()) {
J
jorlow@chromium.org 已提交
1006 1007 1008 1009 1010 1011 1012
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

1013
    if (mem == nullptr) {
X
Xing Jin 已提交
1014 1015
      mem = new MemTable(internal_comparator_, mem_rep_factory_,
        NumberLevels(), options_);
1016
      mem->Ref();
J
jorlow@chromium.org 已提交
1017
    }
1018
    status = WriteBatchInternal::InsertInto(&batch, mem, &options_);
J
jorlow@chromium.org 已提交
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
    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;
    }

1030 1031
    if (!external_table &&
        mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
1032
      status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
1033 1034 1035 1036 1037
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
        break;
      }
1038
      delete mem->Unref();
1039
      mem = nullptr;
J
jorlow@chromium.org 已提交
1040 1041 1042
    }
  }

1043
  if (status.ok() && mem != nullptr && !external_table) {
1044
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
1045 1046 1047 1048
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

1049 1050 1051
  if (mem != nullptr && !external_table) {
    delete mem->Unref();
  }
J
jorlow@chromium.org 已提交
1052 1053 1054
  return status;
}

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

  Status s;
  {
    mutex_.Unlock();
1071 1072
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1073
                   user_comparator(), newest_snapshot,
1074
                   earliest_seqno_in_memtable, true);
I
Igor Canadi 已提交
1075
    LogFlush(options_.info_log);
1076 1077 1078
    mutex_.Lock();
  }

K
Kai Liu 已提交
1079 1080 1081
  Log(options_.info_log, "Level-0 table #%lu: %lu bytes %s",
      (unsigned long) meta.number,
      (unsigned long) meta.file_size,
J
jorlow@chromium.org 已提交
1082 1083
      s.ToString().c_str());
  delete iter;
1084

1085
  pending_outputs_.erase(meta.number);
1086 1087 1088 1089 1090 1091

  // 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,
1092 1093
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1094 1095
  }

1096 1097 1098
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
1099
  stats.files_out_levelnp1 = 1;
1100
  stats_[level].Add(stats);
1101
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
J
jorlow@chromium.org 已提交
1102 1103 1104
  return s;
}

1105

1106
Status DBImpl::WriteLevel0Table(std::vector<MemTable*> &mems, VersionEdit* edit,
1107
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
1108
  mutex_.AssertHeld();
1109 1110 1111 1112 1113
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
1114

1115 1116
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1117
    mems[0]->GetFirstSequenceNumber();
1118
  Version* base = versions_->current();
1119
  base->Ref();          // it is likely that we do not need this reference
1120 1121 1122
  Status s;
  {
    mutex_.Unlock();
1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134
    std::vector<Iterator*> list;
    for (MemTable* m : mems) {
      Log(options_.info_log,
          "Flushing memtable with log file: %lu\n",
          (unsigned long)m->GetLogNumber());
      list.push_back(m->NewIterator());
    }
    Iterator* iter = NewMergingIterator(&internal_comparator_, &list[0],
                                        list.size());
    Log(options_.info_log,
        "Level-0 flush table #%lu: started",
        (unsigned long)meta.number);
1135 1136 1137 1138 1139
    // 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);
1140 1141
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1142
                   user_comparator(), newest_snapshot,
1143
                   earliest_seqno_in_memtable, enable_compression);
I
Igor Canadi 已提交
1144
    LogFlush(options_.info_log);
1145 1146 1147 1148 1149
    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());
1150 1151
    mutex_.Lock();
  }
1152 1153
  base->Unref();

1154 1155 1156 1157 1158 1159 1160 1161 1162

  // 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 已提交
1163
  // committed so that other threads can recognize this file as a
1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176
  // 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.
1177
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1178
        options_.compaction_style == kCompactionStyleLevel) {
1179 1180 1181
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
1182 1183
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1184 1185 1186 1187 1188 1189
  }

  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
  stats_[level].Add(stats);
1190
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
1191 1192 1193
  return s;
}

I
Igor Canadi 已提交
1194 1195
Status DBImpl::FlushMemTableToOutputFile(bool* madeProgress,
                                         DeletionState& deletion_state) {
1196 1197 1198
  mutex_.AssertHeld();
  assert(imm_.size() != 0);

1199
  if (!imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
1200 1201
    Log(options_.info_log, "FlushMemTableToOutputFile already in progress");
    Status s = Status::IOError("FlushMemTableToOutputFile already in progress");
1202 1203 1204 1205 1206
    return s;
  }

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
1207 1208 1209
  std::vector<MemTable*> mems;
  imm_.PickMemtablesToFlush(&mems);
  if (mems.empty()) {
1210 1211 1212 1213 1214 1215
    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
1216 1217 1218
  // 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.
1219
  MemTable* m = mems[0];
1220 1221
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
  // 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());
  }
1232

1233
  // This will release and re-acquire the mutex.
1234
  Status s = WriteLevel0Table(mems, edit, &file_number);
1235

1236
  if (s.ok() && shutting_down_.Acquire_Load()) {
1237 1238 1239
    s = Status::IOError(
      "Database shutdown started during memtable compaction"
    );
1240
  }
J
jorlow@chromium.org 已提交
1241

1242
  // Replace immutable memtable with the generated Table
1243
  s = imm_.InstallMemtableFlushResults(
1244
    mems, versions_.get(), s, &mutex_, options_.info_log.get(),
1245
    file_number, pending_outputs_, &deletion_state.memtables_to_free);
J
jorlow@chromium.org 已提交
1246 1247

  if (s.ok()) {
I
Igor Canadi 已提交
1248
    InstallSuperVersion(deletion_state);
1249 1250 1251
    if (madeProgress) {
      *madeProgress = 1;
    }
1252

1253
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1254

1255
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1256
      // add to deletion state
1257 1258 1259 1260
      deletion_state.log_delete_files.insert(
          deletion_state.log_delete_files.end(),
          logs_to_delete.begin(),
          logs_to_delete.end());
1261
    }
J
jorlow@chromium.org 已提交
1262 1263 1264 1265
  }
  return s;
}

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

  if (reduce_level) {
1285
    ReFitLevel(max_level_with_files, target_level);
1286
  }
I
Igor Canadi 已提交
1287
  LogFlush(options_.info_log);
1288 1289 1290 1291 1292 1293
}

// return the same level if it cannot be moved
int DBImpl::FindMinimumEmptyLevelFitting(int level) {
  mutex_.AssertHeld();
  int minimum_level = level;
1294
  for (int i = level - 1; i > 0; --i) {
1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
    // 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;
}

1306
void DBImpl::ReFitLevel(int level, int target_level) {
1307 1308
  assert(level < NumberLevels());

I
Igor Canadi 已提交
1309 1310 1311 1312 1313
  SuperVersion* superversion_to_free = nullptr;
  SuperVersion* new_superversion =
      new SuperVersion(options_.max_write_buffer_number);

  mutex_.Lock();
1314 1315 1316

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

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1326
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1327
    Log(options_.info_log,
1328 1329
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1330 1331 1332 1333
    bg_cv_.Wait();
  }

  // move to a smaller level
1334 1335 1336 1337
  int to_level = target_level;
  if (target_level < 0) {
    to_level = FindMinimumEmptyLevelFitting(level);
  }
1338 1339 1340 1341 1342 1343 1344 1345 1346 1347

  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);
1348 1349
      edit.AddFile(to_level, f->number, f->file_size, f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1350 1351 1352 1353 1354
    }
    Log(options_.info_log, "Apply version edit:\n%s",
        edit.DebugString().data());

    auto status = versions_->LogAndApply(&edit, &mutex_);
I
Igor Canadi 已提交
1355 1356
    superversion_to_free = InstallSuperVersion(new_superversion);
    new_superversion = nullptr;
1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367

    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 已提交
1368 1369 1370 1371

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

1374
int DBImpl::NumberLevels(const ColumnFamilyHandle& column_family) {
1375
  return options_.num_levels;
1376 1377
}

1378
int DBImpl::MaxMemCompactionLevel(const ColumnFamilyHandle& column_family) {
1379
  return options_.max_mem_compaction_level;
1380 1381
}

1382
int DBImpl::Level0StopWriteTrigger(const ColumnFamilyHandle& column_family) {
1383
  return options_.level0_stop_writes_trigger;
1384 1385
}

1386 1387
Status DBImpl::Flush(const FlushOptions& options,
                     const ColumnFamilyHandle& column_family) {
H
heyongqiang 已提交
1388 1389 1390 1391
  Status status = FlushMemTable(options);
  return status;
}

1392
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1393 1394 1395
  return versions_->LastSequence();
}

1396
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
1397
                               unique_ptr<TransactionLogIterator>* iter) {
1398

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

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

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

1426 1427
Status DBImpl::RetainProbableWalFiles(VectorLogPtr& all_logs,
                                      const SequenceNumber target) {
1428
  long start = 0; // signed to avoid overflow when target is < first file.
1429
  long end = static_cast<long>(all_logs.size()) - 1;
1430
  // Binary Search. avoid opening all files.
1431 1432
  while (end >= start) {
    long mid = start + (end - start) / 2;  // Avoid overflow.
1433 1434
    SequenceNumber current_seq_num = all_logs.at(mid)->StartSequence();
    if (current_seq_num == target) {
1435
      end = mid;
1436
      break;
1437
    } else if (current_seq_num < target) {
1438
      start = mid + 1;
1439
    } else {
1440
      end = mid - 1;
1441 1442
    }
  }
1443 1444 1445
  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);
1446 1447 1448
  return Status::OK();
}

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

1459 1460
Status DBImpl::ReadFirstRecord(const WalFileType type, const uint64_t number,
                               WriteBatch* const result) {
1461

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

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

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

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


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

1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
  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);
}

1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541
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);
1542 1543 1544
  if (!status.ok()) {
    return status;
  }
1545
  log_files.reserve(log_files.size() + all_files.size());
1546 1547 1548 1549 1550 1551
  VectorLogPtr::iterator pos_start;
  if (!log_files.empty()) {
    pos_start = log_files.end() - 1;
  } else {
    pos_start = log_files.begin();
  }
1552
  for (const auto& f : all_files) {
1553 1554
    uint64_t number;
    FileType type;
1555
    if (ParseFileName(f, &number, &type) && type == kLogFile){
1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573

      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))));
1574 1575
    }
  }
1576
  CompareLogByPointer compare_log_files;
1577
  std::sort(pos_start, log_files.end(), compare_log_files);
1578 1579 1580
  return status;
}

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

G
Gabor Cselle 已提交
1584 1585
  InternalKey begin_storage, end_storage;

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

  MutexLock l(&mutex_);
1608

1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627
  // 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");
1628 1629
    bg_cv_.Wait();
  }
1630

1631 1632
  Log(options_.info_log, "Manual compaction starting");

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

1644 1645 1646
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
J
jorlow@chromium.org 已提交
1647 1648
}

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

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

1672
Status DBImpl::TEST_FlushMemTable() {
H
heyongqiang 已提交
1673 1674 1675
  return FlushMemTable(FlushOptions());
}

1676 1677
Status DBImpl::TEST_WaitForFlushMemTable() {
  return WaitForFlushMemTable();
1678 1679 1680
}

Status DBImpl::TEST_WaitForCompact() {
1681
  // Wait until the compaction completes
1682 1683 1684 1685 1686

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

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

1695
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
1696
  mutex_.AssertHeld();
1697 1698
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
J
jorlow@chromium.org 已提交
1699 1700 1701
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
  } else {
1702 1703 1704 1705 1706 1707 1708 1709 1710
    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);
    }

1711 1712 1713 1714
    // 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.
1715 1716 1717
    if ((manual_compaction_ ||
         versions_->NeedsCompaction() ||
         (is_flush_pending && (options_.max_background_flushes <= 0))) &&
1718 1719 1720
        bg_compaction_scheduled_ < options_.max_background_compactions &&
        (!bg_manual_only_ || manual_compaction_)) {

1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734
      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 已提交
1735 1736
Status DBImpl::BackgroundFlush(bool* madeProgress,
                               DeletionState& deletion_state) {
1737 1738 1739 1740
  Status stat;
  while (stat.ok() &&
         imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
    Log(options_.info_log,
1741
        "BackgroundCallFlush doing FlushMemTableToOutputFile, flush slots available %d",
1742
        options_.max_background_flushes - bg_flush_scheduled_);
I
Igor Canadi 已提交
1743
    stat = FlushMemTableToOutputFile(madeProgress, deletion_state);
J
jorlow@chromium.org 已提交
1744
  }
1745
  return stat;
J
jorlow@chromium.org 已提交
1746 1747
}

1748
void DBImpl::BackgroundCallFlush() {
1749
  bool madeProgress = false;
I
Igor Canadi 已提交
1750
  DeletionState deletion_state(options_.max_write_buffer_number, true);
1751 1752 1753
  assert(bg_flush_scheduled_);
  MutexLock l(&mutex_);

I
Igor Canadi 已提交
1754
  Status s;
1755
  if (!shutting_down_.Acquire_Load()) {
I
Igor Canadi 已提交
1756
    s = BackgroundFlush(&madeProgress, deletion_state);
1757 1758 1759 1760 1761 1762 1763 1764 1765
    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 已提交
1766
      LogFlush(options_.info_log);
1767 1768 1769 1770 1771
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
  }

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

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

1789

1790 1791 1792 1793
void DBImpl::TEST_PurgeObsoleteteWAL() {
  PurgeObsoleteWALFiles();
}

1794
void DBImpl::BackgroundCallCompaction() {
1795
  bool madeProgress = false;
I
Igor Canadi 已提交
1796
  DeletionState deletion_state(options_.max_write_buffer_number, true);
H
Haobo Xu 已提交
1797 1798 1799

  MaybeDumpStats();

J
jorlow@chromium.org 已提交
1800
  MutexLock l(&mutex_);
1801
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1802
  assert(bg_compaction_scheduled_);
I
Igor Canadi 已提交
1803
  Status s;
H
hans@chromium.org 已提交
1804
  if (!shutting_down_.Acquire_Load()) {
I
Igor Canadi 已提交
1805
    s = BackgroundCompaction(&madeProgress, deletion_state);
1806 1807 1808 1809 1810 1811 1812 1813 1814
    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 已提交
1815
      LogFlush(options_.info_log);
1816 1817 1818
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
J
jorlow@chromium.org 已提交
1819
  }
1820

I
Igor Canadi 已提交
1821 1822 1823 1824 1825
  // 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());
1826

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

1834
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1835

1836 1837
  MaybeScheduleLogDBDeployStats();

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

J
jorlow@chromium.org 已提交
1846 1847
}

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

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

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

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

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

  if (is_manual) {
G
Gabor Cselle 已提交
1936
    ManualCompaction* m = manual_compaction_;
1937 1938 1939
    if (!status.ok()) {
      m->done = true;
    }
1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951
    // 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 已提交
1952 1953 1954 1955 1956 1957
    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;
    }
1958
    m->in_progress = false; // not being processed anymore
1959
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1960
  }
1961
  return status;
J
jorlow@chromium.org 已提交
1962 1963
}

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

    // 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 已提交
1982 1983 1984 1985
  }
  delete compact;
}

1986 1987 1988 1989 1990
// 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();
1991 1992
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
1993
  int filesNeeded = compact->compaction->num_input_files(1);
1994
  for (int i = 0; i < filesNeeded; i++) {
1995 1996 1997 1998 1999 2000 2001 2002 2003
    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();
2004
  for (const auto file_number : compact->allocated_file_numbers) {
2005 2006 2007 2008 2009
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

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

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

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

S
Siying Dong 已提交
2043 2044 2045 2046
    CompressionType compression_type = GetCompressionType(
        options_, compact->compaction->output_level(),
        compact->compaction->enable_compression());

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

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

  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;
2074
  compact->builder.reset();
J
jorlow@chromium.org 已提交
2075 2076

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

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


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
2115 2116 2117 2118 2119

  // 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.
2120
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
2121 2122 2123 2124 2125 2126 2127 2128
    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");
  }

2129
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
2130 2131 2132 2133 2134 2135 2136 2137 2138
      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 已提交
2139
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2140 2141
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
2142 2143
        (options_.compaction_style == kCompactionStyleUniversal) ?
          level : level + 1,
2144 2145
        out.number, out.file_size, out.smallest, out.largest,
        out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2146
  }
2147
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
2148 2149
}

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

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

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
2195
  assert(compact->builder == nullptr);
2196
  assert(!compact->outfile);
2197 2198 2199

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
H
Haobo Xu 已提交
2200
  SequenceNumber latest_snapshot = 0;
2201 2202 2203 2204 2205
  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 已提交
2206
  } else {
H
Haobo Xu 已提交
2207
    latest_snapshot = compact->existing_snapshots.back();
2208 2209 2210 2211
    // 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 已提交
2212 2213
  }

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

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

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

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

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

J
jorlow@chromium.org 已提交
2262
    Slice key = input->key();
2263
    Slice value = input->value();
H
Haobo Xu 已提交
2264

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

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2274
    bool drop = false;
2275
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2276 2277
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2278 2279
      // 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 已提交
2280 2281 2282
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2283
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2284 2285 2286 2287 2288 2289 2290 2291
    } 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;
2292
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2293 2294

        // apply the compaction filter to the first occurrence of the user key
2295
        if (compaction_filter &&
H
Haobo Xu 已提交
2296 2297 2298 2299 2300 2301 2302 2303 2304 2305
            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 =
2306
            compaction_filter->Filter(compact->compaction->level(),
S
Siying Dong 已提交
2307 2308 2309
                                               ikey.user_key, value,
                                               &compaction_filter_value,
                                               &value_changed);
H
Haobo Xu 已提交
2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321
          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();
2322
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2323 2324 2325 2326 2327
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }

J
jorlow@chromium.org 已提交
2328 2329
      }

2330 2331 2332
      // 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.
2333 2334 2335 2336 2337 2338
      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);
2339 2340 2341 2342 2343

      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 已提交
2344
        // Hidden by an newer entry for same user key
2345
        // TODO: why not > ?
2346
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2347
        drop = true;    // (A)
2348
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2349
      } else if (ikey.type == kTypeDeletion &&
2350
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
2351 2352 2353 2354 2355 2356 2357 2358 2359
                 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;
2360
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2361 2362 2363 2364 2365 2366 2367
      } 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 已提交
2368
        merge.MergeUntil(input.get(), prev_snapshot, bottommost_level,
2369
                         options_.statistics.get());
2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387
        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 已提交
2388 2389 2390
      }

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

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

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

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

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

        // 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 已提交
2476 2477
        }

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

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

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

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

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

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

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

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

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

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

J
jorlow@chromium.org 已提交
2584 2585 2586
  return status;
}

2587 2588 2589 2590
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
2591
  std::vector<MemTable*> mem; // includes both mem_ and imm_
I
Igor Canadi 已提交
2592
  DBImpl *db;
2593 2594 2595 2596
};

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

J
jorlow@chromium.org 已提交
2615 2616
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
2617
  IterState* cleanup = new IterState;
J
jorlow@chromium.org 已提交
2618
  mutex_.Lock();
2619
  *latest_snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2620

2621
  // Collect together all needed child iterators for mem
J
jorlow@chromium.org 已提交
2622
  std::vector<Iterator*> list;
2623
  mem_->Ref();
2624
  list.push_back(mem_->NewIterator(options));
J
Jim Paton 已提交
2625

2626 2627 2628 2629 2630 2631 2632 2633
  cleanup->mem.push_back(mem_);

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

  // Collect iterators for files in L0 - Ln
2639
  versions_->current()->AddIterators(options, storage_options_, &list);
J
jorlow@chromium.org 已提交
2640
  Iterator* internal_iter =
2641
      NewMergingIterator(&internal_comparator_, &list[0], list.size());
J
jorlow@chromium.org 已提交
2642
  versions_->current()->Ref();
2643 2644

  cleanup->mu = &mutex_;
I
Igor Canadi 已提交
2645
  cleanup->db = this;
2646
  cleanup->version = versions_->current();
2647
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2648 2649 2650 2651 2652 2653 2654 2655 2656 2657

  mutex_.Unlock();
  return internal_iter;
}

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

J
jorlow@chromium.org 已提交
2658
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
2659 2660 2661 2662
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
2663
Status DBImpl::Get(const ReadOptions& options,
2664
                   const ColumnFamilyHandle& column_family, const Slice& key,
J
jorlow@chromium.org 已提交
2665
                   std::string* value) {
2666 2667 2668
  return GetImpl(options, key, value);
}

I
Igor Canadi 已提交
2669 2670 2671 2672 2673 2674 2675 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
// 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;
}

2707 2708 2709
Status DBImpl::GetImpl(const ReadOptions& options,
                       const Slice& key,
                       std::string* value,
2710
                       bool* value_found) {
2711
  Status s;
2712

2713
  StopWatch sw(env_, options_.statistics.get(), DB_GET);
2714
  SequenceNumber snapshot;
2715
  if (options.snapshot != nullptr) {
2716 2717 2718
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2719
  }
2720

I
Igor Canadi 已提交
2721 2722 2723
  // This can be replaced by using atomics and spinlock instead of big mutex
  mutex_.Lock();
  SuperVersion* get_version = super_version_->Ref();
2724
  mutex_.Unlock();
I
Igor Canadi 已提交
2725

2726
  bool have_stat_update = false;
2727
  Version::GetStats stats;
2728

2729
  // Prepare to store a list of merge operations if merge occurs.
2730
  MergeContext merge_context;
2731

2732
  // First look in the memtable, then in the immutable memtable (if any).
2733
  // s is both in/out. When in, s could either be OK or MergeInProgress.
2734
  // merge_operands will contain the sequence of merges in the latter case.
2735
  LookupKey lkey(key, snapshot);
I
Igor Canadi 已提交
2736
  if (get_version->mem->Get(lkey, value, &s, merge_context, options_)) {
2737
    // Done
2738
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
2739
  } else if (get_version->imm.Get(lkey, value, &s, merge_context, options_)) {
2740
    // Done
2741
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
2742
  } else {
I
Igor Canadi 已提交
2743 2744
    get_version->current->Get(options, lkey, value, &s, &merge_context, &stats,
                              options_, value_found);
2745
    have_stat_update = true;
2746
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
2747
  }
2748

I
Igor Canadi 已提交
2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769
  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;
2770
  }
2771

2772
  // Note, tickers are atomic now - no lock protection needed any more.
2773 2774
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
2775
  return s;
J
jorlow@chromium.org 已提交
2776 2777
}

2778 2779 2780 2781
std::vector<Status> DBImpl::MultiGet(
    const ReadOptions& options,
    const std::vector<ColumnFamilyHandle>& column_family,
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
2782

2783
  StopWatch sw(env_, options_.statistics.get(), DB_MULTIGET);
2784
  SequenceNumber snapshot;
2785 2786
  std::vector<MemTable*> to_delete;

2787
  mutex_.Lock();
2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806
  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;

2807 2808
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
2809

2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820
  // 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;

  // 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.
2821 2822
  // merge_operands will contain the sequence of merges in the latter case.
  for (int i=0; i<numKeys; ++i) {
2823
    merge_context.Clear();
2824 2825 2826 2827
    Status& s = statList[i];
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
2828
    if (mem->Get(lkey, value, &s, merge_context, options_)) {
2829
      // Done
2830
    } else if (imm.Get(lkey, value, &s, merge_context, options_)) {
2831 2832
      // Done
    } else {
2833
      current->Get(options, lkey, value, &s, &merge_context, &stats, options_);
2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845
      have_stat_update = true;
    }

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

  // Post processing (decrement reference counts and record statistics)
  mutex_.Lock();
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
2846
    MaybeScheduleFlushOrCompaction();
2847
  }
2848 2849
  MemTable* m = mem->Unref();
  imm.UnrefAll(&to_delete);
2850
  current->Unref();
2851 2852
  mutex_.Unlock();

2853 2854 2855 2856
  // free up all obsolete memtables outside the mutex
  delete m;
  for (MemTable* v: to_delete) delete v;

2857 2858 2859
  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);
2860 2861 2862 2863

  return statList;
}

2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& options,
                                  const Slice& column_family,
                                  ColumnFamilyHandle* handle) {
  VersionEdit edit(0);
  edit.AddColumnFamily(column_family.ToString());
  MutexLock l(&mutex_);
  ++versions_->max_column_family_;
  handle->id = versions_->max_column_family_;
  edit.SetColumnFamily(handle->id);
  return versions_->LogAndApply(&edit, &mutex_);
}

Status DBImpl::DropColumnFamily(const ColumnFamilyHandle& column_family) {
  VersionEdit edit(0);
  edit.DropColumnFamily();
  edit.SetColumnFamily(column_family.id);
  MutexLock l(&mutex_);
  return versions_->LogAndApply(&edit, &mutex_);
}

2884
bool DBImpl::KeyMayExist(const ReadOptions& options,
2885 2886
                         const ColumnFamilyHandle& column_family,
                         const Slice& key, std::string* value,
2887 2888
                         bool* value_found) {
  if (value_found != nullptr) {
K
Kai Liu 已提交
2889 2890
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
2891
  }
2892 2893
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
K
Kai Liu 已提交
2894 2895 2896 2897 2898 2899
  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();
2900 2901
}

2902 2903
Iterator* DBImpl::NewIterator(const ReadOptions& options,
                              const ColumnFamilyHandle& column_family) {
J
jorlow@chromium.org 已提交
2904
  SequenceNumber latest_snapshot;
T
Tyler Harter 已提交
2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917
  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 已提交
2918 2919
}

2920 2921 2922 2923 2924 2925 2926 2927
Status DBImpl::NewIterators(
    const ReadOptions& options,
    const std::vector<ColumnFamilyHandle>& column_family,
    std::vector<Iterator*>* iterators) {
  // TODO
  return Status::NotSupported("Not yet!");
}

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

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

// Convenience methods
2939 2940 2941 2942
Status DBImpl::Put(const WriteOptions& o,
                   const ColumnFamilyHandle& column_family, const Slice& key,
                   const Slice& val) {
  return DB::Put(o, column_family, key, val);
J
jorlow@chromium.org 已提交
2943 2944
}

2945 2946
Status DBImpl::Merge(const WriteOptions& o,
                     const ColumnFamilyHandle& column_family, const Slice& key,
2947 2948 2949 2950
                     const Slice& val) {
  if (!options_.merge_operator) {
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
2951
    return DB::Merge(o, column_family, key, val);
2952 2953 2954
  }
}

2955 2956 2957 2958
Status DBImpl::Delete(const WriteOptions& options,
                      const ColumnFamilyHandle& column_family,
                      const Slice& key) {
  return DB::Delete(options, column_family, key);
J
jorlow@chromium.org 已提交
2959 2960
}

2961 2962 2963 2964
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
2965
  w.disableWAL = options.disableWAL;
2966
  w.done = false;
2967

2968
  StopWatch sw(env_, options_.statistics.get(), DB_WRITE);
I
Igor Canadi 已提交
2969
  mutex_.Lock();
2970 2971 2972 2973
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
2974 2975 2976 2977 2978

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

2979
  if (w.done) {
I
Igor Canadi 已提交
2980
    mutex_.Unlock();
2981
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
2982
    return w.status;
2983 2984
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
2985 2986 2987
  }

  // May temporarily unlock and wait.
I
Igor Canadi 已提交
2988 2989
  SuperVersion* superversion_to_free = nullptr;
  Status status = MakeRoomForWrite(my_batch == nullptr, &superversion_to_free);
D
dgrogan@chromium.org 已提交
2990
  uint64_t last_sequence = versions_->LastSequence();
2991
  Writer* last_writer = &w;
2992
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
2993 2994 2995 2996 2997
    // 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.
2998
    WriteBatch* updates = BuildBatchGroup(&last_writer);
2999

3000 3001 3002 3003
    // 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_.
3004
    {
3005
      mutex_.Unlock();
3006 3007 3008 3009 3010
      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
3011 3012 3013
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3014 3015
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3016 3017
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3018 3019 3020
      }

      if (!options.disableWAL) {
3021 3022
        StopWatchNano timer(env_);
        StartPerfTimer(&timer);
3023 3024 3025 3026
        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());
3027
        BumpPerfTime(&perf_context.wal_write_time, &timer);
H
heyongqiang 已提交
3028
        if (status.ok() && options.sync) {
3029
          if (options_.use_fsync) {
3030
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3031
            status = log_->file()->Fsync();
3032
          } else {
3033
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3034
            status = log_->file()->Sync();
3035
          }
H
heyongqiang 已提交
3036
        }
3037 3038
      }
      if (status.ok()) {
3039 3040
        status = WriteBatchInternal::InsertInto(updates, mem_, &options_, this,
                                                options_.filter_deletes);
3041 3042 3043 3044 3045 3046 3047
        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!");
        }
3048 3049
        SetTickerCount(options_.statistics.get(),
                       SEQUENCE_NUMBER, last_sequence);
3050 3051 3052
      }
      mutex_.Lock();
      if (status.ok()) {
3053
        versions_->SetLastSequence(last_sequence);
3054
      }
J
jorlow@chromium.org 已提交
3055
    }
3056
    if (updates == &tmp_batch_) tmp_batch_.Clear();
J
jorlow@chromium.org 已提交
3057
  }
I
Igor Canadi 已提交
3058 3059 3060
  if (options_.paranoid_checks && !status.ok() && bg_error_.ok()) {
    bg_error_ = status; // stop compaction & fail any further writes
  }
3061

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

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

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

  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 已提交
3110 3111 3112 3113 3114 3115
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

3116
    if (w->batch != nullptr) {
3117 3118 3119 3120 3121 3122 3123 3124 3125
      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
3126
        result = &tmp_batch_;
3127 3128 3129 3130 3131 3132 3133 3134 3135 3136
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

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

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

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

3300
    } else {
3301 3302 3303 3304 3305
      unique_ptr<WritableFile> lfile;
      MemTable* memtmp = nullptr;

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

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

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

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

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

J
jorlow@chromium.org 已提交
3373 3374
  MutexLock l(&mutex_);
  Slice in = property;
3375
  Slice prefix("rocksdb.");
J
jorlow@chromium.org 已提交
3376 3377 3378 3379 3380 3381 3382
  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();
3383
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
3384 3385
      return false;
    } else {
3386
      char buf[100];
D
dgrogan@chromium.org 已提交
3387 3388
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
3389
      *value = buf;
J
jorlow@chromium.org 已提交
3390 3391
      return true;
    }
3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407 3408
  } 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;

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

    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;
3418 3419
    uint64_t total_bytes_written = 0;
    uint64_t total_bytes_read = 0;
M
Mark Callaghan 已提交
3420
    uint64_t micros_up = env_->NowMicros() - started_at_;
3421 3422
    // Add "+1" to make sure seconds_up is > 0 and avoid NaN later
    double seconds_up = (micros_up + 1) / 1000000.0;
3423
    uint64_t total_slowdown = 0;
J
Jim Paton 已提交
3424
    uint64_t total_slowdown_count = 0;
3425 3426 3427 3428
    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 已提交
3429

3430 3431 3432 3433 3434 3435 3436 3437 3438 3439
    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 已提交
3440
    // Pardon the long line but I think it is easier to read this way.
3441 3442
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
3443
             "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 已提交
3444
             "--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
3445 3446
             );
    value->append(buf);
3447
    for (int level = 0; level < NumberLevels(); level++) {
3448 3449
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
3450 3451 3452 3453 3454 3455
        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
3456 3457 3458
            : (stats_[level].bytes_written +
               stats_[level].bytes_readnp1 +
               stats_[level].bytes_readn) /
M
Mark Callaghan 已提交
3459 3460
                (double) stats_[level].bytes_readn;

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

3464 3465
        snprintf(
            buf, sizeof(buf),
3466
            "%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",
3467 3468 3469
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
3470
            versions_->NumLevelBytes(level) /
3471
                versions_->MaxBytesForLevel(level),
3472
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
3473 3474 3475 3476 3477 3478
            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,
3479 3480
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
3481
            (stats_[level].bytes_written / 1048576.0) /
3482
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
3483 3484 3485 3486
            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,
3487
            stats_[level].count,
J
Jim Paton 已提交
3488 3489
            stall_leveln_slowdown_[level] / 1000000.0,
            (unsigned long) stall_leveln_slowdown_count_[level]);
3490
        total_slowdown += stall_leveln_slowdown_[level];
J
Jim Paton 已提交
3491
        total_slowdown_count += stall_leveln_slowdown_count_[level];
3492 3493 3494
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
3495

3496 3497 3498 3499
    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_;
3500 3501 3502 3503 3504 3505
    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);

3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523
    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 已提交
3524
    snprintf(buf, sizeof(buf),
3525 3526
             "Compaction IO cumulative (GB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
3527
             user_bytes_written / (1048576.0 * 1024),
3528 3529 3530 3531 3532 3533 3534 3535
             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",
3536
             user_bytes_written / 1048576.0 / seconds_up,
3537 3538 3539 3540 3541 3542 3543 3544
             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",
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 3576
             (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 已提交
3577 3578
    value->append(buf);

3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600
    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",
3601 3602 3603 3604
             (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 已提交
3605 3606 3607 3608
    value->append(buf);

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

J
Jim Paton 已提交
3616 3617 3618 3619 3620 3621 3622 3623 3624
    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);

3625 3626 3627
    last_stats_.compaction_bytes_read_ = total_bytes_read;
    last_stats_.compaction_bytes_written_ = total_bytes_written;
    last_stats_.ingest_bytes_ = user_bytes_written;
3628
    last_stats_.seconds_up_ = seconds_up;
3629 3630 3631 3632 3633
    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;
3634

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

J
jorlow@chromium.org 已提交
3644 3645 3646
  return false;
}

3647 3648
void DBImpl::GetApproximateSizes(const ColumnFamilyHandle& column_family,
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671
  // 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);
I
Igor Canadi 已提交
3707
  DeletionState deletion_state(0, 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
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
3782 3783
Status DB::Put(const WriteOptions& opt, const ColumnFamilyHandle& column_family,
               const Slice& key, const Slice& value) {
J
jorlow@chromium.org 已提交
3784
  WriteBatch batch;
3785
  batch.Put(column_family, key, value);
J
jorlow@chromium.org 已提交
3786 3787 3788
  return Write(opt, &batch);
}

3789 3790
Status DB::Delete(const WriteOptions& opt,
                  const ColumnFamilyHandle& column_family, const Slice& key) {
J
jorlow@chromium.org 已提交
3791
  WriteBatch batch;
3792
  batch.Delete(column_family, key);
J
jorlow@chromium.org 已提交
3793 3794 3795
  return Write(opt, &batch);
}

3796 3797
Status DB::Merge(const WriteOptions& opt,
                 const ColumnFamilyHandle& column_family, const Slice& key,
3798 3799
                 const Slice& value) {
  WriteBatch batch;
3800
  batch.Merge(column_family, key, value);
3801 3802 3803
  return Write(opt, &batch);
}

3804 3805 3806 3807 3808
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
                              const Slice& column_family,
                              ColumnFamilyHandle* handle) {
  return Status::NotSupported("");
3809 3810
}
Status DB::DropColumnFamily(const ColumnFamilyHandle& column_family) {
3811
  return Status::NotSupported("");
3812 3813
}

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

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

3820
  if (options.block_cache != nullptr && options.no_block_cache) {
3821
    return Status::InvalidArgument(
3822
        "no_block_cache is true while block_cache is not nullptr");
3823
  }
3824

J
jorlow@chromium.org 已提交
3825
  DBImpl* impl = new DBImpl(options, dbname);
3826 3827 3828 3829 3830 3831 3832
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
3833 3834 3835 3836
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
3837
  impl->mutex_.Lock();
3838
  VersionEdit edit(impl->NumberLevels());
3839
  s = impl->Recover(&edit); // Handles create_if_missing, error_if_exists
J
jorlow@chromium.org 已提交
3840
  if (s.ok()) {
3841
    uint64_t new_log_number = impl->versions_->NewFileNumber();
3842
    unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
3843
    soptions.use_mmap_writes = false;
3844 3845 3846 3847 3848
    s = options.env->NewWritableFile(
      LogFileName(impl->options_.wal_dir, new_log_number),
      &lfile,
      soptions
    );
J
jorlow@chromium.org 已提交
3849
    if (s.ok()) {
3850
      lfile->SetPreallocationBlockSize(1.1 * options.write_buffer_size);
3851
      edit.SetLogNumber(new_log_number);
3852
      impl->logfile_number_ = new_log_number;
3853
      impl->log_.reset(new log::Writer(std::move(lfile)));
3854
      s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
J
jorlow@chromium.org 已提交
3855 3856
    }
    if (s.ok()) {
I
Igor Canadi 已提交
3857
      delete impl->InstallSuperVersion(new DBImpl::SuperVersion());
3858
      impl->mem_->SetLogNumber(impl->logfile_number_);
J
jorlow@chromium.org 已提交
3859
      impl->DeleteObsoleteFiles();
3860
      impl->MaybeScheduleFlushOrCompaction();
3861
      impl->MaybeScheduleLogDBDeployStats();
J
jorlow@chromium.org 已提交
3862 3863 3864
    }
  }
  impl->mutex_.Unlock();
3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877

  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 已提交
3878 3879 3880 3881 3882 3883 3884 3885
  if (s.ok()) {
    *dbptr = impl;
  } else {
    delete impl;
  }
  return s;
}

3886 3887 3888 3889 3890 3891 3892 3893
Status DB::OpenWithColumnFamilies(
    const DBOptions& db_options, const std::string& name,
    const std::vector<ColumnFamilyDescriptor>& column_families,
    std::vector<ColumnFamilyHandle>* handles, DB** dbptr) {
  // TODO
  return Status::NotSupported("Working on it");
}

3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
  Options options(db_options, ColumnFamilyOptions());
  InternalKeyComparator* icmp = new InternalKeyComparator(options.comparator);
  TableCache* table_cache = new TableCache(name, &options, EnvOptions(options),
                                           db_options.max_open_files - 10);
  VersionSet* version_set =
      new VersionSet(name, &options, EnvOptions(options), table_cache, icmp);

  version_set->Recover();
  column_families->clear();
  for (auto cf : version_set->column_families_) {
    column_families->push_back(cf.first);
  }

  delete version_set;
  delete table_cache;
  delete icmp;
3913 3914 3915
  return Status::NotSupported("Working on it");
}

3916 3917 3918
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
3919
Status DestroyDB(const std::string& dbname, const Options& options) {
3920 3921 3922 3923 3924
  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 已提交
3925
  std::vector<std::string> filenames;
3926 3927
  std::vector<std::string> archiveFiles;

3928
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
3929 3930
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
3931 3932 3933 3934 3935 3936 3937

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

J
jorlow@chromium.org 已提交
3939 3940 3941 3942 3943
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
3944 3945
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
3946 3947 3948
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
3949
    for (size_t i = 0; i < filenames.size(); i++) {
3950
      if (ParseFileName(filenames[i], &number, &type) &&
3951
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
3952 3953 3954
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
3955 3956
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
3957 3958 3959
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
3960 3961 3962 3963 3964
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
3965

3966
    env->GetChildren(archivedir, &archiveFiles);
3967 3968
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
3969 3970
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
3971
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
3972 3973 3974 3975 3976
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
3977
    // ignore case where no archival directory is present.
3978
    env->DeleteDir(archivedir);
3979

J
jorlow@chromium.org 已提交
3980
    env->UnlockFile(lock);  // Ignore error since state is already gone
3981
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
3982
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
3983
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
3984 3985 3986 3987
  }
  return result;
}

3988 3989
//
// A global method that can dump out the build version
3990
void dumpLeveldbBuildVersion(Logger * log) {
3991
  Log(log, "Git sha %s", rocksdb_build_git_sha);
3992
  Log(log, "Compile time %s %s",
3993
      rocksdb_build_compile_time, rocksdb_build_compile_date);
3994 3995
}

3996
}  // namespace rocksdb