version_set.cc 50.1 KB
Newer Older
J
jorlow@chromium.org 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
// 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/version_set.h"

#include <algorithm>
#include <stdio.h>
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
#include "db/table_cache.h"
14 15
#include "leveldb/env.h"
#include "leveldb/table_builder.h"
J
jorlow@chromium.org 已提交
16 17 18 19 20 21 22
#include "table/merger.h"
#include "table/two_level_iterator.h"
#include "util/coding.h"
#include "util/logging.h"

namespace leveldb {

G
Gabor Cselle 已提交
23 24 25 26 27 28 29 30
static int64_t TotalFileSize(const std::vector<FileMetaData*>& files) {
  int64_t sum = 0;
  for (size_t i = 0; i < files.size(); i++) {
    sum += files[i]->file_size;
  }
  return sum;
}

J
jorlow@chromium.org 已提交
31 32 33 34 35 36 37 38 39 40 41 42
namespace {
std::string IntSetToString(const std::set<uint64_t>& s) {
  std::string result = "{";
  for (std::set<uint64_t>::const_iterator it = s.begin();
       it != s.end();
       ++it) {
    result += (result.size() > 1) ? "," : "";
    result += NumberToString(*it);
  }
  result += "}";
  return result;
}
H
Hans Wennborg 已提交
43
}  // namespace
J
jorlow@chromium.org 已提交
44 45 46

Version::~Version() {
  assert(refs_ == 0);
47 48 49 50 51 52

  // Remove from linked list
  prev_->next_ = next_;
  next_->prev_ = prev_;

  // Drop references to files
53
  for (int level = 0; level < vset_->NumberLevels(); level++) {
D
dgrogan@chromium.org 已提交
54
    for (size_t i = 0; i < files_[level].size(); i++) {
J
jorlow@chromium.org 已提交
55
      FileMetaData* f = files_[level][i];
56
      assert(f->refs > 0);
J
jorlow@chromium.org 已提交
57 58 59 60 61 62
      f->refs--;
      if (f->refs <= 0) {
        delete f;
      }
    }
  }
63
  delete[] files_;
J
jorlow@chromium.org 已提交
64 65
}

66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
int FindFile(const InternalKeyComparator& icmp,
             const std::vector<FileMetaData*>& files,
             const Slice& key) {
  uint32_t left = 0;
  uint32_t right = files.size();
  while (left < right) {
    uint32_t mid = (left + right) / 2;
    const FileMetaData* f = files[mid];
    if (icmp.InternalKeyComparator::Compare(f->largest.Encode(), key) < 0) {
      // Key at "mid.largest" is < "target".  Therefore all
      // files at or before "mid" are uninteresting.
      left = mid + 1;
    } else {
      // Key at "mid.largest" is >= "target".  Therefore all files
      // after "mid" are uninteresting.
      right = mid;
    }
  }
  return right;
}

G
Gabor Cselle 已提交
87 88 89 90 91 92 93 94 95 96 97 98 99 100
static bool AfterFile(const Comparator* ucmp,
                      const Slice* user_key, const FileMetaData* f) {
  // NULL user_key occurs before all keys and is therefore never after *f
  return (user_key != NULL &&
          ucmp->Compare(*user_key, f->largest.user_key()) > 0);
}

static bool BeforeFile(const Comparator* ucmp,
                       const Slice* user_key, const FileMetaData* f) {
  // NULL user_key occurs after all keys and is therefore never before *f
  return (user_key != NULL &&
          ucmp->Compare(*user_key, f->smallest.user_key()) < 0);
}

101 102
bool SomeFileOverlapsRange(
    const InternalKeyComparator& icmp,
G
Gabor Cselle 已提交
103
    bool disjoint_sorted_files,
104
    const std::vector<FileMetaData*>& files,
G
Gabor Cselle 已提交
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
    const Slice* smallest_user_key,
    const Slice* largest_user_key) {
  const Comparator* ucmp = icmp.user_comparator();
  if (!disjoint_sorted_files) {
    // Need to check against all files
    for (int i = 0; i < files.size(); i++) {
      const FileMetaData* f = files[i];
      if (AfterFile(ucmp, smallest_user_key, f) ||
          BeforeFile(ucmp, largest_user_key, f)) {
        // No overlap
      } else {
        return true;  // Overlap
      }
    }
    return false;
  }

  // Binary search over file list
  uint32_t index = 0;
  if (smallest_user_key != NULL) {
    // Find the earliest possible internal key for smallest_user_key
    InternalKey small(*smallest_user_key, kMaxSequenceNumber,kValueTypeForSeek);
    index = FindFile(icmp, files, small.Encode());
  }

  if (index >= files.size()) {
    // beginning of range is after all files, so no overlap.
    return false;
  }

  return !BeforeFile(ucmp, largest_user_key, files[index]);
136 137
}

J
jorlow@chromium.org 已提交
138 139 140
// An internal iterator.  For a given version/level pair, yields
// information about the files in the level.  For a given entry, key()
// is the largest key that occurs in the file, and value() is an
J
jorlow@chromium.org 已提交
141 142
// 16-byte value containing the file number and file size, both
// encoded using EncodeFixed64.
J
jorlow@chromium.org 已提交
143 144
class Version::LevelFileNumIterator : public Iterator {
 public:
145
  LevelFileNumIterator(const InternalKeyComparator& icmp,
J
jorlow@chromium.org 已提交
146
                       const std::vector<FileMetaData*>* flist)
147
      : icmp_(icmp),
J
jorlow@chromium.org 已提交
148 149 150 151 152 153 154
        flist_(flist),
        index_(flist->size()) {        // Marks as invalid
  }
  virtual bool Valid() const {
    return index_ < flist_->size();
  }
  virtual void Seek(const Slice& target) {
155
    index_ = FindFile(icmp_, *flist_, target);
J
jorlow@chromium.org 已提交
156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
  }
  virtual void SeekToFirst() { index_ = 0; }
  virtual void SeekToLast() {
    index_ = flist_->empty() ? 0 : flist_->size() - 1;
  }
  virtual void Next() {
    assert(Valid());
    index_++;
  }
  virtual void Prev() {
    assert(Valid());
    if (index_ == 0) {
      index_ = flist_->size();  // Marks as invalid
    } else {
      index_--;
    }
  }
  Slice key() const {
    assert(Valid());
    return (*flist_)[index_]->largest.Encode();
  }
  Slice value() const {
    assert(Valid());
    EncodeFixed64(value_buf_, (*flist_)[index_]->number);
J
jorlow@chromium.org 已提交
180
    EncodeFixed64(value_buf_+8, (*flist_)[index_]->file_size);
J
jorlow@chromium.org 已提交
181 182 183 184 185 186
    return Slice(value_buf_, sizeof(value_buf_));
  }
  virtual Status status() const { return Status::OK(); }
 private:
  const InternalKeyComparator icmp_;
  const std::vector<FileMetaData*>* const flist_;
D
dgrogan@chromium.org 已提交
187
  uint32_t index_;
J
jorlow@chromium.org 已提交
188

J
jorlow@chromium.org 已提交
189 190
  // Backing store for value().  Holds the file number and size.
  mutable char value_buf_[16];
J
jorlow@chromium.org 已提交
191 192 193 194 195 196
};

static Iterator* GetFileIterator(void* arg,
                                 const ReadOptions& options,
                                 const Slice& file_value) {
  TableCache* cache = reinterpret_cast<TableCache*>(arg);
J
jorlow@chromium.org 已提交
197
  if (file_value.size() != 16) {
J
jorlow@chromium.org 已提交
198 199 200
    return NewErrorIterator(
        Status::Corruption("FileReader invoked with unexpected value"));
  } else {
J
jorlow@chromium.org 已提交
201 202 203
    return cache->NewIterator(options,
                              DecodeFixed64(file_value.data()),
                              DecodeFixed64(file_value.data() + 8));
J
jorlow@chromium.org 已提交
204 205 206 207 208 209
  }
}

Iterator* Version::NewConcatenatingIterator(const ReadOptions& options,
                                            int level) const {
  return NewTwoLevelIterator(
210
      new LevelFileNumIterator(vset_->icmp_, &files_[level]),
J
jorlow@chromium.org 已提交
211 212 213 214 215 216
      &GetFileIterator, vset_->table_cache_, options);
}

void Version::AddIterators(const ReadOptions& options,
                           std::vector<Iterator*>* iters) {
  // Merge all level zero files together since they may overlap
D
dgrogan@chromium.org 已提交
217
  for (size_t i = 0; i < files_[0].size(); i++) {
J
jorlow@chromium.org 已提交
218
    iters->push_back(
J
jorlow@chromium.org 已提交
219 220
        vset_->table_cache_->NewIterator(
            options, files_[0][i]->number, files_[0][i]->file_size));
J
jorlow@chromium.org 已提交
221 222 223 224 225
  }

  // For levels > 0, we can use a concatenating iterator that sequentially
  // walks through the non-overlapping files in the level, opening them
  // lazily.
226
  for (int level = 1; level < vset_->NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
227 228 229 230 231 232
    if (!files_[level].empty()) {
      iters->push_back(NewConcatenatingIterator(options, level));
    }
  }
}

S
Sanjay Ghemawat 已提交
233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249
// Callback from TableCache::Get()
namespace {
enum SaverState {
  kNotFound,
  kFound,
  kDeleted,
  kCorrupt,
};
struct Saver {
  SaverState state;
  const Comparator* ucmp;
  Slice user_key;
  std::string* value;
};
}
static void SaveValue(void* arg, const Slice& ikey, const Slice& v) {
  Saver* s = reinterpret_cast<Saver*>(arg);
250
  ParsedInternalKey parsed_key;
S
Sanjay Ghemawat 已提交
251 252 253 254 255 256 257 258
  if (!ParseInternalKey(ikey, &parsed_key)) {
    s->state = kCorrupt;
  } else {
    if (s->ucmp->Compare(parsed_key.user_key, s->user_key) == 0) {
      s->state = (parsed_key.type == kTypeValue) ? kFound : kDeleted;
      if (s->state == kFound) {
        s->value->assign(v.data(), v.size());
      }
259 260 261 262 263 264 265 266
    }
  }
}

static bool NewestFirst(FileMetaData* a, FileMetaData* b) {
  return a->number > b->number;
}

267 268 269 270 271 272 273 274 275
Version::Version(VersionSet* vset)
    : vset_(vset), next_(this), prev_(this), refs_(0),
      file_to_compact_(NULL),
      file_to_compact_level_(-1),
      compaction_score_(-1),
      compaction_level_(-1) {
  files_ = new std::vector<FileMetaData*>[vset->NumberLevels()];
}

276 277 278 279 280 281 282 283 284 285 286 287
Status Version::Get(const ReadOptions& options,
                    const LookupKey& k,
                    std::string* value,
                    GetStats* stats) {
  Slice ikey = k.internal_key();
  Slice user_key = k.user_key();
  const Comparator* ucmp = vset_->icmp_.user_comparator();
  Status s;

  stats->seek_file = NULL;
  stats->seek_file_level = -1;
  FileMetaData* last_file_read = NULL;
288
  int last_file_read_level = -1;
289 290 291 292 293 294

  // We can search level-by-level since entries never hop across
  // levels.  Therefore we are guaranteed that if we find data
  // in an smaller level, later levels are irrelevant.
  std::vector<FileMetaData*> tmp;
  FileMetaData* tmp2;
295
  for (int level = 0; level < vset_->NumberLevels(); level++) {
296 297 298 299 300 301 302 303 304
    size_t num_files = files_[level].size();
    if (num_files == 0) continue;

    // Get the list of files to search in this level
    FileMetaData* const* files = &files_[level][0];
    if (level == 0) {
      // Level-0 files may overlap each other.  Find all files that
      // overlap user_key and process them in order from newest to oldest.
      tmp.reserve(num_files);
D
dgrogan@chromium.org 已提交
305
      for (uint32_t i = 0; i < num_files; i++) {
306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335
        FileMetaData* f = files[i];
        if (ucmp->Compare(user_key, f->smallest.user_key()) >= 0 &&
            ucmp->Compare(user_key, f->largest.user_key()) <= 0) {
          tmp.push_back(f);
        }
      }
      if (tmp.empty()) continue;

      std::sort(tmp.begin(), tmp.end(), NewestFirst);
      files = &tmp[0];
      num_files = tmp.size();
    } else {
      // Binary search to find earliest index whose largest key >= ikey.
      uint32_t index = FindFile(vset_->icmp_, files_[level], ikey);
      if (index >= num_files) {
        files = NULL;
        num_files = 0;
      } else {
        tmp2 = files[index];
        if (ucmp->Compare(user_key, tmp2->smallest.user_key()) < 0) {
          // All of "tmp2" is past any data for user_key
          files = NULL;
          num_files = 0;
        } else {
          files = &tmp2;
          num_files = 1;
        }
      }
    }

D
dgrogan@chromium.org 已提交
336
    for (uint32_t i = 0; i < num_files; ++i) {
337 338 339
      if (last_file_read != NULL && stats->seek_file == NULL) {
        // We have had more than one seek for this read.  Charge the 1st file.
        stats->seek_file = last_file_read;
340
        stats->seek_file_level = last_file_read_level;
341 342 343 344
      }

      FileMetaData* f = files[i];
      last_file_read = f;
345
      last_file_read_level = level;
346

S
Sanjay Ghemawat 已提交
347 348 349 350 351 352 353 354
      Saver saver;
      saver.state = kNotFound;
      saver.ucmp = ucmp;
      saver.user_key = user_key;
      saver.value = value;
      s = vset_->table_cache_->Get(options, f->number, f->file_size,
                                   ikey, &saver, SaveValue);
      if (!s.ok()) {
355
        return s;
S
Sanjay Ghemawat 已提交
356 357 358 359 360 361 362 363 364 365 366
      }
      switch (saver.state) {
        case kNotFound:
          break;      // Keep searching in other files
        case kFound:
          return s;
        case kDeleted:
          s = Status::NotFound(Slice());  // Use empty error message for speed
          return s;
        case kCorrupt:
          s = Status::Corruption("corrupted key for ", user_key);
367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
          return s;
      }
    }
  }

  return Status::NotFound(Slice());  // Use an empty error message for speed
}

bool Version::UpdateStats(const GetStats& stats) {
  FileMetaData* f = stats.seek_file;
  if (f != NULL) {
    f->allowed_seeks--;
    if (f->allowed_seeks <= 0 && file_to_compact_ == NULL) {
      file_to_compact_ = f;
      file_to_compact_level_ = stats.seek_file_level;
      return true;
    }
  }
  return false;
}

J
jorlow@chromium.org 已提交
388 389 390 391 392
void Version::Ref() {
  ++refs_;
}

void Version::Unref() {
393
  assert(this != &vset_->dummy_versions_);
J
jorlow@chromium.org 已提交
394 395 396
  assert(refs_ >= 1);
  --refs_;
  if (refs_ == 0) {
397
    delete this;
J
jorlow@chromium.org 已提交
398 399 400
  }
}

401
bool Version::OverlapInLevel(int level,
G
Gabor Cselle 已提交
402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
                             const Slice* smallest_user_key,
                             const Slice* largest_user_key) {
  return SomeFileOverlapsRange(vset_->icmp_, (level > 0), files_[level],
                               smallest_user_key, largest_user_key);
}

int Version::PickLevelForMemTableOutput(
    const Slice& smallest_user_key,
    const Slice& largest_user_key) {
  int level = 0;
  if (!OverlapInLevel(0, &smallest_user_key, &largest_user_key)) {
    // Push to next level if there is no overlap in next level,
    // and the #bytes overlapping in the level after that are limited.
    InternalKey start(smallest_user_key, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey limit(largest_user_key, 0, static_cast<ValueType>(0));
    std::vector<FileMetaData*> overlaps;
418 419
    int max_mem_compact_level = vset_->options_->max_mem_compaction_level;
    while (max_mem_compact_level > 0 && level < max_mem_compact_level) {
G
Gabor Cselle 已提交
420 421 422
      if (OverlapInLevel(level + 1, &smallest_user_key, &largest_user_key)) {
        break;
      }
423 424 425 426
      if (level + 2 >= vset_->NumberLevels()) {
	level++;
	break;
      }
G
Gabor Cselle 已提交
427 428
      GetOverlappingInputs(level + 2, &start, &limit, &overlaps);
      const int64_t sum = TotalFileSize(overlaps);
429
      if (sum > vset_->MaxGrandParentOverlapBytes(level)) {
G
Gabor Cselle 已提交
430 431 432 433 434
        break;
      }
      level++;
    }
  }
435

G
Gabor Cselle 已提交
436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453
  return level;
}

// Store in "*inputs" all files in "level" that overlap [begin,end]
void Version::GetOverlappingInputs(
    int level,
    const InternalKey* begin,
    const InternalKey* end,
    std::vector<FileMetaData*>* inputs) {
  inputs->clear();
  Slice user_begin, user_end;
  if (begin != NULL) {
    user_begin = begin->user_key();
  }
  if (end != NULL) {
    user_end = end->user_key();
  }
  const Comparator* user_cmp = vset_->icmp_.user_comparator();
H
Hans Wennborg 已提交
454 455 456 457 458
  for (size_t i = 0; i < files_[level].size(); ) {
    FileMetaData* f = files_[level][i++];
    const Slice file_start = f->smallest.user_key();
    const Slice file_limit = f->largest.user_key();
    if (begin != NULL && user_cmp->Compare(file_limit, user_begin) < 0) {
G
Gabor Cselle 已提交
459
      // "f" is completely before specified range; skip it
H
Hans Wennborg 已提交
460
    } else if (end != NULL && user_cmp->Compare(file_start, user_end) > 0) {
G
Gabor Cselle 已提交
461 462 463
      // "f" is completely after specified range; skip it
    } else {
      inputs->push_back(f);
H
Hans Wennborg 已提交
464 465 466 467 468 469 470 471 472 473 474 475 476
      if (level == 0) {
        // Level-0 files may overlap each other.  So check if the newly
        // added file has expanded the range.  If so, restart search.
        if (begin != NULL && user_cmp->Compare(file_start, user_begin) < 0) {
          user_begin = file_start;
          inputs->clear();
          i = 0;
        } else if (end != NULL && user_cmp->Compare(file_limit, user_end) > 0) {
          user_end = file_limit;
          inputs->clear();
          i = 0;
        }
      }
G
Gabor Cselle 已提交
477 478
    }
  }
479 480
}

J
jorlow@chromium.org 已提交
481 482
std::string Version::DebugString() const {
  std::string r;
483
  for (int level = 0; level < vset_->NumberLevels(); level++) {
484 485 486 487 488
    // E.g.,
    //   --- level 1 ---
    //   17:123['a' .. 'd']
    //   20:43['e' .. 'g']
    r.append("--- level ");
J
jorlow@chromium.org 已提交
489
    AppendNumberTo(&r, level);
490
    r.append(" ---\n");
J
jorlow@chromium.org 已提交
491
    const std::vector<FileMetaData*>& files = files_[level];
D
dgrogan@chromium.org 已提交
492
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
493 494 495 496
      r.push_back(' ');
      AppendNumberTo(&r, files[i]->number);
      r.push_back(':');
      AppendNumberTo(&r, files[i]->file_size);
G
Gabor Cselle 已提交
497 498 499 500 501
      r.append("[");
      r.append(files[i]->smallest.DebugString());
      r.append(" .. ");
      r.append(files[i]->largest.DebugString());
      r.append("]\n");
J
jorlow@chromium.org 已提交
502 503 504 505 506 507 508 509 510 511
    }
  }
  return r;
}

// A helper class so we can efficiently apply a whole sequence
// of edits to a particular state without creating intermediate
// Versions that contain full copies of the intermediate state.
class VersionSet::Builder {
 private:
512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
  // Helper to sort by v->files_[file_number].smallest
  struct BySmallestKey {
    const InternalKeyComparator* internal_comparator;

