version_set.cc 49.6 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 796 797
      std::string record;
      edit->EncodeTo(&record);
      s = descriptor_log_->AddRecord(record);
      if (s.ok()) {
        s = descriptor_file_->Sync();
      }
J
jorlow@chromium.org 已提交
798 799
    }

800 801 802 803 804 805 806
    // 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 已提交
807 808 809 810
  }

  // Install the new version
  if (s.ok()) {
811
    AppendVersion(v);
812 813
    log_number_ = edit->log_number_;
    prev_log_number_ = edit->prev_log_number_;
J
jorlow@chromium.org 已提交
814 815 816 817 818 819 820 821 822 823 824 825 826 827
  } 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;
}

828
Status VersionSet::Recover() {
J
jorlow@chromium.org 已提交
829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854
  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);

  std::string dscname = dbname_ + "/" + current;
  SequentialFile* file;
  s = env_->NewSequentialFile(dscname, &file);
  if (!s.ok()) {
    return s;
  }

  bool have_log_number = false;
855
  bool have_prev_log_number = false;
J
jorlow@chromium.org 已提交
856 857 858
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
859 860 861
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
  uint64_t prev_log_number = 0;
J
jorlow@chromium.org 已提交
862 863 864 865 866
  Builder builder(this, current_);

  {
    LogReporter reporter;
    reporter.status = &s;
867
    log::Reader reader(file, &reporter, true/*checksum*/, 0/*initial_offset*/);
J
jorlow@chromium.org 已提交
868 869 870
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
871
      VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
872 873 874 875 876 877 878 879 880 881 882 883 884 885 886
      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_) {
887
        log_number = edit.log_number_;
J
jorlow@chromium.org 已提交
888 889 890
        have_log_number = true;
      }

891 892 893 894 895
      if (edit.has_prev_log_number_) {
        prev_log_number = edit.prev_log_number_;
        have_prev_log_number = true;
      }

J
jorlow@chromium.org 已提交
896 897 898 899 900 901
      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

      if (edit.has_last_sequence_) {
902
        last_sequence = edit.last_sequence_;
J
jorlow@chromium.org 已提交
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917
        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");
    }
918 919 920 921

    if (!have_prev_log_number) {
      prev_log_number = 0;
    }
922 923 924

    MarkFileNumberUsed(prev_log_number);
    MarkFileNumberUsed(log_number);
J
jorlow@chromium.org 已提交
925 926 927 928 929
  }

  if (s.ok()) {
    Version* v = new Version(this);
    builder.SaveTo(v);
930 931 932 933 934 935 936 937
    // 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;
J
jorlow@chromium.org 已提交
938 939 940 941 942
  }

  return s;
}

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 1043 1044
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;
1045

1046
    printf("manifest_file_number %d next_file_number %d last_sequence %d log_number %d  prev_log_number %d\n",
1047
           manifest_file_number_, next_file_number_,
1048 1049 1050
           last_sequence, log_number, prev_log_number);
    printf("%s \n", v->DebugString().c_str());
  }
1051

1052 1053 1054 1055

  return s;
}

1056 1057 1058 1059 1060 1061
void VersionSet::MarkFileNumberUsed(uint64_t number) {
  if (next_file_number_ <= number) {
    next_file_number_ = number + 1;
  }
}

1062
void VersionSet::Finalize(Version* v) {
J
jorlow@chromium.org 已提交
1063 1064 1065 1066
  // Precomputed best level for next compaction
  int best_level = -1;
  double best_score = -1;

1067
  for (int level = 0; level < NumberLevels()-1; level++) {
1068
    double score;
J
jorlow@chromium.org 已提交
1069
    if (level == 0) {
1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
      // 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).
1081
      score = v->files_[level].size() /
1082
          static_cast<double>(options_->level0_file_num_compaction_trigger);
1083 1084 1085 1086
    } 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 已提交
1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102
    }

    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
1103
  VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
1104 1105 1106
  edit.SetComparatorName(icmp_.user_comparator()->Name());

  // Save compaction pointers
1107
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1108 1109 1110 1111 1112 1113 1114 1115
    if (!compact_pointer_[level].empty()) {
      InternalKey key;
      key.DecodeFrom(compact_pointer_[level]);
      edit.SetCompactPointer(level, key);
    }
  }

  // Save files
