db_impl.cc 142.7 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>
T
Tomislav Novak 已提交
20
#include <utility>
21
#include <vector>
22

J
jorlow@chromium.org 已提交
23 24
#include "db/builder.h"
#include "db/dbformat.h"
25
#include "db/db_iter.h"
J
jorlow@chromium.org 已提交
26 27 28 29
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
30
#include "db/memtablelist.h"
31
#include "db/merge_context.h"
32
#include "db/merge_helper.h"
T
Tyler Harter 已提交
33
#include "db/prefix_filter_iterator.h"
J
jorlow@chromium.org 已提交
34
#include "db/table_cache.h"
K
kailiu 已提交
35
#include "db/table_properties_collector.h"
T
Tomislav Novak 已提交
36
#include "db/tailing_iter.h"
37
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
38 39
#include "db/version_set.h"
#include "db/write_batch_internal.h"
40
#include "port/port.h"
41 42
#include "rocksdb/compaction_filter.h"
#include "rocksdb/db.h"
43
#include "rocksdb/column_family.h"
44 45 46 47
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
#include "rocksdb/statistics.h"
#include "rocksdb/status.h"
S
Siying Dong 已提交
48 49
#include "rocksdb/table.h"
#include "port/port.h"
J
jorlow@chromium.org 已提交
50
#include "table/block.h"
51
#include "table/block_based_table_factory.h"
J
jorlow@chromium.org 已提交
52 53
#include "table/merger.h"
#include "table/two_level_iterator.h"
54 55
#include "util/auto_roll_logger.h"
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
56
#include "util/coding.h"
I
Igor Canadi 已提交
57
#include "util/hash_skiplist_rep.h"
J
jorlow@chromium.org 已提交
58 59
#include "util/logging.h"
#include "util/mutexlock.h"
60
#include "util/perf_context_imp.h"
61
#include "util/stop_watch.h"
62
#include "util/autovector.h"
J
jorlow@chromium.org 已提交
63

64
namespace rocksdb {
J
jorlow@chromium.org 已提交
65

66
const std::string default_column_family_name("default");
67

68 69
void dumpLeveldbBuildVersion(Logger * log);

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

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

J
jorlow@chromium.org 已提交
82 83 84
struct DBImpl::CompactionState {
  Compaction* const compaction;

85 86 87 88 89
  // 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 已提交
90 91 92 93 94 95

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

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

  uint64_t total_bytes;

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

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
113 114 115 116 117 118 119

  // 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 已提交
120 121 122
};

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

X
Xing Jin 已提交
140 141 142 143 144 145
  // 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;
  }

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

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

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

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

J
jorlow@chromium.org 已提交
210 211 212
  return result;
}

S
Siying Dong 已提交
213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
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;
  }
}

234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255
CompressionType GetCompressionFlush(const Options& options) {
  // Compressing memtable flushes might not help unless the sequential load
  // optimization is used for leveled compaction. Otherwise the CPU and
  // latency overhead is not offset by saving much space.

  bool can_compress;

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

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

J
jorlow@chromium.org 已提交
256 257
DBImpl::DBImpl(const Options& options, const std::string& dbname)
    : env_(options.env),
H
heyongqiang 已提交
258
      dbname_(dbname),
J
jorlow@chromium.org 已提交
259
      internal_comparator_(options.comparator),
260 261
      options_(SanitizeOptions(dbname, &internal_comparator_,
                               &internal_filter_policy_, options)),
H
heyongqiang 已提交
262
      internal_filter_policy_(options.filter_policy),
J
jorlow@chromium.org 已提交
263
      owns_info_log_(options_.info_log != options.info_log),
264
      db_lock_(nullptr),
H
Haobo Xu 已提交
265
      mutex_(options.use_adaptive_mutex),
266
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
267
      bg_cv_(&mutex_),
268
      logfile_number_(0),
T
Tomislav Novak 已提交
269
      super_version_number_(0),
270
      tmp_batch_(),
271
      bg_compaction_scheduled_(0),
272
      bg_manual_only_(0),
273
      bg_flush_scheduled_(0),
274
      bg_logstats_scheduled_(false),
275 276
      manual_compaction_(nullptr),
      logger_(nullptr),
277
      disable_delete_obsolete_files_(0),
I
Igor Canadi 已提交
278
      delete_obsolete_files_last_run_(options.env->NowMicros()),
279
      purge_wal_files_last_run_(0),
280
      last_stats_dump_time_microsec_(0),
281
      default_interval_to_delete_obsolete_WAL_(600),
M
Mark Callaghan 已提交
282 283 284
      stall_level0_slowdown_(0),
      stall_memtable_compaction_(0),
      stall_level0_num_files_(0),
J
Jim Paton 已提交
285 286 287
      stall_level0_slowdown_count_(0),
      stall_memtable_compaction_count_(0),
      stall_level0_num_files_count_(0),
288
      started_at_(options.env->NowMicros()),
289
      flush_on_destroy_(false),
290
      stats_(options.num_levels),
291
      delayed_writes_(0),
292 293 294
      storage_options_(options),
      bg_work_gate_closed_(false),
      refitting_level_(false) {
295

H
heyongqiang 已提交
296
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
297 298

  stall_leveln_slowdown_.resize(options.num_levels);
J
Jim Paton 已提交
299 300
  stall_leveln_slowdown_count_.resize(options.num_levels);
  for (int i = 0; i < options.num_levels; ++i) {
301
    stall_leveln_slowdown_[i] = 0;
J
Jim Paton 已提交
302 303
    stall_leveln_slowdown_count_[i] = 0;
  }
304

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

310 311
  versions_.reset(new VersionSet(dbname_, &options_, storage_options_,
                                 table_cache_.get(), &internal_comparator_));
312

313 314
  dumpLeveldbBuildVersion(options_.info_log.get());
  options_.Dump(options_.info_log.get());
315

316
  char name[100];
317
  Status st = env_->GetHostName(name, 100L);
318
  if (st.ok()) {
319 320 321 322 323 324
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
325

I
Igor Canadi 已提交
326
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
327 328 329 330
}

DBImpl::~DBImpl() {
  // Wait for background work to finish
331
  if (flush_on_destroy_ && default_cfd_->mem->GetFirstSequenceNumber() != 0) {
332 333
    FlushMemTable(FlushOptions());
  }
334
  mutex_.Lock();
335
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
336 337 338
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
339
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
340 341 342
  }
  mutex_.Unlock();

343
  if (db_lock_ != nullptr) {
J
jorlow@chromium.org 已提交
344 345 346
    env_->UnlockFile(db_lock_);
  }

I
Igor Canadi 已提交
347
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
348 349
}

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

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

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

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

  log_.reset();
  versions_.reset();
  table_cache_.reset();
379 380
}

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

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

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

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

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

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

460 461
// Returns the list of live files in 'sst_live' and the list
// of all files in the filesystem in 'all_files'.
I
Igor Canadi 已提交
462 463 464 465 466 467 468
// 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 已提交
469 470
  mutex_.AssertHeld();

471
  // if deletion is disabled, do nothing
472
  if (disable_delete_obsolete_files_ > 0) {
473 474 475
    return;
  }

476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
  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 已提交
492 493 494
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
495 496 497 498 499
  // 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();

500 501 502 503 504 505 506 507
  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 已提交
508 509 510 511
  deletion_state.sst_live.assign(pending_outputs_.begin(),
                                 pending_outputs_.end());
  versions_->AddLiveFiles(&deletion_state.sst_live);

512 513 514 515 516 517 518 519 520 521 522 523 524
  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()
      );
525
    }
526
  }
527 528
}

D
Dhruba Borthakur 已提交
529
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
530
// belong to live files are posibly removed. Also, removes all the
531
// files in sst_delete_files and log_delete_files.
532
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
533
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
534 535 536 537 538 539 540 541

  // 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 已提交
542 543 544 545
  // 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 已提交
546 547 548
    return;
  }

J
jorlow@chromium.org 已提交
549 550
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
551
  std::vector<std::string> old_log_files;
552

553 554
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
555 556
  std::unordered_set<uint64_t> live_set(state.sst_live.begin(),
                                        state.sst_live.end());
I
Igor Canadi 已提交
557

558 559 560
  state.all_files.reserve(state.all_files.size() +
      state.sst_delete_files.size());
  for (auto file : state.sst_delete_files) {
I
Igor Canadi 已提交
561
    state.all_files.push_back(TableFileName("", file->number).substr(1));
562
    delete file;
I
Igor Canadi 已提交
563 564
  }

565 566 567
  state.all_files.reserve(state.all_files.size() +
      state.log_delete_files.size());
  for (auto filenum : state.log_delete_files) {
I
Igor Canadi 已提交
568
    if (filenum > 0) {
I
Igor Canadi 已提交
569
      state.all_files.push_back(LogFileName("", filenum).substr(1));
I
Igor Canadi 已提交
570 571
    }
  }
572

I
Igor Canadi 已提交
573 574 575 576 577 578
  // 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++) {
579
    if (ParseFileName(state.all_files[i], &number, &type)) {
J
jorlow@chromium.org 已提交
580 581 582
      bool keep = true;
      switch (type) {
        case kLogFile:
I
Igor Canadi 已提交
583 584
          keep = ((number >= state.log_number) ||
                  (number == state.prev_log_number));
J
jorlow@chromium.org 已提交
585 586 587 588
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
I
Igor Canadi 已提交
589
          keep = (number >= state.manifest_file_number);
J
jorlow@chromium.org 已提交
590 591
          break;
        case kTableFile:
592
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
593 594 595 596
          break;
        case kTempFile:
          // Any temp files that are currently being written to must
          // be recorded in pending_outputs_, which is inserted into "live"
597
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
598
          break;
H
heyongqiang 已提交
599 600 601
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
602
            old_log_files.push_back(state.all_files[i]);
H
heyongqiang 已提交
603 604
          }
          break;
J
jorlow@chromium.org 已提交
605 606
        case kCurrentFile:
        case kDBLockFile:
M
Mayank Agarwal 已提交
607
        case kIdentityFile:
K
Kosie van der Merwe 已提交
608
        case kMetaDatabase:
J
jorlow@chromium.org 已提交
609 610 611 612 613 614
          keep = true;
          break;
      }

      if (!keep) {
        if (type == kTableFile) {
I
Igor Canadi 已提交
615 616
          // evict from cache
          table_cache_->Evict(number);
J
jorlow@chromium.org 已提交
617
        }
I
Igor Canadi 已提交
618 619
        std::string fname = ((type == kLogFile) ? options_.wal_dir : dbname_) +
            "/" + state.all_files[i];
K
Kai Liu 已提交
620 621 622 623
        Log(options_.info_log,
            "Delete type=%d #%lu",
            int(type),
            (unsigned long)number);
624

I
Igor Canadi 已提交
625 626 627
        Status st;
        if (type == kLogFile && (options_.WAL_ttl_seconds > 0 ||
              options_.WAL_size_limit_MB > 0)) {
I
Igor Canadi 已提交
628 629
            st = env_->RenameFile(fname,
                ArchivedLogFileName(options_.wal_dir, number));
I
Igor Canadi 已提交
630
            if (!st.ok()) {
K
Kai Liu 已提交
631
              Log(options_.info_log,
I
Igor Canadi 已提交
632 633
                  "RenameFile logfile #%lu FAILED -- %s\n",
                  (unsigned long)number, st.ToString().c_str());
I
Igor Canadi 已提交
634
            }
635
        } else {
I
Igor Canadi 已提交
636
          st = env_->DeleteFile(fname);
637
          if (!st.ok()) {
I
Igor Canadi 已提交
638 639
            Log(options_.info_log, "Delete type=%d #%lu FAILED -- %s\n",
                int(type), (unsigned long)number, st.ToString().c_str());
640
          }
H
heyongqiang 已提交
641
        }
J
jorlow@chromium.org 已提交
642 643 644
      }
    }
  }
H
heyongqiang 已提交
645

646
  // Delete old info log files.
K
Kai Liu 已提交
647 648 649 650 651
  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 已提交
652
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
653
    size_t end = old_log_file_count - options_.keep_log_file_num;
654
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
655
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
656 657
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
658 659 660
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
661
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
662
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
663 664 665 666 667
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
668
  FindObsoleteFiles(deletion_state, true);
D
Dhruba Borthakur 已提交
669
  PurgeObsoleteFiles(deletion_state);
670 671
}

672 673 674 675 676 677 678 679
// 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
680
void DBImpl::PurgeObsoleteWALFiles() {
681 682 683 684 685 686
  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;
  }