    bool operator()(FileMetaData* f1, FileMetaData* f2) const {
      int r = internal_comparator->Compare(f1->smallest, f2->smallest);
      if (r != 0) {
        return (r < 0);
      } else {
        // Break ties by file number
        return (f1->number < f2->number);
      }
    }
  };

  typedef std::set<FileMetaData*, BySmallestKey> FileSet;
  struct LevelState {
    std::set<uint64_t> deleted_files;
    FileSet* added_files;
  };

J
jorlow@chromium.org 已提交
533
  VersionSet* vset_;
534
  Version* base_;
535
  LevelState* levels_;
J
jorlow@chromium.org 已提交
536 537 538 539

 public:
  // Initialize a builder with the files from *base and other info from *vset
  Builder(VersionSet* vset, Version* base)
540 541 542
      : vset_(vset),
        base_(base) {
    base_->Ref();
543
    levels_ = new LevelState[vset_->NumberLevels()];
544 545
    BySmallestKey cmp;
    cmp.internal_comparator = &vset_->icmp_;
546
    for (int level = 0; level < vset_->NumberLevels(); level++) {
547
      levels_[level].added_files = new FileSet(cmp);
J
jorlow@chromium.org 已提交
548 549 550 551
    }
  }

  ~Builder() {
552
    for (int level = 0; level < vset_->NumberLevels(); level++) {
553 554 555 556 557 558 559 560
      const FileSet* added = levels_[level].added_files;
      std::vector<FileMetaData*> to_unref;
      to_unref.reserve(added->size());
      for (FileSet::const_iterator it = added->begin();
          it != added->end(); ++it) {
        to_unref.push_back(*it);
      }
      delete added;
D
dgrogan@chromium.org 已提交
561
      for (uint32_t i = 0; i < to_unref.size(); i++) {
562
        FileMetaData* f = to_unref[i];
J
jorlow@chromium.org 已提交
563 564 565 566 567 568
        f->refs--;
        if (f->refs <= 0) {
          delete f;
        }
      }
    }
569
    delete[] levels_;
570
    base_->Unref();
J
jorlow@chromium.org 已提交
571 572 573 574 575
  }

