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

J
jorlow@chromium.org 已提交
31 32
Version::~Version() {
  assert(refs_ == 0);
33 34 35 36 37 38

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

  // Drop references to files
39
  for (int level = 0; level < vset_->NumberLevels(); level++) {
D
dgrogan@chromium.org 已提交
40
    for (size_t i = 0; i < files_[level].size(); i++) {
J
jorlow@chromium.org 已提交
41
      FileMetaData* f = files_[level][i];
42
      assert(f->refs > 0);
J
jorlow@chromium.org 已提交
43 44 45 46 47 48
      f->refs--;
      if (f->refs <= 0) {
        delete f;
      }
    }
  }
49
  delete[] files_;
J
jorlow@chromium.org 已提交
50 51
}

52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
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 已提交
73 74
static bool AfterFile(const Comparator* ucmp,
                      const Slice* user_key, const FileMetaData* f) {
A
Abhishek Kona 已提交
75 76
  // nullptr user_key occurs before all keys and is therefore never after *f
  return (user_key != nullptr &&
G
Gabor Cselle 已提交
77 78 79 80 81
          ucmp->Compare(*user_key, f->largest.user_key()) > 0);
}

static bool BeforeFile(const Comparator* ucmp,
                       const Slice* user_key, const FileMetaData* f) {
A
Abhishek Kona 已提交
82 83
  // nullptr user_key occurs after all keys and is therefore never before *f
  return (user_key != nullptr &&
G
Gabor Cselle 已提交
84 85 86
          ucmp->Compare(*user_key, f->smallest.user_key()) < 0);
}

87 88
bool SomeFileOverlapsRange(
    const InternalKeyComparator& icmp,
G
Gabor Cselle 已提交
89
    bool disjoint_sorted_files,
90
    const std::vector<FileMetaData*>& files,
G
Gabor Cselle 已提交
91 92 93 94 95
    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
H
heyongqiang 已提交
96
    for (size_t i = 0; i < files.size(); i++) {
G
Gabor Cselle 已提交
97 98 99 100 101 102 103 104 105 106 107 108 109
      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;
A
Abhishek Kona 已提交
110
  if (smallest_user_key != nullptr) {
G
Gabor Cselle 已提交
111 112 113 114 115 116 117 118 119 120 121
    // 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]);
122 123
}

J
jorlow@chromium.org 已提交
124 125 126
// 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 已提交
127 128
// 16-byte value containing the file number and file size, both
// encoded using EncodeFixed64.
J
jorlow@chromium.org 已提交
129 130
class Version::LevelFileNumIterator : public Iterator {
 public:
131
  LevelFileNumIterator(const InternalKeyComparator& icmp,
J
jorlow@chromium.org 已提交
132
                       const std::vector<FileMetaData*>* flist)
133
      : icmp_(icmp),
J
jorlow@chromium.org 已提交
134 135 136 137 138 139 140
        flist_(flist),
        index_(flist->size()) {        // Marks as invalid
  }
  virtual bool Valid() const {
    return index_ < flist_->size();
  }
  virtual void Seek(const Slice& target) {
141
    index_ = FindFile(icmp_, *flist_, target);
J
jorlow@chromium.org 已提交
142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165
  }
  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 已提交
166
    EncodeFixed64(value_buf_+8, (*flist_)[index_]->file_size);
J
jorlow@chromium.org 已提交
167 168 169 170 171 172
    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 已提交
173
  uint32_t index_;
J
jorlow@chromium.org 已提交
174

J
jorlow@chromium.org 已提交
175 176
  // Backing store for value().  Holds the file number and size.
  mutable char value_buf_[16];
J
jorlow@chromium.org 已提交
177 178 179 180
};

static Iterator* GetFileIterator(void* arg,
                                 const ReadOptions& options,
181
                                 const EnvOptions& soptions,
J
jorlow@chromium.org 已提交
182 183
                                 const Slice& file_value) {
  TableCache* cache = reinterpret_cast<TableCache*>(arg);
J
jorlow@chromium.org 已提交
184
  if (file_value.size() != 16) {
J
jorlow@chromium.org 已提交
185 186 187
    return NewErrorIterator(
        Status::Corruption("FileReader invoked with unexpected value"));
  } else {
J
jorlow@chromium.org 已提交
188
    return cache->NewIterator(options,
189
                              soptions,
J
jorlow@chromium.org 已提交
190 191
                              DecodeFixed64(file_value.data()),
                              DecodeFixed64(file_value.data() + 8));
J
jorlow@chromium.org 已提交
192 193 194 195
  }
}

Iterator* Version::NewConcatenatingIterator(const ReadOptions& options,
196
                                            const StorageOptions& soptions,
J
jorlow@chromium.org 已提交
197 198
                                            int level) const {
  return NewTwoLevelIterator(
199
      new LevelFileNumIterator(vset_->icmp_, &files_[level]),
200
      &GetFileIterator, vset_->table_cache_, options, soptions);
J
jorlow@chromium.org 已提交
201 202 203
}

void Version::AddIterators(const ReadOptions& options,
204
                           const StorageOptions& soptions,
J
jorlow@chromium.org 已提交
205 206
                           std::vector<Iterator*>* iters) {
  // Merge all level zero files together since they may overlap
D
dgrogan@chromium.org 已提交
207
  for (size_t i = 0; i < files_[0].size(); i++) {
J
jorlow@chromium.org 已提交
208
    iters->push_back(
J
jorlow@chromium.org 已提交
209
        vset_->table_cache_->NewIterator(
210
            options, soptions, files_[0][i]->number, files_[0][i]->file_size));
J
jorlow@chromium.org 已提交
211 212 213 214 215
  }

  // For levels > 0, we can use a concatenating iterator that sequentially
  // walks through the non-overlapping files in the level, opening them
  // lazily.
216
  for (int level = 1; level < vset_->NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
217
    if (!files_[level].empty()) {
218
      iters->push_back(NewConcatenatingIterator(options, soptions, level));
J
jorlow@chromium.org 已提交
219 220 221 222
    }
  }
}

S
Sanjay Ghemawat 已提交
223 224 225 226 227 228 229 230 231 232 233 234 235
// Callback from TableCache::Get()
namespace {
enum SaverState {
  kNotFound,
  kFound,
  kDeleted,
  kCorrupt,
};
struct Saver {
  SaverState state;
  const Comparator* ucmp;
  Slice user_key;
  std::string* value;
236
  bool didIO;    // did we do any disk io?
S
Sanjay Ghemawat 已提交
237 238
};
}
239
static void SaveValue(void* arg, const Slice& ikey, const Slice& v, bool didIO){
S
Sanjay Ghemawat 已提交
240
  Saver* s = reinterpret_cast<Saver*>(arg);
241
  ParsedInternalKey parsed_key;
242
  s->didIO = didIO;
S
Sanjay Ghemawat 已提交
243 244 245 246 247 248 249 250
  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());
      }
251 252 253 254 255 256 257 258
    }
  }
}

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

259
Version::Version(VersionSet* vset, uint64_t version_number)
260
    : vset_(vset), next_(this), prev_(this), refs_(0),
261
      files_by_size_(vset->NumberLevels()),
262
      next_file_to_compact_by_size_(vset->NumberLevels()),
A
Abhishek Kona 已提交
263
      file_to_compact_(nullptr),
264
      file_to_compact_level_(-1),
265 266 267 268
      compaction_score_(vset->NumberLevels()),
      compaction_level_(vset->NumberLevels()),
      offset_manifest_file_(0),
      version_number_(version_number) {
269 270 271
  files_ = new std::vector<FileMetaData*>[vset->NumberLevels()];
}

272 273 274 275 276 277 278 279 280
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;

A
Abhishek Kona 已提交
281
  stats->seek_file = nullptr;
282
  stats->seek_file_level = -1;
A
Abhishek Kona 已提交
283
  FileMetaData* last_file_read = nullptr;
284
  int last_file_read_level = -1;
285 286 287 288 289 290

  // 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;
291
  for (int level = 0; level < vset_->NumberLevels(); level++) {
292 293 294 295 296 297 298 299 300
    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 已提交
301
      for (uint32_t i = 0; i < num_files; i++) {
302 303 304 305 306 307 308 309 310 311 312 313 314 315 316
        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) {
A
Abhishek Kona 已提交
317
        files = nullptr;
318 319 320 321 322
        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
A
Abhishek Kona 已提交
323
          files = nullptr;
324 325 326 327 328 329 330 331
          num_files = 0;
        } else {
          files = &tmp2;
          num_files = 1;
        }
      }
    }

