version_set.cc 100.0 KB
Newer Older
1 2 3 4 5
//  Copyright (c) 2013, Facebook, Inc.  All rights reserved.
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
//
J
jorlow@chromium.org 已提交
6 7 8 9 10
// 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"
I
Igor Canadi 已提交
11

L
liuhuahang 已提交
12
#ifndef __STDC_FORMAT_MACROS
I
Igor Canadi 已提交
13
#define __STDC_FORMAT_MACROS
L
liuhuahang 已提交
14 15
#endif

I
Igor Canadi 已提交
16
#include <inttypes.h>
J
jorlow@chromium.org 已提交
17
#include <algorithm>
I
Igor Canadi 已提交
18
#include <map>
I
Igor Canadi 已提交
19
#include <set>
20
#include <climits>
21
#include <unordered_map>
22
#include <vector>
J
jorlow@chromium.org 已提交
23
#include <stdio.h>
K
kailiu 已提交
24

J
jorlow@chromium.org 已提交
25 26 27 28
#include "db/filename.h"
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
29
#include "db/merge_context.h"
J
jorlow@chromium.org 已提交
30
#include "db/table_cache.h"
31
#include "db/compaction.h"
32 33
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
K
kailiu 已提交
34
#include "table/table_reader.h"
J
jorlow@chromium.org 已提交
35 36
#include "table/merger.h"
#include "table/two_level_iterator.h"
37
#include "table/format.h"
38
#include "table/plain_table_factory.h"
39
#include "table/meta_blocks.h"
40
#include "table/get_context.h"
J
jorlow@chromium.org 已提交
41 42
#include "util/coding.h"
#include "util/logging.h"
43
#include "util/stop_watch.h"
J
jorlow@chromium.org 已提交
44

