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

#include "db/db_impl.h"

I
Igor Canadi 已提交
12 13 14
#define __STDC_FORMAT_MACROS

#include <inttypes.h>
J
jorlow@chromium.org 已提交
15
#include <algorithm>
16 17
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
18
#include <set>
19
#include <stdexcept>
20 21
#include <stdint.h>
#include <string>
22
#include <unordered_set>
23
#include <unordered_map>
T
Tomislav Novak 已提交
24
#include <utility>
25
#include <vector>
26

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

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

73
const std::string default_column_family_name("default");
74

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

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

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

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

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

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

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

  uint64_t total_bytes;

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

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
120 121 122 123 124

  // Create a client visible context of this compaction
  CompactionFilter::Context GetFilterContext() {
    CompactionFilter::Context context;
    context.is_full_compaction = compaction->IsFullCompaction();
125
    context.is_manual_compaction = compaction->IsManualCompaction();
126 127
    return context;
  }
J
jorlow@chromium.org 已提交
128 129
};

130
namespace {
J
jorlow@chromium.org 已提交
131
// Fix user-supplied options to be reasonable
132
template <class T, class V>
J
jorlow@chromium.org 已提交
133
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
D
dgrogan@chromium.org 已提交
134 135
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
J
jorlow@chromium.org 已提交
136
}
137 138
}  // anonymous namespace

J
jorlow@chromium.org 已提交
139 140
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
141
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
142
                        const Options& src) {
143 144 145 146 147 148 149
  auto db_options = SanitizeOptions(dbname, DBOptions(src));
  auto cf_options = SanitizeOptions(icmp, ipolicy, ColumnFamilyOptions(src));
  return Options(db_options, cf_options);
}

DBOptions SanitizeOptions(const std::string& dbname, const DBOptions& src) {
  DBOptions result = src;
150 151
  // result.max_open_files means an "infinite" open files.
  if (result.max_open_files != -1) {
152
    ClipToRange(&result.max_open_files, 20, 1000000);
153
  }
154 155 156
  if (result.max_background_flushes == 0) {
    result.max_background_flushes = 1;
  }
157

158
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
159 160
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
161 162
    if (!s.ok()) {
      // No place suitable for logging
163
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
164 165
    }
  }
166 167 168 169 170

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

J
jorlow@chromium.org 已提交
172 173 174
  return result;
}

S
Siying Dong 已提交
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
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;
  }
}

196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217
CompressionType GetCompressionFlush(const Options& options) {
  // Compressing memtable flushes might not help unless the sequential load
  // optimization is used for leveled compaction. Otherwise the CPU and
  // latency overhead is not offset by saving much space.

  bool can_compress;

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

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

I
Igor Canadi 已提交
218
DBImpl::DBImpl(const DBOptions& options, const std::string& dbname)
J
jorlow@chromium.org 已提交
219
    : env_(options.env),
H
heyongqiang 已提交
220
      dbname_(dbname),
I
Igor Canadi 已提交
221
      options_(SanitizeOptions(dbname, options)),
222
      db_lock_(nullptr),
H
Haobo Xu 已提交
223
      mutex_(options.use_adaptive_mutex),
224
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
225
      bg_cv_(&mutex_),
226
      logfile_number_(0),
227
      default_cf_handle_(nullptr),
228
      tmp_batch_(),
229
      bg_schedule_needed_(false),
230
      bg_compaction_scheduled_(0),
231
      bg_manual_only_(0),
232
      bg_flush_scheduled_(0),
233
      bg_logstats_scheduled_(false),
234 235
      manual_compaction_(nullptr),
      logger_(nullptr),
236
      disable_delete_obsolete_files_(0),
I
Igor Canadi 已提交
237
      delete_obsolete_files_last_run_(options.env->NowMicros()),
238
      purge_wal_files_last_run_(0),
239
      last_stats_dump_time_microsec_(0),
240
      default_interval_to_delete_obsolete_WAL_(600),
241
      flush_on_destroy_(false),
242
      delayed_writes_(0),
243 244
      storage_options_(options),
      bg_work_gate_closed_(false),
245 246
      refitting_level_(false),
      opened_successfully_(false) {
H
heyongqiang 已提交
247
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
248

J
jorlow@chromium.org 已提交
249
  // Reserve ten files or so for other uses and give the rest to TableCache.
250 251
  // Give a large number for setting of "infinite" open files.
  const int table_cache_size =
252 253 254 255 256 257
      (options_.max_open_files == -1) ? 4194304 : options_.max_open_files - 10;
  // Reserve ten files or so for other uses and give the rest to TableCache.
  table_cache_ =
      NewLRUCache(table_cache_size, options_.table_cache_numshardbits,
                  options_.table_cache_remove_scan_count_limit);

I
Igor Canadi 已提交
258 259
  versions_.reset(
      new VersionSet(dbname_, &options_, storage_options_, table_cache_.get()));
260 261
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
262

K
kailiu 已提交
263
  DumpLeveldbBuildVersion(options_.info_log.get());
264
  options_.Dump(options_.info_log.get());
265

266
  char name[100];
K
kailiu 已提交
267 268
  Status s = env_->GetHostName(name, 100L);
  if (s.ok()) {
269 270 271 272 273 274
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
275

I
Igor Canadi 已提交
276
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
277 278 279
}

DBImpl::~DBImpl() {
I
Igor Canadi 已提交
280 281 282 283 284 285
  // only the default CFD is alive at this point
  if (default_cf_handle_ != nullptr) {
    auto default_cfd = default_cf_handle_->cfd();
    if (flush_on_destroy_ &&
        default_cfd->mem()->GetFirstSequenceNumber() != 0) {
      FlushMemTable(default_cfd, FlushOptions());
286
    }
287
  }
I
Igor Canadi 已提交
288 289 290

  mutex_.Lock();
  // Wait for background work to finish
291
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
292 293 294
  while (bg_compaction_scheduled_ ||
         bg_flush_scheduled_ ||
         bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
295
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
296
  }
297

I
Igor Canadi 已提交
298 299 300 301 302
  if (default_cf_handle_ != nullptr) {
    // we need to delete handle outside of lock because it does its own locking
    mutex_.Unlock();
    delete default_cf_handle_;
    mutex_.Lock();
303 304
  }

305 306 307 308 309 310 311 312 313 314 315 316 317 318 319
  if (options_.allow_thread_local) {
    // Clean up obsolete files due to SuperVersion release.
    // (1) Need to delete to obsolete files before closing because RepairDB()
    // scans all existing files in the file system and builds manifest file.
    // Keeping obsolete files confuses the repair process.
    // (2) Need to check if we Open()/Recover() the DB successfully before
    // deleting because if VersionSet recover fails (may be due to corrupted
    // manifest file), it is not able to identify live files correctly. As a
    // result, all "live" files can get deleted by accident. However, corrupted
    // manifest is recoverable by RepairDB().
    if (opened_successfully_) {
      DeletionState deletion_state;
      FindObsoleteFiles(deletion_state, true);
      // manifest number starting from 2
      deletion_state.manifest_file_number = 1;
320 321 322
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
323 324 325
    }
  }

326
  // versions need to be destroyed before table_cache since it can hold
327 328
  // references to table_cache.
  versions_.reset();
329
  mutex_.Unlock();
I
Igor Canadi 已提交
330 331 332
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
333

I
Igor Canadi 已提交
334
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
335 336
}

A
Abhishek Kona 已提交
337 338 339
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
340

J
jorlow@chromium.org 已提交
341
Status DBImpl::NewDB() {
342
  VersionEdit new_db;
343
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
344 345 346 347
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
348
  unique_ptr<WritableFile> file;
I
Igor Canadi 已提交
349 350
  Status s = env_->NewWritableFile(manifest, &file,
                                   storage_options_.AdaptForLogWrite());
J
jorlow@chromium.org 已提交
351 352 353
  if (!s.ok()) {
    return s;
  }
354
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
355
  {
356
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373
    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 {
374
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
375 376 377 378
    *s = Status::OK();
  }
}

379
const Status DBImpl::CreateArchivalDirectory() {
380
  if (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0) {
381
    std::string archivalPath = ArchivalDirectory(options_.wal_dir);
382 383 384 385 386
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

387
void DBImpl::PrintStatistics() {
388
  auto dbstats = options_.statistics.get();
389 390
  if (dbstats) {
    Log(options_.info_log,
391 392
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
393 394 395
  }
}

396
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
397 398 399 400 401 402 403 404 405 406 407 408 409
  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;
410
    GetProperty("rocksdb.stats", &stats);
H
Haobo Xu 已提交
411
    Log(options_.info_log, "%s", stats.c_str());
412
    PrintStatistics();
413 414 415
  }
}

416
// Returns the list of live files in 'sst_live' and the list
K
kailiu 已提交
417
// of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
418 419 420 421 422 423 424
// 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 已提交
425 426
  mutex_.AssertHeld();

427
  // if deletion is disabled, do nothing
428
  if (disable_delete_obsolete_files_ > 0) {
429 430 431
    return;
  }

432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
  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 已提交
448 449 450
  // get obsolete files
  versions_->GetObsoleteFiles(&deletion_state.sst_delete_files);

I
Igor Canadi 已提交
451 452
  // store the current filenum, lognum, etc
  deletion_state.manifest_file_number = versions_->ManifestFileNumber();
453
  deletion_state.log_number = versions_->MinLogNumber();
I
Igor Canadi 已提交
454 455
  deletion_state.prev_log_number = versions_->PrevLogNumber();

456 457 458 459 460 461 462 463
  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 已提交
464 465 466 467
  deletion_state.sst_live.assign(pending_outputs_.begin(),
                                 pending_outputs_.end());
  versions_->AddLiveFiles(&deletion_state.sst_live);

468
  if (doing_the_full_scan) {
K
kailiu 已提交
469 470 471 472 473
    // set of all files in the directory. We'll exclude files that are still
    // alive in the subsequent processings.
    env_->GetChildren(
        dbname_, &deletion_state.candidate_files
    ); // Ignore errors
474 475 476 477 478

    //Add log files in wal_dir
    if (options_.wal_dir != dbname_) {
      std::vector<std::string> log_files;
      env_->GetChildren(options_.wal_dir, &log_files); // Ignore errors
K
kailiu 已提交
479 480
      deletion_state.candidate_files.insert(
        deletion_state.candidate_files.end(),
481 482 483
        log_files.begin(),
        log_files.end()
      );
484
    }
485
  }
486 487
}

D
Dhruba Borthakur 已提交
488
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
489
// belong to live files are posibly removed. Also, removes all the
490
// files in sst_delete_files and log_delete_files.
491
// It is not necessary to hold the mutex when invoking this method.
D
Dhruba Borthakur 已提交
492
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
493 494
  // we'd better have sth to delete
  assert(state.HaveSomethingToDelete());
495

I
Igor Canadi 已提交
496 497 498 499
  // 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 已提交
500 501
    return;
  }
502

503

504 505
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
K
kailiu 已提交
506 507 508
  std::unordered_set<uint64_t> sst_live(
      state.sst_live.begin(), state.sst_live.end()
  );
I
Igor Canadi 已提交
509

K
kailiu 已提交
510 511 512 513 514 515 516
  auto& candidate_files = state.candidate_files;
  candidate_files.reserve(
      candidate_files.size() +
      state.sst_delete_files.size() +
      state.log_delete_files.size());
  // We may ignore the dbname when generating the file names.
  const char* kDumbDbName = "";
517
  for (auto file : state.sst_delete_files) {
K
kailiu 已提交
518 519 520
    candidate_files.push_back(
        TableFileName(kDumbDbName, file->number).substr(1)
    );
521
    delete file;
I
Igor Canadi 已提交
522 523
  }

K
kailiu 已提交
524 525 526 527 528
  for (auto file_num : state.log_delete_files) {
    if (file_num > 0) {
      candidate_files.push_back(
          LogFileName(kDumbDbName, file_num).substr(1)
      );
I
Igor Canadi 已提交
529 530
    }
  }
531

K
kailiu 已提交
532
  // dedup state.candidate_files so we don't try to delete the same
I
Igor Canadi 已提交
533
  // file twice
K
kailiu 已提交
534 535 536 537 538
  sort(candidate_files.begin(), candidate_files.end());
  candidate_files.erase(
      unique(candidate_files.begin(), candidate_files.end()),
      candidate_files.end()
  );
J
jorlow@chromium.org 已提交
539

540 541
  std::vector<std::string> old_info_log_files;

K
kailiu 已提交
542 543 544 545 546 547 548
  for (const auto& to_delete : candidate_files) {
    uint64_t number;
    FileType type;
    // Ignore file if we cannot recognize it.
    if (!ParseFileName(to_delete, &number, &type)) {
      continue;
    }
J
jorlow@chromium.org 已提交
549

K
kailiu 已提交
550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571
    bool keep = true;
    switch (type) {
      case kLogFile:
        keep = ((number >= state.log_number) ||
                (number == state.prev_log_number));
        break;
      case kDescriptorFile:
        // Keep my manifest file, and any newer incarnations'
        // (in case there is a race that allows other incarnations)
        keep = (number >= state.manifest_file_number);
        break;
      case kTableFile:
        keep = (sst_live.find(number) != sst_live.end());
        break;
      case kTempFile:
        // Any temp files that are currently being written to must
        // be recorded in pending_outputs_, which is inserted into "live"
        keep = (sst_live.find(number) != sst_live.end());
        break;
      case kInfoLogFile:
        keep = true;
        if (number != 0) {
572
          old_info_log_files.push_back(to_delete);
J
jorlow@chromium.org 已提交
573
        }
K
kailiu 已提交
574 575 576 577 578 579 580 581 582 583 584 585 586 587 588
        break;
      case kCurrentFile:
      case kDBLockFile:
      case kIdentityFile:
      case kMetaDatabase:
        keep = true;
        break;
    }

    if (keep) {
      continue;
    }

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

K
kailiu 已提交
592 593 594 595
    std::string fname = ((type == kLogFile) ? options_.wal_dir : dbname_) +
        "/" + to_delete;
    if (type == kLogFile &&
        (options_.WAL_ttl_seconds > 0 || options_.WAL_size_limit_MB > 0)) {
596 597 598 599 600
      auto archived_log_name = ArchivedLogFileName(options_.wal_dir, number);
      Status s = env_->RenameFile(fname, archived_log_name);
      Log(options_.info_log,
          "Move log file %s to %s -- %s\n",
          fname.c_str(), archived_log_name.c_str(), s.ToString().c_str());
K
kailiu 已提交
601 602
    } else {
      Status s = env_->DeleteFile(fname);
603 604 605
      Log(options_.info_log, "Delete %s type=%d #%lu -- %s\n",
          fname.c_str(), type, (unsigned long)number,
          s.ToString().c_str());
J
jorlow@chromium.org 已提交
606 607
    }
  }
H
heyongqiang 已提交
608

609
  // Delete old info log files.
610
  size_t old_info_log_file_count = old_info_log_files.size();
K
Kai Liu 已提交
611 612
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
613
  if (old_info_log_file_count >= options_.keep_log_file_num &&
K
Kai Liu 已提交
614
      options_.db_log_dir.empty()) {
615 616
    std::sort(old_info_log_files.begin(), old_info_log_files.end());
    size_t end = old_info_log_file_count - options_.keep_log_file_num;
617
    for (unsigned int i = 0; i <= end; i++) {
618 619 620 621 622 623 624
      std::string& to_delete = old_info_log_files.at(i);
      Log(options_.info_log, "Delete info log file %s\n", to_delete.c_str());
      Status s = env_->DeleteFile(dbname_ + "/" + to_delete);
      if (!s.ok()) {
        Log(options_.info_log, "Delete info log file %s FAILED -- %s\n",
            to_delete.c_str(), s.ToString().c_str());
      }
H
heyongqiang 已提交
625 626
    }
  }
627
  PurgeObsoleteWALFiles();
I
Igor Canadi 已提交
628
  LogFlush(options_.info_log);
D
Dhruba Borthakur 已提交
629 630 631 632 633
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
I
Igor Canadi 已提交
634
  FindObsoleteFiles(deletion_state, true);
635 636 637
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
638 639
}

640 641 642 643 644 645 646 647
// 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
648
void DBImpl::PurgeObsoleteWALFiles() {
649 650 651 652 653 654
  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;
  }

655 656
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
657 658 659 660 661 662 663 664
  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_;
665

666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705
  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;
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 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759
      }

      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;