  // Apply all of the edits in *edit to the current state.
  void Apply(VersionEdit* edit) {
    // Update compaction pointers
D
dgrogan@chromium.org 已提交
576
    for (size_t i = 0; i < edit->compact_pointers_.size(); i++) {
J
jorlow@chromium.org 已提交
577 578 579 580 581 582 583 584 585 586 587 588
      const int level = edit->compact_pointers_[i].first;
      vset_->compact_pointer_[level] =
          edit->compact_pointers_[i].second.Encode().ToString();
    }

    // Delete files
    const VersionEdit::DeletedFileSet& del = edit->deleted_files_;
    for (VersionEdit::DeletedFileSet::const_iterator iter = del.begin();
         iter != del.end();
         ++iter) {
      const int level = iter->first;
      const uint64_t number = iter->second;
589
      levels_[level].deleted_files.insert(number);
J
jorlow@chromium.org 已提交
590 591 592
    }

    // Add new files
D
dgrogan@chromium.org 已提交
593
    for (size_t i = 0; i < edit->new_files_.size(); i++) {
J
jorlow@chromium.org 已提交
594 595 596
      const int level = edit->new_files_[i].first;
      FileMetaData* f = new FileMetaData(edit->new_files_[i].second);
      f->refs = 1;
597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613

      // We arrange to automatically compact this file after
      // a certain number of seeks.  Let's assume:
      //   (1) One seek costs 10ms
      //   (2) Writing or reading 1MB costs 10ms (100MB/s)
      //   (3) A compaction of 1MB does 25MB of IO:
      //         1MB read from this level
      //         10-12MB read from next level (boundaries may be misaligned)
      //         10-12MB written to next level
      // This implies that 25 seeks cost the same as the compaction
      // of 1MB of data.  I.e., one seek costs approximately the
      // same as the compaction of 40KB of data.  We are a little
      // conservative and allow approximately one seek for every 16KB
      // of data before triggering a compaction.
      f->allowed_seeks = (f->file_size / 16384);
      if (f->allowed_seeks < 100) f->allowed_seeks = 100;

614 615
      levels_[level].deleted_files.erase(f->number);
      levels_[level].added_files->insert(f);
J
jorlow@chromium.org 已提交
616 617 618 619 620
    }
  }