45
namespace rocksdb {
J
jorlow@chromium.org 已提交
46

47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
namespace {

// Find File in FileLevel data structure
// Within an index range defined by left and right
int FindFileInRange(const InternalKeyComparator& icmp,
    const FileLevel& file_level,
    const Slice& key,
    uint32_t left,
    uint32_t right) {
  while (left < right) {
    uint32_t mid = (left + right) / 2;
    const FdWithKeyRange& f = file_level.files[mid];
    if (icmp.InternalKeyComparator::Compare(f.largest_key, 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;
}

bool NewestFirstBySeqNo(FileMetaData* a, FileMetaData* b) {
  if (a->smallest_seqno != b->smallest_seqno) {
    return a->smallest_seqno > b->smallest_seqno;
  }
  if (a->largest_seqno != b->largest_seqno) {
    return a->largest_seqno > b->largest_seqno;
  }
  // Break ties by file number
  return a->fd.GetNumber() > b->fd.GetNumber();
}

bool BySmallestKey(FileMetaData* a, FileMetaData* b,
                   const InternalKeyComparator* cmp) {
  int r = cmp->Compare(a->smallest, b->smallest);
  if (r != 0) {
    return (r < 0);
  }
  // Break ties by file number
  return (a->fd.GetNumber() < b->fd.GetNumber());
}

// Class to help choose the next file to search for the particular key.
// Searches and returns files level by level.
// We can search level-by-level since entries never hop across
// levels. Therefore we are guaranteed that if we find data
// in a smaller level, later levels are irrelevant (unless we
// are MergeInProgress).
class FilePicker {
 public:
  FilePicker(
      std::vector<FileMetaData*>* files,
      const Slice& user_key,
      const Slice& ikey,
      autovector<FileLevel>* file_levels,
      unsigned int num_levels,
      FileIndexer* file_indexer,
      const Comparator* user_comparator,
      const InternalKeyComparator* internal_comparator)
      : num_levels_(num_levels),
        curr_level_(-1),
        search_left_bound_(0),
        search_right_bound_(FileIndexer::kLevelMaxIndex),
114
#ifndef NDEBUG
115
        files_(files),
116
#endif
117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
        file_levels_(file_levels),
        user_key_(user_key),
        ikey_(ikey),
        file_indexer_(file_indexer),
        user_comparator_(user_comparator),
        internal_comparator_(internal_comparator) {
    // Setup member variables to search first level.
    search_ended_ = !PrepareNextLevel();
    if (!search_ended_) {
      // Prefetch Level 0 table data to avoid cache miss if possible.
      for (unsigned int i = 0; i < (*file_levels_)[0].num_files; ++i) {
        auto* r = (*file_levels_)[0].files[i].fd.table_reader;
        if (r) {
          r->Prepare(ikey);
        }
      }
    }
  }

  FdWithKeyRange* GetNextFile() {
    while (!search_ended_) {  // Loops over different levels.
      while (curr_index_in_curr_level_ < curr_file_level_->num_files) {
        // Loops over all files in current level.
        FdWithKeyRange* f = &curr_file_level_->files[curr_index_in_curr_level_];
        int cmp_largest = -1;

        // Do key range filtering of files or/and fractional cascading if:
        // (1) not all the files are in level 0, or
        // (2) there are more than 3 Level 0 files
        // If there are only 3 or less level 0 files in the system, we skip
        // the key range filtering. In this case, more likely, the system is
        // highly tuned to minimize number of tables queried by each query,
        // so it is unlikely that key range filtering is more efficient than
        // querying the files.
        if (num_levels_ > 1 || curr_file_level_->num_files > 3) {
          // Check if key is within a file's range. If search left bound and
          // right bound point to the same find, we are sure key falls in
          // range.
          assert(
              curr_level_ == 0 ||
              curr_index_in_curr_level_ == start_index_in_curr_level_ ||
              user_comparator_->Compare(user_key_,
                ExtractUserKey(f->smallest_key)) <= 0);

          int cmp_smallest = user_comparator_->Compare(user_key_,
              ExtractUserKey(f->smallest_key));
          if (cmp_smallest >= 0) {
            cmp_largest = user_comparator_->Compare(user_key_,
                ExtractUserKey(f->largest_key));
          }

          // Setup file search bound for the next level based on the
          // comparison results
          if (curr_level_ > 0) {
            file_indexer_->GetNextLevelIndex(curr_level_,
                                            curr_index_in_curr_level_,
                                            cmp_smallest, cmp_largest,
                                            &search_left_bound_,
                                            &search_right_bound_);
          }
          // Key falls out of current file's range
          if (cmp_smallest < 0 || cmp_largest > 0) {
            if (curr_level_ == 0) {
              ++curr_index_in_curr_level_;
              continue;
            } else {
              // Search next level.
              break;
            }
          }
        }
#ifndef NDEBUG
        // Sanity check to make sure that the files are correctly sorted
        if (prev_file_) {
          if (curr_level_ != 0) {
            int comp_sign = internal_comparator_->Compare(
                prev_file_->largest_key, f->smallest_key);
            assert(comp_sign < 0);
          } else {
            // level == 0, the current file cannot be newer than the previous
            // one. Use compressed data structure, has no attribute seqNo
            assert(curr_index_in_curr_level_ > 0);
            assert(!NewestFirstBySeqNo(files_[0][curr_index_in_curr_level_],
                  files_[0][curr_index_in_curr_level_-1]));
          }
        }
        prev_file_ = f;
#endif
        if (curr_level_ > 0 && cmp_largest < 0) {
          // No more files to search in this level.
          search_ended_ = !PrepareNextLevel();
        } else {
          ++curr_index_in_curr_level_;
        }
        return f;
      }
      // Start searching next level.
      search_ended_ = !PrepareNextLevel();
    }
    // Search ended.
    return nullptr;
  }

 private:
  unsigned int num_levels_;
  unsigned int curr_level_;
  int search_left_bound_;
  int search_right_bound_;
225
#ifndef NDEBUG
226
  std::vector<FileMetaData*>* files_;
227
#endif
228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307
  autovector<FileLevel>* file_levels_;
  bool search_ended_;
  FileLevel* curr_file_level_;
  unsigned int curr_index_in_curr_level_;
  unsigned int start_index_in_curr_level_;
  Slice user_key_;
  Slice ikey_;
  FileIndexer* file_indexer_;
  const Comparator* user_comparator_;
  const InternalKeyComparator* internal_comparator_;
#ifndef NDEBUG
  FdWithKeyRange* prev_file_;
#endif

  // Setup local variables to search next level.
  // Returns false if there are no more levels to search.
  bool PrepareNextLevel() {
    curr_level_++;
    while (curr_level_ < num_levels_) {
      curr_file_level_ = &(*file_levels_)[curr_level_];
      if (curr_file_level_->num_files == 0) {
        // When current level is empty, the search bound generated from upper
        // level must be [0, -1] or [0, FileIndexer::kLevelMaxIndex] if it is
        // also empty.
        assert(search_left_bound_ == 0);
        assert(search_right_bound_ == -1 ||
               search_right_bound_ == FileIndexer::kLevelMaxIndex);
        // Since current level is empty, it will need to search all files in
        // the next level
        search_left_bound_ = 0;
        search_right_bound_ = FileIndexer::kLevelMaxIndex;
        curr_level_++;
        continue;
      }

      // Some files may overlap each other. We find
      // all files that overlap user_key and process them in order from
      // newest to oldest. In the context of merge-operator, this can occur at
      // any level. Otherwise, it only occurs at Level-0 (since Put/Deletes
      // are always compacted into a single entry).
      int32_t start_index;
      if (curr_level_ == 0) {
        // On Level-0, we read through all files to check for overlap.
        start_index = 0;
      } else {
        // On Level-n (n>=1), files are sorted. Binary search to find the
        // earliest file whose largest key >= ikey. Search left bound and
        // right bound are used to narrow the range.
        if (search_left_bound_ == search_right_bound_) {
          start_index = search_left_bound_;
        } else if (search_left_bound_ < search_right_bound_) {
          if (search_right_bound_ == FileIndexer::kLevelMaxIndex) {
            search_right_bound_ = curr_file_level_->num_files - 1;
          }
          start_index = FindFileInRange(*internal_comparator_,
              *curr_file_level_, ikey_,
              search_left_bound_, search_right_bound_);
        } else {
          // search_left_bound > search_right_bound, key does not exist in
          // this level. Since no comparision is done in this level, it will
          // need to search all files in the next level.
          search_left_bound_ = 0;
          search_right_bound_ = FileIndexer::kLevelMaxIndex;
          curr_level_++;
          continue;
        }
      }
      start_index_in_curr_level_ = start_index;
      curr_index_in_curr_level_ = start_index;
#ifndef NDEBUG
      prev_file_ = nullptr;
#endif
      return true;
    }
    // curr_level_ = num_levels_. So, no more levels to search.
    return false;
  }
};
}  // anonymous namespace

J
jorlow@chromium.org 已提交
308 309
Version::~Version() {
  assert(refs_ == 0);
310 311 312 313 314 315

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

  // Drop references to files
316
  for (int level = 0; level < num_levels_; level++) {
D
dgrogan@chromium.org 已提交
317
    for (size_t i = 0; i < files_[level].size(); i++) {
J
jorlow@chromium.org 已提交
318
      FileMetaData* f = files_[level][i];
319
      assert(f->refs > 0);
J
jorlow@chromium.org 已提交
320 321
      f->refs--;
      if (f->refs <= 0) {
322
        if (f->table_reader_handle) {
323
          cfd_->table_cache()->ReleaseHandle(f->table_reader_handle);
324 325
          f->table_reader_handle = nullptr;
        }
I
Igor Canadi 已提交
326
        vset_->obsolete_files_.push_back(f);
J
jorlow@chromium.org 已提交
327 328 329
      }
    }
  }
330
  delete[] files_;
J
jorlow@chromium.org 已提交
331 332
}

333 334 335 336 337 338
int FindFile(const InternalKeyComparator& icmp,
             const FileLevel& file_level,
             const Slice& key) {
  return FindFileInRange(icmp, file_level, key, 0, file_level.num_files);
}

F
Feng Zhu 已提交
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368
void DoGenerateFileLevel(FileLevel* file_level,
        const std::vector<FileMetaData*>& files,
        Arena* arena) {
  assert(file_level);
  assert(files.size() >= 0);
  assert(arena);

  size_t num = files.size();
  file_level->num_files = num;
  char* mem = arena->AllocateAligned(num * sizeof(FdWithKeyRange));
  file_level->files = new (mem)FdWithKeyRange[num];

  for (size_t i = 0; i < num; i++) {
    Slice smallest_key = files[i]->smallest.Encode();
    Slice largest_key = files[i]->largest.Encode();

    // Copy key slice to sequential memory
    size_t smallest_size = smallest_key.size();
    size_t largest_size = largest_key.size();
    mem = arena->AllocateAligned(smallest_size + largest_size);
    memcpy(mem, smallest_key.data(), smallest_size);
    memcpy(mem + smallest_size, largest_key.data(), largest_size);

    FdWithKeyRange& f = file_level->files[i];
    f.fd = files[i]->fd;
    f.smallest_key = Slice(mem, smallest_size);
    f.largest_key = Slice(mem + smallest_size, largest_size);
  }
}

G
Gabor Cselle 已提交
369
static bool AfterFile(const Comparator* ucmp,
370
                      const Slice* user_key, const FdWithKeyRange* f) {
A
Abhishek Kona 已提交
371 372
  // nullptr user_key occurs before all keys and is therefore never after *f
  return (user_key != nullptr &&
373
          ucmp->Compare(*user_key, ExtractUserKey(f->largest_key)) > 0);
G
Gabor Cselle 已提交
374 375 376
}

static bool BeforeFile(const Comparator* ucmp,
377
                       const Slice* user_key, const FdWithKeyRange* f) {
A
Abhishek Kona 已提交
378 379
  // nullptr user_key occurs after all keys and is therefore never before *f
  return (user_key != nullptr &&
380
          ucmp->Compare(*user_key, ExtractUserKey(f->smallest_key)) < 0);
G
Gabor Cselle 已提交
381 382
}

383 384
bool SomeFileOverlapsRange(
    const InternalKeyComparator& icmp,
G
Gabor Cselle 已提交
385
    bool disjoint_sorted_files,
386
    const FileLevel& file_level,
G
Gabor Cselle 已提交
387 388 389 390 391
    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
392 393
    for (size_t i = 0; i < file_level.num_files; i++) {
      const FdWithKeyRange* f = &(file_level.files[i]);
G
Gabor Cselle 已提交
394 395 396 397 398 399 400 401 402 403 404 405
      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 已提交
406
  if (smallest_user_key != nullptr) {
G
Gabor Cselle 已提交
407 408
    // Find the earliest possible internal key for smallest_user_key
    InternalKey small(*smallest_user_key, kMaxSequenceNumber,kValueTypeForSeek);
409
    index = FindFile(icmp, file_level, small.Encode());
G
Gabor Cselle 已提交
410 411
  }

412
  if (index >= file_level.num_files) {
G
Gabor Cselle 已提交
413 414 415 416
    // beginning of range is after all files, so no overlap.
    return false;
  }

417
  return !BeforeFile(ucmp, largest_user_key, &file_level.files[index]);
418 419
}

J
jorlow@chromium.org 已提交
420 421 422
// 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 已提交
423 424
// 16-byte value containing the file number and file size, both
// encoded using EncodeFixed64.
J
jorlow@chromium.org 已提交
425 426
class Version::LevelFileNumIterator : public Iterator {
 public:
427
  LevelFileNumIterator(const InternalKeyComparator& icmp,
F
Feng Zhu 已提交
428
                       const FileLevel* flevel)
429
      : icmp_(icmp),
F
Feng Zhu 已提交
430 431
        flevel_(flevel),
        index_(flevel->num_files),
432
        current_value_(0, 0, 0) {  // Marks as invalid
J
jorlow@chromium.org 已提交
433 434
  }
  virtual bool Valid() const {
F
Feng Zhu 已提交
435
    return index_ < flevel_->num_files;
J
jorlow@chromium.org 已提交
436 437
  }
  virtual void Seek(const Slice& target) {
F
Feng Zhu 已提交
438
    index_ = FindFile(icmp_, *flevel_, target);
J
jorlow@chromium.org 已提交
439 440 441
  }
  virtual void SeekToFirst() { index_ = 0; }
  virtual void SeekToLast() {
F
Feng Zhu 已提交
442
    index_ = (flevel_->num_files == 0) ? 0 : flevel_->num_files - 1;
J
jorlow@chromium.org 已提交
443 444 445 446 447 448 449 450
  }
  virtual void Next() {
    assert(Valid());
    index_++;
  }
  virtual void Prev() {
    assert(Valid());
    if (index_ == 0) {
F
Feng Zhu 已提交
451
      index_ = flevel_->num_files;  // Marks as invalid
J
jorlow@chromium.org 已提交
452 453 454 455 456 457
    } else {
      index_--;
    }
  }
  Slice key() const {
    assert(Valid());
F
Feng Zhu 已提交
458
    return flevel_->files[index_].largest_key;
J
jorlow@chromium.org 已提交
459 460 461
  }
  Slice value() const {
    assert(Valid());
F
Feng Zhu 已提交
462 463 464

    auto file_meta = flevel_->files[index_];
    current_value_ = file_meta.fd;
465
    return Slice(reinterpret_cast<const char*>(&current_value_),
466
                 sizeof(FileDescriptor));
J
jorlow@chromium.org 已提交
467 468 469 470
  }
  virtual Status status() const { return Status::OK(); }
 private:
  const InternalKeyComparator icmp_;
F
Feng Zhu 已提交
471
  const FileLevel* flevel_;
D
dgrogan@chromium.org 已提交
472
  uint32_t index_;
473
  mutable FileDescriptor current_value_;
J
jorlow@chromium.org 已提交
474 475
};

L
Lei Jin 已提交
476 477 478 479 480 481 482 483 484 485 486 487
class Version::LevelFileIteratorState : public TwoLevelIteratorState {
 public:
  LevelFileIteratorState(TableCache* table_cache,
    const ReadOptions& read_options, const EnvOptions& env_options,
    const InternalKeyComparator& icomparator, bool for_compaction,
    bool prefix_enabled)
    : TwoLevelIteratorState(prefix_enabled),
      table_cache_(table_cache), read_options_(read_options),
      env_options_(env_options), icomparator_(icomparator),
      for_compaction_(for_compaction) {}

  Iterator* NewSecondaryIterator(const Slice& meta_handle) override {
488
    if (meta_handle.size() != sizeof(FileDescriptor)) {
L
Lei Jin 已提交
489 490 491
      return NewErrorIterator(
          Status::Corruption("FileReader invoked with unexpected value"));
    } else {
492 493 494 495 496
      const FileDescriptor* fd =
          reinterpret_cast<const FileDescriptor*>(meta_handle.data());
      return table_cache_->NewIterator(
          read_options_, env_options_, icomparator_, *fd,
          nullptr /* don't need reference to table*/, for_compaction_);
L
Lei Jin 已提交
497
    }
J
jorlow@chromium.org 已提交
498 499
  }

L
Lei Jin 已提交
500 501 502 503 504 505 506 507 508 509 510
  bool PrefixMayMatch(const Slice& internal_key) override {
    return true;
  }

 private:
  TableCache* table_cache_;
  const ReadOptions read_options_;
  const EnvOptions& env_options_;
  const InternalKeyComparator& icomparator_;
  bool for_compaction_;
};
T
Tyler Harter 已提交
511

512 513 514
Status Version::GetTableProperties(std::shared_ptr<const TableProperties>* tp,
                                   const FileMetaData* file_meta,
                                   const std::string* fname) {
I
Igor Canadi 已提交
515
  auto table_cache = cfd_->table_cache();
516
  auto ioptions = cfd_->ioptions();
517
  Status s = table_cache->GetTableProperties(
518
      vset_->env_options_, cfd_->internal_comparator(), file_meta->fd,
519 520 521 522 523 524 525 526 527 528 529 530 531 532 533
      tp, true /* no io */);
  if (s.ok()) {
    return s;
  }

  // We only ignore error type `Incomplete` since it's by design that we
  // disallow table when it's not in table cache.
  if (!s.IsIncomplete()) {
    return s;
  }

  // 2. Table is not present in table cache, we'll read the table properties
  // directly from the properties block in the file.
  std::unique_ptr<RandomAccessFile> file;
  if (fname != nullptr) {
534
    s = ioptions->env->NewRandomAccessFile(
535
        *fname, &file, vset_->env_options_);
536
  } else {
537
    s = ioptions->env->NewRandomAccessFile(
538
        TableFileName(vset_->db_options_->db_paths, file_meta->fd.GetNumber(),
539
                      file_meta->fd.GetPathId()),
540
        &file, vset_->env_options_);
541 542 543 544 545 546 547 548 549 550 551
  }
  if (!s.ok()) {
    return s;
  }

  TableProperties* raw_table_properties;
  // By setting the magic number to kInvalidTableMagicNumber, we can by
  // pass the magic number check in the footer.
  s = ReadTableProperties(
      file.get(), file_meta->fd.GetFileSize(),
      Footer::kInvalidTableMagicNumber /* table's magic number */,
552
      vset_->env_, ioptions->info_log, &raw_table_properties);
553 554 555
  if (!s.ok()) {
    return s;
  }
556
  RecordTick(ioptions->statistics, NUMBER_DIRECT_LOAD_TABLE_PROPERTIES);
557 558 559 560 561 562

  *tp = std::shared_ptr<const TableProperties>(raw_table_properties);
  return s;
}

Status Version::GetPropertiesOfAllTables(TablePropertiesCollection* props) {
563 564
  for (int level = 0; level < num_levels_; level++) {
    for (const auto& file_meta : files_[level]) {
565
      auto fname =
566
          TableFileName(vset_->db_options_->db_paths, file_meta->fd.GetNumber(),
567
                        file_meta->fd.GetPathId());
568 569 570
      // 1. If the table is already present in table cache, load table
      // properties from there.
      std::shared_ptr<const TableProperties> table_properties;
571
      Status s = GetTableProperties(&table_properties, file_meta, &fname);
572 573
      if (s.ok()) {
        props->insert({fname, table_properties});
574
      } else {
575 576 577 578 579 580 581 582
        return s;
      }
    }
  }

  return Status::OK();
}

583 584 585 586 587
size_t Version::GetMemoryUsageByTableReaders() {
  size_t total_usage = 0;
  for (auto& file_level : file_levels_) {
    for (size_t i = 0; i < file_level.num_files; i++) {
      total_usage += cfd_->table_cache()->GetMemoryUsageByTableReader(
588
          vset_->env_options_, cfd_->internal_comparator(),
589 590 591 592 593 594
          file_level.files[i].fd);
    }
  }
  return total_usage;
}

S
sdong 已提交
595 596 597 598 599
uint64_t Version::GetEstimatedActiveKeys() {
  // Estimation will be not accurate when:
  // (1) there is merge keys
  // (2) keys are directly overwritten
  // (3) deletion on non-existing keys
600 601 602 603 604 605 606 607 608 609 610 611 612
  // (4) low number of samples
  if (num_samples_ == 0) {
    return 0;
  }

  if (num_samples_ < files_->size()) {
    // casting to avoid overflowing
    return static_cast<uint64_t>(static_cast<double>(
        accumulated_num_non_deletions_ - accumulated_num_deletions_) *
        files_->size() / num_samples_);
  } else {
    return accumulated_num_non_deletions_ - accumulated_num_deletions_;
  }
S
sdong 已提交
613 614
}

615 616 617 618
void Version::AddIterators(const ReadOptions& read_options,
                           const EnvOptions& soptions,
                           MergeIteratorBuilder* merge_iter_builder) {
  // Merge all level zero files together since they may overlap
F
Feng Zhu 已提交
619 620
  for (size_t i = 0; i < file_levels_[0].num_files; i++) {
    const auto& file = file_levels_[0].files[i];
621
    merge_iter_builder->AddIterator(cfd_->table_cache()->NewIterator(
F
Feng Zhu 已提交
622
        read_options, soptions, cfd_->internal_comparator(), file.fd, nullptr,
623 624 625 626 627 628 629
        false, merge_iter_builder->GetArena()));
  }

  // For levels > 0, we can use a concatenating iterator that sequentially
  // walks through the non-overlapping files in the level, opening them
  // lazily.
  for (int level = 1; level < num_levels_; level++) {
F
Feng Zhu 已提交
630
    if (file_levels_[level].num_files != 0) {
631 632 633 634
      merge_iter_builder->AddIterator(NewTwoLevelIterator(
          new LevelFileIteratorState(
              cfd_->table_cache(), read_options, soptions,
              cfd_->internal_comparator(), false /* for_compaction */,
635
              cfd_->ioptions()->prefix_extractor != nullptr),
F
Feng Zhu 已提交
636 637
          new LevelFileNumIterator(cfd_->internal_comparator(),
              &file_levels_[level]), merge_iter_builder->GetArena()));
638 639 640 641
    }
  }
}

642

643

644 645 646
Version::Version(ColumnFamilyData* cfd, VersionSet* vset,
                 uint64_t version_number)
    : cfd_(cfd),
647 648
      internal_comparator_((cfd == nullptr) ? nullptr
                                            : &cfd->internal_comparator()),
S
sdong 已提交
649 650
      user_comparator_(
          (cfd == nullptr) ? nullptr : internal_comparator_->user_comparator()),
651 652
      table_cache_((cfd == nullptr) ? nullptr : cfd->table_cache()),
      merge_operator_((cfd == nullptr) ? nullptr
653 654
                                       : cfd->ioptions()->merge_operator),
      info_log_((cfd == nullptr) ? nullptr : cfd->ioptions()->info_log),
655
      db_statistics_((cfd == nullptr) ? nullptr
656
                                      : cfd->ioptions()->statistics),
657 658 659
      // cfd is nullptr if Version is dummy
      num_levels_(cfd == nullptr ? 0 : cfd->NumberLevels()),
      num_non_empty_levels_(num_levels_),
660 661 662
      file_indexer_(cfd == nullptr
                        ? nullptr
                        : cfd->internal_comparator().user_comparator()),
663
      vset_(vset),
664 665 666 667 668 669 670 671
      next_(this),
      prev_(this),
      refs_(0),
      files_(new std::vector<FileMetaData*>[num_levels_]),
      files_by_size_(num_levels_),
      next_file_to_compact_by_size_(num_levels_),
      compaction_score_(num_levels_),
      compaction_level_(num_levels_),
672
      version_number_(version_number),
673 674 675 676 677 678
      accumulated_file_size_(0),
      accumulated_raw_key_size_(0),
      accumulated_raw_value_size_(0),
      accumulated_num_non_deletions_(0),
      accumulated_num_deletions_(0),
      num_samples_(0) {
679
  if (cfd != nullptr && cfd->current() != nullptr) {
680 681 682 683 684 685 686 687
      accumulated_file_size_ = cfd->current()->accumulated_file_size_;
      accumulated_raw_key_size_ = cfd->current()->accumulated_raw_key_size_;
      accumulated_raw_value_size_ =
          cfd->current()->accumulated_raw_value_size_;
      accumulated_num_non_deletions_ =
          cfd->current()->accumulated_num_non_deletions_;
      accumulated_num_deletions_ = cfd->current()->accumulated_num_deletions_;
      num_samples_ = cfd->current()->num_samples_;
688
  }
689
}
690

691
void Version::Get(const ReadOptions& read_options,
692 693
                  const LookupKey& k,
                  std::string* value,
694
                  Status* status,
695
                  MergeContext* merge_context,
696
                  bool* value_found) {
697 698
  Slice ikey = k.internal_key();
  Slice user_key = k.user_key();
699 700

  assert(status->ok() || status->IsMergeInProgress());
701 702 703 704

  GetContext get_context(user_comparator_, merge_operator_, info_log_,
      db_statistics_, status->ok() ? GetContext::kNotFound : GetContext::kMerge,
      user_key, value, value_found, merge_context);
705

706 707 708 709
  FilePicker fp(files_, user_key, ikey, &file_levels_, num_non_empty_levels_,
      &file_indexer_, user_comparator_, internal_comparator_);
  FdWithKeyRange* f = fp.GetNextFile();
  while (f != nullptr) {
710 711
    *status = table_cache_->Get(read_options, *internal_comparator_, f->fd,
                                ikey, &get_context);
712 713 714
    // TODO: examine the behavior for corrupted key
    if (!status->ok()) {
      return;
715
    }
716

717 718 719 720 721
    switch (get_context.State()) {
      case GetContext::kNotFound:
        // Keep searching in other files
        break;
      case GetContext::kFound:
722
        return;
723 724 725
      case GetContext::kDeleted:
        // Use empty error message for speed
        *status = Status::NotFound();
726
        return;
727
      case GetContext::kCorrupt:
728 729
        *status = Status::Corruption("corrupted key for ", user_key);
        return;
730
      case GetContext::kMerge:
731
        break;
732
    }
733
    f = fp.GetNextFile();
734 735
  }

736
  if (GetContext::kMerge == get_context.State()) {
737 738 739 740 741
    if (!merge_operator_) {
      *status =  Status::InvalidArgument(
          "merge_operator is not properly initialized.");
      return;
    }
742 743
    // merge_operands are in saver and we hit the beginning of the key history
    // do a final merge of nullptr and operands;
744
    if (merge_operator_->FullMerge(user_key, nullptr,
745
                                   merge_context->GetOperands(), value,
746
                                   info_log_)) {
747 748
      *status = Status::OK();
    } else {
749
      RecordTick(db_statistics_, NUMBER_MERGE_FAILURES);
750 751 752
      *status = Status::Corruption("could not perform end-of-key merge for ",
                                   user_key);
    }
753
  } else {
754
    *status = Status::NotFound(); // Use an empty error message for speed
755
  }
756 757
}

758 759 760
void Version::GenerateFileLevels() {
  file_levels_.resize(num_non_empty_levels_);
  for (int level = 0; level < num_non_empty_levels_; level++) {
F
Feng Zhu 已提交
761
    DoGenerateFileLevel(&file_levels_[level], files_[level], &arena_);
762 763 764
  }
}

765 766
void Version::PrepareApply(const MutableCFOptions& mutable_cf_options,
                           std::vector<uint64_t>& size_being_compacted) {
767
  UpdateAccumulatedStats();
768
  ComputeCompactionScore(mutable_cf_options, size_being_compacted);
769 770
  UpdateFilesBySize();
  UpdateNumNonEmptyLevels();
771
  file_indexer_.UpdateIndex(&arena_, num_non_empty_levels_, files_);
772
  GenerateFileLevels();
773 774
}

775
bool Version::MaybeInitializeFileMetaData(FileMetaData* file_meta) {
776 777
  if (file_meta->init_stats_from_file ||
      file_meta->compensated_file_size > 0) {
778 779 780 781
    return false;
  }
  std::shared_ptr<const TableProperties> tp;
  Status s = GetTableProperties(&tp, file_meta);
782
  file_meta->init_stats_from_file = true;
783
  if (!s.ok()) {
784
    Log(vset_->db_options_->info_log,
785 786
        "Unable to load table properties for file %" PRIu64 " --- %s\n",
        file_meta->fd.GetNumber(), s.ToString().c_str());
787 788 789 790 791 792 793 794 795 796 797
    return false;
  }
  if (tp.get() == nullptr) return false;
  file_meta->num_entries = tp->num_entries;
  file_meta->num_deletions = GetDeletedKeys(tp->user_collected_properties);
  file_meta->raw_value_size = tp->raw_value_size;
  file_meta->raw_key_size = tp->raw_key_size;

  return true;
}

798 799 800 801 802 803 804 805 806 807 808 809
void Version::UpdateAccumulatedStats(FileMetaData* file_meta) {
  assert(file_meta->init_stats_from_file);
  accumulated_file_size_ += file_meta->fd.GetFileSize();
  accumulated_raw_key_size_ += file_meta->raw_key_size;
  accumulated_raw_value_size_ += file_meta->raw_value_size;
  accumulated_num_non_deletions_ +=
      file_meta->num_entries - file_meta->num_deletions;
  accumulated_num_deletions_ += file_meta->num_deletions;
  num_samples_++;
}

void Version::UpdateAccumulatedStats() {
810 811
  static const int kDeletionWeightOnCompaction = 2;

812 813
  // maximum number of table properties loaded from files.
  const int kMaxInitCount = 20;
814
  int init_count = 0;
815 816 817 818 819 820 821 822 823 824 825 826
  // here only the first kMaxInitCount files which haven't been
  // initialized from file will be updated with num_deletions.
  // The motivation here is to cap the maximum I/O per Version creation.
  // The reason for choosing files from lower-level instead of higher-level
  // is that such design is able to propagate the initialization from
  // lower-level to higher-level:  When the num_deletions of lower-level
  // files are updated, it will make the lower-level files have accurate
  // compensated_file_size, making lower-level to higher-level compaction
  // will be triggered, which creates higher-level files whose num_deletions
  // will be updated here.
  for (int level = 0;
       level < num_levels_ && init_count < kMaxInitCount; ++level) {
827 828 829
    for (auto* file_meta : files_[level]) {
      if (MaybeInitializeFileMetaData(file_meta)) {
        // each FileMeta will be initialized only once.
830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
        UpdateAccumulatedStats(file_meta);
        if (++init_count >= kMaxInitCount) {
          break;
        }
      }
    }
  }
  // In case all sampled-files contain only deletion entries, then we
  // load the table-property of a file in higher-level to initialize
  // that value.
  for (int level = num_levels_ - 1;
       accumulated_raw_value_size_ == 0 && level >= 0; --level) {
    for (int i = static_cast<int>(files_[level].size()) - 1;
        accumulated_raw_value_size_ == 0 && i >= 0; --i) {
      if (MaybeInitializeFileMetaData(files_[level][i])) {
        UpdateAccumulatedStats(files_[level][i]);
846 847 848 849 850 851 852 853 854
      }
    }
  }

  uint64_t average_value_size = GetAverageValueSize();

  // compute the compensated size
  for (int level = 0; level < num_levels_; level++) {
    for (auto* file_meta : files_[level]) {
855 856 857 858 859 860 861
      // Here we only compute compensated_file_size for those file_meta
      // which compensated_file_size is uninitialized (== 0).
      if (file_meta->compensated_file_size == 0) {
        file_meta->compensated_file_size = file_meta->fd.GetFileSize() +
            file_meta->num_deletions * average_value_size *
            kDeletionWeightOnCompaction;
      }
862 863 864 865
    }
  }
}

866
void Version::ComputeCompactionScore(
867
    const MutableCFOptions& mutable_cf_options,
868
    std::vector<uint64_t>& size_being_compacted) {
869 870 871
  double max_score = 0;
  int max_score_level = 0;

872 873 874
  int max_input_level =
      cfd_->compaction_picker()->MaxInputLevel(NumberLevels());
  for (int level = 0; level <= max_input_level; level++) {
875 876 877 878 879 880 881 882 883 884 885 886 887 888
    double score;
    if (level == 0) {
      // 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).
      int numfiles = 0;
I
Igor Canadi 已提交
889
      uint64_t total_size = 0;
890 891
      for (unsigned int i = 0; i < files_[level].size(); i++) {
        if (!files_[level][i]->being_compacted) {
892
          total_size += files_[level][i]->compensated_file_size;
893 894 895
          numfiles++;
        }
      }
896
      if (cfd_->ioptions()->compaction_style == kCompactionStyleFIFO) {
I
Igor Canadi 已提交
897 898
        score = static_cast<double>(total_size) /
                cfd_->options()->compaction_options_fifo.max_table_files_size;
899
      } else if (numfiles >= mutable_cf_options.level0_stop_writes_trigger) {
I
Igor Canadi 已提交
900
        // If we are slowing down writes, then we better compact that first
901
        score = 1000000;
902 903
      } else if (numfiles >=
          mutable_cf_options.level0_slowdown_writes_trigger) {
904 905 906
        score = 10000;
      } else {
        score = static_cast<double>(numfiles) /
907
                mutable_cf_options.level0_file_num_compaction_trigger;
908 909 910 911
      }
    } else {
      // Compute the ratio of current size to size limit.
      const uint64_t level_bytes =
912
          TotalCompensatedFileSize(files_[level]) - size_being_compacted[level];
913
      score = static_cast<double>(level_bytes) /
914
              mutable_cf_options.MaxBytesForLevel(level);
915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946
      if (max_score < score) {
        max_score = score;
        max_score_level = level;
      }
    }
    compaction_level_[level] = level;
    compaction_score_[level] = score;
  }

  // update the max compaction score in levels 1 to n-1
  max_compaction_score_ = max_score;
  max_compaction_score_level_ = max_score_level;

  // 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++) {
    for (int j = i + 1; j < NumberLevels() - 1; j++) {
      if (compaction_score_[i] < compaction_score_[j]) {
        double score = compaction_score_[i];
        int level = compaction_level_[i];
        compaction_score_[i] = compaction_score_[j];
        compaction_level_[i] = compaction_level_[j];
        compaction_score_[j] = score;
        compaction_level_[j] = level;
      }
    }
  }
}

namespace {
// Compator that is used to sort files based on their size
// In normal mode: descending size
947 948 949 950
bool CompareCompensatedSizeDescending(const Version::Fsize& first,
                                      const Version::Fsize& second) {
  return (first.file->compensated_file_size >
      second.file->compensated_file_size);
951
}
K
kailiu 已提交
952
} // anonymous namespace
953

954 955 956 957 958 959 960 961 962 963 964
void Version::UpdateNumNonEmptyLevels() {
  num_non_empty_levels_ = num_levels_;
  for (int i = num_levels_ - 1; i >= 0; i--) {
    if (files_[i].size() != 0) {
      return;
    } else {
      num_non_empty_levels_ = i;
    }
  }
}

965
void Version::UpdateFilesBySize() {
966 967
  if (cfd_->ioptions()->compaction_style == kCompactionStyleFIFO ||
      cfd_->ioptions()->compaction_style == kCompactionStyleUniversal) {
I
Igor Canadi 已提交
968 969 970
    // don't need this
    return;
  }
971
  // No need to sort the highest level because it is never compacted.
972
  for (int level = 0; level < NumberLevels() - 1; level++) {
973
    const std::vector<FileMetaData*>& files = files_[level];
974
    auto& files_by_size = files_by_size_[level];
975 976 977 978 979 980 981 982 983 984
    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
985 986 987
    size_t num = Version::number_of_files_to_sort_;
    if (num > temp.size()) {
      num = temp.size();
988
    }
989 990
    std::partial_sort(temp.begin(), temp.begin() + num, temp.end(),
                      CompareCompensatedSizeDescending);
991 992 993 994 995 996 997 998 999 1000 1001
    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);
    }
    next_file_to_compact_by_size_[level] = 0;
    assert(files_[level].size() == files_by_size_[level].size());
  }
}

J
jorlow@chromium.org 已提交
1002 1003 1004 1005
void Version::Ref() {
  ++refs_;
}

1006
bool Version::Unref() {
J
jorlow@chromium.org 已提交
1007 1008 1009
  assert(refs_ >= 1);
  --refs_;
  if (refs_ == 0) {
1010
    delete this;
1011
    return true;
J
jorlow@chromium.org 已提交
1012
  }
1013
  return false;
J
jorlow@chromium.org 已提交
1014 1015
}

1016 1017 1018 1019 1020 1021 1022 1023
bool Version::NeedsCompaction() const {
  // In universal compaction case, this check doesn't really
  // check the compaction condition, but checks num of files threshold
  // only. We are not going to miss any compaction opportunity
  // but it's likely that more compactions are scheduled but
  // ending up with nothing to do. We can improve it later.
  // TODO(sdong): improve this function to be accurate for universal
  //              compactions.
1024 1025 1026 1027
  int max_input_level =
      cfd_->compaction_picker()->MaxInputLevel(NumberLevels());

  for (int i = 0; i <= max_input_level; i++) {
1028 1029 1030 1031 1032 1033 1034
    if (compaction_score_[i] >= 1) {
      return true;
    }
  }
  return false;
}

1035
bool Version::OverlapInLevel(int level,
G
Gabor Cselle 已提交
1036 1037
                             const Slice* smallest_user_key,
                             const Slice* largest_user_key) {
1038
  return SomeFileOverlapsRange(cfd_->internal_comparator(), (level > 0),
1039
                               file_levels_[level], smallest_user_key,
1040
                               largest_user_key);
G
Gabor Cselle 已提交
1041 1042 1043
}

int Version::PickLevelForMemTableOutput(
1044
    const MutableCFOptions& mutable_cf_options,
G
Gabor Cselle 已提交
1045 1046 1047 1048 1049 1050 1051 1052 1053
    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;
1054
    int max_mem_compact_level = cfd_->options()->max_mem_compaction_level;
1055
    while (max_mem_compact_level > 0 && level < max_mem_compact_level) {
G
Gabor Cselle 已提交
1056 1057 1058
      if (OverlapInLevel(level + 1, &smallest_user_key, &largest_user_key)) {
        break;
      }
1059
      if (level + 2 >= num_levels_) {
1060 1061
        level++;
        break;
1062
      }
G
Gabor Cselle 已提交
1063
      GetOverlappingInputs(level + 2, &start, &limit, &overlaps);
1064
      const uint64_t sum = TotalFileSize(overlaps);
1065
      if (sum > mutable_cf_options.MaxGrandParentOverlapBytes(level)) {
G
Gabor Cselle 已提交
1066 1067 1068 1069 1070
        break;
      }
      level++;
    }
  }
1071

G
Gabor Cselle 已提交
1072 1073 1074 1075
  return level;
}

// Store in "*inputs" all files in "level" that overlap [begin,end]
A
Abhishek Kona 已提交
1076
// If hint_index is specified, then it points to a file in the
1077 1078
// overlapping range.
// The file_index returns a pointer to any file in an overlapping range.
1079 1080 1081 1082 1083 1084
void Version::GetOverlappingInputs(int level,
                                   const InternalKey* begin,
                                   const InternalKey* end,
                                   std::vector<FileMetaData*>* inputs,
                                   int hint_index,
                                   int* file_index) {
G
Gabor Cselle 已提交
1085 1086
  inputs->clear();
  Slice user_begin, user_end;
A
Abhishek Kona 已提交
1087
  if (begin != nullptr) {
G
Gabor Cselle 已提交
1088 1089
    user_begin = begin->user_key();
  }
A
Abhishek Kona 已提交
1090
  if (end != nullptr) {
G
Gabor Cselle 已提交
1091 1092
    user_end = end->user_key();
  }
1093 1094 1095
  if (file_index) {
    *file_index = -1;
  }
1096
  const Comparator* user_cmp = cfd_->internal_comparator().user_comparator();
A
Abhishek Kona 已提交
1097
  if (begin != nullptr && end != nullptr && level > 0) {
1098 1099
    GetOverlappingInputsBinarySearch(level, user_begin, user_end, inputs,
      hint_index, file_index);
1100 1101
    return;
  }
1102 1103 1104 1105
  for (size_t i = 0; i < file_levels_[level].num_files; ) {
    FdWithKeyRange* f = &(file_levels_[level].files[i++]);
    const Slice file_start = ExtractUserKey(f->smallest_key);
    const Slice file_limit = ExtractUserKey(f->largest_key);
A
Abhishek Kona 已提交
1106
    if (begin != nullptr && user_cmp->Compare(file_limit, user_begin) < 0) {
G
Gabor Cselle 已提交
1107
      // "f" is completely before specified range; skip it
A
Abhishek Kona 已提交
1108
    } else if (end != nullptr && user_cmp->Compare(file_start, user_end) > 0) {
G
Gabor Cselle 已提交
1109 1110
      // "f" is completely after specified range; skip it
    } else {
1111
      inputs->push_back(files_[level][i-1]);
H
Hans Wennborg 已提交
1112 1113 1114
      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 已提交
1115
        if (begin != nullptr && user_cmp->Compare(file_start, user_begin) < 0) {
H
Hans Wennborg 已提交
1116 1117 1118
          user_begin = file_start;
          inputs->clear();
          i = 0;
A
Abhishek Kona 已提交
1119 1120
        } else if (end != nullptr
            && user_cmp->Compare(file_limit, user_end) > 0) {
H
Hans Wennborg 已提交
1121 1122 1123 1124
          user_end = file_limit;
          inputs->clear();
          i = 0;
        }
1125
      } else if (file_index) {
1126
        *file_index = i-1;
H
Hans Wennborg 已提交
1127
      }
G
Gabor Cselle 已提交
1128 1129
    }
  }
1130 1131
}

