db_impl.cc 142.1 KB
Newer Older
1 2 3 4 5
//  Copyright (c) 2013, Facebook, Inc.  All rights reserved.
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
//
J
jorlow@chromium.org 已提交
6 7 8 9 10 11 12
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file. See the AUTHORS file for names of contributors.

#include "db/db_impl.h"

#include <algorithm>
13 14
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
15
#include <set>
16
#include <stdexcept>
17 18
#include <stdint.h>
#include <string>
19
#include <unordered_set>
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 = versions_->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 1119 1120 1121 1122 1123 1124 1125

  // re-acquire the most current version
  base = versions_->current();

  // There could be multiple threads writing to its own level-0 file.
  // The pending_outputs cannot be cleared here, otherwise this newly
  // created file might not be considered as a live-file by another
  // compaction thread that is concurrently deleting obselete files.
  // The pending_outputs can be cleared only after the new version is
A
Abhishek Kona 已提交
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 1242
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->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 = versions_->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 1326 1327
  if (to_level < level) {
    Log(options_.info_log, "Before refitting:\n%s",
        versions_->current()->DebugString().data());

1328
    VersionEdit edit;
1329 1330
    for (const auto& f : versions_->current()->files_[level]) {
      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 1346 1347 1348 1349 1350 1351

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

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

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
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_ ||
1718
         versions_->current()->NeedsCompaction() ||
1719
         (is_flush_pending && (options_.max_background_flushes <= 0))) &&
1720 1721 1722
        bg_compaction_scheduled_ < options_.max_background_compactions &&
        (!bg_manual_only_ || manual_compaction_)) {

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

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

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

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

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

1790

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

1795 1796 1797 1798 1799
uint64_t DBImpl::TEST_GetLevel0TotalSize() {
  MutexLock l(&mutex_);
  return versions_->current()->NumLevelBytes(0);
}

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

  MaybeDumpStats();

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

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

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

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

1842 1843
  MaybeScheduleLogDBDeployStats();

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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


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

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

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

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

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

2209
  assert(versions_->current()->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
      versions_->current()->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 2653 2654
  versions_->current()->Ref();
  version = versions_->current();
  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();
T
Tomislav Novak 已提交
2692 2693 2694 2695 2696 2697 2698 2699 2700
  version = versions_->current();
  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 versions_->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 3302 3303 3304
      uint64_t slowdown =
          SlowdownAmount(versions_->current()->NumLevelFiles(0),
                         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 (versions_->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 3358 3359
    } else if (allow_hard_rate_limit_delay && options_.hard_rate_limit > 1.0 &&
               (score = versions_->current()->MaxCompactionScore()) >
                   options_.hard_rate_limit) {
3360
      // Delay a write when the compaction score for any level is too large.
3361
      int max_level = versions_->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 3384 3385
    } else if (allow_soft_rate_limit_delay && options_.soft_rate_limit > 0.0 &&
               (score = versions_->current()->MaxCompactionScore()) >
                   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 = versions_->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
             "Level  Files Size(MB) Score Time(sec)  Read(MB) Write(MB)    Rn(MB)  Rnp1(MB)  Wnew(MB) RW-Amplify Read(MB/s) Write(MB/s)      Rn     Rnp1     Wnp1     NewW    Count  Ln-stall Stall-cnt\n"
J
Jim Paton 已提交
3544
             "--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\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
        snprintf(
            buf, sizeof(buf),
3566
            "%3d %8d %8.0f %5.1f %9.0f %9.0f %9.0f %9.0f %9.0f %9.0f %10.1f %9.1f %11.1f %8d %8d %8d %8d %8d %9.1f %9lu\n",
3567 3568
            level,
            files,
3569 3570
            current->NumLevelBytes(level) / 1048576.0,
            current->NumLevelBytes(level) /
3571
                versions_->MaxBytesForLevel(level),
3572
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
3573 3574 3575 3576 3577 3578
            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,
3579 3580
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
3581
            (stats_[level].bytes_written / 1048576.0) /
3582
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
3583 3584 3585 3586
            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,
3587
            stats_[level].count,
J
Jim Paton 已提交
3588 3589
            stall_leveln_slowdown_[level] / 1000000.0,
            (unsigned long) stall_leveln_slowdown_count_[level]);
3590
        total_slowdown += stall_leveln_slowdown_[level];
J
Jim Paton 已提交
3591
        total_slowdown_count += stall_leveln_slowdown_count_[level];
3592 3593 3594
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
3595

3596 3597 3598 3599
    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_;
3600 3601 3602 3603 3604 3605
    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);

3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623
    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 已提交
3624
    snprintf(buf, sizeof(buf),
3625 3626
             "Compaction IO cumulative (GB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
3627
             user_bytes_written / (1048576.0 * 1024),
3628 3629 3630 3631 3632 3633 3634 3635
             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",
3636
             user_bytes_written / 1048576.0 / seconds_up,
3637 3638 3639 3640 3641 3642 3643 3644
             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",
3645 3646 3647 3648 3649 3650 3651 3652 3653 3654 3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676
             (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 已提交
3677 3678
    value->append(buf);

3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691 3692 3693 3694 3695 3696 3697 3698 3699 3700
    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",
3701 3702 3703 3704
             (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 已提交
3705 3706 3707 3708
    value->append(buf);

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
3709
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
3710 3711
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
3712
            stall_memtable_compaction_ / 1000000.0,
3713
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
3714 3715
    value->append(buf);

J
Jim Paton 已提交
3716 3717 3718 3719 3720 3721 3722 3723 3724
    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);

3725 3726 3727
    last_stats_.compaction_bytes_read_ = total_bytes_read;
    last_stats_.compaction_bytes_written_ = total_bytes_written;
    last_stats_.ingest_bytes_ = user_bytes_written;
3728
    last_stats_.seconds_up_ = seconds_up;
3729 3730 3731 3732 3733
    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;
3734

3735
    return true;
G
Gabor Cselle 已提交
3736 3737 3738
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
3739
  } else if (in == "num-immutable-mem-table") {
3740
    *value = std::to_string(default_cfd_->imm.size());
3741
    return true;
J
jorlow@chromium.org 已提交
3742
  }
3743

J
jorlow@chromium.org 已提交
3744 3745 3746
  return false;
}

3747 3748
void DBImpl::GetApproximateSizes(const ColumnFamilyHandle& column_family,
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771
  // TODO(opt): better implementation
  Version* v;
  {
    MutexLock l(&mutex_);
    versions_->current()->Ref();
    v = versions_->current();
  }

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

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

3772 3773 3774 3775 3776 3777 3778
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

3779 3780 3781
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
3782 3783 3784 3785
  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());
3786 3787 3788
    return Status::InvalidArgument("Invalid file name");
  }

3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802
  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;
  }

3803 3804 3805
  int level;
  FileMetaData metadata;
  int maxlevel = NumberLevels();
3806
  VersionEdit edit;
I
Igor Canadi 已提交
3807
  DeletionState deletion_state(0, true);
D
Dhruba Borthakur 已提交
3808 3809 3810 3811
  {
    MutexLock l(&mutex_);
    status = versions_->GetMetadataForFile(number, &level, &metadata);
    if (!status.ok()) {
3812 3813
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
3814 3815 3816
      return Status::InvalidArgument("File not found");
    }
    assert((level > 0) && (level < maxlevel));
3817

D
Dhruba Borthakur 已提交
3818 3819
    // If the file is being compacted no need to delete.
    if (metadata.being_compacted) {
3820
      Log(options_.info_log,
3821
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
3822
      return Status::OK();
3823 3824
    }

D
Dhruba Borthakur 已提交
3825 3826 3827 3828
    // 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++) {
3829
      if (versions_->current()->NumLevelFiles(i) != 0) {
D
Dhruba Borthakur 已提交
3830
        Log(options_.info_log,
3831
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
3832 3833 3834 3835
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
3836 3837
    status = versions_->LogAndApply(default_cfd_, &edit, &mutex_,
                                    db_directory_.get());
I
Igor Canadi 已提交
3838
    if (status.ok()) {
3839
      InstallSuperVersion(default_cfd_, deletion_state);
I
Igor Canadi 已提交
3840
    }
I
Igor Canadi 已提交
3841
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
3842
  } // lock released here
I
Igor Canadi 已提交
3843
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
3844 3845
  // remove files outside the db-lock
  PurgeObsoleteFiles(deletion_state);
3846 3847 3848
  return status;
}

3849
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData> *metadata) {
3850 3851 3852 3853
  MutexLock l(&mutex_);
  return versions_->GetLiveFilesMetaData(metadata);
}

3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880
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 已提交
3881 3882
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
3883 3884
Status DB::Put(const WriteOptions& opt, const ColumnFamilyHandle& column_family,
               const Slice& key, const Slice& value) {
3885 3886 3887 3888
  // 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);
3889
  batch.Put(column_family.id, key, value);
J
jorlow@chromium.org 已提交
3890 3891 3892
  return Write(opt, &batch);
}

3893 3894
Status DB::Delete(const WriteOptions& opt,
                  const ColumnFamilyHandle& column_family, const Slice& key) {
J
jorlow@chromium.org 已提交
3895
  WriteBatch batch;
3896
  batch.Delete(column_family.id, key);
J
jorlow@chromium.org 已提交
3897 3898 3899
  return Write(opt, &batch);
}

3900 3901
Status DB::Merge(const WriteOptions& opt,
                 const ColumnFamilyHandle& column_family, const Slice& key,
3902 3903
                 const Slice& value) {
  WriteBatch batch;
3904
  batch.Merge(column_family.id, key, value);
3905 3906 3907
  return Write(opt, &batch);
}

3908 3909
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
3910
                              const std::string& column_family_name,
3911 3912
                              ColumnFamilyHandle* handle) {
  return Status::NotSupported("");
3913 3914
}
Status DB::DropColumnFamily(const ColumnFamilyHandle& column_family) {
3915
  return Status::NotSupported("");
3916 3917
}

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

J
Jim Paton 已提交
3920
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
3921 3922 3923 3924 3925 3926 3927 3928 3929 3930 3931 3932 3933 3934
  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) {
3935
  *dbptr = nullptr;
H
Haobo Xu 已提交
3936
  EnvOptions soptions;
3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947
  // 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 已提交
3948

3949
  if (options.block_cache != nullptr && options.no_block_cache) {
3950
    return Status::InvalidArgument(
3951
        "no_block_cache is true while block_cache is not nullptr");
3952
  }
3953

J
jorlow@chromium.org 已提交
3954
  DBImpl* impl = new DBImpl(options, dbname);
3955 3956 3957 3958 3959 3960 3961
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
3962 3963 3964 3965
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
3966
  impl->mutex_.Lock();
3967 3968
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
3969
  if (s.ok()) {
3970
    uint64_t new_log_number = impl->versions_->NewFileNumber();
3971
    unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
3972
    soptions.use_mmap_writes = false;
3973
    s = impl->options_.env->NewWritableFile(
3974 3975 3976 3977
      LogFileName(impl->options_.wal_dir, new_log_number),
      &lfile,
      soptions
    );
J
jorlow@chromium.org 已提交
3978
    if (s.ok()) {
3979
      lfile->SetPreallocationBlockSize(1.1 * impl->options_.write_buffer_size);
I
Igor Canadi 已提交
3980
      VersionEdit edit;
3981
      edit.SetLogNumber(new_log_number);
3982
      impl->logfile_number_ = new_log_number;
3983
      impl->log_.reset(new log::Writer(std::move(lfile)));
3984
      s = impl->versions_->LogAndApply(impl->default_cfd_, &edit, &impl->mutex_,
3985
                                       impl->db_directory_.get());
J
jorlow@chromium.org 已提交
3986 3987
    }
    if (s.ok()) {
3988 3989 3990
      // set column family handles
      handles->clear();
      for (auto cf : column_families) {
I
Igor Canadi 已提交
3991 3992 3993 3994 3995 3996 3997
        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));
3998
      }
I
Igor Canadi 已提交
3999 4000
    }
    if (s.ok()) {
4001 4002 4003 4004
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
        delete impl->InstallSuperVersion(cfd, new SuperVersion());
        cfd->mem->SetLogNumber(impl->logfile_number_);
      }
