db_impl.cc 74.6 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 13
#include <set>
#include <string>
#include <stdint.h>
#include <vector>
14

J
jorlow@chromium.org 已提交
15 16 17 18 19 20 21
#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"
22
#include "db/memtablelist.h"
J
jorlow@chromium.org 已提交
23 24 25
#include "db/table_cache.h"
#include "db/version_set.h"
#include "db/write_batch_internal.h"
26
#include "db/transaction_log_iterator_impl.h"
27 28
#include "leveldb/db.h"
#include "leveldb/env.h"
29
#include "leveldb/statistics.h"
30 31 32
#include "leveldb/status.h"
#include "leveldb/table.h"
#include "leveldb/table_builder.h"
J
jorlow@chromium.org 已提交
33 34 35 36 37 38 39
#include "port/port.h"
#include "table/block.h"
#include "table/merger.h"
#include "table/two_level_iterator.h"
#include "util/coding.h"
#include "util/logging.h"
#include "util/mutexlock.h"
40
#include "util/build_version.h"
41
#include "util/auto_split_logger.h"
J
jorlow@chromium.org 已提交
42 43 44

namespace leveldb {

45 46
void dumpLeveldbBuildVersion(Logger * log);

47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
static Status NewLogger(const std::string& dbname,
                        const std::string& db_log_dir,
                        Env* env,
                        size_t max_log_file_size,
                        Logger** logger) {
  std::string db_absolute_path;
  env->GetAbsolutePath(dbname, &db_absolute_path);

  if (max_log_file_size > 0) { // need to auto split the log file?
    AutoSplitLogger<Logger>* auto_split_logger =
      new AutoSplitLogger<Logger>(env, dbname, db_log_dir, max_log_file_size);
    Status s = auto_split_logger->GetStatus();
    if (!s.ok()) {
      delete auto_split_logger;
    } else {
      *logger = auto_split_logger;
    }
    return s;
  } else {
    // Open a log file in the same directory as the db
    env->CreateDir(dbname);  // In case it does not exist
    std::string fname = InfoLogFileName(dbname, db_absolute_path, db_log_dir);
    env->RenameFile(fname, OldInfoLogFileName(dbname, env->NowMicros(),
                                              db_absolute_path, db_log_dir));
    return env->NewLogger(fname, logger);
  }
}

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

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

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

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

  // Files produced by compaction
  struct Output {
    uint64_t number;
    uint64_t file_size;
    InternalKey smallest, largest;
  };
  std::vector<Output> outputs;
103
  std::list<uint64_t> allocated_file_numbers;
J
jorlow@chromium.org 已提交
104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120

  // State kept for output being generated
  WritableFile* outfile;
  TableBuilder* builder;

  uint64_t total_bytes;

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

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

D
Dhruba Borthakur 已提交
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
struct DBImpl::DeletionState {

  // the set of all live files that cannot be deleted
  std::set<uint64_t> live;

  // 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;

  // the list of all files to be evicted from the table cahce
  std::vector<uint64_t> files_to_evict;
};

J
jorlow@chromium.org 已提交
137 138 139
// Fix user-supplied options to be reasonable
template <class T,class V>
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
D
dgrogan@chromium.org 已提交
140 141
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
J
jorlow@chromium.org 已提交
142 143 144
}
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
S
Sanjay Ghemawat 已提交
145
                        const InternalFilterPolicy* ipolicy,
J
jorlow@chromium.org 已提交
146 147 148
                        const Options& src) {
  Options result = src;
  result.comparator = icmp;
S
Sanjay Ghemawat 已提交
149 150 151 152
  result.filter_policy = (src.filter_policy != NULL) ? ipolicy : NULL;
  ClipToRange(&result.max_open_files,            20,     50000);
  ClipToRange(&result.write_buffer_size,         64<<10, 1<<30);
  ClipToRange(&result.block_size,                1<<10,  4<<20);
J
jorlow@chromium.org 已提交
153
  if (result.info_log == NULL) {
154 155
    Status s = NewLogger(dbname, result.db_log_dir, src.env,
                         result.max_log_file_size, &result.info_log);
J
jorlow@chromium.org 已提交
156 157
    if (!s.ok()) {
      // No place suitable for logging
158
      result.info_log = NULL;
J
jorlow@chromium.org 已提交
159 160
    }
  }
161
  if (result.block_cache == NULL && !result.no_block_cache) {
162 163
    result.block_cache = NewLRUCache(8 << 20);
  }
164 165
  if (src.compression_per_level != NULL) {
    result.compression_per_level = new CompressionType[src.num_levels];
166
    for (int i = 0; i < src.num_levels; i++) {
167 168 169
      result.compression_per_level[i] = src.compression_per_level[i];
    }
  }
J
jorlow@chromium.org 已提交
170 171 172 173 174
  return result;
}

DBImpl::DBImpl(const Options& options, const std::string& dbname)
    : env_(options.env),
H
heyongqiang 已提交
175
      dbname_(dbname),
J
jorlow@chromium.org 已提交
176
      internal_comparator_(options.comparator),
S
Sanjay Ghemawat 已提交
177 178
      options_(SanitizeOptions(
          dbname, &internal_comparator_, &internal_filter_policy_, options)),
H
heyongqiang 已提交
179
      internal_filter_policy_(options.filter_policy),
J
jorlow@chromium.org 已提交
180
      owns_info_log_(options_.info_log != options.info_log),
181
      owns_cache_(options_.block_cache != options.block_cache),
J
jorlow@chromium.org 已提交
182 183 184
      db_lock_(NULL),
      shutting_down_(NULL),
      bg_cv_(&mutex_),
185
      mem_(new MemTable(internal_comparator_, NumberLevels())),
J
jorlow@chromium.org 已提交
186
      logfile_(NULL),
187
      logfile_number_(0),
J
jorlow@chromium.org 已提交
188
      log_(NULL),
189
      tmp_batch_(new WriteBatch),
190
      bg_compaction_scheduled_(0),
191
      bg_logstats_scheduled_(false),
192
      manual_compaction_(NULL),
193
      logger_(NULL),
194
      disable_delete_obsolete_files_(false),
195
      delete_obsolete_files_last_run_(0),
M
Mark Callaghan 已提交
196 197 198
      stall_level0_slowdown_(0),
      stall_memtable_compaction_(0),
      stall_level0_num_files_(0),
199
      stall_leveln_slowdown_(0),
200
      started_at_(options.env->NowMicros()),
201
      flush_on_destroy_(false),
202
      delayed_writes_(0) {
203
  mem_->Ref();
204

H
heyongqiang 已提交
205
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
206
  stats_ = new CompactionStats[options.num_levels];
J
jorlow@chromium.org 已提交
207
  // Reserve ten files or so for other uses and give the rest to TableCache.
208
  const int table_cache_size = options_.max_open_files - 10;
J
jorlow@chromium.org 已提交
209 210 211 212
  table_cache_ = new TableCache(dbname_, &options_, table_cache_size);

  versions_ = new VersionSet(dbname_, &options_, table_cache_,
                             &internal_comparator_);
213

214
  dumpLeveldbBuildVersion(options_.info_log);
215 216
  options_.Dump(options_.info_log);

217 218 219 220 221
#ifdef USE_SCRIBE
  logger_ = new ScribeLogger("localhost", 1456);
#endif

  char name[100];
222
  Status st = env_->GetHostName(name, 100L);
223 224 225 226 227 228 229
  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;
230

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

DBImpl::~DBImpl() {
  // Wait for background work to finish
235 236 237
  if (flush_on_destroy_) {
    FlushMemTable(FlushOptions());
  }
238
  mutex_.Lock();
J
jorlow@chromium.org 已提交
239
  shutting_down_.Release_Store(this);  // Any non-NULL value is ok
240
  while (bg_compaction_scheduled_ || bg_logstats_scheduled_) {
H
hans@chromium.org 已提交
241
    bg_cv_.Wait();
J
jorlow@chromium.org 已提交
242 243 244 245 246 247 248 249
  }
  mutex_.Unlock();

  if (db_lock_ != NULL) {
    env_->UnlockFile(db_lock_);
  }

  delete versions_;
250
  if (mem_ != NULL) mem_->Unref();
251
  imm_.UnrefAll();
252
  delete tmp_batch_;
J
jorlow@chromium.org 已提交
253 254 255
  delete log_;
  delete logfile_;
  delete table_cache_;
256
  delete[] stats_;
J
jorlow@chromium.org 已提交
257 258 259 260

  if (owns_info_log_) {
    delete options_.info_log;
  }
261 262 263
  if (owns_cache_) {
    delete options_.block_cache;
  }
264 265 266
  if (options_.compression_per_level != NULL) {
    delete options_.compression_per_level;
  }
267 268

  delete logger_;
J
jorlow@chromium.org 已提交
269 270
}

A
Abhishek Kona 已提交
271
// Do not flush and close database elegantly. Simulate a crash.
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293
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.
  if (db_lock_ != NULL) {
    env_->UnlockFile(db_lock_);
  }
}


