db_impl.cc 97.0 KB
Newer Older
J
jorlow@chromium.org 已提交
1 2 3 4 5 6 7
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file. See the AUTHORS file for names of contributors.

#include "db/db_impl.h"

#include <algorithm>
8 9
#include <climits>
#include <cstdio>
J
jorlow@chromium.org 已提交
10 11 12
#include <set>
#include <string>
#include <stdint.h>
13
#include <stdexcept>
J
jorlow@chromium.org 已提交
14
#include <vector>
15
#include <unordered_set>
16

J
jorlow@chromium.org 已提交
17 18 19 20 21 22 23
#include "db/builder.h"
#include "db/db_iter.h"
#include "db/dbformat.h"
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
24
#include "db/memtablelist.h"
25
#include "db/merge_helper.h"
J
jorlow@chromium.org 已提交
26 27 28
#include "db/table_cache.h"
#include "db/version_set.h"
#include "db/write_batch_internal.h"
29
#include "db/transaction_log_iterator_impl.h"
30
#include "leveldb/compaction_filter.h"
31 32
#include "leveldb/db.h"
#include "leveldb/env.h"
33
#include "leveldb/merge_operator.h"
34
#include "leveldb/statistics.h"
35 36
#include "leveldb/status.h"
#include "leveldb/table_builder.h"
J
jorlow@chromium.org 已提交
37 38 39
#include "port/port.h"
#include "table/block.h"
#include "table/merger.h"
H
Haobo Xu 已提交
40
#include "table/table.h"
J
jorlow@chromium.org 已提交
41
#include "table/two_level_iterator.h"
42 43
#include "util/auto_roll_logger.h"
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
44 45 46
#include "util/coding.h"
#include "util/logging.h"
#include "util/mutexlock.h"
47
#include "util/stop_watch.h"
J
jorlow@chromium.org 已提交
48 49 50

namespace leveldb {

51 52
void dumpLeveldbBuildVersion(Logger * log);

53 54 55 56 57
// Information kept for every waiting writer
struct DBImpl::Writer {
  Status status;
  WriteBatch* batch;
  bool sync;
H
heyongqiang 已提交
58
  bool disableWAL;
59 60 61 62 63 64
  bool done;
  port::CondVar cv;

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

J
jorlow@chromium.org 已提交
65 66 67
struct DBImpl::CompactionState {
  Compaction* const compaction;

68 69 70 71 72
  // 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 已提交
73 74 75 76 77 78 79 80

  // Files produced by compaction
  struct Output {
    uint64_t number;
    uint64_t file_size;
    InternalKey smallest, largest;
  };
  std::vector<Output> outputs;
81
  std::list<uint64_t> allocated_file_numbers;
J
jorlow@chromium.org 已提交
82 83

  // State kept for output being generated
84 85
  unique_ptr<WritableFile> outfile;
  unique_ptr<TableBuilder> builder;
J
jorlow@chromium.org 已提交
86 87 88 89 90 91 92 93 94 95 96

  uint64_t total_bytes;

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

  explicit CompactionState(Compaction* c)
      : compaction(c),
        total_bytes(0) {
  }
};

D
Dhruba Borthakur 已提交
97 98
struct DBImpl::DeletionState {

99 100
  // the list of all live files that cannot be deleted
  std::vector<uint64_t> live;
D
Dhruba Borthakur 已提交
101 102 103 104 105 106 107 108

  // a list of all siles that exists in the db directory
  std::vector<std::string> allfiles;

  // the current filenumber, lognumber and prevlognumber
  // that corresponds to the set of files in 'live'.
  uint64_t filenumber, lognumber, prevlognumber;

109
  // the list of all files to be evicted from the table cache
D
Dhruba Borthakur 已提交
110 111 112
  std::vector<uint64_t> files_to_evict;
};

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

  result.min_write_buffer_number_to_merge = std::min(
    result.min_write_buffer_number_to_merge, result.max_write_buffer_number-1);
133
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
134 135
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
136 137
    if (!s.ok()) {
      // No place suitable for logging
138
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
139 140
    }
  }
141
  if (result.block_cache == nullptr && !result.no_block_cache) {
142 143
    result.block_cache = NewLRUCache(8 << 20);
  }
144
  result.compression_per_level = src.compression_per_level;
145 146 147
  if (result.block_size_deviation < 0 || result.block_size_deviation > 100) {
    result.block_size_deviation = 0;
  }
148 149 150
  if (result.max_mem_compaction_level >= result.num_levels) {
    result.max_mem_compaction_level = result.num_levels - 1;
  }
J
jorlow@chromium.org 已提交
151 152 153 154 155
  return result;
}

DBImpl::DBImpl(const Options& options, const std::string& dbname)
    : env_(options.env),
H
heyongqiang 已提交
156
      dbname_(dbname),
J
jorlow@chromium.org 已提交
157
      internal_comparator_(options.comparator),
S
Sanjay Ghemawat 已提交
158 159
      options_(SanitizeOptions(
          dbname, &internal_comparator_, &internal_filter_policy_, options)),
H
heyongqiang 已提交
160
      internal_filter_policy_(options.filter_policy),
J
jorlow@chromium.org 已提交
161
      owns_info_log_(options_.info_log != options.info_log),
162
      db_lock_(nullptr),
H
Haobo Xu 已提交
163
      mutex_(options.use_adaptive_mutex),
164
      shutting_down_(nullptr),
J
jorlow@chromium.org 已提交
165
      bg_cv_(&mutex_),
J
Jim Paton 已提交
166 167 168
      mem_rep_factory_(options_.memtable_factory),
      mem_(new MemTable(internal_comparator_,
        mem_rep_factory_, NumberLevels())),
169
      logfile_number_(0),
170
      tmp_batch_(),
171
      bg_compaction_scheduled_(0),
172
      bg_logstats_scheduled_(false),
173 174
      manual_compaction_(nullptr),
      logger_(nullptr),
175
      disable_delete_obsolete_files_(false),
176
      delete_obsolete_files_last_run_(0),
177
      purge_wal_files_last_run_(0),
178
      last_stats_dump_time_microsec_(0),
M
Mark Callaghan 已提交
179 180 181
      stall_level0_slowdown_(0),
      stall_memtable_compaction_(0),
      stall_level0_num_files_(0),
J
Jim Paton 已提交
182 183 184
      stall_level0_slowdown_count_(0),
      stall_memtable_compaction_count_(0),
      stall_level0_num_files_count_(0),
185
      started_at_(options.env->NowMicros()),
186
      flush_on_destroy_(false),
187
      stats_(options.num_levels),
188
      delayed_writes_(0),
189
      last_flushed_sequence_(0),
190 191 192
      storage_options_(options),
      bg_work_gate_closed_(false),
      refitting_level_(false) {
193

194
  mem_->Ref();
195

H
heyongqiang 已提交
196
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
197 198

  stall_leveln_slowdown_.resize(options.num_levels);
J
Jim Paton 已提交
199 200
  stall_leveln_slowdown_count_.resize(options.num_levels);
  for (int i = 0; i < options.num_levels; ++i) {
201
    stall_leveln_slowdown_[i] = 0;
J
Jim Paton 已提交
202 203
    stall_leveln_slowdown_count_[i] = 0;
  }
204

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

210 211
  versions_.reset(new VersionSet(dbname_, &options_, storage_options_,
                                 table_cache_.get(), &internal_comparator_));
212

213 214
  dumpLeveldbBuildVersion(options_.info_log.get());
  options_.Dump(options_.info_log.get());
215

216
#ifdef USE_SCRIBE
217
  logger_.reset(new ScribeLogger("localhost", 1456));
218 219 220
#endif

  char name[100];
221
  Status st = env_->GetHostName(name, 100L);
222 223 224 225 226 227 228
  if(st.ok()) {
    host_name_ = name;
  } else {
    Log(options_.info_log, "Can't get hostname, use localhost as host name.");
    host_name_ = "localhost";
  }
  last_log_ts = 0;
229

J
jorlow@chromium.org 已提交
230 231 232 233
}

DBImpl::~DBImpl() {
  // Wait for background work to finish
234
  if (flush_on_destroy_ && mem_->GetFirstSequenceNumber() != 0) {
235 236
    FlushMemTable(FlushOptions());
  }
237
  mutex_.Lock();
238
  shutting_down_.Release_Store(this);  // Any non-nullptr value is ok
239
  while (bg_compaction_scheduled_ || bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
240
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
241 242 243
  }
  mutex_.Unlock();

244
  if (db_lock_ != nullptr) {
J
jorlow@chromium.org 已提交
245 246 247
    env_->UnlockFile(db_lock_);
  }

248
  if (mem_ != nullptr) mem_->Unref();
249
  imm_.UnrefAll();
J
jorlow@chromium.org 已提交
250 251
}

A
Abhishek Kona 已提交
252
// Do not flush and close database elegantly. Simulate a crash.
253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
void DBImpl::TEST_Destroy_DBImpl() {
  // ensure that no new memtable flushes can occur
  flush_on_destroy_ = false;

  // wait till all background compactions are done.
  mutex_.Lock();
  while (bg_compaction_scheduled_ || bg_logstats_scheduled_) {
    bg_cv_.Wait();
  }

  // Prevent new compactions from occuring.
  const int LargeNumber = 10000000;
  bg_compaction_scheduled_ += LargeNumber;
  mutex_.Unlock();

  // force release the lock file.
269
  if (db_lock_ != nullptr) {
270 271
    env_->UnlockFile(db_lock_);
  }
272 273 274 275

  log_.reset();
  versions_.reset();
  table_cache_.reset();
276 277
}

A
Abhishek Kona 已提交
278 279 280
uint64_t DBImpl::TEST_Current_Manifest_FileNo() {
  return versions_->ManifestFileNumber();
}
281