760 761
    }
  }
D
Dhruba Borthakur 已提交
762 763
}

764
Status DBImpl::Recover(
765 766
    const std::vector<ColumnFamilyDescriptor>& column_families, bool read_only,
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
767 768
  mutex_.AssertHeld();

769
  assert(db_lock_ == nullptr);
I
Igor Canadi 已提交
770
  if (!read_only) {
771 772 773 774 775 776 777 778 779 780 781 782
    // 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;
    }

783 784 785 786 787
    s = env_->NewDirectory(dbname_, &db_directory_);
    if (!s.ok()) {
      return s;
    }

788
    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
789 790 791
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
792

793 794
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
795
        // TODO: add merge_operator name check
796 797 798 799 800 801 802
        s = NewDB();
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
803 804
      }
    } else {
805 806 807 808
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
809
    }
M
Mayank Agarwal 已提交
810 811 812 813 814 815 816
    // 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 已提交
817 818
  }

I
Igor Canadi 已提交
819
  Status s = versions_->Recover(column_families);
J
jorlow@chromium.org 已提交
820 821
  if (s.ok()) {
    SequenceNumber max_sequence(0);
822 823
    default_cf_handle_ = new ColumnFamilyHandleImpl(
        versions_->GetColumnFamilySet()->GetDefault(), this, &mutex_);
824 825 826 827 828 829 830

    // 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
831
    // produced by an older version of rocksdb.
832
    const uint64_t min_log = versions_->MinLogNumber();
833 834
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
835
    s = env_->GetChildren(options_.wal_dir, &filenames);
836 837
    if (!s.ok()) {
      return s;
838
    }
K
kailiu 已提交
839

840 841
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
K
kailiu 已提交
842 843
      uint64_t number;
      FileType type;
844 845 846 847 848
      if (ParseFileName(filenames[i], &number, &type)
          && type == kLogFile
          && ((number >= min_log) || (number == prev_log))) {
        logs.push_back(number);
      }
J
jorlow@chromium.org 已提交
849
    }
850

H
heyongqiang 已提交
851 852 853 854 855 856
    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");
    }

857 858
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
K
kailiu 已提交
859
    for (const auto& log : logs) {
860 861 862
      // The previous incarnation may not have written any MANIFEST
      // records after allocating this log number.  So we manually
      // update the file number allocation counter in VersionSet.
K
kailiu 已提交
863
      versions_->MarkFileNumberUsed(log);
K
Kai Liu 已提交
864
      s = RecoverLogFile(log, &max_sequence, read_only);
865
    }
I
Igor Canadi 已提交
866 867
    SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                   versions_->LastSequence());
J
jorlow@chromium.org 已提交
868 869 870 871 872
  }

  return s;
}

I
Igor Canadi 已提交
873 874
Status DBImpl::RecoverLogFile(uint64_t log_number, SequenceNumber* max_sequence,
                              bool read_only) {
J
jorlow@chromium.org 已提交
875 876
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
877
    Logger* info_log;
J
jorlow@chromium.org 已提交
878
    const char* fname;
879 880
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
881
    virtual void Corruption(size_t bytes, const Status& s) {
882
      Log(info_log, "%s%s: dropping %d bytes; %s",
883
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
884
          fname, static_cast<int>(bytes), s.ToString().c_str());
885
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
886 887 888 889 890
    }
  };

  mutex_.AssertHeld();

891
  std::unordered_map<int, VersionEdit> version_edits;
892
  // no need to refcount because iteration is under mutex
893 894
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    VersionEdit edit;
895 896
    edit.SetColumnFamily(cfd->GetID());
    version_edits.insert({cfd->GetID(), edit});
897
  }
I
Igor Canadi 已提交
898

J
jorlow@chromium.org 已提交
899
  // Open the log file
900
  std::string fname = LogFileName(options_.wal_dir, log_number);
901
  unique_ptr<SequentialFile> file;
902
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
903 904 905 906 907 908 909 910
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
911
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
912
  reporter.fname = fname.c_str();
913 914
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
915 916 917 918
  // 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).
919
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
920
                     0/*initial_offset*/);
K
Kai Liu 已提交
921 922
  Log(options_.info_log, "Recovering log #%lu",
      (unsigned long) log_number);
J
jorlow@chromium.org 已提交
923 924 925 926 927

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
I
Igor Canadi 已提交
928
  while (reader.ReadRecord(&record, &scratch)) {
J
jorlow@chromium.org 已提交
929 930 931 932 933 934 935
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

936
    status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
937
        &batch, column_family_memtables_.get(), true, log_number);
938

J
jorlow@chromium.org 已提交
939 940
    MaybeIgnoreError(&status);
    if (!status.ok()) {
I
Igor Canadi 已提交
941
      return status;
J
jorlow@chromium.org 已提交
942 943 944 945 946 947 948 949
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

950
    if (!read_only) {
951 952
      // no need to refcount since client still doesn't have access
      // to the DB and can not drop column families while we iterate
953
      for (auto cfd : *versions_->GetColumnFamilySet()) {
954 955
        if (cfd->mem()->ApproximateMemoryUsage() >
            cfd->options()->write_buffer_size) {
956
          // If this asserts, it means that InsertInto failed in
957 958
          // filtering updates to already-flushed column families
          assert(cfd->GetLogNumber() <= log_number);
959
          auto iter = version_edits.find(cfd->GetID());
960 961
          assert(iter != version_edits.end());
          VersionEdit* edit = &iter->second;
962
          status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
963 964 965 966 967 968 969 970
          // we still want to clear the memtable, even if the recovery failed
          cfd->CreateNewMemtable();
          if (!status.ok()) {
            // Reflect errors immediately so that conditions like full
            // file-systems cause the DB::Open() to fail.
            return status;
          }
        }
J
jorlow@chromium.org 已提交
971 972 973 974
      }
    }
  }

975 976 977 978
  if (versions_->LastSequence() < *max_sequence) {
    versions_->SetLastSequence(*max_sequence);
  }

979
  if (!read_only) {
980 981
    // no need to refcount since client still doesn't have access
    // to the DB and can not drop column families while we iterate
982
    for (auto cfd : *versions_->GetColumnFamilySet()) {
983
      auto iter = version_edits.find(cfd->GetID());
984 985 986
      assert(iter != version_edits.end());
      VersionEdit* edit = &iter->second;

987 988 989 990
      if (cfd->GetLogNumber() > log_number) {
        // Column family cfd has already flushed the data
        // from log_number. Memtable has to be empty because
        // we filter the updates based on log_number
991
        // (in WriteBatch::InsertInto)
992 993 994 995 996 997 998
        assert(cfd->mem()->GetFirstSequenceNumber() == 0);
        assert(edit->NumEntries() == 0);
        continue;
      }

      // flush the final memtable (if non-empty)
      if (cfd->mem()->GetFirstSequenceNumber() != 0) {
999
        status = WriteLevel0TableForRecovery(cfd, cfd->mem(), edit);
1000
      }
1001 1002 1003 1004 1005
      // we still want to clear the memtable, even if the recovery failed
      cfd->CreateNewMemtable();
      if (!status.ok()) {
        return status;
      }
J
jorlow@chromium.org 已提交
1006

1007 1008 1009 1010 1011 1012 1013
      // write MANIFEST with update
      // writing log number in the manifest means that any log file
      // with number strongly less than (log_number + 1) is already
      // recovered and should be ignored on next reincarnation.
      // Since we already recovered log_number, we want all logs
      // with numbers `<= log_number` (includes this one) to be ignored
      edit->SetLogNumber(log_number + 1);
1014 1015 1016
      // we must mark the next log number as used, even though it's
      // not actually used. that is because VersionSet assumes
      // VersionSet::next_file_number_ always to be strictly greater than any
I
Igor Canadi 已提交
1017
      // log number
1018
      versions_->MarkFileNumberUsed(log_number + 1);
1019 1020 1021 1022 1023
      status = versions_->LogAndApply(cfd, edit, &mutex_);
      if (!status.ok()) {
        return status;
      }
    }
1024
  }
I
Igor Canadi 已提交
1025

J
jorlow@chromium.org 已提交
1026 1027 1028
  return status;
}

1029 1030
Status DBImpl::WriteLevel0TableForRecovery(ColumnFamilyData* cfd, MemTable* mem,
                                           VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1031
  mutex_.AssertHeld();
1032
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1033 1034 1035 1036
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
1037 1038 1039
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
K
Kai Liu 已提交
1040 1041
  Log(options_.info_log, "Level-0 table #%lu: started",
      (unsigned long) meta.number);
1042 1043 1044 1045

  Status s;
  {
    mutex_.Unlock();
I
Igor Canadi 已提交
1046
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1047
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1048
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1049
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1050
    LogFlush(options_.info_log);
1051 1052 1053
    mutex_.Lock();
  }

K
Kai Liu 已提交
1054 1055 1056
  Log(options_.info_log, "Level-0 table #%lu: %lu bytes %s",
      (unsigned long) meta.number,
      (unsigned long) meta.file_size,
J
jorlow@chromium.org 已提交
1057 1058
      s.ToString().c_str());
  delete iter;
1059

1060
  pending_outputs_.erase(meta.number);
1061 1062 1063 1064 1065 1066

  // 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,
1067 1068
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1069 1070
  }

I
Igor Canadi 已提交
1071
  InternalStats::CompactionStats stats;
1072 1073
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
1074
  stats.files_out_levelnp1 = 1;
1075
  cfd->internal_stats()->AddCompactionStats(level, stats);
1076
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
J
jorlow@chromium.org 已提交
1077 1078 1079
  return s;
}

1080
Status DBImpl::WriteLevel0Table(ColumnFamilyData* cfd,
1081
                                autovector<MemTable*>& mems, VersionEdit* edit,
1082
                                uint64_t* filenumber, LogBuffer* log_buffer) {
J
jorlow@chromium.org 已提交
1083
  mutex_.AssertHeld();
1084 1085 1086 1087 1088
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
1089

1090 1091
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
1092
    mems[0]->GetFirstSequenceNumber();
1093
  Version* base = cfd->current();
1094
  base->Ref();          // it is likely that we do not need this reference
1095 1096 1097
  Status s;
  {
    mutex_.Unlock();
H
Haobo Xu 已提交
1098
    log_buffer->FlushBufferToLog();
K
Kai Liu 已提交
1099
    std::vector<Iterator*> memtables;
1100
    for (MemTable* m : mems) {
1101 1102
      Log(options_.info_log,
          "[CF %u] Flushing memtable with next log file: %lu\n", cfd->GetID(),
1103
          (unsigned long)m->GetNextLogNumber());
K
Kai Liu 已提交
1104
      memtables.push_back(m->NewIterator());
1105
    }
1106 1107 1108
    Iterator* iter = NewMergingIterator(env_, &cfd->internal_comparator(),
                                        &memtables[0], memtables.size());
    Log(options_.info_log, "Level-0 flush table #%lu: started",
1109
        (unsigned long)meta.number);
1110

I
Igor Canadi 已提交
1111
    s = BuildTable(dbname_, env_, *cfd->options(), storage_options_,
1112
                   cfd->table_cache(), iter, &meta, cfd->internal_comparator(),
1113
                   newest_snapshot, earliest_seqno_in_memtable,
I
Igor Canadi 已提交
1114
                   GetCompressionFlush(*cfd->options()));
I
Igor Canadi 已提交
1115
    LogFlush(options_.info_log);
1116 1117 1118 1119 1120
    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());
1121 1122 1123
    if (!options_.disableDataSync) {
      db_directory_->Fsync();
    }
1124 1125
    mutex_.Lock();
  }
1126 1127
  base->Unref();

1128
  // re-acquire the most current version
1129
  base = cfd->current();
1130 1131 1132 1133 1134 1135

  // 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 已提交
1136
  // committed so that other threads can recognize this file as a
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149
  // 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.
1150
    if (base != nullptr && options_.max_background_compactions <= 1 &&
1151
        cfd->options()->compaction_style == kCompactionStyleLevel) {
1152 1153 1154
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
1155 1156
                  meta.smallest, meta.largest,
                  meta.smallest_seqno, meta.largest_seqno);
1157 1158
  }

I
Igor Canadi 已提交
1159
  InternalStats::CompactionStats stats;
1160 1161
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
1162
  cfd->internal_stats()->AddCompactionStats(level, stats);
1163
  RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES, meta.file_size);
1164 1165 1166
  return s;
}

1167 1168
Status DBImpl::FlushMemTableToOutputFile(ColumnFamilyData* cfd,
                                         bool* madeProgress,
H
Haobo Xu 已提交
1169 1170
                                         DeletionState& deletion_state,
                                         LogBuffer* log_buffer) {
1171
  mutex_.AssertHeld();
1172
  assert(cfd->imm()->size() != 0);
1173
  assert(cfd->imm()->IsFlushPending());
1174 1175 1176

  // Save the contents of the earliest memtable as a new Table
  uint64_t file_number;
K
Kai Liu 已提交
1177
  autovector<MemTable*> mems;
1178
  cfd->imm()->PickMemtablesToFlush(&mems);
1179
  if (mems.empty()) {
H
Haobo Xu 已提交
1180
    LogToBuffer(log_buffer, "Nothing in memstore to flush");
L
Lei Jin 已提交
1181
    return Status::OK();
1182 1183 1184
  }

  // record the logfile_number_ before we release the mutex
1185 1186 1187
  // 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.
1188
  MemTable* m = mems[0];
1189 1190
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
1191 1192
  // SetLogNumber(log_num) indicates logs with number smaller than log_num
  // will no longer be picked up for recovery.
1193 1194
  edit->SetLogNumber(mems.back()->GetNextLogNumber());
  edit->SetColumnFamily(cfd->GetID());
1195

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

1199
  if (s.ok() && shutting_down_.Acquire_Load() && cfd->IsDropped()) {
L
Lei Jin 已提交
1200
    s = Status::ShutdownInProgress(
1201
        "Column family closed during memtable flush");
1202
  }
J
jorlow@chromium.org 已提交
1203

L
Lei Jin 已提交
1204
  if (!s.ok()) {
1205
    cfd->imm()->RollbackMemtableFlush(mems, file_number, &pending_outputs_);
L
Lei Jin 已提交
1206
    return s;
1207
  }
J
jorlow@chromium.org 已提交
1208

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

  if (s.ok()) {
1215
    InstallSuperVersion(cfd, deletion_state);
1216 1217 1218
    if (madeProgress) {
      *madeProgress = 1;
    }
1219

1220
    MaybeScheduleLogDBDeployStats();
I
Igor Canadi 已提交
1221

1222
    if (disable_delete_obsolete_files_ == 0) {
I
Igor Canadi 已提交
1223
      // add to deletion state
1224 1225 1226 1227 1228
      while (alive_log_files_.size() &&
             *alive_log_files_.begin() < versions_->MinLogNumber()) {
        deletion_state.log_delete_files.push_back(*alive_log_files_.begin());
        alive_log_files_.pop_front();
      }
1229
    }
J
jorlow@chromium.org 已提交
1230 1231 1232 1233
  }
  return s;
}

1234
Status DBImpl::CompactRange(ColumnFamilyHandle* column_family,
1235 1236
                            const Slice* begin, const Slice* end,
                            bool reduce_level, int target_level) {
1237 1238
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
I
Igor Canadi 已提交
1239 1240

  Status s = FlushMemTable(cfd, FlushOptions());
L
Lei Jin 已提交
1241 1242 1243 1244 1245
  if (!s.ok()) {
    LogFlush(options_.info_log);
    return s;
  }

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

  if (reduce_level) {
I
Igor Canadi 已提交
1272
    s = ReFitLevel(cfd, max_level_with_files, target_level);
1273
  }
I
Igor Canadi 已提交
1274
  LogFlush(options_.info_log);
L
Lei Jin 已提交
1275 1276

  return s;
1277 1278 1279
}

// return the same level if it cannot be moved
I
Igor Canadi 已提交
1280
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd, int level) {
1281
  mutex_.AssertHeld();
I
Igor Canadi 已提交
1282
  Version* current = cfd->current();
1283
  int minimum_level = level;
1284
  for (int i = level - 1; i > 0; --i) {
1285
    // stop if level i is not empty
1286
    if (current->NumLevelFiles(i) > 0) break;
1287
    // stop if level i is too small (cannot fit the level files)
I
Igor Canadi 已提交
1288
    if (cfd->compaction_picker()->MaxBytesForLevel(i) <
1289 1290 1291
        current->NumLevelBytes(level)) {
      break;
    }
1292 1293 1294 1295 1296 1297

    minimum_level = i;
  }
  return minimum_level;
}

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