J
jorlow@chromium.org 已提交
294
Status DBImpl::NewDB() {
295
  VersionEdit new_db(NumberLevels());
J
jorlow@chromium.org 已提交
296
  new_db.SetComparatorName(user_comparator()->Name());
297
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

  const std::string manifest = DescriptorFileName(dbname_, 1);
  WritableFile* file;
  Status s = env_->NewWritableFile(manifest, &file);
  if (!s.ok()) {
    return s;
  }
  {
    log::Writer log(file);
    std::string record;
    new_db.EncodeTo(&record);
    s = log.AddRecord(record);
    if (s.ok()) {
      s = file->Close();
    }
  }
  delete file;
  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 {
330
    Log(options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
331 332 333 334
    *s = Status::OK();
  }
}

335 336
const Status DBImpl::CreateArchivalDirectory() {
  if (options_.WAL_ttl_seconds > 0) {
337
    std::string archivalPath = ArchivalDirectory(dbname_);
338 339 340 341 342
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

D
Dhruba Borthakur 已提交
343 344 345 346 347
// 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();

348 349 350 351 352
  // if deletion is disabled, do nothing
  if (disable_delete_obsolete_files_) {
    return;
  }

353 354 355 356 357
  // 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();
358
    if (delete_obsolete_files_last_run_ +
359 360 361 362 363 364
        options_.delete_obsolete_files_period_micros > now_micros) {
      return;
    }
    delete_obsolete_files_last_run_ = now_micros;
  }

J
jorlow@chromium.org 已提交
365
  // Make a set of all of the live files
D
Dhruba Borthakur 已提交
366 367 368 369 370
  deletion_state.live = pending_outputs_;
  versions_->AddLiveFiles(&deletion_state.live);

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

D
Dhruba Borthakur 已提交
372 373 374 375 376 377 378 379 380 381 382
  // 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 已提交
383 384
  uint64_t number;
  FileType type;
H
heyongqiang 已提交
385
  std::vector<std::string> old_log_files;
386

D
Dhruba Borthakur 已提交
387 388
  for (size_t i = 0; i < state.allfiles.size(); i++) {
    if (ParseFileName(state.allfiles[i], &number, &type)) {
J
jorlow@chromium.org 已提交
389 390 391
      bool keep = true;
      switch (type) {
        case kLogFile:
D
Dhruba Borthakur 已提交
392 393
          keep = ((number >= state.lognumber) ||
                  (number == state.prevlognumber));
J
jorlow@chromium.org 已提交
394 395 396 397
          break;
        case kDescriptorFile:
          // Keep my manifest file, and any newer incarnations'
          // (in case there is a race that allows other incarnations)
D
Dhruba Borthakur 已提交
398
          keep = (number >= state.filenumber);
J
jorlow@chromium.org 已提交
399 400
          break;
        case kTableFile:
D
Dhruba Borthakur 已提交
401
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
402 403 404 405
          break;
        case kTempFile:
          // Any temp files that are currently being written to must
          // be recorded in pending_outputs_, which is inserted into "live"
D
Dhruba Borthakur 已提交
406
          keep = (state.live.find(number) != state.live.end());
J
jorlow@chromium.org 已提交
407
          break;
H
heyongqiang 已提交
408 409 410
        case kInfoLogFile:
          keep = true;
          if (number != 0) {
D
Dhruba Borthakur 已提交
411
            old_log_files.push_back(state.allfiles[i]);
H
heyongqiang 已提交
412 413
          }
          break;
J
jorlow@chromium.org 已提交
414 415 416 417 418 419 420 421
        case kCurrentFile:
        case kDBLockFile:
          keep = true;
          break;
      }

      if (!keep) {
        if (type == kTableFile) {
D
Dhruba Borthakur 已提交
422 423
          // record the files to be evicted from the cache
          state.files_to_evict.push_back(number);
J
jorlow@chromium.org 已提交
424
        }
425
        Log(options_.info_log, "Delete type=%d #%lld\n",
J
jorlow@chromium.org 已提交
426 427
            int(type),
            static_cast<unsigned long long>(number));
428
        if (type == kLogFile && options_.WAL_ttl_seconds > 0) {
429 430
          Status st = env_->RenameFile(LogFileName(dbname_, number),
                                       ArchivedLogFileName(dbname_, number));
431 432
          if (!st.ok()) {
            Log(options_.info_log, "RenameFile type=%d #%lld FAILED\n",
H
heyongqiang 已提交
433 434
              int(type),
              static_cast<unsigned long long>(number));
435 436
          }
        } else {
437
          Status st = env_->DeleteFile(dbname_ + "/" + state.allfiles[i]);
438 439 440 441 442
          if(!st.ok()) {
            Log(options_.info_log, "Delete type=%d #%lld FAILED\n",
                int(type),
                static_cast<unsigned long long>(number));
          }
H
heyongqiang 已提交
443
        }
J
jorlow@chromium.org 已提交
444 445 446
      }
    }
  }
H
heyongqiang 已提交
447 448

  // Delete old log files.
H
heyongqiang 已提交
449 450 451 452
  int old_log_file_count = old_log_files.size();
  if (old_log_file_count >= KEEP_LOG_FILE_NUM &&
      !options_.db_log_dir.empty()) {
    std::sort(old_log_files.begin(), old_log_files.end());
H
heyongqiang 已提交
453
    for (int i = 0; i >= (old_log_file_count - KEEP_LOG_FILE_NUM); i++) {
H
heyongqiang 已提交
454
      std::string& to_delete = old_log_files.at(i);
D
Dhruba Borthakur 已提交
455 456
      // Log(options_.info_log, "Delete type=%d %s\n",
      //     int(kInfoLogFile), to_delete.c_str());
H
heyongqiang 已提交
457 458 459
      env_->DeleteFile(dbname_ + "/" + to_delete);
    }
  }
J
jorlow@chromium.org 已提交
460 461
}

D
Dhruba Borthakur 已提交
462 463 464 465 466 467 468 469 470 471 472 473
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);
474 475 476 477 478 479
  PurgeObsoleteWALFiles();
}

void DBImpl::PurgeObsoleteWALFiles() {
  if (options_.WAL_ttl_seconds != ULONG_MAX && options_.WAL_ttl_seconds > 0) {
    std::vector<std::string> WALFiles;
480
    std::string archivalDir = ArchivalDirectory(dbname_);
481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504
    env_->GetChildren(archivalDir, &WALFiles);
    int64_t currentTime;
    const Status status = env_->GetCurrentTime(&currentTime);
    assert(status.ok());
    for (std::vector<std::string>::iterator it = WALFiles.begin();
         it != WALFiles.end();
         ++it) {

    uint64_t fileMTime;
    const std::string filePath = archivalDir + "/" + *it;
    const Status s = env_->GetFileModificationTime(filePath, &fileMTime);
    if (s.ok()) {
      if (status.ok() &&
         (currentTime - fileMTime > options_.WAL_ttl_seconds)) {
        Status delStatus = env_->DeleteFile(filePath);
        if (!delStatus.ok()) {
          Log(options_.info_log,
              "Failed Deleting a WAL file Error : i%s",
              delStatus.ToString().c_str());
        }
      }
    } // Ignore errors.
    }
  }
D
Dhruba Borthakur 已提交
505 506
}

H
heyongqiang 已提交
507 508
Status DBImpl::Recover(VersionEdit* edit, bool no_log_recory,
    bool error_if_log_file_exist) {
J
jorlow@chromium.org 已提交
509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537
  mutex_.AssertHeld();

  // Ignore error from CreateDir since the creation of the DB is
  // committed only when the descriptor is created, and this directory
  // may already exist from a previous failed creation attempt.
  env_->CreateDir(dbname_);
  assert(db_lock_ == NULL);
  Status s = env_->LockFile(LockFileName(dbname_), &db_lock_);
  if (!s.ok()) {
    return s;
  }

  if (!env_->FileExists(CurrentFileName(dbname_))) {
    if (options_.create_if_missing) {
      s = NewDB();
      if (!s.ok()) {
        return s;
      }
    } else {
      return Status::InvalidArgument(
          dbname_, "does not exist (create_if_missing is false)");
    }
  } else {
    if (options_.error_if_exists) {
      return Status::InvalidArgument(
          dbname_, "exists (error_if_exists is true)");
    }
  }

538
  s = versions_->Recover();
J
jorlow@chromium.org 已提交
539 540
  if (s.ok()) {
    SequenceNumber max_sequence(0);
541 542 543 544 545 546 547 548 549 550 551 552 553 554

    // 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;
555
    }
556 557 558 559 560 561 562 563 564
    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 已提交
565
    }
566

H
heyongqiang 已提交
567 568 569 570 571 572 573 574 575 576
    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");
    }

    if (no_log_recory) {
      return s;
    }

577 578 579 580
    // 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++) {
      s = RecoverLogFile(logs[i], edit, &max_sequence);
581 582 583 584 585

      // 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]);
586 587
    }

J
jorlow@chromium.org 已提交
588
    if (s.ok()) {
589 590 591
      if (versions_->LastSequence() < max_sequence) {
        versions_->SetLastSequence(max_sequence);
      }
J
jorlow@chromium.org 已提交
592 593 594 595 596 597 598 599 600 601 602
    }
  }

  return s;
}