1116
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1117
    const std::vector<FileMetaData*>& files = current_->files_[level];
D
dgrogan@chromium.org 已提交
1118
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130
      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);
1131
  assert(level < NumberLevels());
J
jorlow@chromium.org 已提交
1132 1133 1134
  return current_->files_[level].size();
}

1135
const char* VersionSet::LevelSummary(LevelSummaryStorage* scratch) const {
1136 1137 1138 1139 1140 1141 1142 1143 1144 1145
  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, "]");
1146 1147 1148
  return scratch->buffer;
}

1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163
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 已提交
1164 1165
uint64_t VersionSet::ApproximateOffsetOf(Version* v, const InternalKey& ikey) {
  uint64_t result = 0;
1166
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1167
    const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
1168
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184
      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 已提交
1185
            ReadOptions(), files[i]->number, files[i]->file_size, &tableptr);
J
jorlow@chromium.org 已提交
1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196
        if (tableptr != NULL) {
          result += tableptr->ApproximateOffsetOf(ikey.Encode());
        }
        delete iter;
      }
    }
  }
  return result;
}

void VersionSet::AddLiveFiles(std::set<uint64_t>* live) {
1197 1198 1199
  for (Version* v = dummy_versions_.next_;
       v != &dummy_versions_;
       v = v->next_) {
1200
    for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1201
      const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
1202
      for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1203 1204 1205 1206 1207 1208
        live->insert(files[i]->number);
      }
    }
  }
}

1209 1210
int64_t VersionSet::NumLevelBytes(int level) const {
  assert(level >= 0);
1211
  assert(level < NumberLevels());
1212
  return TotalFileSize(current_->files_[level]);
J
jorlow@chromium.org 已提交
1213 1214 1215 1216
}

int64_t VersionSet::MaxNextLevelOverlappingBytes() {
  int64_t result = 0;
1217
  std::vector<FileMetaData*> overlaps;
1218
  for (int level = 1; level < NumberLevels() - 1; level++) {
D
dgrogan@chromium.org 已提交
1219
    for (size_t i = 0; i < current_->files_[level].size(); i++) {
1220
      const FileMetaData* f = current_->files_[level][i];
G
Gabor Cselle 已提交
1221 1222
      current_->GetOverlappingInputs(level+1, &f->smallest, &f->largest,
                                     &overlaps);
J
jorlow@chromium.org 已提交
1223
      const int64_t sum = TotalFileSize(overlaps);
1224 1225 1226 1227 1228 1229 1230 1231
      if (sum > result) {
        result = sum;
      }
    }
  }
  return result;
}

J
jorlow@chromium.org 已提交
1232 1233 1234 1235 1236 1237 1238 1239 1240
// 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 已提交
1241
  for (size_t i = 0; i < inputs.size(); i++) {
J
jorlow@chromium.org 已提交
1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
    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;
      }
    }
  }
}

1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268
// 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 已提交
1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283
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 已提交
1284
        for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1285 1286
          list[num++] = table_cache_->NewIterator(
              options, files[i]->number, files[i]->file_size);
J
jorlow@chromium.org 已提交
1287 1288 1289 1290
        }
      } else {
        // Create concatenating iterator for the files from this level
        list[num++] = NewTwoLevelIterator(
1291
            new Version::LevelFileNumIterator(icmp_, &c->inputs_[which]),
J
jorlow@chromium.org 已提交
1292 1293 1294 1295 1296 1297 1298 1299 1300 1301
            &GetFileIterator, table_cache_, options);
      }
    }
  }
  assert(num <= space);
  Iterator* result = NewMergingIterator(&icmp_, list, num);
  delete[] list;
  return result;
}

1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327
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 已提交
1328
Compaction* VersionSet::PickCompaction() {
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338
  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);
1339 1340 1341
    assert(level+1 < NumberLevels());
    c = new Compaction(level, MaxFileSizeForLevel(level),
        MaxGrandParentOverlapBytes(level), NumberLevels());
1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357

    // 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_;
1358 1359
    c = new Compaction(level, MaxFileSizeForLevel(level),
		MaxGrandParentOverlapBytes(level), NumberLevels());