I
Igor Canadi 已提交
1301
  SuperVersion* superversion_to_free = nullptr;
1302
  SuperVersion* new_superversion = new SuperVersion();
I
Igor Canadi 已提交
1303 1304

  mutex_.Lock();
1305 1306 1307

  // only allow one thread refitting
  if (refitting_level_) {
I
Igor Canadi 已提交
1308
    mutex_.Unlock();
1309
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
I
Igor Canadi 已提交
1310
    delete new_superversion;
L
Lei Jin 已提交
1311
    return Status::NotSupported("another thread is refitting");
1312 1313 1314 1315 1316
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
1317
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_) {
1318
    Log(options_.info_log,
1319 1320
        "RefitLevel: waiting for background threads to stop: %d %d",
        bg_compaction_scheduled_, bg_flush_scheduled_);
1321 1322 1323 1324
    bg_cv_.Wait();
  }

  // move to a smaller level
1325 1326
  int to_level = target_level;
  if (target_level < 0) {
I
Igor Canadi 已提交
1327
    to_level = FindMinimumEmptyLevelFitting(cfd, level);
1328
  }
1329 1330 1331

  assert(to_level <= level);

L
Lei Jin 已提交
1332
  Status status;
1333 1334
  if (to_level < level) {
    Log(options_.info_log, "Before refitting:\n%s",
I
Igor Canadi 已提交
1335
        cfd->current()->DebugString().data());
1336

1337
    VersionEdit edit;
I
Igor Canadi 已提交
1338 1339
    edit.SetColumnFamily(cfd->GetID());
    for (const auto& f : cfd->current()->files_[level]) {
1340
      edit.DeleteFile(level, f->number);
1341 1342
      edit.AddFile(to_level, f->number, f->file_size, f->smallest, f->largest,
                   f->smallest_seqno, f->largest_seqno);
1343 1344 1345 1346
    }
    Log(options_.info_log, "Apply version edit:\n%s",
        edit.DebugString().data());

I
Igor Canadi 已提交
1347
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
1348
    superversion_to_free = cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
1349
    new_superversion = nullptr;
1350 1351 1352 1353 1354

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

    if (status.ok()) {
      Log(options_.info_log, "After refitting:\n%s",
I
Igor Canadi 已提交
1355
          cfd->current()->DebugString().data());
1356 1357 1358 1359 1360
    }
  }

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
I
Igor Canadi 已提交
1361 1362 1363 1364

  mutex_.Unlock();
  delete superversion_to_free;
  delete new_superversion;
L
Lei Jin 已提交
1365
  return status;
G
Gabor Cselle 已提交
1366 1367
}

1368 1369 1370
int DBImpl::NumberLevels(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->NumberLevels();
1371 1372
}

1373 1374 1375
int DBImpl::MaxMemCompactionLevel(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->max_mem_compaction_level;
1376 1377
}

1378 1379 1380
int DBImpl::Level0StopWriteTrigger(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->options()->level0_stop_writes_trigger;
1381 1382
}

1383
Status DBImpl::Flush(const FlushOptions& options,
1384 1385 1386
                     ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return FlushMemTable(cfh->cfd(), options);
H
heyongqiang 已提交
1387 1388
}

1389
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
1390 1391 1392
  return versions_->LastSequence();
}

1393 1394 1395
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {
1396

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

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

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

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

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

1454 1455
Status DBImpl::ReadFirstRecord(const WalFileType type, const uint64_t number,
                               WriteBatch* const result) {
1456

1457
  if (type == kAliveLogFile) {
1458
    std::string fname = LogFileName(options_.wal_dir, number);
1459
    Status status = ReadFirstLine(fname, result);
L
Lei Jin 已提交
1460 1461 1462
    if (status.ok() || env_->FileExists(fname)) {
      // return OK or any error that is not caused non-existing file
      return status;
1463
    }
L
Lei Jin 已提交
1464 1465 1466 1467 1468 1469 1470 1471 1472

    //  check if the file got moved to archive.
    std::string archived_file =
      ArchivedLogFileName(options_.wal_dir, number);
    Status s = ReadFirstLine(archived_file, result);
    if (s.ok() || env_->FileExists(archived_file)) {
      return s;
    }
    return Status::NotFound("Log File has been deleted: " + archived_file);
1473
  } else if (type == kArchivedLogFile) {
1474
    std::string fname = ArchivedLogFileName(options_.wal_dir, number);
1475 1476 1477
    Status status = ReadFirstLine(fname, result);
    return status;
  }
1478
  return Status::NotSupported("File Type Not Known: " + std::to_string(type));
1479 1480 1481 1482 1483 1484 1485 1486
}

Status DBImpl::ReadFirstLine(const std::string& fname,
                             WriteBatch* const batch) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;
L
Lei Jin 已提交
1487 1488 1489

    Status* status;
    bool ignore_error;  // true if options_.paranoid_checks==false
1490 1491
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
L
Lei Jin 已提交
1492
          (this->ignore_error ? "(ignoring error) " : ""),
1493
          fname, static_cast<int>(bytes), s.ToString().c_str());
L
Lei Jin 已提交
1494 1495 1496 1497
      if (this->status->ok()) {
        // only keep the first error
        *this->status = s;
      }
1498 1499 1500
    }
  };

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

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


  LogReporter reporter;
  reporter.env = env_;
1511
  reporter.info_log = options_.info_log.get();
1512
  reporter.fname = fname.c_str();
L
Lei Jin 已提交
1513 1514
  reporter.status = &status;
  reporter.ignore_error = !options_.paranoid_checks;
1515
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1516 1517 1518
                     0/*initial_offset*/);
  std::string scratch;
  Slice record;
1519

L
Lei Jin 已提交
1520 1521
  if (reader.ReadRecord(&record, &scratch) &&
      (status.ok() || !options_.paranoid_checks)) {
1522 1523 1524 1525
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      //  TODO read record's till the first no corrupt entry?
L
Lei Jin 已提交
1526 1527 1528
    } else {
      WriteBatchInternal::SetContents(batch, record);
      return Status::OK();
1529 1530
    }
  }
L
Lei Jin 已提交
1531 1532 1533 1534 1535 1536

  // ReadRecord returns false on EOF, which is deemed as OK() by Reader
  if (status.ok()) {
    status = Status::Corruption("eof reached");
  }
  return status;
1537 1538
}

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

      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))));
1584 1585
    }
  }
1586
  CompareLogByPointer compare_log_files;
1587
  std::sort(pos_start, log_files.end(), compare_log_files);
1588 1589 1590
  return status;
}

I
Igor Canadi 已提交
1591 1592
Status DBImpl::RunManualCompaction(ColumnFamilyData* cfd, int input_level,
                                   int output_level, const Slice* begin,
L
Lei Jin 已提交
1593
                                   const Slice* end) {
1594
  assert(input_level >= 0);
1595

G
Gabor Cselle 已提交
1596 1597
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1598
  ManualCompaction manual;
I
Igor Canadi 已提交
1599
  manual.cfd = cfd;
1600 1601
  manual.input_level = input_level;
  manual.output_level = output_level;
G
Gabor Cselle 已提交
1602
  manual.done = false;
1603
  manual.in_progress = false;
1604 1605 1606
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
I
Igor Canadi 已提交
1607
      cfd->options()->compaction_style == kCompactionStyleUniversal) {
1608
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1609 1610 1611 1612
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1613
  if (end == nullptr ||
I
Igor Canadi 已提交
1614
      cfd->options()->compaction_style == kCompactionStyleUniversal) {
1615
    manual.end = nullptr;
G
Gabor Cselle 已提交
1616 1617 1618 1619 1620 1621
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1622

1623 1624 1625 1626 1627 1628
  // 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
1629
  // RunManualCompaction(), i.e. during that time no other compaction will
1630 1631 1632
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
1633
  // RunManualCompaction() from getting to the second while loop below.
1634 1635 1636 1637 1638 1639 1640 1641
  // 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");
1642 1643
    bg_cv_.Wait();
  }
1644

1645 1646
  Log(options_.info_log, "Manual compaction starting");

1647 1648 1649 1650
  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 已提交
1651
      bg_cv_.Wait();
1652 1653 1654
    } else {
      manual_compaction_ = &manual;
      MaybeScheduleFlushOrCompaction();
G
Gabor Cselle 已提交
1655
    }
H
hans@chromium.org 已提交
1656
  }
1657

1658 1659 1660
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
L
Lei Jin 已提交
1661
  return manual.status;
J
jorlow@chromium.org 已提交
1662 1663
}

1664 1665 1666 1667 1668 1669 1670 1671 1672 1673
Status DBImpl::TEST_CompactRange(int level, const Slice* begin,
                                 const Slice* end,
                                 ColumnFamilyHandle* column_family) {
  ColumnFamilyData* cfd;
  if (column_family == nullptr) {
    cfd = default_cf_handle_->cfd();
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    cfd = cfh->cfd();
  }
I
Igor Canadi 已提交
1674
  int output_level =
1675
      (cfd->options()->compaction_style == kCompactionStyleUniversal)
I
Igor Canadi 已提交
1676 1677
          ? level
          : level + 1;
1678
  return RunManualCompaction(cfd, level, output_level, begin, end);
1679 1680
}

1681 1682
Status DBImpl::FlushMemTable(ColumnFamilyData* cfd,
                             const FlushOptions& options) {
1683 1684
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1685
  if (s.ok() && options.wait) {
1686
    // Wait until the compaction completes
1687
    s = WaitForFlushMemTable(cfd);
1688 1689
  }
  return s;
J
jorlow@chromium.org 已提交
1690 1691
}

1692
Status DBImpl::WaitForFlushMemTable(ColumnFamilyData* cfd) {
1693 1694 1695
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1696
  while (cfd->imm()->size() > 0 && bg_error_.ok()) {
1697 1698
    bg_cv_.Wait();
  }
1699
  if (!bg_error_.ok()) {
1700 1701 1702
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1703 1704
}

1705
Status DBImpl::TEST_FlushMemTable() {
1706
  return FlushMemTable(default_cf_handle_->cfd(), FlushOptions());
H
heyongqiang 已提交
1707 1708
}

1709 1710 1711 1712 1713 1714 1715 1716 1717
Status DBImpl::TEST_WaitForFlushMemTable(ColumnFamilyHandle* column_family) {
  ColumnFamilyData* cfd;
  if (column_family == nullptr) {
    cfd = default_cf_handle_->cfd();
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    cfd = cfh->cfd();
  }
  return WaitForFlushMemTable(cfd);
1718 1719 1720
}

Status DBImpl::TEST_WaitForCompact() {
1721
  // Wait until the compaction completes
1722 1723 1724 1725 1726

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

1727
  MutexLock l(&mutex_);
1728 1729
  while ((bg_compaction_scheduled_ || bg_flush_scheduled_) &&
         bg_error_.ok()) {
1730 1731 1732
    bg_cv_.Wait();
  }
  return bg_error_;
1733 1734
}

1735
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
1736
  mutex_.AssertHeld();
1737
  bg_schedule_needed_ = false;
1738 1739
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
J
jorlow@chromium.org 已提交
1740 1741 1742
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
  } else {
1743
    bool is_flush_pending = false;
1744
    // no need to refcount since we're under a mutex
1745
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1746
      if (cfd->imm()->IsFlushPending()) {
1747 1748 1749
        is_flush_pending = true;
      }
    }
1750
    if (is_flush_pending) {
1751
      // memtable flush needed
1752 1753 1754 1755 1756 1757
      // max_background_compactions should not be 0, because that means
      // flush will never get executed
      assert(options_.max_background_flushes != 0);
      if (bg_flush_scheduled_ < options_.max_background_flushes) {
        bg_flush_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::HIGH);
1758 1759
      } else {
        bg_schedule_needed_ = true;
1760
      }
1761
    }
1762
    bool is_compaction_needed = false;
1763
    // no need to refcount since we're under a mutex
1764
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
1765
      if (cfd->current()->NeedsCompaction()) {
1766 1767 1768 1769
        is_compaction_needed = true;
        break;
      }
    }
1770

1771 1772
    // Schedule BGWorkCompaction if there's a compaction pending
    // Do it only if max_background_compactions hasn't been reached and, in case
1773
    // bg_manual_only_ > 0, if it's a manual compaction.
1774
    if ((manual_compaction_ || is_compaction_needed) &&
1775
        (!bg_manual_only_ || manual_compaction_)) {
1776 1777 1778 1779 1780 1781
      if (bg_compaction_scheduled_ < options_.max_background_compactions) {
        bg_compaction_scheduled_++;
        env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW);
      } else {
        bg_schedule_needed_ = true;
      }
1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793
    }
  }
}

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

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

I
Igor Canadi 已提交
1794
Status DBImpl::BackgroundFlush(bool* madeProgress,
H
Haobo Xu 已提交
1795 1796
                               DeletionState& deletion_state,
                               LogBuffer* log_buffer) {
1797 1798 1799 1800 1801 1802 1803
  mutex_.AssertHeld();
  // call_status is failure if at least one flush was a failure. even if
  // flushing one column family reports a failure, we will continue flushing
  // other column families. however, call_status will be a failure in that case.
  Status call_status;
  autovector<ColumnFamilyData*> to_delete;
  // refcounting in iteration
1804
  for (auto cfd : *versions_->GetColumnFamilySet()) {
1805 1806 1807
    cfd->Ref();
    Status flush_status;
    while (flush_status.ok() && cfd->imm()->IsFlushPending()) {
1808 1809 1810 1811
      Log(options_.info_log,
          "BackgroundCallFlush doing FlushMemTableToOutputFile with column "
          "family %u, flush slots available %d",
          cfd->GetID(), options_.max_background_flushes - bg_flush_scheduled_);
1812 1813
      flush_status = FlushMemTableToOutputFile(cfd, madeProgress,
                                               deletion_state, log_buffer);
1814 1815 1816
    }
    if (call_status.ok() && !flush_status.ok()) {
      call_status = flush_status;
1817
    }
1818 1819 1820 1821 1822 1823
    if (cfd->Unref()) {
      to_delete.push_back(cfd);
    }
  }
  for (auto cfd : to_delete) {
    delete cfd;
J
jorlow@chromium.org 已提交
1824
  }
1825
  return call_status;
J
jorlow@chromium.org 已提交
1826 1827
}

1828
void DBImpl::BackgroundCallFlush() {
1829
  bool madeProgress = false;
K
Kai Liu 已提交
1830
  DeletionState deletion_state(true);
1831 1832
  assert(bg_flush_scheduled_);

H
Haobo Xu 已提交
1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859
  LogBuffer log_buffer(INFO, options_.info_log.get());
  {
    MutexLock l(&mutex_);

    Status s;
    if (!shutting_down_.Acquire_Load()) {
      s = BackgroundFlush(&madeProgress, deletion_state, &log_buffer);
      if (!s.ok()) {
        // Wait a little bit before retrying background compaction in
        // case this is an environmental problem and we do not want to
        // chew up resources for failed compactions for the duration of
        // the problem.
        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();
        log_buffer.FlushBufferToLog();
        LogFlush(options_.info_log);
        env_->SleepForMicroseconds(1000000);
        mutex_.Lock();
      }
    }

    // If !s.ok(), this means that Flush failed. In that case, we want
    // to delete all obsolete files and we force FindObsoleteFiles()
    FindObsoleteFiles(deletion_state, !s.ok());
    // delete unnecessary files if any, this is done outside the mutex
1860
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
1861
      mutex_.Unlock();
1862 1863 1864 1865 1866
      // Have to flush the info logs before bg_flush_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
1867
      log_buffer.FlushBufferToLog();
1868 1869 1870
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
1871 1872
      mutex_.Lock();
    }
I
Igor Canadi 已提交
1873

H
Haobo Xu 已提交
1874
    bg_flush_scheduled_--;
1875 1876 1877 1878 1879
    // Any time the mutex is released After finding the work to do, another
    // thread might execute MaybeScheduleFlushOrCompaction(). It is possible
    // that there is a pending job but it is not scheduled because of the
    // max thread limit.
    if (madeProgress || bg_schedule_needed_) {
H
Haobo Xu 已提交
1880 1881 1882
      MaybeScheduleFlushOrCompaction();
    }
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
1883 1884 1885 1886
    // IMPORTANT: there should be no code after calling SignalAll. This call may
    // signal the DB destructor that it's OK to proceed with destruction. In
    // that case, all DB variables will be dealloacated and referencing them
    // will cause trouble.
1887
  }