1132 1133 1134 1135 1136 1137 1138 1139
// 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,
1140 1141 1142
    std::vector<FileMetaData*>* inputs,
    int hint_index,
    int* file_index) {
1143 1144 1145 1146 1147
  assert(level > 0);
  int min = 0;
  int mid = 0;
  int max = files_[level].size() -1;
  bool foundOverlap = false;
1148
  const Comparator* user_cmp = cfd_->internal_comparator().user_comparator();
1149 1150 1151 1152 1153 1154 1155 1156 1157

  // 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) {
1158
    mid = (min + max)/2;
1159 1160 1161
    FdWithKeyRange* f = &(file_levels_[level].files[mid]);
    const Slice file_start = ExtractUserKey(f->smallest_key);
    const Slice file_limit = ExtractUserKey(f->largest_key);
1162 1163 1164 1165 1166 1167 1168 1169 1170
    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 已提交
1171

1172 1173 1174 1175
  // If there were no overlapping files, return immediately.
  if (!foundOverlap) {
    return;
  }
1176 1177 1178 1179
  // returns the index where an overlap is found
  if (file_index) {
    *file_index = mid;
  }
1180 1181
  ExtendOverlappingInputs(level, user_begin, user_end, inputs, mid);
}
A
Abhishek Kona 已提交
1182