J
jorlow@chromium.org 已提交
282
Status DBImpl::NewDB() {
283
  VersionEdit new_db(NumberLevels());
J
jorlow@chromium.org 已提交
284
  new_db.SetComparatorName(user_comparator()->Name());
285
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
286 287 288 289
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
290
  unique_ptr<WritableFile> file;
291
  Status s = env_->NewWritableFile(manifest, &file, storage_options_);
J
jorlow@chromium.org 已提交
292 293 294
  if (!s.ok()) {
    return s;
  }
295
  file->SetPreallocationBlockSize(options_.manifest_preallocation_size);
J
jorlow@chromium.org 已提交
296
  {
297
    log::Writer log(std::move(file));
J
jorlow@chromium.org 已提交
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
    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 {
315
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
316 317 318 319
    *s = Status::OK();
  }
}

320 321
const Status DBImpl::CreateArchivalDirectory() {
  if (options_.WAL_ttl_seconds > 0) {
322
    std::string archivalPath = ArchivalDirectory(dbname_);
323 324 325 326 327
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

328 329 330 331
void DBImpl::PrintStatistics() {
  auto dbstats = options_.statistics;
  if (dbstats) {
    Log(options_.info_log,
332 333
        "STATISTCS:\n %s",
        dbstats->ToString().c_str());
334 335 336
  }
}

337
void DBImpl::MaybeDumpStats() {
H
Haobo Xu 已提交
338 339 340 341 342 343 344 345 346 347 348 349 350 351 352
  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;
    GetProperty("leveldb.stats", &stats);
    Log(options_.info_log, "%s", stats.c_str());
353
    PrintStatistics();
354 355 356
  }
}

D
Dhruba Borthakur 已提交
357 358 359 360 361
// Returns the list of live files in 'live' and the list
// of all files in the filesystem in 'allfiles'.
void DBImpl::FindObsoleteFiles(DeletionState& deletion_state) {
  mutex_.AssertHeld();

362 363 364 365 366
  // if deletion is disabled, do nothing
  if (disable_delete_obsolete_files_) {
    return;
  }

367 368 369 370 371
  // This method is costly when the number of files is large.
  // Do not allow it to trigger more often than once in
  // delete_obsolete_files_period_micros.
  if (options_.delete_obsolete_files_period_micros != 0) {
    const uint64_t now_micros = env_->NowMicros();
372
    if (delete_obsolete_files_last_run_ +
373 374 375 376 377 378
        options_.delete_obsolete_files_period_micros > now_micros) {
      return;
    }
    delete_obsolete_files_last_run_ = now_micros;
  }

379 380 381 382
  // Make a list of all of the live files; set is slow, should not
  // be used.
  deletion_state.live.assign(pending_outputs_.begin(),
                             pending_outputs_.end());
D
Dhruba Borthakur 已提交
383 384 385 386
  versions_->AddLiveFiles(&deletion_state.live);

  // set of all files in the directory
  env_->GetChildren(dbname_, &deletion_state.allfiles); // Ignore errors
J
jorlow@chromium.org 已提交
387

D
Dhruba Borthakur 已提交
388 389 390 391 392 393 394 395 396 397 398
  // store the current filenum, lognum, etc
  deletion_state.filenumber = versions_->ManifestFileNumber();
  deletion_state.lognumber = versions_->LogNumber();
  deletion_state.prevlognumber = versions_->PrevLogNumber();
}

// Diffs the files listed in filenames and those that do not
// belong to live files are posibly removed. If the removed file
// is a sst file, then it returns the file number in files_to_evict.
// It is not necesary to hold the mutex when invoking this method.
void DBImpl::PurgeObsoleteFiles(DeletionState& state) {
J
jorlow@chromium.org 已提交
399 400
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
401
  std::vector<std::string> old_log_files;
402

403 404 405 406 407
  // Now, convert live list to an unordered set, WITHOUT mutex held;
  // set is slow.
  std::unordered_set<uint64_t> live_set(state.live.begin(),
                                        state.live.end());

D
Dhruba Borthakur 已提交
408 409
  for (size_t i = 0; i < state.allfiles.size(); i++) {
    if (ParseFileName(state.allfiles[i], &number, &type)) {
J
jorlow@chromium.org 已提交
410 411 412
      bool keep = true;
      switch (type) {
        case kLogFile:
D
Dhruba Borthakur 已提交
413 414
          keep = ((number >= state.lognumber) ||
                  (number == state.prevlognumber));
J
jorlow@chromium.org 已提交
415 416 417 418
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
D
Dhruba Borthakur 已提交
419
          keep = (number >= state.filenumber);
J
jorlow@chromium.org 已提交
420 421
          break;
        case kTableFile:
422
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
423 424 425 426
          break;
        case kTempFile:
          // Any temp files that are currently being written to must
          // be recorded in pending_outputs_, which is inserted into "live"
427
          keep = (live_set.find(number) != live_set.end());
J
jorlow@chromium.org 已提交
428
          break;
H
heyongqiang 已提交
429 430 431
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
D
Dhruba Borthakur 已提交
432
            old_log_files.push_back(state.allfiles[i]);
H
heyongqiang 已提交
433 434
          }
          break;
J
jorlow@chromium.org 已提交
435 436
        case kCurrentFile:
        case kDBLockFile:
K
Kosie van der Merwe 已提交
437
        case kMetaDatabase:
J
jorlow@chromium.org 已提交
438 439 440 441 442 443
          keep = true;
          break;
      }

      if (!keep) {
        if (type == kTableFile) {
D
Dhruba Borthakur 已提交
444 445
          // record the files to be evicted from the cache
          state.files_to_evict.push_back(number);
J
jorlow@chromium.org 已提交
446
        }
447
        Log(options_.info_log, "Delete type=%d #%lu", int(type), number);
448
        if (type == kLogFile && options_.WAL_ttl_seconds > 0) {
449 450 451 452 453
          Status st = env_->RenameFile(
            LogFileName(dbname_, number),
            ArchivedLogFileName(dbname_, number)
          );

454
          if (!st.ok()) {
455 456
            Log(
              options_.info_log, "RenameFile type=%d #%lu FAILED",
H
heyongqiang 已提交
457
              int(type),
458 459
              number
            );
460 461
          }
        } else {
462
          Status st = env_->DeleteFile(dbname_ + "/" + state.allfiles[i]);
463 464 465 466 467
          if(!st.ok()) {
            Log(options_.info_log, "Delete type=%d #%lld FAILED\n",
                int(type),
                static_cast<unsigned long long>(number));
          }
H
heyongqiang 已提交
468
        }
J
jorlow@chromium.org 已提交
469 470 471
      }
    }
  }
H
heyongqiang 已提交
472 473

  // Delete old log files.
K
Kai Liu 已提交
474 475 476 477 478
  size_t old_log_file_count = old_log_files.size();
  // NOTE: Currently we only support log purge when options_.db_log_dir is
  // located in `dbname` directory.
  if (old_log_file_count >= options_.keep_log_file_num &&
      options_.db_log_dir.empty()) {
H
heyongqiang 已提交
479
    std::sort(old_log_files.begin(), old_log_files.end());
K
Kai Liu 已提交
480
    size_t end = old_log_file_count - options_.keep_log_file_num;
481
    for (unsigned int i = 0; i <= end; i++) {
H
heyongqiang 已提交
482
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
483 484
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
485 486 487
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
488
  PurgeObsoleteWALFiles();
J
jorlow@chromium.org 已提交
489 490
}

D
Dhruba Borthakur 已提交
491 492 493 494 495 496 497 498 499 500 501 502
void DBImpl::EvictObsoleteFiles(DeletionState& state) {
  for (unsigned int i = 0; i < state.files_to_evict.size(); i++) {
    table_cache_->Evict(state.files_to_evict[i]);
  }
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
  DeletionState deletion_state;
  FindObsoleteFiles(deletion_state);
  PurgeObsoleteFiles(deletion_state);
  EvictObsoleteFiles(deletion_state);
503 504 505
}

void DBImpl::PurgeObsoleteWALFiles() {
506 507 508 509 510
  int64_t current_time;
  Status s = env_->GetCurrentTime(&current_time);
  uint64_t now_micros = static_cast<uint64_t>(current_time);
  assert(s.ok());

511
  if (options_.WAL_ttl_seconds != ULONG_MAX && options_.WAL_ttl_seconds > 0) {
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527
    if (purge_wal_files_last_run_ + options_.WAL_ttl_seconds > now_micros) {
      return;
    }
    std::vector<std::string> wal_files;
    std::string archival_dir = ArchivalDirectory(dbname_);
    env_->GetChildren(archival_dir, &wal_files);
    for (const auto& f : wal_files) {
      uint64_t file_m_time;
      const std::string file_path = archival_dir + "/" + f;
      const Status s = env_->GetFileModificationTime(file_path, &file_m_time);
      if (s.ok() && (now_micros - file_m_time > options_.WAL_ttl_seconds)) {
        Status status = env_->DeleteFile(file_path);
        if (!status.ok()) {
          Log(options_.info_log,
              "Failed Deleting a WAL file Error : i%s",
              status.ToString().c_str());
528
        }
529
      } // Ignore errors.
530 531
    }
  }
532
  purge_wal_files_last_run_ = now_micros;
D
Dhruba Borthakur 已提交
533 534
}

535 536 537
// If externalTable is set, then apply recovered transactions
// to that table. This is used for readonly mode.
Status DBImpl::Recover(VersionEdit* edit, MemTable* external_table,
H
heyongqiang 已提交
538
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
539 540
  mutex_.AssertHeld();

541
  assert(db_lock_ == nullptr);
542
  if (!external_table) {
543 544 545 546 547 548 549 550 551 552 553 554 555
    // We call CreateDirIfMissing() as the directory may already exist (if we
    // are reopening a DB), when this happens we don't want creating the
    // directory to cause an error. However, we need to check if creating the
    // directory fails or else we may get an obscure message about the lock
    // file not existing. One real-world example of this occurring is if
    // env->CreateDirIfMissing() doesn't create intermediate directories, e.g.
    // when dbname_ is "dir/db" but when "dir" doesn't exist.
    Status s = env_->CreateDirIfMissing(dbname_);
    if (!s.ok()) {
      return s;
    }

    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
556 557 558
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
559

560 561
    if (!env_->FileExists(CurrentFileName(dbname_))) {
      if (options_.create_if_missing) {
562
        // TODO: add merge_operator name check
563 564 565 566 567 568 569
        s = NewDB();
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
570 571
      }
    } else {
572 573 574 575
      if (options_.error_if_exists) {
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
J
jorlow@chromium.org 已提交
576 577 578
    }
  }

579
  Status s = versions_->Recover();
J
jorlow@chromium.org 已提交
580 581
  if (s.ok()) {
    SequenceNumber max_sequence(0);
582 583 584 585 586 587 588 589 590 591 592 593 594 595

    // 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
    // produced by an older version of leveldb.
    const uint64_t min_log = versions_->LogNumber();
    const uint64_t prev_log = versions_->PrevLogNumber();
    std::vector<std::string> filenames;
    s = env_->GetChildren(dbname_, &filenames);
    if (!s.ok()) {
      return s;
596
    }
597 598 599 600 601 602 603 604 605
    uint64_t number;
    FileType type;
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
      if (ParseFileName(filenames[i], &number, &type)
          && type == kLogFile
          && ((number >= min_log) || (number == prev_log))) {
        logs.push_back(number);
      }
J
jorlow@chromium.org 已提交
606
    }
607

H
heyongqiang 已提交
608 609 610 611 612 613
    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");
    }

614 615 616
    // Recover in the order in which the logs were generated
    std::sort(logs.begin(), logs.end());
    for (size_t i = 0; i < logs.size(); i++) {
617
      s = RecoverLogFile(logs[i], edit, &max_sequence, external_table);
618 619 620 621
      // The previous incarnation may not have written any MANIFEST
      // records after allocating this log number.  So we manually
      // update the file number allocation counter in VersionSet.
      versions_->MarkFileNumberUsed(logs[i]);
622 623
    }

J
jorlow@chromium.org 已提交
624
    if (s.ok()) {
625 626
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
627 628 629
        last_flushed_sequence_ = max_sequence;
      } else {
        last_flushed_sequence_ = versions_->LastSequence();
630
      }
J
jorlow@chromium.org 已提交
631 632 633 634 635 636 637 638
    }
  }

  return s;
}

Status DBImpl::RecoverLogFile(uint64_t log_number,
                              VersionEdit* edit,
639 640
                              SequenceNumber* max_sequence,
                              MemTable* external_table) {
J
jorlow@chromium.org 已提交
641 642
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
643
    Logger* info_log;
J
jorlow@chromium.org 已提交
644
    const char* fname;
645 646
    Status* status;  // nullptr if options_.paranoid_checks==false or
                     //            options_.skip_log_error_on_recovery==true
J
jorlow@chromium.org 已提交
647
    virtual void Corruption(size_t bytes, const Status& s) {
648
      Log(info_log, "%s%s: dropping %d bytes; %s",
649
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
650
          fname, static_cast<int>(bytes), s.ToString().c_str());
651
      if (this->status != nullptr && this->status->ok()) *this->status = s;
J
jorlow@chromium.org 已提交
652 653 654 655 656 657 658
    }
  };

  mutex_.AssertHeld();

  // Open the log file
  std::string fname = LogFileName(dbname_, log_number);
659
  unique_ptr<SequentialFile> file;
660
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
J
jorlow@chromium.org 已提交
661 662 663 664 665 666 667 668
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
669
  reporter.info_log = options_.info_log.get();
J
jorlow@chromium.org 已提交
670
  reporter.fname = fname.c_str();
671 672
  reporter.status = (options_.paranoid_checks &&
                     !options_.skip_log_error_on_recovery ? &status : nullptr);
J
jorlow@chromium.org 已提交
673 674 675 676
  // 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).
677
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
678
                     0/*initial_offset*/);
679
  Log(options_.info_log, "Recovering log #%llu",
J
jorlow@chromium.org 已提交
680 681 682 683 684 685
      (unsigned long long) log_number);

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
686
  MemTable* mem = nullptr;
687 688 689
  if (external_table) {
    mem = external_table;
  }
690
  while (reader.ReadRecord(&record, &scratch) && status.ok()) {
J
jorlow@chromium.org 已提交
691 692 693 694 695 696 697
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

698
    if (mem == nullptr) {
J
Jim Paton 已提交
699 700
      mem = new MemTable(internal_comparator_,
        mem_rep_factory_, NumberLevels());
701
      mem->Ref();
J
jorlow@chromium.org 已提交
702
    }
703
    status = WriteBatchInternal::InsertInto(&batch, mem, &options_);
J
jorlow@chromium.org 已提交
704 705 706 707 708 709 710 711 712 713 714
    MaybeIgnoreError(&status);
    if (!status.ok()) {
      break;
    }
    const SequenceNumber last_seq =
        WriteBatchInternal::Sequence(&batch) +
        WriteBatchInternal::Count(&batch) - 1;
    if (last_seq > *max_sequence) {
      *max_sequence = last_seq;
    }

715 716
    if (!external_table &&
        mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
717
      status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
718 719 720 721 722
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
        break;
      }
723
      mem->Unref();
724
      mem = nullptr;
J
jorlow@chromium.org 已提交
725 726 727
    }
  }