  // Save the current state in *v.
  void SaveTo(Version* v) {
621 622
    BySmallestKey cmp;
    cmp.internal_comparator = &vset_->icmp_;
623
    for (int level = 0; level < vset_->NumberLevels(); level++) {
624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647
      // Merge the set of added files with the set of pre-existing files.
      // Drop any deleted files.  Store the result in *v.
      const std::vector<FileMetaData*>& base_files = base_->files_[level];
      std::vector<FileMetaData*>::const_iterator base_iter = base_files.begin();
      std::vector<FileMetaData*>::const_iterator base_end = base_files.end();
      const FileSet* added = levels_[level].added_files;
      v->files_[level].reserve(base_files.size() + added->size());
      for (FileSet::const_iterator added_iter = added->begin();
           added_iter != added->end();
           ++added_iter) {
        // Add all smaller files listed in base_
        for (std::vector<FileMetaData*>::const_iterator bpos
                 = std::upper_bound(base_iter, base_end, *added_iter, cmp);
             base_iter != bpos;
             ++base_iter) {
          MaybeAddFile(v, level, *base_iter);
        }

        MaybeAddFile(v, level, *added_iter);
      }

      // Add remaining base files
      for (; base_iter != base_end; ++base_iter) {
        MaybeAddFile(v, level, *base_iter);
J
jorlow@chromium.org 已提交
648
      }
649 650 651 652

#ifndef NDEBUG
      // Make sure there is no overlap in levels > 0
      if (level > 0) {
D
dgrogan@chromium.org 已提交
653
        for (uint32_t i = 1; i < v->files_[level].size(); i++) {
654 655 656 657
          const InternalKey& prev_end = v->files_[level][i-1]->largest;
          const InternalKey& this_begin = v->files_[level][i]->smallest;
          if (vset_->icmp_.Compare(prev_end, this_begin) >= 0) {
            fprintf(stderr, "overlapping ranges in same level %s vs. %s\n",
G
Gabor Cselle 已提交
658 659
                    prev_end.DebugString().c_str(),
                    this_begin.DebugString().c_str());
660 661 662 663 664 665 666 667 668 669 670 671
            abort();
          }
        }
      }
#endif
    }
  }

  void MaybeAddFile(Version* v, int level, FileMetaData* f) {
    if (levels_[level].deleted_files.count(f->number) > 0) {
      // File is deleted: do nothing
    } else {
672 673 674 675 676 677
      std::vector<FileMetaData*>* files = &v->files_[level];
      if (level > 0 && !files->empty()) {
        // Must not overlap
        assert(vset_->icmp_.Compare((*files)[files->size()-1]->largest,
                                    f->smallest) < 0);
      }
678
      f->refs++;
679
      files->push_back(f);
J
jorlow@chromium.org 已提交
680 681 682 683 684 685 686 687 688 689 690 691 692 693 694
    }
  }
};

VersionSet::VersionSet(const std::string& dbname,
                       const Options* options,
                       TableCache* table_cache,
                       const InternalKeyComparator* cmp)
    : env_(options->env),
      dbname_(dbname),
      options_(options),
      table_cache_(table_cache),
      icmp_(*cmp),
      next_file_number_(2),
      manifest_file_number_(0),  // Filled by Recover()
695 696 697
      last_sequence_(0),
      log_number_(0),
      prev_log_number_(0),
J
jorlow@chromium.org 已提交
698 699
      descriptor_file_(NULL),
      descriptor_log_(NULL),
700 701
      dummy_versions_(this),
      current_(NULL) {
702 703 704 705 706
	compact_pointer_ = new std::string[options_->num_levels];
	max_file_size_ = new uint64_t[options_->num_levels];
	level_max_bytes_ = new uint64_t[options->num_levels];
	int target_file_size_multiplier = options_->target_file_size_multiplier;
	int max_bytes_multiplier = options_->max_bytes_for_level_multiplier;
707 708 709 710 711 712 713 714
	for (int i = 0; i < options_->num_levels; i++) {
	  if (i > 1) {
	    max_file_size_[i] = max_file_size_[i-1] * target_file_size_multiplier;
	    level_max_bytes_[i] = level_max_bytes_[i-1] * max_bytes_multiplier;
	  } else {
	    max_file_size_[i] = options_->target_file_size_base;
	    level_max_bytes_[i] = options_->max_bytes_for_level_base;
	  }
715
	}
716
  AppendVersion(new Version(this));
J
jorlow@chromium.org 已提交
717 718 719
}

VersionSet::~VersionSet() {
720 721
  current_->Unref();
  assert(dummy_versions_.next_ == &dummy_versions_);  // List must be empty
722 723 724
  delete[] compact_pointer_;
  delete[] max_file_size_;
  delete[] level_max_bytes_;
J
jorlow@chromium.org 已提交
725 726 727 728
  delete descriptor_log_;
  delete descriptor_file_;
}

729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745
void VersionSet::AppendVersion(Version* v) {
  // Make "v" current
  assert(v->refs_ == 0);
  assert(v != current_);
  if (current_ != NULL) {
    current_->Unref();
  }
  current_ = v;
  v->Ref();

  // Append to linked list
  v->prev_ = dummy_versions_.prev_;
  v->next_ = &dummy_versions_;
  v->prev_->next_ = v;
  v->next_->prev_ = v;
}

746
Status VersionSet::LogAndApply(VersionEdit* edit, port::Mutex* mu) {
747 748 749 750 751 752 753 754 755 756 757
  if (edit->has_log_number_) {
    assert(edit->log_number_ >= log_number_);
    assert(edit->log_number_ < next_file_number_);
  } else {
    edit->SetLogNumber(log_number_);
  }

  if (!edit->has_prev_log_number_) {
    edit->SetPrevLogNumber(prev_log_number_);
  }

J
jorlow@chromium.org 已提交
758
  edit->SetNextFile(next_file_number_);
759
  edit->SetLastSequence(last_sequence_);
J
jorlow@chromium.org 已提交
760 761 762 763 764 765 766

  Version* v = new Version(this);
  {
    Builder builder(this, current_);
    builder.Apply(edit);
    builder.SaveTo(v);
  }
767
  Finalize(v);
J
jorlow@chromium.org 已提交
768 769 770

  // Initialize new descriptor log file if necessary by creating
  // a temporary file that contains a snapshot of the current version.
771 772 773
  std::string new_manifest_file;
  Status s;
  if (descriptor_log_ == NULL) {
774 775
    // No reason to unlock *mu here since we only hit this path in the
    // first call to LogAndApply (when opening the database).
776 777 778 779 780 781 782
    assert(descriptor_file_ == NULL);
    new_manifest_file = DescriptorFileName(dbname_, manifest_file_number_);
    edit->SetNextFile(next_file_number_);
    s = env_->NewWritableFile(new_manifest_file, &descriptor_file_);
    if (s.ok()) {
      descriptor_log_ = new log::Writer(descriptor_file_);
      s = WriteSnapshot(descriptor_log_);
J
jorlow@chromium.org 已提交
783 784 785
    }
  }

786 787 788 789 790
  // Unlock during expensive MANIFEST log write
  {
    mu->Unlock();

    // Write new record to MANIFEST log
J
jorlow@chromium.org 已提交
791
    if (s.ok()) {
792 793 794 795
      std::string record;
      edit->EncodeTo(&record);
      s = descriptor_log_->AddRecord(record);
      if (s.ok()) {
796 797 798 799 800
        if (options_->use_fsync) {
          s = descriptor_file_->Fsync();
        } else {
          s = descriptor_file_->Sync();
        }
801
      }
J
jorlow@chromium.org 已提交
802 803
    }

804 805 806 807 808 809 810
    // If we just created a new descriptor file, install it by writing a
    // new CURRENT file that points to it.
    if (s.ok() && !new_manifest_file.empty()) {
      s = SetCurrentFile(env_, dbname_, manifest_file_number_);
    }

    mu->Lock();
J
jorlow@chromium.org 已提交
811 812 813 814
  }

  // Install the new version
  if (s.ok()) {
815
    AppendVersion(v);
816 817
    log_number_ = edit->log_number_;
    prev_log_number_ = edit->prev_log_number_;
J
jorlow@chromium.org 已提交
818 819 820 821 822 823 824 825 826 827 828 829 830 831
  } else {
    delete v;
    if (!new_manifest_file.empty()) {
      delete descriptor_log_;
      delete descriptor_file_;
      descriptor_log_ = NULL;
      descriptor_file_ = NULL;
      env_->DeleteFile(new_manifest_file);
    }
  }

  return s;
}

832
Status VersionSet::Recover() {
J
jorlow@chromium.org 已提交
833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
  struct LogReporter : public log::Reader::Reporter {
    Status* status;
    virtual void Corruption(size_t bytes, const Status& s) {
      if (this->status->ok()) *this->status = s;
    }
  };

  // Read "CURRENT" file, which contains a pointer to the current manifest file
  std::string current;
  Status s = ReadFileToString(env_, CurrentFileName(dbname_), &current);
  if (!s.ok()) {
    return s;
  }
  if (current.empty() || current[current.size()-1] != '\n') {
    return Status::Corruption("CURRENT file does not end with newline");
  }
  current.resize(current.size() - 1);

H
heyongqiang 已提交
851 852 853
  Log(options_->info_log, "Recovering from manifest file:%s\n",
      current.c_str());

J
jorlow@chromium.org 已提交
854 855 856 857 858 859 860 861
  std::string dscname = dbname_ + "/" + current;
  SequentialFile* file;
  s = env_->NewSequentialFile(dscname, &file);
  if (!s.ok()) {
    return s;
  }

  bool have_log_number = false;
862
  bool have_prev_log_number = false;
J
jorlow@chromium.org 已提交
863 864 865
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
866 867 868
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
  uint64_t prev_log_number = 0;
J
jorlow@chromium.org 已提交
869 870 871 872 873
  Builder builder(this, current_);

  {
    LogReporter reporter;
    reporter.status = &s;
874
    log::Reader reader(file, &reporter, true/*checksum*/, 0/*initial_offset*/);
J
jorlow@chromium.org 已提交
875 876 877
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
878
      VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
879 880 881 882 883 884 885 886 887 888 889 890 891 892 893
      s = edit.DecodeFrom(record);
      if (s.ok()) {
        if (edit.has_comparator_ &&
            edit.comparator_ != icmp_.user_comparator()->Name()) {
          s = Status::InvalidArgument(
              edit.comparator_ + "does not match existing comparator ",
              icmp_.user_comparator()->Name());
        }
      }

      if (s.ok()) {
        builder.Apply(&edit);
      }

      if (edit.has_log_number_) {
894
        log_number = edit.log_number_;
J
jorlow@chromium.org 已提交
895 896 897
        have_log_number = true;
      }

898 899 900 901 902
      if (edit.has_prev_log_number_) {
        prev_log_number = edit.prev_log_number_;
        have_prev_log_number = true;
      }

J
jorlow@chromium.org 已提交
903 904 905 906 907 908
      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

      if (edit.has_last_sequence_) {
909
        last_sequence = edit.last_sequence_;
J
jorlow@chromium.org 已提交
910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
        have_last_sequence = true;
      }
    }
  }
  delete file;
  file = NULL;