D
dgrogan@chromium.org 已提交
332
    for (uint32_t i = 0; i < num_files; ++i) {
333 334

      FileMetaData* f = files[i];
S
Sanjay Ghemawat 已提交
335 336 337 338 339
      Saver saver;
      saver.state = kNotFound;
      saver.ucmp = ucmp;
      saver.user_key = user_key;
      saver.value = value;
340 341
      saver.didIO = false;
      bool tableIO = false;
S
Sanjay Ghemawat 已提交
342
      s = vset_->table_cache_->Get(options, f->number, f->file_size,
343
                                   ikey, &saver, SaveValue, &tableIO);
S
Sanjay Ghemawat 已提交
344
      if (!s.ok()) {
345
        return s;
S
Sanjay Ghemawat 已提交
346
      }
347

A
Abhishek Kona 已提交
348
      if (last_file_read != nullptr && stats->seek_file == nullptr) {
349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365
        // We have had more than one seek for this read.  Charge the 1st file.
        stats->seek_file = last_file_read;
        stats->seek_file_level = last_file_read_level;
      }

      // If we did any IO as part of the read, then we remember it because
      // it is a possible candidate for seek-based compaction. saver.didIO
      // is true if the block had to be read in from storage and was not
      // pre-exisiting in the block cache. Also, if this file was not pre-
      // existing in the table cache and had to be freshly opened that needed
      // the index blocks to be read-in, then tableIO is true. One thing
      // to note is that the index blocks are not part of the block cache.
      if (saver.didIO || tableIO) {
        last_file_read = f;
        last_file_read_level = level;
      }

S
Sanjay Ghemawat 已提交
366 367 368 369 370 371 372 373 374 375
      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);
376 377 378 379 380 381 382 383 384 385
          return s;
      }
    }
  }

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

bool Version::UpdateStats(const GetStats& stats) {
  FileMetaData* f = stats.seek_file;
A
Abhishek Kona 已提交
386
  if (f != nullptr) {
387
    f->allowed_seeks--;
A
Abhishek Kona 已提交
388
    if (f->allowed_seeks <= 0 && file_to_compact_ == nullptr) {
389 390 391 392 393 394 395 396
      file_to_compact_ = f;
      file_to_compact_level_ = stats.seek_file_level;
      return true;
    }
  }
  return false;
}

J
jorlow@chromium.org 已提交
397 398 399 400 401
void Version::Ref() {
  ++refs_;
}

void Version::Unref() {
402
  assert(this != &vset_->dummy_versions_);
J
jorlow@chromium.org 已提交
403 404 405
  assert(refs_ >= 1);
  --refs_;
  if (refs_ == 0) {
406
    delete this;
J
jorlow@chromium.org 已提交
407 408 409
  }
}

410
bool Version::OverlapInLevel(int level,
G
Gabor Cselle 已提交
411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426
                             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;
427 428
    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 已提交
429 430 431
      if (OverlapInLevel(level + 1, &smallest_user_key, &largest_user_key)) {
        break;
      }
432
      if (level + 2 >= vset_->NumberLevels()) {
433 434
        level++;
        break;
435
      }
G
Gabor Cselle 已提交
436 437
      GetOverlappingInputs(level + 2, &start, &limit, &overlaps);
      const int64_t sum = TotalFileSize(overlaps);
438
      if (sum > vset_->MaxGrandParentOverlapBytes(level)) {
G
Gabor Cselle 已提交
439 440 441 442 443
        break;
      }
      level++;
    }
  }
444

G
Gabor Cselle 已提交
445 446 447 448
  return level;
}

// Store in "*inputs" all files in "level" that overlap [begin,end]
A
Abhishek Kona 已提交
449
// If hint_index is specified, then it points to a file in the
450 451
// overlapping range.
// The file_index returns a pointer to any file in an overlapping range.
G
Gabor Cselle 已提交
452 453 454 455
void Version::GetOverlappingInputs(
    int level,
    const InternalKey* begin,
    const InternalKey* end,
456 457 458
    std::vector<FileMetaData*>* inputs,
    int hint_index,
    int* file_index) {
G
Gabor Cselle 已提交
459 460
  inputs->clear();
  Slice user_begin, user_end;
A
Abhishek Kona 已提交
461
  if (begin != nullptr) {
G
Gabor Cselle 已提交
462 463
    user_begin = begin->user_key();
  }
A
Abhishek Kona 已提交
464
  if (end != nullptr) {
G
Gabor Cselle 已提交
465 466
    user_end = end->user_key();
  }
467 468 469
  if (file_index) {
    *file_index = -1;
  }
G
Gabor Cselle 已提交
470
  const Comparator* user_cmp = vset_->icmp_.user_comparator();
A
Abhishek Kona 已提交
471
  if (begin != nullptr && end != nullptr && level > 0) {
472 473
    GetOverlappingInputsBinarySearch(level, user_begin, user_end, inputs,
      hint_index, file_index);
474 475
    return;
  }
H
Hans Wennborg 已提交
476 477 478 479
  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();
A
Abhishek Kona 已提交
480
    if (begin != nullptr && user_cmp->Compare(file_limit, user_begin) < 0) {
G
Gabor Cselle 已提交
481
      // "f" is completely before specified range; skip it
A
Abhishek Kona 已提交
482
    } else if (end != nullptr && user_cmp->Compare(file_start, user_end) > 0) {
G
Gabor Cselle 已提交
483 484 485
      // "f" is completely after specified range; skip it
    } else {
      inputs->push_back(f);
H
Hans Wennborg 已提交
486 487 488
      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.
A
Abhishek Kona 已提交
489
        if (begin != nullptr && user_cmp->Compare(file_start, user_begin) < 0) {
H
Hans Wennborg 已提交
490 491 492
          user_begin = file_start;
          inputs->clear();
          i = 0;
A
Abhishek Kona 已提交
493 494
        } else if (end != nullptr
            && user_cmp->Compare(file_limit, user_end) > 0) {
H
Hans Wennborg 已提交
495 496 497 498
          user_end = file_limit;
          inputs->clear();
          i = 0;
        }
499
      } else if (file_index) {
500
        *file_index = i-1;
H
Hans Wennborg 已提交
501
      }
G
Gabor Cselle 已提交
502 503
    }
  }
504 505
}

506 507 508 509 510 511 512 513
// Store in "*inputs" all files in "level" that overlap [begin,end]
// Employ binary search to find at least one file that overlaps the
// specified range. From that file, iterate backwards and
// forwards to find all overlapping files.
void Version::GetOverlappingInputsBinarySearch(
    int level,
    const Slice& user_begin,
    const Slice& user_end,
514 515 516
    std::vector<FileMetaData*>* inputs,
    int hint_index,
    int* file_index) {
517 518 519 520 521 522
  assert(level > 0);
  int min = 0;
  int mid = 0;
  int max = files_[level].size() -1;
  bool foundOverlap = false;
  const Comparator* user_cmp = vset_->icmp_.user_comparator();
523 524 525 526 527 528 529 530 531

  // if the caller already knows the index of a file that has overlap,
  // then we can skip the binary search.
  if (hint_index != -1) {
    mid = hint_index;
    foundOverlap = true;
  }

  while (!foundOverlap && min <= max) {
532 533 534 535 536 537 538 539 540 541 542 543 544
    mid = (min + max)/2;
    FileMetaData* f = files_[level][mid];
    const Slice file_start = f->smallest.user_key();
    const Slice file_limit = f->largest.user_key();
    if (user_cmp->Compare(file_limit, user_begin) < 0) {
      min = mid + 1;
    } else if (user_cmp->Compare(user_end, file_start) < 0) {
      max = mid - 1;
    } else {
      foundOverlap = true;
      break;
    }
  }
A
Abhishek Kona 已提交
545

546 547 548 549
  // If there were no overlapping files, return immediately.
  if (!foundOverlap) {
    return;
  }
550 551 552 553
  // returns the index where an overlap is found
  if (file_index) {
    *file_index = mid;
  }
554 555
  ExtendOverlappingInputs(level, user_begin, user_end, inputs, mid);
}
A
Abhishek Kona 已提交
556

557 558 559 560 561 562 563 564 565
// Store in "*inputs" all files in "level" that overlap [begin,end]
// The midIndex specifies the index of at least one file that
// overlaps the specified range. From that file, iterate backward
// and forward to find all overlapping files.
void Version::ExtendOverlappingInputs(
    int level,
    const Slice& user_begin,
    const Slice& user_end,
    std::vector<FileMetaData*>* inputs,
566
    unsigned int midIndex) {
567 568

  const Comparator* user_cmp = vset_->icmp_.user_comparator();
569 570 571 572 573 574 575 576 577 578 579 580 581 582
#ifndef NDEBUG
  {
    // assert that the file at midIndex overlaps with the range
    assert(midIndex < files_[level].size());
    FileMetaData* f = files_[level][midIndex];
    const Slice fstart = f->smallest.user_key();
    const Slice flimit = f->largest.user_key();
    if (user_cmp->Compare(fstart, user_begin) >= 0) {
      assert(user_cmp->Compare(fstart, user_end) <= 0);
    } else {
      assert(user_cmp->Compare(flimit, user_begin) >= 0);
    }
  }
#endif
583 584
  int startIndex = midIndex + 1;
  int endIndex = midIndex;
585
  int count __attribute__((unused)) = 0;
586 587

  // check backwards from 'mid' to lower indices
588
  for (int i = midIndex; i >= 0 ; i--) {
589 590 591
    FileMetaData* f = files_[level][i];
    const Slice file_limit = f->largest.user_key();
    if (user_cmp->Compare(file_limit, user_begin) >= 0) {
592 593
      startIndex = i;
      assert((count++, true));
594 595 596 597 598
    } else {
      break;
    }
  }
  // check forward from 'mid+1' to higher indices
599
  for (unsigned int i = midIndex+1; i < files_[level].size(); i++) {
600 601 602
    FileMetaData* f = files_[level][i];
    const Slice file_start = f->smallest.user_key();
    if (user_cmp->Compare(file_start, user_end) <= 0) {
603 604
      assert((count++, true));
      endIndex = i;
605 606 607 608
    } else {
      break;
    }
  }
609 610 611 612 613
  assert(count == endIndex - startIndex + 1);

  // insert overlapping files into vector
  for (int i = startIndex; i <= endIndex; i++) {
    FileMetaData* f = files_[level][i];
614
    inputs->push_back(f);
615
  }
616 617
}