J
jorlow@chromium.org 已提交
1888 1889
}

1890

1891 1892 1893 1894
void DBImpl::TEST_PurgeObsoleteteWAL() {
  PurgeObsoleteWALFiles();
}

1895 1896
uint64_t DBImpl::TEST_GetLevel0TotalSize() {
  MutexLock l(&mutex_);
1897
  return default_cf_handle_->cfd()->current()->NumLevelBytes(0);
1898 1899
}

1900
void DBImpl::BackgroundCallCompaction() {
1901
  bool madeProgress = false;
K
Kai Liu 已提交
1902
  DeletionState deletion_state(true);
H
Haobo Xu 已提交
1903 1904

  MaybeDumpStats();
H
Haobo Xu 已提交
1905
  LogBuffer log_buffer(INFO, options_.info_log.get());
1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920
  {
    MutexLock l(&mutex_);
    // Log(options_.info_log, "XXX BG Thread %llx process new work item",
    //     pthread_self());
    assert(bg_compaction_scheduled_);
    Status s;
    if (!shutting_down_.Acquire_Load()) {
      s = BackgroundCompaction(&madeProgress, deletion_state, &log_buffer);
      if (!s.ok()) {
        // Wait a little bit before retrying background compaction in
        // case this is an environmental problem and we do not want to
        // chew up resources for failed compactions for the duration of
        // the problem.
        bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
        mutex_.Unlock();
H
Haobo Xu 已提交
1921
        log_buffer.FlushBufferToLog();
1922 1923 1924 1925 1926 1927 1928
        Log(options_.info_log, "Waiting after background compaction error: %s",
            s.ToString().c_str());
        LogFlush(options_.info_log);
        env_->SleepForMicroseconds(1000000);
        mutex_.Lock();
      }
    }
H
Haobo Xu 已提交
1929

1930 1931 1932 1933 1934 1935 1936
    // If !s.ok(), this means that Compaction failed. In that case, we want
    // to delete all obsolete files we might have created and we force
    // FindObsoleteFiles(). This is because deletion_state does not catch
    // all created files if compaction failed.
    FindObsoleteFiles(deletion_state, !s.ok());

    // delete unnecessary files if any, this is done outside the mutex
1937
    if (deletion_state.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
1938
      mutex_.Unlock();
1939 1940 1941 1942 1943
      // Have to flush the info logs before bg_compaction_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
1944
      log_buffer.FlushBufferToLog();
1945 1946 1947
      if (deletion_state.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(deletion_state);
      }
1948 1949
      mutex_.Lock();
    }
D
Dhruba Borthakur 已提交
1950

1951
    bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1952

1953
    MaybeScheduleLogDBDeployStats();
1954

1955 1956 1957
    // Previous compaction may have produced too many files in a level,
    // So reschedule another compaction if we made progress in the
    // last compaction.
1958 1959 1960 1961 1962 1963
    //
    // Also, any time the mutex is released After finding the work to do,
    // another thread might execute MaybeScheduleFlushOrCompaction(). It is
    // possible  that there is a pending job but it is not scheduled because of
    // the max thread limit.
    if (madeProgress || bg_schedule_needed_) {
1964 1965 1966
      MaybeScheduleFlushOrCompaction();
    }
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
1967 1968 1969 1970
    // IMPORTANT: there should be no code after calling SignalAll. This call may
    // signal the DB destructor that it's OK to proceed with destruction. In
    // that case, all DB variables will be dealloacated and referencing them
    // will cause trouble.
1971
  }
J
jorlow@chromium.org 已提交
1972 1973
}

A
Abhishek Kona 已提交
1974
Status DBImpl::BackgroundCompaction(bool* madeProgress,
1975 1976
                                    DeletionState& deletion_state,
                                    LogBuffer* log_buffer) {
1977
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1978
  mutex_.AssertHeld();
1979

1980
  unique_ptr<Compaction> c;
1981 1982
  bool is_manual = (manual_compaction_ != nullptr) &&
                   (manual_compaction_->in_progress == false);
1983 1984
  InternalKey manual_end_storage;
  InternalKey* manual_end = &manual_end_storage;
H
hans@chromium.org 已提交
1985
  if (is_manual) {
G
Gabor Cselle 已提交
1986
    ManualCompaction* m = manual_compaction_;
1987
    m->in_progress = true;
I
Igor Canadi 已提交
1988 1989
    c.reset(m->cfd->CompactRange(m->input_level, m->output_level, m->begin,
                                 m->end, &manual_end));
1990
    if (!c) {
1991
      m->done = true;
G
Gabor Cselle 已提交
1992
    }
1993 1994
    LogToBuffer(
        log_buffer,
1995 1996
        "Manual compaction from level-%d to level-%d from %s .. %s; will stop "
        "at %s\n",
1997
        m->input_level, m->output_level,
G
Gabor Cselle 已提交
1998 1999
        (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
        (m->end ? m->end->DebugString().c_str() : "(end)"),
2000 2001 2002
        ((m->done || manual_end == nullptr)
             ? "(end)"
             : manual_end->DebugString().c_str()));
I
Igor Canadi 已提交
2003
  } else {
2004
    // no need to refcount in iteration since it's always under a mutex
I
Igor Canadi 已提交
2005
    for (auto cfd : *versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
2006
      if (!cfd->options()->disable_auto_compactions) {
2007
        c.reset(cfd->PickCompaction(log_buffer));
I
Igor Canadi 已提交
2008 2009 2010 2011 2012 2013
        if (c != nullptr) {
          // update statistics
          MeasureTime(options_.statistics.get(), NUM_FILES_IN_SINGLE_COMPACTION,
                      c->inputs(0)->size());
          break;
        }
I
Igor Canadi 已提交
2014 2015
      }
    }
J
jorlow@chromium.org 已提交
2016 2017 2018
  }

  Status status;
2019
  if (!c) {
H
hans@chromium.org 已提交
2020
    // Nothing to do
2021
    LogToBuffer(log_buffer, "Compaction nothing to do");
H
hans@chromium.org 已提交
2022
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
2023
    // Move file to next level
2024
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
2025 2026 2027
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
2028 2029
                       f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
I
Igor Canadi 已提交
2030
    status = versions_->LogAndApply(c->column_family_data(), c->edit(), &mutex_,
2031
                                    db_directory_.get());
I
Igor Canadi 已提交
2032
    InstallSuperVersion(c->column_family_data(), deletion_state);
2033 2034 2035
    if (options_.allow_thread_local) {
      c->column_family_data()->ResetThreadLocalSuperVersions();
    }
2036

2037
    Version::LevelSummaryStorage tmp;
2038
    LogToBuffer(log_buffer, "Moved #%lld to level-%d %lld bytes %s: %s\n",
2039
        static_cast<unsigned long long>(f->number), c->level() + 1,
J
jorlow@chromium.org 已提交
2040
        static_cast<unsigned long long>(f->file_size),
I
Igor Canadi 已提交
2041 2042
        status.ToString().c_str(), c->input_version()->LevelSummary(&tmp));
    c->ReleaseCompactionFiles(status);
2043
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2044
  } else {
2045
    MaybeScheduleFlushOrCompaction(); // do more compaction work in parallel.
2046
    CompactionState* compact = new CompactionState(c.get());
H
Haobo Xu 已提交
2047
    status = DoCompactionWork(compact, deletion_state, log_buffer);
2048
    CleanupCompaction(compact, status);
I
Igor Canadi 已提交
2049
    c->ReleaseCompactionFiles(status);
2050
    c->ReleaseInputs();
2051
    *madeProgress = true;
J
jorlow@chromium.org 已提交
2052
  }
2053
  c.reset();
J
jorlow@chromium.org 已提交
2054 2055 2056 2057 2058 2059

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
2060 2061
    Log(WARN, options_.info_log, "Compaction error: %s",
        status.ToString().c_str());
J
jorlow@chromium.org 已提交
2062 2063 2064 2065
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
2066 2067

  if (is_manual) {
G
Gabor Cselle 已提交
2068
    ManualCompaction* m = manual_compaction_;
2069
    if (!status.ok()) {
L
Lei Jin 已提交
2070
      m->status = status;
2071 2072
      m->done = true;
    }
2073 2074 2075 2076 2077 2078 2079 2080 2081
    // 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.
2082 2083 2084 2085 2086
    //
    // 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) {
2087 2088
      m->done = true;
    }
G
Gabor Cselle 已提交
2089 2090 2091
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
2092
      // Universal compaction should always compact the whole range
I
Igor Canadi 已提交
2093
      assert(m->cfd->options()->compaction_style != kCompactionStyleUniversal);
2094
      m->tmp_storage = *manual_end;
G
Gabor Cselle 已提交
2095 2096
      m->begin = &m->tmp_storage;
    }
2097
    m->in_progress = false; // not being processed anymore
2098
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
2099
  }
2100
  return status;
J
jorlow@chromium.org 已提交
2101 2102
}

2103
void DBImpl::CleanupCompaction(CompactionState* compact, Status status) {
J
jorlow@chromium.org 已提交
2104
  mutex_.AssertHeld();
2105
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
2106 2107
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
2108
    compact->builder.reset();
J
jorlow@chromium.org 已提交
2109
  } else {
2110
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
2111
  }
D
dgrogan@chromium.org 已提交
2112
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2113 2114
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
2115 2116 2117 2118

    // If this file was inserted into the table cache then remove
    // them here because this compaction was not committed.
    if (!status.ok()) {
I
Igor Canadi 已提交
2119
      TableCache::Evict(table_cache_.get(), out.number);
2120
    }
J
jorlow@chromium.org 已提交
2121 2122 2123 2124
  }
  delete compact;
}

2125
// Allocate the file numbers for the output file. We allocate as
2126
// many output file numbers as there are files in level+1 (at least one)
2127 2128 2129
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
2130 2131
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
2132
  int filesNeeded = compact->compaction->num_input_files(1);
2133
  for (int i = 0; i < std::max(filesNeeded, 1); i++) {
2134 2135 2136 2137 2138 2139 2140 2141 2142
    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();
2143
  for (const auto file_number : compact->allocated_file_numbers) {
2144 2145 2146 2147 2148
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
2149
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
2150 2151
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
2152
  uint64_t file_number;
2153 2154 2155 2156 2157 2158 2159
  // 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 已提交
2160 2161 2162 2163 2164
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
2165 2166 2167 2168
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
2169
  out.smallest_seqno = out.largest_seqno = 0;
2170
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
2171 2172 2173

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

J
jorlow@chromium.org 已提交
2176
  if (s.ok()) {
2177 2178
    // Over-estimate slightly so we don't end up just barely crossing
    // the threshold.
I
Igor Canadi 已提交
2179
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2180
    compact->outfile->SetPreallocationBlockSize(
I
Igor Canadi 已提交
2181
        1.1 * cfd->compaction_picker()->MaxFileSizeForLevel(
2182
                  compact->compaction->output_level()));
2183

I
Igor Canadi 已提交
2184 2185 2186
    CompressionType compression_type =
        GetCompressionType(*cfd->options(), compact->compaction->output_level(),
                           compact->compaction->enable_compression());
S
Siying Dong 已提交
2187

2188
    compact->builder.reset(
I
Igor Canadi 已提交
2189
        NewTableBuilder(*cfd->options(), cfd->internal_comparator(),
2190
                        compact->outfile.get(), compression_type));
J
jorlow@chromium.org 已提交
2191
  }
I
Igor Canadi 已提交
2192
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2193 2194 2195 2196 2197
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
2198
  assert(compact != nullptr);
2199
  assert(compact->outfile);
2200
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215

  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;
2216
  compact->builder.reset();
J
jorlow@chromium.org 已提交
2217 2218

  // Finish and check for file errors
2219
  if (s.ok() && !options_.disableDataSync) {
2220
    if (options_.use_fsync) {
2221
      StopWatch sw(env_, options_.statistics.get(),
2222
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2223 2224
      s = compact->outfile->Fsync();
    } else {
2225
      StopWatch sw(env_, options_.statistics.get(),
2226
                   COMPACTION_OUTFILE_SYNC_MICROS, false);
2227 2228
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
2229 2230 2231 2232
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
2233
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
2234 2235 2236

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
I
Igor Canadi 已提交
2237
    ColumnFamilyData* cfd = compact->compaction->column_family_data();
2238
    FileMetaData meta(output_number, current_bytes);
I
Igor Canadi 已提交
2239
    Iterator* iter = cfd->table_cache()->NewIterator(
2240
        ReadOptions(), storage_options_, cfd->internal_comparator(), meta);
J
jorlow@chromium.org 已提交
2241 2242 2243
    s = iter->status();
    delete iter;
    if (s.ok()) {
2244
      Log(options_.info_log,
K
Kai Liu 已提交
2245 2246 2247 2248
          "Generated table #%lu: %lu keys, %lu bytes",
          (unsigned long) output_number,
          (unsigned long) current_entries,
          (unsigned long) current_bytes);
J
jorlow@chromium.org 已提交
2249 2250 2251 2252 2253 2254 2255 2256
    }
  }
  return s;
}


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
2257 2258 2259 2260 2261

  // 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.
2262
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
2263 2264 2265 2266
    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),
I
Igor Canadi 已提交
2267
      compact->compaction->output_level());
L
Lei Jin 已提交
2268
    return Status::Corruption("Compaction input files inconsistent");
2269 2270
  }

2271
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
2272 2273 2274
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
I
Igor Canadi 已提交
2275
      compact->compaction->output_level(),
J
jorlow@chromium.org 已提交
2276 2277 2278 2279
      static_cast<long long>(compact->total_bytes));

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
D
dgrogan@chromium.org 已提交
2280
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
2281 2282
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
2283 2284
        compact->compaction->output_level(), out.number, out.file_size,
        out.smallest, out.largest, out.smallest_seqno, out.largest_seqno);
J
jorlow@chromium.org 已提交
2285
  }
I
Igor Canadi 已提交
2286 2287 2288
  return versions_->LogAndApply(compact->compaction->column_family_data(),
                                compact->compaction->edit(), &mutex_,
                                db_directory_.get());
J
jorlow@chromium.org 已提交
2289 2290
}

2291 2292 2293 2294 2295 2296 2297 2298
//
// 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(
2299 2300
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
2301
  SequenceNumber prev __attribute__((unused)) = 0;
2302 2303 2304
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
2305
      *prev_snapshot = prev;
2306
      return cur;
2307
    }
2308 2309
    prev = cur; // assignment
    assert(prev);
2310 2311
  }
  Log(options_.info_log,
K
Kai Liu 已提交
2312 2313 2314
      "Looking for seqid %lu but maxseqid is %lu",
      (unsigned long)in,
      (unsigned long)snapshots[snapshots.size()-1]);
2315 2316 2317 2318
  assert(0);
  return 0;
}

I
Igor Canadi 已提交
2319
Status DBImpl::DoCompactionWork(CompactionState* compact,
H
Haobo Xu 已提交
2320 2321
                                DeletionState& deletion_state,
                                LogBuffer* log_buffer) {
2322
  assert(compact);
2323
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions
I
Igor Canadi 已提交
2324
  ColumnFamilyData* cfd = compact->compaction->column_family_data();
A
Abhishek Kona 已提交
2325
  Log(options_.info_log,
I
Igor Canadi 已提交
2326 2327 2328 2329
      "[CF %u] Compacting %d@%d + %d@%d files, score %.2f slots available %d",
      cfd->GetID(), compact->compaction->num_input_files(0),
      compact->compaction->level(), compact->compaction->num_input_files(1),
      compact->compaction->output_level(), compact->compaction->score(),
2330
      options_.max_background_compactions - bg_compaction_scheduled_);
M
Mike Lin 已提交
2331
  char scratch[2345];
2332
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
2333
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
2334

2335 2336
  assert(compact->compaction->input_version()->NumLevelFiles(
             compact->compaction->level()) > 0);
2337
  assert(compact->builder == nullptr);
2338
  assert(!compact->outfile);
2339 2340 2341

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
H
Haobo Xu 已提交
2342
  SequenceNumber latest_snapshot = 0;
2343 2344 2345 2346 2347
  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 已提交
2348
  } else {
H
Haobo Xu 已提交
2349
    latest_snapshot = compact->existing_snapshots.back();
2350 2351 2352 2353
    // 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 已提交
2354 2355
  }