Status DBImpl::RecoverLogFile(uint64_t log_number,
                              VersionEdit* edit,
                              SequenceNumber* max_sequence) {
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
603
    Logger* info_log;
J
jorlow@chromium.org 已提交
604 605 606
    const char* fname;
    Status* status;  // NULL if options_.paranoid_checks==false
    virtual void Corruption(size_t bytes, const Status& s) {
607
      Log(info_log, "%s%s: dropping %d bytes; %s",
J
jorlow@chromium.org 已提交
608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634
          (this->status == NULL ? "(ignoring error) " : ""),
          fname, static_cast<int>(bytes), s.ToString().c_str());
      if (this->status != NULL && this->status->ok()) *this->status = s;
    }
  };

  mutex_.AssertHeld();

  // Open the log file
  std::string fname = LogFileName(dbname_, log_number);
  SequentialFile* file;
  Status status = env_->NewSequentialFile(fname, &file);
  if (!status.ok()) {
    MaybeIgnoreError(&status);
    return status;
  }

  // Create the log reader.
  LogReporter reporter;
  reporter.env = env_;
  reporter.info_log = options_.info_log;
  reporter.fname = fname.c_str();
  reporter.status = (options_.paranoid_checks ? &status : NULL);
  // 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).
635 636
  log::Reader reader(file, &reporter, true/*checksum*/,
                     0/*initial_offset*/);
637
  Log(options_.info_log, "Recovering log #%llu",
J
jorlow@chromium.org 已提交
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
      (unsigned long long) log_number);

  // Read all the records and add to a memtable
  std::string scratch;
  Slice record;
  WriteBatch batch;
  MemTable* mem = NULL;
  while (reader.ReadRecord(&record, &scratch) &&
         status.ok()) {
    if (record.size() < 12) {
      reporter.Corruption(
          record.size(), Status::Corruption("log record too small"));
      continue;
    }
    WriteBatchInternal::SetContents(&batch, record);

    if (mem == NULL) {
655
      mem = new MemTable(internal_comparator_, NumberLevels());
656
      mem->Ref();
J
jorlow@chromium.org 已提交
657 658 659 660 661 662 663 664 665 666 667 668 669 670
    }
    status = WriteBatchInternal::InsertInto(&batch, mem);
    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;
    }

    if (mem->ApproximateMemoryUsage() > options_.write_buffer_size) {
671
      status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
672 673 674 675 676
      if (!status.ok()) {
        // Reflect errors immediately so that conditions like full
        // file-systems cause the DB::Open() to fail.
        break;
      }
677
      mem->Unref();
J
jorlow@chromium.org 已提交
678 679 680 681 682
      mem = NULL;
    }
  }

  if (status.ok() && mem != NULL) {
683
    status = WriteLevel0TableForRecovery(mem, edit);
J
jorlow@chromium.org 已提交
684 685 686 687
    // Reflect errors immediately so that conditions like full
    // file-systems cause the DB::Open() to fail.
  }

688
  if (mem != NULL) mem->Unref();
J
jorlow@chromium.org 已提交
689 690 691 692
  delete file;
  return status;
}

693
Status DBImpl::WriteLevel0TableForRecovery(MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
694
  mutex_.AssertHeld();
695
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
696 697 698 699
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
700
  Log(options_.info_log, "Level-0 table #%llu: started",
J
jorlow@chromium.org 已提交
701
      (unsigned long long) meta.number);
702 703 704 705

  Status s;
  {
    mutex_.Unlock();
706
    s = BuildTable(dbname_, env_, options_, table_cache_, iter, &meta);
707 708 709
    mutex_.Lock();
  }

710
  Log(options_.info_log, "Level-0 table #%llu: %lld bytes %s",
J
jorlow@chromium.org 已提交
711 712 713 714
      (unsigned long long) meta.number,
      (unsigned long long) meta.file_size,
      s.ToString().c_str());
  delete iter;
715

716
  pending_outputs_.erase(meta.number);
717 718 719 720 721 722 723 724 725

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

726 727 728
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros;
  stats.bytes_written = meta.file_size;
M
Mark Callaghan 已提交
729
  stats.files_out_levelnp1 = 1;
730
  stats_[level].Add(stats);
J
jorlow@chromium.org 已提交
731 732 733
  return s;
}

734 735 736

Status DBImpl::WriteLevel0Table(MemTable* mem, VersionEdit* edit,
                                uint64_t* filenumber) {
J
jorlow@chromium.org 已提交
737
  mutex_.AssertHeld();
738 739 740 741 742 743 744 745
  const uint64_t start_micros = env_->NowMicros();
  FileMetaData meta;
  meta.number = versions_->NewFileNumber();
  *filenumber = meta.number;
  pending_outputs_.insert(meta.number);
  Iterator* iter = mem->NewIterator();
  Log(options_.info_log, "Level-0 flush table #%llu: started",
      (unsigned long long) meta.number);
J
jorlow@chromium.org 已提交
746

747 748
  Version* base = versions_->current();
  base->Ref();
749 750 751 752 753 754
  Status s;
  {
    mutex_.Unlock();
    s = BuildTable(dbname_, env_, options_, table_cache_, iter, &meta);
    mutex_.Lock();
  }
755 756
  base->Unref();

757 758 759 760 761 762 763 764 765 766 767 768 769 770
  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 已提交
771
  // committed so that other threads can recognize this file as a
772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825
  // 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.
    if (base != NULL && options_.max_background_compactions <= 1) {
      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);

  if (!imm_.IsFlushPending()) {
    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;
  MemTable* m = imm_.PickMemtableToFlush();
  if (m == NULL) {
    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
  VersionEdit* edit = m->GetEdits();
  edit->SetPrevLogNumber(0);
  edit->SetLogNumber(logfile_number_);  // Earlier logs no longer needed

  Status s = WriteLevel0Table(m, edit, &file_number);

826 827 828
  if (s.ok() && shutting_down_.Acquire_Load()) {
    s = Status::IOError("Deleting DB during memtable compaction");
  }
J
jorlow@chromium.org 已提交
829

830
  // Replace immutable memtable with the generated Table
A
Abhishek Kona 已提交
831
  s = imm_.InstallMemtableFlushResults(m, versions_, s, &mutex_,
832
                        options_.info_log, file_number, pending_outputs_);
J
jorlow@chromium.org 已提交
833 834

  if (s.ok()) {
835 836 837
    if (madeProgress) {
      *madeProgress = 1;
    }
838
    MaybeScheduleLogDBDeployStats();
D
Dhruba Borthakur 已提交
839 840 841
    // 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 已提交
842 843 844 845
  }
  return s;
}

G
Gabor Cselle 已提交
846 847 848 849 850
void DBImpl::CompactRange(const Slice* begin, const Slice* end) {
  int max_level_with_files = 1;
  {
    MutexLock l(&mutex_);
    Version* base = versions_->current();
851
    for (int level = 1; level < NumberLevels(); level++) {
G
Gabor Cselle 已提交
852 853 854 855 856 857 858 859 860 861 862
      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);
  }
}

863
int DBImpl::NumberLevels() {
864
  return options_.num_levels;
865 866 867
}

int DBImpl::MaxMemCompactionLevel() {
868
  return options_.max_mem_compaction_level;
869 870 871
}

int DBImpl::Level0StopWriteTrigger() {
872
  return options_.level0_stop_writes_trigger;
873 874
}

H
heyongqiang 已提交
875 876 877 878 879
Status DBImpl::Flush(const FlushOptions& options) {
  Status status = FlushMemTable(options);
  return status;
}

880 881 882 883
SequenceNumber DBImpl::GetLatestSequenceNumber() {
  return versions_->LastSequence();
}

884 885 886 887 888 889 890 891 892 893 894 895 896 897 898
Status DBImpl::GetUpdatesSince(SequenceNumber seq,
                               TransactionLogIterator** iter) {

  //  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.
899
  s = ListAllWALFiles(ArchivalDirectory(dbname_), &walFiles, kArchivedLogFile);
900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 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 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042
  if (!s.ok()) {
    return s;
  }

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

  std::vector<LogFile>* probableWALFiles = new std::vector<LogFile>();
  FindProbableWALFiles(&walFiles, probableWALFiles, seq);
  if (probableWALFiles->empty()) {
    return Status::IOError(" No wal files for the given seqNo. found");
  }

  TransactionLogIteratorImpl* impl =
    new TransactionLogIteratorImpl(dbname_, &options_, seq, probableWALFiles);
  *iter = impl;
  return Status::OK();
}

Status DBImpl::FindProbableWALFiles(std::vector<LogFile>* const allLogs,
                                    std::vector<LogFile>* const result,
                                    const SequenceNumber target) {
  assert(allLogs != NULL);
  assert(result != NULL);

  std::sort(allLogs->begin(), allLogs->end());
  size_t start = 0;
  size_t end = allLogs->size();
  // Binary Search. avoid opening all files.
  while (start < end) {
    int mid = (start + end) / 2;
    WriteBatch batch;
    Status s = ReadFirstRecord(allLogs->at(mid), &batch);
    if (!s.ok()) {
      return s;
    }
    SequenceNumber currentSeqNum = WriteBatchInternal::Sequence(&batch);
    if (currentSeqNum == target) {
      start = mid;
      end = mid;
    } else if (currentSeqNum < target) {
      start = mid;
    } else {
      end = mid;
    }
  }
  assert( start == end);
  for( size_t i = start; i < allLogs->size(); ++i) {
    result->push_back(allLogs->at(i));
  }
  return Status::OK();
}

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;
    Status* status;  // NULL if options_.paranoid_checks==false
    virtual void Corruption(size_t bytes, const Status& s) {
      Log(info_log, "%s%s: dropping %d bytes; %s",
          (this->status == NULL ? "(ignoring error) " : ""),
          fname, static_cast<int>(bytes), s.ToString().c_str());
      if (this->status != NULL && this->status->ok()) *this->status = s;
    }
  };

  SequentialFile* file;
  Status status = env_->NewSequentialFile(fname, &file);

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


  LogReporter reporter;
  reporter.env = env_;
  reporter.info_log = options_.info_log;
  reporter.fname = fname.c_str();
  reporter.status = (options_.paranoid_checks ? &status : NULL);
  log::Reader reader(file, &reporter, true/*checksum*/,
                     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) {
  assert(logFiles != NULL);
  std::vector<std::string> allFiles;
  const Status status = env_->GetChildren(path, &allFiles);
  if (!status.ok()) {
    return status;
  }
  for(std::vector<std::string>::iterator it = allFiles.begin();
      it != allFiles.end();
      ++it) {
    uint64_t number;
    FileType type;
    if (ParseFileName(*it, &number, &type) && type == kLogFile){
      logFiles->push_back(LogFile(number, logType));
    }
  }
  return status;
}

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

G
Gabor Cselle 已提交
1046 1047
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
1048 1049
  ManualCompaction manual;
  manual.level = level;
G
Gabor Cselle 已提交
1050
  manual.done = false;
1051
  manual.in_progress = false;
G
Gabor Cselle 已提交
1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065
  if (begin == NULL) {
    manual.begin = NULL;
  } else {
    begin_storage = InternalKey(*begin, kMaxSequenceNumber, kValueTypeForSeek);
    manual.begin = &begin_storage;
  }
  if (end == NULL) {
    manual.end = NULL;
  } else {
    end_storage = InternalKey(*end, 0, static_cast<ValueType>(0));
    manual.end = &end_storage;
  }

  MutexLock l(&mutex_);
1066

A
Abhishek Kona 已提交
1067 1068 1069 1070
  // 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
1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
  // 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 已提交
1084 1085 1086 1087 1088
  while (!manual.done) {
    while (manual_compaction_ != NULL) {
      bg_cv_.Wait();
    }
    manual_compaction_ = &manual;
1089 1090 1091
    if (bg_compaction_scheduled_ == LargeNumber) {
      bg_compaction_scheduled_ = newvalue;
    }
G
Gabor Cselle 已提交
1092 1093 1094 1095
    MaybeScheduleCompaction();
    while (manual_compaction_ == &manual) {
      bg_cv_.Wait();
    }
H
hans@chromium.org 已提交
1096
  }
1097 1098 1099 1100 1101 1102 1103 1104 1105
  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 已提交
1106 1107
}