1360 1361
    c->inputs_[0].push_back(current_->file_to_compact_);
  } else {
J
jorlow@chromium.org 已提交
1362 1363 1364 1365 1366 1367 1368 1369
    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) {
1370 1371
    InternalKey smallest, largest;
    GetRange(c->inputs_[0], &smallest, &largest);
J
jorlow@chromium.org 已提交
1372 1373 1374
    // 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 已提交
1375
    current_->GetOverlappingInputs(0, &smallest, &largest, &c->inputs_[0]);
J
jorlow@chromium.org 已提交
1376 1377 1378
    assert(!c->inputs_[0].empty());
  }

1379 1380 1381 1382 1383 1384 1385 1386 1387 1388
  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 已提交
1389
  current_->GetOverlappingInputs(level+1, &smallest, &largest, &c->inputs_[1]);
J
jorlow@chromium.org 已提交
1390

1391 1392 1393 1394
  // 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 已提交
1395 1396 1397 1398
  // 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 已提交
1399
    current_->GetOverlappingInputs(level, &all_start, &all_limit, &expanded0);
1400 1401 1402
    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);
1403
    uint64_t limit = ExpandedCompactionByteSizeLimit(level);
1404
    if (expanded0.size() > c->inputs_[0].size() &&
1405
        inputs1_size + expanded0_size < limit) {
J
jorlow@chromium.org 已提交
1406 1407 1408
      InternalKey new_start, new_limit;
      GetRange(expanded0, &new_start, &new_limit);
      std::vector<FileMetaData*> expanded1;
G
Gabor Cselle 已提交
1409 1410
      current_->GetOverlappingInputs(level+1, &new_start, &new_limit,
                                     &expanded1);
J
jorlow@chromium.org 已提交
1411
      if (expanded1.size() == c->inputs_[1].size()) {
1412
        Log(options_->info_log,
1413
            "Expanding@%d %d+%d (%ld+%ld bytes) to %d+%d (%ld+%ld bytes)\n",
J
jorlow@chromium.org 已提交
1414 1415 1416
            level,
            int(c->inputs_[0].size()),
            int(c->inputs_[1].size()),
1417
            long(inputs0_size), long(inputs1_size),
J
jorlow@chromium.org 已提交
1418
            int(expanded0.size()),
1419 1420
            int(expanded1.size()),
            long(expanded0_size), long(inputs1_size));
J
jorlow@chromium.org 已提交
1421 1422 1423 1424
        smallest = new_start;
        largest = new_limit;
        c->inputs_[0] = expanded0;
        c->inputs_[1] = expanded1;
1425
        GetRange2(c->inputs_[0], c->inputs_[1], &all_start, &all_limit);
J
jorlow@chromium.org 已提交
1426 1427 1428 1429
      }
    }
  }

1430 1431
  // Compute the set of grandparent files that overlap this compaction
  // (parent == level+1; grandparent == level+2)
1432
  if (level + 2 < NumberLevels()) {
G
Gabor Cselle 已提交
1433 1434
    current_->GetOverlappingInputs(level + 2, &all_start, &all_limit,
                                   &c->grandparents_);
1435 1436
  }

J
jorlow@chromium.org 已提交
1437
  if (false) {
1438
    Log(options_->info_log, "Compacting %d '%s' .. '%s'",
J
jorlow@chromium.org 已提交
1439
        level,
G
Gabor Cselle 已提交
1440 1441
        smallest.DebugString().c_str(),
        largest.DebugString().c_str());
J
jorlow@chromium.org 已提交
1442 1443 1444 1445 1446 1447 1448
  }

  // 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();
1449
  c->edit_->SetCompactPointer(level, largest);
J
jorlow@chromium.org 已提交
1450 1451 1452 1453
}

Compaction* VersionSet::CompactRange(
    int level,
G
Gabor Cselle 已提交
1454 1455
    const InternalKey* begin,
    const InternalKey* end) {
J
jorlow@chromium.org 已提交
1456
  std::vector<FileMetaData*> inputs;
G
Gabor Cselle 已提交
1457
  current_->GetOverlappingInputs(level, begin, end, &inputs);
J
jorlow@chromium.org 已提交
1458 1459 1460 1461
  if (inputs.empty()) {
    return NULL;
  }

G
Gabor Cselle 已提交
1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473
  // 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;
    }
  }