687 688
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
689 690 691 692 693 694 695 696
  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_;
697

698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737
  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;
738
        }
739 740 741 742 743 744 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 786 787 788 789 790 791
      }

      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;
792 793
    }
  }
D
Dhruba Borthakur 已提交
794 795
}

796
Status DBImpl::Recover(
797 798
    const std::vector<ColumnFamilyDescriptor>& column_families, bool read_only,
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
799 800
  mutex_.AssertHeld();

801
  assert(db_lock_ == nullptr);
I
Igor Canadi 已提交
802
  if (!read_only) {
803 804 805 806 807 808 809 810 811 812 813 814
    // 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;
    }

815 816 817 818 819
    s = env_->NewDirectory(dbname_, &db_directory_);
    if (!s.ok()) {
      return s;
    }

820
    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
821 822 823
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
824

825 826
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
827
        // TODO: add merge_operator name check
828 829 830 831 832 833 834
        s = NewDB();
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
835 836
      }
    } else {
837 838 839 840
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
841
    }
M
Mayank Agarwal 已提交
842 843 844 845 846 847 848
    // 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 已提交
849 850
  }

I
Igor Canadi 已提交
851
  Status s = versions_->Recover(column_families);
J
jorlow@chromium.org 已提交
852 853
  if (s.ok()) {
    SequenceNumber max_sequence(0);
854
    default_cfd_ = versions_->GetColumnFamilySet()->GetDefault();
855 856 857 858 859 860 861

    // 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
862
    // produced by an older version of rocksdb.
863 864 865
    const uint64_t min_log = versions_->LogNumber();
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
866
    s = env_->GetChildren(options_.wal_dir, &filenames);
867 868
    if (!s.ok()) {
      return s;
869
    }
870 871 872 873 874 875 876 877 878
    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 已提交
879
    }
880

H
heyongqiang 已提交
881 882 883 884 885 886
    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");
    }

887 888
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
I
Igor Canadi 已提交
889
    for (size_t i = 0; s.ok() && i < logs.size(); i++) {
890 891 892 893
      // 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]);
I
Igor Canadi 已提交
894
      s = RecoverLogFile(logs[i], &max_sequence, read_only);
895 896
    }

J
jorlow@chromium.org 已提交
897
    if (s.ok()) {
898 899 900
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
      }
901
      SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
902
                     versions_->LastSequence());
J
jorlow@chromium.org 已提交
903 904 905 906 907 908
    }
  }

  return s;
}

I
Igor Canadi 已提交
909 910
Status DBImpl::RecoverLogFile(uint64_t log_number, SequenceNumber* max_sequence,
                              bool read_only) {
J
jorlow@chromium.org 已提交
911 912
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
913
    Logger* info_log;
J
jorlow@chromium.org 已提交
914
    const char* fname;
915 916
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
917
    virtual void Corruption(size_t bytes, const Status& s) {
918
      Log(info_log, "%s%s: dropping %d bytes; %s",
919
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
920
          fname, static_cast<int>(bytes), s.ToString().c_str());
921
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
922 923 924 925 926
    }
  };

  mutex_.AssertHeld();

I
Igor Canadi 已提交
927 928
  VersionEdit edit;

J
jorlow@chromium.org 已提交
929
  // Open the log file
930
  std::string fname = LogFileName(options_.wal_dir, log_number);
931
  unique_ptr<SequentialFile> file;
932
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
933 934 935 936 937 938 939 940
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
941
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
942
  reporter.fname = fname.c_str();
943 944
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
945 946 947 948
  // 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).
949
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
950
                     0/*initial_offset*/);
K
Kai Liu 已提交
951 952
  Log(options_.info_log, "Recovering log #%lu",
      (unsigned long) log_number);
J
jorlow@chromium.org 已提交
953 954 955 956 957

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
I
Igor Canadi 已提交
958 959
  bool memtable_empty = true;
  while (reader.ReadRecord(&record, &scratch)) {
J
jorlow@chromium.org 已提交
960 961 962 963 964 965 966
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

967 968
    status =
        WriteBatchInternal::InsertInto(&batch, default_cfd_->mem, &options_);
I
Igor Canadi 已提交
969
    memtable_empty = false;
J
jorlow@chromium.org 已提交
970 971
    MaybeIgnoreError(&status);
    if (!status.ok()) {
I
Igor Canadi 已提交
972
      return status;
J
jorlow@chromium.org 已提交
973 974 975 976 977 978 979 980
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

981 982 983
    if (!read_only && default_cfd_->mem->ApproximateMemoryUsage() >
                          options_.write_buffer_size) {
      status = WriteLevel0TableForRecovery(default_cfd_->mem, &edit);
I
Igor Canadi 已提交
984
      // we still want to clear memtable, even if the recovery failed
985
      default_cfd_->CreateNewMemtable();
I
Igor Canadi 已提交
986
      memtable_empty = true;
J
jorlow@chromium.org 已提交
987 988 989
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
I
Igor Canadi 已提交
990
        return status;
J
jorlow@chromium.org 已提交
991 992 993 994
      }
    }
  }

I
Igor Canadi 已提交
995
  if (!memtable_empty && !read_only) {
996 997
    status = WriteLevel0TableForRecovery(default_cfd_->mem, &edit);
    default_cfd_->CreateNewMemtable();
I
Igor Canadi 已提交
998 999 1000
    if (!status.ok()) {
      return status;
    }
J
jorlow@chromium.org 已提交
1001 1002
  }

I
Igor Canadi 已提交
1003 1004 1005 1006 1007 1008 1009 1010 1011
  if (edit.NumEntries() > 0) {
    // if read_only, NumEntries() will be 0
    assert(!read_only);
    // writing log number in the manifest means that any log file
    // with number strongly less than (log_number + 1) is already
    // recovered and should be ignored on next reincarnation.
    // Since we already recovered log_number, we want all logs
    // with numbers `<= log_number` (includes this one) to be ignored
    edit.SetLogNumber(log_number + 1);
1012
    status = versions_->LogAndApply(default_cfd_, &edit, &mutex_);
1013
  }
I
Igor Canadi 已提交
1014

J
jorlow@chromium.org 已提交
1015 1016 1017
  return status;
}

1018
Status DBImpl::WriteLevel0TableForRecovery(MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1019
  mutex_.AssertHeld();
1020
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1021 1022 1023 1024
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
1025 1026 1027
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
K
Kai Liu 已提交
1028 1029
  Log(options_.info_log, "Level-0 table #%lu: started",
      (unsigned long) meta.number);
1030 1031 1032 1033

  Status s;
  {
    mutex_.Unlock();
1034 1035
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1036
                   user_comparator(), newest_snapshot,
1037 1038
                   earliest_seqno_in_memtable,
                   GetCompressionFlush(options_));
I
Igor Canadi 已提交
1039
    LogFlush(options_.info_log);
1040 1041 1042
    mutex_.Lock();
  }

K
Kai Liu 已提交
1043 1044 1045
  Log(options_.info_log, "Level-0 table #%lu: %lu bytes %s",
      (unsigned long) meta.number,
      (unsigned long) meta.file_size,
J
jorlow@chromium.org 已提交
1046 1047
      s.ToString().c_str());
  delete iter;
1048

1049
  pending_outputs_.erase(meta.number);
1050 1051 1052 1053 1054 1055

  // 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,
1056 1057
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1058 1059
  }

1060 1061 1062
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
1063
  stats.files_out_levelnp1 = 1;
1064
  stats_[level].Add(stats);
1065
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
J
jorlow@chromium.org 已提交
1066 1067 1068
  return s;
}

1069

1070
Status DBImpl::WriteLevel0Table(std::vector<MemTable*> &mems, VersionEdit* edit,
1071
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
1072
  mutex_.AssertHeld();
1073 1074 1075 1076 1077
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
1078

1079 1080
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1081
    mems[0]->GetFirstSequenceNumber();
1082
  Version* base = default_cfd_->current;
1083
  base->Ref();          // it is likely that we do not need this reference
1084 1085 1086
  Status s;
  {
    mutex_.Unlock();
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
    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);
1099

1100 1101
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
1102
                   user_comparator(), newest_snapshot,
1103
                   earliest_seqno_in_memtable, GetCompressionFlush(options_));
I
Igor Canadi 已提交
1104
    LogFlush(options_.info_log);
1105 1106 1107 1108 1109
    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());
1110 1111 1112
    if (!options_.disableDataSync) {
      db_directory_->Fsync();
    }
1113 1114
    mutex_.Lock();
  }
1115 1116
  base->Unref();

1117 1118

  // re-acquire the most current version
1119
  base = default_cfd_->current;
1120 1121 1122 1123 1124 1125

  // 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 已提交
1126
  // committed so that other threads can recognize this file as a
1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
  // 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.
1140
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1141
        options_.compaction_style == kCompactionStyleLevel) {
1142 1143 1144
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
1145 1146
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1147 1148 1149 1150 1151 1152
  }

  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
  stats_[level].Add(stats);
1153
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
1154 1155 1156
  return s;
}

I
Igor Canadi 已提交
1157 1158
Status DBImpl::FlushMemTableToOutputFile(bool* madeProgress,
                                         DeletionState& deletion_state) {
1159
  mutex_.AssertHeld();
1160
  assert(default_cfd_->imm.size() != 0);
1161

1162
  if (!default_cfd_->imm.IsFlushPending()) {
1163 1164
    Log(options_.info_log, "FlushMemTableToOutputFile already in progress");
    Status s = Status::IOError("FlushMemTableToOutputFile already in progress");
1165 1166 1167 1168 1169
    return s;
  }

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
1170
  std::vector<MemTable*> mems;
1171
  default_cfd_->imm.PickMemtablesToFlush(&mems);
1172
  if (mems.empty()) {
1173 1174 1175 1176 1177 1178
    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
1179 1180 1181
  // 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.
1182
  MemTable* m = mems[0];
1183 1184
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1185 1186 1187 1188 1189 1190 1191 1192 1193 1194
  // 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());
  }
1195

1196
  // This will release and re-acquire the mutex.
1197
  Status s = WriteLevel0Table(mems, edit, &file_number);
1198

1199
  if (s.ok() && shutting_down_.Acquire_Load()) {
1200 1201 1202
    s = Status::IOError(
      "Database shutdown started during memtable compaction"
    );
1203
  }
J
jorlow@chromium.org 已提交
1204

1205
  // Replace immutable memtable with the generated Table
1206
  s = default_cfd_->imm.InstallMemtableFlushResults(
1207 1208 1209
      default_cfd_, mems, versions_.get(), s, &mutex_, options_.info_log.get(),
      file_number, pending_outputs_, &deletion_state.memtables_to_free,
      db_directory_.get());
J
jorlow@chromium.org 已提交
1210 1211

  if (s.ok()) {
1212
    InstallSuperVersion(default_cfd_, deletion_state);
1213 1214 1215
    if (madeProgress) {
      *madeProgress = 1;
    }
1216

1217
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1218

1219
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1220
      // add to deletion state
1221 1222 1223 1224
      deletion_state.log_delete_files.insert(
          deletion_state.log_delete_files.end(),
          logs_to_delete.begin(),
          logs_to_delete.end());
1225
    }
J
jorlow@chromium.org 已提交
1226 1227 1228 1229
  }
  return s;
}

1230 1231 1232
Status DBImpl::CompactRange(const ColumnFamilyHandle& column_family,
                            const Slice* begin, const Slice* end,
                            bool reduce_level, int target_level) {
L
Lei Jin 已提交
1233 1234 1235 1236 1237 1238
  Status s = FlushMemTable(FlushOptions());
  if (!s.ok()) {
    LogFlush(options_.info_log);
    return s;
  }

G
Gabor Cselle 已提交
1239 1240 1241
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
1242
    Version* base = default_cfd_->current;
1243
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
1244 1245 1246 1247 1248
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
1249 1250 1251 1252 1253
  for (int level = 0; level <= max_level_with_files; level++) {
    // in case the compaction is unversal or if we're compacting the
    // bottom-most level, the output level will be the same as input one
    if (options_.compaction_style == kCompactionStyleUniversal ||
        level == max_level_with_files) {
L
Lei Jin 已提交
1254
      s = RunManualCompaction(level, level, begin, end);
1255
    } else {
L
Lei Jin 已提交
1256 1257 1258 1259 1260
      s = RunManualCompaction(level, level + 1, begin, end);
    }
    if (!s.ok()) {
      LogFlush(options_.info_log);
      return s;
1261
    }
G
Gabor Cselle 已提交
1262
  }
1263 1264

  if (reduce_level) {
L
Lei Jin 已提交
1265
    s = ReFitLevel(max_level_with_files, target_level);
1266
  }
I
Igor Canadi 已提交
1267
  LogFlush(options_.info_log);
L
Lei Jin 已提交
1268 1269

  return s;
1270 1271 1272 1273 1274
}