  if (s.ok()) {
    if (!have_next_file) {
      s = Status::Corruption("no meta-nextfile entry in descriptor");
    } else if (!have_log_number) {
      s = Status::Corruption("no meta-lognumber entry in descriptor");
    } else if (!have_last_sequence) {
      s = Status::Corruption("no last-sequence-number entry in descriptor");
    }
925 926 927 928

    if (!have_prev_log_number) {
      prev_log_number = 0;
    }
929 930 931

    MarkFileNumberUsed(prev_log_number);
    MarkFileNumberUsed(log_number);
J
jorlow@chromium.org 已提交
932 933 934 935 936
  }

  if (s.ok()) {
    Version* v = new Version(this);
    builder.SaveTo(v);
937 938 939 940 941 942 943 944
    // Install recovered version
    Finalize(v);
    AppendVersion(v);
    manifest_file_number_ = next_file;
    next_file_number_ = next_file + 1;
    last_sequence_ = last_sequence;
    log_number_ = log_number;
    prev_log_number_ = prev_log_number;
H
heyongqiang 已提交
945

H
heyongqiang 已提交
946
    Log(options_->info_log, "Recovered from manifest file:%s succeeded,"
H
heyongqiang 已提交
947 948 949 950 951
        "manifest_file_number is %lld, next_file_number is %lld, "
        "last_sequence is %lld, log_number is %lld,"
        "prev_log_number is %lld\n",
        current.c_str(), manifest_file_number_, next_file_number_,
        last_sequence_, log_number_, prev_log_number_);
J
jorlow@chromium.org 已提交
952 953 954 955 956
  }

  return s;
}

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 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058
Status VersionSet::DumpManifest(Options& options, std::string& dscname) {
  struct LogReporter : public log::Reader::Reporter {
    Status* status;
    virtual void Corruption(size_t bytes, const Status& s) {
      if (this->status->ok()) *this->status = s;
    }
  };

  // Open the specified manifest file.
  SequentialFile* file;
  Status s = options.env->NewSequentialFile(dscname, &file);
  if (!s.ok()) {
    return s;
  }

  bool have_log_number = false;
  bool have_prev_log_number = false;
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
  uint64_t prev_log_number = 0;
  VersionSet::Builder builder(this, current_);

  {
    LogReporter reporter;
    reporter.status = &s;
    log::Reader reader(file, &reporter, true/*checksum*/, 0/*initial_offset*/);
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
      VersionEdit edit(NumberLevels());
      s = edit.DecodeFrom(record);
      if (s.ok()) {
        if (edit.has_comparator_ &&
            edit.comparator_ != icmp_.user_comparator()->Name()) {
          s = Status::InvalidArgument(
              edit.comparator_ + "does not match existing comparator ",
              icmp_.user_comparator()->Name());
        }
      }

      if (s.ok()) {
        builder.Apply(&edit);
      }

      if (edit.has_log_number_) {
        log_number = edit.log_number_;
        have_log_number = true;
      }

      if (edit.has_prev_log_number_) {
        prev_log_number = edit.prev_log_number_;
        have_prev_log_number = true;
      }

      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

      if (edit.has_last_sequence_) {
        last_sequence = edit.last_sequence_;
        have_last_sequence = true;
      }
    }
  }
  delete file;
  file = NULL;

  if (s.ok()) {
    if (!have_next_file) {
      s = Status::Corruption("no meta-nextfile entry in descriptor");
      printf("no meta-nextfile entry in descriptor");
    } else if (!have_log_number) {
      s = Status::Corruption("no meta-lognumber entry in descriptor");
      printf("no meta-lognumber entry in descriptor");
    } else if (!have_last_sequence) {
      printf("no last-sequence-number entry in descriptor");
      s = Status::Corruption("no last-sequence-number entry in descriptor");
    }

    if (!have_prev_log_number) {
      prev_log_number = 0;
    }

    MarkFileNumberUsed(prev_log_number);
    MarkFileNumberUsed(log_number);
  }

  if (s.ok()) {
    Version* v = new Version(this);
    builder.SaveTo(v);
    // Install recovered version
    Finalize(v);
    AppendVersion(v);
    manifest_file_number_ = next_file;
    next_file_number_ = next_file + 1;
    last_sequence_ = last_sequence;
    log_number_ = log_number;
    prev_log_number_ = prev_log_number;
1059

1060
    printf("manifest_file_number %d next_file_number %d last_sequence %d log_number %d  prev_log_number %d\n",
1061
           manifest_file_number_, next_file_number_,
1062 1063 1064
           last_sequence, log_number, prev_log_number);
    printf("%s \n", v->DebugString().c_str());
  }