H
heyongqiang 已提交
1108
Status DBImpl::FlushMemTable(const FlushOptions& options) {
1109 1110
  // NULL batch means just wait for earlier writes to be done
  Status s = Write(WriteOptions(), NULL);
H
heyongqiang 已提交
1111
  if (s.ok() && options.wait) {
1112
    // Wait until the compaction completes
H
heyongqiang 已提交
1113
    s = WaitForCompactMemTable();
1114 1115
  }
  return s;
J
jorlow@chromium.org 已提交
1116 1117
}

H
heyongqiang 已提交
1118
Status DBImpl::WaitForCompactMemTable() {
1119 1120 1121
  Status s;
  // Wait until the compaction completes
  MutexLock l(&mutex_);
1122
  while (imm_.size() > 0 && bg_error_.ok()) {
1123 1124
    bg_cv_.Wait();
  }
1125
  if (imm_.size() != 0) {
1126 1127 1128
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
1129 1130 1131 1132 1133 1134
}

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

1135
Status DBImpl::TEST_WaitForCompactMemTable() {
1136
  return WaitForCompactMemTable();
1137 1138 1139
}

Status DBImpl::TEST_WaitForCompact() {
1140 1141 1142 1143 1144 1145
  // Wait until the compaction completes
  MutexLock l(&mutex_);
  while (bg_compaction_scheduled_ && bg_error_.ok()) {
    bg_cv_.Wait();
  }
  return bg_error_;
1146 1147
}

J
jorlow@chromium.org 已提交
1148 1149
void DBImpl::MaybeScheduleCompaction() {
  mutex_.AssertHeld();
1150
  if (bg_compaction_scheduled_ >= options_.max_background_compactions) {
J
jorlow@chromium.org 已提交
1151 1152 1153
    // Already scheduled
  } else if (shutting_down_.Acquire_Load()) {
    // DB is being deleted; no more background compactions
1154
  } else if (!imm_.IsFlushPending() &&
H
hans@chromium.org 已提交
1155 1156
             manual_compaction_ == NULL &&
             !versions_->NeedsCompaction()) {
J
jorlow@chromium.org 已提交
1157 1158
    // No work to be done
  } else {
1159
    bg_compaction_scheduled_++;
J
jorlow@chromium.org 已提交
1160 1161 1162 1163 1164 1165 1166 1167 1168
    env_->Schedule(&DBImpl::BGWork, this);
  }
}

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

void DBImpl::BackgroundCall() {
1169
  bool madeProgress;
D
Dhruba Borthakur 已提交
1170
  DeletionState deletion_state;
J
jorlow@chromium.org 已提交
1171
  MutexLock l(&mutex_);
1172
  // Log(options_.info_log, "XXX BG Thread %llx process new work item", pthread_self());
J
jorlow@chromium.org 已提交
1173
  assert(bg_compaction_scheduled_);
H
hans@chromium.org 已提交
1174
  if (!shutting_down_.Acquire_Load()) {
1175
    Status s = BackgroundCompaction(&madeProgress, deletion_state);
1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
    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 已提交
1188
  }
1189

D
Dhruba Borthakur 已提交
1190 1191 1192 1193
  // delete unnecessary files if any, this is done outside the mutex
  if (!deletion_state.live.empty()) {
    mutex_.Unlock();
    PurgeObsoleteFiles(deletion_state);
D
Dhruba Borthakur 已提交
1194
    EvictObsoleteFiles(deletion_state);
1195
    mutex_.Lock();
D
Dhruba Borthakur 已提交
1196 1197
  }

1198
  bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
1199

1200 1201
  MaybeScheduleLogDBDeployStats();

J
jorlow@chromium.org 已提交
1202
  // Previous compaction may have produced too many files in a level,
A
Abhishek Kona 已提交
1203
  // So reschedule another compaction if we made progress in the
1204 1205 1206 1207
  // last compaction.
  if (madeProgress) {
    MaybeScheduleCompaction();
  }
H
hans@chromium.org 已提交
1208
  bg_cv_.SignalAll();
J
jorlow@chromium.org 已提交
1209 1210
}

A
Abhishek Kona 已提交
1211
Status DBImpl::BackgroundCompaction(bool* madeProgress,
1212
  DeletionState& deletion_state) {
1213
  *madeProgress = false;
J
jorlow@chromium.org 已提交
1214
  mutex_.AssertHeld();
1215

1216
  while (imm_.IsFlushPending()) {
A
Abhishek Kona 已提交
1217
    Log(options_.info_log,
1218 1219 1220 1221 1222 1223
        "BackgroundCompaction doing CompactMemTable, compaction slots available %d",
        options_.max_background_compactions - bg_compaction_scheduled_);
    Status stat = CompactMemTable(madeProgress);
    if (!stat.ok()) {
      return stat;
    }
1224 1225
  }

1226
  Compaction* c = NULL;
1227 1228
  bool is_manual = (manual_compaction_ != NULL) &&
                   (manual_compaction_->in_progress == false);
G
Gabor Cselle 已提交
1229
  InternalKey manual_end;
H
hans@chromium.org 已提交
1230
  if (is_manual) {
G
Gabor Cselle 已提交
1231
    ManualCompaction* m = manual_compaction_;
1232 1233
    assert(!m->in_progress);
    m->in_progress = true; // another thread cannot pick up the same work
G
Gabor Cselle 已提交
1234 1235 1236 1237 1238 1239 1240
    c = versions_->CompactRange(m->level, m->begin, m->end);
    m->done = (c == NULL);
    if (c != NULL) {
      manual_end = c->input(0, c->num_input_files(0) - 1)->largest;
    }
    Log(options_.info_log,
        "Manual compaction at level-%d from %s .. %s; will stop at %s\n",
H
hans@chromium.org 已提交
1241
        m->level,
G
Gabor Cselle 已提交
1242 1243 1244
        (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
        (m->end ? m->end->DebugString().c_str() : "(end)"),
        (m->done ? "(end)" : manual_end.DebugString().c_str()));
1245
  } else if (!options_.disable_auto_compactions) {
H
hans@chromium.org 已提交
1246
    c = versions_->PickCompaction();
J
jorlow@chromium.org 已提交
1247 1248 1249
  }

  Status status;
H
hans@chromium.org 已提交
1250 1251
  if (c == NULL) {
    // Nothing to do
1252
    Log(options_.info_log, "Compaction nothing to do");
H
hans@chromium.org 已提交
1253
  } else if (!is_manual && c->IsTrivialMove()) {
J
jorlow@chromium.org 已提交
1254
    // Move file to next level
1255
    assert(c->num_input_files(0) == 1);
J
jorlow@chromium.org 已提交
1256 1257 1258 1259
    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);
1260
    status = versions_->LogAndApply(c->edit(), &mutex_);
H
hans@chromium.org 已提交
1261
    VersionSet::LevelSummaryStorage tmp;
1262
    Log(options_.info_log, "Moved #%lld to level-%d %lld bytes %s: %s\n",
J
jorlow@chromium.org 已提交
1263 1264 1265
        static_cast<unsigned long long>(f->number),
        c->level() + 1,
        static_cast<unsigned long long>(f->file_size),
H
hans@chromium.org 已提交
1266 1267
        status.ToString().c_str(),
        versions_->LevelSummary(&tmp));
1268
    versions_->ReleaseCompactionFiles(c, status);
1269
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1270 1271 1272 1273
  } else {
    CompactionState* compact = new CompactionState(c);
    status = DoCompactionWork(compact);
    CleanupCompaction(compact);
1274
    versions_->ReleaseCompactionFiles(c, status);
1275
    c->ReleaseInputs();
D
Dhruba Borthakur 已提交
1276
    FindObsoleteFiles(deletion_state);
1277
    *madeProgress = true;
J
jorlow@chromium.org 已提交
1278 1279 1280 1281 1282 1283 1284 1285
  }
  delete c;

  if (status.ok()) {
    // Done
  } else if (shutting_down_.Acquire_Load()) {
    // Ignore compaction errors found during shutting down
  } else {
1286
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1287 1288 1289 1290 1291
        "Compaction error: %s", status.ToString().c_str());
    if (options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
1292 1293

  if (is_manual) {
G
Gabor Cselle 已提交
1294
    ManualCompaction* m = manual_compaction_;
1295 1296 1297
    if (!status.ok()) {
      m->done = true;
    }
G
Gabor Cselle 已提交
1298 1299 1300 1301 1302 1303
    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;
    }
1304
    m->in_progress = false; // not being processed anymore
H
hans@chromium.org 已提交
1305 1306
    manual_compaction_ = NULL;
  }
1307
  return status;
J
jorlow@chromium.org 已提交
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319
}

void DBImpl::CleanupCompaction(CompactionState* compact) {
  mutex_.AssertHeld();
  if (compact->builder != NULL) {
    // May happen if we get a shutdown call in the middle of compaction
    compact->builder->Abandon();
    delete compact->builder;
  } else {
    assert(compact->outfile == NULL);
  }
  delete compact->outfile;
D
dgrogan@chromium.org 已提交
1320
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1321 1322 1323 1324 1325 1326
    const CompactionState::Output& out = compact->outputs[i];
    pending_outputs_.erase(out.number);
  }
  delete compact;
}