728
  if (status.ok() && mem != nullptr && !external_table) {
729
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
730 731 732 733
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

734
  if (mem != nullptr && !external_table) mem->Unref();
J
jorlow@chromium.org 已提交
735 736 737
  return status;
}

738
Status DBImpl::WriteLevel0TableForRecovery(MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
739
  mutex_.AssertHeld();
740
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
741 742 743 744
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
745 746 747
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
    mem->GetFirstSequenceNumber();
748
  Log(options_.info_log, "Level-0 table #%llu: started",
J
jorlow@chromium.org 已提交
749
      (unsigned long long) meta.number);
750 751 752 753

  Status s;
  {
    mutex_.Unlock();
754 755
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
756 757
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
758 759 760
    mutex_.Lock();
  }

761
  Log(options_.info_log, "Level-0 table #%llu: %lld bytes %s",
J
jorlow@chromium.org 已提交
762 763 764 765
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;
766

767
  pending_outputs_.erase(meta.number);
768 769 770 771 772 773 774 775 776

  // 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,
                  meta.smallest, meta.largest);
  }

777 778 779
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
780
  stats.files_out_levelnp1 = 1;
781
  stats_[level].Add(stats);
J
jorlow@chromium.org 已提交
782 783 784
  return s;
}

785

786
Status DBImpl::WriteLevel0Table(std::vector<MemTable*> &mems, VersionEdit* edit,
787
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
788
  mutex_.AssertHeld();
789 790 791 792 793
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
794 795 796 797 798 799 800

  std::vector<Iterator*> list;
  for (MemTable* m : mems) {
    list.push_back(m->NewIterator());
  }
  Iterator* iter = NewMergingIterator(&internal_comparator_, &list[0],
                                      list.size());
801 802
  const SequenceNumber newest_snapshot = snapshots_.GetNewest();
  const SequenceNumber earliest_seqno_in_memtable =
803
    mems[0]->GetFirstSequenceNumber();
804 805
  Log(options_.info_log, "Level-0 flush table #%llu: started",
      (unsigned long long) meta.number);
J
jorlow@chromium.org 已提交
806

807
  Version* base = versions_->current();
808
  base->Ref();          // it is likely that we do not need this reference
809 810 811
  Status s;
  {
    mutex_.Unlock();
812 813
    s = BuildTable(dbname_, env_, options_, storage_options_,
                   table_cache_.get(), iter, &meta,
814 815
                   user_comparator(), newest_snapshot,
                   earliest_seqno_in_memtable);
816 817
    mutex_.Lock();
  }
818 819
  base->Unref();

820 821 822 823 824 825 826 827 828 829 830 831 832 833
  Log(options_.info_log, "Level-0 flush table #%llu: %lld bytes %s",
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;

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

  // There could be multiple threads writing to its own level-0 file.
  // The pending_outputs cannot be cleared here, otherwise this newly
  // created file might not be considered as a live-file by another
  // compaction thread that is concurrently deleting obselete files.
  // The pending_outputs can be cleared only after the new version is
A
Abhishek Kona 已提交
834
  // committed so that other threads can recognize this file as a
835 836 837 838 839 840 841 842 843 844 845 846 847
  // 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.
848
    if (base != nullptr && options_.max_background_compactions <= 1) {
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
      level = base->PickLevelForMemTableOutput(min_user_key, max_user_key);
    }
    edit->AddFile(level, meta.number, meta.file_size,
                  meta.smallest, meta.largest);
  }

  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
  stats_[level].Add(stats);
  return s;
}

Status DBImpl::CompactMemTable(bool* madeProgress) {
  mutex_.AssertHeld();
  assert(imm_.size() != 0);

866
  if (!imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
867 868 869 870 871 872 873 874
    Log(options_.info_log, "Memcompaction already in progress");
    Status s = Status::IOError("Memcompaction already in progress");
    return s;
  }

  // Save the contents of the earliest memtable as a new Table
  // This will release and re-acquire the mutex.
  uint64_t file_number;
875 876 877
  std::vector<MemTable*> mems;
  imm_.PickMemtablesToFlush(&mems);
  if (mems.empty()) {
878 879 880 881 882 883
    Log(options_.info_log, "Nothing in memstore to flush");
    Status s = Status::IOError("Nothing in memstore to flush");
    return s;
  }

  // record the logfile_number_ before we release the mutex
884
  MemTable* m = mems[0];
885 886
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
887
  edit->SetLogNumber(m->GetLogNumber());  // Earlier logs no longer needed
888

889
  Status s = WriteLevel0Table(mems, edit, &file_number);
890

891
  if (s.ok() && shutting_down_.Acquire_Load()) {
892 893 894
    s = Status::IOError(
      "Database shutdown started during memtable compaction"
    );
895
  }
J
jorlow@chromium.org 已提交
896

897
  // Replace immutable memtable with the generated Table
898
  s = imm_.InstallMemtableFlushResults(
899
    mems, versions_.get(), s, &mutex_, options_.info_log.get(),
900
    file_number, pending_outputs_);
J
jorlow@chromium.org 已提交
901 902

  if (s.ok()) {
903 904 905
    if (madeProgress) {
      *madeProgress = 1;
    }
906
    MaybeScheduleLogDBDeployStats();
D
Dhruba Borthakur 已提交
907 908 909
    // we could have deleted obsolete files here, but it is not
    // absolutely necessary because it could be also done as part
    // of other background compaction
J
jorlow@chromium.org 已提交
910 911 912 913
  }
  return s;
}

914 915
void DBImpl::CompactRange(const Slice* begin, const Slice* end,
                          bool reduce_level) {
G
Gabor Cselle 已提交
916 917 918 919
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->current();
920
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
921 922 923 924 925 926 927 928 929
      if (base->OverlapInLevel(level, begin, end)) {
        max_level_with_files = level;
      }
    }
  }
  TEST_CompactMemTable(); // TODO(sanjay): Skip if memtable does not overlap
  for (int level = 0; level < max_level_with_files; level++) {
    TEST_CompactRange(level, begin, end);
  }
930 931 932 933 934 935 936 937 938 939

  if (reduce_level) {
    ReFitLevel(max_level_with_files);
  }
}

// return the same level if it cannot be moved
int DBImpl::FindMinimumEmptyLevelFitting(int level) {
  mutex_.AssertHeld();
  int minimum_level = level;
940
  for (int i = level - 1; i > 0; --i) {
941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001
    // stop if level i is not empty
    if (versions_->NumLevelFiles(i) > 0) break;

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

    minimum_level = i;
  }
  return minimum_level;
}

void DBImpl::ReFitLevel(int level) {
  assert(level < NumberLevels());

  MutexLock l(&mutex_);

  // only allow one thread refitting
  if (refitting_level_) {
    Log(options_.info_log, "ReFitLevel: another thread is refitting");
    return;
  }
  refitting_level_ = true;

  // wait for all background threads to stop
  bg_work_gate_closed_ = true;
  while (bg_compaction_scheduled_ > 0) {
    Log(options_.info_log,
        "RefitLevel: waiting for background threads to stop: %d",
        bg_compaction_scheduled_);
    bg_cv_.Wait();
  }

  // move to a smaller level
  int to_level = FindMinimumEmptyLevelFitting(level);

  assert(to_level <= level);

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

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

    auto status = versions_->LogAndApply(&edit, &mutex_);

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

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

  refitting_level_ = false;
  bg_work_gate_closed_ = false;
G
Gabor Cselle 已提交
1002 1003
}

1004
int DBImpl::NumberLevels() {
1005
  return options_.num_levels;
1006 1007 1008
}

int DBImpl::MaxMemCompactionLevel() {
1009
  return options_.max_mem_compaction_level;
1010 1011 1012
}

int DBImpl::Level0StopWriteTrigger() {
1013
  return options_.level0_stop_writes_trigger;
1014 1015
}

H
heyongqiang 已提交
1016 1017 1018 1019 1020
Status DBImpl::Flush(const FlushOptions& options) {
  Status status = FlushMemTable(options);
  return status;
}

1021 1022 1023 1024
SequenceNumber DBImpl::GetLatestSequenceNumber() {
  return versions_->LastSequence();
}

1025
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
1026
                               unique_ptr<TransactionLogIterator>* iter) {
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039

  //  Get All Log Files.
  //  Sort Files
  //  Get the first entry from each file.
  //  Do binary search and open files and find the seq number.

  std::vector<LogFile> walFiles;
  // list wal files in main db dir.
  Status s = ListAllWALFiles(dbname_, &walFiles, kAliveLogFile);
  if (!s.ok()) {
    return s;
  }
  //  list wal files in archive dir.
1040 1041 1042 1043 1044 1045
  std::string archivedir = ArchivalDirectory(dbname_);
  if (env_->FileExists(archivedir)) {
    s = ListAllWALFiles(archivedir, &walFiles, kArchivedLogFile);
    if (!s.ok()) {
      return s;
    }
1046 1047 1048 1049 1050 1051 1052
  }

  if (walFiles.empty()) {
    return Status::IOError(" NO WAL Files present in the db");
  }
  //  std::shared_ptr would have been useful here.

1053 1054 1055
  std::unique_ptr<std::vector<LogFile>> probableWALFiles(
    new std::vector<LogFile>());
  s = FindProbableWALFiles(&walFiles, probableWALFiles.get(), seq);
1056 1057
  if (!s.ok()) {
    return s;
1058
  }
1059
  iter->reset(
1060 1061
    new TransactionLogIteratorImpl(dbname_,
                                   &options_,
1062
                                   storage_options_,
1063
                                   seq,
1064
                                   std::move(probableWALFiles),
1065 1066
                                   &last_flushed_sequence_));
  iter->get()->Next();
1067
  return iter->get()->status();
1068 1069 1070 1071 1072
}

Status DBImpl::FindProbableWALFiles(std::vector<LogFile>* const allLogs,
                                    std::vector<LogFile>* const result,
                                    const SequenceNumber target) {
1073 1074
  assert(allLogs != nullptr);
  assert(result != nullptr);
1075 1076

  std::sort(allLogs->begin(), allLogs->end());
1077 1078
  long start = 0; // signed to avoid overflow when target is < first file.
  long end = static_cast<long>(allLogs->size()) - 1;
1079
  // Binary Search. avoid opening all files.
1080 1081
  while (end >= start) {
    long mid = start + (end - start) / 2;  // Avoid overflow.
1082 1083 1084
    WriteBatch batch;
    Status s = ReadFirstRecord(allLogs->at(mid), &batch);
    if (!s.ok()) {
1085 1086
      if (CheckFileExistsAndEmpty(allLogs->at(mid))) {
        allLogs->erase(allLogs->begin() + mid);
1087
        --end;
1088 1089
        continue;
      }
1090 1091 1092 1093 1094 1095
      return s;
    }
    SequenceNumber currentSeqNum = WriteBatchInternal::Sequence(&batch);
    if (currentSeqNum == target) {
      start = mid;
      end = mid;
1096
      break;
1097
    } else if (currentSeqNum < target) {
1098
      start = mid + 1;
1099
    } else {
1100
      end = mid - 1;
1101 1102
    }
  }
1103 1104
  size_t startIndex = std::max(0l, end); // end could be -ve.
  for( size_t i = startIndex; i < allLogs->size(); ++i) {
1105 1106
    result->push_back(allLogs->at(i));
  }
1107 1108 1109 1110
  if (result->empty()) {
    return Status::IOError(
        "No probable files. Check if the db contains log files");
  }
1111 1112 1113
  return Status::OK();
}

1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131
bool DBImpl::CheckFileExistsAndEmpty(const LogFile& file) {
  if (file.type == kAliveLogFile) {
    const std::string fname = LogFileName(dbname_, file.logNumber);
    uint64_t file_size;
    Status s = env_->GetFileSize(fname, &file_size);
    if (s.ok() && file_size == 0) {
      return true;
    }
  }
  const std::string fname = ArchivedLogFileName(dbname_, file.logNumber);
  uint64_t file_size;
  Status s = env_->GetFileSize(fname, &file_size);
  if (s.ok() && file_size == 0) {
    return true;
  }
  return false;
}