1065

1066 1067 1068 1069

  return s;
}

1070 1071 1072 1073 1074 1075
void VersionSet::MarkFileNumberUsed(uint64_t number) {
  if (next_file_number_ <= number) {
    next_file_number_ = number + 1;
  }
}

1076
void VersionSet::Finalize(Version* v) {
J
jorlow@chromium.org 已提交
1077 1078 1079 1080
  // Precomputed best level for next compaction
  int best_level = -1;
  double best_score = -1;

1081
  for (int level = 0; level < NumberLevels()-1; level++) {
1082
    double score;
J
jorlow@chromium.org 已提交
1083
    if (level == 0) {
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094
      // We treat level-0 specially by bounding the number of files
      // instead of number of bytes for two reasons:
      //
      // (1) With larger write-buffer sizes, it is nice not to do too
      // many level-0 compactions.
      //
      // (2) The files in level-0 are merged on every read and
      // therefore we wish to avoid too many files when the individual
      // file size is small (perhaps because of a small write-buffer
      // setting, or very high compression ratios, or lots of
      // overwrites/deletions).
1095
      score = v->files_[level].size() /
1096
          static_cast<double>(options_->level0_file_num_compaction_trigger);
1097 1098 1099 1100
    } else {
      // Compute the ratio of current size to size limit.
      const uint64_t level_bytes = TotalFileSize(v->files_[level]);
      score = static_cast<double>(level_bytes) / MaxBytesForLevel(level);
J
jorlow@chromium.org 已提交
1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
    }

    if (score > best_score) {
      best_level = level;
      best_score = score;
    }
  }

  v->compaction_level_ = best_level;
  v->compaction_score_ = best_score;
}

Status VersionSet::WriteSnapshot(log::Writer* log) {
  // TODO: Break up into multiple records to reduce memory usage on recovery?

  // Save metadata
1117
  VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
1118 1119 1120
  edit.SetComparatorName(icmp_.user_comparator()->Name());

  // Save compaction pointers
1121
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1122 1123 1124 1125 1126 1127 1128 1129
    if (!compact_pointer_[level].empty()) {
      InternalKey key;
      key.DecodeFrom(compact_pointer_[level]);
      edit.SetCompactPointer(level, key);
    }
  }

  // Save files
1130
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1131
    const std::vector<FileMetaData*>& files = current_->files_[level];
D
dgrogan@chromium.org 已提交
1132
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144
      const FileMetaData* f = files[i];
      edit.AddFile(level, f->number, f->file_size, f->smallest, f->largest);
    }
  }

  std::string record;
  edit.EncodeTo(&record);
  return log->AddRecord(record);
}

int VersionSet::NumLevelFiles(int level) const {
  assert(level >= 0);
1145
  assert(level < NumberLevels());
J
jorlow@chromium.org 已提交
1146 1147 1148
  return current_->files_[level].size();
}

1149
const char* VersionSet::LevelSummary(LevelSummaryStorage* scratch) const {
1150 1151 1152 1153 1154 1155 1156 1157 1158 1159
  int len = snprintf(scratch->buffer, sizeof(scratch->buffer), "files[");
  for (int i = 0; i < NumberLevels(); i++) {
    int sz = sizeof(scratch->buffer) - len;
    int ret = snprintf(scratch->buffer + len, sz, "%d ",
        int(current_->files_[i].size()));
    if (ret < 0 || ret >= sz)
      break;
    len += ret;
  }
  snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "]");
1160 1161 1162
  return scratch->buffer;
}

1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
const char* VersionSet::LevelDataSizeSummary(
    LevelSummaryStorage* scratch) const {
  int len = snprintf(scratch->buffer, sizeof(scratch->buffer), "files_size[");
  for (int i = 0; i < NumberLevels(); i++) {
    int sz = sizeof(scratch->buffer) - len;
    int ret = snprintf(scratch->buffer + len, sz, "%ld ",
        NumLevelBytes(i));
    if (ret < 0 || ret >= sz)
      break;
    len += ret;
  }
  snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "]");
  return scratch->buffer;
}

J
jorlow@chromium.org 已提交
1178 1179
uint64_t VersionSet::ApproximateOffsetOf(Version* v, const InternalKey& ikey) {
  uint64_t result = 0;
1180
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1181
    const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
1182
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198
      if (icmp_.Compare(files[i]->largest, ikey) <= 0) {
        // Entire file is before "ikey", so just add the file size
        result += files[i]->file_size;
      } else if (icmp_.Compare(files[i]->smallest, ikey) > 0) {
        // Entire file is after "ikey", so ignore
        if (level > 0) {
          // Files other than level 0 are sorted by meta->smallest, so
          // no further files in this level will contain data for
          // "ikey".
          break;
        }
      } else {
        // "ikey" falls in the range for this table.  Add the
        // approximate offset of "ikey" within the table.
        Table* tableptr;
        Iterator* iter = table_cache_->NewIterator(
J
jorlow@chromium.org 已提交
1199
            ReadOptions(), files[i]->number, files[i]->file_size, &tableptr);
J
jorlow@chromium.org 已提交
1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210
        if (tableptr != NULL) {
          result += tableptr->ApproximateOffsetOf(ikey.Encode());
        }
        delete iter;
      }
    }
  }
  return result;
}

void VersionSet::AddLiveFiles(std::set<uint64_t>* live) {
1211 1212 1213
  for (Version* v = dummy_versions_.next_;
       v != &dummy_versions_;
       v = v->next_) {
1214
    for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1215
      const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
1216
      for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1217 1218 1219 1220 1221 1222
        live->insert(files[i]->number);
      }
    }
  }
}

1223 1224
int64_t VersionSet::NumLevelBytes(int level) const {
  assert(level >= 0);
1225
  assert(level < NumberLevels());
1226
  return TotalFileSize(current_->files_[level]);
J
jorlow@chromium.org 已提交
1227 1228 1229 1230
}

int64_t VersionSet::MaxNextLevelOverlappingBytes() {
  int64_t result = 0;
1231
  std::vector<FileMetaData*> overlaps;
1232
  for (int level = 1; level < NumberLevels() - 1; level++) {
D
dgrogan@chromium.org 已提交
1233
    for (size_t i = 0; i < current_->files_[level].size(); i++) {
1234
      const FileMetaData* f = current_->files_[level][i];
G
Gabor Cselle 已提交
1235 1236
      current_->GetOverlappingInputs(level+1, &f->smallest, &f->largest,
                                     &overlaps);
J
jorlow@chromium.org 已提交
1237
      const int64_t sum = TotalFileSize(overlaps);
1238 1239 1240 1241 1242 1243 1244 1245
      if (sum > result) {
        result = sum;
      }
    }
  }
  return result;
}

J
jorlow@chromium.org 已提交
1246 1247 1248 1249 1250 1251 1252 1253 1254
// Stores the minimal range that covers all entries in inputs in
// *smallest, *largest.
// REQUIRES: inputs is not empty
void VersionSet::GetRange(const std::vector<FileMetaData*>& inputs,
                          InternalKey* smallest,
                          InternalKey* largest) {
  assert(!inputs.empty());
  smallest->Clear();
  largest->Clear();
D
dgrogan@chromium.org 已提交
1255
  for (size_t i = 0; i < inputs.size(); i++) {
J
jorlow@chromium.org 已提交
1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
    FileMetaData* f = inputs[i];
    if (i == 0) {
      *smallest = f->smallest;
      *largest = f->largest;
    } else {
      if (icmp_.Compare(f->smallest, *smallest) < 0) {
        *smallest = f->smallest;
      }
      if (icmp_.Compare(f->largest, *largest) > 0) {
        *largest = f->largest;
      }
    }
  }
}