1183 1184 1185 1186
// 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.
F
Feng Zhu 已提交
1187
// Use FileLevel in searching, make it faster
1188 1189 1190 1191 1192
void Version::ExtendOverlappingInputs(
    int level,
    const Slice& user_begin,
    const Slice& user_end,
    std::vector<FileMetaData*>* inputs,
1193
    unsigned int midIndex) {
1194

1195
  const Comparator* user_cmp = cfd_->internal_comparator().user_comparator();
1196
  const FdWithKeyRange* files = file_levels_[level].files;
1197 1198 1199
#ifndef NDEBUG
  {
    // assert that the file at midIndex overlaps with the range
1200 1201 1202 1203
    assert(midIndex < file_levels_[level].num_files);
    const FdWithKeyRange* f = &files[midIndex];
    const Slice fstart = ExtractUserKey(f->smallest_key);
    const Slice flimit = ExtractUserKey(f->largest_key);
1204 1205 1206 1207 1208 1209 1210
    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
1211 1212
  int startIndex = midIndex + 1;
  int endIndex = midIndex;
1213
  int count __attribute__((unused)) = 0;
1214 1215

  // check backwards from 'mid' to lower indices
1216
  for (int i = midIndex; i >= 0 ; i--) {
1217 1218
    const FdWithKeyRange* f = &files[i];
    const Slice file_limit = ExtractUserKey(f->largest_key);
1219
    if (user_cmp->Compare(file_limit, user_begin) >= 0) {
1220 1221
      startIndex = i;
      assert((count++, true));
1222 1223 1224 1225 1226
    } else {
      break;
    }
  }
  // check forward from 'mid+1' to higher indices
1227 1228 1229
  for (unsigned int i = midIndex+1; i < file_levels_[level].num_files; i++) {
    const FdWithKeyRange* f = &files[i];
    const Slice file_start = ExtractUserKey(f->smallest_key);
1230
    if (user_cmp->Compare(file_start, user_end) <= 0) {
1231 1232
      assert((count++, true));
      endIndex = i;
1233 1234 1235 1236
    } else {
      break;
    }
  }
1237 1238 1239 1240 1241
  assert(count == endIndex - startIndex + 1);

  // insert overlapping files into vector
  for (int i = startIndex; i <= endIndex; i++) {
    FileMetaData* f = files_[level][i];
1242
    inputs->push_back(f);
1243
  }
1244 1245
}

1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259
// Returns true iff the first or last file in inputs contains
// an overlapping user key to the file "just outside" of it (i.e.
// just after the last file, or just before the first file)
// REQUIRES: "*inputs" is a sorted list of non-overlapping files
bool Version::HasOverlappingUserKey(
    const std::vector<FileMetaData*>* inputs,
    int level) {

  // If inputs empty, there is no overlap.
  // If level == 0, it is assumed that all needed files were already included.
  if (inputs->empty() || level == 0){
    return false;
  }

1260
  const Comparator* user_cmp = cfd_->internal_comparator().user_comparator();
1261 1262 1263
  const FileLevel& file_level = file_levels_[level];
  const FdWithKeyRange* files = file_levels_[level].files;
  const size_t kNumFiles = file_level.num_files;
1264 1265

  // Check the last file in inputs against the file after it
1266
  size_t last_file = FindFile(cfd_->internal_comparator(), file_level,
1267
                              inputs->back()->largest.Encode());
1268
  assert(last_file < kNumFiles);  // File should exist!
1269
  if (last_file < kNumFiles-1) {                    // If not the last file
1270 1271 1272 1273
    const Slice last_key_in_input = ExtractUserKey(
        files[last_file].largest_key);
    const Slice first_key_after = ExtractUserKey(
        files[last_file+1].smallest_key);
1274 1275 1276 1277 1278 1279 1280
    if (user_cmp->Compare(last_key_in_input, first_key_after) == 0) {
      // The last user key in input overlaps with the next file's first key
      return true;
    }
  }

  // Check the first file in inputs against the file just before it
1281
  size_t first_file = FindFile(cfd_->internal_comparator(), file_level,
1282
                               inputs->front()->smallest.Encode());
1283
  assert(first_file <= last_file);   // File should exist!
1284
  if (first_file > 0) {                                 // If not first file
1285 1286 1287 1288
    const Slice& first_key_in_input = ExtractUserKey(
        files[first_file].smallest_key);
    const Slice& last_key_before = ExtractUserKey(
        files[first_file-1].largest_key);
1289 1290 1291 1292 1293 1294 1295 1296 1297
    if (user_cmp->Compare(first_key_in_input, last_key_before) == 0) {
      // The first user key in input overlaps with the previous file's last key
      return true;
    }
  }

  return false;
}

1298
uint64_t Version::NumLevelBytes(int level) const {
1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
  assert(level >= 0);
  assert(level < NumberLevels());
  return TotalFileSize(files_[level]);
}

const char* Version::LevelSummary(LevelSummaryStorage* scratch) const {
  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(files_[i].size()));
    if (ret < 0 || ret >= sz) break;
    len += ret;
  }
I
Igor Canadi 已提交
1312 1313 1314 1315
  if (len > 0) {
    // overwrite the last space
    --len;
  }
1316 1317 1318 1319 1320 1321 1322 1323 1324
  snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "]");
  return scratch->buffer;
}

const char* Version::LevelFileSummary(FileSummaryStorage* scratch,
                                      int level) const {
  int len = snprintf(scratch->buffer, sizeof(scratch->buffer), "files_size[");
  for (const auto& f : files_[level]) {
    int sz = sizeof(scratch->buffer) - len;
I
Igor Canadi 已提交
1325
    char sztxt[16];
1326
    AppendHumanBytes(f->fd.GetFileSize(), sztxt, sizeof(sztxt));
1327
    int ret = snprintf(scratch->buffer + len, sz,
1328 1329
                       "#%" PRIu64 "(seq=%" PRIu64 ",sz=%s,%d) ",
                       f->fd.GetNumber(), f->smallest_seqno, sztxt,
I
Igor Canadi 已提交
1330
                       static_cast<int>(f->being_compacted));
1331 1332 1333 1334
    if (ret < 0 || ret >= sz)
      break;
    len += ret;
  }
I
Igor Canadi 已提交
1335 1336 1337 1338
  // overwrite the last space (only if files_[level].size() is non-zero)
  if (files_[level].size() && len > 0) {
    --len;
  }
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
  snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "]");
  return scratch->buffer;
}

int64_t Version::MaxNextLevelOverlappingBytes() {
  uint64_t result = 0;
  std::vector<FileMetaData*> overlaps;
  for (int level = 1; level < NumberLevels() - 1; level++) {
    for (const auto& f : files_[level]) {
      GetOverlappingInputs(level + 1, &f->smallest, &f->largest, &overlaps);
      const uint64_t sum = TotalFileSize(overlaps);
      if (sum > result) {
        result = sum;
      }
    }
  }
  return result;
}

1358
void Version::AddLiveFiles(std::vector<FileDescriptor>* live) {
1359 1360 1361
  for (int level = 0; level < NumberLevels(); level++) {
    const std::vector<FileMetaData*>& files = files_[level];
    for (const auto& file : files) {
1362
      live->push_back(file->fd);
1363 1364 1365 1366
    }
  }
}

Z
Zheng Shao 已提交
1367
std::string Version::DebugString(bool hex) const {
J
jorlow@chromium.org 已提交
1368
  std::string r;
1369
  for (int level = 0; level < num_levels_; level++) {
1370 1371 1372 1373 1374
    // E.g.,
    //   --- level 1 ---
    //   17:123['a' .. 'd']
    //   20:43['e' .. 'g']
    r.append("--- level ");
J
jorlow@chromium.org 已提交
1375
    AppendNumberTo(&r, level);
1376 1377
    r.append(" --- version# ");
    AppendNumberTo(&r, version_number_);
1378
    r.append(" ---\n");
J
jorlow@chromium.org 已提交
1379
    const std::vector<FileMetaData*>& files = files_[level];
D
dgrogan@chromium.org 已提交
1380
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
1381
      r.push_back(' ');
1382
      AppendNumberTo(&r, files[i]->fd.GetNumber());
J
jorlow@chromium.org 已提交
1383
      r.push_back(':');
1384
      AppendNumberTo(&r, files[i]->fd.GetFileSize());
G
Gabor Cselle 已提交
1385
      r.append("[");
Z
Zheng Shao 已提交
1386
      r.append(files[i]->smallest.DebugString(hex));
G
Gabor Cselle 已提交
1387
      r.append(" .. ");
Z
Zheng Shao 已提交
1388
      r.append(files[i]->largest.DebugString(hex));
G
Gabor Cselle 已提交
1389
      r.append("]\n");
J
jorlow@chromium.org 已提交
1390 1391 1392 1393 1394
    }
  }
  return r;
}

1395 1396 1397 1398 1399
// this is used to batch writes to the manifest file
struct VersionSet::ManifestWriter {
  Status status;
  bool done;
  port::CondVar cv;
1400
  ColumnFamilyData* cfd;
1401
  VersionEdit* edit;
A
Abhishek Kona 已提交
1402

1403 1404 1405
  explicit ManifestWriter(port::Mutex* mu, ColumnFamilyData* cfd,
                          VersionEdit* e)
      : done(false), cv(mu), cfd(cfd), edit(e) {}
1406 1407
};

J
jorlow@chromium.org 已提交
1408 1409 1410 1411 1412
// 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:
1413
  // Helper to sort v->files_
1414
  // kLevel0 -- NewestFirstBySeqNo
1415 1416 1417
  // kLevelNon0 -- BySmallestKey
  struct FileComparator {
    enum SortMethod {
1418 1419
      kLevel0 = 0,
      kLevelNon0 = 1,
1420
    } sort_method;
1421 1422 1423
    const InternalKeyComparator* internal_comparator;

    bool operator()(FileMetaData* f1, FileMetaData* f2) const {
1424
      switch (sort_method) {
1425
        case kLevel0:
1426 1427 1428
          return NewestFirstBySeqNo(f1, f2);
        case kLevelNon0:
          return BySmallestKey(f1, f2, internal_comparator);
1429
      }
1430
      assert(false);
1431
      return false;
1432 1433 1434
    }
  };

1435
  typedef std::set<FileMetaData*, FileComparator> FileSet;
1436 1437 1438 1439 1440
  struct LevelState {
    std::set<uint64_t> deleted_files;
    FileSet* added_files;
  };

1441
  ColumnFamilyData* cfd_;
1442
  Version* base_;
1443
  LevelState* levels_;
1444 1445
  FileComparator level_zero_cmp_;
  FileComparator level_nonzero_cmp_;
J
jorlow@chromium.org 已提交
1446 1447

 public:
1448
  Builder(ColumnFamilyData* cfd) : cfd_(cfd), base_(cfd->current()) {
1449
    base_->Ref();
1450
    levels_ = new LevelState[base_->NumberLevels()];
1451
    level_zero_cmp_.sort_method = FileComparator::kLevel0;
1452
    level_nonzero_cmp_.sort_method = FileComparator::kLevelNon0;
1453
    level_nonzero_cmp_.internal_comparator = &cfd->internal_comparator();
1454 1455

    levels_[0].added_files = new FileSet(level_zero_cmp_);
1456
    for (int level = 1; level < base_->NumberLevels(); level++) {
1457
        levels_[level].added_files = new FileSet(level_nonzero_cmp_);
J
jorlow@chromium.org 已提交
1458 1459 1460 1461
    }
  }

  ~Builder() {
1462
    for (int level = 0; level < base_->NumberLevels(); level++) {
1463 1464 1465 1466 1467 1468 1469 1470
      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 已提交
1471
      for (uint32_t i = 0; i < to_unref.size(); i++) {
1472
        FileMetaData* f = to_unref[i];
J
jorlow@chromium.org 已提交
1473 1474
        f->refs--;
        if (f->refs <= 0) {
1475
          if (f->table_reader_handle) {
1476
            cfd_->table_cache()->ReleaseHandle(f->table_reader_handle);
1477 1478
            f->table_reader_handle = nullptr;
          }
J
jorlow@chromium.org 已提交
1479 1480 1481 1482
          delete f;
        }
      }
    }
K
kailiu 已提交
1483

1484
    delete[] levels_;
1485
    base_->Unref();
J
jorlow@chromium.org 已提交
1486 1487
  }

1488 1489
  void CheckConsistency(Version* v) {
#ifndef NDEBUG
1490
    // make sure the files are sorted correctly
1491
    for (int level = 0; level < v->NumberLevels(); level++) {
1492 1493 1494 1495 1496
      for (size_t i = 1; i < v->files_[level].size(); i++) {
        auto f1 = v->files_[level][i - 1];
        auto f2 = v->files_[level][i];
        if (level == 0) {
          assert(level_zero_cmp_(f1, f2));
1497
          assert(f1->largest_seqno > f2->largest_seqno);
1498 1499 1500 1501
        } else {
          assert(level_nonzero_cmp_(f1, f2));

          // Make sure there is no overlap in levels > 0
1502 1503
          if (cfd_->internal_comparator().Compare(f1->largest, f2->smallest) >=
              0) {
1504
            fprintf(stderr, "overlapping ranges in same level %s vs. %s\n",
1505 1506
                    (f1->largest).DebugString().c_str(),
                    (f2->smallest).DebugString().c_str());
1507 1508 1509 1510 1511 1512 1513 1514
            abort();
          }
        }
      }
    }
#endif
  }