2356
  // Is this compaction producing files at the bottommost level?
2357
  bool bottommost_level = compact->compaction->BottomMostLevel();
2358

2359 2360 2361
  // Allocate the output file numbers before we release the lock
  AllocateCompactionOutputFileNumbers(compact);

J
jorlow@chromium.org 已提交
2362 2363
  // Release mutex while we're actually doing the compaction work
  mutex_.Unlock();
H
Haobo Xu 已提交
2364 2365
  // flush log buffer immediately after releasing the mutex
  log_buffer->FlushBufferToLog();
J
jorlow@chromium.org 已提交
2366

2367
  const uint64_t start_micros = env_->NowMicros();
2368
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
2369 2370 2371 2372 2373
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
2374 2375
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
2376
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2377
  std::string compaction_filter_value;
H
Haobo Xu 已提交
2378
  std::vector<char> delete_key; // for compaction filter
I
Igor Canadi 已提交
2379 2380 2381
  MergeHelper merge(
      cfd->user_comparator(), cfd->options()->merge_operator.get(),
      options_.info_log.get(), false /* internal key corruption is expected */);
I
Igor Canadi 已提交
2382
  auto compaction_filter = cfd->options()->compaction_filter;
2383 2384
  std::unique_ptr<CompactionFilter> compaction_filter_from_factory = nullptr;
  if (!compaction_filter) {
2385 2386
    auto context = compact->GetFilterContext();
    compaction_filter_from_factory =
I
Igor Canadi 已提交
2387 2388
        cfd->options()->compaction_filter_factory->CreateCompactionFilter(
            context);
2389 2390
    compaction_filter = compaction_filter_from_factory.get();
  }
2391

2392 2393
  while (input->Valid() && !shutting_down_.Acquire_Load() &&
         !cfd->IsDropped()) {
J
jorlow@chromium.org 已提交
2394
    Slice key = input->key();
2395
    Slice value = input->value();
H
Haobo Xu 已提交
2396

2397
    if (compact->compaction->ShouldStopBefore(key) &&
2398
        compact->builder != nullptr) {
2399
      status = FinishCompactionOutputFile(compact, input.get());
2400 2401 2402 2403 2404 2405
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
2406
    bool drop = false;
2407
    bool current_entry_is_merging = false;
J
jorlow@chromium.org 已提交
2408 2409
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
2410 2411
      // 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 已提交
2412 2413 2414
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
2415
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
2416 2417
    } else {
      if (!has_current_user_key ||
2418 2419
          cfd->user_comparator()->Compare(ikey.user_key,
                                          Slice(current_user_key)) != 0) {
J
jorlow@chromium.org 已提交
2420 2421 2422 2423
        // 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;
2424
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
2425 2426

        // apply the compaction filter to the first occurrence of the user key
2427
        if (compaction_filter &&
H
Haobo Xu 已提交
2428 2429 2430 2431 2432 2433 2434 2435 2436
            ikey.type == kTypeValue &&
            (visible_at_tip || ikey.sequence > latest_snapshot)) {
          // If the user has specified a compaction filter and the sequence
          // number is greater than any external snapshot, then invoke the
          // filter.
          // If the return value of the compaction filter is true, replace
          // the entry with a delete marker.
          bool value_changed = false;
          compaction_filter_value.clear();
I
Igor Canadi 已提交
2437 2438 2439
          bool to_delete = compaction_filter->Filter(
              compact->compaction->level(), ikey.user_key, value,
              &compaction_filter_value, &value_changed);
H
Haobo Xu 已提交
2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451
          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();
2452
            RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_USER);
H
Haobo Xu 已提交
2453 2454 2455 2456
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }
J
jorlow@chromium.org 已提交
2457 2458
      }

2459 2460 2461
      // 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.
2462 2463 2464 2465 2466 2467
      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);
2468 2469 2470 2471 2472

      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 已提交
2473
        // Hidden by an newer entry for same user key
2474
        // TODO: why not > ?
2475
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
2476
        drop = true;    // (A)
2477
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
2478
      } else if (ikey.type == kTypeDeletion &&
2479
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
2480 2481 2482 2483 2484 2485 2486 2487 2488
                 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;
2489
        RecordTick(options_.statistics.get(), COMPACTION_KEY_DROP_OBSOLETE);
2490 2491 2492 2493 2494 2495 2496
      } 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 已提交
2497
        merge.MergeUntil(input.get(), prev_snapshot, bottommost_level,
2498
                         options_.statistics.get());
2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516
        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 已提交
2517 2518 2519
      }

      last_sequence_for_key = ikey.sequence;
2520
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
2521 2522
    }
#if 0
2523
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
2524
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
2525
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
2526
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
2527
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
2528
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
2529 2530
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
2531 2532 2533
#endif

    if (!drop) {
2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549
      // 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);
2550
      }
2551

2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562
      // 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.
2563
        if (bottommost_level && ikey.sequence < earliest_snapshot &&
2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574
            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
2575

2576 2577 2578 2579 2580 2581 2582
        // Open output file if necessary
        if (compact->builder == nullptr) {
          status = OpenCompactionOutputFile(compact);
          if (!status.ok()) {
            break;
          }
        }
2583 2584

        SequenceNumber seqno = GetInternalKeySeqno(newkey);
2585 2586
        if (compact->builder->NumEntries() == 0) {
          compact->current_output()->smallest.DecodeFrom(newkey);
2587 2588 2589 2590
          compact->current_output()->smallest_seqno = seqno;
        } else {
          compact->current_output()->smallest_seqno =
            std::min(compact->current_output()->smallest_seqno, seqno);
2591 2592 2593
        }
        compact->current_output()->largest.DecodeFrom(newkey);
        compact->builder->Add(newkey, value);
2594 2595
        compact->current_output()->largest_seqno =
          std::max(compact->current_output()->largest_seqno, seqno);
2596 2597 2598 2599 2600 2601 2602 2603

        // 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 已提交
2604 2605
        }

2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625
        // 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 已提交
2626 2627 2628 2629 2630
          break;
        }
      }
    }

2631
    // MergeUntil has moved input to the next entry
2632
    if (!current_entry_is_merging) {
2633 2634
      input->Next();
    }
J
jorlow@chromium.org 已提交
2635 2636
  }

2637
  if (status.ok() && (shutting_down_.Acquire_Load() || cfd->IsDropped())) {
L
Lei Jin 已提交
2638 2639
    status = Status::ShutdownInProgress(
        "Database shutdown started during compaction");
J
jorlow@chromium.org 已提交
2640
  }
2641
  if (status.ok() && compact->builder != nullptr) {
2642
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
2643 2644 2645 2646
  }
  if (status.ok()) {
    status = input->status();
  }
2647
  input.reset();
J
jorlow@chromium.org 已提交
2648

2649 2650 2651
  if (!options_.disableDataSync) {
    db_directory_->Fsync();
  }
I
Igor Canadi 已提交
2652 2653

  InternalStats::CompactionStats stats;
2654
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
I
Igor Canadi 已提交
2655
  MeasureTime(options_.statistics.get(), COMPACTION_TIME, stats.micros);
M
Mark Callaghan 已提交
2656 2657
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
2658 2659

  int num_output_files = compact->outputs.size();
2660
  if (compact->builder != nullptr) {
P
Pascal Borreli 已提交
2661
    // An error occurred so ignore the last output.
2662 2663 2664 2665
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
2666

2667
  for (int i = 0; i < compact->compaction->num_input_files(0); i++) {
M
Mark Callaghan 已提交
2668
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;
2669 2670 2671
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(0, i)->file_size);
  }
M
Mark Callaghan 已提交
2672

2673
  for (int i = 0; i < compact->compaction->num_input_files(1); i++) {
M
Mark Callaghan 已提交
2674
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;
2675 2676 2677
    RecordTick(options_.statistics.get(), COMPACT_READ_BYTES,
               compact->compaction->input(1, i)->file_size);
  }
M
Mark Callaghan 已提交
2678

2679
  for (int i = 0; i < num_output_files; i++) {
2680
    stats.bytes_written += compact->outputs[i].file_size;
2681 2682
    RecordTick(options_.statistics.get(), COMPACT_WRITE_BYTES,
               compact->outputs[i].file_size);
2683 2684
  }

I
Igor Canadi 已提交
2685
  LogFlush(options_.info_log);
J
jorlow@chromium.org 已提交
2686
  mutex_.Lock();
2687 2688
  cfd->internal_stats()->AddCompactionStats(compact->compaction->output_level(),
                                            stats);
J
jorlow@chromium.org 已提交
2689

2690 2691 2692 2693
  // 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 已提交
2694 2695
  if (status.ok()) {
    status = InstallCompactionResults(compact);
I
Igor Canadi 已提交
2696
    InstallSuperVersion(cfd, deletion_state);
J
jorlow@chromium.org 已提交
2697
  }
2698
  Version::LevelSummaryStorage tmp;
2699
  Log(options_.info_log,
M
Mark Callaghan 已提交
2700
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
2701 2702
      "MB in(%.1f, %.1f) out(%.1f), read-write-amplify(%.1f) "
      "write-amplify(%.1f) %s\n",
2703
      cfd->current()->LevelSummary(&tmp),
M
Mark Callaghan 已提交
2704
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
2705 2706 2707 2708
          (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 已提交
2709
      stats.bytes_written / 1048576.0,
2710
      (stats.bytes_written + stats.bytes_readnp1 + stats.bytes_readn) /
2711 2712
          (double)stats.bytes_readn,
      stats.bytes_written / (double)stats.bytes_readn,
2713
      status.ToString().c_str());
M
Mark Callaghan 已提交
2714

J
jorlow@chromium.org 已提交
2715 2716 2717
  return status;
}

2718 2719
namespace {
struct IterState {
2720 2721
  IterState(DBImpl* db, port::Mutex* mu, SuperVersion* super_version)
      : db(db), mu(mu), super_version(super_version) {}
2722 2723

  DBImpl* db;
2724
  port::Mutex* mu;
2725
  SuperVersion* super_version;
2726 2727 2728 2729
};

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

  bool need_cleanup = state->super_version->Unref();
  if (need_cleanup) {
2733 2734
    DBImpl::DeletionState deletion_state;

2735 2736 2737 2738 2739 2740
    state->mu->Lock();
    state->super_version->Cleanup();
    state->db->FindObsoleteFiles(deletion_state, false, true);
    state->mu->Unlock();

    delete state->super_version;
2741 2742 2743
    if (deletion_state.HaveSomethingToDelete()) {
      state->db->PurgeObsoleteFiles(deletion_state);
    }
I
Igor Canadi 已提交
2744
  }
T
Tomislav Novak 已提交
2745

2746 2747
  delete state;
}
H
Hans Wennborg 已提交
2748
}  // namespace
2749

J
jorlow@chromium.org 已提交
2750
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
2751 2752
                                      ColumnFamilyData* cfd,
                                      SuperVersion* super_version) {
I
Igor Canadi 已提交
2753
  std::vector<Iterator*> iterator_list;
2754 2755
  // Collect iterator for mutable mem
  iterator_list.push_back(super_version->mem->NewIterator(options));
T
Tomislav Novak 已提交
2756
  // Collect all needed child iterators for immutable memtables
2757
  super_version->imm->AddIterators(options, &iterator_list);
T
Tomislav Novak 已提交
2758
  // Collect iterators for files in L0 - Ln
2759 2760
  super_version->current->AddIterators(options, storage_options_,
                                       &iterator_list);
2761 2762 2763
  Iterator* internal_iter =
      NewMergingIterator(env_, &cfd->internal_comparator(), &iterator_list[0],
                         iterator_list.size());
2764 2765

  IterState* cleanup = new IterState(this, &mutex_, super_version);
2766
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2767 2768 2769 2770

  return internal_iter;
}

2771 2772 2773 2774
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

2775 2776 2777 2778 2779 2780 2781 2782 2783
Iterator* DBImpl::TEST_NewInternalIterator(ColumnFamilyHandle* column_family) {
  ColumnFamilyData* cfd;
  if (column_family == nullptr) {
    cfd = default_cf_handle_->cfd();
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    cfd = cfh->cfd();
  }

2784
  mutex_.Lock();
2785
  SuperVersion* super_version = cfd->GetSuperVersion()->Ref();
2786
  mutex_.Unlock();
I
Igor Canadi 已提交
2787 2788 2789
  ReadOptions roptions;
  roptions.prefix_seek = true;
  return NewInternalIterator(roptions, cfd, super_version);
J
jorlow@chromium.org 已提交
2790 2791
}

T
Tomislav Novak 已提交
2792
std::pair<Iterator*, Iterator*> DBImpl::GetTailingIteratorPair(
2793
    const ReadOptions& options, ColumnFamilyData* cfd,
T
Tomislav Novak 已提交
2794 2795 2796
    uint64_t* superversion_number) {

  mutex_.Lock();
2797
  SuperVersion* super_version = cfd->GetSuperVersion()->Ref();
T
Tomislav Novak 已提交
2798
  if (superversion_number != nullptr) {
2799
    *superversion_number = cfd->GetSuperVersionNumber();
T
Tomislav Novak 已提交
2800 2801 2802
  }
  mutex_.Unlock();

2803
  Iterator* mutable_iter = super_version->mem->NewIterator(options);
T
Tomislav Novak 已提交
2804
  // create a DBIter that only uses memtable content; see NewIterator()
I
Igor Canadi 已提交
2805
  mutable_iter =
I
Igor Canadi 已提交
2806 2807
      NewDBIterator(&dbname_, env_, *cfd->options(), cfd->user_comparator(),
                    mutable_iter, kMaxSequenceNumber);
T
Tomislav Novak 已提交
2808 2809

  std::vector<Iterator*> list;
2810 2811
  super_version->imm->AddIterators(options, &list);
  super_version->current->AddIterators(options, storage_options_, &list);
2812 2813
  Iterator* immutable_iter = NewMergingIterator(
      env_, &cfd->internal_comparator(), &list[0], list.size());
T
Tomislav Novak 已提交
2814 2815

  // create a DBIter that only uses memtable content; see NewIterator()
2816
  immutable_iter =
I
Igor Canadi 已提交
2817 2818
      NewDBIterator(&dbname_, env_, *cfd->options(), cfd->user_comparator(),
                    immutable_iter, kMaxSequenceNumber);
T
Tomislav Novak 已提交
2819

2820 2821 2822 2823 2824 2825 2826 2827
  // register cleanups
  mutable_iter->RegisterCleanup(CleanupIteratorState,
    new IterState(this, &mutex_, super_version), nullptr);

  // bump the ref one more time since it will be Unref'ed twice
  immutable_iter->RegisterCleanup(CleanupIteratorState,
    new IterState(this, &mutex_, super_version->Ref()), nullptr);

T
Tomislav Novak 已提交
2828 2829 2830
  return std::make_pair(mutable_iter, immutable_iter);
}

2831 2832 2833 2834 2835 2836 2837 2838 2839
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes(
    ColumnFamilyHandle* column_family) {
  ColumnFamilyData* cfd;
  if (column_family == nullptr) {
    cfd = default_cf_handle_->cfd();
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    cfd = cfh->cfd();
  }
2840
  MutexLock l(&mutex_);
2841
  return cfd->current()->MaxNextLevelOverlappingBytes();
2842 2843
}

J
jorlow@chromium.org 已提交
2844
Status DBImpl::Get(const ReadOptions& options,
2845
                   ColumnFamilyHandle* column_family, const Slice& key,
J
jorlow@chromium.org 已提交
2846
                   std::string* value) {
2847
  return GetImpl(options, column_family, key, value);
2848 2849
}