1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
// Stores the minimal range that covers all entries in inputs1 and inputs2
// in *smallest, *largest.
// REQUIRES: inputs is not empty
void VersionSet::GetRange2(const std::vector<FileMetaData*>& inputs1,
                           const std::vector<FileMetaData*>& inputs2,
                           InternalKey* smallest,
                           InternalKey* largest) {
  std::vector<FileMetaData*> all = inputs1;
  all.insert(all.end(), inputs2.begin(), inputs2.end());
  GetRange(all, smallest, largest);
}

J
jorlow@chromium.org 已提交
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297
Iterator* VersionSet::MakeInputIterator(Compaction* c) {
  ReadOptions options;
  options.verify_checksums = options_->paranoid_checks;
  options.fill_cache = false;

  // Level-0 files have to be merged together.  For other levels,
  // we will make a concatenating iterator per level.
  // TODO(opt): use concatenating iterator for level-0 if there is no overlap
  const int space = (c->level() == 0 ? c->inputs_[0].size() + 1 : 2);
  Iterator** list = new Iterator*[space];
  int num = 0;
  for (int which = 0; which < 2; which++) {
    if (!c->inputs_[which].empty()) {
      if (c->level() + which == 0) {
        const std::vector<FileMetaData*>& files = c->inputs_[which];
D
dgrogan@chromium.org 已提交
1298
        for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1299 1300
          list[num++] = table_cache_->NewIterator(
              options, files[i]->number, files[i]->file_size);
J
jorlow@chromium.org 已提交
1301 1302 1303 1304
        }
      } else {
        // Create concatenating iterator for the files from this level
        list[num++] = NewTwoLevelIterator(
1305
            new Version::LevelFileNumIterator(icmp_, &c->inputs_[which]),
J
jorlow@chromium.org 已提交
1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
            &GetFileIterator, table_cache_, options);
      }
    }
  }
  assert(num <= space);
  Iterator* result = NewMergingIterator(&icmp_, list, num);
  delete[] list;
  return result;
}

1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341
double VersionSet::MaxBytesForLevel(int level) {
	// Note: the result for level zero is not really used since we set
	// the level-0 compaction threshold based on number of files.
  assert(level >= 0);
  assert(level < NumberLevels());
	return level_max_bytes_[level];
}

uint64_t VersionSet::MaxFileSizeForLevel(int level) {
  assert(level >= 0);
  assert(level < NumberLevels());
	return max_file_size_[level];
}

uint64_t VersionSet::ExpandedCompactionByteSizeLimit(int level) {
  uint64_t result = MaxFileSizeForLevel(level);
  result *= options_->expanded_compaction_factor;
  return result;
}

uint64_t VersionSet::MaxGrandParentOverlapBytes(int level) {
  uint64_t result = MaxFileSizeForLevel(level);
  result *= options_->max_grandparent_overlap_factor;
  return result;
}

J
jorlow@chromium.org 已提交
1342
Compaction* VersionSet::PickCompaction() {
1343 1344 1345 1346 1347 1348 1349 1350 1351 1352
  Compaction* c;
  int level;

  // We prefer compactions triggered by too much data in a level over
  // the compactions triggered by seeks.
  const bool size_compaction = (current_->compaction_score_ >= 1);
  const bool seek_compaction = (current_->file_to_compact_ != NULL);
  if (size_compaction) {
    level = current_->compaction_level_;
    assert(level >= 0);
1353 1354 1355
    assert(level+1 < NumberLevels());
    c = new Compaction(level, MaxFileSizeForLevel(level),
        MaxGrandParentOverlapBytes(level), NumberLevels());
1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371

    // Pick the first file that comes after compact_pointer_[level]
    for (size_t i = 0; i < current_->files_[level].size(); i++) {
      FileMetaData* f = current_->files_[level][i];
      if (compact_pointer_[level].empty() ||
          icmp_.Compare(f->largest.Encode(), compact_pointer_[level]) > 0) {
        c->inputs_[0].push_back(f);
        break;
      }
    }
    if (c->inputs_[0].empty()) {
      // Wrap-around to the beginning of the key space
      c->inputs_[0].push_back(current_->files_[level][0]);
    }
  } else if (seek_compaction) {
    level = current_->file_to_compact_level_;
1372 1373
    c = new Compaction(level, MaxFileSizeForLevel(level),
		MaxGrandParentOverlapBytes(level), NumberLevels());
1374 1375
    c->inputs_[0].push_back(current_->file_to_compact_);
  } else {
J
jorlow@chromium.org 已提交
1376 1377 1378 1379 1380 1381 1382 1383
    return NULL;
  }

  c->input_version_ = current_;
  c->input_version_->Ref();

  // Files in level 0 may overlap each other, so pick up all overlapping ones
  if (level == 0) {
1384 1385
    InternalKey smallest, largest;
    GetRange(c->inputs_[0], &smallest, &largest);
J
jorlow@chromium.org 已提交
1386 1387 1388
    // Note that the next call will discard the file we placed in
    // c->inputs_[0] earlier and replace it with an overlapping set
    // which will include the picked file.
G
Gabor Cselle 已提交
1389
    current_->GetOverlappingInputs(0, &smallest, &largest, &c->inputs_[0]);
J
jorlow@chromium.org 已提交
1390 1391 1392
    assert(!c->inputs_[0].empty());
  }

1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
  SetupOtherInputs(c);

  return c;
}

void VersionSet::SetupOtherInputs(Compaction* c) {
  const int level = c->level();
  InternalKey smallest, largest;
  GetRange(c->inputs_[0], &smallest, &largest);

G
Gabor Cselle 已提交
1403
  current_->GetOverlappingInputs(level+1, &smallest, &largest, &c->inputs_[1]);
J
jorlow@chromium.org 已提交
1404

1405 1406 1407 1408
  // Get entire range covered by compaction
  InternalKey all_start, all_limit;
  GetRange2(c->inputs_[0], c->inputs_[1], &all_start, &all_limit);

J
jorlow@chromium.org 已提交
1409 1410 1411 1412
  // See if we can grow the number of inputs in "level" without
  // changing the number of "level+1" files we pick up.
  if (!c->inputs_[1].empty()) {
    std::vector<FileMetaData*> expanded0;
G
Gabor Cselle 已提交
1413
    current_->GetOverlappingInputs(level, &all_start, &all_limit, &expanded0);
1414 1415 1416
    const int64_t inputs0_size = TotalFileSize(c->inputs_[0]);
    const int64_t inputs1_size = TotalFileSize(c->inputs_[1]);
    const int64_t expanded0_size = TotalFileSize(expanded0);
1417
    uint64_t limit = ExpandedCompactionByteSizeLimit(level);
1418
    if (expanded0.size() > c->inputs_[0].size() &&
1419
        inputs1_size + expanded0_size < limit) {
J
jorlow@chromium.org 已提交
1420 1421 1422
      InternalKey new_start, new_limit;
      GetRange(expanded0, &new_start, &new_limit);
      std::vector<FileMetaData*> expanded1;
G
Gabor Cselle 已提交
1423 1424
      current_->GetOverlappingInputs(level+1, &new_start, &new_limit,
                                     &expanded1);
J
jorlow@chromium.org 已提交
1425
      if (expanded1.size() == c->inputs_[1].size()) {
1426
        Log(options_->info_log,
1427
            "Expanding@%d %d+%d (%ld+%ld bytes) to %d+%d (%ld+%ld bytes)\n",
J
jorlow@chromium.org 已提交
1428 1429 1430
            level,
            int(c->inputs_[0].size()),
            int(c->inputs_[1].size()),
1431
            long(inputs0_size), long(inputs1_size),
J
jorlow@chromium.org 已提交
1432
            int(expanded0.size()),
1433 1434
            int(expanded1.size()),
            long(expanded0_size), long(inputs1_size));
J
jorlow@chromium.org 已提交
1435 1436 1437 1438
        smallest = new_start;
        largest = new_limit;
        c->inputs_[0] = expanded0;
        c->inputs_[1] = expanded1;
1439
        GetRange2(c->inputs_[0], c->inputs_[1], &all_start, &all_limit);
J
jorlow@chromium.org 已提交
1440 1441 1442 1443
      }
    }
  }

1444 1445
  // Compute the set of grandparent files that overlap this compaction
  // (parent == level+1; grandparent == level+2)
1446
  if (level + 2 < NumberLevels()) {
G
Gabor Cselle 已提交
1447 1448
    current_->GetOverlappingInputs(level + 2, &all_start, &all_limit,
                                   &c->grandparents_);
1449 1450
  }

J
jorlow@chromium.org 已提交
1451
  if (false) {
1452
    Log(options_->info_log, "Compacting %d '%s' .. '%s'",
J
jorlow@chromium.org 已提交
1453
        level,
G
Gabor Cselle 已提交
1454 1455
        smallest.DebugString().c_str(),
        largest.DebugString().c_str());
J
jorlow@chromium.org 已提交
1456 1457 1458 1459 1460 1461 1462
  }

  // Update the place where we will do the next compaction for this level.
  // We update this immediately instead of waiting for the VersionEdit
  // to be applied so that if the compaction fails, we will try a different
  // key range next time.
  compact_pointer_[level] = largest.Encode().ToString();
1463
  c->edit_->SetCompactPointer(level, largest);
J
jorlow@chromium.org 已提交
1464 1465 1466 1467
}