1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159
Status DBImpl::ReadFirstRecord(const LogFile& file, WriteBatch* const result) {

  if (file.type == kAliveLogFile) {
    std::string fname = LogFileName(dbname_, file.logNumber);
    Status status = ReadFirstLine(fname, result);
    if (!status.ok()) {
      //  check if the file got moved to archive.
      std::string archivedFile = ArchivedLogFileName(dbname_, file.logNumber);
      Status s = ReadFirstLine(archivedFile, result);
      if (!s.ok()) {
        return Status::IOError("Log File Has been deleted");
      }
    }
    return Status::OK();
  } else if (file.type == kArchivedLogFile) {
    std::string fname = ArchivedLogFileName(dbname_, file.logNumber);
    Status status = ReadFirstLine(fname, result);
    return status;
  }
  return Status::NotSupported("File Type Not Known");
}

Status DBImpl::ReadFirstLine(const std::string& fname,
                             WriteBatch* const batch) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
    Logger* info_log;
    const char* fname;
1160
    Status* status;  // nullptr if options_.paranoid_checks==false
1161 1162
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
1163
          (this->status == nullptr ? "(ignoring error) " : ""),
1164
          fname, static_cast<int>(bytes), s.ToString().c_str());
1165
      if (this->status != nullptr && this->status->ok()) *this->status = s;
1166 1167 1168
    }
  };

1169
  unique_ptr<SequentialFile> file;
1170
  Status status = env_->NewSequentialFile(fname, &file, storage_options_);
1171 1172 1173 1174 1175 1176 1177 1178

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


  LogReporter reporter;
  reporter.env = env_;
1179
  reporter.info_log = options_.info_log.get();
1180
  reporter.fname = fname.c_str();
1181
  reporter.status = (options_.paranoid_checks ? &status : nullptr);
1182
  log::Reader reader(std::move(file), &reporter, true/*checksum*/,
1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201
                     0/*initial_offset*/);
  std::string scratch;
  Slice record;
  if (reader.ReadRecord(&record, &scratch) && status.ok()) {
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      return Status::IOError("Corruption noted");
      //  TODO read record's till the first no corrupt entry?
    }
    WriteBatchInternal::SetContents(batch, record);
    return Status::OK();
  }
  return Status::IOError("Error reading from file " + fname);
}

Status DBImpl::ListAllWALFiles(const std::string& path,
                               std::vector<LogFile>* const logFiles,
                               WalFileType logType) {
1202
  assert(logFiles != nullptr);
1203 1204 1205 1206 1207
  std::vector<std::string> allFiles;
  const Status status = env_->GetChildren(path, &allFiles);
  if (!status.ok()) {
    return status;
  }
1208
  for (const auto& f : allFiles) {
1209 1210
    uint64_t number;
    FileType type;
1211
    if (ParseFileName(f, &number, &type) && type == kLogFile){
1212 1213 1214 1215 1216 1217
      logFiles->push_back(LogFile(number, logType));
    }
  }
  return status;
}

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

G
Gabor Cselle 已提交
1221 1222
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1223 1224
  ManualCompaction manual;
  manual.level = level;
G
Gabor Cselle 已提交
1225
  manual.done = false;
1226
  manual.in_progress = false;
1227 1228
  if (begin == nullptr) {
    manual.begin = nullptr;
G
Gabor Cselle 已提交
1229 1230 1231 1232
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
1233 1234
  if (end == nullptr) {
    manual.end = nullptr;
G
Gabor Cselle 已提交
1235 1236 1237 1238 1239 1240
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1241

A
Abhishek Kona 已提交
1242 1243 1244 1245
  // When a manual compaction arrives, temporarily throttle down
  // the number of background compaction threads to 1. This is
  // needed to ensure that this manual compaction can compact
  // any range of keys/files. We artificialy increase
1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258
  // bg_compaction_scheduled_ by a large number, this causes
  // the system to have a single background thread. Now,
  // this manual compaction can progress without stomping
  // on any other concurrent compactions.
  const int LargeNumber = 10000000;
  const int newvalue = options_.max_background_compactions-1;
  bg_compaction_scheduled_ += LargeNumber;
  while (bg_compaction_scheduled_ > LargeNumber) {
    Log(options_.info_log, "Manual compaction request waiting for background threads to fall below 1");
    bg_cv_.Wait();
  }
  Log(options_.info_log, "Manual compaction starting");

G
Gabor Cselle 已提交
1259
  while (!manual.done) {
1260
    while (manual_compaction_ != nullptr) {
G
Gabor Cselle 已提交
1261 1262 1263
      bg_cv_.Wait();
    }
    manual_compaction_ = &manual;
1264 1265 1266
    if (bg_compaction_scheduled_ == LargeNumber) {
      bg_compaction_scheduled_ = newvalue;
    }
G
Gabor Cselle 已提交
1267 1268 1269 1270
    MaybeScheduleCompaction();
    while (manual_compaction_ == &manual) {
      bg_cv_.Wait();
    }
H
hans@chromium.org 已提交
1271
  }
1272 1273 1274 1275 1276 1277 1278 1279 1280
  assert(!manual.in_progress);

  // wait till there are no background threads scheduled
  bg_compaction_scheduled_ += LargeNumber;
  while (bg_compaction_scheduled_ > LargeNumber + newvalue) {
    Log(options_.info_log, "Manual compaction resetting background threads");
    bg_cv_.Wait();
  }
  bg_compaction_scheduled_ = 0;
J
jorlow@chromium.org 已提交
1281 1282
}

H
heyongqiang 已提交
1283
Status DBImpl::FlushMemTable(const FlushOptions& options) {
1284 1285
  // nullptr batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), nullptr);
H
heyongqiang 已提交
1286
  if (s.ok() && options.wait) {
1287
    // Wait until the compaction completes
H
heyongqiang 已提交
1288
    s = WaitForCompactMemTable();
1289 1290
  }
  return s;
J
jorlow@chromium.org 已提交
1291 1292
}

H
heyongqiang 已提交
1293
Status DBImpl::WaitForCompactMemTable() {
1294 1295 1296
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1297
  while (imm_.size() > 0 && bg_error_.ok()) {
1298 1299
    bg_cv_.Wait();
  }
1300
  if (imm_.size() != 0) {
1301 1302 1303
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1304 1305 1306 1307 1308 1309
}

Status DBImpl::TEST_CompactMemTable() {
  return FlushMemTable(FlushOptions());
}

1310
Status DBImpl::TEST_WaitForCompactMemTable() {
1311
  return WaitForCompactMemTable();
1312 1313 1314
}

Status DBImpl::TEST_WaitForCompact() {
1315 1316 1317 1318 1319 1320
  // Wait until the compaction completes
  MutexLock l(&mutex_);
  while (bg_compaction_scheduled_ && bg_error_.ok()) {
    bg_cv_.Wait();
  }
  return bg_error_;
1321 1322
}

J
jorlow@chromium.org 已提交
1323 1324
void DBImpl::MaybeScheduleCompaction() {
  mutex_.AssertHeld();
1325 1326 1327
  if (bg_work_gate_closed_) {
    // gate closed for backgrond work
  } else if (bg_compaction_scheduled_ >= options_.max_background_compactions) {
J
jorlow@chromium.org 已提交
1328 1329 1330
    // Already scheduled
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
1331
  } else if (!imm_.IsFlushPending(options_.min_write_buffer_number_to_merge) &&
1332
             manual_compaction_ == nullptr &&
H
hans@chromium.org 已提交
1333
             !versions_->NeedsCompaction()) {
J
jorlow@chromium.org 已提交
1334 1335
    // No work to be done
  } else {
1336
    bg_compaction_scheduled_++;
J
jorlow@chromium.org 已提交
1337 1338 1339 1340 1341 1342 1343 1344
    env_->Schedule(&DBImpl::BGWork, this);
  }
}

void DBImpl::BGWork(void* db) {
  reinterpret_cast<DBImpl*>(db)->BackgroundCall();
}

1345 1346 1347 1348
void DBImpl::TEST_PurgeObsoleteteWAL() {
  PurgeObsoleteWALFiles();
}

J
jorlow@chromium.org 已提交
1349
void DBImpl::BackgroundCall() {
1350
  bool madeProgress = false;
D
Dhruba Borthakur 已提交
1351
  DeletionState deletion_state;
H
Haobo Xu 已提交
1352 1353 1354

  MaybeDumpStats();

J
jorlow@chromium.org 已提交
1355
  MutexLock l(&mutex_);
1356
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1357
  assert(bg_compaction_scheduled_);
H
hans@chromium.org 已提交
1358
  if (!shutting_down_.Acquire_Load()) {
1359
    Status s = BackgroundCompaction(&madeProgress, deletion_state);
1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371
    if (!s.ok()) {
      // Wait a little bit before retrying background compaction in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed compactions for the duration of
      // the problem.
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      Log(options_.info_log, "Waiting after background compaction error: %s",
          s.ToString().c_str());
      mutex_.Unlock();
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
    }
J
jorlow@chromium.org 已提交
1372
  }
1373

D
Dhruba Borthakur 已提交
1374 1375 1376 1377
  // delete unnecessary files if any, this is done outside the mutex
  if (!deletion_state.live.empty()) {
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
D
Dhruba Borthakur 已提交
1378
    EvictObsoleteFiles(deletion_state);
1379
    mutex_.Lock();
1380

D
Dhruba Borthakur 已提交
1381 1382
  }

1383
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1384

1385 1386
  MaybeScheduleLogDBDeployStats();

J
jorlow@chromium.org 已提交
1387
  // Previous compaction may have produced too many files in a level,
A
Abhishek Kona 已提交
1388
  // So reschedule another compaction if we made progress in the
1389 1390 1391 1392
  // last compaction.
  if (madeProgress) {
    MaybeScheduleCompaction();
  }
H
hans@chromium.org 已提交
1393
  bg_cv_.SignalAll();
1394

J
jorlow@chromium.org 已提交
1395 1396
}

A
Abhishek Kona 已提交
1397
Status DBImpl::BackgroundCompaction(bool* madeProgress,
1398
  DeletionState& deletion_state) {
1399
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1400
  mutex_.AssertHeld();
1401

1402
  while (imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
A
Abhishek Kona 已提交
1403
    Log(options_.info_log,
1404 1405 1406 1407 1408 1409
        "BackgroundCompaction doing CompactMemTable, compaction slots available %d",
        options_.max_background_compactions - bg_compaction_scheduled_);
    Status stat = CompactMemTable(madeProgress);
    if (!stat.ok()) {
      return stat;
    }
1410 1411
  }

1412
  unique_ptr<Compaction> c;
1413
  bool is_manual = (manual_compaction_ != nullptr) &&
1414
                   (manual_compaction_->in_progress == false);
G
Gabor Cselle 已提交
1415
  InternalKey manual_end;
H
hans@chromium.org 已提交
1416
  if (is_manual) {
G
Gabor Cselle 已提交
1417
    ManualCompaction* m = manual_compaction_;
1418 1419
    assert(!m->in_progress);
    m->in_progress = true; // another thread cannot pick up the same work
1420 1421
    c.reset(versions_->CompactRange(m->level, m->begin, m->end));
    if (c) {
G
Gabor Cselle 已提交
1422
      manual_end = c->input(0, c->num_input_files(0) - 1)->largest;
1423 1424
    } else {
      m->done = true;
G
Gabor Cselle 已提交
1425 1426 1427
    }
    Log(options_.info_log,
        "Manual compaction at level-%d from %s .. %s; will stop at %s\n",
H
hans@chromium.org 已提交
1428
        m->level,
G
Gabor Cselle 已提交
1429 1430 1431
        (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
        (m->end ? m->end->DebugString().c_str() : "(end)"),
        (m->done ? "(end)" : manual_end.DebugString().c_str()));
1432
  } else if (!options_.disable_auto_compactions) {
1433
    c.reset(versions_->PickCompaction());
J
jorlow@chromium.org 已提交
1434 1435 1436
  }

  Status status;
1437
  if (!c) {
H
hans@chromium.org 已提交
1438
    // Nothing to do
1439
    Log(options_.info_log, "Compaction nothing to do");
H
hans@chromium.org 已提交
1440
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
1441
    // Move file to next level
1442
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
1443 1444 1445 1446
    FileMetaData* f = c->input(0, 0);
    c->edit()->DeleteFile(c->level(), f->number);
    c->edit()->AddFile(c->level() + 1, f->number, f->file_size,
                       f->smallest, f->largest);
1447
    status = versions_->LogAndApply(c->edit(), &mutex_);
H
hans@chromium.org 已提交
1448
    VersionSet::LevelSummaryStorage tmp;
1449
    Log(options_.info_log, "Moved #%lld to level-%d %lld bytes %s: %s\n",
J
jorlow@chromium.org 已提交
1450 1451 1452
        static_cast<unsigned long long>(f->number),
        c->level() + 1,
        static_cast<unsigned long long>(f->file_size),
H
hans@chromium.org 已提交
1453 1454
        status.ToString().c_str(),
        versions_->LevelSummary(&tmp));
1455
    versions_->ReleaseCompactionFiles(c.get(), status);
1456
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1457
  } else {
1458
    MaybeScheduleCompaction(); // do more compaction work in parallel.
1459
    CompactionState* compact = new CompactionState(c.get());
J
jorlow@chromium.org 已提交
1460 1461
    status = DoCompactionWork(compact);
    CleanupCompaction(compact);
1462
    versions_->ReleaseCompactionFiles(c.get(), status);
1463
    c->ReleaseInputs();
D
Dhruba Borthakur 已提交
1464
    FindObsoleteFiles(deletion_state);
1465
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1466
  }
1467
  c.reset();
J
jorlow@chromium.org 已提交
1468 1469 1470 1471 1472 1473

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
1474
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1475 1476 1477 1478 1479
        "Compaction error: %s", status.ToString().c_str());
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
1480 1481

  if (is_manual) {
G
Gabor Cselle 已提交
1482
    ManualCompaction* m = manual_compaction_;
1483 1484 1485
    if (!status.ok()) {
      m->done = true;
    }
G
Gabor Cselle 已提交
1486 1487 1488 1489 1490 1491
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
      m->tmp_storage = manual_end;
      m->begin = &m->tmp_storage;
    }
1492
    m->in_progress = false; // not being processed anymore
1493
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
1494
  }
1495
  return status;
J
jorlow@chromium.org 已提交
1496 1497 1498 1499
}

void DBImpl::CleanupCompaction(CompactionState* compact) {
  mutex_.AssertHeld();
1500
  if (compact->builder != nullptr) {
J
jorlow@chromium.org 已提交
1501 1502
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
1503
    compact->builder.reset();
J
jorlow@chromium.org 已提交
1504
  } else {
1505
    assert(compact->outfile == nullptr);
J
jorlow@chromium.org 已提交
1506
  }
D
dgrogan@chromium.org 已提交
1507
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1508 1509 1510 1511 1512 1513
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
  }
  delete compact;
}