// return the same level if it cannot be moved
int DBImpl::FindMinimumEmptyLevelFitting(int level) {
  mutex_.AssertHeld();
1275
  Version* current = default_cfd_->current;
1276
  int minimum_level = level;
1277
  for (int i = level - 1; i > 0; --i) {
1278
    // stop if level i is not empty
1279
    if (current->NumLevelFiles(i) > 0) break;
1280
    // stop if level i is too small (cannot fit the level files)
1281
    if (versions_->MaxBytesForLevel(i) < current->NumLevelBytes(level)) break;
1282 1283 1284 1285 1286 1287

    minimum_level = i;
  }
  return minimum_level;
}

L
Lei Jin 已提交
1288
Status DBImpl::ReFitLevel(int level, int target_level) {
1289 1290
  assert(level < NumberLevels());

I
Igor Canadi 已提交
1291 1292 1293 1294 1295
  SuperVersion* superversion_to_free = nullptr;
  SuperVersion* new_superversion =
      new SuperVersion(options_.max_write_buffer_number);

  mutex_.Lock();
1296 1297 1298

  // only allow one thread refitting
  if (refitting_level_) {
I
Igor Canadi 已提交
1299
    mutex_.Unlock();
1300
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
I
Igor Canadi 已提交
1301
    delete new_superversion;
L
Lei Jin 已提交
1302
    return Status::NotSupported("another thread is refitting");
1303 1304 1305 1306 1307
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1308
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1309
    Log(options_.info_log,
1310 1311
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1312 1313 1314 1315
    bg_cv_.Wait();
  }

  // move to a smaller level
1316 1317 1318 1319
  int to_level = target_level;
  if (target_level < 0) {
    to_level = FindMinimumEmptyLevelFitting(level);
  }
1320 1321 1322

  assert(to_level <= level);

L
Lei Jin 已提交
1323
  Status status;
1324 1325
  if (to_level < level) {
    Log(options_.info_log, "Before refitting:\n%s",
1326
        default_cfd_->current->DebugString().data());
1327

1328
    VersionEdit edit;
1329
    for (const auto& f : default_cfd_->current->files_[level]) {
1330
      edit.DeleteFile(level, f->number);
1331 1332
      edit.AddFile(to_level, f->number, f->file_size, f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1333 1334 1335 1336
    }
    Log(options_.info_log, "Apply version edit:\n%s",
        edit.DebugString().data());

1337 1338
    status = versions_->LogAndApply(default_cfd_, &edit, &mutex_,
                                    db_directory_.get());
1339
    superversion_to_free = InstallSuperVersion(default_cfd_, new_superversion);
I
Igor Canadi 已提交
1340
    new_superversion = nullptr;
1341 1342 1343 1344 1345

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

    if (status.ok()) {
      Log(options_.info_log, "After refitting:\n%s",
1346
          default_cfd_->current->DebugString().data());
1347 1348 1349 1350 1351
    }
  }

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
1352 1353 1354 1355

  mutex_.Unlock();
  delete superversion_to_free;
  delete new_superversion;
L
Lei Jin 已提交
1356
  return status;
G
Gabor Cselle 已提交
1357 1358
}

1359
int DBImpl::NumberLevels(const ColumnFamilyHandle& column_family) {
1360
  return options_.num_levels;
1361 1362
}

1363
int DBImpl::MaxMemCompactionLevel(const ColumnFamilyHandle& column_family) {
1364
  return options_.max_mem_compaction_level;
1365 1366
}

1367
int DBImpl::Level0StopWriteTrigger(const ColumnFamilyHandle& column_family) {
1368
  return options_.level0_stop_writes_trigger;
1369 1370
}

T
Tomislav Novak 已提交
1371 1372 1373 1374
uint64_t DBImpl::CurrentVersionNumber() const {
  return super_version_number_.load();
}

1375 1376
Status DBImpl::Flush(const FlushOptions& options,
                     const ColumnFamilyHandle& column_family) {
H
heyongqiang 已提交
1377 1378 1379 1380
  Status status = FlushMemTable(options);
  return status;
}

1381
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1382 1383 1384
  return versions_->LastSequence();
}

1385
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
1386
                               unique_ptr<TransactionLogIterator>* iter) {
1387

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

1395 1396
  std::unique_ptr<VectorLogPtr> wal_files(new VectorLogPtr);
  Status s = GetSortedWalFiles(*wal_files);
1397 1398 1399 1400
  if (!s.ok()) {
    return s;
  }

1401
  s = RetainProbableWalFiles(*wal_files, seq);
1402 1403
  if (!s.ok()) {
    return s;
1404
  }
1405
  iter->reset(
1406
    new TransactionLogIteratorImpl(options_.wal_dir,
1407
                                   &options_,
1408
                                   storage_options_,
1409
                                   seq,
1410
                                   std::move(wal_files),
1411
                                   this));
1412
  return (*iter)->status();
1413 1414
}

1415 1416
Status DBImpl::RetainProbableWalFiles(VectorLogPtr& all_logs,
                                      const SequenceNumber target) {
1417
  long start = 0; // signed to avoid overflow when target is < first file.
1418
  long end = static_cast<long>(all_logs.size()) - 1;
1419
  // Binary Search. avoid opening all files.
1420 1421
  while (end >= start) {
    long mid = start + (end - start) / 2;  // Avoid overflow.
1422 1423
    SequenceNumber current_seq_num = all_logs.at(mid)->StartSequence();
    if (current_seq_num == target) {
1424
      end = mid;
1425
      break;
1426
    } else if (current_seq_num < target) {
1427
      start = mid + 1;
1428
    } else {
1429
      end = mid - 1;
1430 1431
    }
  }
1432 1433 1434
  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);
1435 1436 1437
  return Status::OK();
}

1438 1439 1440
bool DBImpl::CheckWalFileExistsAndEmpty(const WalFileType type,
                                        const uint64_t number) {
  const std::string fname = (type == kAliveLogFile) ?
1441 1442
    LogFileName(options_.wal_dir, number) :
    ArchivedLogFileName(options_.wal_dir, number);
1443 1444
  uint64_t file_size;
  Status s = env_->GetFileSize(fname, &file_size);
1445
  return (s.ok() && (file_size == 0));
1446 1447
}

1448 1449
Status DBImpl::ReadFirstRecord(const WalFileType type, const uint64_t number,
                               WriteBatch* const result) {
1450

1451
  if (type == kAliveLogFile) {
1452
    std::string fname = LogFileName(options_.wal_dir, number);
1453 1454 1455
    Status status = ReadFirstLine(fname, result);
    if (!status.ok()) {
      //  check if the file got moved to archive.
1456 1457
      std::string archived_file =
        ArchivedLogFileName(options_.wal_dir, number);
1458
      Status s = ReadFirstLine(archived_file, result);
1459
      if (!s.ok()) {
1460
        return Status::IOError("Log File has been deleted: " + archived_file);
1461 1462 1463
      }
    }
    return Status::OK();
1464
  } else if (type == kArchivedLogFile) {
1465
    std::string fname = ArchivedLogFileName(options_.wal_dir, number);
1466 1467 1468
    Status status = ReadFirstLine(fname, result);
    return status;
  }
1469
  return Status::NotSupported("File Type Not Known: " + std::to_string(type));
1470 1471 1472 1473 1474 1475 1476 1477
}

Status DBImpl::ReadFirstLine(const std::string& fname,
                             WriteBatch* const batch) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;
1478
    Status* status;  // nullptr if options_.paranoid_checks==false
1479 1480
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
1481
          (this->status == nullptr ? "(ignoring error) " : ""),
1482
          fname, static_cast<int>(bytes), s.ToString().c_str());
1483
      if (this->status != nullptr && this->status->ok()) *this->status = s;
1484 1485 1486
    }
  };

1487
  unique_ptr<SequentialFile> file;
1488
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
1489 1490 1491 1492 1493 1494 1495 1496

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


  LogReporter reporter;
  reporter.env = env_;
1497
  reporter.info_log = options_.info_log.get();
1498
  reporter.fname = fname.c_str();
1499
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
1500
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1501 1502 1503
                     0/*initial_offset*/);
  std::string scratch;
  Slice record;
1504

1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517
  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);
}

1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530
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);
1531 1532 1533
  if (!status.ok()) {
    return status;
  }
1534
  log_files.reserve(log_files.size() + all_files.size());
1535 1536 1537 1538 1539 1540
  VectorLogPtr::iterator pos_start;
  if (!log_files.empty()) {
    pos_start = log_files.end() - 1;
  } else {
    pos_start = log_files.begin();
  }
1541
  for (const auto& f : all_files) {
1542 1543
    uint64_t number;
    FileType type;
1544
    if (ParseFileName(f, &number, &type) && type == kLogFile){
1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562

      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))));
1563 1564
    }
  }
1565
  CompareLogByPointer compare_log_files;
1566
  std::sort(pos_start, log_files.end(), compare_log_files);
1567 1568 1569
  return status;
}

L
Lei Jin 已提交
1570 1571 1572 1573
Status DBImpl::RunManualCompaction(int input_level,
                                   int output_level,
                                   const Slice* begin,
                                   const Slice* end) {
1574
  assert(input_level >= 0);
1575

G
Gabor Cselle 已提交
1576 1577
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1578
  ManualCompaction manual;
1579 1580
  manual.input_level = input_level;
  manual.output_level = output_level;
G
Gabor Cselle 已提交
1581
  manual.done = false;
1582
  manual.in_progress = false;
1583 1584 1585 1586
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
      options_.compaction_style == kCompactionStyleUniversal) {
1587
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1588 1589 1590 1591
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1592 1593
  if (end == nullptr ||
      options_.compaction_style == kCompactionStyleUniversal) {
1594
    manual.end = nullptr;
G
Gabor Cselle 已提交
1595 1596 1597 1598 1599 1600
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1601

1602 1603 1604 1605 1606 1607
  // 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
1608
  // RunManualCompaction(), i.e. during that time no other compaction will
1609 1610 1611
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
1612
  // RunManualCompaction() from getting to the second while loop below.
1613 1614 1615 1616 1617 1618 1619 1620
  // 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");
1621 1622
    bg_cv_.Wait();
  }
1623

1624 1625
  Log(options_.info_log, "Manual compaction starting");

1626 1627 1628 1629
  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 已提交
1630
      bg_cv_.Wait();
1631 1632 1633
    } else {
      manual_compaction_ = &manual;
      MaybeScheduleFlushOrCompaction();
G
Gabor Cselle 已提交
1634
    }
H
hans@chromium.org 已提交
1635
  }
1636

1637 1638 1639
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
L
Lei Jin 已提交
1640
  return manual.status;
J
jorlow@chromium.org 已提交
1641 1642
}

L
Lei Jin 已提交
1643 1644 1645
Status DBImpl::TEST_CompactRange(int level,
                                 const Slice* begin,
                                 const Slice* end) {
1646 1647 1648
  int output_level = (options_.compaction_style == kCompactionStyleUniversal)
                         ? level
                         : level + 1;
L
Lei Jin 已提交
1649
  return RunManualCompaction(level, output_level, begin, end);
1650 1651
}

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

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

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

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

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

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

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

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

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

1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735
      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 已提交