1327 1328 1329 1330 1331 1332 1333 1334
// 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();
  assert(compact != NULL);
  assert(compact->builder == NULL);
  int filesNeeded = compact->compaction->num_input_files(1);
1335
  for (int i = 0; i < filesNeeded; i++) {
1336 1337 1338 1339 1340 1341 1342 1343 1344
    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();
A
Abhishek Kona 已提交
1345
  for (std::list<uint64_t>::iterator it =
1346 1347 1348 1349 1350 1351 1352 1353
       compact->allocated_file_numbers.begin();
       it != compact->allocated_file_numbers.end(); ++it) {
    uint64_t file_number = *it;
    pending_outputs_.erase(file_number);
    // Log(options_.info_log, "XXX releasing unused file num %d", file_number);
  }
}

J
jorlow@chromium.org 已提交
1354 1355 1356 1357
Status DBImpl::OpenCompactionOutputFile(CompactionState* compact) {
  assert(compact != NULL);
  assert(compact->builder == NULL);
  uint64_t file_number;
1358 1359 1360 1361 1362 1363 1364
  // 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 已提交
1365 1366 1367 1368 1369
    mutex_.Lock();
    file_number = versions_->NewFileNumber();
    pending_outputs_.insert(file_number);
    mutex_.Unlock();
  }
1370 1371 1372 1373 1374
  CompactionState::Output out;
  out.number = file_number;
  out.smallest.Clear();
  out.largest.Clear();
  compact->outputs.push_back(out);
J
jorlow@chromium.org 已提交
1375 1376 1377 1378 1379

  // Make the output file
  std::string fname = TableFileName(dbname_, file_number);
  Status s = env_->NewWritableFile(fname, &compact->outfile);
  if (s.ok()) {
1380 1381
    compact->builder = new TableBuilder(options_, compact->outfile,
                                        compact->compaction->level() + 1);
J
jorlow@chromium.org 已提交
1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409
  }
  return s;
}