Z
Zheng Shao 已提交
618
std::string Version::DebugString(bool hex) const {
J
jorlow@chromium.org 已提交
619
  std::string r;
620
  for (int level = 0; level < vset_->NumberLevels(); level++) {
621 622 623 624 625
    // E.g.,
    //   --- level 1 ---
    //   17:123['a' .. 'd']
    //   20:43['e' .. 'g']
    r.append("--- level ");
J
jorlow@chromium.org 已提交
626
    AppendNumberTo(&r, level);
627 628
    r.append(" --- version# ");
    AppendNumberTo(&r, version_number_);
629
    r.append(" ---\n");
J
jorlow@chromium.org 已提交
630
    const std::vector<FileMetaData*>& files = files_[level];
D
dgrogan@chromium.org 已提交
631
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
632 633 634 635
      r.push_back(' ');
      AppendNumberTo(&r, files[i]->number);
      r.push_back(':');
      AppendNumberTo(&r, files[i]->file_size);
G
Gabor Cselle 已提交
636
      r.append("[");
Z
Zheng Shao 已提交
637
      r.append(files[i]->smallest.DebugString(hex));
G
Gabor Cselle 已提交
638
      r.append(" .. ");
Z
Zheng Shao 已提交
639
      r.append(files[i]->largest.DebugString(hex));
G
Gabor Cselle 已提交
640
      r.append("]\n");
J
jorlow@chromium.org 已提交
641 642 643 644 645
    }
  }
  return r;
}

646 647 648 649 650 651
// this is used to batch writes to the manifest file
struct VersionSet::ManifestWriter {
  Status status;
  bool done;
  port::CondVar cv;
  VersionEdit* edit;
A
Abhishek Kona 已提交
652 653

  explicit ManifestWriter(port::Mutex* mu, VersionEdit* e) :
654 655 656
             done(false), cv(mu), edit(e) {}
};

J
jorlow@chromium.org 已提交
657 658 659 660 661
// 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:
662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682
  // 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 已提交
683
  VersionSet* vset_;
684
  Version* base_;
685
  LevelState* levels_;
J
jorlow@chromium.org 已提交
686 687 688 689

 public:
  // Initialize a builder with the files from *base and other info from *vset
  Builder(VersionSet* vset, Version* base)
690 691 692
      : vset_(vset),
        base_(base) {
    base_->Ref();
693
    levels_ = new LevelState[vset_->NumberLevels()];
694 695
    BySmallestKey cmp;
    cmp.internal_comparator = &vset_->icmp_;
696
    for (int level = 0; level < vset_->NumberLevels(); level++) {
697
      levels_[level].added_files = new FileSet(cmp);
J
jorlow@chromium.org 已提交
698 699 700 701
    }
  }

  ~Builder() {
702
    for (int level = 0; level < vset_->NumberLevels(); level++) {
703 704 705 706 707 708 709 710
      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 已提交
711
      for (uint32_t i = 0; i < to_unref.size(); i++) {
712
        FileMetaData* f = to_unref[i];
J
jorlow@chromium.org 已提交
713 714 715 716 717 718
        f->refs--;
        if (f->refs <= 0) {
          delete f;
        }
      }
    }
719
    delete[] levels_;
720
    base_->Unref();
J
jorlow@chromium.org 已提交
721 722
  }

723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742
  void CheckConsistency(Version* v) {
#ifndef NDEBUG
    for (int level = 0; level < vset_->NumberLevels(); level++) {
      // Make sure there is no overlap in levels > 0
      if (level > 0) {
        for (uint32_t i = 1; i < v->files_[level].size(); i++) {
          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",
                    prev_end.DebugString().c_str(),
                    this_begin.DebugString().c_str());
            abort();
          }
        }
      }
    }
#endif
  }

743 744 745 746
  void CheckConsistencyForDeletes(
    VersionEdit* edit,
    unsigned int number,
    int level) {
747 748 749 750 751
#ifndef NDEBUG
      // a file to be deleted better exist in the previous version
      bool found = false;
      for (int l = 0; !found && l < edit->number_levels_; l++) {
        const std::vector<FileMetaData*>& base_files = base_->files_[l];
752
        for (unsigned int i = 0; i < base_files.size(); i++) {
753 754 755 756 757 758 759 760 761 762 763 764
          FileMetaData* f = base_files[i];
          if (f->number == number) {
            found =  true;
            break;
          }
        }
      }
      // if the file did not exist in the previous version, then it
      // is possibly moved from lower level to higher level in current
      // version
      for (int l = level+1; !found && l < edit->number_levels_; l++) {
        const FileSet* added = levels_[l].added_files;
D
Dhruba Borthakur 已提交
765 766 767 768 769 770 771 772 773 774 775 776 777
        for (FileSet::const_iterator added_iter = added->begin();
             added_iter != added->end(); ++added_iter) {
          FileMetaData* f = *added_iter;
          if (f->number == number) {
            found = true;
            break;
          }
        }
      }

      // maybe this file was added in a previous edit that was Applied
      if (!found) {
        const FileSet* added = levels_[level].added_files;
778 779 780 781 782 783 784 785 786 787 788 789 790
        for (FileSet::const_iterator added_iter = added->begin();
             added_iter != added->end(); ++added_iter) {
          FileMetaData* f = *added_iter;
          if (f->number == number) {
            found = true;
            break;
          }
        }
      }
      assert(found);
#endif
  }

J
jorlow@chromium.org 已提交
791 792
  // Apply all of the edits in *edit to the current state.
  void Apply(VersionEdit* edit) {
793 794
    CheckConsistency(base_);

J
jorlow@chromium.org 已提交
795
    // Update compaction pointers
D
dgrogan@chromium.org 已提交
796
    for (size_t i = 0; i < edit->compact_pointers_.size(); i++) {
J
jorlow@chromium.org 已提交
797 798 799 800 801 802 803 804 805 806 807 808
      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;
809
      levels_[level].deleted_files.insert(number);
810
      CheckConsistencyForDeletes(edit, number, level);
J
jorlow@chromium.org 已提交
811 812 813
    }

    // Add new files
D
dgrogan@chromium.org 已提交
814
    for (size_t i = 0; i < edit->new_files_.size(); i++) {
J
jorlow@chromium.org 已提交
815 816 817
      const int level = edit->new_files_[i].first;
      FileMetaData* f = new FileMetaData(edit->new_files_[i].second);
      f->refs = 1;
818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834

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

835 836
      levels_[level].deleted_files.erase(f->number);
      levels_[level].added_files->insert(f);
J
jorlow@chromium.org 已提交
837 838 839 840 841
    }
  }

  // Save the current state in *v.
  void SaveTo(Version* v) {
842 843
    CheckConsistency(base_);
    CheckConsistency(v);
844 845
    BySmallestKey cmp;
    cmp.internal_comparator = &vset_->icmp_;
846
    for (int level = 0; level < vset_->NumberLevels(); level++) {
847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870
      // 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 已提交
871
      }
872
    }
873
    CheckConsistency(v);
874 875 876 877 878 879
  }

  void MaybeAddFile(Version* v, int level, FileMetaData* f) {
    if (levels_[level].deleted_files.count(f->number) > 0) {
      // File is deleted: do nothing
    } else {
880 881 882 883 884 885
      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);
      }
886
      f->refs++;
887
      files->push_back(f);
J
jorlow@chromium.org 已提交
888 889 890 891 892 893
    }
  }
};

VersionSet::VersionSet(const std::string& dbname,
                       const Options* options,
894
                       const StorageOptions& storage_options,
J
jorlow@chromium.org 已提交
895 896 897 898 899 900 901 902 903
                       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()
904 905 906
      last_sequence_(0),
      log_number_(0),
      prev_log_number_(0),
907
      num_levels_(options_->num_levels),
908
      dummy_versions_(this),
A
Abhishek Kona 已提交
909
      current_(nullptr),
910
      compactions_in_progress_(options_->num_levels),
A
Abhishek Kona 已提交
911
      current_version_number_(0),
912 913 914
      last_observed_manifest_size_(0),
      storage_options_(storage_options),
      storage_options_compactions_(storage_options_)  {
915
  compact_pointer_ = new std::string[options_->num_levels];
916
  Init(options_->num_levels);
917
  AppendVersion(new Version(this, current_version_number_++));
J
jorlow@chromium.org 已提交
918 919 920
}

VersionSet::~VersionSet() {
921 922
  current_->Unref();
  assert(dummy_versions_.next_ == &dummy_versions_);  // List must be empty
923 924 925
  delete[] compact_pointer_;
  delete[] max_file_size_;
  delete[] level_max_bytes_;
J
jorlow@chromium.org 已提交
926 927
}

928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943
void VersionSet::Init(int num_levels) {
  max_file_size_ = new uint64_t[num_levels];
  level_max_bytes_ = new uint64_t[num_levels];
  int target_file_size_multiplier = options_->target_file_size_multiplier;
  int max_bytes_multiplier = options_->max_bytes_for_level_multiplier;
  for (int i = 0; i < 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;
    }
  }
}