J
jorlow@chromium.org 已提交
4005
      impl->DeleteObsoleteFiles();
4006
      impl->MaybeScheduleFlushOrCompaction();
4007
      impl->MaybeScheduleLogDBDeployStats();
4008
      s = impl->db_directory_->Fsync();
J
jorlow@chromium.org 已提交
4009 4010
    }
  }
4011

4012 4013
  if (s.ok() && impl->options_.compaction_style == kCompactionStyleUniversal) {
    Version* current = impl->versions_->current();
4014
    for (int i = 1; i < impl->NumberLevels(); i++) {
4015
      int num_files = current->NumLevelFiles(i);
4016 4017 4018 4019 4020 4021 4022 4023
      if (num_files > 0) {
        s = Status::InvalidArgument("Not all files are at level 0. Cannot "
          "open with universal compaction style.");
        break;
      }
    }
  }

4024 4025
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4026 4027 4028
  if (s.ok()) {
    *dbptr = impl;
  } else {
4029
    handles->clear();
J
jorlow@chromium.org 已提交
4030 4031 4032 4033 4034
    delete impl;
  }
  return s;
}

4035 4036 4037
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4038
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4039 4040
}

4041 4042 4043
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4044
Status DestroyDB(const std::string& dbname, const Options& options) {
4045 4046 4047 4048 4049
  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 已提交
4050
  std::vector<std::string> filenames;
4051 4052
  std::vector<std::string> archiveFiles;

4053
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
4054 4055
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
4056 4057 4058 4059 4060 4061 4062

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

J
jorlow@chromium.org 已提交
4064 4065 4066 4067 4068
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
4069 4070
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4071 4072 4073
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
4074
    for (size_t i = 0; i < filenames.size(); i++) {
4075
      if (ParseFileName(filenames[i], &number, &type) &&
4076
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4077 4078 4079
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
4080 4081
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
4082 4083 4084
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
4085 4086 4087 4088 4089
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4090

4091
    env->GetChildren(archivedir, &archiveFiles);
4092 4093
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4094 4095
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4096
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4097 4098 4099 4100 4101
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4102
    // ignore case where no archival directory is present.
4103
    env->DeleteDir(archivedir);
4104

J
jorlow@chromium.org 已提交
4105
    env->UnlockFile(lock);  // Ignore error since state is already gone
4106
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4107
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4108
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4109 4110 4111 4112
  }
  return result;
}

4113 4114
//
// A global method that can dump out the build version
4115
void dumpLeveldbBuildVersion(Logger * log) {
4116
  Log(log, "Git sha %s", rocksdb_build_git_sha);
4117
  Log(log, "Compile time %s %s",
4118
      rocksdb_build_compile_time, rocksdb_build_compile_date);
4119 4120
}

4121
}  // namespace rocksdb