1474 1475
  Compaction* c = new Compaction(level, limit,
		MaxGrandParentOverlapBytes(level), NumberLevels());
J
jorlow@chromium.org 已提交
1476 1477 1478
  c->input_version_ = current_;
  c->input_version_->Ref();
  c->inputs_[0] = inputs;
1479
  SetupOtherInputs(c);
J
jorlow@chromium.org 已提交
1480 1481 1482
  return c;
}

1483 1484
Compaction::Compaction(int level, uint64_t target_file_size,
		uint64_t max_grandparent_overlap_bytes, int number_levels)
J
jorlow@chromium.org 已提交
1485
    : level_(level),
1486 1487 1488
      max_output_file_size_(target_file_size),
      maxGrandParentOverlapBytes_(max_grandparent_overlap_bytes),
      number_levels_(number_levels),
1489 1490
      input_version_(NULL),
      grandparent_index_(0),
J
jorlow@chromium.org 已提交
1491 1492
      seen_key_(false),
      overlapped_bytes_(0) {
1493 1494 1495
	edit_ = new VersionEdit(number_levels_);
	level_ptrs_ = new size_t[number_levels_];
  for (int i = 0; i < number_levels_; i++) {
J
jorlow@chromium.org 已提交
1496 1497 1498 1499 1500
    level_ptrs_[i] = 0;
  }
}

Compaction::~Compaction() {
1501 1502
	delete[] level_ptrs_;
	delete edit_;
J
jorlow@chromium.org 已提交
1503 1504 1505 1506 1507
  if (input_version_ != NULL) {
    input_version_->Unref();
  }
}

1508
bool Compaction::IsTrivialMove() const {
J
jorlow@chromium.org 已提交
1509
  // Avoid a move if there is lots of overlapping grandparent data.
1510 1511
  // Otherwise, the move could create a parent file that will require
  // a very expensive merge later on.
J
jorlow@chromium.org 已提交
1512 1513
  return (num_input_files(0) == 1 &&
          num_input_files(1) == 0 &&
1514
          TotalFileSize(grandparents_) <= maxGrandParentOverlapBytes_);
1515 1516
}

J
jorlow@chromium.org 已提交
1517 1518
void Compaction::AddInputDeletions(VersionEdit* edit) {
  for (int which = 0; which < 2; which++) {
D
dgrogan@chromium.org 已提交
1519
    for (size_t i = 0; i < inputs_[which].size(); i++) {
J
jorlow@chromium.org 已提交
1520 1521 1522 1523 1524 1525 1526 1527
      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();
1528
  for (int lvl = level_ + 2; lvl < number_levels_; lvl++) {
J
jorlow@chromium.org 已提交
1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545
    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;
}

1546
bool Compaction::ShouldStopBefore(const Slice& internal_key) {
1547 1548 1549
  // Scan to find earliest grandparent file that contains key.
  const InternalKeyComparator* icmp = &input_version_->vset_->icmp_;
  while (grandparent_index_ < grandparents_.size() &&
1550 1551
      icmp->Compare(internal_key,
                    grandparents_[grandparent_index_]->largest.Encode()) > 0) {
J
jorlow@chromium.org 已提交
1552 1553 1554
    if (seen_key_) {
      overlapped_bytes_ += grandparents_[grandparent_index_]->file_size;
    }
1555 1556
    grandparent_index_++;
  }
J
jorlow@chromium.org 已提交
1557
  seen_key_ = true;
1558

1559
  if (overlapped_bytes_ > maxGrandParentOverlapBytes_) {
J
jorlow@chromium.org 已提交
1560 1561
    // Too much overlap for current output; start new output
    overlapped_bytes_ = 0;
1562 1563 1564 1565 1566 1567
    return true;
  } else {
    return false;
  }
}

J
jorlow@chromium.org 已提交
1568 1569 1570 1571 1572 1573 1574
void Compaction::ReleaseInputs() {
  if (input_version_ != NULL) {
    input_version_->Unref();
    input_version_ = NULL;
  }
}

H
heyongqiang 已提交
1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607
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 已提交
1608
}  // namespace leveldb