944 945 946 947
void VersionSet::AppendVersion(Version* v) {
  // Make "v" current
  assert(v->refs_ == 0);
  assert(v != current_);
A
Abhishek Kona 已提交
948
  if (current_ != nullptr) {
949
    assert(current_->refs_ > 0);
950 951 952 953 954 955 956 957 958 959 960 961
    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;
}

962 963
Status VersionSet::LogAndApply(VersionEdit* edit, port::Mutex* mu,
    bool new_descriptor_log) {
964
  mu->AssertHeld();
965

966 967 968 969 970
  // queue our request
  ManifestWriter w(mu, edit);
  manifest_writers_.push_back(&w);
  while (!w.done && &w != manifest_writers_.front()) {
    w.cv.Wait();
971
  }
972 973 974
  if (w.done) {
    return w.status;
  }
A
Abhishek Kona 已提交
975

976 977 978
  std::vector<VersionEdit*> batch_edits;
  Version* v = new Version(this, current_version_number_++);
  Builder builder(this, current_);
979

980 981 982
  // process all requests in the queue
  ManifestWriter* last_writer = &w;
  assert(!manifest_writers_.empty());
983
  assert(manifest_writers_.front() == &w);
984 985 986
  std::deque<ManifestWriter*>::iterator iter = manifest_writers_.begin();
  for (; iter != manifest_writers_.end(); ++iter) {
    last_writer = *iter;
A
Abhishek Kona 已提交
987
    LogAndApplyHelper(&builder, v, last_writer->edit, mu);
988
    batch_edits.push_back(last_writer->edit);
J
jorlow@chromium.org 已提交
989
  }
990
  builder.SaveTo(v);
J
jorlow@chromium.org 已提交
991 992 993

  // Initialize new descriptor log file if necessary by creating
  // a temporary file that contains a snapshot of the current version.
994
  std::string new_manifest_file;
995
  uint64_t new_manifest_file_size = 0;
996
  Status s;
A
Abhishek Kona 已提交
997 998

  //  No need to perform this check if a new Manifest is being created anyways.
999 1000
  if (!descriptor_log_ ||
      last_observed_manifest_size_ > options_->max_manifest_file_size) {
A
Abhishek Kona 已提交
1001 1002 1003 1004
    new_descriptor_log = true;
    manifest_file_number_ = NewFileNumber(); // Change manifest file no.
  }

1005
  if (!descriptor_log_ || new_descriptor_log) {
1006 1007
    // No reason to unlock *mu here since we only hit this path in the
    // first call to LogAndApply (when opening the database).
1008
    assert(!descriptor_log_ || new_descriptor_log);
1009 1010
    new_manifest_file = DescriptorFileName(dbname_, manifest_file_number_);
    edit->SetNextFile(next_file_number_);
1011
    unique_ptr<WritableFile> descriptor_file;
1012 1013
    s = env_->NewWritableFile(new_manifest_file, &descriptor_file,
                              storage_options_);
1014
    if (s.ok()) {
1015 1016
      descriptor_log_.reset(new log::Writer(std::move(descriptor_file)));
      s = WriteSnapshot(descriptor_log_.get());
J
jorlow@chromium.org 已提交
1017 1018 1019
    }
  }

1020 1021
  // Unlock during expensive MANIFEST log write. New writes cannot get here
  // because &w is ensuring that all new writes get queued.
1022
  {
1023 1024 1025 1026
    // calculate the amount of data being compacted at every level
    std::vector<uint64_t> size_being_compacted(NumberLevels()-1);
    SizeBeingCompacted(size_being_compacted);

1027
    mu->Unlock();
1028

1029
    // The calls to Finalize and UpdateFilesBySize are cpu-heavy
1030
    // and is best called outside the mutex.
1031
    Finalize(v, size_being_compacted);
1032
    UpdateFilesBySize(v);
1033 1034

    // Write new record to MANIFEST log
J
jorlow@chromium.org 已提交
1035
    if (s.ok()) {
1036
      std::string record;
1037 1038 1039 1040 1041 1042 1043
      for (unsigned int i = 0; i < batch_edits.size(); i++) {
        batch_edits[i]->EncodeTo(&record);
        s = descriptor_log_->AddRecord(record);
        if (!s.ok()) {
          break;
        }
      }
1044
      if (s.ok()) {
1045
        if (options_->use_fsync) {
1046
          s = descriptor_log_->file()->Fsync();
1047
        } else {
1048
          s = descriptor_log_->file()->Sync();
1049
        }
1050
      }
1051 1052 1053 1054 1055
      if (!s.ok()) {
        Log(options_->info_log, "MANIFEST write: %s\n", s.ToString().c_str());
        if (ManifestContains(record)) {
          Log(options_->info_log,
              "MANIFEST contains log record despite error; advancing to new "
1056 1057
              "version to prevent mismatch between in-memory and logged state"
              " If paranoid is set, then the db is now in readonly mode.");
1058 1059 1060
          s = Status::OK();
        }
      }
J
jorlow@chromium.org 已提交
1061 1062
    }

1063 1064 1065 1066 1067 1068
    // 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_);
    }

1069
    // find offset in manifest file where this version is stored.
1070
    new_manifest_file_size = descriptor_log_->file()->GetFileSize();
A
Abhishek Kona 已提交
1071

1072
    mu->Lock();
A
Abhishek Kona 已提交
1073 1074 1075
    // cache the manifest_file_size so that it can be used to rollover in the
    // next call to LogAndApply
    last_observed_manifest_size_ = new_manifest_file_size;
J
jorlow@chromium.org 已提交
1076 1077 1078 1079
  }

  // Install the new version
  if (s.ok()) {
1080
    v->offset_manifest_file_ = new_manifest_file_size;
1081
    AppendVersion(v);
1082 1083
    log_number_ = edit->log_number_;
    prev_log_number_ = edit->prev_log_number_;
1084

J
jorlow@chromium.org 已提交
1085
  } else {
A
Abhishek Kona 已提交
1086
    Log(options_->info_log, "Error in committing version %ld",
1087
        v->GetVersionNumber());
J
jorlow@chromium.org 已提交
1088 1089
    delete v;
    if (!new_manifest_file.empty()) {
1090
      descriptor_log_.reset();
J
jorlow@chromium.org 已提交
1091 1092 1093 1094
      env_->DeleteFile(new_manifest_file);
    }
  }

1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109
  // wake up all the waiting writers
  while (true) {
    ManifestWriter* ready = manifest_writers_.front();
    manifest_writers_.pop_front();
    if (ready != &w) {
      ready->status = s;
      ready->done = true;
      ready->cv.Signal();
    }
    if (ready == last_writer) break;
  }
  // Notify new head of write queue
  if (!manifest_writers_.empty()) {
    manifest_writers_.front()->cv.Signal();
  }
J
jorlow@chromium.org 已提交
1110 1111 1112
  return s;
}

1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133
void VersionSet::LogAndApplyHelper(Builder* builder, Version* v,
  VersionEdit* edit, port::Mutex* mu) {
  mu->AssertHeld();

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

  edit->SetNextFile(next_file_number_);
  edit->SetLastSequence(last_sequence_);

  builder->Apply(edit);
}

1134
Status VersionSet::Recover() {
J
jorlow@chromium.org 已提交
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152
  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 已提交
1153 1154 1155
  Log(options_->info_log, "Recovering from manifest file:%s\n",
      current.c_str());

J
jorlow@chromium.org 已提交
1156
  std::string dscname = dbname_ + "/" + current;
1157
  unique_ptr<SequentialFile> file;
1158
  s = env_->NewSequentialFile(dscname, &file, storage_options_);
J
jorlow@chromium.org 已提交
1159 1160 1161
  if (!s.ok()) {
    return s;
  }
1162 1163 1164 1165 1166
  uint64_t manifest_file_size;
  s = env_->GetFileSize(dscname, &manifest_file_size);
  if (!s.ok()) {
    return s;
  }
J
jorlow@chromium.org 已提交
1167 1168

  bool have_log_number = false;
1169
  bool have_prev_log_number = false;
J
jorlow@chromium.org 已提交
1170 1171 1172
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
1173 1174 1175
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
  uint64_t prev_log_number = 0;
J
jorlow@chromium.org 已提交
1176 1177 1178 1179 1180
  Builder builder(this, current_);

  {
    LogReporter reporter;
    reporter.status = &s;
1181 1182
    log::Reader reader(std::move(file), &reporter, true/*checksum*/,
                       0/*initial_offset*/);
J
jorlow@chromium.org 已提交
1183 1184 1185
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
1186
      VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201
      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_) {
1202
        log_number = edit.log_number_;
J
jorlow@chromium.org 已提交
1203 1204 1205
        have_log_number = true;
      }

1206 1207 1208 1209 1210
      if (edit.has_prev_log_number_) {
        prev_log_number = edit.prev_log_number_;
        have_prev_log_number = true;
      }

J
jorlow@chromium.org 已提交
1211 1212 1213 1214 1215 1216
      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

      if (edit.has_last_sequence_) {
1217
        last_sequence = edit.last_sequence_;
J
jorlow@chromium.org 已提交
1218 1219 1220 1221
        have_last_sequence = true;
      }
    }
  }