1736 1737
Status DBImpl::BackgroundFlush(bool* madeProgress,
                               DeletionState& deletion_state) {
1738
  Status stat;
1739
  while (stat.ok() && default_cfd_->imm.IsFlushPending()) {
1740
    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 1795
uint64_t DBImpl::TEST_GetLevel0TotalSize() {
  MutexLock l(&mutex_);
1796
  return default_cfd_->current->NumLevelBytes(0);
1797 1798
}

1799
void DBImpl::BackgroundCallCompaction() {
1800
  bool madeProgress = false;
I
Igor Canadi 已提交
1801
  DeletionState deletion_state(options_.max_write_buffer_number, true);
H
Haobo Xu 已提交
1802 1803 1804

  MaybeDumpStats();

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

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

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

1839
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1840

1841 1842
  MaybeScheduleLogDBDeployStats();

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

J
jorlow@chromium.org 已提交
1851 1852
}

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

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

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

  Status status;
1899
  if (!c) {
H
hans@chromium.org 已提交
1900
    // Nothing to do
1901
    Log(options_.info_log, "Compaction nothing to do");
H
hans@chromium.org 已提交
1902
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
1903
    // Move file to next level
1904
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
1905 1906 1907
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
1908 1909
                       f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
1910 1911
    status = versions_->LogAndApply(default_cfd_, c->edit(), &mutex_,
                                    db_directory_.get());
1912 1913
    InstallSuperVersion(default_cfd_, deletion_state);

1914
    Version::LevelSummaryStorage tmp;
1915
    Log(options_.info_log, "Moved #%lld to level-%d %lld bytes %s: %s\n",
1916
        static_cast<unsigned long long>(f->number), c->level() + 1,
J
jorlow@chromium.org 已提交
1917
        static_cast<unsigned long long>(f->file_size),
1918
        status.ToString().c_str(), default_cfd_->current->LevelSummary(&tmp));
1919
    versions_->ReleaseCompactionFiles(c.get(), status);
1920
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1921
  } else {
1922
    MaybeScheduleFlushOrCompaction(); // do more compaction work in parallel.
1923
    CompactionState* compact = new CompactionState(c.get());
I
Igor Canadi 已提交
1924
    status = DoCompactionWork(compact, deletion_state);
1925
    CleanupCompaction(compact, status);
1926
    versions_->ReleaseCompactionFiles(c.get(), status);
1927
    c->ReleaseInputs();
1928
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1929
  }
1930
  c.reset();
J
jorlow@chromium.org 已提交
1931 1932 1933 1934 1935 1936

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

  if (is_manual) {
G
Gabor Cselle 已提交
1945
    ManualCompaction* m = manual_compaction_;
1946
    if (!status.ok()) {
L
Lei Jin 已提交
1947
      m->status = status;
1948 1949
      m->done = true;
    }
1950 1951 1952 1953 1954 1955 1956 1957 1958
    // 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.
1959 1960 1961 1962 1963
    //
    // Stop the compaction if manual_end points to nullptr -- this means
    // that we compacted the whole range. manual_end should always point
    // to nullptr in case of universal compaction
    if (manual_end == nullptr) {
1964 1965
      m->done = true;
    }
G
Gabor Cselle 已提交
1966 1967 1968
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
1969 1970 1971
      // Universal compaction should always compact the whole range
      assert(options_.compaction_style != kCompactionStyleUniversal);
      m->tmp_storage = *manual_end;
G
Gabor Cselle 已提交
1972 1973
      m->begin = &m->tmp_storage;
    }
1974
    m->in_progress = false; // not being processed anymore
1975
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1976
  }
1977
  return status;
J
jorlow@chromium.org 已提交
1978 1979
}

1980
void DBImpl::CleanupCompaction(CompactionState* compact, Status status) {
J
jorlow@chromium.org 已提交
1981
  mutex_.AssertHeld();
1982
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
1983 1984
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
1985
    compact->builder.reset();
J
jorlow@chromium.org 已提交
1986
  } else {
1987
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
1988
  }
D
dgrogan@chromium.org 已提交
1989
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1990 1991
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
1992 1993 1994 1995 1996 1997

    // 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 已提交
1998 1999 2000 2001
  }
  delete compact;
}

2002
// Allocate the file numbers for the output file. We allocate as
2003
// many output file numbers as there are files in level+1 (at least one)
2004 2005 2006
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2007 2008
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
2009
  int filesNeeded = compact->compaction->num_input_files(1);
2010
  for (int i = 0; i < std::max(filesNeeded, 1); i++) {
2011 2012 2013 2014 2015 2016 2017 2018 2019
    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();
2020
  for (const auto file_number : compact->allocated_file_numbers) {
2021 2022 2023 2024 2025
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
2026
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
2027 2028
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
2029
  uint64_t file_number;
2030 2031 2032 2033 2034 2035 2036
  // 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 已提交
2037 2038 2039 2040 2041
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
2042 2043 2044 2045
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
2046
  out.smallest_seqno = out.largest_seqno = 0;
2047
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
2048 2049 2050

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

J
jorlow@chromium.org 已提交
2053
  if (s.ok()) {
2054 2055 2056
    // Over-estimate slightly so we don't end up just barely crossing
    // the threshold.
    compact->outfile->SetPreallocationBlockSize(
2057
      1.1 * versions_->MaxFileSizeForLevel(compact->compaction->output_level()));
2058

S
Siying Dong 已提交
2059 2060 2061 2062
    CompressionType compression_type = GetCompressionType(
        options_, compact->compaction->output_level(),
        compact->compaction->enable_compression());

S
Siying Dong 已提交
2063
    compact->builder.reset(
S
Siying Dong 已提交
2064
        GetTableBuilder(options_, compact->outfile.get(), compression_type));
J
jorlow@chromium.org 已提交
2065
  }
I
Igor Canadi 已提交
2066
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2067 2068 2069 2070 2071
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
2072
  assert(compact != nullptr);
2073
  assert(compact->outfile);
2074
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089

  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;
2090
  compact->builder.reset();
J
jorlow@chromium.org 已提交
2091 2092

  // Finish and check for file errors
2093
  if (s.ok() && !options_.disableDataSync) {
2094
    if (options_.use_fsync) {
2095
      StopWatch sw(env_, options_.statistics.get(),
2096
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2097 2098
      s = compact->outfile->Fsync();
    } else {
2099
      StopWatch sw(env_, options_.statistics.get(),
2100
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2101 2102
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
2103 2104 2105 2106
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
2107
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
2108 2109 2110

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
2111
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
2112
                                               storage_options_,
J
jorlow@chromium.org 已提交
2113 2114
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
2115 2116 2117
    s = iter->status();
    delete iter;
    if (s.ok()) {
2118
      Log(options_.info_log,
K
Kai Liu 已提交
2119 2120 2121 2122
          "Generated table #%lu: %lu keys, %lu bytes",
          (unsigned long) output_number,
          (unsigned long) current_entries,
          (unsigned long) current_bytes);
J
jorlow@chromium.org 已提交
2123 2124 2125 2126 2127 2128 2129 2130
    }
  }
  return s;
}


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
2131 2132 2133 2134 2135

  // 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.
2136
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
2137 2138 2139 2140 2141 2142 2143 2144
    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");
  }

2145
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
2146 2147 2148 2149 2150 2151 2152 2153
      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());
D
dgrogan@chromium.org 已提交
2154
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2155 2156
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
2157 2158
        compact->compaction->output_level(), out.number, out.file_size,
        out.smallest, out.largest, out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2159
  }
2160 2161
  return versions_->LogAndApply(default_cfd_, compact->compaction->edit(),
                                &mutex_, db_directory_.get());
J
jorlow@chromium.org 已提交
2162 2163
}

2164 2165 2166 2167 2168 2169 2170 2171
//
// 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(
2172 2173
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
2174
  SequenceNumber prev __attribute__((unused)) = 0;
2175 2176 2177
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
2178
      *prev_snapshot = prev;
2179
      return cur;
2180
    }
2181 2182
    prev = cur; // assignment
    assert(prev);
2183 2184
  }
  Log(options_.info_log,
K
Kai Liu 已提交
2185 2186 2187
      "Looking for seqid %lu but maxseqid is %lu",
      (unsigned long)in,
      (unsigned long)snapshots[snapshots.size()-1]);
2188 2189 2190 2191
  assert(0);
  return 0;
}

I
Igor Canadi 已提交
2192 2193
Status DBImpl::DoCompactionWork(CompactionState* compact,
                                DeletionState& deletion_state) {
2194
  assert(compact);
2195
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions
A
Abhishek Kona 已提交
2196
  Log(options_.info_log,
2197
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
2198 2199 2200
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
2201
      compact->compaction->output_level(),
2202
      compact->compaction->score(),
2203
      options_.max_background_compactions - bg_compaction_scheduled_);
2204 2205
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
2206
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
2207

2208 2209
  assert(compact->compaction->input_version()->NumLevelFiles(
             compact->compaction->level()) > 0);
2210
  assert(compact->builder == nullptr);
2211
  assert(!compact->outfile);
2212 2213 2214

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
H
Haobo Xu 已提交
2215
  SequenceNumber latest_snapshot = 0;
2216 2217 2218 2219 2220
  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 已提交
2221
  } else {
H
Haobo Xu 已提交
2222
    latest_snapshot = compact->existing_snapshots.back();
2223 2224 2225 2226
    // 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 已提交
2227 2228
  }

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

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

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

2238
  const uint64_t start_micros = env_->NowMicros();
2239
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
2240 2241 2242 2243 2244
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
2245 2246
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
2247
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2248
  std::string compaction_filter_value;
H
Haobo Xu 已提交
2249
  std::vector<char> delete_key; // for compaction filter
2250
  MergeHelper merge(user_comparator(), options_.merge_operator.get(),
2251 2252
                    options_.info_log.get(),
                    false /* internal key corruption is expected */);
2253 2254 2255
  auto compaction_filter = options_.compaction_filter;
  std::unique_ptr<CompactionFilter> compaction_filter_from_factory = nullptr;
  if (!compaction_filter) {
2256 2257 2258
    auto context = compact->GetFilterContext();
    compaction_filter_from_factory =
      options_.compaction_filter_factory->CreateCompactionFilter(context);
2259 2260
    compaction_filter = compaction_filter_from_factory.get();
  }
2261

J
jorlow@chromium.org 已提交
2262
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
2263
    // Prioritize immutable compaction work
2264
    // TODO: remove memtable flush from normal compaction work
2265
    if (default_cfd_->imm.imm_flush_needed.NoBarrier_Load() != nullptr) {
2266
      const uint64_t imm_start = env_->NowMicros();
I
Igor Canadi 已提交
2267
      LogFlush(options_.info_log);
2268
      mutex_.Lock();
2269
      if (default_cfd_->imm.IsFlushPending()) {
I
Igor Canadi 已提交
2270
        FlushMemTableToOutputFile(nullptr, deletion_state);
H
hans@chromium.org 已提交
2271
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
2272 2273 2274 2275 2276
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
2277
    Slice key = input->key();
2278
    Slice value = input->value();
H
Haobo Xu 已提交
2279

2280
    if (compact->compaction->ShouldStopBefore(key) &&
2281
        compact->builder != nullptr) {
2282
      status = FinishCompactionOutputFile(compact, input.get());
2283 2284 2285 2286 2287 2288
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2289
    bool drop = false;
2290
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2291 2292
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2293 2294
      // 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 已提交
2295 2296 2297
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2298
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2299 2300 2301 2302 2303 2304 2305 2306
    } 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;
2307
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2308 2309

        // apply the compaction filter to the first occurrence of the user key
2310
        if (compaction_filter &&
H
Haobo Xu 已提交
2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
            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 =
2321
            compaction_filter->Filter(compact->compaction->level(),
S
Siying Dong 已提交
2322 2323 2324
                                               ikey.user_key, value,
                                               &compaction_filter_value,
                                               &value_changed);
H
Haobo Xu 已提交
2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336
          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();
2337
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2338 2339 2340 2341 2342
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }

J
jorlow@chromium.org 已提交
2343 2344
      }

2345 2346 2347
      // 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.
2348 2349 2350 2351 2352 2353
      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);
2354 2355 2356 2357 2358

      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 已提交
2359
        // Hidden by an newer entry for same user key
2360
        // TODO: why not > ?
2361
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2362
        drop = true;    // (A)
2363
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2364
      } else if (ikey.type == kTypeDeletion &&
2365
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
2366 2367 2368 2369 2370 2371 2372 2373 2374
                 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;
2375
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2376 2377 2378 2379 2380 2381 2382
      } 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 已提交
2383
        merge.MergeUntil(input.get(), prev_snapshot, bottommost_level,
2384
                         options_.statistics.get());
2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402
        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 已提交
2403 2404 2405
      }

      last_sequence_for_key = ikey.sequence;
2406
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
2407 2408
    }
#if 0
2409
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
2410
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
2411
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
2412
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
2413
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
2414
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
2415 2416
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
2417 2418 2419
#endif

    if (!drop) {
2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435
      // 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);
2436
      }
2437

2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448
      // 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.