1515
  void CheckConsistencyForDeletes(VersionEdit* edit, uint64_t number,
1516
                                  int level) {
1517 1518 1519
#ifndef NDEBUG
      // a file to be deleted better exist in the previous version
      bool found = false;
1520
      for (int l = 0; !found && l < base_->NumberLevels(); l++) {
1521
        const std::vector<FileMetaData*>& base_files = base_->files_[l];
1522
        for (unsigned int i = 0; i < base_files.size(); i++) {
1523
          FileMetaData* f = base_files[i];
1524
          if (f->fd.GetNumber() == number) {
1525 1526 1527 1528 1529 1530 1531 1532
            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
1533
      for (int l = level+1; !found && l < base_->NumberLevels(); l++) {
1534
        const FileSet* added = levels_[l].added_files;
D
Dhruba Borthakur 已提交
1535 1536 1537
        for (FileSet::const_iterator added_iter = added->begin();
             added_iter != added->end(); ++added_iter) {
          FileMetaData* f = *added_iter;
1538
          if (f->fd.GetNumber() == number) {
D
Dhruba Borthakur 已提交
1539 1540 1541 1542 1543 1544 1545 1546 1547
            found = true;
            break;
          }
        }
      }

      // maybe this file was added in a previous edit that was Applied
      if (!found) {
        const FileSet* added = levels_[level].added_files;
1548 1549 1550
        for (FileSet::const_iterator added_iter = added->begin();
             added_iter != added->end(); ++added_iter) {
          FileMetaData* f = *added_iter;
1551
          if (f->fd.GetNumber() == number) {
1552 1553 1554 1555 1556
            found = true;
            break;
          }
        }
      }
1557
      if (!found) {
I
Igor Canadi 已提交
1558
        fprintf(stderr, "not found %" PRIu64 "\n", number);
1559
      }
1560 1561 1562 1563
      assert(found);
#endif
  }

J
jorlow@chromium.org 已提交
1564 1565
  // Apply all of the edits in *edit to the current state.
  void Apply(VersionEdit* edit) {
1566 1567
    CheckConsistency(base_);

J
jorlow@chromium.org 已提交
1568 1569
    // Delete files
    const VersionEdit::DeletedFileSet& del = edit->deleted_files_;
K
kailiu 已提交
1570 1571 1572
    for (const auto& del_file : del) {
      const auto level = del_file.first;
      const auto number = del_file.second;
1573
      levels_[level].deleted_files.insert(number);
1574
      CheckConsistencyForDeletes(edit, number, level);
J
jorlow@chromium.org 已提交
1575 1576 1577
    }

    // Add new files
K
kailiu 已提交
1578 1579 1580
    for (const auto& new_file : edit->new_files_) {
      const int level = new_file.first;
      FileMetaData* f = new FileMetaData(new_file.second);
J
jorlow@chromium.org 已提交
1581
      f->refs = 1;
1582

1583
      levels_[level].deleted_files.erase(f->fd.GetNumber());
1584
      levels_[level].added_files->insert(f);
J
jorlow@chromium.org 已提交
1585 1586 1587 1588 1589
    }
  }

  // Save the current state in *v.
  void SaveTo(Version* v) {
1590 1591
    CheckConsistency(base_);
    CheckConsistency(v);
1592

1593
    for (int level = 0; level < base_->NumberLevels(); level++) {
1594
      const auto& cmp = (level == 0) ? level_zero_cmp_ : level_nonzero_cmp_;
1595 1596
      // Merge the set of added files with the set of pre-existing files.
      // Drop any deleted files.  Store the result in *v.
K
kailiu 已提交
1597 1598 1599 1600 1601 1602 1603
      const auto& base_files = base_->files_[level];
      auto base_iter = base_files.begin();
      auto base_end = base_files.end();
      const auto& added_files = *levels_[level].added_files;
      v->files_[level].reserve(base_files.size() + added_files.size());

      for (const auto& added : added_files) {
1604
        // Add all smaller files listed in base_
K
kailiu 已提交
1605
        for (auto bpos = std::upper_bound(base_iter, base_end, added, cmp);
1606 1607 1608 1609 1610
             base_iter != bpos;
             ++base_iter) {
          MaybeAddFile(v, level, *base_iter);
        }

K
kailiu 已提交
1611
        MaybeAddFile(v, level, added);
1612 1613 1614 1615 1616
      }

      // Add remaining base files
      for (; base_iter != base_end; ++base_iter) {
        MaybeAddFile(v, level, *base_iter);
J
jorlow@chromium.org 已提交
1617
      }
1618
    }
1619

1620
    CheckConsistency(v);
1621 1622
  }

1623
  void LoadTableHandlers() {
1624
    for (int level = 0; level < cfd_->NumberLevels(); level++) {
1625 1626
      for (auto& file_meta : *(levels_[level].added_files)) {
        assert (!file_meta->table_reader_handle);
1627
        cfd_->table_cache()->FindTable(
1628
            base_->vset_->env_options_, cfd_->internal_comparator(),
I
Igor Canadi 已提交
1629
            file_meta->fd, &file_meta->table_reader_handle, false);
1630 1631
        if (file_meta->table_reader_handle != nullptr) {
          // Load table_reader
1632
          file_meta->fd.table_reader =
1633 1634 1635
              cfd_->table_cache()->GetTableReaderFromHandle(
                  file_meta->table_reader_handle);
        }
1636 1637 1638 1639
      }
    }
  }

1640
  void MaybeAddFile(Version* v, int level, FileMetaData* f) {
1641
    if (levels_[level].deleted_files.count(f->fd.GetNumber()) > 0) {
1642 1643
      // File is deleted: do nothing
    } else {
K
kailiu 已提交
1644
      auto* files = &v->files_[level];
1645 1646
      if (level > 0 && !files->empty()) {
        // Must not overlap
1647 1648
        assert(cfd_->internal_comparator().Compare(
                   (*files)[files->size() - 1]->largest, f->smallest) < 0);
1649
      }
1650
      f->refs++;
1651
      files->push_back(f);
J
jorlow@chromium.org 已提交
1652 1653 1654 1655
    }
  }
};

1656 1657
VersionSet::VersionSet(const std::string& dbname, const DBOptions* db_options,
                       const EnvOptions& env_options, Cache* table_cache,
1658
                       WriteController* write_controller)
1659
    : column_family_set_(new ColumnFamilySet(dbname, db_options, env_options,
1660
                                             table_cache, write_controller)),
1661
      env_(db_options->env),
J
jorlow@chromium.org 已提交
1662
      dbname_(dbname),
1663
      db_options_(db_options),
J
jorlow@chromium.org 已提交
1664 1665
      next_file_number_(2),
      manifest_file_number_(0),  // Filled by Recover()
1666
      pending_manifest_file_number_(0),
1667 1668
      last_sequence_(0),
      prev_log_number_(0),
A
Abhishek Kona 已提交
1669
      current_version_number_(0),
1670
      manifest_file_size_(0),
1671 1672
      env_options_(env_options),
      env_options_compactions_(env_options_) {}
J
jorlow@chromium.org 已提交
1673 1674

VersionSet::~VersionSet() {
1675 1676 1677
  // we need to delete column_family_set_ because its destructor depends on
  // VersionSet
  column_family_set_.reset();
1678 1679 1680 1681
  for (auto file : obsolete_files_) {
    delete file;
  }
  obsolete_files_.clear();
1682 1683
}

1684 1685
void VersionSet::AppendVersion(ColumnFamilyData* column_family_data,
                               Version* v) {
1686 1687
  // Make "v" current
  assert(v->refs_ == 0);
1688 1689 1690 1691 1692
  Version* current = column_family_data->current();
  assert(v != current);
  if (current != nullptr) {
    assert(current->refs_ > 0);
    current->Unref();
1693
  }
1694
  column_family_data->SetCurrent(v);
1695 1696 1697
  v->Ref();

  // Append to linked list
1698 1699
  v->prev_ = column_family_data->dummy_versions()->prev_;
  v->next_ = column_family_data->dummy_versions();
1700 1701 1702 1703
  v->prev_->next_ = v;
  v->next_->prev_ = v;
}

1704
Status VersionSet::LogAndApply(ColumnFamilyData* column_family_data,
1705
                               const MutableCFOptions& mutable_cf_options,
1706
                               VersionEdit* edit, port::Mutex* mu,
1707
                               Directory* db_directory, bool new_descriptor_log,
1708
                               const ColumnFamilyOptions* new_cf_options) {
1709
  mu->AssertHeld();
1710

I
Igor Canadi 已提交
1711 1712 1713 1714
  // column_family_data can be nullptr only if this is column_family_add.
  // in that case, we also need to specify ColumnFamilyOptions
  if (column_family_data == nullptr) {
    assert(edit->is_column_family_add_);
1715
    assert(new_cf_options != nullptr);
1716
  }
1717

1718
  // queue our request
1719
  ManifestWriter w(mu, column_family_data, edit);
1720 1721 1722
  manifest_writers_.push_back(&w);
  while (!w.done && &w != manifest_writers_.front()) {
    w.cv.Wait();
1723
  }
1724 1725 1726
  if (w.done) {
    return w.status;
  }
I
Igor Canadi 已提交
1727 1728 1729 1730 1731 1732 1733 1734 1735 1736
  if (column_family_data != nullptr && column_family_data->IsDropped()) {
    // if column family is dropped by the time we get here, no need to write
    // anything to the manifest
    manifest_writers_.pop_front();
    // Notify new head of write queue
    if (!manifest_writers_.empty()) {
      manifest_writers_.front()->cv.Signal();
    }
    return Status::OK();
  }
A
Abhishek Kona 已提交
1737

1738
  std::vector<VersionEdit*> batch_edits;
1739 1740
  Version* v = nullptr;
  std::unique_ptr<Builder> builder(nullptr);
1741

1742 1743 1744
  // process all requests in the queue
  ManifestWriter* last_writer = &w;
  assert(!manifest_writers_.empty());
1745
  assert(manifest_writers_.front() == &w);
1746 1747
  if (edit->IsColumnFamilyManipulation()) {
    // no group commits for column family add or drop
I
Igor Canadi 已提交
1748
    LogAndApplyCFHelper(edit);
1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763
    batch_edits.push_back(edit);
  } else {
    v = new Version(column_family_data, this, current_version_number_++);
    builder.reset(new Builder(column_family_data));
    for (const auto& writer : manifest_writers_) {
      if (writer->edit->IsColumnFamilyManipulation() ||
          writer->cfd->GetID() != column_family_data->GetID()) {
        // no group commits for column family add or drop
        // also, group commits across column families are not supported
        break;
      }
      last_writer = writer;
      LogAndApplyHelper(column_family_data, builder.get(), v, last_writer->edit,
                        mu);
      batch_edits.push_back(last_writer->edit);
1764
    }
1765
    builder->SaveTo(v);
J
jorlow@chromium.org 已提交
1766 1767 1768 1769
  }

  // Initialize new descriptor log file if necessary by creating
  // a temporary file that contains a snapshot of the current version.
1770
  uint64_t new_manifest_file_size = 0;
1771
  Status s;
A
Abhishek Kona 已提交
1772

1773
  assert(pending_manifest_file_number_ == 0);
1774
  if (!descriptor_log_ ||
1775
      manifest_file_size_ > db_options_->max_manifest_file_size) {
1776 1777
    pending_manifest_file_number_ = NewFileNumber();
    batch_edits.back()->SetNextFile(next_file_number_);
A
Abhishek Kona 已提交
1778
    new_descriptor_log = true;
1779 1780
  } else {
    pending_manifest_file_number_ = manifest_file_number_;
A
Abhishek Kona 已提交
1781 1782
  }

I
Igor Canadi 已提交
1783
  if (new_descriptor_log) {
1784
    // if we're writing out new snapshot make sure to persist max column family
I
Igor Canadi 已提交
1785 1786 1787
    if (column_family_set_->GetMaxColumnFamily() > 0) {
      edit->SetMaxColumnFamily(column_family_set_->GetMaxColumnFamily());
    }
J
jorlow@chromium.org 已提交
1788 1789
  }

1790
  // Unlock during expensive operations. New writes cannot get here
1791
  // because &w is ensuring that all new writes get queued.
1792
  {
1793 1794 1795 1796 1797 1798 1799
    std::vector<uint64_t> size_being_compacted;
    if (!edit->IsColumnFamilyManipulation()) {
      size_being_compacted.resize(v->NumberLevels() - 1);
      // calculate the amount of data being compacted at every level
      column_family_data->compaction_picker()->SizeBeingCompacted(
          size_being_compacted);
    }
1800

1801
    mu->Unlock();
1802

1803 1804
    if (!edit->IsColumnFamilyManipulation() &&
        db_options_->max_open_files == -1) {
1805 1806
      // unlimited table cache. Pre-load table handle now.
      // Need to do it out of the mutex.
1807
      builder->LoadTableHandlers();
1808 1809
    }

I
Igor Canadi 已提交
1810 1811
    // This is fine because everything inside of this block is serialized --
    // only one thread can be here at the same time
I
Igor Canadi 已提交
1812
    if (new_descriptor_log) {
1813
      // create manifest file
1814
      Log(db_options_->info_log,
1815
          "Creating manifest %" PRIu64 "\n", pending_manifest_file_number_);
I
Igor Canadi 已提交
1816
      unique_ptr<WritableFile> descriptor_file;
1817 1818
      s = env_->NewWritableFile(
          DescriptorFileName(dbname_, pending_manifest_file_number_),
1819
          &descriptor_file, env_->OptimizeForManifestWrite(env_options_));
I
Igor Canadi 已提交
1820
      if (s.ok()) {
1821
        descriptor_file->SetPreallocationBlockSize(
1822
            db_options_->manifest_preallocation_size);
I
Igor Canadi 已提交
1823 1824 1825 1826 1827
        descriptor_log_.reset(new log::Writer(std::move(descriptor_file)));
        s = WriteSnapshot(descriptor_log_.get());
      }
    }

1828
    if (!edit->IsColumnFamilyManipulation()) {
1829
      // This is cpu-heavy operations, which should be called outside mutex.
1830
      v->PrepareApply(mutable_cf_options, size_being_compacted);
1831
    }
1832 1833

    // Write new record to MANIFEST log
J
jorlow@chromium.org 已提交
1834
    if (s.ok()) {
I
Igor Canadi 已提交
1835 1836 1837
      for (auto& e : batch_edits) {
        std::string record;
        e->EncodeTo(&record);
1838 1839 1840 1841 1842
        s = descriptor_log_->AddRecord(record);
        if (!s.ok()) {
          break;
        }
      }
1843
      if (s.ok() && db_options_->disableDataSync == false) {
1844 1845
        if (db_options_->use_fsync) {
          StopWatch sw(env_, db_options_->statistics.get(),
1846
                       MANIFEST_FILE_SYNC_MICROS);
1847
          s = descriptor_log_->file()->Fsync();
1848
        } else {
1849
          StopWatch sw(env_, db_options_->statistics.get(),
1850
                       MANIFEST_FILE_SYNC_MICROS);
1851
          s = descriptor_log_->file()->Sync();
1852
        }
1853
      }
1854
      if (!s.ok()) {
1855 1856
        Log(db_options_->info_log, "MANIFEST write: %s\n",
            s.ToString().c_str());
I
Igor Canadi 已提交
1857 1858 1859 1860
        bool all_records_in = true;
        for (auto& e : batch_edits) {
          std::string record;
          e->EncodeTo(&record);
1861
          if (!ManifestContains(pending_manifest_file_number_, record)) {
I
Igor Canadi 已提交
1862 1863 1864 1865 1866
            all_records_in = false;
            break;
          }
        }
        if (all_records_in) {
1867
          Log(db_options_->info_log,
1868
              "MANIFEST contains log record despite error; advancing to new "
1869 1870
              "version to prevent mismatch between in-memory and logged state"
              " If paranoid is set, then the db is now in readonly mode.");
1871 1872 1873
          s = Status::OK();
        }
      }
J
jorlow@chromium.org 已提交
1874 1875
    }

1876 1877
    // If we just created a new descriptor file, install it by writing a
    // new CURRENT file that points to it.
1878
    if (s.ok() && new_descriptor_log) {
1879
      s = SetCurrentFile(env_, dbname_, pending_manifest_file_number_,
1880
                         db_options_->disableDataSync ? nullptr : db_directory);
1881
      if (s.ok() && pending_manifest_file_number_ > manifest_file_number_) {
I
Igor Canadi 已提交
1882
        // delete old manifest file
1883
        Log(db_options_->info_log,
1884 1885
            "Deleting manifest %" PRIu64 " current manifest %" PRIu64 "\n",
            manifest_file_number_, pending_manifest_file_number_);
I
Igor Canadi 已提交
1886 1887
        // we don't care about an error here, PurgeObsoleteFiles will take care
        // of it later
1888
        env_->DeleteFile(DescriptorFileName(dbname_, manifest_file_number_));
I
Igor Canadi 已提交
1889
      }
1890 1891
    }

1892 1893 1894 1895
    if (s.ok()) {
      // find offset in manifest file where this version is stored.
      new_manifest_file_size = descriptor_log_->file()->GetFileSize();
    }
A
Abhishek Kona 已提交
1896

1897
    LogFlush(db_options_->info_log);
1898
    mu->Lock();
J
jorlow@chromium.org 已提交
1899 1900 1901 1902
  }

  // Install the new version
  if (s.ok()) {
1903 1904 1905
    if (edit->is_column_family_add_) {
      // no group commit on column family add
      assert(batch_edits.size() == 1);
1906 1907
      assert(new_cf_options != nullptr);
      CreateColumnFamily(*new_cf_options, edit);
1908 1909
    } else if (edit->is_column_family_drop_) {
      assert(batch_edits.size() == 1);
I
Igor Canadi 已提交
1910
      column_family_data->SetDropped();
1911 1912 1913 1914
      if (column_family_data->Unref()) {
        delete column_family_data;
      }
    } else {
I
Igor Canadi 已提交
1915 1916 1917 1918 1919 1920 1921 1922
      uint64_t max_log_number_in_batch  = 0;
      for (auto& e : batch_edits) {
        if (e->has_log_number_) {
          max_log_number_in_batch =
              std::max(max_log_number_in_batch, e->log_number_);
        }
      }
      if (max_log_number_in_batch != 0) {
I
Igor Canadi 已提交
1923
        assert(column_family_data->GetLogNumber() <= max_log_number_in_batch);
I
Igor Canadi 已提交
1924 1925
        column_family_data->SetLogNumber(max_log_number_in_batch);
      }
1926 1927 1928
      AppendVersion(column_family_data, v);
    }

1929
    manifest_file_number_ = pending_manifest_file_number_;
1930
    manifest_file_size_ = new_manifest_file_size;
1931
    prev_log_number_ = edit->prev_log_number_;
J
jorlow@chromium.org 已提交
1932
  } else {
1933
    Log(db_options_->info_log, "Error in committing version %lu to [%s]",
I
Igor Canadi 已提交
1934 1935
        (unsigned long)v->GetVersionNumber(),
        column_family_data->GetName().c_str());
J
jorlow@chromium.org 已提交
1936
    delete v;
1937
    if (new_descriptor_log) {
1938
      Log(db_options_->info_log,
1939 1940
        "Deleting manifest %" PRIu64 " current manifest %" PRIu64 "\n",
        manifest_file_number_, pending_manifest_file_number_);
1941
      descriptor_log_.reset();
1942 1943
      env_->DeleteFile(
          DescriptorFileName(dbname_, pending_manifest_file_number_));
J
jorlow@chromium.org 已提交
1944 1945
    }
  }
1946
  pending_manifest_file_number_ = 0;
J
jorlow@chromium.org 已提交
1947

1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962
  // 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 已提交
1963 1964 1965
  return s;
}

I
Igor Canadi 已提交
1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976
void VersionSet::LogAndApplyCFHelper(VersionEdit* edit) {
  assert(edit->IsColumnFamilyManipulation());
  edit->SetNextFile(next_file_number_);
  edit->SetLastSequence(last_sequence_);
  if (edit->is_column_family_drop_) {
    // if we drop column family, we have to make sure to save max column family,
    // so that we don't reuse existing ID
    edit->SetMaxColumnFamily(column_family_set_->GetMaxColumnFamily());
  }
}

1977 1978 1979
void VersionSet::LogAndApplyHelper(ColumnFamilyData* cfd, Builder* builder,
                                   Version* v, VersionEdit* edit,
                                   port::Mutex* mu) {
1980
  mu->AssertHeld();
I
Igor Canadi 已提交
1981
  assert(!edit->IsColumnFamilyManipulation());
1982

1983 1984
  if (edit->has_log_number_) {
    assert(edit->log_number_ >= cfd->GetLogNumber());
I
Igor Canadi 已提交
1985
    assert(edit->log_number_ < next_file_number_);
I
Igor Canadi 已提交
1986
  }
1987

I
Igor Canadi 已提交
1988 1989 1990 1991 1992 1993
  if (!edit->has_prev_log_number_) {
    edit->SetPrevLogNumber(prev_log_number_);
  }
  edit->SetNextFile(next_file_number_);
  edit->SetLastSequence(last_sequence_);

1994
  builder->Apply(edit);
1995 1996
}

I
Igor Canadi 已提交
1997
Status VersionSet::Recover(
1998 1999
    const std::vector<ColumnFamilyDescriptor>& column_families,
    bool read_only) {
I
Igor Canadi 已提交
2000 2001 2002 2003 2004 2005 2006
  std::unordered_map<std::string, ColumnFamilyOptions> cf_name_to_options;
  for (auto cf : column_families) {
    cf_name_to_options.insert({cf.name, cf.options});
  }
  // keeps track of column families in manifest that were not found in
  // column families parameters. if those column families are not dropped
  // by subsequent manifest records, Recover() will return failure status
I
Igor Canadi 已提交
2007
  std::unordered_map<int, std::string> column_families_not_found;
J
jorlow@chromium.org 已提交
2008 2009

  // Read "CURRENT" file, which contains a pointer to the current manifest file
K
kailiu 已提交
2010 2011 2012 2013
  std::string manifest_filename;
  Status s = ReadFileToString(
      env_, CurrentFileName(dbname_), &manifest_filename
  );
J
jorlow@chromium.org 已提交
2014 2015 2016
  if (!s.ok()) {
    return s;
  }
K
kailiu 已提交
2017 2018
  if (manifest_filename.empty() ||
      manifest_filename.back() != '\n') {
J
jorlow@chromium.org 已提交
2019 2020
    return Status::Corruption("CURRENT file does not end with newline");
  }
K
kailiu 已提交
2021 2022
  // remove the trailing '\n'
  manifest_filename.resize(manifest_filename.size() - 1);
2023 2024 2025 2026 2027 2028
  FileType type;
  bool parse_ok =
      ParseFileName(manifest_filename, &manifest_file_number_, &type);
  if (!parse_ok || type != kDescriptorFile) {
    return Status::Corruption("CURRENT file corrupted");
  }
J
jorlow@chromium.org 已提交
2029

2030
  Log(db_options_->info_log, "Recovering from manifest file: %s\n",
K
kailiu 已提交
2031
      manifest_filename.c_str());
H
heyongqiang 已提交
2032

K
kailiu 已提交
2033 2034
  manifest_filename = dbname_ + "/" + manifest_filename;
  unique_ptr<SequentialFile> manifest_file;
2035
  s = env_->NewSequentialFile(manifest_filename, &manifest_file,
2036
                              env_options_);
J
jorlow@chromium.org 已提交
2037 2038 2039
  if (!s.ok()) {
    return s;
  }
2040
  uint64_t manifest_file_size;
K
kailiu 已提交
2041
  s = env_->GetFileSize(manifest_filename, &manifest_file_size);
2042 2043 2044
  if (!s.ok()) {
    return s;
  }
J
jorlow@chromium.org 已提交
2045 2046

  bool have_log_number = false;
2047
  bool have_prev_log_number = false;
J
jorlow@chromium.org 已提交
2048 2049 2050
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
2051 2052 2053
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
  uint64_t prev_log_number = 0;
2054
  uint32_t max_column_family = 0;
2055
  std::unordered_map<uint32_t, Builder*> builders;
J
jorlow@chromium.org 已提交
2056

2057
  // add default column family
2058
  auto default_cf_iter = cf_name_to_options.find(kDefaultColumnFamilyName);
I
Igor Canadi 已提交
2059
  if (default_cf_iter == cf_name_to_options.end()) {
I
Igor Canadi 已提交
2060
    return Status::InvalidArgument("Default column family not specified");
I
Igor Canadi 已提交
2061
  }
I
Igor Canadi 已提交
2062
  VersionEdit default_cf_edit;
2063
  default_cf_edit.AddColumnFamily(kDefaultColumnFamilyName);
I
Igor Canadi 已提交
2064 2065 2066 2067
  default_cf_edit.SetColumnFamily(0);
  ColumnFamilyData* default_cfd =
      CreateColumnFamily(default_cf_iter->second, &default_cf_edit);
  builders.insert({0, new Builder(default_cfd)});
2068

J
jorlow@chromium.org 已提交
2069
  {
I
Igor Canadi 已提交
2070
    VersionSet::LogReporter reporter;
J
jorlow@chromium.org 已提交
2071
    reporter.status = &s;
K
kailiu 已提交
2072 2073
    log::Reader reader(std::move(manifest_file), &reporter, true /*checksum*/,
                       0 /*initial_offset*/);
J
jorlow@chromium.org 已提交
2074 2075 2076
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2077
      VersionEdit edit;
J
jorlow@chromium.org 已提交
2078
      s = edit.DecodeFrom(record);
2079 2080
      if (!s.ok()) {
        break;
J
jorlow@chromium.org 已提交
2081 2082
      }

I
Igor Canadi 已提交
2083 2084 2085 2086 2087
      // Not found means that user didn't supply that column
      // family option AND we encountered column family add
      // record. Once we encounter column family drop record,
      // we will delete the column family from
      // column_families_not_found.
I
Igor Canadi 已提交
2088 2089 2090
      bool cf_in_not_found =
          column_families_not_found.find(edit.column_family_) !=
          column_families_not_found.end();
I
Igor Canadi 已提交
2091 2092
      // in builders means that user supplied that column family
      // option AND that we encountered column family add record
I
Igor Canadi 已提交
2093 2094 2095 2096 2097 2098
      bool cf_in_builders =
          builders.find(edit.column_family_) != builders.end();

      // they can't both be true
      assert(!(cf_in_not_found && cf_in_builders));

2099 2100
      ColumnFamilyData* cfd = nullptr;

2101
      if (edit.is_column_family_add_) {
I
Igor Canadi 已提交
2102 2103 2104 2105 2106 2107 2108
        if (cf_in_builders || cf_in_not_found) {
          s = Status::Corruption(
              "Manifest adding the same column family twice");
          break;
        }
        auto cf_options = cf_name_to_options.find(edit.column_family_name_);
        if (cf_options == cf_name_to_options.end()) {
I
Igor Canadi 已提交
2109 2110
          column_families_not_found.insert(
              {edit.column_family_, edit.column_family_name_});
I
Igor Canadi 已提交
2111
        } else {
2112 2113
          cfd = CreateColumnFamily(cf_options->second, &edit);
          builders.insert({edit.column_family_, new Builder(cfd)});
I
Igor Canadi 已提交
2114
        }
2115
      } else if (edit.is_column_family_drop_) {
I
Igor Canadi 已提交
2116 2117 2118 2119 2120
        if (cf_in_builders) {
          auto builder = builders.find(edit.column_family_);
          assert(builder != builders.end());
          delete builder->second;
          builders.erase(builder);
2121
          cfd = column_family_set_->GetColumnFamily(edit.column_family_);
2122 2123
          if (cfd->Unref()) {
            delete cfd;
2124
            cfd = nullptr;
2125 2126 2127 2128
          } else {
            // who else can have reference to cfd!?
            assert(false);
          }
I
Igor Canadi 已提交
2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142
        } else if (cf_in_not_found) {
          column_families_not_found.erase(edit.column_family_);
        } else {
          s = Status::Corruption(
              "Manifest - dropping non-existing column family");
          break;
        }
      } else if (!cf_in_not_found) {
        if (!cf_in_builders) {
          s = Status::Corruption(
              "Manifest record referencing unknown column family");
          break;
        }

2143
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
I
Igor Canadi 已提交
2144 2145
        // this should never happen since cf_in_builders is true
        assert(cfd != nullptr);
2146
        if (edit.max_level_ >= cfd->current()->NumberLevels()) {
2147 2148 2149 2150 2151
          s = Status::InvalidArgument(
              "db has more levels than options.num_levels");
          break;
        }

2152 2153 2154 2155 2156 2157 2158 2159 2160
        // if it is not column family add or column family drop,
        // then it's a file add/delete, which should be forwarded
        // to builder
        auto builder = builders.find(edit.column_family_);
        assert(builder != builders.end());
        builder->second->Apply(&edit);
      }

      if (cfd != nullptr) {
2161
        if (edit.has_log_number_) {
I
Igor Canadi 已提交
2162
          if (cfd->GetLogNumber() > edit.log_number_) {
2163
            Log(db_options_->info_log,
2164 2165
                "MANIFEST corruption detected, but ignored - Log numbers in "
                "records NOT monotonically increasing");
2166 2167 2168
          } else {
            cfd->SetLogNumber(edit.log_number_);
            have_log_number = true;
I
Igor Canadi 已提交
2169
          }
2170
        }
I
Igor Canadi 已提交
2171 2172 2173 2174 2175 2176 2177
        if (edit.has_comparator_ &&
            edit.comparator_ != cfd->user_comparator()->Name()) {
          s = Status::InvalidArgument(
              cfd->user_comparator()->Name(),
              "does not match existing comparator " + edit.comparator_);
          break;
        }
2178 2179
      }

2180 2181 2182 2183 2184
      if (edit.has_prev_log_number_) {
        prev_log_number = edit.prev_log_number_;
        have_prev_log_number = true;
      }

J
jorlow@chromium.org 已提交
2185 2186 2187 2188 2189
      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

2190 2191 2192 2193
      if (edit.has_max_column_family_) {
        max_column_family = edit.max_column_family_;
      }

J
jorlow@chromium.org 已提交
2194
      if (edit.has_last_sequence_) {
2195
        last_sequence = edit.last_sequence_;
J
jorlow@chromium.org 已提交
2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208
        have_last_sequence = true;
      }
    }
  }

  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");
    }
2209 2210 2211 2212

    if (!have_prev_log_number) {
      prev_log_number = 0;
    }
2213

2214 2215
    column_family_set_->UpdateMaxColumnFamily(max_column_family);

2216 2217
    MarkFileNumberUsed(prev_log_number);
    MarkFileNumberUsed(log_number);
J
jorlow@chromium.org 已提交
2218 2219
  }