1222
  file.reset();
J
jorlow@chromium.org 已提交
1223 1224 1225 1226 1227 1228 1229 1230 1231

  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");
    }
1232 1233 1234 1235

    if (!have_prev_log_number) {
      prev_log_number = 0;
    }
1236 1237 1238

    MarkFileNumberUsed(prev_log_number);
    MarkFileNumberUsed(log_number);
J
jorlow@chromium.org 已提交
1239 1240 1241
  }

  if (s.ok()) {
1242
    Version* v = new Version(this, current_version_number_++);
J
jorlow@chromium.org 已提交
1243
    builder.SaveTo(v);
1244

1245
    // Install recovered version
1246 1247 1248 1249
    std::vector<uint64_t> size_being_compacted(NumberLevels()-1);
    SizeBeingCompacted(size_being_compacted);
    Finalize(v, size_being_compacted);

1250
    v->offset_manifest_file_ = manifest_file_size;
1251 1252 1253 1254 1255 1256
    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 已提交
1257

H
heyongqiang 已提交
1258
    Log(options_->info_log, "Recovered from manifest file:%s succeeded,"
1259 1260 1261
        "manifest_file_number is %ld, next_file_number is %ld, "
        "last_sequence is %ld, log_number is %ld,"
        "prev_log_number is %ld\n",
H
heyongqiang 已提交
1262 1263
        current.c_str(), manifest_file_number_, next_file_number_,
        last_sequence_, log_number_, prev_log_number_);
J
jorlow@chromium.org 已提交
1264 1265 1266 1267 1268
  }

  return s;
}

1269
Status VersionSet::DumpManifest(Options& options, std::string& dscname,
Z
Zheng Shao 已提交
1270
    bool verbose, bool hex) {
1271 1272 1273 1274 1275 1276 1277 1278
  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.
1279
  unique_ptr<SequentialFile> file;
1280
  Status s = options.env->NewSequentialFile(dscname, &file, storage_options_);
1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292
  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;
1293
  int count = 0;
1294 1295 1296 1297 1298
  VersionSet::Builder builder(this, current_);

  {
    LogReporter reporter;
    reporter.status = &s;
1299 1300
    log::Reader reader(std::move(file), &reporter, true/*checksum*/,
                       0/*initial_offset*/);
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314
    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());
        }
      }

1315 1316
      // Write out each individual edit
      if (verbose) {
A
Abhishek Kona 已提交
1317
        printf("*************************Edit[%d] = %s\n",
1318 1319 1320 1321
                count, edit.DebugString().c_str());
      }
      count++;

1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346
      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;
      }
    }
  }
1347
  file.reset();
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369

  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()) {
1370
    Version* v = new Version(this, 0);
1371
    builder.SaveTo(v);
1372

1373
    // Install recovered version
1374 1375 1376 1377
    std::vector<uint64_t> size_being_compacted(NumberLevels()-1);
    SizeBeingCompacted(size_being_compacted);
    Finalize(v, size_being_compacted);

1378 1379 1380 1381 1382 1383
    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;
1384

1385
    printf("manifest_file_number %ld next_file_number %ld last_sequence %ld log_number %ld  prev_log_number %ld\n",
1386
           manifest_file_number_, next_file_number_,
1387
           last_sequence, log_number, prev_log_number);
Z
Zheng Shao 已提交
1388
    printf("%s \n", v->DebugString(hex).c_str());
1389
  }
1390

1391 1392 1393
  return s;
}

1394 1395 1396 1397 1398 1399
void VersionSet::MarkFileNumberUsed(uint64_t number) {
  if (next_file_number_ <= number) {
    next_file_number_ = number + 1;
  }
}

1400 1401
void VersionSet::Finalize(Version* v,
  std::vector<uint64_t>& size_being_compacted) {
J
jorlow@chromium.org 已提交
1402

1403
  double max_score = 0;
1404 1405
  int max_score_level = 0;

1406
  for (int level = 0; level < NumberLevels()-1; level++) {
1407
    double score;
J
jorlow@chromium.org 已提交
1408
    if (level == 0) {
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
      // 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).
1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435
      int numfiles = 0;
      for (unsigned int i = 0; i < v->files_[level].size(); i++) {
        if (!v->files_[level][i]->being_compacted) {
          numfiles++;
        }
      }

      // If we are slowing down writes, then we better compact that first
      if (numfiles >= options_->level0_stop_writes_trigger) {
        score = 1000000;
        // Log(options_->info_log, "XXX score l0 = 1000000000 max");
      } else if (numfiles >= options_->level0_slowdown_writes_trigger) {
        score = 10000;
        // Log(options_->info_log, "XXX score l0 = 1000000 medium");
      } else {
        score = numfiles /
1436
          static_cast<double>(options_->level0_file_num_compaction_trigger);
1437 1438 1439 1440
        if (score >= 1) {
          // Log(options_->info_log, "XXX score l0 = %d least", (int)score);
        }
      }
1441 1442
    } else {
      // Compute the ratio of current size to size limit.
1443
      const uint64_t level_bytes = TotalFileSize(v->files_[level]) -
1444
                                   size_being_compacted[level];
1445
      score = static_cast<double>(level_bytes) / MaxBytesForLevel(level);
1446 1447 1448
      if (score > 1) {
        // Log(options_->info_log, "XXX score l%d = %d ", level, (int)score);
      }
1449 1450
      if (max_score < score) {
        max_score = score;
1451
        max_score_level = level;
1452
      }
J
jorlow@chromium.org 已提交
1453
    }
1454 1455 1456 1457
    v->compaction_level_[level] = level;
    v->compaction_score_[level] = score;
  }

1458 1459
  // update the max compaction score in levels 1 to n-1
  v->max_compaction_score_ = max_score;
1460
  v->max_compaction_score_level_ = max_score_level;
1461

1462 1463 1464
  // sort all the levels based on their score. Higher scores get listed
  // first. Use bubble sort because the number of entries are small.
  for(int i = 0; i <  NumberLevels()-2; i++) {
1465
    for (int j = i+1; j < NumberLevels()-1; j++) {
1466 1467 1468 1469 1470 1471 1472 1473
      if (v->compaction_score_[i] < v->compaction_score_[j]) {
        double score = v->compaction_score_[i];
        int level = v->compaction_level_[i];
        v->compaction_score_[i] = v->compaction_score_[j];
        v->compaction_level_[i] = v->compaction_level_[j];
        v->compaction_score_[j] = score;
        v->compaction_level_[j] = level;
      }
J
jorlow@chromium.org 已提交
1474 1475 1476 1477
    }
  }
}

1478
// a static compator used to sort files based on their size
A
Abhishek Kona 已提交
1479
static bool compareSize(const VersionSet::Fsize& first,
1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518
  const VersionSet::Fsize& second) {
  return (first.file->file_size > second.file->file_size);
}

// sort all files in level1 to level(n-1) based on file size
void VersionSet::UpdateFilesBySize(Version* v) {

  // No need to sort the highest level because it is never compacted.
  for (int level = 0; level < NumberLevels()-1; level++) {

    const std::vector<FileMetaData*>& files = v->files_[level];
    std::vector<int>& files_by_size = v->files_by_size_[level];
    assert(files_by_size.size() == 0);

    // populate a temp vector for sorting based on size
    std::vector<Fsize> temp(files.size());
    for (unsigned int i = 0; i < files.size(); i++) {
      temp[i].index = i;
      temp[i].file = files[i];
    }

    // sort the top number_of_files_to_sort_ based on file size
    int num = Version::number_of_files_to_sort_;
    if (num > (int)temp.size()) {
      num = temp.size();
    }
    std::partial_sort(temp.begin(),  temp.begin() + num,
                      temp.end(), compareSize);
    assert(temp.size() == files.size());

    // initialize files_by_size_
    for (unsigned int i = 0; i < temp.size(); i++) {
      files_by_size.push_back(temp[i].index);
    }
    v->next_file_to_compact_by_size_[level] = 0;
    assert(v->files_[level].size() == v->files_by_size_[level].size());
  }
}

J
jorlow@chromium.org 已提交
1519 1520 1521 1522
Status VersionSet::WriteSnapshot(log::Writer* log) {
  // TODO: Break up into multiple records to reduce memory usage on recovery?

  // Save metadata
1523
  VersionEdit edit(NumberLevels());
J
jorlow@chromium.org 已提交
1524 1525 1526
  edit.SetComparatorName(icmp_.user_comparator()->Name());

  // Save compaction pointers
1527
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1528 1529 1530 1531 1532 1533 1534 1535
    if (!compact_pointer_[level].empty()) {
      InternalKey key;
      key.DecodeFrom(compact_pointer_[level]);
      edit.SetCompactPointer(level, key);
    }
  }

  // Save files
1536
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1537
    const std::vector<FileMetaData*>& files = current_->files_[level];
D
dgrogan@chromium.org 已提交
1538
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550
      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);
1551
  assert(level < NumberLevels());
J
jorlow@chromium.org 已提交
1552 1553 1554
  return current_->files_[level].size();
}