2449 2450
        if (options_.compaction_style == kCompactionStyleLevel &&
            bottommost_level && ikey.sequence < earliest_snapshot &&
2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461
            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
2462

2463 2464 2465 2466 2467 2468 2469
        // Open output file if necessary
        if (compact->builder == nullptr) {
          status = OpenCompactionOutputFile(compact);
          if (!status.ok()) {
            break;
          }
        }
2470 2471

        SequenceNumber seqno = GetInternalKeySeqno(newkey);
2472 2473
        if (compact->builder->NumEntries() == 0) {
          compact->current_output()->smallest.DecodeFrom(newkey);
2474 2475 2476 2477
          compact->current_output()->smallest_seqno = seqno;
        } else {
          compact->current_output()->smallest_seqno =
            std::min(compact->current_output()->smallest_seqno, seqno);
2478 2479 2480
        }
        compact->current_output()->largest.DecodeFrom(newkey);
        compact->builder->Add(newkey, value);
2481 2482
        compact->current_output()->largest_seqno =
          std::max(compact->current_output()->largest_seqno, seqno);
2483 2484 2485 2486 2487 2488 2489 2490

        // 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 已提交
2491 2492
        }

2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512
        // 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 已提交
2513 2514 2515 2516 2517
          break;
        }
      }
    }

2518
    // MergeUntil has moved input to the next entry
2519
    if (!current_entry_is_merging) {
2520 2521
      input->Next();
    }
J
jorlow@chromium.org 已提交
2522 2523 2524
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
2525
    status = Status::IOError("Database shutdown started during compaction");
J
jorlow@chromium.org 已提交
2526
  }
2527
  if (status.ok() && compact->builder != nullptr) {
2528
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
2529 2530 2531 2532
  }
  if (status.ok()) {
    status = input->status();
  }
2533
  input.reset();
J
jorlow@chromium.org 已提交
2534

2535 2536 2537
  if (!options_.disableDataSync) {
    db_directory_->Fsync();
  }
2538 2539
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
I
Igor Canadi 已提交
2540
  MeasureTime(options_.statistics.get(), COMPACTION_TIME, stats.micros);
M
Mark Callaghan 已提交
2541 2542
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
2543 2544

  int num_output_files = compact->outputs.size();
2545
  if (compact->builder != nullptr) {
P
Pascal Borreli 已提交
2546
    // An error occurred so ignore the last output.
2547 2548 2549 2550
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
2551

2552
  for (int i = 0; i < compact->compaction->num_input_files(0); i++) {
M
Mark Callaghan 已提交
2553
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;
2554 2555 2556
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(0, i)->file_size);
  }
M
Mark Callaghan 已提交
2557

2558
  for (int i = 0; i < compact->compaction->num_input_files(1); i++) {
M
Mark Callaghan 已提交
2559
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;
2560 2561 2562
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(1, i)->file_size);
  }
M
Mark Callaghan 已提交
2563

2564
  for (int i = 0; i < num_output_files; i++) {
2565
    stats.bytes_written += compact->outputs[i].file_size;
2566 2567
    RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
               compact->outputs[i].file_size);
2568 2569
  }

I
Igor Canadi 已提交
2570
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2571
  mutex_.Lock();
2572
  stats_[compact->compaction->output_level()].Add(stats);
J
jorlow@chromium.org 已提交
2573

2574 2575 2576 2577
  // 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 已提交
2578 2579
  if (status.ok()) {
    status = InstallCompactionResults(compact);
2580
    InstallSuperVersion(default_cfd_, deletion_state);
J
jorlow@chromium.org 已提交
2581
  }
2582
  Version::LevelSummaryStorage tmp;
2583
  Log(options_.info_log,
M
Mark Callaghan 已提交
2584
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
2585 2586
      "MB in(%.1f, %.1f) out(%.1f), read-write-amplify(%.1f) "
      "write-amplify(%.1f) %s\n",
2587
      compact->compaction->input_version()->LevelSummary(&tmp),
M
Mark Callaghan 已提交
2588
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
2589 2590 2591 2592
          (double)stats.micros,
      compact->compaction->output_level(), stats.files_in_leveln,
      stats.files_in_levelnp1, stats.files_out_levelnp1,
      stats.bytes_readn / 1048576.0, stats.bytes_readnp1 / 1048576.0,
M
Mark Callaghan 已提交
2593
      stats.bytes_written / 1048576.0,
2594
      (stats.bytes_written + stats.bytes_readnp1 + stats.bytes_readn) /
2595 2596
          (double)stats.bytes_readn,
      stats.bytes_written / (double)stats.bytes_readn,
2597
      status.ToString().c_str());
M
Mark Callaghan 已提交
2598

J
jorlow@chromium.org 已提交
2599 2600 2601
  return status;
}

2602 2603 2604
namespace {
struct IterState {
  port::Mutex* mu;
I
Igor Canadi 已提交
2605 2606 2607
  Version* version = nullptr;
  MemTable* mem = nullptr;
  MemTableListVersion* imm = nullptr;
I
Igor Canadi 已提交
2608
  DBImpl *db;
2609 2610 2611 2612
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
2613 2614
  DBImpl::DeletionState deletion_state(state->db->GetOptions().
                                       max_write_buffer_number);
2615
  state->mu->Lock();
2616 2617 2618 2619 2620
  if (state->mem) { // not set for immutable iterator
    MemTable* m = state->mem->Unref();
    if (m != nullptr) {
      deletion_state.memtables_to_free.push_back(m);
    }
2621
  }
T
Tomislav Novak 已提交
2622 2623 2624
  if (state->version) {  // not set for memtable-only iterator
    state->version->Unref();
  }
I
Igor Canadi 已提交
2625 2626 2627
  if (state->imm) {  // not set for memtable-only iterator
    state->imm->Unref(&deletion_state.memtables_to_free);
  }
I
Igor Canadi 已提交
2628 2629
  // fast path FindObsoleteFiles
  state->db->FindObsoleteFiles(deletion_state, false, true);
2630
  state->mu->Unlock();
I
Igor Canadi 已提交
2631
  state->db->PurgeObsoleteFiles(deletion_state);
T
Tomislav Novak 已提交
2632

2633 2634
  delete state;
}
H
Hans Wennborg 已提交
2635
}  // namespace
2636

J
jorlow@chromium.org 已提交
2637 2638
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
2639
  IterState* cleanup = new IterState;
2640
  MemTable* mutable_mem;
I
Igor Canadi 已提交
2641
  MemTableListVersion* immutable_mems;
2642
  Version* version;
J
jorlow@chromium.org 已提交
2643

2644
  // Collect together all needed child iterators for mem
2645 2646
  mutex_.Lock();
  *latest_snapshot = versions_->LastSequence();
2647 2648 2649
  mutable_mem = default_cfd_->mem;
  mutable_mem->Ref();
  immutable_mems = default_cfd_->imm.current();
I
Igor Canadi 已提交
2650
  immutable_mems->Ref();
2651 2652
  version = default_cfd_->current;
  version->Ref();
2653 2654
  mutex_.Unlock();

I
Igor Canadi 已提交
2655 2656 2657 2658
  std::vector<Iterator*> iterator_list;
  iterator_list.push_back(mutable_mem->NewIterator(options));
  cleanup->mem = mutable_mem;
  cleanup->imm = immutable_mems;
T
Tomislav Novak 已提交
2659
  // Collect all needed child iterators for immutable memtables
I
Igor Canadi 已提交
2660
  immutable_mems->AddIterators(options, &iterator_list);
T
Tomislav Novak 已提交
2661
  // Collect iterators for files in L0 - Ln
I
Igor Canadi 已提交
2662 2663 2664
  version->AddIterators(options, storage_options_, &iterator_list);
  Iterator* internal_iter = NewMergingIterator(
      &internal_comparator_, &iterator_list[0], iterator_list.size());
2665
  cleanup->version = version;
2666
  cleanup->mu = &mutex_;
I
Igor Canadi 已提交
2667
  cleanup->db = this;
2668
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2669 2670 2671 2672 2673 2674 2675 2676 2677

  return internal_iter;
}

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

T
Tomislav Novak 已提交
2678 2679 2680 2681 2682
std::pair<Iterator*, Iterator*> DBImpl::GetTailingIteratorPair(
    const ReadOptions& options,
    uint64_t* superversion_number) {

  MemTable* mutable_mem;
I
Igor Canadi 已提交
2683
  MemTableListVersion* immutable_mems;
T
Tomislav Novak 已提交
2684 2685 2686 2687
  Version* version;

  // get all child iterators and bump their refcounts under lock
  mutex_.Lock();
2688
  mutable_mem = default_cfd_->mem;
T
Tomislav Novak 已提交
2689
  mutable_mem->Ref();
2690
  immutable_mems = default_cfd_->imm.current();
I
Igor Canadi 已提交
2691
  immutable_mems->Ref();
2692
  version = default_cfd_->current;
T
Tomislav Novak 已提交
2693 2694 2695 2696 2697 2698 2699 2700
  version->Ref();
  if (superversion_number != nullptr) {
    *superversion_number = CurrentVersionNumber();
  }
  mutex_.Unlock();

  Iterator* mutable_iter = mutable_mem->NewIterator(options);
  IterState* mutable_cleanup = new IterState();
I
Igor Canadi 已提交
2701
  mutable_cleanup->mem = mutable_mem;
T
Tomislav Novak 已提交
2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712
  mutable_cleanup->db = this;
  mutable_cleanup->mu = &mutex_;
  mutable_iter->RegisterCleanup(CleanupIteratorState, mutable_cleanup, nullptr);

  // create a DBIter that only uses memtable content; see NewIterator()
  mutable_iter = NewDBIterator(&dbname_, env_, options_, user_comparator(),
                               mutable_iter, kMaxSequenceNumber);

  Iterator* immutable_iter;
  IterState* immutable_cleanup = new IterState();
  std::vector<Iterator*> list;
I
Igor Canadi 已提交
2713 2714
  immutable_mems->AddIterators(options, &list);
  immutable_cleanup->imm = immutable_mems;
T
Tomislav Novak 已提交
2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731
  version->AddIterators(options, storage_options_, &list);
  immutable_cleanup->version = version;
  immutable_cleanup->db = this;
  immutable_cleanup->mu = &mutex_;

  immutable_iter =
    NewMergingIterator(&internal_comparator_, &list[0], list.size());
  immutable_iter->RegisterCleanup(CleanupIteratorState, immutable_cleanup,
                                  nullptr);

  // create a DBIter that only uses memtable content; see NewIterator()
  immutable_iter = NewDBIterator(&dbname_, env_, options_, user_comparator(),
                                 immutable_iter, kMaxSequenceNumber);

  return std::make_pair(mutable_iter, immutable_iter);
}

J
jorlow@chromium.org 已提交
2732
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
2733
  MutexLock l(&mutex_);
2734
  return default_cfd_->current->MaxNextLevelOverlappingBytes();
2735 2736
}

J
jorlow@chromium.org 已提交
2737
Status DBImpl::Get(const ReadOptions& options,
2738
                   const ColumnFamilyHandle& column_family, const Slice& key,
J
jorlow@chromium.org 已提交
2739
                   std::string* value) {
2740 2741 2742
  return GetImpl(options, key, value);
}

I
Igor Canadi 已提交
2743 2744 2745 2746 2747 2748 2749 2750 2751 2752
// 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
2753 2754
void DBImpl::InstallSuperVersion(ColumnFamilyData* cfd,
                                 DeletionState& deletion_state) {
I
Igor Canadi 已提交
2755 2756 2757 2758
  // if new_superversion == nullptr, it means somebody already used it
  SuperVersion* new_superversion =
    (deletion_state.new_superversion != nullptr) ?
    deletion_state.new_superversion : new SuperVersion();
2759
  SuperVersion* old_superversion = InstallSuperVersion(cfd, new_superversion);
I
Igor Canadi 已提交
2760 2761 2762 2763 2764 2765 2766 2767 2768
  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;
  }
}

2769 2770
SuperVersion* DBImpl::InstallSuperVersion(ColumnFamilyData* cfd,
                                          SuperVersion* new_superversion) {
I
Igor Canadi 已提交
2771
  mutex_.AssertHeld();
2772 2773 2774 2775 2776 2777 2778
  new_superversion->Init(cfd->mem, cfd->imm.current(), cfd->current);
  SuperVersion* old_superversion = cfd->super_version;
  cfd->super_version = new_superversion;
  if (cfd->id == 0) {
    // TODO this is only for default column family for now
    ++super_version_number_;
  }
I
Igor Canadi 已提交
2779 2780 2781 2782 2783 2784 2785
  if (old_superversion != nullptr && old_superversion->Unref()) {
    old_superversion->Cleanup();
    return old_superversion; // will let caller delete outside of mutex
  }
  return nullptr;
}