I
Igor Canadi 已提交
2220
  // there were some column families in the MANIFEST that weren't specified
2221
  // in the argument. This is OK in read_only mode
2222
  if (read_only == false && !column_families_not_found.empty()) {
2223
    std::string list_of_not_found;
I
Igor Canadi 已提交
2224 2225
    for (const auto& cf : column_families_not_found) {
      list_of_not_found += ", " + cf.second;
2226 2227
    }
    list_of_not_found = list_of_not_found.substr(2);
I
Igor Canadi 已提交
2228
    s = Status::InvalidArgument(
I
Igor Canadi 已提交
2229 2230
        "You have to open all column families. Column families not opened: " +
        list_of_not_found);
I
Igor Canadi 已提交
2231 2232
  }

J
jorlow@chromium.org 已提交
2233
  if (s.ok()) {
I
Igor Canadi 已提交
2234
    for (auto cfd : *column_family_set_) {
2235 2236 2237 2238
      auto builders_iter = builders.find(cfd->GetID());
      assert(builders_iter != builders.end());
      auto builder = builders_iter->second;

2239
      if (db_options_->max_open_files == -1) {
2240 2241
      // unlimited table cache. Pre-load table handle now.
      // Need to do it out of the mutex.
2242 2243
        builder->LoadTableHandlers();
      }
2244

2245
      Version* v = new Version(cfd, this, current_version_number_++);
2246
      builder->SaveTo(v);
2247

2248
      // Install recovered version
2249
      std::vector<uint64_t> size_being_compacted(v->NumberLevels() - 1);
2250
      cfd->compaction_picker()->SizeBeingCompacted(size_being_compacted);
2251
      v->PrepareApply(*cfd->GetLatestMutableCFOptions(), size_being_compacted);
I
Igor Canadi 已提交
2252
      AppendVersion(cfd, v);
2253
    }
2254

2255
    manifest_file_size_ = manifest_file_size;
2256 2257 2258
    next_file_number_ = next_file + 1;
    last_sequence_ = last_sequence;
    prev_log_number_ = prev_log_number;
H
heyongqiang 已提交
2259

2260
    Log(db_options_->info_log,
2261
        "Recovered from manifest file:%s succeeded,"
K
Kai Liu 已提交
2262 2263
        "manifest_file_number is %lu, next_file_number is %lu, "
        "last_sequence is %lu, log_number is %lu,"
2264 2265
        "prev_log_number is %lu,"
        "max_column_family is %u\n",
2266 2267 2268
        manifest_filename.c_str(), (unsigned long)manifest_file_number_,
        (unsigned long)next_file_number_, (unsigned long)last_sequence_,
        (unsigned long)log_number, (unsigned long)prev_log_number_,
2269
        column_family_set_->GetMaxColumnFamily());
2270 2271

    for (auto cfd : *column_family_set_) {
2272
      Log(db_options_->info_log,
I
Igor Canadi 已提交
2273 2274
          "Column family [%s] (ID %u), log number is %" PRIu64 "\n",
          cfd->GetName().c_str(), cfd->GetID(), cfd->GetLogNumber());
2275
    }
J
jorlow@chromium.org 已提交
2276 2277
  }