Status DBImpl::FinishCompactionOutputFile(CompactionState* compact,
                                          Iterator* input) {
  assert(compact != NULL);
  assert(compact->outfile != NULL);
  assert(compact->builder != NULL);

  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;
  delete compact->builder;
  compact->builder = NULL;

  // Finish and check for file errors
1410
  if (s.ok() && !options_.disableDataSync) {
1411 1412 1413 1414 1415
    if (options_.use_fsync) {
      s = compact->outfile->Fsync();
    } else {
      s = compact->outfile->Sync();
    }
J
jorlow@chromium.org 已提交
1416 1417 1418 1419 1420 1421 1422 1423 1424
  }
  if (s.ok()) {
    s = compact->outfile->Close();
  }
  delete compact->outfile;
  compact->outfile = NULL;

  if (s.ok() && current_entries > 0) {
    // Verify that the table is usable
J
jorlow@chromium.org 已提交
1425 1426 1427
    Iterator* iter = table_cache_->NewIterator(ReadOptions(),
                                               output_number,
                                               current_bytes);
J
jorlow@chromium.org 已提交
1428 1429 1430
    s = iter->status();
    delete iter;
    if (s.ok()) {
1431
      Log(options_.info_log,
J
jorlow@chromium.org 已提交
1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443
          "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();
1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458

  // 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.
  if (options_.paranoid_checks &&
      !versions_->VerifyCompactionFileConsistency(compact->compaction)) {
    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");
  }

1459
  Log(options_.info_log,  "Compacted %d@%d + %d@%d files => %lld bytes",
J
jorlow@chromium.org 已提交
1460 1461 1462 1463 1464 1465 1466 1467 1468
      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 已提交
1469
  for (size_t i = 0; i < compact->outputs.size(); i++) {
J
jorlow@chromium.org 已提交
1470 1471 1472 1473 1474
    const CompactionState::Output& out = compact->outputs[i];
    compact->compaction->edit()->AddFile(
        level + 1,
        out.number, out.file_size, out.smallest, out.largest);
  }
1475
  return versions_->LogAndApply(compact->compaction->edit(), &mutex_);
J
jorlow@chromium.org 已提交
1476 1477
}

1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503
//
// 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(
  SequenceNumber in, std::vector<SequenceNumber>& snapshots) {
  SequenceNumber prev;
  prev = 0;
  for (std::vector<SequenceNumber>::iterator it = snapshots.begin();
       it < snapshots.end(); it++) {
    assert (prev <= *it);
    if (*it >= in) {
      return *it;
    }
    assert(prev = *it); // assignment
  }
  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 已提交
1504
Status DBImpl::DoCompactionWork(CompactionState* compact) {
1505 1506
  int64_t imm_micros = 0;  // Micros spent doing imm_ compactions

A
Abhishek Kona 已提交
1507
  Log(options_.info_log,
1508
      "Compacting %d@%d + %d@%d files, score %.2f slots available %d",
J
jorlow@chromium.org 已提交
1509 1510 1511
      compact->compaction->num_input_files(0),
      compact->compaction->level(),
      compact->compaction->num_input_files(1),
1512
      compact->compaction->level() + 1,
1513
      compact->compaction->score(),
1514
      options_.max_background_compactions - bg_compaction_scheduled_);
1515 1516
  char scratch[256];
  compact->compaction->Summary(scratch, sizeof(scratch));
H
heyongqiang 已提交
1517
  Log(options_.info_log, "Compaction start summary: %s\n", scratch);
J
jorlow@chromium.org 已提交
1518 1519 1520 1521

  assert(versions_->NumLevelFiles(compact->compaction->level()) > 0);
  assert(compact->builder == NULL);
  assert(compact->outfile == NULL);
1522 1523 1524 1525 1526 1527 1528 1529

  SequenceNumber visible_at_tip = 0;
  SequenceNumber earliest_snapshot;
  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 已提交
1530
  } else {
1531 1532 1533 1534
    // 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 已提交
1535 1536
  }

1537 1538 1539
  // Allocate the output file numbers before we release the lock
  AllocateCompactionOutputFileNumbers(compact);

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

1543
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1544 1545 1546 1547 1548 1549 1550
  Iterator* input = versions_->MakeInputIterator(compact->compaction);
  input->SeekToFirst();
  Status status;
  ParsedInternalKey ikey;
  std::string current_user_key;
  bool has_current_user_key = false;
  SequenceNumber last_sequence_for_key = kMaxSequenceNumber;
1551
  SequenceNumber visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1552
  for (; input->Valid() && !shutting_down_.Acquire_Load(); ) {
1553
    // Prioritize immutable compaction work
1554
    if (imm_.imm_flush_needed.NoBarrier_Load() != NULL) {
1555 1556
      const uint64_t imm_start = env_->NowMicros();
      mutex_.Lock();
1557
      if (imm_.IsFlushPending()) {
1558
        CompactMemTable();
H
hans@chromium.org 已提交
1559
        bg_cv_.SignalAll();  // Wakeup MakeRoomForWrite() if necessary
1560 1561 1562 1563 1564
      }
      mutex_.Unlock();
      imm_micros += (env_->NowMicros() - imm_start);
    }

J
jorlow@chromium.org 已提交
1565
    Slice key = input->key();
1566 1567
    Slice value = input->value();
    Slice* compaction_filter_value = NULL;
1568
    if (compact->compaction->ShouldStopBefore(key) &&
1569 1570 1571 1572 1573 1574 1575 1576
        compact->builder != NULL) {
      status = FinishCompactionOutputFile(compact, input);
      if (!status.ok()) {
        break;
      }
    }

    // Handle key/value, add to state, etc.
J
jorlow@chromium.org 已提交
1577 1578 1579 1580 1581 1582
    bool drop = false;
    if (!ParseInternalKey(key, &ikey)) {
      // Do not hide error keys
      current_user_key.clear();
      has_current_user_key = false;
      last_sequence_for_key = kMaxSequenceNumber;
1583
      visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1584 1585 1586 1587 1588 1589 1590 1591
    } 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;
1592
        visible_in_snapshot = kMaxSequenceNumber;
J
jorlow@chromium.org 已提交
1593 1594
      }

1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
      // 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.
      SequenceNumber visible = visible_at_tip ? visible_at_tip :
                                 findEarliestVisibleSnapshot(ikey.sequence,
                                 compact->existing_snapshots);

      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 已提交
1606
        // Hidden by an newer entry for same user key
1607
        assert(last_sequence_for_key >= ikey.sequence);
J
jorlow@chromium.org 已提交
1608
        drop = true;    // (A)
1609
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_NEWER_ENTRY);
J
jorlow@chromium.org 已提交
1610
      } else if (ikey.type == kTypeDeletion &&
1611
                 ikey.sequence <= earliest_snapshot &&
J
jorlow@chromium.org 已提交
1612 1613 1614 1615 1616 1617 1618 1619 1620
                 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;
1621
        RecordTick(options_.statistics, COMPACTION_KEY_DROP_OBSOLETE);
1622 1623
      } else if (options_.CompactionFilter != NULL &&
                 ikey.type != kTypeDeletion &&
1624
                 ikey.sequence < earliest_snapshot) {
1625 1626
        // If the user has specified a compaction filter, then invoke
        // it. If this key is not visible via any snapshot and the
1627 1628
        // return value of the compaction filter is true and then
        // drop this key from the output.
1629 1630
        drop = options_.CompactionFilter(options_.compaction_filter_args,
                         compact->compaction->level(),
1631 1632
                         ikey.user_key, value, &compaction_filter_value);

1633 1634 1635
        if (drop) {
          RecordTick(options_.statistics, COMPACTION_KEY_DROP_USER);
        }
1636 1637 1638 1639 1640
        // If the application wants to change the value, then do so here.
        if (compaction_filter_value != NULL) {
          value = *compaction_filter_value;
          delete compaction_filter_value;
        }
J
jorlow@chromium.org 已提交
1641 1642 1643
      }

      last_sequence_for_key = ikey.sequence;
1644
      visible_in_snapshot = visible;
J
jorlow@chromium.org 已提交
1645 1646
    }
#if 0
1647
    Log(options_.info_log,
J
jorlow@chromium.org 已提交
1648 1649 1650
        "  Compact: %s, seq %d, type: %d %d, drop: %d, is_base: %d, "
        "%d smallest_snapshot: %d",
        ikey.user_key.ToString().c_str(),
D
dgrogan@chromium.org 已提交
1651
        (int)ikey.sequence, ikey.type, kTypeValue, drop,
J
jorlow@chromium.org 已提交
1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667
        compact->compaction->IsBaseLevelForKey(ikey.user_key),
        (int)last_sequence_for_key, (int)compact->smallest_snapshot);
#endif

    if (!drop) {
      // Open output file if necessary
      if (compact->builder == NULL) {
        status = OpenCompactionOutputFile(compact);
        if (!status.ok()) {
          break;
        }
      }
      if (compact->builder->NumEntries() == 0) {
        compact->current_output()->smallest.DecodeFrom(key);
      }
      compact->current_output()->largest.DecodeFrom(key);
1668
      compact->builder->Add(key, value);
J
jorlow@chromium.org 已提交
1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694

      // Close output file if it is big enough
      if (compact->builder->FileSize() >=
          compact->compaction->MaxOutputFileSize()) {
        status = FinishCompactionOutputFile(compact, input);
        if (!status.ok()) {
          break;
        }
      }
    }

    input->Next();
  }

  if (status.ok() && shutting_down_.Acquire_Load()) {
    status = Status::IOError("Deleting DB during compaction");
  }
  if (status.ok() && compact->builder != NULL) {
    status = FinishCompactionOutputFile(compact, input);
  }
  if (status.ok()) {
    status = input->status();
  }
  delete input;
  input = NULL;

1695 1696
  CompactionStats stats;
  stats.micros = env_->NowMicros() - start_micros - imm_micros;
M
Mark Callaghan 已提交
1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707

  stats.files_in_leveln = compact->compaction->num_input_files(0);
  stats.files_in_levelnp1 = compact->compaction->num_input_files(1);
  stats.files_out_levelnp1 = compact->outputs.size();

  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;

D
dgrogan@chromium.org 已提交
1708
  for (size_t i = 0; i < compact->outputs.size(); i++) {
1709 1710 1711
    stats.bytes_written += compact->outputs[i].file_size;
  }

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

1715 1716 1717 1718
  // 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 已提交
1719 1720 1721
  if (status.ok()) {
    status = InstallCompactionResults(compact);
  }
1722
  VersionSet::LevelSummaryStorage tmp;
1723
  Log(options_.info_log,
M
Mark Callaghan 已提交
1724
      "compacted to: %s, %.1f MB/sec, level %d, files in(%d, %d) out(%d) "
1725
      "MB in(%.1f, %.1f) out(%.1f), amplify(%.1f) %s\n",
M
Mark Callaghan 已提交
1726 1727 1728 1729 1730 1731 1732 1733 1734
      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) /
1735 1736
          (double) stats.bytes_readn,
      status.ToString().c_str());
M
Mark Callaghan 已提交
1737

J
jorlow@chromium.org 已提交
1738 1739 1740
  return status;
}

1741 1742 1743 1744
namespace {
struct IterState {
  port::Mutex* mu;
  Version* version;
1745
  std::vector<MemTable*> mem; // includes both mem_ and imm_
1746 1747 1748 1749 1750
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
  state->mu->Lock();
1751 1752 1753
  for (unsigned int i = 0; i < state->mem.size(); i++) {
    state->mem[i]->Unref();
  }
1754 1755 1756 1757
  state->version->Unref();
  state->mu->Unlock();
  delete state;
}
H
Hans Wennborg 已提交
1758
}  // namespace
1759

J
jorlow@chromium.org 已提交
1760 1761
Iterator* DBImpl::NewInternalIterator(const ReadOptions& options,
                                      SequenceNumber* latest_snapshot) {
1762
  IterState* cleanup = new IterState;
J
jorlow@chromium.org 已提交
1763
  mutex_.Lock();
1764
  *latest_snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1765

1766
  // Collect together all needed child iterators for mem
J
jorlow@chromium.org 已提交
1767
  std::vector<Iterator*> list;
1768
  mem_->Ref();
1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779
  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);
1780
  }
1781 1782

  // Collect iterators for files in L0 - Ln
J
jorlow@chromium.org 已提交
1783 1784 1785 1786
  versions_->current()->AddIterators(options, &list);
  Iterator* internal_iter =
      NewMergingIterator(&internal_comparator_, &list[0], list.size());
  versions_->current()->Ref();
1787 1788 1789 1790

  cleanup->mu = &mutex_;
  cleanup->version = versions_->current();
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, NULL);
J
jorlow@chromium.org 已提交
1791 1792 1793 1794 1795 1796 1797 1798 1799 1800

  mutex_.Unlock();
  return internal_iter;
}

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

J
jorlow@chromium.org 已提交
1801
int64_t DBImpl::TEST_MaxNextLevelOverlappingBytes() {
1802 1803 1804 1805
  MutexLock l(&mutex_);
  return versions_->MaxNextLevelOverlappingBytes();
}

J
jorlow@chromium.org 已提交
1806 1807 1808
Status DBImpl::Get(const ReadOptions& options,
                   const Slice& key,
                   std::string* value) {
1809 1810 1811 1812 1813 1814 1815
  Status s;
  MutexLock l(&mutex_);
  SequenceNumber snapshot;
  if (options.snapshot != NULL) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_;
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
1816
  }