2786 2787 2788
Status DBImpl::GetImpl(const ReadOptions& options,
                       const Slice& key,
                       std::string* value,
2789
                       bool* value_found) {
2790
  Status s;
2791

2792
  StopWatch sw(env_, options_.statistics.get(), DB_GET, false);
2793
  SequenceNumber snapshot;
2794
  if (options.snapshot != nullptr) {
2795 2796 2797
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2798
  }
2799

I
Igor Canadi 已提交
2800 2801
  // This can be replaced by using atomics and spinlock instead of big mutex
  mutex_.Lock();
2802
  SuperVersion* get_version = default_cfd_->super_version->Ref();
2803
  mutex_.Unlock();
I
Igor Canadi 已提交
2804

2805
  bool have_stat_update = false;
2806
  Version::GetStats stats;
2807

2808
  // Prepare to store a list of merge operations if merge occurs.
2809
  MergeContext merge_context;
2810

2811
  // First look in the memtable, then in the immutable memtable (if any).
2812
  // s is both in/out. When in, s could either be OK or MergeInProgress.
2813
  // merge_operands will contain the sequence of merges in the latter case.
2814
  LookupKey lkey(key, snapshot);
I
Igor Canadi 已提交
2815
  if (get_version->mem->Get(lkey, value, &s, merge_context, options_)) {
2816
    // Done
2817
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
2818
  } else if (get_version->imm->Get(lkey, value, &s, merge_context, options_)) {
2819
    // Done
2820
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
2821
  } else {
I
Igor Canadi 已提交
2822 2823
    get_version->current->Get(options, lkey, value, &s, &merge_context, &stats,
                              options_, value_found);
2824
    have_stat_update = true;
2825
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
2826
  }
2827

I
Igor Canadi 已提交
2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848
  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;
2849
  }
2850

2851
  // Note, tickers are atomic now - no lock protection needed any more.
2852 2853
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
2854
  return s;
J
jorlow@chromium.org 已提交
2855 2856
}

2857 2858 2859 2860
std::vector<Status> DBImpl::MultiGet(
    const ReadOptions& options,
    const std::vector<ColumnFamilyHandle>& column_family,
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
2861

2862
  StopWatch sw(env_, options_.statistics.get(), DB_MULTIGET, false);
2863
  SequenceNumber snapshot;
2864 2865
  std::vector<MemTable*> to_delete;

2866
  mutex_.Lock();
2867 2868 2869 2870 2871 2872
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }

2873 2874 2875 2876
  // TODO only works for default column family
  MemTable* mem = default_cfd_->mem;
  MemTableListVersion* imm = default_cfd_->imm.current();
  Version* current = default_cfd_->current;
2877
  mem->Ref();
I
Igor Canadi 已提交
2878
  imm->Ref();
2879 2880 2881 2882 2883 2884 2885 2886
  current->Ref();

  // Unlock while reading from files and memtables

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

2887 2888
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
2889

2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900
  // 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.
2901 2902
  // merge_operands will contain the sequence of merges in the latter case.
  for (int i=0; i<numKeys; ++i) {
2903
    merge_context.Clear();
2904 2905 2906 2907
    Status& s = statList[i];
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
2908
    if (mem->Get(lkey, value, &s, merge_context, options_)) {
2909
      // Done
I
Igor Canadi 已提交
2910
    } else if (imm->Get(lkey, value, &s, merge_context, options_)) {
2911 2912
      // Done
    } else {
2913
      current->Get(options, lkey, value, &s, &merge_context, &stats, options_);
2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925
      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)) {
2926
    MaybeScheduleFlushOrCompaction();
2927
  }
2928
  MemTable* m = mem->Unref();
I
Igor Canadi 已提交
2929
  imm->Unref(&to_delete);
2930
  current->Unref();
2931 2932
  mutex_.Unlock();

2933 2934 2935 2936
  // free up all obsolete memtables outside the mutex
  delete m;
  for (MemTable* v: to_delete) delete v;

2937 2938 2939
  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);
2940 2941 2942 2943

  return statList;
}

2944
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& options,
2945
                                  const std::string& column_family_name,
2946
                                  ColumnFamilyHandle* handle) {
2947
  MutexLock l(&mutex_);
2948
  if (versions_->GetColumnFamilySet()->Exists(column_family_name)) {
I
Igor Canadi 已提交
2949 2950
    return Status::InvalidArgument("Column family already exists");
  }
2951
  VersionEdit edit;
2952
  edit.AddColumnFamily(column_family_name);
I
Igor Canadi 已提交
2953
  handle->id = versions_->GetColumnFamilySet()->GetNextColumnFamilyID();
2954
  edit.SetColumnFamily(handle->id);
2955
  Status s = versions_->LogAndApply(default_cfd_, &edit, &mutex_);
2956 2957
  if (s.ok()) {
    // add to internal data structures
2958
    versions_->CreateColumnFamily(options, &edit);
2959 2960
  }
  return s;
2961 2962 2963
}

Status DBImpl::DropColumnFamily(const ColumnFamilyHandle& column_family) {
2964 2965 2966
  if (column_family.id == 0) {
    return Status::InvalidArgument("Can't drop default column family");
  }
2967
  MutexLock l(&mutex_);
I
Igor Canadi 已提交
2968
  if (!versions_->GetColumnFamilySet()->Exists(column_family.id)) {
2969 2970
    return Status::NotFound("Column family not found");
  }
I
Igor Canadi 已提交
2971 2972 2973
  VersionEdit edit;
  edit.DropColumnFamily();
  edit.SetColumnFamily(column_family.id);
2974
  Status s = versions_->LogAndApply(default_cfd_, &edit, &mutex_);
2975 2976
  if (s.ok()) {
    // remove from internal data structures
2977
    versions_->DropColumnFamily(&edit);
2978 2979
  }
  return s;
2980 2981
}

2982
bool DBImpl::KeyMayExist(const ReadOptions& options,
2983 2984
                         const ColumnFamilyHandle& column_family,
                         const Slice& key, std::string* value,
2985 2986
                         bool* value_found) {
  if (value_found != nullptr) {
K
Kai Liu 已提交
2987 2988
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
2989
  }
2990 2991
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
K
Kai Liu 已提交
2992 2993 2994 2995 2996 2997
  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();
2998 2999
}

3000 3001
Iterator* DBImpl::NewIterator(const ReadOptions& options,
                              const ColumnFamilyHandle& column_family) {
T
Tomislav Novak 已提交
3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016
  Iterator* iter;

  if (options.tailing) {
    iter = new TailingIterator(this, options, user_comparator());
  } else {
    SequenceNumber latest_snapshot;
    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));
  }

T
Tyler Harter 已提交
3017 3018 3019 3020 3021 3022 3023
  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 已提交
3024 3025
}

3026 3027 3028 3029 3030 3031 3032 3033
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 已提交
3034 3035
const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
3036
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
3037 3038 3039 3040
}

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

// Convenience methods
3045 3046 3047 3048
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 已提交
3049 3050
}

3051 3052
Status DBImpl::Merge(const WriteOptions& o,
                     const ColumnFamilyHandle& column_family, const Slice& key,
3053 3054 3055 3056
                     const Slice& val) {
  if (!options_.merge_operator) {
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
3057
    return DB::Merge(o, column_family, key, val);
3058 3059 3060
  }
}

3061 3062 3063 3064
Status DBImpl::Delete(const WriteOptions& options,
                      const ColumnFamilyHandle& column_family,
                      const Slice& key) {
  return DB::Delete(options, column_family, key);
J
jorlow@chromium.org 已提交
3065 3066
}

3067 3068 3069 3070
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
3071
  w.disableWAL = options.disableWAL;
3072
  w.done = false;
3073

3074
  StopWatch sw(env_, options_.statistics.get(), DB_WRITE, false);
I
Igor Canadi 已提交
3075
  mutex_.Lock();
3076 3077 3078 3079
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
3080 3081 3082 3083 3084

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

3085
  if (w.done) {
I
Igor Canadi 已提交
3086
    mutex_.Unlock();
3087
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
3088
    return w.status;
3089 3090
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
3091 3092 3093
  }

  // May temporarily unlock and wait.
I
Igor Canadi 已提交
3094 3095
  SuperVersion* superversion_to_free = nullptr;
  Status status = MakeRoomForWrite(my_batch == nullptr, &superversion_to_free);
D
dgrogan@chromium.org 已提交
3096
  uint64_t last_sequence = versions_->LastSequence();
3097
  Writer* last_writer = &w;
3098
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
3099 3100
    autovector<WriteBatch*> write_batch_group;
    BuildBatchGroup(&last_writer, &write_batch_group);
3101

3102 3103 3104
    // 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
3105
    // into default_cfd_->mem.
3106
    {
3107
      mutex_.Unlock();
3108 3109 3110 3111 3112 3113 3114 3115 3116 3117
      WriteBatch* updates = nullptr;
      if (write_batch_group.size() == 1) {
        updates = write_batch_group[0];
      } else {
        updates = &tmp_batch_;
        for (size_t i = 0; i < write_batch_group.size(); ++i) {
          WriteBatchInternal::Append(updates, write_batch_group[i]);
        }
      }

3118 3119 3120 3121 3122
      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
3123 3124 3125
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3126 3127
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3128 3129
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3130 3131 3132
      }

      if (!options.disableWAL) {
3133 3134
        StopWatchNano timer(env_);
        StartPerfTimer(&timer);
3135 3136 3137 3138
        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());
3139
        BumpPerfTime(&perf_context.wal_write_time, &timer);
H
heyongqiang 已提交
3140
        if (status.ok() && options.sync) {
3141
          if (options_.use_fsync) {
3142
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3143
            status = log_->file()->Fsync();
3144
          } else {
3145
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3146
            status = log_->file()->Sync();
3147
          }
H
heyongqiang 已提交
3148
        }
3149 3150
      }
      if (status.ok()) {
3151 3152
        status = WriteBatchInternal::InsertInto(updates, default_cfd_->mem,
                                                &options_, this,
3153
                                                options_.filter_deletes);
3154 3155 3156 3157 3158 3159 3160
        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!");
        }
I
Igor Canadi 已提交
3161 3162
        SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                       last_sequence);
3163
      }
3164
      if (updates == &tmp_batch_) tmp_batch_.Clear();
3165 3166
      mutex_.Lock();
      if (status.ok()) {
3167
        versions_->SetLastSequence(last_sequence);
3168
      }
J
jorlow@chromium.org 已提交
3169 3170
    }
  }
I
Igor Canadi 已提交
3171 3172 3173
  if (options_.paranoid_checks && !status.ok() && bg_error_.ok()) {
    bg_error_ = status; // stop compaction & fail any further writes
  }
3174

3175 3176 3177 3178 3179 3180 3181
  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
3182
    }
3183 3184
    if (ready == last_writer) break;
  }
3185

3186 3187 3188
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
3189
  }
I
Igor Canadi 已提交
3190 3191
  mutex_.Unlock();
  delete superversion_to_free;
J
jorlow@chromium.org 已提交
3192 3193 3194
  return status;
}

3195
// REQUIRES: Writer list must be non-empty
3196
// REQUIRES: First writer must have a non-nullptr batch
3197 3198
void DBImpl::BuildBatchGroup(Writer** last_writer,
                             autovector<WriteBatch*>* write_batch_group) {
3199 3200
  assert(!writers_.empty());
  Writer* first = writers_.front();
3201
  assert(first->batch != nullptr);
3202 3203

  size_t size = WriteBatchInternal::ByteSize(first->batch);
3204
  write_batch_group->push_back(first->batch);
3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223

  // 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 已提交
3224 3225 3226 3227 3228 3229
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

3230
    if (w->batch != nullptr) {
3231 3232 3233 3234 3235 3236
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

3237
      write_batch_group->push_back(w->batch);
3238 3239 3240 3241 3242
    }
    *last_writer = w;
  }
}

3243 3244 3245
// 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 已提交
3246 3247 3248 3249
// if n < bottom, return 0;
// if n >= top, return 1000;
// otherwise, let r = (n - bottom) /
//                    (top - bottom)
3250 3251 3252 3253
//  and return r^2 * 1000.
// The goal of this formula is to gradually increase the rate at which writes
// are slowed. We also tried linear delay (r * 1000), but it seemed to do
// slightly worse. There is no other particular reason for choosing quadratic.
M
Mark Callaghan 已提交
3254
uint64_t DBImpl::SlowdownAmount(int n, double bottom, double top) {
3255
  uint64_t delay;
J
Jim Paton 已提交
3256
  if (n >= top) {
3257 3258
    delay = 1000;
  }
J
Jim Paton 已提交
3259
  else if (n < bottom) {
3260 3261 3262 3263
    delay = 0;
  }
  else {
    // If we are here, we know that:
J
Jim Paton 已提交
3264
    //   level0_start_slowdown <= n < level0_slowdown
3265
    // since the previous two conditions are false.
M
Mark Callaghan 已提交
3266 3267
    double how_much =
      (double) (n - bottom) /
J
Jim Paton 已提交
3268
              (top - bottom);
M
Mark Callaghan 已提交
3269
    delay = std::max(how_much * how_much * 1000, 100.0);
3270 3271 3272 3273 3274
  }
  assert(delay <= 1000);
  return delay;
}