1555
const char* VersionSet::LevelSummary(LevelSummaryStorage* scratch) const {
1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
  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, "]");
1566 1567 1568
  return scratch->buffer;
}

1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583
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;
}

1584 1585 1586 1587 1588
// Opens the mainfest file and reads all records
// till it finds the record we are looking for.
bool VersionSet::ManifestContains(const std::string& record) const {
  std::string fname = DescriptorFileName(dbname_, manifest_file_number_);
  Log(options_->info_log, "ManifestContains: checking %s\n", fname.c_str());
1589
  unique_ptr<SequentialFile> file;
1590
  Status s = env_->NewSequentialFile(fname, &file, storage_options_);
1591 1592
  if (!s.ok()) {
    Log(options_->info_log, "ManifestContains: %s\n", s.ToString().c_str());
1593 1594 1595
    Log(options_->info_log,
        "ManifestContains: is unable to reopen the manifest file  %s",
        fname.c_str());
1596 1597
    return false;
  }
A
Abhishek Kona 已提交
1598
  log::Reader reader(std::move(file), nullptr, true/*checksum*/, 0);
1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
  Slice r;
  std::string scratch;
  bool result = false;
  while (reader.ReadRecord(&r, &scratch)) {
    if (r == Slice(record)) {
      result = true;
      break;
    }
  }
  Log(options_->info_log, "ManifestContains: result = %d\n", result ? 1 : 0);
  return result;
}


J
jorlow@chromium.org 已提交
1613 1614
uint64_t VersionSet::ApproximateOffsetOf(Version* v, const InternalKey& ikey) {
  uint64_t result = 0;
1615
  for (int level = 0; level < NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
1616
    const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
1617
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633
      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(
1634 1635
            ReadOptions(), storage_options_, files[i]->number,
            files[i]->file_size, &tableptr);
A
Abhishek Kona 已提交
1636
        if (tableptr != nullptr) {
J
jorlow@chromium.org 已提交
1637 1638 1639 1640 1641 1642 1643 1644 1645
          result += tableptr->ApproximateOffsetOf(ikey.Encode());
        }
        delete iter;
      }
    }
  }
  return result;
}

1646 1647 1648
void VersionSet::AddLiveFiles(std::vector<uint64_t>* live_list) {
  // pre-calculate space requirement
  int64_t total_files = 0;
1649 1650 1651
  for (Version* v = dummy_versions_.next_;
       v != &dummy_versions_;
       v = v->next_) {
1652
    for (int level = 0; level < NumberLevels(); level++) {
1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665
      total_files += v->files_[level].size();
    }
  }

  // just one time extension to the right size
  live_list->reserve(live_list->size() + total_files);

  for (Version* v = dummy_versions_.next_;
       v != &dummy_versions_;
       v = v->next_) {
    for (int level = 0; level < NumberLevels(); level++) {
      for (const auto& f : v->files_[level]) {
        live_list->push_back(f->number);
J
jorlow@chromium.org 已提交
1666 1667 1668 1669 1670
      }
    }
  }
}

1671 1672 1673 1674 1675 1676 1677 1678 1679 1680
void VersionSet::AddLiveFilesCurrentVersion(std::set<uint64_t>* live) {
  Version* v = current_;
  for (int level = 0; level < NumberLevels(); level++) {
    const std::vector<FileMetaData*>& files = v->files_[level];
    for (size_t i = 0; i < files.size(); i++) {
      live->insert(files[i]->number);
    }
  }
}

1681 1682
int64_t VersionSet::NumLevelBytes(int level) const {
  assert(level >= 0);
1683
  assert(level < NumberLevels());
1684 1685
  assert(current_);
  return TotalFileSize(current_->files_[level]);
J
jorlow@chromium.org 已提交
1686 1687 1688 1689
}

int64_t VersionSet::MaxNextLevelOverlappingBytes() {
  int64_t result = 0;
1690
  std::vector<FileMetaData*> overlaps;
1691
  for (int level = 1; level < NumberLevels() - 1; level++) {
D
dgrogan@chromium.org 已提交
1692
    for (size_t i = 0; i < current_->files_[level].size(); i++) {
1693
      const FileMetaData* f = current_->files_[level][i];
G
Gabor Cselle 已提交
1694 1695
      current_->GetOverlappingInputs(level+1, &f->smallest, &f->largest,
                                     &overlaps);
J
jorlow@chromium.org 已提交
1696
      const int64_t sum = TotalFileSize(overlaps);
1697 1698 1699 1700 1701 1702 1703 1704
      if (sum > result) {
        result = sum;
      }
    }
  }
  return result;
}

J
jorlow@chromium.org 已提交
1705 1706 1707 1708 1709 1710 1711 1712 1713
// 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 已提交
1714
  for (size_t i = 0; i < inputs.size(); i++) {
J
jorlow@chromium.org 已提交
1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729
    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;
      }
    }
  }
}

1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741
// 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 已提交
1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756
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 已提交
1757
        for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1758
          list[num++] = table_cache_->NewIterator(
1759 1760
              options, storage_options_compactions_,
              files[i]->number, files[i]->file_size);
J
jorlow@chromium.org 已提交
1761 1762 1763 1764
        }
      } else {
        // Create concatenating iterator for the files from this level
        list[num++] = NewTwoLevelIterator(
1765
            new Version::LevelFileNumIterator(icmp_, &c->inputs_[which]),
1766
            &GetFileIterator, table_cache_, options, storage_options_);
J
jorlow@chromium.org 已提交
1767 1768 1769 1770 1771 1772 1773 1774 1775
      }
    }
  }
  assert(num <= space);
  Iterator* result = NewMergingIterator(&icmp_, list, num);
  delete[] list;
  return result;
}

1776
double VersionSet::MaxBytesForLevel(int level) {
A
Abhishek Kona 已提交
1777 1778
  // Note: the result for level zero is not really used since we set
  // the level-0 compaction threshold based on number of files.
1779 1780
  assert(level >= 0);
  assert(level < NumberLevels());
A
Abhishek Kona 已提交
1781
  return level_max_bytes_[level];
1782 1783 1784 1785 1786
}

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

1790
int64_t VersionSet::ExpandedCompactionByteSizeLimit(int level) {
1791 1792 1793 1794 1795
  uint64_t result = MaxFileSizeForLevel(level);
  result *= options_->expanded_compaction_factor;
  return result;
}

1796
int64_t VersionSet::MaxGrandParentOverlapBytes(int level) {
1797 1798 1799 1800 1801
  uint64_t result = MaxFileSizeForLevel(level);
  result *= options_->max_grandparent_overlap_factor;
  return result;
}

A
Abhishek Kona 已提交
1802
// verify that the files listed in this compaction are present
1803 1804
// in the current version
bool VersionSet::VerifyCompactionFileConsistency(Compaction* c) {
1805
#ifndef NDEBUG
1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
  if (c->input_version_ != current_) {
    Log(options_->info_log, "VerifyCompactionFileConsistency version mismatch");
  }

  // verify files in level
  int level = c->level();
  for (int i = 0; i < c->num_input_files(0); i++) {
    uint64_t number = c->input(0,i)->number;

    // look for this file in the current version
    bool found = false;
    for (unsigned int j = 0; j < current_->files_[level].size(); j++) {
      FileMetaData* f = current_->files_[level][j];
      if (f->number == number) {
        found = true;
        break;
      }
    }
    if (!found) {
      return false; // input files non existant in current version
    }
  }
  // verify level+1 files
  level++;
  for (int i = 0; i < c->num_input_files(1); i++) {
    uint64_t number = c->input(1,i)->number;

    // look for this file in the current version
    bool found = false;
    for (unsigned int j = 0; j < current_->files_[level].size(); j++) {
      FileMetaData* f = current_->files_[level][j];
      if (f->number == number) {
        found = true;
        break;
      }
    }
    if (!found) {
      return false; // input files non existant in current version
    }
  }
1846
#endif
1847 1848 1849 1850 1851
  return true;     // everything good
}

// Clear all files to indicate that they are not being compacted
// Delete this compaction from the list of running compactions.
1852
void VersionSet::ReleaseCompactionFiles(Compaction* c, Status status) {
1853 1854
  c->MarkFilesBeingCompacted(false);
  compactions_in_progress_[c->level()].erase(c);
1855 1856 1857
  if (!status.ok()) {
    c->ResetNextCompactionIndex();
  }
1858 1859 1860
}

// The total size of files that are currently being compacted
1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873
// at at every level upto the penultimate level.
void VersionSet::SizeBeingCompacted(std::vector<uint64_t>& sizes) {
  for (int level = 0; level < NumberLevels()-1; level++) {
    uint64_t total = 0;
    for (std::set<Compaction*>::iterator it =
         compactions_in_progress_[level].begin();
         it != compactions_in_progress_[level].end();
         ++it) {
      Compaction* c = (*it);
      assert(c->level() == level);
      for (int i = 0; i < c->num_input_files(0); i++) {
        total += c->input(0,i)->file_size;
      }
1874
    }
1875
    sizes[level] = total;
1876 1877 1878
  }
}