2278 2279 2280 2281
  for (auto builder : builders) {
    delete builder.second;
  }

J
jorlow@chromium.org 已提交
2282 2283 2284
  return s;
}

I
Igor Canadi 已提交
2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309
Status VersionSet::ListColumnFamilies(std::vector<std::string>* column_families,
                                      const std::string& dbname, Env* env) {
  // these are just for performance reasons, not correcntes,
  // so we're fine using the defaults
  EnvOptions soptions;
  // Read "CURRENT" file, which contains a pointer to the current manifest file
  std::string current;
  Status s = ReadFileToString(env, CurrentFileName(dbname), &current);
  if (!s.ok()) {
    return s;
  }
  if (current.empty() || current[current.size()-1] != '\n') {
    return Status::Corruption("CURRENT file does not end with newline");
  }
  current.resize(current.size() - 1);

  std::string dscname = dbname + "/" + current;
  unique_ptr<SequentialFile> file;
  s = env->NewSequentialFile(dscname, &file, soptions);
  if (!s.ok()) {
    return s;
  }

  std::map<uint32_t, std::string> column_family_names;
  // default column family is always implicitly there
2310
  column_family_names.insert({0, kDefaultColumnFamilyName});
I
Igor Canadi 已提交
2311 2312 2313 2314 2315 2316 2317
  VersionSet::LogReporter reporter;
  reporter.status = &s;
  log::Reader reader(std::move(file), &reporter, true /*checksum*/,
                     0 /*initial_offset*/);
  Slice record;
  std::string scratch;
  while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2318 2319 2320 2321 2322 2323
    VersionEdit edit;
    s = edit.DecodeFrom(record);
    if (!s.ok()) {
      break;
    }
    if (edit.is_column_family_add_) {
2324 2325 2326 2327 2328
      if (column_family_names.find(edit.column_family_) !=
          column_family_names.end()) {
        s = Status::Corruption("Manifest adding the same column family twice");
        break;
      }
2329 2330 2331
      column_family_names.insert(
          {edit.column_family_, edit.column_family_name_});
    } else if (edit.is_column_family_drop_) {
2332 2333 2334 2335 2336 2337
      if (column_family_names.find(edit.column_family_) ==
          column_family_names.end()) {
        s = Status::Corruption(
            "Manifest - dropping non-existing column family");
        break;
      }
2338 2339
      column_family_names.erase(edit.column_family_);
    }
I
Igor Canadi 已提交
2340 2341 2342 2343 2344 2345
  }

  column_families->clear();
  if (s.ok()) {
    for (const auto& iter : column_family_names) {
      column_families->push_back(iter.second);
2346
    }
I
Igor Canadi 已提交
2347 2348 2349 2350
  }

  return s;
}
2351

I
Igor Canadi 已提交
2352
#ifndef ROCKSDB_LITE
2353 2354
Status VersionSet::ReduceNumberOfLevels(const std::string& dbname,
                                        const Options* options,
2355
                                        const EnvOptions& env_options,
2356 2357 2358 2359 2360 2361
                                        int new_levels) {
  if (new_levels <= 1) {
    return Status::InvalidArgument(
        "Number of levels needs to be bigger than 1");
  }

I
Igor Canadi 已提交
2362 2363 2364 2365
  ColumnFamilyOptions cf_options(*options);
  std::shared_ptr<Cache> tc(NewLRUCache(
      options->max_open_files - 10, options->table_cache_numshardbits,
      options->table_cache_remove_scan_count_limit));
2366
  WriteController wc;
2367
  VersionSet versions(dbname, options, env_options, tc.get(), &wc);
2368 2369
  Status status;

2370
  std::vector<ColumnFamilyDescriptor> dummy;
2371
  ColumnFamilyDescriptor dummy_descriptor(kDefaultColumnFamilyName,
I
Igor Canadi 已提交
2372
                                          ColumnFamilyOptions(*options));
I
Igor Canadi 已提交
2373
  dummy.push_back(dummy_descriptor);
2374
  status = versions.Recover(dummy);
2375 2376 2377 2378
  if (!status.ok()) {
    return status;
  }

2379
  Version* current_version =
2380
      versions.GetColumnFamilySet()->GetDefault()->current();
2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409
  int current_levels = current_version->NumberLevels();

  if (current_levels <= new_levels) {
    return Status::OK();
  }

  // Make sure there are file only on one level from
  // (new_levels-1) to (current_levels-1)
  int first_nonempty_level = -1;
  int first_nonempty_level_filenum = 0;
  for (int i = new_levels - 1; i < current_levels; i++) {
    int file_num = current_version->NumLevelFiles(i);
    if (file_num != 0) {
      if (first_nonempty_level < 0) {
        first_nonempty_level = i;
        first_nonempty_level_filenum = file_num;
      } else {
        char msg[255];
        snprintf(msg, sizeof(msg),
                 "Found at least two levels containing files: "
                 "[%d:%d],[%d:%d].\n",
                 first_nonempty_level, first_nonempty_level_filenum, i,
                 file_num);
        return Status::InvalidArgument(msg);
      }
    }
  }

  std::vector<FileMetaData*>* old_files_list = current_version->files_;
I
Igor Canadi 已提交
2410 2411 2412
  // we need to allocate an array with the old number of levels size to
  // avoid SIGSEGV in WriteSnapshot()
  // however, all levels bigger or equal to new_levels will be empty
2413
  std::vector<FileMetaData*>* new_files_list =
I
Igor Canadi 已提交
2414
      new std::vector<FileMetaData*>[current_levels];
2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426
  for (int i = 0; i < new_levels - 1; i++) {
    new_files_list[i] = old_files_list[i];
  }

  if (first_nonempty_level > 0) {
    new_files_list[new_levels - 1] = old_files_list[first_nonempty_level];
  }

  delete[] current_version->files_;
  current_version->files_ = new_files_list;
  current_version->num_levels_ = new_levels;

2427
  MutableCFOptions mutable_cf_options(*options, ImmutableCFOptions(*options));
2428 2429 2430
  VersionEdit ve;
  port::Mutex dummy_mutex;
  MutexLock l(&dummy_mutex);
2431 2432 2433
  return versions.LogAndApply(
      versions.GetColumnFamilySet()->GetDefault(),
      mutable_cf_options, &ve, &dummy_mutex, nullptr, true);
2434 2435
}

I
Igor Canadi 已提交
2436 2437
Status VersionSet::DumpManifest(Options& options, std::string& dscname,
                                bool verbose, bool hex) {
2438
  // Open the specified manifest file.
2439
  unique_ptr<SequentialFile> file;
2440
  Status s = options.env->NewSequentialFile(dscname, &file, env_options_);
2441 2442 2443 2444 2445 2446 2447 2448 2449 2450
  if (!s.ok()) {
    return s;
  }

  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 prev_log_number = 0;
2451
  int count = 0;
2452 2453 2454 2455 2456
  std::unordered_map<uint32_t, std::string> comparators;
  std::unordered_map<uint32_t, Builder*> builders;

  // add default column family
  VersionEdit default_cf_edit;
2457
  default_cf_edit.AddColumnFamily(kDefaultColumnFamilyName);
2458 2459 2460
  default_cf_edit.SetColumnFamily(0);
  ColumnFamilyData* default_cfd =
      CreateColumnFamily(ColumnFamilyOptions(options), &default_cf_edit);
2461
  builders.insert({0, new Builder(default_cfd)});
2462 2463

  {
I
Igor Canadi 已提交
2464
    VersionSet::LogReporter reporter;
2465
    reporter.status = &s;
2466 2467
    log::Reader reader(std::move(file), &reporter, true/*checksum*/,
                       0/*initial_offset*/);
2468 2469 2470
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2471
      VersionEdit edit;
2472
      s = edit.DecodeFrom(record);
2473 2474
      if (!s.ok()) {
        break;
2475 2476
      }

2477 2478
      // Write out each individual edit
      if (verbose) {
A
Abhishek Kona 已提交
2479
        printf("*************************Edit[%d] = %s\n",
2480
                count, edit.DebugString(hex).c_str());
2481 2482 2483
      }
      count++;

2484 2485 2486 2487 2488
      bool cf_in_builders =
          builders.find(edit.column_family_) != builders.end();

      if (edit.has_comparator_) {
        comparators.insert({edit.column_family_, edit.comparator_});
2489 2490
      }

2491 2492
      ColumnFamilyData* cfd = nullptr;

2493 2494 2495 2496 2497 2498
      if (edit.is_column_family_add_) {
        if (cf_in_builders) {
          s = Status::Corruption(
              "Manifest adding the same column family twice");
          break;
        }
2499 2500
        cfd = CreateColumnFamily(ColumnFamilyOptions(options), &edit);
        builders.insert({edit.column_family_, new Builder(cfd)});
2501 2502 2503 2504 2505 2506 2507 2508 2509 2510
      } else if (edit.is_column_family_drop_) {
        if (!cf_in_builders) {
          s = Status::Corruption(
              "Manifest - dropping non-existing column family");
          break;
        }
        auto builder_iter = builders.find(edit.column_family_);
        delete builder_iter->second;
        builders.erase(builder_iter);
        comparators.erase(edit.column_family_);
2511
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
2512 2513 2514
        assert(cfd != nullptr);
        cfd->Unref();
        delete cfd;
2515
        cfd = nullptr;
2516 2517 2518 2519 2520 2521 2522
      } else {
        if (!cf_in_builders) {
          s = Status::Corruption(
              "Manifest record referencing unknown column family");
          break;
        }

2523
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
2524 2525 2526 2527 2528 2529 2530 2531 2532
        // this should never happen since cf_in_builders is true
        assert(cfd != nullptr);

        // if it is not column family add or column family drop,
        // then it's a file add/delete, which should be forwarded
        // to builder
        auto builder = builders.find(edit.column_family_);
        assert(builder != builders.end());
        builder->second->Apply(&edit);
2533 2534
      }

2535 2536 2537 2538
      if (cfd != nullptr && edit.has_log_number_) {
        cfd->SetLogNumber(edit.log_number_);
      }

2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552
      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;
      }
2553 2554 2555 2556

      if (edit.has_max_column_family_) {
        column_family_set_->UpdateMaxColumnFamily(edit.max_column_family_);
      }
2557 2558
    }
  }
2559
  file.reset();
2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575

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

  if (s.ok()) {
2576 2577 2578 2579 2580 2581 2582
    for (auto cfd : *column_family_set_) {
      auto builders_iter = builders.find(cfd->GetID());
      assert(builders_iter != builders.end());
      auto builder = builders_iter->second;

      Version* v = new Version(cfd, this, current_version_number_++);
      builder->SaveTo(v);
I
Igor Canadi 已提交
2583 2584
      std::vector<uint64_t> size_being_compacted(v->NumberLevels() - 1);
      cfd->compaction_picker()->SizeBeingCompacted(size_being_compacted);
2585
      v->PrepareApply(*cfd->GetLatestMutableCFOptions(), size_being_compacted);
2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599
      delete builder;

      printf("--------------- Column family \"%s\"  (ID %u) --------------\n",
             cfd->GetName().c_str(), (unsigned int)cfd->GetID());
      printf("log number: %lu\n", (unsigned long)cfd->GetLogNumber());
      auto comparator = comparators.find(cfd->GetID());
      if (comparator != comparators.end()) {
        printf("comparator: %s\n", comparator->second.c_str());
      } else {
        printf("comparator: <NO COMPARATOR>\n");
      }
      printf("%s \n", v->DebugString(hex).c_str());
      delete v;
    }
2600

2601 2602 2603
    next_file_number_ = next_file + 1;
    last_sequence_ = last_sequence;
    prev_log_number_ = prev_log_number;
2604

2605
    printf(
2606
        "next_file_number %lu last_sequence "
2607
        "%lu  prev_log_number %lu max_column_family %u\n",
2608 2609
        (unsigned long)next_file_number_, (unsigned long)last_sequence,
        (unsigned long)prev_log_number,
2610
        column_family_set_->GetMaxColumnFamily());
2611
  }