1514 1515 1516 1517 1518
// Allocate the file numbers for the output file. We allocate as
// many output file numbers as there are files in level+1.
// Insert them into pending_outputs so that they do not get deleted.
void DBImpl::AllocateCompactionOutputFileNumbers(CompactionState* compact) {
  mutex_.AssertHeld();
1519 1520
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
1521
  int filesNeeded = compact->compaction->num_input_files(1);
1522
  for (int i = 0; i < filesNeeded; i++) {
1523 1524 1525 1526 1527 1528 1529 1530 1531
    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();
1532
  for (const auto file_number : compact->allocated_file_numbers) {
1533 1534 1535 1536 1537
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
1538
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
1539 1540
  assert(compact != nullptr);
  assert(compact->builder == nullptr);
J
jorlow@chromium.org 已提交
1541
  uint64_t file_number;
1542 1543 1544 1545 1546 1547 1548
  // 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 已提交
1549 1550 1551 1552 1553
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
1554 1555 1556 1557 1558
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
1559 1560 1561

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

J
jorlow@chromium.org 已提交
1564
  if (s.ok()) {
1565 1566 1567 1568 1569
    // Over-estimate slightly so we don't end up just barely crossing
    // the threshold.
    compact->outfile->SetPreallocationBlockSize(
      1.1 * versions_->MaxFileSizeForLevel(compact->compaction->level() + 1));

1570 1571
    compact->builder.reset(new TableBuilder(options_, compact->outfile.get(),
                                            compact->compaction->level() + 1));
J
jorlow@chromium.org 已提交
1572 1573 1574 1575 1576 1577
  }
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
1578
  assert(compact != nullptr);
1579
  assert(compact->outfile);
1580
  assert(compact->builder != nullptr);
J
jorlow@chromium.org 已提交
1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595

  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;
1596
  compact->builder.reset();
J
jorlow@chromium.org 已提交
1597 1598

  // Finish and check for file errors
1599
  if (s.ok() && !options_.disableDataSync) {
1600
    if (options_.use_fsync) {
1601
      StopWatch sw(env_, options_.statistics, COMPACTION_OUTFILE_SYNC_MICROS);
1602 1603
      s = compact->outfile->Fsync();
    } else {
1604
      StopWatch sw(env_, options_.statistics, COMPACTION_OUTFILE_SYNC_MICROS);
1605 1606
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
1607 1608 1609 1610
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
1611
  compact->outfile.reset();
J
jorlow@chromium.org 已提交
1612 1613 1614

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
1615
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
1616
                                               storage_options_,
J
jorlow@chromium.org 已提交
1617 1618
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
1619 1620 1621
    s = iter->status();
    delete iter;
    if (s.ok()) {
1622
      Log(options_.info_log,
J
jorlow@chromium.org 已提交
1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
          "Generated table #%llu: %lld keys, %lld bytes",
          (unsigned long long) output_number,
          (unsigned long long) current_entries,
          (unsigned long long) current_bytes);
    }
  }
  return s;
}


Status DBImpl::InstallCompactionResults(CompactionState* compact) {
  mutex_.AssertHeld();
1635 1636 1637 1638 1639

  // 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.
1640
  if (!versions_->VerifyCompactionFileConsistency(compact->compaction)) {
1641 1642 1643 1644 1645 1646 1647 1648
    Log(options_.info_log,  "Compaction %d@%d + %d@%d files aborted",
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1);
    return Status::IOError("Compaction input files inconsistent");
  }

1649
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
1650 1651 1652 1653 1654 1655 1656 1657 1658
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
      compact->compaction->level() + 1,
      static_cast<long long>(compact->total_bytes));

  // Add compaction outputs
  compact->compaction->AddInputDeletions(compact->compaction->edit());
  const int level = compact->compaction->level();
D
dgrogan@chromium.org 已提交
1659
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1660 1661 1662 1663 1664
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
        level + 1,
        out.number, out.file_size, out.smallest, out.largest);
  }
1665
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
1666 1667
}

1668 1669 1670 1671 1672 1673 1674 1675
//
// 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(
1676 1677
  SequenceNumber in, std::vector<SequenceNumber>& snapshots,
  SequenceNumber* prev_snapshot) {
1678
  SequenceNumber prev __attribute__((unused)) = 0;
1679 1680 1681
  for (const auto cur : snapshots) {
    assert(prev <= cur);
    if (cur >= in) {
1682
      *prev_snapshot = prev;
1683
      return cur;
1684
    }
1685 1686
    prev = cur; // assignment
    assert(prev);
1687 1688 1689 1690 1691 1692 1693 1694
  }
  Log(options_.info_log,
      "Looking for seqid %ld but maxseqid is %ld", in,
      snapshots[snapshots.size()-1]);
  assert(0);
  return 0;
}

J
jorlow@chromium.org 已提交
1695
Status DBImpl::DoCompactionWork(CompactionState* compact) {
1696 1697
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions

A
Abhishek Kona 已提交
1698
  Log(options_.info_log,
1699
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
1700 1701 1702
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
1703
      compact->compaction->level() + 1,
1704
      compact->compaction->score(),
1705
      options_.max_background_compactions - bg_compaction_scheduled_);
1706 1707
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
1708
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
1709 1710

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
1711
  assert(compact->builder == nullptr);
1712
  assert(!compact->outfile);
1713 1714 1715

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
H
Haobo Xu 已提交
1716
  SequenceNumber latest_snapshot = 0;
1717 1718 1719 1720 1721
  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 已提交
1722
  } else {
H
Haobo Xu 已提交
1723
    latest_snapshot = compact->existing_snapshots.back();
1724 1725 1726 1727
    // 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 已提交
1728 1729
  }

1730 1731 1732 1733 1734 1735 1736 1737 1738 1739
  // Is this compaction producing files at the bottommost level?
  bool bottommost_level = true;
  for (int i = compact->compaction->level() + 2;
       i < versions_->NumberLevels(); i++) {
    if (versions_->NumLevelFiles(i) > 0) {
      bottommost_level = false;
      break;
    }
  }

1740 1741 1742
  // Allocate the output file numbers before we release the lock
  AllocateCompactionOutputFileNumbers(compact);

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

1746
  const uint64_t start_micros = env_->NowMicros();
1747
  unique_ptr<Iterator> input(versions_->MakeInputIterator(compact->compaction));
J
jorlow@chromium.org 已提交
1748 1749 1750 1751 1752
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
1753 1754
  SequenceNumber last_sequence_for_key __attribute__((unused)) =
    kMaxSequenceNumber;
1755
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
1756
  std::string compaction_filter_value;
H
Haobo Xu 已提交
1757
  std::vector<char> delete_key; // for compaction filter
1758 1759 1760
  MergeHelper merge(user_comparator(), options_.merge_operator,
                    options_.info_log.get(),
                    false /* internal key corruption is expected */);
J
jorlow@chromium.org 已提交
1761
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
1762
    // Prioritize immutable compaction work
1763
    if (imm_.imm_flush_needed.NoBarrier_Load() != nullptr) {
1764 1765
      const uint64_t imm_start = env_->NowMicros();
      mutex_.Lock();
1766
      if (imm_.IsFlushPending(options_.min_write_buffer_number_to_merge)) {
1767
        CompactMemTable();
H
hans@chromium.org 已提交
1768
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
1769 1770 1771 1772 1773
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
1774
    Slice key = input->key();
1775
    Slice value = input->value();
H
Haobo Xu 已提交
1776

1777
    if (compact->compaction->ShouldStopBefore(key) &&
1778
        compact->builder != nullptr) {
1779
      status = FinishCompactionOutputFile(compact, input.get());
1780 1781 1782 1783 1784 1785
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
1786
    bool drop = false;
1787
    bool current_entry_is_merged = false;
J
jorlow@chromium.org 已提交
1788 1789
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
1790 1791
      // 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 已提交
1792 1793 1794
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
1795
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1796 1797 1798 1799 1800 1801 1802 1803
    } else {
      if (!has_current_user_key ||
          user_comparator()->Compare(ikey.user_key,
                                     Slice(current_user_key)) != 0) {
        // First occurrence of this user key
        current_user_key.assign(ikey.user_key.data(), ikey.user_key.size());
        has_current_user_key = true;
        last_sequence_for_key = kMaxSequenceNumber;
1804
        visible_in_snapshot = kMaxSequenceNumber;
H
Haobo Xu 已提交
1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839

        // apply the compaction filter to the first occurrence of the user key
        if (options_.compaction_filter &&
            ikey.type == kTypeValue &&
            (visible_at_tip || ikey.sequence > latest_snapshot)) {
          // If the user has specified a compaction filter and the sequence
          // number is greater than any external snapshot, then invoke the
          // filter.
          // If the return value of the compaction filter is true, replace
          // the entry with a delete marker.
          bool value_changed = false;
          compaction_filter_value.clear();
          bool to_delete =
            options_.compaction_filter->Filter(compact->compaction->level(),
                                               ikey.user_key, value,
                                               &compaction_filter_value,
                                               &value_changed);
          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();
            RecordTick(options_.statistics, COMPACTION_KEY_DROP_USER);
          } else if (value_changed) {
            value = compaction_filter_value;
          }
        }

J
jorlow@chromium.org 已提交
1840 1841
      }

1842 1843 1844
      // 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.
1845 1846 1847 1848 1849 1850
      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);
1851 1852 1853 1854 1855

      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 已提交
1856
        // Hidden by an newer entry for same user key
1857
        // TODO: why not > ?
1858
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
1859
        drop = true;    // (A)
1860
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
1861
      } else if (ikey.type == kTypeDeletion &&
1862
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
1863 1864 1865 1866 1867 1868 1869 1870 1871
                 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;
1872
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_OBSOLETE);
1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885
      } 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.
        merge.MergeUntil(input.get(), prev_snapshot, bottommost_level);
        current_entry_is_merged = true;
        // get the merge result
        key = merge.key();
        ParseInternalKey(key, &ikey);
        value = merge.value();