3275
// REQUIRES: mutex_ is held
3276
// REQUIRES: this thread is currently at the front of the writer queue
I
Igor Canadi 已提交
3277 3278
Status DBImpl::MakeRoomForWrite(bool force,
                                SuperVersion** superversion_to_free) {
3279
  mutex_.AssertHeld();
3280
  assert(!writers_.empty());
3281
  bool allow_delay = !force;
J
Jim Paton 已提交
3282 3283
  bool allow_hard_rate_limit_delay = !force;
  bool allow_soft_rate_limit_delay = !force;
3284
  uint64_t rate_limit_delay_millis = 0;
3285
  Status s;
3286
  double score;
I
Igor Canadi 已提交
3287
  *superversion_to_free = nullptr;
3288

3289 3290 3291 3292 3293
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
3294
    } else if (allow_delay && versions_->NeedSlowdownForNumLevel0Files()) {
3295 3296 3297
      // 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
3298
      // individual write by 0-1ms to reduce latency variance.  Also,
3299 3300
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
3301
      uint64_t slowdown =
3302
          SlowdownAmount(default_cfd_->current->NumLevelFiles(0),
3303 3304
                         options_.level0_slowdown_writes_trigger,
                         options_.level0_stop_writes_trigger);
3305
      mutex_.Unlock();
3306
      uint64_t delayed;
J
Jim Paton 已提交
3307
      {
3308
        StopWatch sw(env_, options_.statistics.get(), STALL_L0_SLOWDOWN_COUNT);
3309
        env_->SleepForMicroseconds(slowdown);
3310
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3311
      }
3312
      RecordTick(options_.statistics.get(), STALL_L0_SLOWDOWN_MICROS, delayed);
3313
      stall_level0_slowdown_ += delayed;
J
Jim Paton 已提交
3314
      stall_level0_slowdown_count_++;
3315 3316
      allow_delay = false;  // Do not delay a single write more than once
      mutex_.Lock();
3317
      delayed_writes_++;
3318 3319
    } else if (!force && (default_cfd_->mem->ApproximateMemoryUsage() <=
                          options_.write_buffer_size)) {
3320
      // There is room in current memtable
3321 3322 3323
      if (allow_delay) {
        DelayLoggingAndReset();
      }
3324
      break;
3325 3326
    } else if (default_cfd_->imm.size() ==
               options_.max_write_buffer_number - 1) {
3327
      // We have filled up the current memtable, but the previous
3328 3329
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
3330
      Log(options_.info_log, "wait for memtable compaction...\n");
3331
      uint64_t stall;
J
Jim Paton 已提交
3332
      {
3333
        StopWatch sw(env_, options_.statistics.get(),
J
Jim Paton 已提交
3334 3335
          STALL_MEMTABLE_COMPACTION_COUNT);
        bg_cv_.Wait();
3336
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3337
      }
3338 3339
      RecordTick(options_.statistics.get(),
                 STALL_MEMTABLE_COMPACTION_MICROS, stall);
3340
      stall_memtable_compaction_ += stall;
J
Jim Paton 已提交
3341
      stall_memtable_compaction_count_++;
3342
    } else if (default_cfd_->current->NumLevelFiles(0) >=
3343
               options_.level0_stop_writes_trigger) {
3344
      // There are too many level-0 files.
3345 3346
      DelayLoggingAndReset();
      Log(options_.info_log, "wait for fewer level0 files...\n");
3347
      uint64_t stall;
J
Jim Paton 已提交
3348
      {
3349 3350
        StopWatch sw(env_, options_.statistics.get(),
                     STALL_L0_NUM_FILES_COUNT);
J
Jim Paton 已提交
3351
        bg_cv_.Wait();
3352
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3353
      }
3354
      RecordTick(options_.statistics.get(), STALL_L0_NUM_FILES_MICROS, stall);
3355
      stall_level0_num_files_ += stall;
J
Jim Paton 已提交
3356
      stall_level0_num_files_count_++;
3357
    } else if (allow_hard_rate_limit_delay && options_.hard_rate_limit > 1.0 &&
3358
               (score = default_cfd_->current->MaxCompactionScore()) >
3359
                   options_.hard_rate_limit) {
3360
      // Delay a write when the compaction score for any level is too large.
3361
      int max_level = default_cfd_->current->MaxCompactionScoreLevel();
3362
      mutex_.Unlock();
3363
      uint64_t delayed;
J
Jim Paton 已提交
3364
      {
3365 3366
        StopWatch sw(env_, options_.statistics.get(),
                     HARD_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3367
        env_->SleepForMicroseconds(1000);
3368
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3369
      }
3370
      stall_leveln_slowdown_[max_level] += delayed;
J
Jim Paton 已提交
3371
      stall_leveln_slowdown_count_[max_level]++;
3372
      // Make sure the following value doesn't round to zero.
3373 3374
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
3375 3376
      RecordTick(options_.statistics.get(),
                 RATE_LIMIT_DELAY_MILLIS, rate_limit);
J
Jim Paton 已提交
3377 3378 3379 3380
      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;
3381
      }
3382
      mutex_.Lock();
3383
    } else if (allow_soft_rate_limit_delay && options_.soft_rate_limit > 0.0 &&
3384
               (score = default_cfd_->current->MaxCompactionScore()) >
3385
                   options_.soft_rate_limit) {
J
Jim Paton 已提交
3386 3387 3388
      // Delay a write when the compaction score for any level is too large.
      // TODO: add statistics
      mutex_.Unlock();
J
Jim Paton 已提交
3389
      {
3390 3391
        StopWatch sw(env_, options_.statistics.get(),
                     SOFT_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3392 3393 3394 3395 3396 3397 3398
        env_->SleepForMicroseconds(SlowdownAmount(
          score,
          options_.soft_rate_limit,
          options_.hard_rate_limit)
        );
        rate_limit_delay_millis += sw.ElapsedMicros();
      }
J
Jim Paton 已提交
3399 3400
      allow_soft_rate_limit_delay = false;
      mutex_.Lock();
3401

3402
    } else {
3403 3404 3405 3406 3407
      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.
3408 3409
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
I
Igor Canadi 已提交
3410
      SuperVersion* new_superversion = nullptr;
3411 3412 3413 3414 3415
      mutex_.Unlock();
      {
        EnvOptions soptions(storage_options_);
        soptions.use_mmap_writes = false;
        DelayLoggingAndReset();
3416 3417
        s = env_->NewWritableFile(LogFileName(options_.wal_dir, new_log_number),
                                  &lfile, soptions);
3418 3419 3420 3421
        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);
3422
          memtmp = new MemTable(internal_comparator_, options_);
I
Igor Canadi 已提交
3423
          new_superversion = new SuperVersion(options_.max_write_buffer_number);
3424 3425 3426
        }
      }
      mutex_.Lock();
3427
      if (!s.ok()) {
H
heyongqiang 已提交
3428
        // Avoid chewing through file number space in a tight loop.
3429
        versions_->ReuseFileNumber(new_log_number);
3430
        assert (!memtmp);
3431 3432
        break;
      }
3433
      logfile_number_ = new_log_number;
3434
      log_.reset(new log::Writer(std::move(lfile)));
3435 3436
      default_cfd_->mem->SetNextLogNumber(logfile_number_);
      default_cfd_->imm.Add(default_cfd_->mem);
3437
      if (force) {
3438
        default_cfd_->imm.FlushRequested();
3439
      }
3440 3441
      default_cfd_->mem = memtmp;
      default_cfd_->mem->Ref();
3442
      Log(options_.info_log,
K
Kai Liu 已提交
3443 3444
          "New memtable created with log file: #%lu\n",
          (unsigned long)logfile_number_);
3445
      default_cfd_->mem->SetLogNumber(logfile_number_);
3446
      force = false;   // Do not force another compaction if have room
3447
      MaybeScheduleFlushOrCompaction();
3448 3449
      *superversion_to_free =
          InstallSuperVersion(default_cfd_, new_superversion);
3450 3451 3452 3453 3454
    }
  }
  return s;
}

I
Igor Canadi 已提交
3455 3456 3457 3458
const std::string& DBImpl::GetName() const {
  return dbname_;
}

3459 3460 3461 3462
Env* DBImpl::GetEnv() const {
  return env_;
}

3463 3464
const Options& DBImpl::GetOptions(const ColumnFamilyHandle& column_family)
    const {
I
Igor Canadi 已提交
3465 3466 3467
  return options_;
}

3468 3469
bool DBImpl::GetProperty(const ColumnFamilyHandle& column_family,
                         const Slice& property, std::string* value) {
3470 3471
  value->clear();

J
jorlow@chromium.org 已提交
3472
  MutexLock l(&mutex_);
3473
  Version* current = default_cfd_->current;
J
jorlow@chromium.org 已提交
3474
  Slice in = property;
3475
  Slice prefix("rocksdb.");
J
jorlow@chromium.org 已提交
3476 3477 3478 3479 3480 3481 3482
  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();
3483
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
3484 3485
      return false;
    } else {
3486
      char buf[100];
D
dgrogan@chromium.org 已提交
3487
      snprintf(buf, sizeof(buf), "%d",
3488
               current->NumLevelFiles(static_cast<int>(level)));
3489
      *value = buf;
J
jorlow@chromium.org 已提交
3490 3491
      return true;
    }
3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502
  } 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,
3503
               current->NumLevelFiles(level),
3504
               current->NumLevelBytes(level) / 1048576.0);
3505 3506 3507 3508
      value->append(buf);
    }
    return true;

3509
  } else if (in == "stats") {
M
Mark Callaghan 已提交
3510
    char buf[1000];
3511 3512 3513 3514 3515 3516 3517

    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;
3518 3519
    uint64_t total_bytes_written = 0;
    uint64_t total_bytes_read = 0;
M
Mark Callaghan 已提交
3520
    uint64_t micros_up = env_->NowMicros() - started_at_;
3521 3522
    // Add "+1" to make sure seconds_up is > 0 and avoid NaN later
    double seconds_up = (micros_up + 1) / 1000000.0;
3523
    uint64_t total_slowdown = 0;
J
Jim Paton 已提交
3524
    uint64_t total_slowdown_count = 0;
3525 3526 3527 3528
    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 已提交
3529

3530 3531 3532 3533 3534 3535 3536 3537 3538 3539
    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 已提交
3540
    // Pardon the long line but I think it is easier to read this way.
3541 3542
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
3543 3544
             "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   msComp   msStall  Ln-stall Stall-cnt\n"
             "------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
3545 3546
             );
    value->append(buf);
3547
    for (int level = 0; level < current->NumberLevels(); level++) {
3548
      int files = current->NumLevelFiles(level);
3549
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
3550 3551 3552 3553 3554 3555
        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
3556 3557 3558
            : (stats_[level].bytes_written +
               stats_[level].bytes_readnp1 +
               stats_[level].bytes_readn) /
M
Mark Callaghan 已提交
3559 3560
                (double) stats_[level].bytes_readn;

3561 3562 3563
        total_bytes_read += bytes_read;
        total_bytes_written += stats_[level].bytes_written;

3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575
        uint64_t stalls = level == 0 ?
            (stall_level0_slowdown_count_ +
             stall_level0_num_files_count_ +
             stall_memtable_compaction_count_) :
            stall_leveln_slowdown_count_[level];

        double stall_us = level == 0 ?
            (stall_level0_slowdown_ +
             stall_level0_num_files_ +
             stall_memtable_compaction_) :
            stall_leveln_slowdown_[level];

3576 3577
        snprintf(
            buf, sizeof(buf),
3578
            "%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 %8d %9.1f %9.1f %9lu\n",
3579 3580
            level,
            files,
3581 3582
            current->NumLevelBytes(level) / 1048576.0,
            current->NumLevelBytes(level) /
3583
                versions_->MaxBytesForLevel(level),
3584
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
3585 3586 3587 3588 3589 3590
            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,
3591 3592
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
3593
            (stats_[level].bytes_written / 1048576.0) /
3594
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
3595 3596 3597 3598
            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,
3599
            stats_[level].count,
3600 3601 3602 3603 3604 3605 3606
            (int) ((double) stats_[level].micros /
                   1000.0 /
                   (stats_[level].count + 1)),
            (double) stall_us / 1000.0 / (stalls + 1),
            stall_us / 1000000.0,
            (unsigned long) stalls);

3607
        total_slowdown += stall_leveln_slowdown_[level];
J
Jim Paton 已提交
3608
        total_slowdown_count += stall_leveln_slowdown_count_[level];
3609 3610 3611
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
3612

3613 3614 3615 3616
    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_;
3617 3618 3619 3620 3621 3622
    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);