1879
Compaction* VersionSet::PickCompactionBySize(int level, double score) {
A
Abhishek Kona 已提交
1880
  Compaction* c = nullptr;
1881 1882 1883

  // level 0 files are overlapping. So we cannot pick more
  // than one concurrent compactions at this level. This
1884
  // could be made better by looking at key-ranges that are
1885 1886
  // being compacted at level 0.
  if (level == 0 && compactions_in_progress_[level].size() == 1) {
A
Abhishek Kona 已提交
1887
    return nullptr;
1888 1889 1890 1891 1892 1893
  }

  assert(level >= 0);
  assert(level+1 < NumberLevels());
  c = new Compaction(level, MaxFileSizeForLevel(level),
      MaxGrandParentOverlapBytes(level), NumberLevels());
1894
  c->score_ = score;
1895

1896 1897 1898
  // Pick the largest file in this level that is not already
  // being compacted
  std::vector<int>& file_size = current_->files_by_size_[level];
1899 1900 1901 1902 1903 1904

  // record the first file that is not yet compacted
  int nextIndex = -1;

  for (unsigned int i = current_->next_file_to_compact_by_size_[level];
       i < file_size.size(); i++) {
1905 1906 1907 1908 1909
    int index = file_size[i];
    FileMetaData* f = current_->files_[level][index];

    // check to verify files are arranged in descending size
    assert((i == file_size.size() - 1) ||
1910
           (i >= Version::number_of_files_to_sort_-1) ||
1911
          (f->file_size >= current_->files_[level][file_size[i+1]]->file_size));
1912 1913 1914 1915 1916 1917

    // do not pick a file to compact if it is being compacted
    // from n-1 level.
    if (f->being_compacted) {
      continue;
    }
1918 1919 1920 1921 1922 1923 1924 1925 1926 1927

    // remember the startIndex for the next call to PickCompaction
    if (nextIndex == -1) {
      nextIndex = i;
    }

    //if (i > Version::number_of_files_to_sort_) {
    //  Log(options_->info_log, "XXX Looking at index %d", i);
    //}

1928 1929
    // Do not pick this file if its parents at level+1 are being compacted.
    // Maybe we can avoid redoing this work in SetupOtherInputs
1930
    int parent_index = -1;
A
Abhishek Kona 已提交
1931
    if (ParentRangeInCompaction(&f->smallest, &f->largest, level,
1932
                                &parent_index)) {
1933
      continue;
1934
    }
1935
    c->inputs_[0].push_back(f);
1936
    c->base_index_ = index;
1937
    c->parent_index_ = parent_index;
1938
    break;
1939
  }
1940

1941 1942
  if (c->inputs_[0].empty()) {
    delete c;
A
Abhishek Kona 已提交
1943
    c = nullptr;
1944
  }
1945 1946 1947 1948

  // store where to start the iteration in the next call to PickCompaction
  current_->next_file_to_compact_by_size_[level] = nextIndex;

1949 1950 1951
  return c;
}

J
jorlow@chromium.org 已提交
1952
Compaction* VersionSet::PickCompaction() {
A
Abhishek Kona 已提交
1953
  Compaction* c = nullptr;
1954
  int level = -1;
1955

1956 1957
  // compute the compactions needed. It is better to do it here
  // and also in LogAndApply(), otherwise the values could be stale.
1958 1959 1960
  std::vector<uint64_t> size_being_compacted(NumberLevels()-1);
  current_->vset_->SizeBeingCompacted(size_being_compacted);
  Finalize(current_, size_being_compacted);
1961

1962 1963
  // We prefer compactions triggered by too much data in a level over
  // the compactions triggered by seeks.
1964 1965 1966
  //
  // Find the compactions by size on all levels.
  for (int i = 0; i < NumberLevels()-1; i++) {
1967 1968
    assert(i == 0 || current_->compaction_score_[i] <=
                     current_->compaction_score_[i-1]);
1969 1970
    level = current_->compaction_level_[i];
    if ((current_->compaction_score_[i] >= 1)) {
1971
      c = PickCompactionBySize(level, current_->compaction_score_[i]);
A
Abhishek Kona 已提交
1972
      if (c != nullptr) {
1973 1974 1975
        break;
      }
    }
1976 1977 1978
  }

  // Find compactions needed by seeks
A
Abhishek Kona 已提交
1979
  if (c == nullptr && (current_->file_to_compact_ != nullptr)) {
1980
    level = current_->file_to_compact_level_;
1981 1982 1983 1984 1985 1986 1987

    // Only allow one level 0 compaction at a time.
    if (level != 0 || compactions_in_progress_[0].empty()) {
      c = new Compaction(level, MaxFileSizeForLevel(level),
      MaxGrandParentOverlapBytes(level), NumberLevels(), true);
      c->inputs_[0].push_back(current_->file_to_compact_);
    }
1988 1989
  }

A
Abhishek Kona 已提交
1990 1991
  if (c == nullptr) {
    return nullptr;
J
jorlow@chromium.org 已提交
1992 1993 1994 1995 1996 1997
  }

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

  // Files in level 0 may overlap each other, so pick up all overlapping ones
1998 1999
  // Two level 0 compaction won't run at the same time, so don't need to worry
  // about files on level 0 being compacted.
J
jorlow@chromium.org 已提交
2000
  if (level == 0) {
2001
    assert(compactions_in_progress_[0].empty());
2002 2003
    InternalKey smallest, largest;
    GetRange(c->inputs_[0], &smallest, &largest);
J
jorlow@chromium.org 已提交
2004 2005 2006
    // 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.
2007
    c->inputs_[0].clear();
2008
    current_->GetOverlappingInputs(0, &smallest, &largest, &c->inputs_[0]);
2009 2010 2011
    if (ParentRangeInCompaction(&smallest, &largest,
                                level, &c->parent_index_)) {
      delete c;
A
Abhishek Kona 已提交
2012
      return nullptr;
2013
    }
J
jorlow@chromium.org 已提交
2014 2015 2016
    assert(!c->inputs_[0].empty());
  }

2017 2018
  SetupOtherInputs(c);

2019 2020 2021 2022 2023 2024
  // mark all the files that are being compacted
  c->MarkFilesBeingCompacted(true);

  // remember this currently undergoing compaction
  compactions_in_progress_[level].insert(c);

2025 2026 2027
  return c;
}

2028
// Returns true if any one of the parent files are being compacted
2029 2030
bool VersionSet::ParentRangeInCompaction(const InternalKey* smallest,
  const InternalKey* largest, int level, int* parent_index) {
2031
  std::vector<FileMetaData*> inputs;
A
Abhishek Kona 已提交
2032

2033
  current_->GetOverlappingInputs(level+1, smallest, largest,
2034
                                 &inputs, *parent_index, parent_index);
2035 2036 2037 2038 2039 2040 2041 2042
  return FilesInCompaction(inputs);
}

// Returns true if any one of specified files are being compacted
bool VersionSet::FilesInCompaction(std::vector<FileMetaData*>& files) {
  for (unsigned int i = 0; i < files.size(); i++) {
    if (files[i]->being_compacted) {
      return true;
A
Abhishek Kona 已提交
2043
    }
2044 2045 2046 2047
  }
  return false;
}

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

2053 2054
  current_->GetOverlappingInputs(level+1, &smallest, &largest, &c->inputs_[1],
                                 c->parent_index_, &c->parent_index_);
J
jorlow@chromium.org 已提交
2055

2056 2057 2058 2059
  // 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 已提交
2060 2061 2062 2063
  // 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;
2064
    current_->GetOverlappingInputs(level, &all_start, &all_limit, &expanded0,
A
Abhishek Kona 已提交
2065
                                   c->base_index_, nullptr);
2066 2067 2068
    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);
2069
    int64_t limit = ExpandedCompactionByteSizeLimit(level);
2070
    if (expanded0.size() > c->inputs_[0].size() &&
2071 2072
        inputs1_size + expanded0_size < limit &&
        !FilesInCompaction(expanded0)) {
J
jorlow@chromium.org 已提交
2073 2074 2075
      InternalKey new_start, new_limit;
      GetRange(expanded0, &new_start, &new_limit);
      std::vector<FileMetaData*> expanded1;
G
Gabor Cselle 已提交
2076
      current_->GetOverlappingInputs(level+1, &new_start, &new_limit,
2077 2078
                                     &expanded1, c->parent_index_,
                                     &c->parent_index_);
A
Abhishek Kona 已提交
2079
      if (expanded1.size() == c->inputs_[1].size() &&
2080
          !FilesInCompaction(expanded1)) {
2081
        Log(options_->info_log,
2082
            "Expanding@%d %d+%d (%ld+%ld bytes) to %d+%d (%ld+%ld bytes)\n",
J
jorlow@chromium.org 已提交
2083 2084 2085
            level,
            int(c->inputs_[0].size()),
            int(c->inputs_[1].size()),
2086
            long(inputs0_size), long(inputs1_size),
J
jorlow@chromium.org 已提交
2087
            int(expanded0.size()),
2088 2089
            int(expanded1.size()),
            long(expanded0_size), long(inputs1_size));
J
jorlow@chromium.org 已提交
2090 2091 2092 2093
        smallest = new_start;
        largest = new_limit;
        c->inputs_[0] = expanded0;
        c->inputs_[1] = expanded1;
2094
        GetRange2(c->inputs_[0], c->inputs_[1], &all_start, &all_limit);
J
jorlow@chromium.org 已提交
2095 2096 2097 2098
      }
    }
  }