J
jorlow@chromium.org 已提交
1886 1887 1888
      }

      last_sequence_for_key = ikey.sequence;
1889
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
1890 1891
    }
#if 0
1892
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1893
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
1894
        "%d smallest_snapshot: %d level: %d bottommost %d",
J
jorlow@chromium.org 已提交
1895
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
1896
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
1897
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
1898 1899
        (int)last_sequence_for_key, (int)earliest_snapshot,
        compact->compaction->level(), bottommost_level);
J
jorlow@chromium.org 已提交
1900 1901 1902
#endif

    if (!drop) {
1903

1904 1905 1906
      char* kptr = (char*)key.data();
      std::string kstr;

1907 1908 1909 1910
      // 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.
1911 1912
      if (bottommost_level && ikey.sequence < earliest_snapshot &&
         ikey.type != kTypeMerge) {
1913
        assert(ikey.type != kTypeDeletion);
1914 1915 1916 1917 1918
        // 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);
1919
      }
1920

1921 1922 1923
      Slice newkey(kptr, key.size());
      assert((key.clear(), 1)); // we do not need 'key' anymore

J
jorlow@chromium.org 已提交
1924
      // Open output file if necessary
1925
      if (compact->builder == nullptr) {
J
jorlow@chromium.org 已提交
1926 1927 1928 1929 1930 1931
        status = OpenCompactionOutputFile(compact);
        if (!status.ok()) {
          break;
        }
      }
      if (compact->builder->NumEntries() == 0) {
1932
        compact->current_output()->smallest.DecodeFrom(newkey);
J
jorlow@chromium.org 已提交
1933
      }
1934 1935
      compact->current_output()->largest.DecodeFrom(newkey);
      compact->builder->Add(newkey, value);
J
jorlow@chromium.org 已提交
1936 1937 1938 1939

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

1947 1948 1949 1950
    // MergeUntil has moved input to the next entry
    if (!current_entry_is_merged) {
      input->Next();
    }
J
jorlow@chromium.org 已提交
1951 1952 1953
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
1954
    status = Status::IOError("Database shutdown started during compaction");
J
jorlow@chromium.org 已提交
1955
  }
1956
  if (status.ok() && compact->builder != nullptr) {
1957
    status = FinishCompactionOutputFile(compact, input.get());
J
jorlow@chromium.org 已提交
1958 1959 1960 1961
  }
  if (status.ok()) {
    status = input->status();
  }
1962
  input.reset();
J
jorlow@chromium.org 已提交
1963

1964 1965
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
A
Abhishek Kona 已提交
1966 1967 1968
  if (options_.statistics) {
    options_.statistics->measureTime(COMPACTION_TIME, stats.micros);
  }
M
Mark Callaghan 已提交
1969 1970
  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
1971 1972

  int num_output_files = compact->outputs.size();
1973
  if (compact->builder != nullptr) {
1974 1975 1976 1977 1978
    // An error occured so ignore the last output.
    assert(num_output_files > 0);
    --num_output_files;
  }
  stats.files_out_levelnp1 = num_output_files;
M
Mark Callaghan 已提交
1979 1980 1981 1982 1983 1984 1985

  for (int i = 0; i < compact->compaction->num_input_files(0); i++)
    stats.bytes_readn += compact->compaction->input(0, i)->file_size;

  for (int i = 0; i < compact->compaction->num_input_files(1); i++)
    stats.bytes_readnp1 += compact->compaction->input(1, i)->file_size;

1986
  for (int i = 0; i < num_output_files; i++) {
1987 1988 1989
    stats.bytes_written += compact->outputs[i].file_size;
  }

J
jorlow@chromium.org 已提交
1990
  mutex_.Lock();
1991
  stats_[compact->compaction->level() + 1].Add(stats);
J
jorlow@chromium.org 已提交
1992

1993 1994 1995 1996
  // 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 已提交
1997 1998 1999
  if (status.ok()) {
    status = InstallCompactionResults(compact);
  }
2000
  VersionSet::LevelSummaryStorage tmp;
2001
  Log(options_.info_log,
M
Mark Callaghan 已提交
2002
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
2003
      "MB in(%.1f, %.1f) out(%.1f), amplify(%.1f) %s\n",
M
Mark Callaghan 已提交
2004 2005 2006 2007 2008 2009 2010 2011 2012
      versions_->LevelSummary(&tmp),
      (stats.bytes_readn + stats.bytes_readnp1 + stats.bytes_written) /
          (double) stats.micros,
      compact->compaction->level() + 1,
      stats.files_in_leveln, stats.files_in_levelnp1, stats.files_out_levelnp1,
      stats.bytes_readn / 1048576.0,
      stats.bytes_readnp1 / 1048576.0,
      stats.bytes_written / 1048576.0,
      (stats.bytes_written + stats.bytes_readnp1) /
2013 2014
          (double) stats.bytes_readn,
      status.ToString().c_str());
M
Mark Callaghan 已提交
2015

J
jorlow@chromium.org 已提交
2016 2017 2018
  return status;
}

2019 2020 2021 2022
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
2023
  std::vector<MemTable*> mem; // includes both mem_ and imm_
2024 2025 2026 2027 2028
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
  state->mu->Lock();
2029 2030 2031
  for (unsigned int i = 0; i < state->mem.size(); i++) {
    state->mem[i]->Unref();
  }
2032 2033 2034 2035
  state->version->Unref();
  state->mu->Unlock();
  delete state;
}
H
Hans Wennborg 已提交
2036
}  // namespace
2037

J
jorlow@chromium.org 已提交
2038 2039
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
2040
  IterState* cleanup = new IterState;
J
jorlow@chromium.org 已提交
2041
  mutex_.Lock();
2042
  *latest_snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2043

2044
  // Collect together all needed child iterators for mem
J
jorlow@chromium.org 已提交
2045
  std::vector<Iterator*> list;
2046
  mem_->Ref();
2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057
  list.push_back(mem_->NewIterator());
  cleanup->mem.push_back(mem_);

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

  // Collect iterators for files in L0 - Ln
2061
  versions_->current()->AddIterators(options, storage_options_, &list);
J
jorlow@chromium.org 已提交
2062 2063 2064
  Iterator* internal_iter =
      NewMergingIterator(&internal_comparator_, &list[0], list.size());
  versions_->current()->Ref();
2065 2066 2067

  cleanup->mu = &mutex_;
  cleanup->version = versions_->current();
2068
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2069 2070 2071 2072 2073 2074 2075 2076 2077 2078

  mutex_.Unlock();
  return internal_iter;
}

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

J
jorlow@chromium.org 已提交
2079
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
2080 2081 2082 2083
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
2084 2085 2086
Status DBImpl::Get(const ReadOptions& options,
                   const Slice& key,
                   std::string* value) {
2087 2088 2089
  return GetImpl(options, key, value);
}

2090
// If no_io is true, then returns Status::NotFound if key is not in memtable,
2091
// immutable-memtable and bloom-filters can guarantee that key is not in db,
2092
// "value" is garbage string if no_io is true
2093 2094 2095
Status DBImpl::GetImpl(const ReadOptions& options,
                       const Slice& key,
                       std::string* value,
2096
                       const bool no_io) {
2097
  Status s;
2098

2099
  StopWatch sw(env_, options_.statistics, DB_GET);
2100
  SequenceNumber snapshot;
2101
  MutexLock l(&mutex_);
2102
  if (options.snapshot != nullptr) {
2103 2104 2105
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2106
  }
2107

2108
  MemTable* mem = mem_;
2109
  MemTableList imm = imm_;
2110
  Version* current = versions_->current();
2111
  mem->Ref();
2112
  imm.RefAll();
2113
  current->Ref();
2114

2115 2116 2117
  // Unlock while reading from files and memtables

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

2121
  // First look in the memtable, then in the immutable memtable (if any).
2122 2123
  // s is both in/out. When in, s could either be OK or MergeInProgress.
  // value will contain the current merge operand in the latter case.
2124
  LookupKey lkey(key, snapshot);
2125
  if (mem->Get(lkey, value, &s, options_, no_io)) {
2126
    // Done
2127
  } else if (imm.Get(lkey, value, &s, options_, no_io)) {
2128 2129
    // Done
  } else {
2130
    current->Get(options, lkey, value, &s, &stats, options_, no_io);
2131
    have_stat_update = true;
2132
  }
2133
  mutex_.Lock();
2134

2135 2136
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
2137 2138
    MaybeScheduleCompaction();
  }
2139
  mem->Unref();
2140
  imm.UnrefAll();
2141
  current->Unref();
2142
  RecordTick(options_.statistics, NUMBER_KEYS_READ);
2143
  RecordTick(options_.statistics, BYTES_READ, value->size());
2144
  return s;
J
jorlow@chromium.org 已提交
2145 2146
}

2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219
std::vector<Status> DBImpl::MultiGet(const ReadOptions& options,
                                     const std::vector<Slice>& keys,
                                     std::vector<std::string>* values) {

  StopWatch sw(env_, options_.statistics, DB_MULTIGET);
  SequenceNumber snapshot;
  MutexLock l(&mutex_);
  if (options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
  }

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

  // Unlock while reading from files and memtables

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

  // Note: this always resizes the values array
  int numKeys = keys.size();
  std::vector<Status> statList(numKeys);
  values->resize(numKeys);

  // Keep track of bytes that we read for statistics-recording later
  uint64_t bytesRead = 0;

  // For each of the given keys, apply the entire "get" process as follows:
  // First look in the memtable, then in the immutable memtable (if any).
  // s is both in/out. When in, s could either be OK or MergeInProgress.
  // value will contain the current merge operand in the latter case.
  for(int i=0; i<numKeys; ++i) {
    Status& s = statList[i];
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
    if (mem->Get(lkey, value, &s, options_)) {
      // Done
    } else if (imm.Get(lkey, value, &s, options_)) {
      // Done
    } else {
      current->Get(options, lkey, value, &s, &stats, options_);
      have_stat_update = true;
    }

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

  // Post processing (decrement reference counts and record statistics)
  mutex_.Lock();
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
    MaybeScheduleCompaction();
  }
  mem->Unref();
  imm.UnrefAll();
  current->Unref();
  RecordTick(options_.statistics, NUMBER_MULTIGET_CALLS);
  RecordTick(options_.statistics, NUMBER_MULTIGET_KEYS_READ, numKeys);
  RecordTick(options_.statistics, NUMBER_MULTIGET_BYTES_READ, bytesRead);

  return statList;
}

2220
bool DBImpl::KeyMayExist(const Slice& key) {
2221 2222 2223 2224 2225
  return KeyMayExistImpl(key, versions_->LastSequence());
}

bool DBImpl::KeyMayExistImpl(const Slice& key,
                             const SequenceNumber read_from_seq) {
2226
  std::string value;
2227 2228 2229 2230 2231
  SnapshotImpl read_from_snapshot;
  read_from_snapshot.number_ = read_from_seq;
  ReadOptions ropts;
  ropts.snapshot = &read_from_snapshot;
  const Status s = GetImpl(ropts, key, &value, true);
2232 2233 2234
  return !s.IsNotFound();
}

J
jorlow@chromium.org 已提交
2235 2236 2237
Iterator* DBImpl::NewIterator(const ReadOptions& options) {
  SequenceNumber latest_snapshot;
  Iterator* internal_iter = NewInternalIterator(options, &latest_snapshot);
2238
  return NewDBIterator(
2239
      &dbname_, env_, options_, user_comparator(), internal_iter,
2240
      (options.snapshot != nullptr
2241 2242
       ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
       : latest_snapshot));
J
jorlow@chromium.org 已提交
2243 2244 2245 2246
}

const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
2247
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
2248 2249 2250 2251
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
2252
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
2253 2254 2255 2256 2257 2258 2259
}

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

2260 2261 2262 2263 2264 2265 2266 2267 2268
Status DBImpl::Merge(const WriteOptions& o, const Slice& key,
                     const Slice& val) {
  if (!options_.merge_operator) {
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
    return DB::Merge(o, key, val);
  }
}