I
Igor Canadi 已提交
2850 2851 2852
// DeletionState gets created and destructed outside of the lock -- we
// use this convinently to:
// * malloc one SuperVersion() outside of the lock -- new_superversion
2853
// * delete SuperVersion()s outside of the lock -- superversions_to_free
I
Igor Canadi 已提交
2854 2855 2856 2857 2858 2859
//
// 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
2860 2861
void DBImpl::InstallSuperVersion(ColumnFamilyData* cfd,
                                 DeletionState& deletion_state) {
2862
  mutex_.AssertHeld();
I
Igor Canadi 已提交
2863 2864 2865 2866
  // if new_superversion == nullptr, it means somebody already used it
  SuperVersion* new_superversion =
    (deletion_state.new_superversion != nullptr) ?
    deletion_state.new_superversion : new SuperVersion();
2867 2868
  SuperVersion* old_superversion =
      cfd->InstallSuperVersion(new_superversion, &mutex_);
I
Igor Canadi 已提交
2869
  deletion_state.new_superversion = nullptr;
2870
  deletion_state.superversions_to_free.push_back(old_superversion);
2871 2872
  // Reset SuperVersions cached in thread local storage
  if (options_.allow_thread_local) {
2873
    cfd->ResetThreadLocalSuperVersions();
2874
  }
I
Igor Canadi 已提交
2875 2876
}

2877
Status DBImpl::GetImpl(const ReadOptions& options,
2878 2879
                       ColumnFamilyHandle* column_family, const Slice& key,
                       std::string* value, bool* value_found) {
2880
  StopWatch sw(env_, options_.statistics.get(), DB_GET, false);
2881 2882
  StopWatchNano snapshot_timer(env_, false);
  StartPerfTimer(&snapshot_timer);
2883

2884 2885 2886
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

2887
  SequenceNumber snapshot;
2888
  if (options.snapshot != nullptr) {
2889 2890 2891
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2892
  }
2893

2894 2895
  // Acquire SuperVersion
  SuperVersion* sv = nullptr;
2896
  ThreadLocalPtr* thread_local_sv = nullptr;
2897 2898 2899 2900 2901 2902 2903 2904
  if (LIKELY(options_.allow_thread_local)) {
    // The SuperVersion is cached in thread local storage to avoid acquiring
    // mutex when SuperVersion does not change since the last use. When a new
    // SuperVersion is installed, the compaction or flush thread cleans up
    // cached SuperVersion in all existing thread local storage. To avoid
    // acquiring mutex for this operation, we use atomic Swap() on the thread
    // local pointer to guarantee exclusive access. If the thread local pointer
    // is being used while a new SuperVersion is installed, the cached
2905 2906 2907 2908
    // SuperVersion can become stale. In that case, the background thread would
    // have swapped in kSVObsolete. We re-check the value at the end of
    // Get, with an atomic compare and swap. The superversion will be released
    // if detected to be stale.
2909 2910
    thread_local_sv = cfd->GetThreadLocalSuperVersion();
    void* ptr = thread_local_sv->Swap(SuperVersion::kSVInUse);
2911 2912 2913 2914 2915 2916 2917
    // Invariant:
    // (1) Scrape (always) installs kSVObsolete in ThreadLocal storage
    // (2) the Swap above (always) installs kSVInUse, ThreadLocal storage
    // should only keep kSVInUse during a GetImpl.
    assert(ptr != SuperVersion::kSVInUse);
    sv = static_cast<SuperVersion*>(ptr);
    if (sv == SuperVersion::kSVObsolete ||
2918
        sv->version_number != cfd->GetSuperVersionNumber()) {
2919
      RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_ACQUIRES);
2920 2921 2922
      SuperVersion* sv_to_delete = nullptr;

      if (sv && sv->Unref()) {
2923
        RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_CLEANUPS);
2924
        mutex_.Lock();
2925 2926
        // TODO underlying resources held by superversion (sst files) might
        // not be released until the next background job.
2927 2928 2929 2930 2931
        sv->Cleanup();
        sv_to_delete = sv;
      } else {
        mutex_.Lock();
      }
2932
      sv = cfd->GetSuperVersion()->Ref();
2933 2934 2935 2936 2937 2938
      mutex_.Unlock();

      delete sv_to_delete;
    }
  } else {
    mutex_.Lock();
2939
    sv = cfd->GetSuperVersion()->Ref();
2940 2941
    mutex_.Unlock();
  }
I
Igor Canadi 已提交
2942

2943
  bool have_stat_update = false;
2944
  Version::GetStats stats;
2945

2946
  // Prepare to store a list of merge operations if merge occurs.
2947
  MergeContext merge_context;
2948

2949
  Status s;
2950
  // First look in the memtable, then in the immutable memtable (if any).
2951
  // s is both in/out. When in, s could either be OK or MergeInProgress.
2952
  // merge_operands will contain the sequence of merges in the latter case.
2953
  LookupKey lkey(key, snapshot);
2954
  BumpPerfTime(&perf_context.get_snapshot_time, &snapshot_timer);
I
Igor Canadi 已提交
2955
  if (sv->mem->Get(lkey, value, &s, merge_context, *cfd->options())) {
2956
    // Done
2957
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
I
Igor Canadi 已提交
2958
  } else if (sv->imm->Get(lkey, value, &s, merge_context, *cfd->options())) {
2959
    // Done
2960
    RecordTick(options_.statistics.get(), MEMTABLE_HIT);
2961
  } else {
2962
    // Done
2963 2964 2965
    StopWatchNano from_files_timer(env_, false);
    StartPerfTimer(&from_files_timer);

2966
    sv->current->Get(options, lkey, value, &s, &merge_context, &stats,
I
Igor Canadi 已提交
2967
                     *cfd->options(), value_found);
2968
    have_stat_update = true;
2969
    BumpPerfTime(&perf_context.get_from_output_files_time, &from_files_timer);
2970
    RecordTick(options_.statistics.get(), MEMTABLE_MISS);
2971
  }
2972

2973 2974
  StopWatchNano post_process_timer(env_, false);
  StartPerfTimer(&post_process_timer);
2975

I
Igor Canadi 已提交
2976
  if (!cfd->options()->disable_seek_compaction && have_stat_update) {
I
Igor Canadi 已提交
2977
    mutex_.Lock();
2978
    if (sv->current->UpdateStats(stats)) {
I
Igor Canadi 已提交
2979 2980 2981
      MaybeScheduleFlushOrCompaction();
    }
    mutex_.Unlock();
2982 2983
  }

2984
  bool unref_sv = true;
2985 2986
  if (LIKELY(options_.allow_thread_local)) {
    // Put the SuperVersion back
2987
    void* expected = SuperVersion::kSVInUse;
2988
    if (thread_local_sv->CompareAndSwap(static_cast<void*>(sv), expected)) {
2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002
      // When we see kSVInUse in the ThreadLocal, we are sure ThreadLocal
      // storage has not been altered and no Scrape has happend. The
      // SuperVersion is still current.
      unref_sv = false;
    } else {
      // ThreadLocal scrape happened in the process of this GetImpl call (after
      // thread local Swap() at the beginning and before CompareAndSwap()).
      // This means the SuperVersion it holds is obsolete.
      assert(expected == SuperVersion::kSVObsolete);
    }
  }

  if (unref_sv) {
    // Release SuperVersion
3003
    if (sv->Unref()) {
I
Igor Canadi 已提交
3004
      mutex_.Lock();
3005
      sv->Cleanup();
I
Igor Canadi 已提交
3006
      mutex_.Unlock();
3007
      delete sv;
3008
      RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_CLEANUPS);
I
Igor Canadi 已提交
3009
    }
3010
    RecordTick(options_.statistics.get(), NUMBER_SUPERVERSION_RELEASES);
3011
  }
3012

3013 3014
  RecordTick(options_.statistics.get(), NUMBER_KEYS_READ);
  RecordTick(options_.statistics.get(), BYTES_READ, value->size());
K
Kai Liu 已提交
3015
  BumpPerfTime(&perf_context.get_post_process_time, &post_process_timer);
3016
  return s;
J
jorlow@chromium.org 已提交
3017 3018
}

3019 3020
std::vector<Status> DBImpl::MultiGet(
    const ReadOptions& options,
3021
    const std::vector<ColumnFamilyHandle*>& column_family,
3022
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
3023

3024
  StopWatch sw(env_, options_.statistics.get(), DB_MULTIGET, false);
3025 3026
  StopWatchNano snapshot_timer(env_, false);
  StartPerfTimer(&snapshot_timer);
K
Kai Liu 已提交
3027

3028
  SequenceNumber snapshot;
3029

3030
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
3031
    ColumnFamilyData* cfd;
3032 3033 3034 3035 3036 3037 3038
    SuperVersion* super_version;
    Version::GetStats stats;
    bool have_stat_update = false;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
3039 3040 3041 3042 3043 3044
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(cf);
    auto cfd = cfh->cfd();
    if (multiget_cf_data.find(cfd->GetID()) == multiget_cf_data.end()) {
      auto mgcfd = new MultiGetColumnFamilyData();
      mgcfd->cfd = cfd;
      multiget_cf_data.insert({cfd->GetID(), mgcfd});
3045 3046 3047
    }
  }

3048
  mutex_.Lock();
3049 3050 3051 3052 3053
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }
3054
  for (auto mgd_iter : multiget_cf_data) {
3055 3056
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
3057
  }
3058
  mutex_.Unlock();
3059

3060 3061
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
3062

3063
  // Note: this always resizes the values array
3064 3065 3066
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
3067 3068

  // Keep track of bytes that we read for statistics-recording later
3069
  uint64_t bytes_read = 0;
3070
  BumpPerfTime(&perf_context.get_snapshot_time, &snapshot_timer);
3071 3072 3073 3074

  // 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.
3075
  // merge_operands will contain the sequence of merges in the latter case.
3076
  for (size_t i = 0; i < num_keys; ++i) {
3077
    merge_context.Clear();
3078
    Status& s = stat_list[i];
3079 3080 3081
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
3082 3083
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
3084 3085 3086
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
I
Igor Canadi 已提交
3087 3088
    auto cfd = mgd->cfd;
    if (super_version->mem->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3089
                                *cfd->options())) {
3090
      // Done
3091
    } else if (super_version->imm->Get(lkey, value, &s, merge_context,
I
Igor Canadi 已提交
3092
                                       *cfd->options())) {
3093 3094
      // Done
    } else {
3095
      super_version->current->Get(options, lkey, value, &s, &merge_context,
I
Igor Canadi 已提交
3096
                                  &mgd->stats, *cfd->options());
3097
      mgd->have_stat_update = true;
3098 3099 3100
    }

    if (s.ok()) {
3101
      bytes_read += value->size();
3102 3103 3104 3105
    }
  }

  // Post processing (decrement reference counts and record statistics)
3106 3107
  StopWatchNano post_process_timer(env_, false);
  StartPerfTimer(&post_process_timer);
3108 3109 3110 3111 3112 3113
  autovector<SuperVersion*> superversions_to_delete;

  bool schedule_flush_or_compaction = false;
  mutex_.Lock();
  for (auto mgd_iter : multiget_cf_data) {
    auto mgd = mgd_iter.second;
I
Igor Canadi 已提交
3114 3115
    auto cfd = mgd->cfd;
    if (!cfd->options()->disable_seek_compaction && mgd->have_stat_update) {
3116 3117 3118
      if (mgd->super_version->current->UpdateStats(mgd->stats)) {
        schedule_flush_or_compaction = true;
      }
3119
    }
3120 3121 3122
    if (mgd->super_version->Unref()) {
      mgd->super_version->Cleanup();
      superversions_to_delete.push_back(mgd->super_version);
3123 3124
    }
  }
3125 3126 3127 3128 3129 3130 3131 3132 3133 3134
  if (schedule_flush_or_compaction) {
    MaybeScheduleFlushOrCompaction();
  }
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
3135
  }
3136

3137
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_CALLS);
3138 3139
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_KEYS_READ, num_keys);
  RecordTick(options_.statistics.get(), NUMBER_MULTIGET_BYTES_READ, bytes_read);
3140
  BumpPerfTime(&perf_context.get_post_process_time, &post_process_timer);
3141

3142
  return stat_list;
3143 3144
}

3145
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& options,
3146
                                  const std::string& column_family_name,
3147
                                  ColumnFamilyHandle** handle) {
I
Igor Canadi 已提交
3148 3149 3150
  *handle = nullptr;
  MutexLock l(&mutex_);

I
Igor Canadi 已提交
3151 3152
  if (versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name) !=
      nullptr) {
I
Igor Canadi 已提交
3153 3154
    return Status::InvalidArgument("Column family already exists");
  }
3155
  VersionEdit edit;
3156
  edit.AddColumnFamily(column_family_name);
3157 3158
  uint32_t new_id = versions_->GetColumnFamilySet()->GetNextColumnFamilyID();
  edit.SetColumnFamily(new_id);
I
Igor Canadi 已提交
3159
  edit.SetLogNumber(logfile_number_);
I
Igor Canadi 已提交
3160
  edit.SetComparatorName(options.comparator->Name());
3161

I
Igor Canadi 已提交
3162 3163
  // LogAndApply will both write the creation in MANIFEST and create
  // ColumnFamilyData object
3164 3165
  Status s = versions_->LogAndApply(nullptr, &edit, &mutex_,
                                    db_directory_.get(), false, &options);
3166
  if (s.ok()) {
3167 3168 3169
    auto cfd =
        versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
    assert(cfd != nullptr);
3170
    delete cfd->InstallSuperVersion(new SuperVersion(), &mutex_);
3171
    *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
3172 3173 3174 3175 3176 3177
    Log(options_.info_log, "Created column family \"%s\" (ID %u)",
        column_family_name.c_str(), (unsigned)cfd->GetID());
  } else {
    Log(options_.info_log, "Creating column family \"%s\" FAILED -- %s",
        column_family_name.c_str(), s.ToString().c_str());
  }
3178
  return s;
3179 3180
}

3181 3182 3183 3184
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
3185 3186
    return Status::InvalidArgument("Can't drop default column family");
  }
3187

I
Igor Canadi 已提交
3188 3189
  VersionEdit edit;
  edit.DropColumnFamily();
3190 3191
  edit.SetColumnFamily(cfd->GetID());

3192
  Status s;
3193 3194 3195 3196 3197 3198 3199 3200
  {
    MutexLock l(&mutex_);
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
      s = versions_->LogAndApply(cfd, &edit, &mutex_);
    }
3201
  }
3202

3203
  if (s.ok()) {
I
Igor Canadi 已提交
3204
    assert(cfd->IsDropped());
3205
    Log(options_.info_log, "Dropped column family with id %u\n", cfd->GetID());
3206
    // Flush the memtables. This will make all WAL files referencing dropped
I
Igor Canadi 已提交
3207
    // column family to be obsolete. They will be deleted once user deletes
3208 3209
    // column family handle
    Write(WriteOptions(), nullptr);  // ignore error
3210 3211 3212 3213 3214
  } else {
    Log(options_.info_log, "Dropping column family with id %u FAILED -- %s\n",
        cfd->GetID(), s.ToString().c_str());
  }

3215
  return s;
3216 3217
}

3218
bool DBImpl::KeyMayExist(const ReadOptions& options,
3219 3220
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
3221
  if (value_found != nullptr) {
K
Kai Liu 已提交
3222 3223
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
3224
  }
3225 3226
  ReadOptions roptions = options;
  roptions.read_tier = kBlockCacheTier; // read from block cache only
3227
  auto s = GetImpl(roptions, column_family, key, value, value_found);
K
Kai Liu 已提交
3228 3229 3230 3231 3232

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

3235
Iterator* DBImpl::NewIterator(const ReadOptions& options,
3236
                              ColumnFamilyHandle* column_family) {
3237 3238
  SequenceNumber latest_snapshot = 0;
  SuperVersion* super_version = nullptr;
3239 3240
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
3241
  if (!options.tailing) {
I
Igor Canadi 已提交
3242
    mutex_.Lock();
3243 3244
    super_version = cfd->GetSuperVersion()->Ref();
    latest_snapshot = versions_->LastSequence();
I
Igor Canadi 已提交
3245
    mutex_.Unlock();
3246
  }
T
Tomislav Novak 已提交
3247

3248
  Iterator* iter;
T
Tomislav Novak 已提交
3249
  if (options.tailing) {
3250
    iter = new TailingIterator(this, options, cfd);
T
Tomislav Novak 已提交
3251
  } else {
3252
    iter = NewInternalIterator(options, cfd, super_version);
T
Tomislav Novak 已提交
3253

I
Igor Canadi 已提交
3254 3255 3256 3257
    auto snapshot =
        options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
            : latest_snapshot;
I
Igor Canadi 已提交
3258
    iter = NewDBIterator(&dbname_, env_, *cfd->options(),
I
Igor Canadi 已提交
3259
                         cfd->user_comparator(), iter, snapshot);
T
Tomislav Novak 已提交
3260 3261
  }

T
Tyler Harter 已提交
3262 3263 3264 3265
  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,
3266
                                    cfd->options()->prefix_extractor.get());
T
Tyler Harter 已提交
3267 3268
  }
  return iter;