2612

2613 2614
  return s;
}
I
Igor Canadi 已提交
2615
#endif  // ROCKSDB_LITE
2616

2617 2618 2619 2620 2621 2622
void VersionSet::MarkFileNumberUsed(uint64_t number) {
  if (next_file_number_ <= number) {
    next_file_number_ = number + 1;
  }
}

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

I
Igor Canadi 已提交
2626 2627
  // WARNING: This method doesn't hold a mutex!!

I
Igor Canadi 已提交
2628 2629
  // This is done without DB mutex lock held, but only within single-threaded
  // LogAndApply. Column family manipulations can only happen within LogAndApply
I
Igor Canadi 已提交
2630
  // (the same single thread), so we're safe to iterate.
I
Igor Canadi 已提交
2631
  for (auto cfd : *column_family_set_) {
2632 2633 2634
    {
      // Store column family info
      VersionEdit edit;
2635
      if (cfd->GetID() != 0) {
2636 2637
        // default column family is always there,
        // no need to explicitly write it
2638 2639
        edit.AddColumnFamily(cfd->GetName());
        edit.SetColumnFamily(cfd->GetID());
I
Igor Canadi 已提交
2640 2641 2642 2643 2644 2645 2646 2647
      }
      edit.SetComparatorName(
          cfd->internal_comparator().user_comparator()->Name());
      std::string record;
      edit.EncodeTo(&record);
      Status s = log->AddRecord(record);
      if (!s.ok()) {
        return s;
2648
      }
2649
    }
2650

2651 2652 2653
    {
      // Save files
      VersionEdit edit;
2654
      edit.SetColumnFamily(cfd->GetID());
2655

I
Igor Canadi 已提交
2656
      for (int level = 0; level < cfd->NumberLevels(); level++) {
2657
        for (const auto& f : cfd->current()->files_[level]) {
2658 2659 2660
          edit.AddFile(level, f->fd.GetNumber(), f->fd.GetPathId(),
                       f->fd.GetFileSize(), f->smallest, f->largest,
                       f->smallest_seqno, f->largest_seqno);
2661 2662
        }
      }
2663
      edit.SetLogNumber(cfd->GetLogNumber());
2664 2665 2666 2667 2668
      std::string record;
      edit.EncodeTo(&record);
      Status s = log->AddRecord(record);
      if (!s.ok()) {
        return s;
2669
      }
2670 2671 2672
    }
  }

I
Igor Canadi 已提交
2673
  return Status::OK();
J
jorlow@chromium.org 已提交
2674 2675
}

2676 2677
// Opens the mainfest file and reads all records
// till it finds the record we are looking for.
2678 2679 2680 2681
bool VersionSet::ManifestContains(uint64_t manifest_file_number,
                                  const std::string& record) const {
  std::string fname =
      DescriptorFileName(dbname_, manifest_file_number);
2682
  Log(db_options_->info_log, "ManifestContains: checking %s\n", fname.c_str());
2683
  unique_ptr<SequentialFile> file;
2684
  Status s = env_->NewSequentialFile(fname, &file, env_options_);
2685
  if (!s.ok()) {
2686 2687
    Log(db_options_->info_log, "ManifestContains: %s\n", s.ToString().c_str());
    Log(db_options_->info_log,
2688 2689
        "ManifestContains: is unable to reopen the manifest file  %s",
        fname.c_str());
2690 2691
    return false;
  }
A
Abhishek Kona 已提交
2692
  log::Reader reader(std::move(file), nullptr, true/*checksum*/, 0);
2693 2694 2695 2696 2697 2698 2699 2700 2701
  Slice r;
  std::string scratch;
  bool result = false;
  while (reader.ReadRecord(&r, &scratch)) {
    if (r == Slice(record)) {
      result = true;
      break;
    }
  }
2702
  Log(db_options_->info_log, "ManifestContains: result = %d\n", result ? 1 : 0);
2703 2704 2705 2706
  return result;
}


J
jorlow@chromium.org 已提交
2707 2708
uint64_t VersionSet::ApproximateOffsetOf(Version* v, const InternalKey& ikey) {
  uint64_t result = 0;
2709
  for (int level = 0; level < v->NumberLevels(); level++) {
J
jorlow@chromium.org 已提交
2710
    const std::vector<FileMetaData*>& files = v->files_[level];
D
dgrogan@chromium.org 已提交
2711
    for (size_t i = 0; i < files.size(); i++) {
I
Igor Canadi 已提交
2712 2713
      if (v->cfd_->internal_comparator().Compare(files[i]->largest, ikey) <=
          0) {
J
jorlow@chromium.org 已提交
2714
        // Entire file is before "ikey", so just add the file size
2715
        result += files[i]->fd.GetFileSize();
I
Igor Canadi 已提交
2716 2717
      } else if (v->cfd_->internal_comparator().Compare(files[i]->smallest,
                                                        ikey) > 0) {
J
jorlow@chromium.org 已提交
2718 2719 2720 2721 2722 2723 2724 2725 2726 2727
        // 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.
S
Siying Dong 已提交
2728
        TableReader* table_reader_ptr;
I
Igor Canadi 已提交
2729
        Iterator* iter = v->cfd_->table_cache()->NewIterator(
2730
            ReadOptions(), env_options_, v->cfd_->internal_comparator(),
2731
            files[i]->fd, &table_reader_ptr);
S
Siying Dong 已提交
2732 2733
        if (table_reader_ptr != nullptr) {
          result += table_reader_ptr->ApproximateOffsetOf(ikey.Encode());
J
jorlow@chromium.org 已提交
2734 2735 2736 2737 2738 2739 2740 2741
        }
        delete iter;
      }
    }
  }
  return result;
}

2742
void VersionSet::AddLiveFiles(std::vector<FileDescriptor>* live_list) {
2743 2744
  // pre-calculate space requirement
  int64_t total_files = 0;
I
Igor Canadi 已提交
2745
  for (auto cfd : *column_family_set_) {
2746 2747
    Version* dummy_versions = cfd->dummy_versions();
    for (Version* v = dummy_versions->next_; v != dummy_versions;
I
Igor Canadi 已提交
2748
         v = v->next_) {
2749
      for (int level = 0; level < v->NumberLevels(); level++) {
2750 2751
        total_files += v->files_[level].size();
      }
2752 2753 2754 2755 2756 2757
    }
  }

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

I
Igor Canadi 已提交
2758
  for (auto cfd : *column_family_set_) {
2759 2760
    Version* dummy_versions = cfd->dummy_versions();
    for (Version* v = dummy_versions->next_; v != dummy_versions;
I
Igor Canadi 已提交
2761
         v = v->next_) {
2762
      for (int level = 0; level < v->NumberLevels(); level++) {
2763
        for (const auto& f : v->files_[level]) {
2764
          live_list->push_back(f->fd);
2765
        }
J
jorlow@chromium.org 已提交
2766 2767 2768 2769 2770 2771
      }
    }
  }
}

Iterator* VersionSet::MakeInputIterator(Compaction* c) {
L
Lei Jin 已提交
2772 2773 2774 2775 2776
  auto cfd = c->column_family_data();
  ReadOptions read_options;
  read_options.verify_checksums =
    cfd->options()->verify_checksums_in_compaction;
  read_options.fill_cache = false;
J
jorlow@chromium.org 已提交
2777 2778 2779 2780

  // 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
2781 2782 2783
  const int space = (c->level() == 0 ?
      c->input_levels(0)->num_files + c->num_input_levels() - 1:
      c->num_input_levels());
J
jorlow@chromium.org 已提交
2784 2785
  Iterator** list = new Iterator*[space];
  int num = 0;
2786
  for (int which = 0; which < c->num_input_levels(); which++) {
F
Feng Zhu 已提交
2787
    if (c->input_levels(which)->num_files != 0) {
2788
      if (c->level(which) == 0) {
F
Feng Zhu 已提交
2789 2790
        const FileLevel* flevel = c->input_levels(which);
        for (size_t i = 0; i < flevel->num_files; i++) {
L
Lei Jin 已提交
2791
          list[num++] = cfd->table_cache()->NewIterator(
2792
              read_options, env_options_compactions_,
F
Feng Zhu 已提交
2793
              cfd->internal_comparator(), flevel->files[i].fd, nullptr,
K
Kai Liu 已提交
2794
              true /* for compaction */);
J
jorlow@chromium.org 已提交
2795 2796 2797
        }
      } else {
        // Create concatenating iterator for the files from this level
L
Lei Jin 已提交
2798
        list[num++] = NewTwoLevelIterator(new Version::LevelFileIteratorState(
2799
              cfd->table_cache(), read_options, env_options_,
L
Lei Jin 已提交
2800 2801 2802
              cfd->internal_comparator(), true /* for_compaction */,
              false /* prefix enabled */),
            new Version::LevelFileNumIterator(cfd->internal_comparator(),
F
Feng Zhu 已提交
2803
                                              c->input_levels(which)));
J
jorlow@chromium.org 已提交
2804 2805 2806 2807
      }
    }
  }
  assert(num <= space);
I
Igor Canadi 已提交
2808
  Iterator* result = NewMergingIterator(
I
Igor Canadi 已提交
2809
      &c->column_family_data()->internal_comparator(), list, num);
J
jorlow@chromium.org 已提交
2810 2811 2812 2813
  delete[] list;
  return result;
}

A
Abhishek Kona 已提交
2814
// verify that the files listed in this compaction are present
2815 2816
// in the current version
bool VersionSet::VerifyCompactionFileConsistency(Compaction* c) {
2817
#ifndef NDEBUG
I
Igor Canadi 已提交
2818
  Version* version = c->column_family_data()->current();
2819
  if (c->input_version() != version) {
2820
    Log(db_options_->info_log,
I
Igor Canadi 已提交
2821 2822
        "[%s] VerifyCompactionFileConsistency version mismatch",
        c->column_family_data()->GetName().c_str());
2823 2824 2825 2826 2827
  }

  // verify files in level
  int level = c->level();
  for (int i = 0; i < c->num_input_files(0); i++) {
2828
    uint64_t number = c->input(0, i)->fd.GetNumber();
2829 2830 2831

    // look for this file in the current version
    bool found = false;
2832 2833
    for (unsigned int j = 0; j < version->files_[level].size(); j++) {
      FileMetaData* f = version->files_[level][j];
2834
      if (f->fd.GetNumber() == number) {
2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845
        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++) {
2846
    uint64_t number = c->input(1, i)->fd.GetNumber();
2847 2848 2849

    // look for this file in the current version
    bool found = false;
2850 2851
    for (unsigned int j = 0; j < version->files_[level].size(); j++) {
      FileMetaData* f = version->files_[level][j];
2852
      if (f->fd.GetNumber() == number) {
2853 2854 2855 2856 2857 2858 2859 2860
        found = true;
        break;
      }
    }
    if (!found) {
      return false; // input files non existant in current version
    }
  }
2861
#endif
2862 2863 2864
  return true;     // everything good
}

2865
Status VersionSet::GetMetadataForFile(uint64_t number, int* filelevel,
2866
                                      FileMetaData** meta,
2867 2868
                                      ColumnFamilyData** cfd) {
  for (auto cfd_iter : *column_family_set_) {
2869
    Version* version = cfd_iter->current();
2870
    for (int level = 0; level < version->NumberLevels(); level++) {
I
Igor Canadi 已提交
2871
      for (const auto& file : version->files_[level]) {
2872
        if (file->fd.GetNumber() == number) {
2873
          *meta = file;
2874
          *filelevel = level;
2875
          *cfd = cfd_iter;
2876 2877
          return Status::OK();
        }
2878 2879 2880 2881 2882 2883
      }
    }
  }
  return Status::NotFound("File not present in any level");
}

2884
void VersionSet::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
I
Igor Canadi 已提交
2885
  for (auto cfd : *column_family_set_) {
I
Igor Canadi 已提交
2886
    for (int level = 0; level < cfd->NumberLevels(); level++) {
2887
      for (const auto& file : cfd->current()->files_[level]) {
2888
        LiveFileMetaData filemetadata;
2889
        filemetadata.column_family_name = cfd->GetName();
2890
        uint32_t path_id = file->fd.GetPathId();
2891 2892
        if (path_id < db_options_->db_paths.size()) {
          filemetadata.db_path = db_options_->db_paths[path_id].path;
2893
        } else {
2894 2895
          assert(!db_options_->db_paths.empty());
          filemetadata.db_path = db_options_->db_paths.back().path;
2896 2897
        }
        filemetadata.name = MakeTableFileName("", file->fd.GetNumber());
2898
        filemetadata.level = level;
2899
        filemetadata.size = file->fd.GetFileSize();
I
Igor Canadi 已提交
2900 2901 2902 2903
        filemetadata.smallestkey = file->smallest.user_key().ToString();
        filemetadata.largestkey = file->largest.user_key().ToString();
        filemetadata.smallest_seqno = file->smallest_seqno;
        filemetadata.largest_seqno = file->largest_seqno;
2904 2905
        metadata->push_back(filemetadata);
      }
2906 2907 2908 2909
    }
  }
}

2910
void VersionSet::GetObsoleteFiles(std::vector<FileMetaData*>* files) {
2911
  files->insert(files->end(), obsolete_files_.begin(), obsolete_files_.end());
I
Igor Canadi 已提交
2912 2913 2914
  obsolete_files_.clear();
}

2915
ColumnFamilyData* VersionSet::CreateColumnFamily(
2916
    const ColumnFamilyOptions& cf_options, VersionEdit* edit) {
2917 2918
  assert(edit->is_column_family_add_);

2919
  Version* dummy_versions = new Version(nullptr, this);
I
Igor Canadi 已提交
2920
  auto new_cfd = column_family_set_->CreateColumnFamily(
2921 2922
      edit->column_family_name_, edit->column_family_, dummy_versions,
      cf_options);
I
Igor Canadi 已提交
2923

2924 2925 2926
  Version* v = new Version(new_cfd, this, current_version_number_++);

  AppendVersion(new_cfd, v);
2927 2928 2929 2930
  // GetLatestMutableCFOptions() is safe here without mutex since the
  // cfd is not available to client
  new_cfd->CreateNewMemtable(MemTableOptions(
        *new_cfd->GetLatestMutableCFOptions(), *new_cfd->options()));
I
Igor Canadi 已提交
2931
  new_cfd->SetLogNumber(edit->log_number_);
2932 2933 2934
  return new_cfd;
}

2935
}  // namespace rocksdb