2099 2100
  // Compute the set of grandparent files that overlap this compaction
  // (parent == level+1; grandparent == level+2)
2101
  if (level + 2 < NumberLevels()) {
G
Gabor Cselle 已提交
2102 2103
    current_->GetOverlappingInputs(level + 2, &all_start, &all_limit,
                                   &c->grandparents_);
2104 2105
  }

J
jorlow@chromium.org 已提交
2106
  if (false) {
2107
    Log(options_->info_log, "Compacting %d '%s' .. '%s'",
J
jorlow@chromium.org 已提交
2108
        level,
G
Gabor Cselle 已提交
2109 2110
        smallest.DebugString().c_str(),
        largest.DebugString().c_str());
J
jorlow@chromium.org 已提交
2111 2112 2113 2114 2115 2116 2117
  }

  // 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();
2118
  c->edit_->SetCompactPointer(level, largest);
J
jorlow@chromium.org 已提交
2119 2120 2121 2122
}

Compaction* VersionSet::CompactRange(
    int level,
G
Gabor Cselle 已提交
2123 2124
    const InternalKey* begin,
    const InternalKey* end) {
J
jorlow@chromium.org 已提交
2125
  std::vector<FileMetaData*> inputs;
G
Gabor Cselle 已提交
2126
  current_->GetOverlappingInputs(level, begin, end, &inputs);
J
jorlow@chromium.org 已提交
2127
  if (inputs.empty()) {
A
Abhishek Kona 已提交
2128
    return nullptr;
J
jorlow@chromium.org 已提交
2129 2130
  }

G
Gabor Cselle 已提交
2131
  // Avoid compacting too much in one shot in case the range is large.
2132 2133
  const uint64_t limit = MaxFileSizeForLevel(level) *
                         options_->source_compaction_factor;
G
Gabor Cselle 已提交
2134
  uint64_t total = 0;
H
heyongqiang 已提交
2135
  for (size_t i = 0; i < inputs.size(); i++) {
G
Gabor Cselle 已提交
2136 2137 2138 2139 2140 2141 2142 2143
    uint64_t s = inputs[i]->file_size;
    total += s;
    if (total >= limit) {
      inputs.resize(i + 1);
      break;
    }
  }

2144
  Compaction* c = new Compaction(level, MaxFileSizeForLevel(level),
2145
    MaxGrandParentOverlapBytes(level), NumberLevels());
J
jorlow@chromium.org 已提交
2146 2147 2148
  c->input_version_ = current_;
  c->input_version_->Ref();
  c->inputs_[0] = inputs;
2149
  SetupOtherInputs(c);
2150 2151 2152 2153 2154

  // These files that are to be manaully compacted do not trample
  // upon other files because manual compactions are processed when
  // the system has a max of 1 background compaction thread.
  c->MarkFilesBeingCompacted(true);
J
jorlow@chromium.org 已提交
2155 2156 2157
  return c;
}

2158
Compaction::Compaction(int level, uint64_t target_file_size,
2159 2160
  uint64_t max_grandparent_overlap_bytes, int number_levels,
  bool seek_compaction)
J
jorlow@chromium.org 已提交
2161
    : level_(level),
2162 2163
      max_output_file_size_(target_file_size),
      maxGrandParentOverlapBytes_(max_grandparent_overlap_bytes),
A
Abhishek Kona 已提交
2164
      input_version_(nullptr),
2165
      number_levels_(number_levels),
2166
      seek_compaction_(seek_compaction),
2167
      grandparent_index_(0),
J
jorlow@chromium.org 已提交
2168
      seen_key_(false),
2169 2170
      overlapped_bytes_(0),
      base_index_(-1),
2171
      parent_index_(-1),
2172 2173
      score_(0),
      level_ptrs_(std::vector<size_t>(number_levels)) {
2174
  edit_ = new VersionEdit(number_levels_);
2175
  for (int i = 0; i < number_levels_; i++) {
J
jorlow@chromium.org 已提交
2176 2177 2178 2179 2180
    level_ptrs_[i] = 0;
  }
}

Compaction::~Compaction() {
A
Abhishek Kona 已提交
2181
  delete edit_;
A
Abhishek Kona 已提交
2182
  if (input_version_ != nullptr) {
J
jorlow@chromium.org 已提交
2183 2184 2185 2186
    input_version_->Unref();
  }
}

2187
bool Compaction::IsTrivialMove() const {
J
jorlow@chromium.org 已提交
2188
  // Avoid a move if there is lots of overlapping grandparent data.
2189 2190
  // Otherwise, the move could create a parent file that will require
  // a very expensive merge later on.
J
jorlow@chromium.org 已提交
2191 2192
  return (num_input_files(0) == 1 &&
          num_input_files(1) == 0 &&
2193
          TotalFileSize(grandparents_) <= maxGrandParentOverlapBytes_);
2194 2195
}

J
jorlow@chromium.org 已提交
2196 2197
void Compaction::AddInputDeletions(VersionEdit* edit) {
  for (int which = 0; which < 2; which++) {
D
dgrogan@chromium.org 已提交
2198
    for (size_t i = 0; i < inputs_[which].size(); i++) {
J
jorlow@chromium.org 已提交
2199 2200 2201 2202 2203 2204 2205 2206
      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();
2207
  for (int lvl = level_ + 2; lvl < number_levels_; lvl++) {
J
jorlow@chromium.org 已提交
2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224
    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;
}

2225
bool Compaction::ShouldStopBefore(const Slice& internal_key) {
2226 2227 2228
  // Scan to find earliest grandparent file that contains key.
  const InternalKeyComparator* icmp = &input_version_->vset_->icmp_;
  while (grandparent_index_ < grandparents_.size() &&
2229 2230
      icmp->Compare(internal_key,
                    grandparents_[grandparent_index_]->largest.Encode()) > 0) {
J
jorlow@chromium.org 已提交
2231 2232 2233
    if (seen_key_) {
      overlapped_bytes_ += grandparents_[grandparent_index_]->file_size;
    }
2234 2235 2236 2237
    assert(grandparent_index_ + 1 >= grandparents_.size() ||
           icmp->Compare(grandparents_[grandparent_index_]->largest.Encode(),
                         grandparents_[grandparent_index_+1]->smallest.Encode())
                         < 0);
2238 2239
    grandparent_index_++;
  }
J
jorlow@chromium.org 已提交
2240
  seen_key_ = true;
2241

2242
  if (overlapped_bytes_ > maxGrandParentOverlapBytes_) {
J
jorlow@chromium.org 已提交
2243 2244
    // Too much overlap for current output; start new output
    overlapped_bytes_ = 0;
2245 2246 2247 2248 2249 2250
    return true;
  } else {
    return false;
  }
}

2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262
// Mark (or clear) each file that is being compacted
void Compaction::MarkFilesBeingCompacted(bool value) {
  for (int i = 0; i < 2; i++) {
    std::vector<FileMetaData*> v = inputs_[i];
    for (unsigned int j = 0; j < inputs_[i].size(); j++) {
      assert(value ? !inputs_[i][j]->being_compacted :
                      inputs_[i][j]->being_compacted);
      inputs_[i][j]->being_compacted = value;
    }
  }
}

J
jorlow@chromium.org 已提交
2263
void Compaction::ReleaseInputs() {
A
Abhishek Kona 已提交
2264
  if (input_version_ != nullptr) {
J
jorlow@chromium.org 已提交
2265
    input_version_->Unref();
A
Abhishek Kona 已提交
2266
    input_version_ = nullptr;
J
jorlow@chromium.org 已提交
2267 2268 2269
  }
}

2270
void Compaction::ResetNextCompactionIndex() {
A
Abhishek Kona 已提交
2271
  input_version_->ResetNextCompactionIndex(level_);
2272 2273
}

H
heyongqiang 已提交
2274 2275 2276 2277
static void InputSummary(std::vector<FileMetaData*>& files,
    char* output,
    int len) {
  int write = 0;
2278
  for (unsigned int i = 0; i < files.size(); i++) {
H
heyongqiang 已提交
2279
    int sz = len - write;
2280
    int ret = snprintf(output + write, sz, "%lu(%lu) ",
H
heyongqiang 已提交
2281 2282 2283 2284 2285 2286 2287 2288 2289
        files.at(i)->number,
        files.at(i)->file_size);
    if (ret < 0 || ret >= sz)
      break;
    write += ret;
  }
}

void Compaction::Summary(char* output, int len) {
2290
  int write = snprintf(output, len,
2291 2292
      "Base version %ld Base level %d, seek compaction:%d, inputs:",
      input_version_->GetVersionNumber(), level_, seek_compaction_);
H
heyongqiang 已提交
2293 2294 2295 2296
  if(write < 0 || write > len)
    return;

  char level_low_summary[100];
2297
  InputSummary(inputs_[0], level_low_summary, sizeof(level_low_summary));
H
heyongqiang 已提交
2298 2299
  char level_up_summary[100];
  if (inputs_[1].size()) {
2300
    InputSummary(inputs_[1], level_up_summary, sizeof(level_up_summary));
H
heyongqiang 已提交
2301 2302 2303 2304 2305 2306 2307 2308
  } else {
    level_up_summary[0] = '\0';
  }

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

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