J
jorlow@chromium.org 已提交
3269 3270
}

3271 3272
Status DBImpl::NewIterators(
    const ReadOptions& options,
I
Igor Canadi 已提交
3273
    const std::vector<ColumnFamilyHandle*>& column_families,
3274
    std::vector<Iterator*>* iterators) {
I
Igor Canadi 已提交
3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312

  if (options.prefix) {
    return Status::NotSupported(
        "NewIterators doesn't support ReadOptions::prefix");
  }

  iterators->clear();
  iterators->reserve(column_families.size());
  SequenceNumber latest_snapshot = 0;
  std::vector<SuperVersion*> super_versions;
  super_versions.reserve(column_families.size());

  if (!options.tailing) {
    mutex_.Lock();
    latest_snapshot = versions_->LastSequence();
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      super_versions.push_back(cfd->GetSuperVersion()->Ref());
    }
    mutex_.Unlock();
  }

  if (options.tailing) {
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      iterators->push_back(new TailingIterator(this, options, cfd));
    }
  } else {
    for (size_t i = 0; i < column_families.size(); ++i) {
      auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_families[i]);
      auto cfd = cfh->cfd();

      auto snapshot =
          options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
              : latest_snapshot;

      auto iter = NewInternalIterator(options, cfd, super_versions[i]);
I
Igor Canadi 已提交
3313
      iter = NewDBIterator(&dbname_, env_, *cfd->options(),
I
Igor Canadi 已提交
3314 3315 3316 3317 3318 3319
                           cfd->user_comparator(), iter, snapshot);
      iterators->push_back(iter);
    }
  }

  return Status::OK();
3320 3321
}

J
jorlow@chromium.org 已提交
3322 3323
const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
3324
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
3325 3326 3327 3328
}

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

// Convenience methods
3333 3334
Status DBImpl::Put(const WriteOptions& o, ColumnFamilyHandle* column_family,
                   const Slice& key, const Slice& val) {
3335
  return DB::Put(o, column_family, key, val);
J
jorlow@chromium.org 已提交
3336 3337
}

3338 3339 3340 3341
Status DBImpl::Merge(const WriteOptions& o, ColumnFamilyHandle* column_family,
                     const Slice& key, const Slice& val) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  if (!cfh->cfd()->options()->merge_operator) {
3342 3343
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
3344
    return DB::Merge(o, column_family, key, val);
3345 3346 3347
  }
}

3348
Status DBImpl::Delete(const WriteOptions& options,
3349
                      ColumnFamilyHandle* column_family, const Slice& key) {
3350
  return DB::Delete(options, column_family, key);
J
jorlow@chromium.org 已提交
3351 3352
}

3353
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
3354 3355
  StopWatchNano pre_post_process_timer(env_, false);
  StartPerfTimer(&pre_post_process_timer);
3356 3357 3358
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
3359
  w.disableWAL = options.disableWAL;
3360
  w.done = false;
3361

3362
  StopWatch sw(env_, options_.statistics.get(), DB_WRITE, false);
I
Igor Canadi 已提交
3363
  mutex_.Lock();
3364 3365 3366 3367
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
3368 3369 3370 3371 3372

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

3373
  if (w.done) {
I
Igor Canadi 已提交
3374
    mutex_.Unlock();
3375
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_OTHER, 1);
3376
    return w.status;
3377 3378
  } else {
    RecordTick(options_.statistics.get(), WRITE_DONE_BY_SELF, 1);
3379 3380
  }

3381
  Status status;
3382 3383
  autovector<ColumnFamilyData*> to_delete;
  // refcounting cfd in iteration
3384
  for (auto cfd : *versions_->GetColumnFamilySet()) {
3385
    cfd->Ref();
3386 3387
    // May temporarily unlock and wait.
    status = MakeRoomForWrite(cfd, my_batch == nullptr);
3388 3389 3390
    if (cfd->Unref()) {
      to_delete.push_back(cfd);
    }
3391 3392 3393 3394
    if (!status.ok()) {
      break;
    }
  }
3395 3396 3397
  for (auto cfd : to_delete) {
    delete cfd;
  }
D
dgrogan@chromium.org 已提交
3398
  uint64_t last_sequence = versions_->LastSequence();
3399
  Writer* last_writer = &w;
3400
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
3401 3402
    autovector<WriteBatch*> write_batch_group;
    BuildBatchGroup(&last_writer, &write_batch_group);
3403

3404 3405 3406
    // 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
3407
    // into memtables
3408
    {
3409
      mutex_.Unlock();
3410 3411 3412 3413 3414 3415 3416 3417 3418 3419
      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]);
        }
      }

3420 3421 3422 3423 3424
      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
3425 3426 3427
      RecordTick(options_.statistics.get(),
                 NUMBER_KEYS_WRITTEN, my_batch_count);
      RecordTick(options_.statistics.get(),
3428 3429
                 BYTES_WRITTEN,
                 WriteBatchInternal::ByteSize(updates));
3430 3431
      if (options.disableWAL) {
        flush_on_destroy_ = true;
3432
      }
3433 3434
      BumpPerfTime(&perf_context.write_pre_and_post_process_time,
                   &pre_post_process_timer);
3435 3436

      if (!options.disableWAL) {
3437 3438
        StopWatchNano timer(env_);
        StartPerfTimer(&timer);
3439 3440 3441 3442
        Slice log_entry = WriteBatchInternal::Contents(updates);
        status = log_->AddRecord(log_entry);
        RecordTick(options_.statistics.get(), WAL_FILE_SYNCED, 1);
        RecordTick(options_.statistics.get(), WAL_FILE_BYTES, log_entry.size());
H
heyongqiang 已提交
3443
        if (status.ok() && options.sync) {
3444
          if (options_.use_fsync) {
3445
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3446
            status = log_->file()->Fsync();
3447
          } else {
3448
            StopWatch(env_, options_.statistics.get(), WAL_FILE_SYNC_MICROS);
3449
            status = log_->file()->Sync();
3450
          }
H
heyongqiang 已提交
3451
        }
3452
        BumpPerfTime(&perf_context.write_wal_time, &timer);
3453 3454
      }
      if (status.ok()) {
3455
        StopWatchNano write_memtable_timer(env_, false);
3456

3457
        StartPerfTimer(&write_memtable_timer);
3458
        status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
3459
            updates, column_family_memtables_.get(), false, 0, this, false);
3460
        BumpPerfTime(&perf_context.write_memtable_time, &write_memtable_timer);
3461

3462
        if (!status.ok()) {
3463 3464 3465
          // Iteration failed (either in-memory writebatch corruption (very
          // bad), or the client specified invalid column family). Return
          // failure.
3466 3467 3468
          // 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.
3469
          return status;
3470
        }
I
Igor Canadi 已提交
3471 3472
        SetTickerCount(options_.statistics.get(), SEQUENCE_NUMBER,
                       last_sequence);
3473
      }
3474
      StartPerfTimer(&pre_post_process_timer);
3475
      if (updates == &tmp_batch_) tmp_batch_.Clear();
3476 3477
      mutex_.Lock();
      if (status.ok()) {
3478
        versions_->SetLastSequence(last_sequence);
3479
      }
J
jorlow@chromium.org 已提交
3480 3481
    }
  }
I
Igor Canadi 已提交
3482 3483 3484
  if (options_.paranoid_checks && !status.ok() && bg_error_.ok()) {
    bg_error_ = status; // stop compaction & fail any further writes
  }
3485

3486 3487 3488 3489 3490 3491 3492
  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
3493
    }
3494 3495
    if (ready == last_writer) break;
  }
3496

3497 3498 3499
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
3500
  }
I
Igor Canadi 已提交
3501
  mutex_.Unlock();
3502 3503
  BumpPerfTime(&perf_context.write_pre_and_post_process_time,
               &pre_post_process_timer);
J
jorlow@chromium.org 已提交
3504 3505 3506
  return status;
}

3507
// REQUIRES: Writer list must be non-empty
3508
// REQUIRES: First writer must have a non-nullptr batch
3509 3510
void DBImpl::BuildBatchGroup(Writer** last_writer,
                             autovector<WriteBatch*>* write_batch_group) {
3511 3512
  assert(!writers_.empty());
  Writer* first = writers_.front();
3513
  assert(first->batch != nullptr);
3514 3515

  size_t size = WriteBatchInternal::ByteSize(first->batch);
3516
  write_batch_group->push_back(first->batch);
3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535

  // 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 已提交
3536 3537 3538 3539 3540 3541
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

3542
    if (w->batch != nullptr) {
3543 3544 3545 3546 3547 3548
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

3549
      write_batch_group->push_back(w->batch);
3550 3551 3552 3553 3554
    }
    *last_writer = w;
  }
}

3555 3556 3557
// 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 已提交
3558 3559 3560 3561
// if n < bottom, return 0;
// if n >= top, return 1000;
// otherwise, let r = (n - bottom) /
//                    (top - bottom)
3562 3563 3564 3565
//  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 已提交
3566
uint64_t DBImpl::SlowdownAmount(int n, double bottom, double top) {
3567
  uint64_t delay;
J
Jim Paton 已提交
3568
  if (n >= top) {
3569 3570
    delay = 1000;
  }
J
Jim Paton 已提交
3571
  else if (n < bottom) {
3572 3573 3574 3575
    delay = 0;
  }
  else {
    // If we are here, we know that:
J
Jim Paton 已提交
3576
    //   level0_start_slowdown <= n < level0_slowdown
3577
    // since the previous two conditions are false.
M
Mark Callaghan 已提交
3578 3579
    double how_much =
      (double) (n - bottom) /
J
Jim Paton 已提交
3580
              (top - bottom);
M
Mark Callaghan 已提交
3581
    delay = std::max(how_much * how_much * 1000, 100.0);
3582 3583 3584 3585 3586
  }
  assert(delay <= 1000);
  return delay;
}

3587
// REQUIRES: mutex_ is held
3588
// REQUIRES: this thread is currently at the front of the writer queue
3589
Status DBImpl::MakeRoomForWrite(ColumnFamilyData* cfd, bool force) {
3590
  mutex_.AssertHeld();
3591
  assert(!writers_.empty());
3592
  bool allow_delay = !force;
J
Jim Paton 已提交
3593 3594
  bool allow_hard_rate_limit_delay = !force;
  bool allow_soft_rate_limit_delay = !force;
3595
  uint64_t rate_limit_delay_millis = 0;
3596
  Status s;
3597
  double score;
3598

3599 3600 3601 3602 3603
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
I
Igor Canadi 已提交
3604 3605
    } else if (cfd->IsDropped()) {
      break;
3606
    } else if (allow_delay && cfd->NeedSlowdownForNumLevel0Files()) {
3607 3608 3609
      // 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
3610
      // individual write by 0-1ms to reduce latency variance.  Also,
3611 3612
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
3613
      uint64_t slowdown =
3614 3615 3616
          SlowdownAmount(cfd->current()->NumLevelFiles(0),
                         cfd->options()->level0_slowdown_writes_trigger,
                         cfd->options()->level0_stop_writes_trigger);
3617
      mutex_.Unlock();
3618
      uint64_t delayed;
J
Jim Paton 已提交
3619
      {
3620
        StopWatch sw(env_, options_.statistics.get(), STALL_L0_SLOWDOWN_COUNT);
3621
        env_->SleepForMicroseconds(slowdown);
3622
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3623
      }
3624
      RecordTick(options_.statistics.get(), STALL_L0_SLOWDOWN_MICROS, delayed);
3625 3626
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_SLOWDOWN,
                                              delayed);
3627 3628
      allow_delay = false;  // Do not delay a single write more than once
      mutex_.Lock();
3629
      delayed_writes_++;
3630 3631
    } else if (!force && (cfd->mem()->ApproximateMemoryUsage() <=
                          cfd->options()->write_buffer_size)) {
3632
      // There is room in current memtable
3633 3634 3635
      if (allow_delay) {
        DelayLoggingAndReset();
      }
3636
      break;
3637 3638
    } else if (cfd->imm()->size() ==
               cfd->options()->max_write_buffer_number - 1) {
3639
      // We have filled up the current memtable, but the previous
3640
      // ones are still being flushed, so we wait.
3641
      DelayLoggingAndReset();
3642 3643
      Log(options_.info_log, "wait for memtable flush...\n");
      MaybeScheduleFlushOrCompaction();
3644
      uint64_t stall;
J
Jim Paton 已提交
3645
      {
3646
        StopWatch sw(env_, options_.statistics.get(),
3647
                     STALL_MEMTABLE_COMPACTION_COUNT);
J
Jim Paton 已提交
3648
        bg_cv_.Wait();
3649
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3650
      }
3651 3652
      RecordTick(options_.statistics.get(),
                 STALL_MEMTABLE_COMPACTION_MICROS, stall);
3653 3654
      cfd->internal_stats()->RecordWriteStall(
          InternalStats::MEMTABLE_COMPACTION, stall);
3655 3656
    } else if (cfd->current()->NumLevelFiles(0) >=
               cfd->options()->level0_stop_writes_trigger) {
3657
      // There are too many level-0 files.
3658 3659
      DelayLoggingAndReset();
      Log(options_.info_log, "wait for fewer level0 files...\n");
3660
      uint64_t stall;
J
Jim Paton 已提交
3661
      {
3662 3663
        StopWatch sw(env_, options_.statistics.get(),
                     STALL_L0_NUM_FILES_COUNT);
J
Jim Paton 已提交
3664
        bg_cv_.Wait();
3665
        stall = sw.ElapsedMicros();
J
Jim Paton 已提交
3666
      }
3667
      RecordTick(options_.statistics.get(), STALL_L0_NUM_FILES_MICROS, stall);
3668 3669
      cfd->internal_stats()->RecordWriteStall(InternalStats::LEVEL0_NUM_FILES,
                                              stall);
I
Igor Canadi 已提交
3670 3671
    } else if (allow_hard_rate_limit_delay &&
               cfd->options()->hard_rate_limit > 1.0 &&
3672 3673
               (score = cfd->current()->MaxCompactionScore()) >
                   cfd->options()->hard_rate_limit) {
3674
      // Delay a write when the compaction score for any level is too large.
3675
      int max_level = cfd->current()->MaxCompactionScoreLevel();
3676
      mutex_.Unlock();
3677
      uint64_t delayed;
J
Jim Paton 已提交
3678
      {
3679 3680
        StopWatch sw(env_, options_.statistics.get(),
                     HARD_RATE_LIMIT_DELAY_COUNT);
J
Jim Paton 已提交
3681
        env_->SleepForMicroseconds(1000);
3682
        delayed = sw.ElapsedMicros();
J
Jim Paton 已提交
3683
      }
3684
      cfd->internal_stats()->RecordLevelNSlowdown(max_level, delayed);
3685
      // Make sure the following value doesn't round to zero.
3686 3687
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
3688 3689
      RecordTick(options_.statistics.get(),
                 RATE_LIMIT_DELAY_MILLIS, rate_limit);
3690
      if (cfd->options()->rate_limit_delay_max_milliseconds > 0 &&
J
Jim Paton 已提交
3691
          rate_limit_delay_millis >=
3692
              (unsigned)cfd->options()->rate_limit_delay_max_milliseconds) {
J
Jim Paton 已提交
3693
        allow_hard_rate_limit_delay = false;
3694
      }
3695
      mutex_.Lock();
3696 3697 3698 3699
    } else if (allow_soft_rate_limit_delay &&
               cfd->options()->soft_rate_limit > 0.0 &&
               (score = cfd->current()->MaxCompactionScore()) >
                   cfd->options()->soft_rate_limit) {
J
Jim Paton 已提交
3700 3701 3702
      // Delay a write when the compaction score for any level is too large.
      // TODO: add statistics
      mutex_.Unlock();
J
Jim Paton 已提交
3703
      {
3704 3705
        StopWatch sw(env_, options_.statistics.get(),
                     SOFT_RATE_LIMIT_DELAY_COUNT);
3706 3707 3708
        env_->SleepForMicroseconds(
            SlowdownAmount(score, cfd->options()->soft_rate_limit,
                           cfd->options()->hard_rate_limit));
J
Jim Paton 已提交
3709 3710
        rate_limit_delay_millis += sw.ElapsedMicros();
      }
J
Jim Paton 已提交
3711 3712
      allow_soft_rate_limit_delay = false;
      mutex_.Lock();
3713

3714
    } else {
3715
      unique_ptr<WritableFile> lfile;
K
kailiu 已提交
3716
      MemTable* new_mem = nullptr;
3717

3718
      // Attempt to switch to a new memtable and trigger flush of old.
3719
      // Do this without holding the dbmutex lock.
3720 3721
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
I
Igor Canadi 已提交
3722
      SuperVersion* new_superversion = nullptr;
3723 3724 3725
      mutex_.Unlock();
      {
        DelayLoggingAndReset();
3726
        s = env_->NewWritableFile(LogFileName(options_.wal_dir, new_log_number),
I
Igor Canadi 已提交
3727
                                  &lfile, storage_options_.AdaptForLogWrite());
3728 3729 3730
        if (s.ok()) {
          // Our final size should be less than write_buffer_size
          // (compression, etc) but err on the side of caution.
3731 3732
          lfile->SetPreallocationBlockSize(1.1 *
                                           cfd->options()->write_buffer_size);
3733
          new_mem = new MemTable(cfd->internal_comparator(), *cfd->options());
3734
          new_superversion = new SuperVersion();
3735 3736 3737
        }
      }
      mutex_.Lock();
3738
      if (!s.ok()) {
H
heyongqiang 已提交
3739
        // Avoid chewing through file number space in a tight loop.
3740
        versions_->ReuseFileNumber(new_log_number);
3741
        assert(!new_mem);
3742 3743
        break;
      }
3744
      logfile_number_ = new_log_number;
3745
      log_.reset(new log::Writer(std::move(lfile)));
3746 3747
      cfd->mem()->SetNextLogNumber(logfile_number_);
      cfd->imm()->Add(cfd->mem());
3748
      if (force) {
3749
        cfd->imm()->FlushRequested();
3750
      }
3751
      new_mem->Ref();
3752
      alive_log_files_.push_back(logfile_number_);
3753 3754 3755 3756 3757 3758 3759 3760 3761 3762
      for (auto cfd : *versions_->GetColumnFamilySet()) {
        // all this is just optimization to delete logs that
        // are no longer needed -- if CF is empty, that means it
        // doesn't need that particular log to stay alive, so we just
        // advance the log number. no need to persist this in the manifest
        if (cfd->mem()->GetFirstSequenceNumber() == 0 &&
            cfd->imm()->size() == 0) {
          cfd->SetLogNumber(logfile_number_);
        }
      }
3763
      cfd->SetMemtable(new_mem);
I
Igor Canadi 已提交
3764 3765 3766
      Log(options_.info_log,
          "[CF %" PRIu32 "] New memtable created with log file: #%lu\n",
          cfd->GetID(), (unsigned long)logfile_number_);
3767
      force = false;  // Do not force another compaction if have room
3768
      MaybeScheduleFlushOrCompaction();
3769
      delete cfd->InstallSuperVersion(new_superversion, &mutex_);
3770 3771 3772 3773 3774
    }
  }
  return s;
}