J
jorlow@chromium.org 已提交
2269 2270 2271 2272
Status DBImpl::Delete(const WriteOptions& options, const Slice& key) {
  return DB::Delete(options, key);
}

2273 2274 2275 2276
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
2277
  w.disableWAL = options.disableWAL;
2278
  w.done = false;
2279

2280
  StopWatch sw(env_, options_.statistics, DB_WRITE);
2281
  MutexLock l(&mutex_);
2282 2283 2284 2285 2286 2287
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
  if (w.done) {
    return w.status;
2288 2289 2290
  }

  // May temporarily unlock and wait.
2291
  Status status = MakeRoomForWrite(my_batch == nullptr);
D
dgrogan@chromium.org 已提交
2292
  uint64_t last_sequence = versions_->LastSequence();
2293
  Writer* last_writer = &w;
2294
  if (status.ok() && my_batch != nullptr) {  // nullptr batch is for compactions
2295
    WriteBatch* updates = BuildBatchGroup(&last_writer);
2296 2297
    const SequenceNumber current_sequence = last_sequence + 1;
    WriteBatchInternal::SetSequence(updates, current_sequence);
2298 2299 2300 2301
    int my_batch_count = WriteBatchInternal::Count(updates);
    last_sequence += my_batch_count;
    // Record statistics
    RecordTick(options_.statistics, NUMBER_KEYS_WRITTEN, my_batch_count);
2302 2303 2304
    RecordTick(options_.statistics,
               BYTES_WRITTEN,
               WriteBatchInternal::ByteSize(updates));
2305 2306 2307 2308
    // Add to log and apply to memtable.  We can release the lock
    // during this phase since &w is currently responsible for logging
    // and protects against concurrent loggers and concurrent writes
    // into mem_.
2309
    {
2310
      mutex_.Unlock();
2311 2312
      if (options.disableWAL) {
        flush_on_destroy_ = true;
2313 2314 2315
      }

      if (!options.disableWAL) {
H
heyongqiang 已提交
2316 2317
        status = log_->AddRecord(WriteBatchInternal::Contents(updates));
        if (status.ok() && options.sync) {
2318
          if (options_.use_fsync) {
2319
            StopWatch(env_, options_.statistics, WAL_FILE_SYNC_MICROS);
2320
            status = log_->file()->Fsync();
2321
          } else {
2322
            StopWatch(env_, options_.statistics, WAL_FILE_SYNC_MICROS);
2323
            status = log_->file()->Sync();
2324
          }
H
heyongqiang 已提交
2325
        }
2326 2327
      }
      if (status.ok()) {
2328 2329
        status = WriteBatchInternal::InsertInto(updates, mem_, &options_, this,
                                                options_.filter_deletes);
2330 2331 2332 2333 2334 2335 2336 2337 2338
        if (!status.ok()) {
          // Panic for in-memory corruptions
          // Note that existing logic was not sound. Any partial failure writing
          // into the memtable would result in a state that some write ops might
          // have succeeded in memtable but Status reports error for all writes.
          throw std::runtime_error("In memory WriteBatch corruption!");
        }
        versions_->SetLastSequence(last_sequence);
        last_flushed_sequence_ = current_sequence;
2339
      }
2340
      mutex_.Lock();
J
jorlow@chromium.org 已提交
2341
    }
2342
    if (updates == &tmp_batch_) tmp_batch_.Clear();
J
jorlow@chromium.org 已提交
2343
  }
2344

2345 2346 2347 2348 2349 2350 2351
  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
2352
    }
2353 2354
    if (ready == last_writer) break;
  }
2355

2356 2357 2358
  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
2359
  }
J
jorlow@chromium.org 已提交
2360 2361 2362
  return status;
}

2363
// REQUIRES: Writer list must be non-empty
2364
// REQUIRES: First writer must have a non-nullptr batch
2365 2366 2367 2368
WriteBatch* DBImpl::BuildBatchGroup(Writer** last_writer) {
  assert(!writers_.empty());
  Writer* first = writers_.front();
  WriteBatch* result = first->batch;
2369
  assert(result != nullptr);
2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390

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

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

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

H
heyongqiang 已提交
2391 2392 2393 2394 2395 2396
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

2397
    if (w->batch != nullptr) {
2398 2399 2400 2401 2402 2403 2404 2405 2406
      size += WriteBatchInternal::ByteSize(w->batch);
      if (size > max_size) {
        // Do not make batch too big
        break;
      }

      // Append to *reuslt
      if (result == first->batch) {
        // Switch to temporary batch instead of disturbing caller's batch
2407
        result = &tmp_batch_;
2408 2409 2410 2411 2412 2413 2414 2415 2416 2417
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

2418
// REQUIRES: mutex_ is held
2419
// REQUIRES: this thread is currently at the front of the writer queue
2420 2421
Status DBImpl::MakeRoomForWrite(bool force) {
  mutex_.AssertHeld();
2422
  assert(!writers_.empty());
2423
  bool allow_delay = !force;
2424 2425
  bool allow_rate_limit_delay = !force;
  uint64_t rate_limit_delay_millis = 0;
2426
  Status s;
2427
  double score;
2428

2429 2430 2431 2432 2433
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
2434 2435
    } else if (
        allow_delay &&
2436
        versions_->NumLevelFiles(0) >=
2437
          options_.level0_slowdown_writes_trigger) {
2438 2439 2440 2441 2442 2443 2444
      // 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
      // individual write by 1ms to reduce latency variance.  Also,
      // this delay hands over some CPU to the compaction thread in
      // case it is sharing the same core as the writer.
      mutex_.Unlock();
2445
      uint64_t t1 = env_->NowMicros();
J
Jim Paton 已提交
2446 2447 2448 2449
      {
        StopWatch sw(env_, options_.statistics, STALL_L0_SLOWDOWN_COUNT);
        env_->SleepForMicroseconds(1000);
      }
2450
      uint64_t delayed = env_->NowMicros() - t1;
2451
      RecordTick(options_.statistics, STALL_L0_SLOWDOWN_MICROS, delayed);
2452
      stall_level0_slowdown_ += delayed;
J
Jim Paton 已提交
2453
      stall_level0_slowdown_count_++;
2454
      allow_delay = false;  // Do not delay a single write more than once
2455 2456
      //Log(options_.info_log,
      //    "delaying write %llu usecs for level0_slowdown_writes_trigger\n",
2457
      //     (long long unsigned int)delayed);
2458
      mutex_.Lock();
2459
      delayed_writes_++;
2460 2461 2462
    } else if (!force &&
               (mem_->ApproximateMemoryUsage() <= options_.write_buffer_size)) {
      // There is room in current memtable
2463 2464 2465
      if (allow_delay) {
        DelayLoggingAndReset();
      }
2466
      break;
2467
    } else if (imm_.size() == options_.max_write_buffer_number - 1) {
2468
      // We have filled up the current memtable, but the previous
2469 2470
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
2471
      Log(options_.info_log, "wait for memtable compaction...\n");
M
Mark Callaghan 已提交
2472
      uint64_t t1 = env_->NowMicros();
J
Jim Paton 已提交
2473 2474 2475 2476 2477
      {
        StopWatch sw(env_, options_.statistics,
          STALL_MEMTABLE_COMPACTION_COUNT);
        bg_cv_.Wait();
      }
2478 2479 2480
      const uint64_t stall = env_->NowMicros() -t1;
      RecordTick(options_.statistics, STALL_MEMTABLE_COMPACTION_MICROS, stall);
      stall_memtable_compaction_ += stall;
J
Jim Paton 已提交
2481
      stall_memtable_compaction_count_++;
2482
    } else if (versions_->NumLevelFiles(0) >=
2483
               options_.level0_stop_writes_trigger) {
2484
      // There are too many level-0 files.
2485
      DelayLoggingAndReset();
M
Mark Callaghan 已提交
2486
      uint64_t t1 = env_->NowMicros();
2487
      Log(options_.info_log, "wait for fewer level0 files...\n");
J
Jim Paton 已提交
2488 2489 2490 2491
      {
        StopWatch sw(env_, options_.statistics, STALL_L0_NUM_FILES_COUNT);
        bg_cv_.Wait();
      }
2492 2493 2494
      const uint64_t stall = env_->NowMicros() - t1;
      RecordTick(options_.statistics, STALL_L0_NUM_FILES_MICROS, stall);
      stall_level0_num_files_ += stall;
J
Jim Paton 已提交
2495
      stall_level0_num_files_count_++;
2496
    } else if (
2497
        allow_rate_limit_delay &&
2498 2499 2500
        options_.rate_limit > 1.0 &&
        (score = versions_->MaxCompactionScore()) > options_.rate_limit) {
      // Delay a write when the compaction score for any level is too large.
2501
      int max_level = versions_->MaxCompactionScoreLevel();
2502
      mutex_.Unlock();
2503
      uint64_t t1 = env_->NowMicros();
J
Jim Paton 已提交
2504 2505 2506 2507
      {
        StopWatch sw(env_, options_.statistics, RATE_LIMIT_DELAY_COUNT);
        env_->SleepForMicroseconds(1000);
      }
2508
      uint64_t delayed = env_->NowMicros() - t1;
2509
      stall_leveln_slowdown_[max_level] += delayed;
J
Jim Paton 已提交
2510
      stall_leveln_slowdown_count_[max_level]++;
2511
      // Make sure the following value doesn't round to zero.
2512 2513 2514
      uint64_t rate_limit = std::max((delayed / 1000), (uint64_t) 1);
      rate_limit_delay_millis += rate_limit;
      RecordTick(options_.statistics, RATE_LIMIT_DELAY_MILLIS, rate_limit);
2515 2516
      if (rate_limit_delay_millis >=
          (unsigned)options_.rate_limit_delay_milliseconds) {
2517 2518 2519 2520 2521
        allow_rate_limit_delay = false;
      }
      // Log(options_.info_log,
      //    "delaying write %llu usecs for rate limits with max score %.2f\n",
      //    (long long unsigned int)delayed, score);
2522
      mutex_.Lock();
2523 2524
    } else {
      // Attempt to switch to a new memtable and trigger compaction of old
2525
      DelayLoggingAndReset();
2526 2527
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
2528
      unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
2529 2530
      EnvOptions soptions(storage_options_);
      soptions.use_mmap_writes = false;
2531 2532 2533 2534 2535
      s = env_->NewWritableFile(
            LogFileName(dbname_, new_log_number),
            &lfile,
            soptions
          );
2536
      if (!s.ok()) {
H
heyongqiang 已提交
2537
        // Avoid chewing through file number space in a tight loop.
2538
        versions_->ReuseFileNumber(new_log_number);
2539 2540
        break;
      }
2541 2542 2543
      // Our final size should be less than write_buffer_size
      // (compression, etc) but err on the side of caution.
      lfile->SetPreallocationBlockSize(1.1 * options_.write_buffer_size);
2544
      logfile_number_ = new_log_number;
2545
      log_.reset(new log::Writer(std::move(lfile)));
2546
      mem_->SetLogNumber(logfile_number_);
2547
      imm_.Add(mem_);
J
Jim Paton 已提交
2548 2549
      mem_ = new MemTable(internal_comparator_,
        mem_rep_factory_, NumberLevels());
2550
      mem_->Ref();
2551 2552 2553 2554 2555 2556 2557 2558 2559 2560
      force = false;   // Do not force another compaction if have room
      MaybeScheduleCompaction();
    }
  }
  return s;
}

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

J
jorlow@chromium.org 已提交
2561 2562 2563 2564 2565 2566 2567 2568 2569 2570
  MutexLock l(&mutex_);
  Slice in = property;
  Slice prefix("leveldb.");
  if (!in.starts_with(prefix)) return false;
  in.remove_prefix(prefix.size());

  if (in.starts_with("num-files-at-level")) {
    in.remove_prefix(strlen("num-files-at-level"));
    uint64_t level;
    bool ok = ConsumeDecimalNumber(&in, &level) && in.empty();
2571
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
2572 2573
      return false;
    } else {
2574
      char buf[100];
D
dgrogan@chromium.org 已提交
2575 2576
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
2577
      *value = buf;
J
jorlow@chromium.org 已提交
2578 2579
      return true;
    }
2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596
  } else if (in == "levelstats") {
    char buf[1000];
    snprintf(buf, sizeof(buf),
             "Level Files Size(MB)\n"
             "--------------------\n");
    value->append(buf);

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

2597
  } else if (in == "stats") {
M
Mark Callaghan 已提交
2598
    char buf[1000];
2599 2600
    uint64_t total_bytes_written = 0;
    uint64_t total_bytes_read = 0;
M
Mark Callaghan 已提交
2601
    uint64_t micros_up = env_->NowMicros() - started_at_;
2602 2603
    // Add "+1" to make sure seconds_up is > 0 and avoid NaN later
    double seconds_up = (micros_up + 1) / 1000000.0;
2604
    uint64_t total_slowdown = 0;
J
Jim Paton 已提交
2605
    uint64_t total_slowdown_count = 0;
2606 2607 2608 2609
    uint64_t interval_bytes_written = 0;
    uint64_t interval_bytes_read = 0;
    uint64_t interval_bytes_new = 0;
    double   interval_seconds_up = 0;
M
Mark Callaghan 已提交
2610 2611

    // Pardon the long line but I think it is easier to read this way.
2612 2613
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
J
Jim Paton 已提交
2614 2615
             "Level  Files Size(MB) Score Time(sec)  Read(MB) Write(MB)    Rn(MB)  Rnp1(MB)  Wnew(MB) Amplify Read(MB/s) Write(MB/s)      Rn     Rnp1     Wnp1     NewW    Count  Ln-stall Stall-cnt\n"
             "--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
2616 2617
             );
    value->append(buf);