1817

1818
  MemTable* mem = mem_;
1819
  MemTableList imm = imm_;
1820
  Version* current = versions_->current();
1821
  mem->Ref();
1822
  imm.RefAll();
1823
  current->Ref();
1824 1825

  bool have_stat_update = false;
1826
  Version::GetStats stats;
1827 1828 1829

  // Unlock while reading from files and memtables
  {
1830
    mutex_.Unlock();
1831 1832
    // First look in the memtable, then in the immutable memtable (if any).
    LookupKey lkey(key, snapshot);
1833
    if (mem->Get(lkey, value, &s)) {
1834
      // Done
1835
    } else if (imm.Get(lkey, value, &s)) {
1836 1837 1838 1839 1840
      // Done
    } else {
      s = current->Get(options, lkey, value, &stats);
      have_stat_update = true;
    }
1841 1842
    mutex_.Lock();
  }
1843

1844 1845
  if (!options_.disable_seek_compaction &&
      have_stat_update && current->UpdateStats(stats)) {
1846 1847
    MaybeScheduleCompaction();
  }
1848
  mem->Unref();
1849
  imm.UnrefAll();
1850 1851
  current->Unref();
  return s;
J
jorlow@chromium.org 已提交
1852 1853 1854 1855 1856
}

Iterator* DBImpl::NewIterator(const ReadOptions& options) {
  SequenceNumber latest_snapshot;
  Iterator* internal_iter = NewInternalIterator(options, &latest_snapshot);
1857 1858 1859 1860 1861
  return NewDBIterator(
      &dbname_, env_, user_comparator(), internal_iter,
      (options.snapshot != NULL
       ? reinterpret_cast<const SnapshotImpl*>(options.snapshot)->number_
       : latest_snapshot));
J
jorlow@chromium.org 已提交
1862 1863 1864 1865
}

const Snapshot* DBImpl::GetSnapshot() {
  MutexLock l(&mutex_);
1866
  return snapshots_.New(versions_->LastSequence());
J
jorlow@chromium.org 已提交
1867 1868 1869 1870
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  MutexLock l(&mutex_);
1871
  snapshots_.Delete(reinterpret_cast<const SnapshotImpl*>(s));
J
jorlow@chromium.org 已提交
1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882
}

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

Status DBImpl::Delete(const WriteOptions& options, const Slice& key) {
  return DB::Delete(options, key);
}

1883 1884 1885 1886
Status DBImpl::Write(const WriteOptions& options, WriteBatch* my_batch) {
  Writer w(&mutex_);
  w.batch = my_batch;
  w.sync = options.sync;
H
heyongqiang 已提交
1887
  w.disableWAL = options.disableWAL;
1888
  w.done = false;
1889

D
dgrogan@chromium.org 已提交
1890
  MutexLock l(&mutex_);
1891 1892 1893 1894 1895 1896 1897 1898 1899 1900
  writers_.push_back(&w);
  while (!w.done && &w != writers_.front()) {
    w.cv.Wait();
  }
  if (w.done) {
    return w.status;
  }

  // May temporarily unlock and wait.
  Status status = MakeRoomForWrite(my_batch == NULL);
D
dgrogan@chromium.org 已提交
1901
  uint64_t last_sequence = versions_->LastSequence();
1902 1903 1904
  Writer* last_writer = &w;
  if (status.ok() && my_batch != NULL) {  // NULL batch is for compactions
    WriteBatch* updates = BuildBatchGroup(&last_writer);
D
dgrogan@chromium.org 已提交
1905 1906 1907
    WriteBatchInternal::SetSequence(updates, last_sequence + 1);
    last_sequence += WriteBatchInternal::Count(updates);

1908 1909 1910 1911
    // 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_.
1912 1913
    {
      mutex_.Unlock();
1914 1915 1916 1917
      if (options.disableWAL) {
        flush_on_destroy_ = true;
      }

H
heyongqiang 已提交
1918 1919 1920
      if (!options.disableWAL) {
        status = log_->AddRecord(WriteBatchInternal::Contents(updates));
        if (status.ok() && options.sync) {
1921 1922 1923 1924 1925
          if (options_.use_fsync) {
            status = logfile_->Fsync();
          } else {
            status = logfile_->Sync();
          }
H
heyongqiang 已提交
1926
        }
1927 1928 1929 1930 1931
      }
      if (status.ok()) {
        status = WriteBatchInternal::InsertInto(updates, mem_);
      }
      mutex_.Lock();
J
jorlow@chromium.org 已提交
1932
    }
1933
    if (updates == tmp_batch_) tmp_batch_->Clear();
1934 1935

    versions_->SetLastSequence(last_sequence);
J
jorlow@chromium.org 已提交
1936
  }
1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953

  while (true) {
    Writer* ready = writers_.front();
    writers_.pop_front();
    if (ready != &w) {
      ready->status = status;
      ready->done = true;
      ready->cv.Signal();
    }
    if (ready == last_writer) break;
  }

  // Notify new head of write queue
  if (!writers_.empty()) {
    writers_.front()->cv.Signal();
  }

J
jorlow@chromium.org 已提交
1954 1955 1956
  return status;
}

1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984
// REQUIRES: Writer list must be non-empty
// REQUIRES: First writer must have a non-NULL batch
WriteBatch* DBImpl::BuildBatchGroup(Writer** last_writer) {
  assert(!writers_.empty());
  Writer* first = writers_.front();
  WriteBatch* result = first->batch;
  assert(result != NULL);

  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 已提交
1985 1986 1987 1988 1989 1990
    if (!w->disableWAL && first->disableWAL) {
      // Do not include a write that needs WAL into a batch that has
      // WAL disabled.
      break;
    }

1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010 2011
    if (w->batch != NULL) {
      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
        result = tmp_batch_;
        assert(WriteBatchInternal::Count(result) == 0);
        WriteBatchInternal::Append(result, first->batch);
      }
      WriteBatchInternal::Append(result, w->batch);
    }
    *last_writer = w;
  }
  return result;
}