I
Igor Canadi 已提交
3775 3776 3777 3778 3779
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

3780 3781
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
3782
  auto version = cfd->current();
3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}

I
Igor Canadi 已提交
3796 3797 3798 3799
const std::string& DBImpl::GetName() const {
  return dbname_;
}

3800 3801 3802 3803
Env* DBImpl::GetEnv() const {
  return env_;
}

3804 3805
const Options& DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
I
Igor Canadi 已提交
3806
  return *cfh->cfd()->options();
I
Igor Canadi 已提交
3807 3808
}

3809
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
3810
                         const Slice& property, std::string* value) {
3811
  value->clear();
3812 3813
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
J
jorlow@chromium.org 已提交
3814
  MutexLock l(&mutex_);
3815
  return cfd->internal_stats()->GetProperty(property, value, cfd);
J
jorlow@chromium.org 已提交
3816 3817
}

3818
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
3819
                                 const Range* range, int n, uint64_t* sizes) {
J
jorlow@chromium.org 已提交
3820 3821
  // TODO(opt): better implementation
  Version* v;
3822 3823
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
J
jorlow@chromium.org 已提交
3824 3825
  {
    MutexLock l(&mutex_);
3826
    v = cfd->current();
3827
    v->Ref();
J
jorlow@chromium.org 已提交
3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844
  }

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

3845 3846 3847 3848 3849 3850 3851
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

3852 3853 3854
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
3855 3856 3857 3858
  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());
3859 3860 3861
    return Status::InvalidArgument("Invalid file name");
  }

3862 3863 3864 3865
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
3866 3867
      Log(options_.info_log, "DeleteFile %s failed - not archived log.\n",
          name.c_str());
3868 3869 3870 3871
      return Status::NotSupported("Delete only supported for archived logs");
    }
    status = env_->DeleteFile(options_.wal_dir + "/" + name.c_str());
    if (!status.ok()) {
3872 3873
      Log(options_.info_log, "DeleteFile %s failed -- %s.\n",
          name.c_str(), status.ToString().c_str());
3874 3875 3876 3877
    }
    return status;
  }

3878
  int level;
3879
  FileMetaData *metadata;
3880
  ColumnFamilyData* cfd;
3881
  VersionEdit edit;
K
Kai Liu 已提交
3882
  DeletionState deletion_state(true);
D
Dhruba Borthakur 已提交
3883 3884
  {
    MutexLock l(&mutex_);
3885
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
3886
    if (!status.ok()) {
3887 3888
      Log(options_.info_log, "DeleteFile %s failed. File not found\n",
                             name.c_str());
D
Dhruba Borthakur 已提交
3889 3890
      return Status::InvalidArgument("File not found");
    }
I
Igor Canadi 已提交
3891
    assert((level > 0) && (level < cfd->NumberLevels()));
3892

D
Dhruba Borthakur 已提交
3893
    // If the file is being compacted no need to delete.
3894
    if (metadata->being_compacted) {
3895
      Log(options_.info_log,
3896
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
D
Dhruba Borthakur 已提交
3897
      return Status::OK();
3898 3899
    }

D
Dhruba Borthakur 已提交
3900 3901 3902
    // Only the files in the last level can be deleted externally.
    // This is to make sure that any deletion tombstones are not
    // lost. Check that the level passed is the last level.
I
Igor Canadi 已提交
3903
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
3904
      if (cfd->current()->NumLevelFiles(i) != 0) {
D
Dhruba Borthakur 已提交
3905
        Log(options_.info_log,
3906
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
D
Dhruba Borthakur 已提交
3907 3908 3909 3910
        return Status::InvalidArgument("File not in last level");
      }
    }
    edit.DeleteFile(level, number);
3911
    status = versions_->LogAndApply(cfd, &edit, &mutex_, db_directory_.get());
I
Igor Canadi 已提交
3912
    if (status.ok()) {
3913
      InstallSuperVersion(cfd, deletion_state);
I
Igor Canadi 已提交
3914
    }
I
Igor Canadi 已提交
3915
    FindObsoleteFiles(deletion_state, false);
D
Dhruba Borthakur 已提交
3916
  } // lock released here
I
Igor Canadi 已提交
3917
  LogFlush(options_.info_log);
I
Igor Canadi 已提交
3918
  // remove files outside the db-lock
3919 3920 3921
  if (deletion_state.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(deletion_state);
  }
3922 3923 3924 3925 3926 3927
  {
    MutexLock l(&mutex_);
    // schedule flush if file deletion means we freed the space for flushes to
    // continue
    MaybeScheduleFlushOrCompaction();
  }
3928 3929 3930
  return status;
}

3931
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
3932
  MutexLock l(&mutex_);
3933
  versions_->GetLiveFilesMetaData(metadata);
3934 3935
}

3936
void DBImpl::TEST_GetFilesMetaData(
3937
    ColumnFamilyHandle* column_family,
3938
    std::vector<std::vector<FileMetaData>>* metadata) {
3939 3940
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
3941 3942 3943
  MutexLock l(&mutex_);
  metadata->resize(NumberLevels());
  for (int level = 0; level < NumberLevels(); level++) {
3944
    const std::vector<FileMetaData*>& files = cfd->current()->files_[level];
3945 3946 3947 3948 3949 3950 3951 3952

    (*metadata)[level].clear();
    for (const auto& f : files) {
      (*metadata)[level].push_back(*f);
    }
  }
}

3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979
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 已提交
3980 3981
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
3982
Status DB::Put(const WriteOptions& opt, ColumnFamilyHandle* column_family,
3983
               const Slice& key, const Slice& value) {
3984 3985 3986 3987
  // 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);
3988
  batch.Put(column_family->GetID(), key, value);
J
jorlow@chromium.org 已提交
3989 3990 3991
  return Write(opt, &batch);
}

3992 3993
Status DB::Delete(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                  const Slice& key) {
J
jorlow@chromium.org 已提交
3994
  WriteBatch batch;
3995
  batch.Delete(column_family->GetID(), key);
J
jorlow@chromium.org 已提交
3996 3997 3998
  return Write(opt, &batch);
}

3999 4000
Status DB::Merge(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                 const Slice& key, const Slice& value) {
4001
  WriteBatch batch;
4002
  batch.Merge(column_family->GetID(), key, value);
4003 4004 4005
  return Write(opt, &batch);
}

4006 4007
// Default implementation -- returns not supported status
Status DB::CreateColumnFamily(const ColumnFamilyOptions& options,
4008
                              const std::string& column_family_name,
4009
                              ColumnFamilyHandle** handle) {
4010
  return Status::NotSupported("");
4011
}
4012
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
4013
  return Status::NotSupported("");
4014 4015
}

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

J
Jim Paton 已提交
4018
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
4019 4020 4021 4022 4023
  DBOptions db_options(options);
  ColumnFamilyOptions cf_options(options);
  std::vector<ColumnFamilyDescriptor> column_families;
  column_families.push_back(
      ColumnFamilyDescriptor(default_column_family_name, cf_options));
4024 4025 4026 4027 4028 4029 4030 4031 4032
  std::vector<ColumnFamilyHandle*> handles;
  Status s = DB::Open(db_options, dbname, column_families, &handles, dbptr);
  if (s.ok()) {
    assert(handles.size() == 1);
    // i can delete the handle since DBImpl is always holding a reference to
    // default column family
    delete handles[0];
  }
  return s;
4033 4034
}

4035 4036
Status DB::Open(const DBOptions& db_options, const std::string& dbname,
                const std::vector<ColumnFamilyDescriptor>& column_families,
4037
                std::vector<ColumnFamilyHandle*>* handles, DB** dbptr) {
4038
  *dbptr = nullptr;
4039
  handles->clear();
J
jorlow@chromium.org 已提交
4040

I
Igor Canadi 已提交
4041 4042 4043 4044 4045 4046 4047 4048
  size_t max_write_buffer_size = 0;
  for (auto cf : column_families) {
    max_write_buffer_size =
        std::max(max_write_buffer_size, cf.options.write_buffer_size);
    if (cf.options.block_cache != nullptr && cf.options.no_block_cache) {
      return Status::InvalidArgument(
          "no_block_cache is true while block_cache is not nullptr");
    }
4049
  }
4050

I
Igor Canadi 已提交
4051
  DBImpl* impl = new DBImpl(db_options, dbname);
4052 4053 4054 4055 4056 4057 4058
  Status s = impl->env_->CreateDirIfMissing(impl->options_.wal_dir);
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
4059 4060 4061 4062
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
4063
  impl->mutex_.Lock();
4064 4065
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
4066
  if (s.ok()) {
4067
    uint64_t new_log_number = impl->versions_->NewFileNumber();
4068
    unique_ptr<WritableFile> lfile;
4069
    EnvOptions soptions(db_options);
4070
    s = impl->options_.env->NewWritableFile(
I
Igor Canadi 已提交
4071 4072
        LogFileName(impl->options_.wal_dir, new_log_number), &lfile,
        soptions.AdaptForLogWrite());
J
jorlow@chromium.org 已提交
4073
    if (s.ok()) {
4074
      lfile->SetPreallocationBlockSize(1.1 * max_write_buffer_size);
4075
      impl->logfile_number_ = new_log_number;
4076
      impl->log_.reset(new log::Writer(std::move(lfile)));
I
Igor Canadi 已提交
4077

4078 4079
      // set column family handles
      for (auto cf : column_families) {
I
Igor Canadi 已提交
4080 4081 4082
        auto cfd =
            impl->versions_->GetColumnFamilySet()->GetColumnFamily(cf.name);
        if (cfd == nullptr) {
I
Igor Canadi 已提交
4083 4084 4085
          s = Status::InvalidArgument("Column family not found: ", cf.name);
          break;
        }
4086 4087
        handles->push_back(
            new ColumnFamilyHandleImpl(cfd, impl, &impl->mutex_));
4088
      }
I
Igor Canadi 已提交
4089 4090
    }
    if (s.ok()) {
4091
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
4092
        delete cfd->InstallSuperVersion(new SuperVersion(), &impl->mutex_);
4093
        impl->alive_log_files_.push_back(impl->logfile_number_);
4094
      }
J
jorlow@chromium.org 已提交
4095
      impl->DeleteObsoleteFiles();
4096
      impl->MaybeScheduleFlushOrCompaction();
4097
      impl->MaybeScheduleLogDBDeployStats();
4098
      s = impl->db_directory_->Fsync();
J
jorlow@chromium.org 已提交
4099 4100
    }
  }
4101

I
Igor Canadi 已提交
4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115
  if (s.ok()) {
    for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
      if (cfd->options()->compaction_style == kCompactionStyleUniversal) {
        Version* current = cfd->current();
        for (int i = 1; i < current->NumberLevels(); ++i) {
          int num_files = current->NumLevelFiles(i);
          if (num_files > 0) {
            s = Status::InvalidArgument("Not all files are at level 0. Cannot "
                "open with universal compaction style.");
            break;
          }
        }
      }
      if (!s.ok()) {
4116 4117 4118 4119 4120
        break;
      }
    }
  }

4121 4122
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4123
  if (s.ok()) {
4124
    impl->opened_successfully_ = true;
J
jorlow@chromium.org 已提交
4125 4126
    *dbptr = impl;
  } else {
4127 4128 4129
    for (auto h : *handles) {
      delete h;
    }
4130
    handles->clear();
J
jorlow@chromium.org 已提交
4131 4132 4133 4134 4135
    delete impl;
  }
  return s;
}

4136 4137 4138
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4139
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4140 4141
}

4142 4143 4144
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4145
Status DestroyDB(const std::string& dbname, const Options& options) {
4146 4147 4148 4149 4150
  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 已提交
4151
  std::vector<std::string> filenames;
4152 4153
  std::vector<std::string> archiveFiles;

4154
  std::string archivedir = ArchivalDirectory(dbname);
J
jorlow@chromium.org 已提交
4155 4156
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
4157 4158 4159 4160 4161 4162 4163

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

J
jorlow@chromium.org 已提交
4165 4166 4167 4168 4169
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
4170 4171
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4172 4173 4174
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
4175
    for (size_t i = 0; i < filenames.size(); i++) {
4176
      if (ParseFileName(filenames[i], &number, &type) &&
4177
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4178 4179 4180
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
4181 4182
        } else if (type == kLogFile) {
          del = env->DeleteFile(soptions.wal_dir + "/" + filenames[i]);
K
Kosie van der Merwe 已提交
4183 4184 4185
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
4186 4187 4188 4189 4190
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4191

4192
    env->GetChildren(archivedir, &archiveFiles);
4193 4194
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4195 4196
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4197
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4198 4199 4200 4201 4202
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4203
    // ignore case where no archival directory is present.
4204
    env->DeleteDir(archivedir);
4205

J
jorlow@chromium.org 已提交
4206
    env->UnlockFile(lock);  // Ignore error since state is already gone
4207
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4208
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4209
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4210 4211 4212 4213
  }
  return result;
}

4214 4215
//
// A global method that can dump out the build version
K
kailiu 已提交
4216
void DumpLeveldbBuildVersion(Logger * log) {
4217
  Log(log, "Git sha %s", rocksdb_build_git_sha);
4218
  Log(log, "Compile time %s %s",
4219
      rocksdb_build_compile_time, rocksdb_build_compile_date);
4220 4221
}

4222
}  // namespace rocksdb