3623 3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640
    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 已提交
3641
    snprintf(buf, sizeof(buf),
3642 3643
             "Compaction IO cumulative (GB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
3644
             user_bytes_written / (1048576.0 * 1024),
3645 3646 3647 3648 3649 3650 3651 3652
             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",
3653
             user_bytes_written / 1048576.0 / seconds_up,
3654 3655 3656 3657 3658 3659 3660 3661
             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",
3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693
             (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 已提交
3694 3695
    value->append(buf);

3696 3697 3698 3699 3700 3701 3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717
    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",
3718 3719 3720 3721
             (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 已提交
3722 3723 3724 3725
    value->append(buf);

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
3726
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
3727 3728
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
3729
            stall_memtable_compaction_ / 1000000.0,
3730
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
3731 3732
    value->append(buf);

J
Jim Paton 已提交
3733 3734 3735 3736 3737 3738 3739 3740 3741
    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);

3742 3743 3744
    last_stats_.compaction_bytes_read_ = total_bytes_read;
    last_stats_.compaction_bytes_written_ = total_bytes_written;
    last_stats_.ingest_bytes_ = user_bytes_written;
3745
    last_stats_.seconds_up_ = seconds_up;
3746 3747 3748 3749 3750
    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;
3751

3752
    return true;
G
Gabor Cselle 已提交
3753
  } else if (in == "sstables") {
3754
    *value = default_cfd_->current->DebugString();
G
Gabor Cselle 已提交
3755
    return true;
3756
  } else if (in == "num-immutable-mem-table") {
3757
    *value = std::to_string(default_cfd_->imm.size());
3758
    return true;
J
jorlow@chromium.org 已提交
3759
  }
3760

J
jorlow@chromium.org 已提交
3761 3762 3763
  return false;
}

3764 3765
void DBImpl::GetApproximateSizes(const ColumnFamilyHandle& column_family,
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
3766 3767 3768 3769
  // TODO(opt): better implementation
  Version* v;
  {
    MutexLock l(&mutex_);
3770 3771
    v = default_cfd_->current;
    v->Ref();
J
jorlow@chromium.org 已提交
3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788
  }

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

3789 3790 3791 3792 3793 3794 3795
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

3796 3797 3798
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
3799 3800 3801 3802
  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());
3803 3804 3805
    return Status::InvalidArgument("Invalid file name");
  }

3806 3807 3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819
  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;
  }

3820 3821 3822
  int level;
  FileMetaData metadata;
  int maxlevel = NumberLevels();
3823
  ColumnFamilyData* cfd;
3824
  VersionEdit edit;
I
Igor Canadi 已提交
3825
  DeletionState deletion_state(0, true);
D
Dhruba Borthakur 已提交
3826 3827
  {
    MutexLock l(&mutex_);
3828
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
3829
    if (!status.ok()) {
3830 3831
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
3832 3833 3834
      return Status::InvalidArgument("File not found");
    }
    assert((level > 0) && (level < maxlevel));
3835

D
Dhruba Borthakur 已提交
3836 3837
    // If the file is being compacted no need to delete.
    if (metadata.being_compacted) {
3838
      Log(options_.info_log,
3839
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
3840
      return Status::OK();
3841 3842
    }

D
Dhruba Borthakur 已提交
3843 3844 3845 3846
    // 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++) {
3847
      if (cfd->current->NumLevelFiles(i) != 0) {
D
Dhruba Borthakur 已提交
3848
        Log(options_.info_log,
3849
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
3850 3851 3852 3853
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
3854
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
I
Igor Canadi 已提交
3855
    if (status.ok()) {
3856
      InstallSuperVersion(cfd, deletion_state);
I
Igor Canadi 已提交
3857
    }
I
Igor Canadi 已提交
3858
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
3859
  } // lock released here
I
Igor Canadi 已提交
3860
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
3861 3862
  // remove files outside the db-lock
  PurgeObsoleteFiles(deletion_state);
3863 3864 3865
  return status;
}

3866
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData> *metadata) {
3867 3868 3869 3870
  MutexLock l(&mutex_);
  return versions_->GetLiveFilesMetaData(metadata);
}

3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897
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 已提交
3898 3899
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
3900 3901
Status DB::Put(const WriteOptions& opt, const ColumnFamilyHandle& column_family,
               const Slice& key, const Slice& value) {
3902 3903 3904 3905
  // Pre-allocate size of write batch conservatively.
  // 8 bytes are taken by header, 4 bytes for count, 1 byte for type,
  // and we allocate 11 extra bytes for key length, as well as value length.
  WriteBatch batch(key.size() + value.size() + 24);
3906
  batch.Put(column_family.id, key, value);
J
jorlow@chromium.org 已提交
3907 3908 3909
  return Write(opt, &batch);
}

3910 3911
Status DB::Delete(const WriteOptions& opt,
                  const ColumnFamilyHandle& column_family, const Slice& key) {
J
jorlow@chromium.org 已提交
3912
  WriteBatch batch;
3913
  batch.Delete(column_family.id, key);
J
jorlow@chromium.org 已提交
3914 3915 3916
  return Write(opt, &batch);
}

3917 3918
Status DB::Merge(const WriteOptions& opt,
                 const ColumnFamilyHandle& column_family, const Slice& key,
3919 3920
                 const Slice& value) {
  WriteBatch batch;
3921
  batch.Merge(column_family.id, key, value);
3922 3923 3924
  return Write(opt, &batch);
}

3925 3926
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
3927
                              const std::string& column_family_name,
3928 3929
                              ColumnFamilyHandle* handle) {
  return Status::NotSupported("");
3930 3931
}
Status DB::DropColumnFamily(const ColumnFamilyHandle& column_family) {
3932
  return Status::NotSupported("");
3933 3934
}

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

J
Jim Paton 已提交
3937
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951
  DBOptions db_options(options);
  ColumnFamilyOptions cf_options(options);
  std::vector<ColumnFamilyDescriptor> column_families;
  column_families.push_back(
      ColumnFamilyDescriptor(default_column_family_name, cf_options));
  std::vector<ColumnFamilyHandle> handles;
  return DB::OpenWithColumnFamilies(db_options, dbname, column_families,
                                    &handles, dbptr);
}

Status DB::OpenWithColumnFamilies(
    const DBOptions& db_options, const std::string& dbname,
    const std::vector<ColumnFamilyDescriptor>& column_families,
    std::vector<ColumnFamilyHandle>* handles, DB** dbptr) {
3952
  *dbptr = nullptr;
H
Haobo Xu 已提交
3953
  EnvOptions soptions;
3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964
  // TODO temporary until we change DBImpl to accept
  // DBOptions instead of Options
  ColumnFamilyOptions default_column_family_options;
  for (auto cfd : column_families) {
    if (cfd.name == default_column_family_name) {
      default_column_family_options = cfd.options;
      break;
    }
  }
  // default options
  Options options(db_options, default_column_family_options);
J
jorlow@chromium.org 已提交
3965

3966
  if (options.block_cache != nullptr && options.no_block_cache) {
3967
    return Status::InvalidArgument(
3968
        "no_block_cache is true while block_cache is not nullptr");
3969
  }
3970

J
jorlow@chromium.org 已提交
3971
  DBImpl* impl = new DBImpl(options, dbname);
3972 3973 3974 3975 3976 3977 3978
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
3979 3980 3981 3982
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
3983
  impl->mutex_.Lock();
3984 3985
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
3986
  if (s.ok()) {
3987
    uint64_t new_log_number = impl->versions_->NewFileNumber();
3988
    unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
3989
    soptions.use_mmap_writes = false;
3990
    s = impl->options_.env->NewWritableFile(
3991 3992 3993 3994
      LogFileName(impl->options_.wal_dir, new_log_number),
      &lfile,
      soptions
    );
J
jorlow@chromium.org 已提交
3995
    if (s.ok()) {
3996
      lfile->SetPreallocationBlockSize(1.1 * impl->options_.write_buffer_size);
I
Igor Canadi 已提交
3997
      VersionEdit edit;
3998
      edit.SetLogNumber(new_log_number);
3999
      impl->logfile_number_ = new_log_number;
4000
      impl->log_.reset(new log::Writer(std::move(lfile)));
4001
      s = impl->versions_->LogAndApply(impl->default_cfd_, &edit, &impl->mutex_,
4002
                                       impl->db_directory_.get());
J
jorlow@chromium.org 已提交
4003 4004
    }
    if (s.ok()) {
4005 4006 4007
      // set column family handles
      handles->clear();
      for (auto cf : column_families) {
I
Igor Canadi 已提交
4008 4009 4010 4011 4012 4013 4014
        if (!impl->versions_->GetColumnFamilySet()->Exists(cf.name)) {
          s = Status::InvalidArgument("Column family not found: ", cf.name);
          handles->clear();
          break;
        }
        uint32_t id = impl->versions_->GetColumnFamilySet()->GetID(cf.name);
        handles->push_back(ColumnFamilyHandle(id));
4015
      }
I
Igor Canadi 已提交
4016 4017
    }
    if (s.ok()) {
4018 4019 4020 4021
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
        delete impl->InstallSuperVersion(cfd, new SuperVersion());
        cfd->mem->SetLogNumber(impl->logfile_number_);
      }
J
jorlow@chromium.org 已提交
4022
      impl->DeleteObsoleteFiles();
4023
      impl->MaybeScheduleFlushOrCompaction();
4024
      impl->MaybeScheduleLogDBDeployStats();
4025
      s = impl->db_directory_->Fsync();
J
jorlow@chromium.org 已提交
4026 4027
    }
  }
4028

4029
  if (s.ok() && impl->options_.compaction_style == kCompactionStyleUniversal) {
4030
    Version* current = impl->default_cfd_->current;
4031
    for (int i = 1; i < impl->NumberLevels(); i++) {
4032
      int num_files = current->NumLevelFiles(i);
4033 4034 4035 4036 4037 4038 4039 4040
      if (num_files > 0) {
        s = Status::InvalidArgument("Not all files are at level 0. Cannot "
          "open with universal compaction style.");
        break;
      }
    }
  }

4041 4042
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4043 4044 4045
  if (s.ok()) {
    *dbptr = impl;
  } else {
4046
    handles->clear();
J
jorlow@chromium.org 已提交
4047 4048 4049 4050 4051
    delete impl;
  }
  return s;
}

4052 4053 4054
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4055
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4056 4057
}

4058 4059 4060
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4061
Status DestroyDB(const std::string& dbname, const Options& options) {
4062 4063 4064 4065 4066
  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 已提交
4067
  std::vector<std::string> filenames;
4068 4069
  std::vector<std::string> archiveFiles;

4070
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
4071 4072
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
4073 4074 4075 4076 4077 4078 4079

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

J
jorlow@chromium.org 已提交
4081 4082 4083 4084 4085
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
4086 4087
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4088 4089 4090
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
4091
    for (size_t i = 0; i < filenames.size(); i++) {
4092
      if (ParseFileName(filenames[i], &number, &type) &&
4093
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4094 4095 4096
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
4097 4098
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
4099 4100 4101
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
4102 4103 4104 4105 4106
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4107

4108
    env->GetChildren(archivedir, &archiveFiles);
4109 4110
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4111 4112
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4113
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4114 4115 4116 4117 4118
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4119
    // ignore case where no archival directory is present.
4120
    env->DeleteDir(archivedir);
4121

J
jorlow@chromium.org 已提交
4122
    env->UnlockFile(lock);  // Ignore error since state is already gone
4123
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4124
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4125
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4126 4127 4128 4129
  }
  return result;
}

4130 4131
//
// A global method that can dump out the build version
4132
void dumpLeveldbBuildVersion(Logger * log) {
4133
  Log(log, "Git sha %s", rocksdb_build_git_sha);
4134
  Log(log, "Compile time %s %s",
4135
      rocksdb_build_compile_time, rocksdb_build_compile_date);
4136 4137
}

4138
}  // namespace rocksdb