Compaction* VersionSet::CompactRange(
    int level,
G
Gabor Cselle 已提交
1468 1469
    const InternalKey* begin,
    const InternalKey* end) {
J
jorlow@chromium.org 已提交
1470
  std::vector<FileMetaData*> inputs;
G
Gabor Cselle 已提交
1471
  current_->GetOverlappingInputs(level, begin, end, &inputs);
J
jorlow@chromium.org 已提交
1472 1473 1474 1475
  if (inputs.empty()) {
    return NULL;
  }

G
Gabor Cselle 已提交
1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487
  // Avoid compacting too much in one shot in case the range is large.
  const uint64_t limit = MaxFileSizeForLevel(level);
  uint64_t total = 0;
  for (int i = 0; i < inputs.size(); i++) {
    uint64_t s = inputs[i]->file_size;
    total += s;
    if (total >= limit) {
      inputs.resize(i + 1);
      break;
    }
  }

1488 1489
  Compaction* c = new Compaction(level, limit,
		MaxGrandParentOverlapBytes(level), NumberLevels());
J
jorlow@chromium.org 已提交
1490 1491 1492
  c->input_version_ = current_;
  c->input_version_->Ref();
  c->inputs_[0] = inputs;
1493
  SetupOtherInputs(c);
J
jorlow@chromium.org 已提交
1494 1495 1496
  return c;
}

1497 1498
Compaction::Compaction(int level, uint64_t target_file_size,
		uint64_t max_grandparent_overlap_bytes, int number_levels)
J
jorlow@chromium.org 已提交
1499
    : level_(level),
1500 1501 1502
      max_output_file_size_(target_file_size),
      maxGrandParentOverlapBytes_(max_grandparent_overlap_bytes),
      number_levels_(number_levels),
1503 1504
      input_version_(NULL),
      grandparent_index_(0),
J
jorlow@chromium.org 已提交
1505 1506
      seen_key_(false),
      overlapped_bytes_(0) {
1507 1508 1509
	edit_ = new VersionEdit(number_levels_);
	level_ptrs_ = new size_t[number_levels_];
  for (int i = 0; i < number_levels_; i++) {
J
jorlow@chromium.org 已提交
1510 1511 1512 1513 1514
    level_ptrs_[i] = 0;
  }
}

Compaction::~Compaction() {
1515 1516
	delete[] level_ptrs_;
	delete edit_;
J
jorlow@chromium.org 已提交
1517 1518 1519 1520 1521
  if (input_version_ != NULL) {
    input_version_->Unref();
  }
}

1522
bool Compaction::IsTrivialMove() const {
J
jorlow@chromium.org 已提交
1523
  // Avoid a move if there is lots of overlapping grandparent data.
1524 1525
  // Otherwise, the move could create a parent file that will require
  // a very expensive merge later on.
J
jorlow@chromium.org 已提交
1526 1527
  return (num_input_files(0) == 1 &&
          num_input_files(1) == 0 &&
1528
          TotalFileSize(grandparents_) <= maxGrandParentOverlapBytes_);
1529 1530
}

J
jorlow@chromium.org 已提交
1531 1532
void Compaction::AddInputDeletions(VersionEdit* edit) {
  for (int which = 0; which < 2; which++) {
D
dgrogan@chromium.org 已提交
1533
    for (size_t i = 0; i < inputs_[which].size(); i++) {
J
jorlow@chromium.org 已提交
1534 1535 1536 1537 1538 1539 1540 1541
      edit->DeleteFile(level_ + which, inputs_[which][i]->number);
    }
  }
}

bool Compaction::IsBaseLevelForKey(const Slice& user_key) {
  // Maybe use binary search to find right entry instead of linear search?
  const Comparator* user_cmp = input_version_->vset_->icmp_.user_comparator();
1542
  for (int lvl = level_ + 2; lvl < number_levels_; lvl++) {
J
jorlow@chromium.org 已提交
1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559
    const std::vector<FileMetaData*>& files = input_version_->files_[lvl];
    for (; level_ptrs_[lvl] < files.size(); ) {
      FileMetaData* f = files[level_ptrs_[lvl]];
      if (user_cmp->Compare(user_key, f->largest.user_key()) <= 0) {
        // We've advanced far enough
        if (user_cmp->Compare(user_key, f->smallest.user_key()) >= 0) {
          // Key falls in this file's range, so definitely not base level
          return false;
        }
        break;
      }
      level_ptrs_[lvl]++;
    }
  }
  return true;
}

1560
bool Compaction::ShouldStopBefore(const Slice& internal_key) {
1561 1562 1563
  // Scan to find earliest grandparent file that contains key.
  const InternalKeyComparator* icmp = &input_version_->vset_->icmp_;
  while (grandparent_index_ < grandparents_.size() &&
1564 1565
      icmp->Compare(internal_key,
                    grandparents_[grandparent_index_]->largest.Encode()) > 0) {
J
jorlow@chromium.org 已提交
1566 1567 1568
    if (seen_key_) {
      overlapped_bytes_ += grandparents_[grandparent_index_]->file_size;
    }
1569 1570
    grandparent_index_++;
  }
J
jorlow@chromium.org 已提交
1571
  seen_key_ = true;
1572

1573
  if (overlapped_bytes_ > maxGrandParentOverlapBytes_) {
J
jorlow@chromium.org 已提交
1574 1575
    // Too much overlap for current output; start new output
    overlapped_bytes_ = 0;
1576 1577 1578 1579 1580 1581
    return true;
  } else {
    return false;
  }
}

J
jorlow@chromium.org 已提交
1582 1583 1584 1585 1586 1587 1588
void Compaction::ReleaseInputs() {
  if (input_version_ != NULL) {
    input_version_->Unref();
    input_version_ = NULL;
  }
}

H
heyongqiang 已提交
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621
static void InputSummary(std::vector<FileMetaData*>& files,
    char* output,
    int len) {
  int write = 0;
  for (int i = 0; i < files.size(); i++) {
    int sz = len - write;
    int ret = snprintf(output + write, sz, "%llu(%llu) ",
        files.at(i)->number,
        files.at(i)->file_size);
    if (ret < 0 || ret >= sz)
      break;
    write += ret;
  }
}

void Compaction::Summary(char* output, int len) {
  int write = snprintf(output, len, "Base level %d, inputs:", level_);
  if(write < 0 || write > len)
    return;

  char level_low_summary[100];
  InputSummary(inputs_[0], level_low_summary, 100);
  char level_up_summary[100];
  if (inputs_[1].size()) {
    InputSummary(inputs_[1], level_up_summary, 100);
  } else {
    level_up_summary[0] = '\0';
  }

  snprintf(output + write, len - write, "[%s],[%s]",
      level_low_summary, level_up_summary);
}

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