2618
    for (int level = 0; level < NumberLevels(); level++) {
2619 2620
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
2621 2622 2623 2624 2625 2626 2627 2628 2629
        int64_t bytes_read = stats_[level].bytes_readn +
                             stats_[level].bytes_readnp1;
        int64_t bytes_new = stats_[level].bytes_written -
                            stats_[level].bytes_readnp1;
        double amplify = (stats_[level].bytes_readn == 0)
            ? 0.0
            : (stats_[level].bytes_written + stats_[level].bytes_readnp1) /
                (double) stats_[level].bytes_readn;

2630 2631 2632
        total_bytes_read += bytes_read;
        total_bytes_written += stats_[level].bytes_written;

2633 2634
        snprintf(
            buf, sizeof(buf),
J
Jim Paton 已提交
2635
            "%3d %8d %8.0f %5.1f %9.0f %9.0f %9.0f %9.0f %9.0f %9.0f %7.1f %9.1f %11.1f %8d %8d %8d %8d %8d %9.1f %9lu\n",
2636 2637 2638
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
2639 2640
            versions_->NumLevelBytes(level) /
            versions_->MaxBytesForLevel(level),
2641
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
2642 2643 2644 2645 2646 2647
            bytes_read / 1048576.0,
            stats_[level].bytes_written / 1048576.0,
            stats_[level].bytes_readn / 1048576.0,
            stats_[level].bytes_readnp1 / 1048576.0,
            bytes_new / 1048576.0,
            amplify,
2648 2649
            // +1 to avoid division by 0
            (bytes_read / 1048576.0) / ((stats_[level].micros+1) / 1000000.0),
2650
            (stats_[level].bytes_written / 1048576.0) /
2651
                ((stats_[level].micros+1) / 1000000.0),
M
Mark Callaghan 已提交
2652 2653 2654 2655
            stats_[level].files_in_leveln,
            stats_[level].files_in_levelnp1,
            stats_[level].files_out_levelnp1,
            stats_[level].files_out_levelnp1 - stats_[level].files_in_levelnp1,
2656
            stats_[level].count,
J
Jim Paton 已提交
2657 2658
            stall_leveln_slowdown_[level] / 1000000.0,
            (unsigned long) stall_leveln_slowdown_count_[level]);
2659
        total_slowdown += stall_leveln_slowdown_[level];
J
Jim Paton 已提交
2660
        total_slowdown_count += stall_leveln_slowdown_count_[level];
2661 2662 2663
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
2664

2665 2666 2667 2668 2669 2670 2671 2672 2673
    interval_bytes_new = stats_[0].bytes_written - last_stats_.bytes_new_;
    interval_bytes_read = total_bytes_read - last_stats_.bytes_read_;
    interval_bytes_written = total_bytes_written - last_stats_.bytes_written_;
    interval_seconds_up = seconds_up - last_stats_.seconds_up_;

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

M
Mark Callaghan 已提交
2674
    snprintf(buf, sizeof(buf),
2675 2676
             "Compaction IO cumulative (GB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
M
Mark Callaghan 已提交
2677
             stats_[0].bytes_written / (1048576.0 * 1024),
2678 2679 2680 2681 2682 2683 2684 2685
             total_bytes_read / (1048576.0 * 1024),
             total_bytes_written / (1048576.0 * 1024),
             (total_bytes_read + total_bytes_written) / (1048576.0 * 1024));
    value->append(buf);

    snprintf(buf, sizeof(buf),
             "Compaction IO cumulative (MB/sec): "
             "%.1f new, %.1f read, %.1f write, %.1f read+write\n",
M
Mark Callaghan 已提交
2686
             stats_[0].bytes_written / 1048576.0 / seconds_up,
2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697
             total_bytes_read / 1048576.0 / seconds_up,
             total_bytes_written / 1048576.0 / seconds_up,
             (total_bytes_read + total_bytes_written) / 1048576.0 / seconds_up);
    value->append(buf);

    // +1 to avoid divide by 0 and NaN
    snprintf(buf, sizeof(buf),
             "Amplification cumulative: %.1f write, %.1f compaction\n",
             (double) total_bytes_written / (stats_[0].bytes_written+1),
             (double) (total_bytes_written + total_bytes_read)
                  / (stats_[0].bytes_written+1));
M
Mark Callaghan 已提交
2698 2699
    value->append(buf);

2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724
    snprintf(buf, sizeof(buf),
             "Compaction IO interval (MB): "
             "%.2f new, %.2f read, %.2f write, %.2f read+write\n",
             interval_bytes_new / 1048576.0,
             interval_bytes_read/ 1048576.0,
             interval_bytes_written / 1048576.0,
             (interval_bytes_read + interval_bytes_written) / 1048576.0);
    value->append(buf);

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

    // +1 to avoid divide by 0 and NaN
    snprintf(buf, sizeof(buf),
             "Amplification interval: %.1f write, %.1f compaction\n",
             (double) interval_bytes_written / (interval_bytes_new+1),
             (double) (interval_bytes_written + interval_bytes_read) /
                  (interval_bytes_new+1));
M
Mark Callaghan 已提交
2725 2726 2727 2728
    value->append(buf);

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
2729
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
2730 2731
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
2732
            stall_memtable_compaction_ / 1000000.0,
2733
            total_slowdown / 1000000.0);
M
Mark Callaghan 已提交
2734 2735
    value->append(buf);

J
Jim Paton 已提交
2736 2737 2738 2739 2740 2741 2742 2743 2744
    snprintf(buf, sizeof(buf),
            "Stalls(count): %lu level0_slowdown, %lu level0_numfiles, "
            "%lu memtable_compaction, %lu leveln_slowdown\n",
            (unsigned long) stall_level0_slowdown_count_,
            (unsigned long) stall_level0_num_files_count_,
            (unsigned long) stall_memtable_compaction_count_,
            (unsigned long) total_slowdown_count);
    value->append(buf);

2745 2746 2747 2748 2749
    last_stats_.bytes_read_ = total_bytes_read;
    last_stats_.bytes_written_ = total_bytes_written;
    last_stats_.bytes_new_ = stats_[0].bytes_written;
    last_stats_.seconds_up_ = seconds_up;

2750
    return true;
G
Gabor Cselle 已提交
2751 2752 2753
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
J
jorlow@chromium.org 已提交
2754
  }
2755

J
jorlow@chromium.org 已提交
2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784
  return false;
}

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

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

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

2785 2786 2787 2788 2789 2790 2791
inline void DBImpl::DelayLoggingAndReset() {
  if (delayed_writes_ > 0) {
    Log(options_.info_log, "delayed %d write...\n", delayed_writes_ );
    delayed_writes_ = 0;
  }
}

J
jorlow@chromium.org 已提交
2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
Status DB::Put(const WriteOptions& opt, const Slice& key, const Slice& value) {
  WriteBatch batch;
  batch.Put(key, value);
  return Write(opt, &batch);
}

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

2806 2807 2808 2809 2810 2811 2812
Status DB::Merge(const WriteOptions& opt, const Slice& key,
                 const Slice& value) {
  WriteBatch batch;
  batch.Merge(key, value);
  return Write(opt, &batch);
}

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

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

2819
  if (options.block_cache != nullptr && options.no_block_cache) {
2820
    return Status::InvalidArgument(
2821
        "no_block_cache is true while block_cache is not nullptr");
2822
  }
J
jorlow@chromium.org 已提交
2823
  DBImpl* impl = new DBImpl(options, dbname);
2824 2825 2826 2827 2828
  Status s = impl->CreateArchivalDirectory();
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
2829
  impl->mutex_.Lock();
2830
  VersionEdit edit(impl->NumberLevels());
2831
  s = impl->Recover(&edit); // Handles create_if_missing, error_if_exists
J
jorlow@chromium.org 已提交
2832
  if (s.ok()) {
2833
    uint64_t new_log_number = impl->versions_->NewFileNumber();
2834
    unique_ptr<WritableFile> lfile;
H
Haobo Xu 已提交
2835
    soptions.use_mmap_writes = false;
2836
    s = options.env->NewWritableFile(LogFileName(dbname, new_log_number),
2837
                                     &lfile, soptions);
J
jorlow@chromium.org 已提交
2838
    if (s.ok()) {
2839
      lfile->SetPreallocationBlockSize(1.1 * options.write_buffer_size);
2840
      edit.SetLogNumber(new_log_number);
2841
      impl->logfile_number_ = new_log_number;
2842
      impl->log_.reset(new log::Writer(std::move(lfile)));
2843
      s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
J
jorlow@chromium.org 已提交
2844 2845 2846
    }
    if (s.ok()) {
      impl->DeleteObsoleteFiles();
2847
      impl->MaybeScheduleCompaction();
2848
      impl->MaybeScheduleLogDBDeployStats();
J
jorlow@chromium.org 已提交
2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859
    }
  }
  impl->mutex_.Unlock();
  if (s.ok()) {
    *dbptr = impl;
  } else {
    delete impl;
  }
  return s;
}

2860 2861 2862
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2863 2864 2865
Status DestroyDB(const std::string& dbname, const Options& options) {
  Env* env = options.env;
  std::vector<std::string> filenames;
2866 2867
  std::vector<std::string> archiveFiles;

J
jorlow@chromium.org 已提交
2868 2869
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2870 2871
  env->GetChildren(ArchivalDirectory(dbname), &archiveFiles);

J
jorlow@chromium.org 已提交
2872 2873 2874 2875 2876
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
2877 2878
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2879 2880 2881
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
2882
    for (size_t i = 0; i < filenames.size(); i++) {
2883
      if (ParseFileName(filenames[i], &number, &type) &&
2884
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
2885 2886 2887 2888 2889 2890
        Status del;
        if (type == kMetaDatabase) {
          del = DestroyDB(dbname + "/" + filenames[i], options);
        } else {
          del = env->DeleteFile(dbname + "/" + filenames[i]);
        }
J
jorlow@chromium.org 已提交
2891 2892 2893 2894 2895
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2896 2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907

    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
      ParseFileName(archiveFiles[i], &number, &type);
      if (type == kLogFile) {
        Status del = env->DeleteFile(ArchivalDirectory(dbname) + "/" +
                                     archiveFiles[i]);
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2908 2909
    // ignore case where no archival directory is present.
    env->DeleteDir(ArchivalDirectory(dbname));
2910

J
jorlow@chromium.org 已提交
2911
    env->UnlockFile(lock);  // Ignore error since state is already gone
2912
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2913 2914 2915 2916 2917
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
  }
  return result;
}

2918 2919
//
// A global method that can dump out the build version
2920 2921
void dumpLeveldbBuildVersion(Logger * log) {
  Log(log, "Git sha %s", leveldb_build_git_sha);
2922 2923
  Log(log, "Compile time %s %s",
      leveldb_build_compile_time, leveldb_build_compile_date);
2924 2925
}

H
Hans Wennborg 已提交
2926
}  // namespace leveldb