2012
// REQUIRES: mutex_ is held
2013
// REQUIRES: this thread is currently at the front of the writer queue
2014 2015
Status DBImpl::MakeRoomForWrite(bool force) {
  mutex_.AssertHeld();
2016
  assert(!writers_.empty());
2017
  bool allow_delay = !force;
2018
  Status s;
2019
  double score;
2020

2021 2022 2023 2024 2025
  while (true) {
    if (!bg_error_.ok()) {
      // Yield previous error
      s = bg_error_;
      break;
2026 2027
    } else if (
        allow_delay &&
2028
        versions_->NumLevelFiles(0) >=
2029
          options_.level0_slowdown_writes_trigger) {
2030 2031 2032 2033 2034 2035 2036
      // 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();
2037
      uint64_t t1 = env_->NowMicros();
2038
      env_->SleepForMicroseconds(1000);
2039 2040
      uint64_t delayed = env_->NowMicros() - t1;
      stall_level0_slowdown_ += delayed;
2041
      allow_delay = false;  // Do not delay a single write more than once
2042 2043
      //Log(options_.info_log,
      //    "delaying write %llu usecs for level0_slowdown_writes_trigger\n",
2044
      //     (long long unsigned int)delayed);
2045
      mutex_.Lock();
2046
      delayed_writes_++;
2047 2048 2049
    } else if (!force &&
               (mem_->ApproximateMemoryUsage() <= options_.write_buffer_size)) {
      // There is room in current memtable
2050 2051 2052
      if (allow_delay) {
        DelayLoggingAndReset();
      }
2053
      break;
2054
    } else if (imm_.size() == options_.max_write_buffer_number - 1) {
2055
      // We have filled up the current memtable, but the previous
2056 2057
      // ones are still being compacted, so we wait.
      DelayLoggingAndReset();
2058
      Log(options_.info_log, "wait for memtable compaction...\n");
M
Mark Callaghan 已提交
2059
      uint64_t t1 = env_->NowMicros();
H
hans@chromium.org 已提交
2060
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2061
      stall_memtable_compaction_ += env_->NowMicros() - t1;
2062
    } else if (versions_->NumLevelFiles(0) >=
2063
               options_.level0_stop_writes_trigger) {
2064
      // There are too many level-0 files.
2065
      DelayLoggingAndReset();
M
Mark Callaghan 已提交
2066
      uint64_t t1 = env_->NowMicros();
2067
      Log(options_.info_log, "wait for fewer level0 files...\n");
H
hans@chromium.org 已提交
2068
      bg_cv_.Wait();
M
Mark Callaghan 已提交
2069
      stall_level0_num_files_ += env_->NowMicros() - t1;
2070 2071 2072 2073 2074 2075
    } else if (
        allow_delay &&
        options_.rate_limit > 1.0 &&
        (score = versions_->MaxCompactionScore()) > options_.rate_limit) {
      // Delay a write when the compaction score for any level is too large.
      mutex_.Unlock();
2076
      uint64_t t1 = env_->NowMicros();
2077
      env_->SleepForMicroseconds(1000);
2078 2079
      uint64_t delayed = env_->NowMicros() - t1;
      stall_leveln_slowdown_ += delayed;
2080 2081
      allow_delay = false;  // Do not delay a single write more than once
      Log(options_.info_log,
2082
          "delaying write %llu usecs for rate limits with max score %.2f\n",
2083
          (long long unsigned int)delayed, score);
2084
      mutex_.Lock();
2085 2086
    } else {
      // Attempt to switch to a new memtable and trigger compaction of old
2087
      DelayLoggingAndReset();
2088 2089 2090 2091 2092
      assert(versions_->PrevLogNumber() == 0);
      uint64_t new_log_number = versions_->NewFileNumber();
      WritableFile* lfile = NULL;
      s = env_->NewWritableFile(LogFileName(dbname_, new_log_number), &lfile);
      if (!s.ok()) {
H
heyongqiang 已提交
2093
        // Avoid chewing through file number space in a tight loop.
A
Abhishek Kona 已提交
2094
  versions_->ReuseFileNumber(new_log_number);
2095 2096 2097 2098 2099
        break;
      }
      delete log_;
      delete logfile_;
      logfile_ = lfile;
2100
      logfile_number_ = new_log_number;
2101
      log_ = new log::Writer(lfile);
2102 2103
      imm_.Add(mem_);
      mem_ = new MemTable(internal_comparator_, NumberLevels());
2104
      mem_->Ref();
2105 2106 2107 2108 2109 2110 2111 2112 2113 2114
      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 已提交
2115 2116 2117 2118 2119 2120 2121 2122 2123 2124
  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();
2125
    if (!ok || (int)level >= NumberLevels()) {
J
jorlow@chromium.org 已提交
2126 2127
      return false;
    } else {
2128
      char buf[100];
D
dgrogan@chromium.org 已提交
2129 2130
      snprintf(buf, sizeof(buf), "%d",
               versions_->NumLevelFiles(static_cast<int>(level)));
2131
      *value = buf;
J
jorlow@chromium.org 已提交
2132 2133
      return true;
    }
2134
  } else if (in == "stats") {
M
Mark Callaghan 已提交
2135 2136 2137 2138 2139 2140
    char buf[1000];
    uint64_t total_bytes = 0;
    uint64_t micros_up = env_->NowMicros() - started_at_;
    double seconds_up = micros_up / 1000000.0;

    // Pardon the long line but I think it is easier to read this way.
2141 2142
    snprintf(buf, sizeof(buf),
             "                               Compactions\n"
M
Mark Callaghan 已提交
2143 2144
             "Level  Files Size(MB) Time(sec)  Read(MB) Write(MB)    Rn(MB)  Rnp1(MB)  Wnew(MB) Amplify Read(MB/s) Write(MB/s)      Rn     Rnp1     Wnp1     NewW    Count\n"
             "------------------------------------------------------------------------------------------------------------------------------------------------------------\n"
2145 2146
             );
    value->append(buf);
2147
    for (int level = 0; level < NumberLevels(); level++) {
2148 2149
      int files = versions_->NumLevelFiles(level);
      if (stats_[level].micros > 0 || files > 0) {
M
Mark Callaghan 已提交
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159
        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;

        total_bytes += bytes_read + stats_[level].bytes_written;
2160 2161
        snprintf(
            buf, sizeof(buf),
M
Mark Callaghan 已提交
2162
            "%3d %8d %8.0f %9.0f %9.0f %9.0f %9.0f %9.0f %9.0f %7.1f %9.1f %11.1f %8d %8d %8d %8d %8d\n",
2163 2164 2165 2166
            level,
            files,
            versions_->NumLevelBytes(level) / 1048576.0,
            stats_[level].micros / 1e6,
M
Mark Callaghan 已提交
2167 2168 2169 2170 2171 2172
            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,
2173 2174 2175
            (bytes_read / 1048576.0) / (stats_[level].micros / 1000000.0),
            (stats_[level].bytes_written / 1048576.0) /
                (stats_[level].micros / 1000000.0),
M
Mark Callaghan 已提交
2176 2177 2178 2179 2180
            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,
            stats_[level].count);
2181 2182 2183
        value->append(buf);
      }
    }
M
Mark Callaghan 已提交
2184 2185

    snprintf(buf, sizeof(buf),
M
Mark Callaghan 已提交
2186
             "Amplification: %.1f rate, %.2f GB in, %.2f GB out, %.2f MB/sec in, %.2f MB/sec out\n",
M
Mark Callaghan 已提交
2187 2188
             (double) total_bytes / stats_[0].bytes_written,
             stats_[0].bytes_written / (1048576.0 * 1024),
M
Mark Callaghan 已提交
2189 2190 2191
             total_bytes / (1048576.0 * 1024),
             stats_[0].bytes_written / 1048576.0 / seconds_up,
             total_bytes / 1048576.0 / seconds_up);
M
Mark Callaghan 已提交
2192 2193 2194 2195 2196 2197 2198
    value->append(buf);

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

    snprintf(buf, sizeof(buf),
            "Stalls(secs): %.3f level0_slowdown, %.3f level0_numfiles, "
2199
            "%.3f memtable_compaction, %.3f leveln_slowdown\n",
M
Mark Callaghan 已提交
2200 2201
            stall_level0_slowdown_ / 1000000.0,
            stall_level0_num_files_ / 1000000.0,
2202 2203
            stall_memtable_compaction_ / 1000000.0,
            stall_leveln_slowdown_ / 1000000.0);
M
Mark Callaghan 已提交
2204 2205
    value->append(buf);

2206
    return true;
G
Gabor Cselle 已提交
2207 2208 2209
  } else if (in == "sstables") {
    *value = versions_->current()->DebugString();
    return true;
J
jorlow@chromium.org 已提交
2210
  }
2211

J
jorlow@chromium.org 已提交
2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240
  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();
  }
}

2241 2242 2243 2244 2245 2246 2247
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 已提交
2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267
// 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);
}

DB::~DB() { }

Status DB::Open(const Options& options, const std::string& dbname,
                DB** dbptr) {
  *dbptr = NULL;

2268 2269 2270 2271
  if (options.block_cache != NULL && options.no_block_cache) {
    return Status::InvalidArgument(
        "no_block_cache is true while block_cache is not NULL");
  }
J
jorlow@chromium.org 已提交
2272
  DBImpl* impl = new DBImpl(options, dbname);
2273 2274 2275 2276 2277
  Status s = impl->CreateArchivalDirectory();
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
2278
  impl->mutex_.Lock();
2279
  VersionEdit edit(impl->NumberLevels());
2280
  s = impl->Recover(&edit); // Handles create_if_missing, error_if_exists
J
jorlow@chromium.org 已提交
2281
  if (s.ok()) {
2282
    uint64_t new_log_number = impl->versions_->NewFileNumber();
J
jorlow@chromium.org 已提交
2283
    WritableFile* lfile;
2284
    s = options.env->NewWritableFile(LogFileName(dbname, new_log_number),
J
jorlow@chromium.org 已提交
2285 2286
                                     &lfile);
    if (s.ok()) {
2287
      edit.SetLogNumber(new_log_number);
J
jorlow@chromium.org 已提交
2288
      impl->logfile_ = lfile;
2289
      impl->logfile_number_ = new_log_number;
J
jorlow@chromium.org 已提交
2290
      impl->log_ = new log::Writer(lfile);
2291
      s = impl->versions_->LogAndApply(&edit, &impl->mutex_);
J
jorlow@chromium.org 已提交
2292 2293 2294
    }
    if (s.ok()) {
      impl->DeleteObsoleteFiles();
2295
      impl->MaybeScheduleCompaction();
2296
      impl->MaybeScheduleLogDBDeployStats();
J
jorlow@chromium.org 已提交
2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307
    }
  }
  impl->mutex_.Unlock();
  if (s.ok()) {
    *dbptr = impl;
  } else {
    delete impl;
  }
  return s;
}

2308 2309 2310
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
Status DestroyDB(const std::string& dbname, const Options& options) {
  Env* env = options.env;
  std::vector<std::string> filenames;
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
  if (filenames.empty()) {
    return Status::OK();
  }

  FileLock* lock;
2321 2322
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2323 2324 2325
  if (result.ok()) {
    uint64_t number;
    FileType type;
D
dgrogan@chromium.org 已提交
2326
    for (size_t i = 0; i < filenames.size(); i++) {
2327
      if (ParseFileName(filenames[i], &number, &type) &&
2328
          type != kDBLockFile) {  // Lock file will be deleted at end
J
jorlow@chromium.org 已提交
2329 2330 2331 2332 2333 2334 2335
        Status del = env->DeleteFile(dbname + "/" + filenames[i]);
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
    env->UnlockFile(lock);  // Ignore error since state is already gone
2336
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2337 2338 2339 2340 2341
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
  }
  return result;
}

2342 2343
//
// A global method that can dump out the build version
2344 2345 2346 2347
void dumpLeveldbBuildVersion(Logger * log) {
  Log(log, "Git sha %s", leveldb_build_git_sha);
  Log(log, "Git datetime %s", leveldb_build_git_datetime);
  Log(log, "Compile time %s %s", leveldb_build_compile_time, leveldb_build_compile_date);
2348 2349
}

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