version_set.cc 126.7 KB
Newer Older
1
//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
2 3 4 5
//  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>
17
#include <stdio.h>
J
jorlow@chromium.org 已提交
18
#include <algorithm>
I
Igor Canadi 已提交
19
#include <map>
I
Igor Canadi 已提交
20
#include <set>
21
#include <climits>
22
#include <unordered_map>
23
#include <vector>
24
#include <string>
K
kailiu 已提交
25

26
#include "db/compaction.h"
J
jorlow@chromium.org 已提交
27
#include "db/filename.h"
28
#include "db/internal_stats.h"
J
jorlow@chromium.org 已提交
29 30 31
#include "db/log_reader.h"
#include "db/log_writer.h"
#include "db/memtable.h"
32
#include "db/merge_context.h"
33
#include "db/merge_helper.h"
34
#include "db/pinned_iterators_manager.h"
J
jorlow@chromium.org 已提交
35
#include "db/table_cache.h"
S
sdong 已提交
36
#include "db/version_builder.h"
37 38
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
39
#include "rocksdb/write_buffer_manager.h"
40 41
#include "table/format.h"
#include "table/get_context.h"
S
sdong 已提交
42
#include "table/internal_iterator.h"
J
jorlow@chromium.org 已提交
43
#include "table/merger.h"
44
#include "table/meta_blocks.h"
45 46 47
#include "table/plain_table_factory.h"
#include "table/table_reader.h"
#include "table/two_level_iterator.h"
J
jorlow@chromium.org 已提交
48
#include "util/coding.h"
49
#include "util/file_reader_writer.h"
J
jorlow@chromium.org 已提交
50
#include "util/logging.h"
51
#include "util/perf_context_imp.h"
52
#include "util/stop_watch.h"
S
sdong 已提交
53
#include "util/sync_point.h"
J
jorlow@chromium.org 已提交
54

55
namespace rocksdb {
J
jorlow@chromium.org 已提交
56

57 58
namespace {

59
// Find File in LevelFilesBrief data structure
60 61
// Within an index range defined by left and right
int FindFileInRange(const InternalKeyComparator& icmp,
62
    const LevelFilesBrief& file_level,
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
    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;
}

// 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:
90 91 92 93 94
  FilePicker(std::vector<FileMetaData*>* files, const Slice& user_key,
             const Slice& ikey, autovector<LevelFilesBrief>* file_levels,
             unsigned int num_levels, FileIndexer* file_indexer,
             const Comparator* user_comparator,
             const InternalKeyComparator* internal_comparator)
95
      : num_levels_(num_levels),
96
        curr_level_(static_cast<unsigned int>(-1)),
97
        returned_file_level_(static_cast<unsigned int>(-1)),
98
        hit_file_level_(static_cast<unsigned int>(-1)),
99 100
        search_left_bound_(0),
        search_right_bound_(FileIndexer::kLevelMaxIndex),
101
#ifndef NDEBUG
102
        files_(files),
103
#endif
104
        level_files_brief_(file_levels),
105
        is_hit_file_last_in_level_(false),
106 107 108 109 110 111 112 113 114
        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.
115 116
      for (unsigned int i = 0; i < (*level_files_brief_)[0].num_files; ++i) {
        auto* r = (*level_files_brief_)[0].files[i].fd.table_reader;
117 118 119 120 121 122 123
        if (r) {
          r->Prepare(ikey);
        }
      }
    }
  }

124 125
  int GetCurrentLevel() { return returned_file_level_; }

126 127 128 129 130
  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_];
131
        hit_file_level_ = curr_level_;
132 133
        is_hit_file_last_in_level_ =
            curr_index_in_curr_level_ == curr_file_level_->num_files - 1;
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
        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
198
        returned_file_level_ = curr_level_;
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
        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;
  }

214 215 216 217
  // getter for current file level
  // for GET_HIT_L0, GET_HIT_L1 & GET_HIT_L2_AND_UP counts
  unsigned int GetHitFileLevel() { return hit_file_level_; }

218 219 220 221
  // Returns true if the most recent "hit file" (i.e., one returned by
  // GetNextFile()) is at the last index in its level.
  bool IsHitFileLastInLevel() { return is_hit_file_last_in_level_; }

222 223 224
 private:
  unsigned int num_levels_;
  unsigned int curr_level_;
225
  unsigned int returned_file_level_;
226
  unsigned int hit_file_level_;
227 228
  int32_t search_left_bound_;
  int32_t search_right_bound_;
229
#ifndef NDEBUG
230
  std::vector<FileMetaData*>* files_;
231
#endif
232
  autovector<LevelFilesBrief>* level_files_brief_;
233
  bool search_ended_;
234
  bool is_hit_file_last_in_level_;
235
  LevelFilesBrief* curr_file_level_;
236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
  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_) {
252
      curr_file_level_ = &(*level_files_brief_)[curr_level_];
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
      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) {
285 286
            search_right_bound_ =
                static_cast<int32_t>(curr_file_level_->num_files) - 1;
287
          }
288 289 290 291
          start_index =
              FindFileInRange(*internal_comparator_, *curr_file_level_, ikey_,
                              static_cast<uint32_t>(search_left_bound_),
                              static_cast<uint32_t>(search_right_bound_));
292 293
        } else {
          // search_left_bound > search_right_bound, key does not exist in
C
clark.kang 已提交
294
          // this level. Since no comparison is done in this level, it will
295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
          // 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

S
sdong 已提交
315 316
VersionStorageInfo::~VersionStorageInfo() { delete[] files_; }

J
jorlow@chromium.org 已提交
317 318
Version::~Version() {
  assert(refs_ == 0);
319 320 321 322 323 324

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

  // Drop references to files
S
sdong 已提交
325 326 327
  for (int level = 0; level < storage_info_.num_levels_; level++) {
    for (size_t i = 0; i < storage_info_.files_[level].size(); i++) {
      FileMetaData* f = storage_info_.files_[level][i];
328
      assert(f->refs > 0);
J
jorlow@chromium.org 已提交
329 330
      f->refs--;
      if (f->refs <= 0) {
331
        if (f->table_reader_handle) {
332
          cfd_->table_cache()->EraseHandle(f->fd, f->table_reader_handle);
333 334
          f->table_reader_handle = nullptr;
        }
I
Igor Canadi 已提交
335
        vset_->obsolete_files_.push_back(f);
J
jorlow@chromium.org 已提交
336 337 338 339 340
      }
    }
  }
}

341
int FindFile(const InternalKeyComparator& icmp,
342
             const LevelFilesBrief& file_level,
343
             const Slice& key) {
344 345
  return FindFileInRange(icmp, file_level, key, 0,
                         static_cast<uint32_t>(file_level.num_files));
346 347
}

348
void DoGenerateLevelFilesBrief(LevelFilesBrief* file_level,
F
Feng Zhu 已提交
349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
        const std::vector<FileMetaData*>& files,
        Arena* arena) {
  assert(file_level);
  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 已提交
377
static bool AfterFile(const Comparator* ucmp,
378
                      const Slice* user_key, const FdWithKeyRange* f) {
A
Abhishek Kona 已提交
379 380
  // nullptr user_key occurs before all keys and is therefore never after *f
  return (user_key != nullptr &&
381
          ucmp->Compare(*user_key, ExtractUserKey(f->largest_key)) > 0);
G
Gabor Cselle 已提交
382 383 384
}

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

391 392
bool SomeFileOverlapsRange(
    const InternalKeyComparator& icmp,
G
Gabor Cselle 已提交
393
    bool disjoint_sorted_files,
394
    const LevelFilesBrief& file_level,
G
Gabor Cselle 已提交
395 396 397 398 399
    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
400 401
    for (size_t i = 0; i < file_level.num_files; i++) {
      const FdWithKeyRange* f = &(file_level.files[i]);
G
Gabor Cselle 已提交
402 403 404 405 406 407 408 409 410 411 412 413
      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 已提交
414
  if (smallest_user_key != nullptr) {
G
Gabor Cselle 已提交
415
    // Find the earliest possible internal key for smallest_user_key
416 417
    InternalKey small;
    small.SetMaxPossibleForUserKey(*smallest_user_key);
418
    index = FindFile(icmp, file_level, small.Encode());
G
Gabor Cselle 已提交
419 420
  }

421
  if (index >= file_level.num_files) {
G
Gabor Cselle 已提交
422 423 424 425
    // beginning of range is after all files, so no overlap.
    return false;
  }

426
  return !BeforeFile(ucmp, largest_user_key, &file_level.files[index]);
427 428
}

429 430
namespace {

J
jorlow@chromium.org 已提交
431 432 433
// 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 已提交
434 435
// 16-byte value containing the file number and file size, both
// encoded using EncodeFixed64.
S
sdong 已提交
436
class LevelFileNumIterator : public InternalIterator {
J
jorlow@chromium.org 已提交
437
 public:
438
  LevelFileNumIterator(const InternalKeyComparator& icmp,
439
                       const LevelFilesBrief* flevel)
440
      : icmp_(icmp),
F
Feng Zhu 已提交
441
        flevel_(flevel),
442
        index_(static_cast<uint32_t>(flevel->num_files)),
443
        current_value_(0, 0, 0) {  // Marks as invalid
J
jorlow@chromium.org 已提交
444
  }
I
Igor Sugak 已提交
445 446
  virtual bool Valid() const override { return index_ < flevel_->num_files; }
  virtual void Seek(const Slice& target) override {
F
Feng Zhu 已提交
447
    index_ = FindFile(icmp_, *flevel_, target);
J
jorlow@chromium.org 已提交
448
  }
A
Aaron Gao 已提交
449 450 451 452
  virtual void SeekForPrev(const Slice& target) override {
    SeekForPrevImpl(target, &icmp_);
  }

I
Igor Sugak 已提交
453 454
  virtual void SeekToFirst() override { index_ = 0; }
  virtual void SeekToLast() override {
455 456 457
    index_ = (flevel_->num_files == 0)
                 ? 0
                 : static_cast<uint32_t>(flevel_->num_files) - 1;
J
jorlow@chromium.org 已提交
458
  }
I
Igor Sugak 已提交
459
  virtual void Next() override {
J
jorlow@chromium.org 已提交
460 461 462
    assert(Valid());
    index_++;
  }
I
Igor Sugak 已提交
463
  virtual void Prev() override {
J
jorlow@chromium.org 已提交
464 465
    assert(Valid());
    if (index_ == 0) {
466
      index_ = static_cast<uint32_t>(flevel_->num_files);  // Marks as invalid
J
jorlow@chromium.org 已提交
467 468 469 470
    } else {
      index_--;
    }
  }
I
Igor Sugak 已提交
471
  Slice key() const override {
J
jorlow@chromium.org 已提交
472
    assert(Valid());
F
Feng Zhu 已提交
473
    return flevel_->files[index_].largest_key;
J
jorlow@chromium.org 已提交
474
  }
I
Igor Sugak 已提交
475
  Slice value() const override {
J
jorlow@chromium.org 已提交
476
    assert(Valid());
F
Feng Zhu 已提交
477 478 479

    auto file_meta = flevel_->files[index_];
    current_value_ = file_meta.fd;
480
    return Slice(reinterpret_cast<const char*>(&current_value_),
481
                 sizeof(FileDescriptor));
J
jorlow@chromium.org 已提交
482
  }
I
Igor Sugak 已提交
483 484
  virtual Status status() const override { return Status::OK(); }

J
jorlow@chromium.org 已提交
485 486
 private:
  const InternalKeyComparator icmp_;
487
  const LevelFilesBrief* flevel_;
D
dgrogan@chromium.org 已提交
488
  uint32_t index_;
489
  mutable FileDescriptor current_value_;
J
jorlow@chromium.org 已提交
490 491
};

492
class LevelFileIteratorState : public TwoLevelIteratorState {
L
Lei Jin 已提交
493
 public:
494
  // @param skip_filters Disables loading/accessing the filter block
L
Lei Jin 已提交
495
  LevelFileIteratorState(TableCache* table_cache,
496 497 498 499
                         const ReadOptions& read_options,
                         const EnvOptions& env_options,
                         const InternalKeyComparator& icomparator,
                         HistogramImpl* file_read_hist, bool for_compaction,
500 501
                         bool prefix_enabled, bool skip_filters, int level,
                         RangeDelAggregator* range_del_agg)
502 503 504 505 506 507
      : TwoLevelIteratorState(prefix_enabled),
        table_cache_(table_cache),
        read_options_(read_options),
        env_options_(env_options),
        icomparator_(icomparator),
        file_read_hist_(file_read_hist),
508
        for_compaction_(for_compaction),
509
        skip_filters_(skip_filters),
510 511
        level_(level),
        range_del_agg_(range_del_agg) {}
L
Lei Jin 已提交
512

S
sdong 已提交
513
  InternalIterator* NewSecondaryIterator(const Slice& meta_handle) override {
514
    if (meta_handle.size() != sizeof(FileDescriptor)) {
S
sdong 已提交
515
      return NewErrorInternalIterator(
L
Lei Jin 已提交
516 517
          Status::Corruption("FileReader invoked with unexpected value"));
    }
A
Andrew Kryczka 已提交
518 519 520
    const FileDescriptor* fd =
        reinterpret_cast<const FileDescriptor*>(meta_handle.data());
    return table_cache_->NewIterator(
521
        read_options_, env_options_, icomparator_, *fd, range_del_agg_,
A
Andrew Kryczka 已提交
522
        nullptr /* don't need reference to table */, file_read_hist_,
523
        for_compaction_, nullptr /* arena */, skip_filters_, level_);
J
jorlow@chromium.org 已提交
524 525
  }

L
Lei Jin 已提交
526 527 528 529 530 531 532 533 534
  bool PrefixMayMatch(const Slice& internal_key) override {
    return true;
  }

 private:
  TableCache* table_cache_;
  const ReadOptions read_options_;
  const EnvOptions& env_options_;
  const InternalKeyComparator& icomparator_;
535
  HistogramImpl* file_read_hist_;
L
Lei Jin 已提交
536
  bool for_compaction_;
537
  bool skip_filters_;
538
  int level_;
539
  RangeDelAggregator* range_del_agg_;
L
Lei Jin 已提交
540
};
T
Tyler Harter 已提交
541

S
sdong 已提交
542 543
// A wrapper of version builder which references the current version in
// constructor and unref it in the destructor.
544
// Both of the constructor and destructor need to be called inside DB Mutex.
S
sdong 已提交
545 546 547
class BaseReferencedVersionBuilder {
 public:
  explicit BaseReferencedVersionBuilder(ColumnFamilyData* cfd)
548
      : version_builder_(new VersionBuilder(
549
            cfd->current()->version_set()->env_options(), cfd->table_cache(),
550
            cfd->current()->storage_info(), cfd->ioptions()->info_log)),
S
sdong 已提交
551 552 553
        version_(cfd->current()) {
    version_->Ref();
  }
554 555 556 557
  ~BaseReferencedVersionBuilder() {
    delete version_builder_;
    version_->Unref();
  }
558
  VersionBuilder* version_builder() { return version_builder_; }
S
sdong 已提交
559 560

 private:
561
  VersionBuilder* version_builder_;
S
sdong 已提交
562 563
  Version* version_;
};
564 565
}  // anonymous namespace

566 567
Status Version::GetTableProperties(std::shared_ptr<const TableProperties>* tp,
                                   const FileMetaData* file_meta,
568
                                   const std::string* fname) const {
I
Igor Canadi 已提交
569
  auto table_cache = cfd_->table_cache();
570
  auto ioptions = cfd_->ioptions();
571
  Status s = table_cache->GetTableProperties(
572
      vset_->env_options_, cfd_->internal_comparator(), file_meta->fd,
573 574 575 576 577 578 579 580 581 582 583 584 585 586 587
      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) {
588
    s = ioptions->env->NewRandomAccessFile(
589
        *fname, &file, vset_->env_options_);
590
  } else {
591
    s = ioptions->env->NewRandomAccessFile(
592
        TableFileName(vset_->db_options_->db_paths, file_meta->fd.GetNumber(),
593
                      file_meta->fd.GetPathId()),
594
        &file, vset_->env_options_);
595 596 597 598 599 600 601 602
  }
  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.
603 604
  std::unique_ptr<RandomAccessFileReader> file_reader(
      new RandomAccessFileReader(std::move(file)));
605
  s = ReadTableProperties(
606
      file_reader.get(), file_meta->fd.GetFileSize(),
607
      Footer::kInvalidTableMagicNumber /* table's magic number */, *ioptions, &raw_table_properties);
608 609 610
  if (!s.ok()) {
    return s;
  }
611
  RecordTick(ioptions->statistics, NUMBER_DIRECT_LOAD_TABLE_PROPERTIES);
612 613 614 615 616 617

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

Status Version::GetPropertiesOfAllTables(TablePropertiesCollection* props) {
618
  Status s;
S
sdong 已提交
619
  for (int level = 0; level < storage_info_.num_levels_; level++) {
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642
    s = GetPropertiesOfAllTables(props, level);
    if (!s.ok()) {
      return s;
    }
  }

  return Status::OK();
}

Status Version::GetPropertiesOfAllTables(TablePropertiesCollection* props,
                                         int level) {
  for (const auto& file_meta : storage_info_.files_[level]) {
    auto fname =
        TableFileName(vset_->db_options_->db_paths, file_meta->fd.GetNumber(),
                      file_meta->fd.GetPathId());
    // 1. If the table is already present in table cache, load table
    // properties from there.
    std::shared_ptr<const TableProperties> table_properties;
    Status s = GetTableProperties(&table_properties, file_meta, &fname);
    if (s.ok()) {
      props->insert({fname, table_properties});
    } else {
      return s;
643 644 645 646 647 648
    }
  }

  return Status::OK();
}

649
Status Version::GetPropertiesOfTablesInRange(
650
    const Range* range, std::size_t n, TablePropertiesCollection* props) const {
651
  for (int level = 0; level < storage_info_.num_non_empty_levels(); level++) {
652
    for (decltype(n) i = 0; i < n; i++) {
653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
      // Convert user_key into a corresponding internal key.
      InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
      InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
      std::vector<FileMetaData*> files;
      storage_info_.GetOverlappingInputs(level, &k1, &k2, &files, -1, nullptr,
                                         false);
      for (const auto& file_meta : files) {
        auto fname =
            TableFileName(vset_->db_options_->db_paths,
                          file_meta->fd.GetNumber(), file_meta->fd.GetPathId());
        if (props->count(fname) == 0) {
          // 1. If the table is already present in table cache, load table
          // properties from there.
          std::shared_ptr<const TableProperties> table_properties;
          Status s = GetTableProperties(&table_properties, file_meta, &fname);
          if (s.ok()) {
            props->insert({fname, table_properties});
          } else {
            return s;
          }
        }
      }
    }
  }

  return Status::OK();
}

681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701
Status Version::GetAggregatedTableProperties(
    std::shared_ptr<const TableProperties>* tp, int level) {
  TablePropertiesCollection props;
  Status s;
  if (level < 0) {
    s = GetPropertiesOfAllTables(&props);
  } else {
    s = GetPropertiesOfAllTables(&props, level);
  }
  if (!s.ok()) {
    return s;
  }

  auto* new_tp = new TableProperties();
  for (const auto& item : props) {
    new_tp->Add(*item.second);
  }
  tp->reset(new_tp);
  return Status::OK();
}

702 703
size_t Version::GetMemoryUsageByTableReaders() {
  size_t total_usage = 0;
S
sdong 已提交
704
  for (auto& file_level : storage_info_.level_files_brief_) {
705 706
    for (size_t i = 0; i < file_level.num_files; i++) {
      total_usage += cfd_->table_cache()->GetMemoryUsageByTableReader(
707
          vset_->env_options_, cfd_->internal_comparator(),
708 709 710 711 712 713
          file_level.files[i].fd);
    }
  }
  return total_usage;
}

714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756
void Version::GetColumnFamilyMetaData(ColumnFamilyMetaData* cf_meta) {
  assert(cf_meta);
  assert(cfd_);

  cf_meta->name = cfd_->GetName();
  cf_meta->size = 0;
  cf_meta->file_count = 0;
  cf_meta->levels.clear();

  auto* ioptions = cfd_->ioptions();
  auto* vstorage = storage_info();

  for (int level = 0; level < cfd_->NumberLevels(); level++) {
    uint64_t level_size = 0;
    cf_meta->file_count += vstorage->LevelFiles(level).size();
    std::vector<SstFileMetaData> files;
    for (const auto& file : vstorage->LevelFiles(level)) {
      uint32_t path_id = file->fd.GetPathId();
      std::string file_path;
      if (path_id < ioptions->db_paths.size()) {
        file_path = ioptions->db_paths[path_id].path;
      } else {
        assert(!ioptions->db_paths.empty());
        file_path = ioptions->db_paths.back().path;
      }
      files.emplace_back(
          MakeTableFileName("", file->fd.GetNumber()),
          file_path,
          file->fd.GetFileSize(),
          file->smallest_seqno,
          file->largest_seqno,
          file->smallest.user_key().ToString(),
          file->largest.user_key().ToString(),
          file->being_compacted);
      level_size += file->fd.GetFileSize();
    }
    cf_meta->levels.emplace_back(
        level, level_size, std::move(files));
    cf_meta->size += level_size;
  }
}


S
sdong 已提交
757
uint64_t VersionStorageInfo::GetEstimatedActiveKeys() const {
758 759
  // Estimation will be inaccurate when:
  // (1) there exist merge keys
S
sdong 已提交
760 761
  // (2) keys are directly overwritten
  // (3) deletion on non-existing keys
762
  // (4) low number of samples
763
  if (current_num_samples_ == 0) {
764 765 766
    return 0;
  }

767
  if (current_num_non_deletions_ <= current_num_deletions_) {
768 769 770
    return 0;
  }

771
  uint64_t est = current_num_non_deletions_ - current_num_deletions_;
772

Y
Yueh-Hsuan Chiang 已提交
773 774 775 776 777
  uint64_t file_count = 0;
  for (int level = 0; level < num_levels_; ++level) {
    file_count += files_[level].size();
  }

778
  if (current_num_samples_ < file_count) {
779
    // casting to avoid overflowing
780
    return
D
Dmitri Smirnov 已提交
781
      static_cast<uint64_t>(
D
Dmitri Smirnov 已提交
782
        (est * static_cast<double>(file_count) / current_num_samples_)
D
Dmitri Smirnov 已提交
783
      );
784
  } else {
785
    return est;
786
  }
S
sdong 已提交
787 788
}

789 790 791 792 793 794 795 796 797 798 799 800 801 802 803
double VersionStorageInfo::GetEstimatedCompressionRatioAtLevel(
    int level) const {
  assert(level < num_levels_);
  uint64_t sum_file_size_bytes = 0;
  uint64_t sum_data_size_bytes = 0;
  for (auto* file_meta : files_[level]) {
    sum_file_size_bytes += file_meta->fd.GetFileSize();
    sum_data_size_bytes += file_meta->raw_key_size + file_meta->raw_value_size;
  }
  if (sum_file_size_bytes == 0) {
    return -1.0;
  }
  return static_cast<double>(sum_data_size_bytes) / sum_file_size_bytes;
}

804 805
void Version::AddIterators(const ReadOptions& read_options,
                           const EnvOptions& soptions,
A
Andrew Kryczka 已提交
806 807
                           MergeIteratorBuilder* merge_iter_builder,
                           RangeDelAggregator* range_del_agg) {
S
sdong 已提交
808
  assert(storage_info_.finalized_);
809

810
  for (int level = 0; level < storage_info_.num_non_empty_levels(); level++) {
A
Andrew Kryczka 已提交
811 812
    AddIteratorsForLevel(read_options, soptions, merge_iter_builder, level,
                         range_del_agg);
813 814 815 816 817 818
  }
}

void Version::AddIteratorsForLevel(const ReadOptions& read_options,
                                   const EnvOptions& soptions,
                                   MergeIteratorBuilder* merge_iter_builder,
A
Andrew Kryczka 已提交
819 820
                                   int level,
                                   RangeDelAggregator* range_del_agg) {
821 822 823 824 825 826
  assert(storage_info_.finalized_);
  if (level >= storage_info_.num_non_empty_levels()) {
    // This is an empty level
    return;
  } else if (storage_info_.LevelFilesBrief(level).num_files == 0) {
    // No files in this level
S
sdong 已提交
827 828 829
    return;
  }

830
  auto* arena = merge_iter_builder->GetArena();
831 832 833 834 835
  if (level == 0) {
    // Merge all level zero files together since they may overlap
    for (size_t i = 0; i < storage_info_.LevelFilesBrief(0).num_files; i++) {
      const auto& file = storage_info_.LevelFilesBrief(0).files[i];
      merge_iter_builder->AddIterator(cfd_->table_cache()->NewIterator(
836 837 838
          read_options, soptions, cfd_->internal_comparator(), file.fd,
          range_del_agg, nullptr, cfd_->internal_stats()->GetFileReadHist(0),
          false, arena, false /* skip_filters */, 0 /* level */));
839
    }
840 841 842 843 844
  } else {
    // For levels > 0, we can use a concatenating iterator that sequentially
    // walks through the non-overlapping files in the level, opening them
    // lazily.
    auto* mem = arena->AllocateAligned(sizeof(LevelFileIteratorState));
A
Andrew Kryczka 已提交
845 846 847 848 849 850 851
    auto* state = new (mem)
        LevelFileIteratorState(cfd_->table_cache(), read_options, soptions,
                               cfd_->internal_comparator(),
                               cfd_->internal_stats()->GetFileReadHist(level),
                               false /* for_compaction */,
                               cfd_->ioptions()->prefix_extractor != nullptr,
                               IsFilterSkipped(level), level, range_del_agg);
852 853 854 855 856
    mem = arena->AllocateAligned(sizeof(LevelFileNumIterator));
    auto* first_level_iter = new (mem) LevelFileNumIterator(
        cfd_->internal_comparator(), &storage_info_.LevelFilesBrief(level));
    merge_iter_builder->AddIterator(
        NewTwoLevelIterator(state, first_level_iter, arena, false));
857 858 859
  }
}

S
sdong 已提交
860 861
VersionStorageInfo::VersionStorageInfo(
    const InternalKeyComparator* internal_comparator,
I
Igor Canadi 已提交
862
    const Comparator* user_comparator, int levels,
863 864
    CompactionStyle compaction_style, VersionStorageInfo* ref_vstorage,
    bool _force_consistency_checks)
S
sdong 已提交
865 866
    : internal_comparator_(internal_comparator),
      user_comparator_(user_comparator),
867
      // cfd is nullptr if Version is dummy
I
Igor Canadi 已提交
868
      num_levels_(levels),
S
sdong 已提交
869
      num_non_empty_levels_(0),
S
sdong 已提交
870 871
      file_indexer_(user_comparator),
      compaction_style_(compaction_style),
872
      files_(new std::vector<FileMetaData*>[num_levels_]),
S
sdong 已提交
873
      base_level_(num_levels_ == 1 ? -1 : 1),
874
      files_by_compaction_pri_(num_levels_),
875
      level0_non_overlapping_(false),
876 877 878
      next_file_to_compact_by_size_(num_levels_),
      compaction_score_(num_levels_),
      compaction_level_(num_levels_),
S
sdong 已提交
879
      l0_delay_trigger_count_(0),
880 881 882 883 884
      accumulated_file_size_(0),
      accumulated_raw_key_size_(0),
      accumulated_raw_value_size_(0),
      accumulated_num_non_deletions_(0),
      accumulated_num_deletions_(0),
885 886 887
      current_num_non_deletions_(0),
      current_num_deletions_(0),
      current_num_samples_(0),
888
      estimated_compaction_needed_bytes_(0),
889 890
      finalized_(false),
      force_consistency_checks_(_force_consistency_checks) {
S
sdong 已提交
891 892 893 894 895 896 897
  if (ref_vstorage != nullptr) {
    accumulated_file_size_ = ref_vstorage->accumulated_file_size_;
    accumulated_raw_key_size_ = ref_vstorage->accumulated_raw_key_size_;
    accumulated_raw_value_size_ = ref_vstorage->accumulated_raw_value_size_;
    accumulated_num_non_deletions_ =
        ref_vstorage->accumulated_num_non_deletions_;
    accumulated_num_deletions_ = ref_vstorage->accumulated_num_deletions_;
898 899 900
    current_num_non_deletions_ = ref_vstorage->current_num_non_deletions_;
    current_num_deletions_ = ref_vstorage->current_num_deletions_;
    current_num_samples_ = ref_vstorage->current_num_samples_;
901
  }
902
}
903

I
Igor Canadi 已提交
904
Version::Version(ColumnFamilyData* column_family_data, VersionSet* vset,
S
sdong 已提交
905
                 uint64_t version_number)
906 907
    : env_(vset->env_),
      cfd_(column_family_data),
I
Igor Canadi 已提交
908 909 910 911 912 913
      info_log_((cfd_ == nullptr) ? nullptr : cfd_->ioptions()->info_log),
      db_statistics_((cfd_ == nullptr) ? nullptr
                                       : cfd_->ioptions()->statistics),
      table_cache_((cfd_ == nullptr) ? nullptr : cfd_->table_cache()),
      merge_operator_((cfd_ == nullptr) ? nullptr
                                        : cfd_->ioptions()->merge_operator),
914 915 916 917 918 919 920 921 922 923
      storage_info_(
          (cfd_ == nullptr) ? nullptr : &cfd_->internal_comparator(),
          (cfd_ == nullptr) ? nullptr : cfd_->user_comparator(),
          cfd_ == nullptr ? 0 : cfd_->NumberLevels(),
          cfd_ == nullptr ? kCompactionStyleLevel
                          : cfd_->ioptions()->compaction_style,
          (cfd_ == nullptr || cfd_->current() == nullptr)
              ? nullptr
              : cfd_->current()->storage_info(),
          cfd_ == nullptr ? false : cfd_->ioptions()->force_consistency_checks),
S
sdong 已提交
924 925 926 927 928 929
      vset_(vset),
      next_(this),
      prev_(this),
      refs_(0),
      version_number_(version_number) {}

930
void Version::Get(const ReadOptions& read_options, const LookupKey& k,
M
Maysam Yabandeh 已提交
931
                  PinnableSlice* pSlice, Status* status,
A
Andrew Kryczka 已提交
932 933
                  MergeContext* merge_context,
                  RangeDelAggregator* range_del_agg, bool* value_found,
934
                  bool* key_exists, SequenceNumber* seq) {
935 936
  Slice ikey = k.internal_key();
  Slice user_key = k.user_key();
937 938

  assert(status->ok() || status->IsMergeInProgress());
939

940 941 942 943 944
  if (key_exists != nullptr) {
    // will falsify below if not found
    *key_exists = true;
  }

945
  PinnedIteratorsManager pinned_iters_mgr;
S
sdong 已提交
946
  GetContext get_context(
S
sdong 已提交
947
      user_comparator(), merge_operator_, info_log_, db_statistics_,
S
sdong 已提交
948
      status->ok() ? GetContext::kNotFound : GetContext::kMerge, user_key,
M
Maysam Yabandeh 已提交
949
      pSlice, value_found, merge_context, range_del_agg, this->env_, seq,
950 951 952 953 954 955
      merge_operator_ ? &pinned_iters_mgr : nullptr);

  // Pin blocks that we read to hold merge operands
  if (merge_operator_) {
    pinned_iters_mgr.StartPinning();
  }
956

S
sdong 已提交
957 958 959 960
  FilePicker fp(
      storage_info_.files_, user_key, ikey, &storage_info_.level_files_brief_,
      storage_info_.num_non_empty_levels_, &storage_info_.file_indexer_,
      user_comparator(), internal_comparator());
961 962
  FdWithKeyRange* f = fp.GetNextFile();
  while (f != nullptr) {
963 964
    *status = table_cache_->Get(
        read_options, *internal_comparator(), f->fd, ikey, &get_context,
965
        cfd_->internal_stats()->GetFileReadHist(fp.GetHitFileLevel()),
966
        IsFilterSkipped(static_cast<int>(fp.GetHitFileLevel()),
967 968
                        fp.IsHitFileLastInLevel()),
        fp.GetCurrentLevel());
969 970 971
    // TODO: examine the behavior for corrupted key
    if (!status->ok()) {
      return;
972
    }
973

974 975 976 977 978
    switch (get_context.State()) {
      case GetContext::kNotFound:
        // Keep searching in other files
        break;
      case GetContext::kFound:
979 980 981 982 983 984 985
        if (fp.GetHitFileLevel() == 0) {
          RecordTick(db_statistics_, GET_HIT_L0);
        } else if (fp.GetHitFileLevel() == 1) {
          RecordTick(db_statistics_, GET_HIT_L1);
        } else if (fp.GetHitFileLevel() >= 2) {
          RecordTick(db_statistics_, GET_HIT_L2_AND_UP);
        }
986
        return;
987 988 989
      case GetContext::kDeleted:
        // Use empty error message for speed
        *status = Status::NotFound();
990
        return;
991
      case GetContext::kCorrupt:
992 993
        *status = Status::Corruption("corrupted key for ", user_key);
        return;
994
      case GetContext::kMerge:
995
        break;
996
    }
997
    f = fp.GetNextFile();
998 999
  }

1000
  if (GetContext::kMerge == get_context.State()) {
1001 1002 1003 1004 1005
    if (!merge_operator_) {
      *status =  Status::InvalidArgument(
          "merge_operator is not properly initialized.");
      return;
    }
1006 1007
    // merge_operands are in saver and we hit the beginning of the key history
    // do a final merge of nullptr and operands;
M
Maysam Yabandeh 已提交
1008 1009 1010 1011 1012 1013 1014
    std::string* str_value = pSlice != nullptr ? pSlice->GetSelf() : nullptr;
    *status = MergeHelper::TimedFullMerge(
        merge_operator_, user_key, nullptr, merge_context->GetOperands(),
        str_value, info_log_, db_statistics_, env_);
    if (LIKELY(pSlice != nullptr)) {
      pSlice->PinSelf();
    }
1015
  } else {
1016 1017 1018
    if (key_exists != nullptr) {
      *key_exists = false;
    }
1019
    *status = Status::NotFound(); // Use an empty error message for speed
1020
  }
1021 1022
}

1023
bool Version::IsFilterSkipped(int level, bool is_file_last_in_level) {
1024 1025
  // Reaching the bottom level implies misses at all upper levels, so we'll
  // skip checking the filters when we predict a hit.
1026 1027
  return cfd_->ioptions()->optimize_filters_for_hits &&
         (level > 0 || is_file_last_in_level) &&
1028 1029 1030
         level == storage_info_.num_non_empty_levels() - 1;
}

S
sdong 已提交
1031
void VersionStorageInfo::GenerateLevelFilesBrief() {
1032
  level_files_brief_.resize(num_non_empty_levels_);
1033
  for (int level = 0; level < num_non_empty_levels_; level++) {
1034 1035
    DoGenerateLevelFilesBrief(
        &level_files_brief_[level], files_[level], &arena_);
1036 1037 1038
  }
}

1039 1040 1041 1042
void Version::PrepareApply(
    const MutableCFOptions& mutable_cf_options,
    bool update_stats) {
  UpdateAccumulatedStats(update_stats);
S
sdong 已提交
1043
  storage_info_.UpdateNumNonEmptyLevels();
1044
  storage_info_.CalculateBaseBytes(*cfd_->ioptions(), mutable_cf_options);
Y
Yi Wu 已提交
1045
  storage_info_.UpdateFilesByCompactionPri(cfd_->ioptions()->compaction_pri);
S
sdong 已提交
1046 1047
  storage_info_.GenerateFileIndexer();
  storage_info_.GenerateLevelFilesBrief();
1048
  storage_info_.GenerateLevel0NonOverlapping();
1049 1050
}

1051
bool Version::MaybeInitializeFileMetaData(FileMetaData* file_meta) {
1052 1053
  if (file_meta->init_stats_from_file ||
      file_meta->compensated_file_size > 0) {
1054 1055 1056 1057
    return false;
  }
  std::shared_ptr<const TableProperties> tp;
  Status s = GetTableProperties(&tp, file_meta);
1058
  file_meta->init_stats_from_file = true;
1059
  if (!s.ok()) {
1060
    Log(InfoLogLevel::ERROR_LEVEL, vset_->db_options_->info_log,
1061 1062
        "Unable to load table properties for file %" PRIu64 " --- %s\n",
        file_meta->fd.GetNumber(), s.ToString().c_str());
1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073
    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;
}

S
sdong 已提交
1074
void VersionStorageInfo::UpdateAccumulatedStats(FileMetaData* file_meta) {
1075 1076 1077 1078 1079 1080 1081
  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;
1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095

  current_num_non_deletions_ +=
      file_meta->num_entries - file_meta->num_deletions;
  current_num_deletions_ += file_meta->num_deletions;
  current_num_samples_++;
}

void VersionStorageInfo::RemoveCurrentStats(FileMetaData* file_meta) {
  if (file_meta->init_stats_from_file) {
    current_num_non_deletions_ -=
        file_meta->num_entries - file_meta->num_deletions;
    current_num_deletions_ -= file_meta->num_deletions;
    current_num_samples_--;
  }
1096 1097
}

1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119
void Version::UpdateAccumulatedStats(bool update_stats) {
  if (update_stats) {
    // maximum number of table properties loaded from files.
    const int kMaxInitCount = 20;
    int init_count = 0;
    // 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 < storage_info_.num_levels_ && init_count < kMaxInitCount;
         ++level) {
      for (auto* file_meta : storage_info_.files_[level]) {
        if (MaybeInitializeFileMetaData(file_meta)) {
          // each FileMeta will be initialized only once.
          storage_info_.UpdateAccumulatedStats(file_meta);
1120 1121 1122 1123 1124 1125
          // when option "max_open_files" is -1, all the file metadata has
          // already been read, so MaybeInitializeFileMetaData() won't incur
          // any I/O cost.
          if (vset_->db_options_->max_open_files == -1) {
            continue;
          }
1126 1127 1128
          if (++init_count >= kMaxInitCount) {
            break;
          }
1129 1130 1131
        }
      }
    }
1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142
    // 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 = storage_info_.num_levels_ - 1;
         storage_info_.accumulated_raw_value_size_ == 0 && level >= 0;
         --level) {
      for (int i = static_cast<int>(storage_info_.files_[level].size()) - 1;
           storage_info_.accumulated_raw_value_size_ == 0 && i >= 0; --i) {
        if (MaybeInitializeFileMetaData(storage_info_.files_[level][i])) {
          storage_info_.UpdateAccumulatedStats(storage_info_.files_[level][i]);
        }
1143 1144 1145 1146
      }
    }
  }

S
sdong 已提交
1147
  storage_info_.ComputeCompensatedSizes();
S
sdong 已提交
1148 1149 1150 1151
}

void VersionStorageInfo::ComputeCompensatedSizes() {
  static const int kDeletionWeightOnCompaction = 2;
1152 1153 1154 1155 1156
  uint64_t average_value_size = GetAverageValueSize();

  // compute the compensated size
  for (int level = 0; level < num_levels_; level++) {
    for (auto* file_meta : files_[level]) {
1157
      // Here we only compute compensated_file_size for those file_meta
I
Igor Canadi 已提交
1158 1159 1160
      // which compensated_file_size is uninitialized (== 0). This is true only
      // for files that have been created right now and no other thread has
      // access to them. That's why we can safely mutate compensated_file_size.
1161
      if (file_meta->compensated_file_size == 0) {
1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172
        file_meta->compensated_file_size = file_meta->fd.GetFileSize();
        // Here we only boost the size of deletion entries of a file only
        // when the number of deletion entries is greater than the number of
        // non-deletion entries in the file.  The motivation here is that in
        // a stable workload, the number of deletion entries should be roughly
        // equal to the number of non-deletion entries.  If we compensate the
        // size of deletion entries in a stable workload, the deletion
        // compensation logic might introduce unwanted effet which changes the
        // shape of LSM tree.
        if (file_meta->num_deletions * 2 >= file_meta->num_entries) {
          file_meta->compensated_file_size +=
1173 1174
              (file_meta->num_deletions * 2 - file_meta->num_entries) *
              average_value_size * kDeletionWeightOnCompaction;
1175
        }
1176
      }
1177 1178 1179 1180
    }
  }
}

S
sdong 已提交
1181 1182
int VersionStorageInfo::MaxInputLevel() const {
  if (compaction_style_ == kCompactionStyleLevel) {
1183
    return num_levels() - 2;
S
sdong 已提交
1184 1185 1186 1187
  }
  return 0;
}

1188 1189 1190 1191
void VersionStorageInfo::EstimateCompactionBytesNeeded(
    const MutableCFOptions& mutable_cf_options) {
  // Only implemented for level-based compaction
  if (compaction_style_ != kCompactionStyleLevel) {
1192
    estimated_compaction_needed_bytes_ = 0;
1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205
    return;
  }

  // Start from Level 0, if level 0 qualifies compaction to level 1,
  // we estimate the size of compaction.
  // Then we move on to the next level and see whether it qualifies compaction
  // to the next level. The size of the level is estimated as the actual size
  // on the level plus the input bytes from the previous level if there is any.
  // If it exceeds, take the exceeded bytes as compaction input and add the size
  // of the compaction size to tatal size.
  // We keep doing it to Level 2, 3, etc, until the last level and return the
  // accumulated bytes.

I
Igor Canadi 已提交
1206
  uint64_t bytes_compact_to_next_level = 0;
1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220
  // Level 0
  bool level0_compact_triggered = false;
  if (static_cast<int>(files_[0].size()) >
      mutable_cf_options.level0_file_num_compaction_trigger) {
    level0_compact_triggered = true;
    for (auto* f : files_[0]) {
      bytes_compact_to_next_level += f->fd.GetFileSize();
    }
    estimated_compaction_needed_bytes_ = bytes_compact_to_next_level;
  } else {
    estimated_compaction_needed_bytes_ = 0;
  }

  // Level 1 and up.
1221
  uint64_t bytes_next_level = 0;
1222
  for (int level = base_level(); level <= MaxInputLevel(); level++) {
I
Igor Canadi 已提交
1223
    uint64_t level_size = 0;
1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237
    if (bytes_next_level > 0) {
#ifndef NDEBUG
      uint64_t level_size2 = 0;
      for (auto* f : files_[level]) {
        level_size2 += f->fd.GetFileSize();
      }
      assert(level_size2 == bytes_next_level);
#endif
      level_size = bytes_next_level;
      bytes_next_level = 0;
    } else {
      for (auto* f : files_[level]) {
        level_size += f->fd.GetFileSize();
      }
1238 1239 1240 1241 1242 1243 1244 1245
    }
    if (level == base_level() && level0_compact_triggered) {
      // Add base level size to compaction if level0 compaction triggered.
      estimated_compaction_needed_bytes_ += level_size;
    }
    // Add size added by previous compaction
    level_size += bytes_compact_to_next_level;
    bytes_compact_to_next_level = 0;
I
Igor Canadi 已提交
1246
    uint64_t level_target = MaxBytesForLevel(level);
1247 1248
    if (level_size > level_target) {
      bytes_compact_to_next_level = level_size - level_target;
1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
      // Estimate the actual compaction fan-out ratio as size ratio between
      // the two levels.

      assert(bytes_next_level == 0);
      if (level + 1 < num_levels_) {
        for (auto* f : files_[level + 1]) {
          bytes_next_level += f->fd.GetFileSize();
        }
      }
      if (bytes_next_level > 0) {
        assert(level_size > 0);
        estimated_compaction_needed_bytes_ += static_cast<uint64_t>(
            static_cast<double>(bytes_compact_to_next_level) *
            (static_cast<double>(bytes_next_level) /
                 static_cast<double>(level_size) +
             1));
      }
1266 1267 1268 1269
    }
  }
}

S
sdong 已提交
1270
void VersionStorageInfo::ComputeCompactionScore(
Y
Yi Wu 已提交
1271
    const ImmutableCFOptions& immutable_cf_options,
1272
    const MutableCFOptions& mutable_cf_options) {
S
sdong 已提交
1273
  for (int level = 0; level <= MaxInputLevel(); level++) {
1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286
    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).
S
sdong 已提交
1287
      int num_sorted_runs = 0;
I
Igor Canadi 已提交
1288
      uint64_t total_size = 0;
1289 1290 1291
      for (auto* f : files_[level]) {
        if (!f->being_compacted) {
          total_size += f->compensated_file_size;
S
sdong 已提交
1292
          num_sorted_runs++;
1293 1294
        }
      }
S
sdong 已提交
1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
      if (compaction_style_ == kCompactionStyleUniversal) {
        // For universal compaction, we use level0 score to indicate
        // compaction score for the whole DB. Adding other levels as if
        // they are L0 files.
        for (int i = 1; i < num_levels(); i++) {
          if (!files_[i].empty() && !files_[i][0]->being_compacted) {
            num_sorted_runs++;
          }
        }
      }

S
sdong 已提交
1306
      if (compaction_style_ == kCompactionStyleFIFO) {
Y
Yi Wu 已提交
1307 1308 1309
        score =
            static_cast<double>(total_size) /
            immutable_cf_options.compaction_options_fifo.max_table_files_size;
1310
      } else {
S
sdong 已提交
1311
        score = static_cast<double>(num_sorted_runs) /
1312
                mutable_cf_options.level0_file_num_compaction_trigger;
1313 1314 1315
      }
    } else {
      // Compute the ratio of current size to size limit.
I
Igor Canadi 已提交
1316 1317
      uint64_t level_bytes_no_compacting = 0;
      for (auto f : files_[level]) {
1318
        if (!f->being_compacted) {
I
Igor Canadi 已提交
1319 1320 1321 1322
          level_bytes_no_compacting += f->compensated_file_size;
        }
      }
      score = static_cast<double>(level_bytes_no_compacting) /
1323
              MaxBytesForLevel(level);
1324 1325 1326 1327 1328 1329 1330
    }
    compaction_level_[level] = level;
    compaction_score_[level] = score;
  }

  // sort all the levels based on their score. Higher scores get listed
  // first. Use bubble sort because the number of entries are small.
1331 1332
  for (int i = 0; i < num_levels() - 2; i++) {
    for (int j = i + 1; j < num_levels() - 1; j++) {
1333 1334 1335 1336 1337 1338 1339 1340 1341 1342
      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;
      }
    }
  }
1343
  ComputeFilesMarkedForCompaction();
1344
  EstimateCompactionBytesNeeded(mutable_cf_options);
1345 1346 1347 1348
}

void VersionStorageInfo::ComputeFilesMarkedForCompaction() {
  files_marked_for_compaction_.clear();
1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361
  int last_qualify_level = 0;

  // Do not include files from the last level with data
  // If table properties collector suggests a file on the last level,
  // we should not move it to a new level.
  for (int level = num_levels() - 1; level >= 1; level--) {
    if (!files_[level].empty()) {
      last_qualify_level = level - 1;
      break;
    }
  }

  for (int level = 0; level <= last_qualify_level; level++) {
1362 1363 1364 1365 1366 1367
    for (auto* f : files_[level]) {
      if (!f->being_compacted && f->marked_for_compaction) {
        files_marked_for_compaction_.emplace_back(level, f);
      }
    }
  }
1368 1369 1370
}

namespace {
1371 1372 1373

// used to sort files by size
struct Fsize {
1374
  size_t index;
1375 1376 1377
  FileMetaData* file;
};

1378 1379
// Compator that is used to sort files based on their size
// In normal mode: descending size
1380
bool CompareCompensatedSizeDescending(const Fsize& first, const Fsize& second) {
1381 1382
  return (first.file->compensated_file_size >
      second.file->compensated_file_size);
1383
}
K
kailiu 已提交
1384
} // anonymous namespace
1385

1386
void VersionStorageInfo::AddFile(int level, FileMetaData* f, Logger* info_log) {
S
sdong 已提交
1387 1388
  auto* level_files = &files_[level];
  // Must not overlap
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406
#ifndef NDEBUG
  if (level > 0 && !level_files->empty() &&
      internal_comparator_->Compare(
          (*level_files)[level_files->size() - 1]->largest, f->smallest) >= 0) {
    auto* f2 = (*level_files)[level_files->size() - 1];
    if (info_log != nullptr) {
      Error(info_log, "Adding new file %" PRIu64
                      " range (%s, %s) to level %d but overlapping "
                      "with existing file %" PRIu64 " %s %s",
            f->fd.GetNumber(), f->smallest.DebugString(true).c_str(),
            f->largest.DebugString(true).c_str(), level, f2->fd.GetNumber(),
            f2->smallest.DebugString(true).c_str(),
            f2->largest.DebugString(true).c_str());
      LogFlush(info_log);
    }
    assert(false);
  }
#endif
S
sdong 已提交
1407 1408 1409 1410
  f->refs++;
  level_files->push_back(f);
}

1411 1412 1413 1414
// Version::PrepareApply() need to be called before calling the function, or
// following functions called:
// 1. UpdateNumNonEmptyLevels();
// 2. CalculateBaseBytes();
1415
// 3. UpdateFilesByCompactionPri();
1416 1417
// 4. GenerateFileIndexer();
// 5. GenerateLevelFilesBrief();
1418
// 6. GenerateLevel0NonOverlapping();
1419 1420 1421
void VersionStorageInfo::SetFinalized() {
  finalized_ = true;
#ifndef NDEBUG
S
sdong 已提交
1422 1423 1424 1425 1426 1427
  if (compaction_style_ != kCompactionStyleLevel) {
    // Not level based compaction.
    return;
  }
  assert(base_level_ < 0 || num_levels() == 1 ||
         (base_level_ >= 1 && base_level_ < num_levels()));
1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455
  // Verify all levels newer than base_level are empty except L0
  for (int level = 1; level < base_level(); level++) {
    assert(NumLevelBytes(level) == 0);
  }
  uint64_t max_bytes_prev_level = 0;
  for (int level = base_level(); level < num_levels() - 1; level++) {
    if (LevelFiles(level).size() == 0) {
      continue;
    }
    assert(MaxBytesForLevel(level) >= max_bytes_prev_level);
    max_bytes_prev_level = MaxBytesForLevel(level);
  }
  int num_empty_non_l0_level = 0;
  for (int level = 0; level < num_levels(); level++) {
    assert(LevelFiles(level).size() == 0 ||
           LevelFiles(level).size() == LevelFilesBrief(level).num_files);
    if (level > 0 && NumLevelBytes(level) > 0) {
      num_empty_non_l0_level++;
    }
    if (LevelFiles(level).size() > 0) {
      assert(level < num_non_empty_levels());
    }
  }
  assert(compaction_level_.size() > 0);
  assert(compaction_level_.size() == compaction_score_.size());
#endif
}

S
sdong 已提交
1456
void VersionStorageInfo::UpdateNumNonEmptyLevels() {
1457 1458 1459 1460 1461 1462 1463 1464 1465 1466
  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;
    }
  }
}

1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507
namespace {
// Sort `temp` based on ratio of overlapping size over file size
void SortFileByOverlappingRatio(
    const InternalKeyComparator& icmp, const std::vector<FileMetaData*>& files,
    const std::vector<FileMetaData*>& next_level_files,
    std::vector<Fsize>* temp) {
  std::unordered_map<uint64_t, uint64_t> file_to_order;
  auto next_level_it = next_level_files.begin();

  for (auto& file : files) {
    uint64_t overlapping_bytes = 0;
    // Skip files in next level that is smaller than current file
    while (next_level_it != next_level_files.end() &&
           icmp.Compare((*next_level_it)->largest, file->smallest) < 0) {
      next_level_it++;
    }

    while (next_level_it != next_level_files.end() &&
           icmp.Compare((*next_level_it)->smallest, file->largest) < 0) {
      overlapping_bytes += (*next_level_it)->fd.file_size;

      if (icmp.Compare((*next_level_it)->largest, file->largest) > 0) {
        // next level file cross large boundary of current file.
        break;
      }
      next_level_it++;
    }

    assert(file->fd.file_size != 0);
    file_to_order[file->fd.GetNumber()] =
        overlapping_bytes * 1024u / file->fd.file_size;
  }

  std::sort(temp->begin(), temp->end(),
            [&](const Fsize& f1, const Fsize& f2) -> bool {
              return file_to_order[f1.file->fd.GetNumber()] <
                     file_to_order[f2.file->fd.GetNumber()];
            });
}
}  // namespace

1508
void VersionStorageInfo::UpdateFilesByCompactionPri(
Y
Yi Wu 已提交
1509
    CompactionPri compaction_pri) {
S
sdong 已提交
1510 1511
  if (compaction_style_ == kCompactionStyleFIFO ||
      compaction_style_ == kCompactionStyleUniversal) {
I
Igor Canadi 已提交
1512 1513 1514
    // don't need this
    return;
  }
1515
  // No need to sort the highest level because it is never compacted.
1516
  for (int level = 0; level < num_levels() - 1; level++) {
1517
    const std::vector<FileMetaData*>& files = files_[level];
1518 1519
    auto& files_by_compaction_pri = files_by_compaction_pri_[level];
    assert(files_by_compaction_pri.size() == 0);
1520 1521 1522

    // populate a temp vector for sorting based on size
    std::vector<Fsize> temp(files.size());
1523
    for (size_t i = 0; i < files.size(); i++) {
1524 1525 1526 1527
      temp[i].index = i;
      temp[i].file = files[i];
    }

S
sdong 已提交
1528 1529
    // sort the top number_of_files_to_sort_ based on file size
    size_t num = VersionStorageInfo::kNumberFilesToSort;
1530 1531
    if (num > temp.size()) {
      num = temp.size();
1532
    }
Y
Yi Wu 已提交
1533
    switch (compaction_pri) {
1534
      case kByCompensatedSize:
1535 1536 1537
        std::partial_sort(temp.begin(), temp.begin() + num, temp.end(),
                          CompareCompensatedSizeDescending);
        break;
1538
      case kOldestLargestSeqFirst:
1539 1540 1541 1542 1543
        std::sort(temp.begin(), temp.end(),
                  [this](const Fsize& f1, const Fsize& f2) -> bool {
                    return f1.file->largest_seqno < f2.file->largest_seqno;
                  });
        break;
1544 1545 1546 1547 1548 1549
      case kOldestSmallestSeqFirst:
        std::sort(temp.begin(), temp.end(),
                  [this](const Fsize& f1, const Fsize& f2) -> bool {
                    return f1.file->smallest_seqno < f2.file->smallest_seqno;
                  });
        break;
1550 1551 1552 1553
      case kMinOverlappingRatio:
        SortFileByOverlappingRatio(*internal_comparator_, files_[level],
                                   files_[level + 1], &temp);
        break;
1554 1555 1556
      default:
        assert(false);
    }
1557 1558
    assert(temp.size() == files.size());

1559
    // initialize files_by_compaction_pri_
1560 1561
    for (size_t i = 0; i < temp.size(); i++) {
      files_by_compaction_pri.push_back(static_cast<int>(temp[i].index));
1562 1563
    }
    next_file_to_compact_by_size_[level] = 0;
1564
    assert(files_[level].size() == files_by_compaction_pri_[level].size());
1565 1566 1567
  }
}

1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
void VersionStorageInfo::GenerateLevel0NonOverlapping() {
  assert(!finalized_);
  level0_non_overlapping_ = true;
  if (level_files_brief_.size() == 0) {
    return;
  }

  // A copy of L0 files sorted by smallest key
  std::vector<FdWithKeyRange> level0_sorted_file(
      level_files_brief_[0].files,
      level_files_brief_[0].files + level_files_brief_[0].num_files);
1579 1580 1581 1582 1583
  std::sort(level0_sorted_file.begin(), level0_sorted_file.end(),
            [this](const FdWithKeyRange& f1, const FdWithKeyRange& f2) -> bool {
              return (internal_comparator_->Compare(f1.smallest_key,
                                                    f2.smallest_key) < 0);
            });
1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594

  for (size_t i = 1; i < level0_sorted_file.size(); ++i) {
    FdWithKeyRange& f = level0_sorted_file[i];
    FdWithKeyRange& prev = level0_sorted_file[i - 1];
    if (internal_comparator_->Compare(prev.largest_key, f.smallest_key) >= 0) {
      level0_non_overlapping_ = false;
      break;
    }
  }
}

J
jorlow@chromium.org 已提交
1595 1596 1597 1598
void Version::Ref() {
  ++refs_;
}

1599
bool Version::Unref() {
J
jorlow@chromium.org 已提交
1600 1601 1602
  assert(refs_ >= 1);
  --refs_;
  if (refs_ == 0) {
1603
    delete this;
1604
    return true;
J
jorlow@chromium.org 已提交
1605
  }
1606
  return false;
J
jorlow@chromium.org 已提交
1607 1608
}

S
sdong 已提交
1609 1610 1611
bool VersionStorageInfo::OverlapInLevel(int level,
                                        const Slice* smallest_user_key,
                                        const Slice* largest_user_key) {
1612 1613 1614 1615
  if (level >= num_non_empty_levels_) {
    // empty level, no overlap
    return false;
  }
S
sdong 已提交
1616
  return SomeFileOverlapsRange(*internal_comparator_, (level > 0),
1617
                               level_files_brief_[level], smallest_user_key,
1618
                               largest_user_key);
G
Gabor Cselle 已提交
1619 1620 1621
}

// Store in "*inputs" all files in "level" that overlap [begin,end]
A
Abhishek Kona 已提交
1622
// If hint_index is specified, then it points to a file in the
1623 1624
// overlapping range.
// The file_index returns a pointer to any file in an overlapping range.
S
sdong 已提交
1625 1626
void VersionStorageInfo::GetOverlappingInputs(
    int level, const InternalKey* begin, const InternalKey* end,
1627 1628
    std::vector<FileMetaData*>* inputs, int hint_index, int* file_index,
    bool expand_range) const {
1629 1630 1631 1632 1633
  if (level >= num_non_empty_levels_) {
    // this level is empty, no overlapping inputs
    return;
  }

G
Gabor Cselle 已提交
1634 1635
  inputs->clear();
  Slice user_begin, user_end;
A
Abhishek Kona 已提交
1636
  if (begin != nullptr) {
G
Gabor Cselle 已提交
1637 1638
    user_begin = begin->user_key();
  }
A
Abhishek Kona 已提交
1639
  if (end != nullptr) {
G
Gabor Cselle 已提交
1640 1641
    user_end = end->user_key();
  }
1642 1643 1644
  if (file_index) {
    *file_index = -1;
  }
S
sdong 已提交
1645
  const Comparator* user_cmp = user_comparator_;
A
Abhishek Kona 已提交
1646
  if (begin != nullptr && end != nullptr && level > 0) {
1647 1648
    GetOverlappingInputsBinarySearch(level, user_begin, user_end, inputs,
      hint_index, file_index);
1649 1650
    return;
  }
1651 1652
  for (size_t i = 0; i < level_files_brief_[level].num_files; ) {
    FdWithKeyRange* f = &(level_files_brief_[level].files[i++]);
1653 1654
    const Slice file_start = ExtractUserKey(f->smallest_key);
    const Slice file_limit = ExtractUserKey(f->largest_key);
A
Abhishek Kona 已提交
1655
    if (begin != nullptr && user_cmp->Compare(file_limit, user_begin) < 0) {
G
Gabor Cselle 已提交
1656
      // "f" is completely before specified range; skip it
A
Abhishek Kona 已提交
1657
    } else if (end != nullptr && user_cmp->Compare(file_start, user_end) > 0) {
G
Gabor Cselle 已提交
1658 1659
      // "f" is completely after specified range; skip it
    } else {
1660
      inputs->push_back(files_[level][i-1]);
1661
      if (level == 0 && expand_range) {
H
Hans Wennborg 已提交
1662 1663
        // 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 已提交
1664
        if (begin != nullptr && user_cmp->Compare(file_start, user_begin) < 0) {
H
Hans Wennborg 已提交
1665 1666 1667
          user_begin = file_start;
          inputs->clear();
          i = 0;
A
Abhishek Kona 已提交
1668 1669
        } else if (end != nullptr
            && user_cmp->Compare(file_limit, user_end) > 0) {
H
Hans Wennborg 已提交
1670 1671 1672 1673
          user_end = file_limit;
          inputs->clear();
          i = 0;
        }
1674
      } else if (file_index) {
1675
        *file_index = static_cast<int>(i) - 1;
H
Hans Wennborg 已提交
1676
      }
G
Gabor Cselle 已提交
1677 1678
    }
  }
1679 1680
}

1681 1682 1683 1684
// 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.
S
sdong 已提交
1685 1686
void VersionStorageInfo::GetOverlappingInputsBinarySearch(
    int level, const Slice& user_begin, const Slice& user_end,
1687
    std::vector<FileMetaData*>* inputs, int hint_index, int* file_index) const {
1688 1689 1690
  assert(level > 0);
  int min = 0;
  int mid = 0;
1691
  int max = static_cast<int>(files_[level].size()) - 1;
1692
  bool foundOverlap = false;
S
sdong 已提交
1693
  const Comparator* user_cmp = user_comparator_;
1694 1695 1696 1697 1698 1699 1700 1701 1702

  // 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) {
1703
    mid = (min + max)/2;
1704
    FdWithKeyRange* f = &(level_files_brief_[level].files[mid]);
1705 1706
    const Slice file_start = ExtractUserKey(f->smallest_key);
    const Slice file_limit = ExtractUserKey(f->largest_key);
1707 1708 1709 1710 1711 1712 1713 1714 1715
    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 已提交
1716

1717 1718 1719 1720
  // If there were no overlapping files, return immediately.
  if (!foundOverlap) {
    return;
  }
1721 1722 1723 1724
  // returns the index where an overlap is found
  if (file_index) {
    *file_index = mid;
  }
1725 1726
  ExtendOverlappingInputs(level, user_begin, user_end, inputs, mid);
}
A
Abhishek Kona 已提交
1727

1728 1729 1730 1731
// 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.
S
sdong 已提交
1732 1733 1734
// Use FileLevel in searching, make it faster
void VersionStorageInfo::ExtendOverlappingInputs(
    int level, const Slice& user_begin, const Slice& user_end,
1735
    std::vector<FileMetaData*>* inputs, unsigned int midIndex) const {
S
sdong 已提交
1736
  const Comparator* user_cmp = user_comparator_;
1737
  const FdWithKeyRange* files = level_files_brief_[level].files;
1738 1739 1740
#ifndef NDEBUG
  {
    // assert that the file at midIndex overlaps with the range
1741
    assert(midIndex < level_files_brief_[level].num_files);
1742 1743 1744
    const FdWithKeyRange* f = &files[midIndex];
    const Slice fstart = ExtractUserKey(f->smallest_key);
    const Slice flimit = ExtractUserKey(f->largest_key);
1745 1746 1747 1748 1749 1750 1751
    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
1752 1753
  int startIndex = midIndex + 1;
  int endIndex = midIndex;
1754
  int count __attribute__((unused)) = 0;
1755 1756

  // check backwards from 'mid' to lower indices
1757
  for (int i = midIndex; i >= 0 ; i--) {
1758 1759
    const FdWithKeyRange* f = &files[i];
    const Slice file_limit = ExtractUserKey(f->largest_key);
1760
    if (user_cmp->Compare(file_limit, user_begin) >= 0) {
1761 1762
      startIndex = i;
      assert((count++, true));
1763 1764 1765 1766 1767
    } else {
      break;
    }
  }
  // check forward from 'mid+1' to higher indices
1768 1769
  for (unsigned int i = midIndex+1;
       i < level_files_brief_[level].num_files; i++) {
1770 1771
    const FdWithKeyRange* f = &files[i];
    const Slice file_start = ExtractUserKey(f->smallest_key);
1772
    if (user_cmp->Compare(file_start, user_end) <= 0) {
1773 1774
      assert((count++, true));
      endIndex = i;
1775 1776 1777 1778
    } else {
      break;
    }
  }
1779 1780 1781 1782 1783
  assert(count == endIndex - startIndex + 1);

  // insert overlapping files into vector
  for (int i = startIndex; i <= endIndex; i++) {
    FileMetaData* f = files_[level][i];
1784
    inputs->push_back(f);
1785
  }
1786 1787
}

1788 1789 1790 1791
// 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
S
sdong 已提交
1792 1793
bool VersionStorageInfo::HasOverlappingUserKey(
    const std::vector<FileMetaData*>* inputs, int level) {
1794 1795 1796 1797 1798 1799 1800

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

S
sdong 已提交
1801 1802
  const Comparator* user_cmp = user_comparator_;
  const rocksdb::LevelFilesBrief& file_level = level_files_brief_[level];
1803
  const FdWithKeyRange* files = level_files_brief_[level].files;
1804
  const size_t kNumFiles = file_level.num_files;
1805 1806

  // Check the last file in inputs against the file after it
S
sdong 已提交
1807
  size_t last_file = FindFile(*internal_comparator_, file_level,
1808
                              inputs->back()->largest.Encode());
1809
  assert(last_file < kNumFiles);  // File should exist!
1810
  if (last_file < kNumFiles-1) {                    // If not the last file
1811 1812 1813 1814
    const Slice last_key_in_input = ExtractUserKey(
        files[last_file].largest_key);
    const Slice first_key_after = ExtractUserKey(
        files[last_file+1].smallest_key);
1815
    if (user_cmp->Equal(last_key_in_input, first_key_after)) {
1816 1817 1818 1819 1820 1821
      // 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
S
sdong 已提交
1822
  size_t first_file = FindFile(*internal_comparator_, file_level,
1823
                               inputs->front()->smallest.Encode());
1824
  assert(first_file <= last_file);   // File should exist!
1825
  if (first_file > 0) {                                 // If not first file
1826 1827 1828 1829
    const Slice& first_key_in_input = ExtractUserKey(
        files[first_file].smallest_key);
    const Slice& last_key_before = ExtractUserKey(
        files[first_file-1].largest_key);
1830
    if (user_cmp->Equal(first_key_in_input, last_key_before)) {
1831 1832 1833 1834 1835 1836 1837 1838
      // The first user key in input overlaps with the previous file's last key
      return true;
    }
  }

  return false;
}

S
sdong 已提交
1839
uint64_t VersionStorageInfo::NumLevelBytes(int level) const {
1840
  assert(level >= 0);
1841
  assert(level < num_levels());
1842 1843 1844
  return TotalFileSize(files_[level]);
}

S
sdong 已提交
1845 1846
const char* VersionStorageInfo::LevelSummary(
    LevelSummaryStorage* scratch) const {
1847
  int len = 0;
S
sdong 已提交
1848
  if (compaction_style_ == kCompactionStyleLevel && num_levels() > 1) {
1849 1850
    assert(base_level_ < static_cast<int>(level_max_bytes_.size()));
    len = snprintf(scratch->buffer, sizeof(scratch->buffer),
S
sdong 已提交
1851
                   "base level %d max bytes base %" PRIu64 " ", base_level_,
1852 1853 1854
                   level_max_bytes_[base_level_]);
  }
  len +=
S
sdong 已提交
1855
      snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "files[");
1856
  for (int i = 0; i < num_levels(); i++) {
1857 1858 1859 1860 1861
    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 已提交
1862 1863 1864 1865
  if (len > 0) {
    // overwrite the last space
    --len;
  }
1866 1867 1868 1869 1870
  len += snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len,
                  "] max score %.2f", compaction_score_[0]);

  if (!files_marked_for_compaction_.empty()) {
    snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len,
1871
             " (%" ROCKSDB_PRIszt " files need compaction)",
1872 1873 1874
             files_marked_for_compaction_.size());
  }

1875 1876 1877
  return scratch->buffer;
}

S
sdong 已提交
1878 1879
const char* VersionStorageInfo::LevelFileSummary(FileSummaryStorage* scratch,
                                                 int level) const {
1880 1881 1882
  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 已提交
1883
    char sztxt[16];
1884
    AppendHumanBytes(f->fd.GetFileSize(), sztxt, sizeof(sztxt));
1885
    int ret = snprintf(scratch->buffer + len, sz,
1886 1887
                       "#%" PRIu64 "(seq=%" PRIu64 ",sz=%s,%d) ",
                       f->fd.GetNumber(), f->smallest_seqno, sztxt,
I
Igor Canadi 已提交
1888
                       static_cast<int>(f->being_compacted));
1889 1890 1891 1892
    if (ret < 0 || ret >= sz)
      break;
    len += ret;
  }
I
Igor Canadi 已提交
1893 1894 1895 1896
  // overwrite the last space (only if files_[level].size() is non-zero)
  if (files_[level].size() && len > 0) {
    --len;
  }
1897 1898 1899 1900
  snprintf(scratch->buffer + len, sizeof(scratch->buffer) - len, "]");
  return scratch->buffer;
}

S
sdong 已提交
1901
int64_t VersionStorageInfo::MaxNextLevelOverlappingBytes() {
1902 1903
  uint64_t result = 0;
  std::vector<FileMetaData*> overlaps;
1904
  for (int level = 1; level < num_levels() - 1; level++) {
1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915
    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;
}

1916 1917 1918 1919 1920 1921 1922 1923 1924 1925
uint64_t VersionStorageInfo::MaxBytesForLevel(int level) const {
  // Note: the result for level zero is not really used since we set
  // the level-0 compaction threshold based on number of files.
  assert(level >= 0);
  assert(level < static_cast<int>(level_max_bytes_.size()));
  return level_max_bytes_[level];
}

void VersionStorageInfo::CalculateBaseBytes(const ImmutableCFOptions& ioptions,
                                            const MutableCFOptions& options) {
S
sdong 已提交
1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
  // Special logic to set number of sorted runs.
  // It is to match the previous behavior when all files are in L0.
  int num_l0_count = static_cast<int>(files_[0].size());
  if (compaction_style_ == kCompactionStyleUniversal) {
    // For universal compaction, we use level0 score to indicate
    // compaction score for the whole DB. Adding other levels as if
    // they are L0 files.
    for (int i = 1; i < num_levels(); i++) {
      if (!files_[i].empty()) {
        num_l0_count++;
      }
    }
  }
  set_l0_delay_trigger_count(num_l0_count);

1941 1942
  level_max_bytes_.resize(ioptions.num_levels);
  if (!ioptions.level_compaction_dynamic_level_bytes) {
S
sdong 已提交
1943
    base_level_ = (ioptions.compaction_style == kCompactionStyleLevel) ? 1 : -1;
1944 1945 1946 1947 1948 1949 1950 1951 1952

    // Calculate for static bytes base case
    for (int i = 0; i < ioptions.num_levels; ++i) {
      if (i == 0 && ioptions.compaction_style == kCompactionStyleUniversal) {
        level_max_bytes_[i] = options.max_bytes_for_level_base;
      } else if (i > 1) {
        level_max_bytes_[i] = MultiplyCheckOverflow(
            MultiplyCheckOverflow(level_max_bytes_[i - 1],
                                  options.max_bytes_for_level_multiplier),
S
sdong 已提交
1953
            options.MaxBytesMultiplerAdditional(i - 1));
1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988
      } else {
        level_max_bytes_[i] = options.max_bytes_for_level_base;
      }
    }
  } else {
    uint64_t max_level_size = 0;

    int first_non_empty_level = -1;
    // Find size of non-L0 level of most data.
    // Cannot use the size of the last level because it can be empty or less
    // than previous levels after compaction.
    for (int i = 1; i < num_levels_; i++) {
      uint64_t total_size = 0;
      for (const auto& f : files_[i]) {
        total_size += f->fd.GetFileSize();
      }
      if (total_size > 0 && first_non_empty_level == -1) {
        first_non_empty_level = i;
      }
      if (total_size > max_level_size) {
        max_level_size = total_size;
      }
    }

    // Prefill every level's max bytes to disallow compaction from there.
    for (int i = 0; i < num_levels_; i++) {
      level_max_bytes_[i] = std::numeric_limits<uint64_t>::max();
    }

    if (max_level_size == 0) {
      // No data for L1 and up. L0 compacts to last level directly.
      // No compaction from L1+ needs to be scheduled.
      base_level_ = num_levels_ - 1;
    } else {
      uint64_t base_bytes_max = options.max_bytes_for_level_base;
1989 1990
      uint64_t base_bytes_min = static_cast<uint64_t>(
          base_bytes_max / options.max_bytes_for_level_multiplier);
1991 1992 1993 1994 1995

      // Try whether we can make last level's target size to be max_level_size
      uint64_t cur_level_size = max_level_size;
      for (int i = num_levels_ - 2; i >= first_non_empty_level; i--) {
        // Round up after dividing
1996 1997
        cur_level_size = static_cast<uint64_t>(
            cur_level_size / options.max_bytes_for_level_multiplier);
1998 1999 2000
      }

      // Calculate base level and its size.
2001
      uint64_t base_level_size;
2002 2003 2004 2005
      if (cur_level_size <= base_bytes_min) {
        // Case 1. If we make target size of last level to be max_level_size,
        // target size of the first non-empty level would be smaller than
        // base_bytes_min. We set it be base_bytes_min.
2006
        base_level_size = base_bytes_min + 1U;
2007 2008 2009 2010 2011 2012 2013 2014 2015
        base_level_ = first_non_empty_level;
        Warn(ioptions.info_log,
             "More existing levels in DB than needed. "
             "max_bytes_for_level_multiplier may not be guaranteed.");
      } else {
        // Find base level (where L0 data is compacted to).
        base_level_ = first_non_empty_level;
        while (base_level_ > 1 && cur_level_size > base_bytes_max) {
          --base_level_;
2016 2017
          cur_level_size = static_cast<uint64_t>(
              cur_level_size / options.max_bytes_for_level_multiplier);
2018 2019 2020 2021
        }
        if (cur_level_size > base_bytes_max) {
          // Even L1 will be too large
          assert(base_level_ == 1);
2022
          base_level_size = base_bytes_max;
2023
        } else {
2024
          base_level_size = cur_level_size;
2025 2026 2027
        }
      }

2028
      uint64_t level_size = base_level_size;
2029 2030
      for (int i = base_level_; i < num_levels_; i++) {
        if (i > base_level_) {
2031 2032
          level_size = MultiplyCheckOverflow(
              level_size, options.max_bytes_for_level_multiplier);
2033 2034 2035 2036 2037 2038 2039
        }
        level_max_bytes_[i] = level_size;
      }
    }
  }
}

A
Andres Notzli 已提交
2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074
uint64_t VersionStorageInfo::EstimateLiveDataSize() const {
  // Estimate the live data size by adding up the size of the last level for all
  // key ranges. Note: Estimate depends on the ordering of files in level 0
  // because files in level 0 can be overlapping.
  uint64_t size = 0;

  auto ikey_lt = [this](InternalKey* x, InternalKey* y) {
    return internal_comparator_->Compare(*x, *y) < 0;
  };
  // (Ordered) map of largest keys in non-overlapping files
  std::map<InternalKey*, FileMetaData*, decltype(ikey_lt)> ranges(ikey_lt);

  for (int l = num_levels_ - 1; l >= 0; l--) {
    bool found_end = false;
    for (auto file : files_[l]) {
      // Find the first file where the largest key is larger than the smallest
      // key of the current file. If this file does not overlap with the
      // current file, none of the files in the map does. If there is
      // no potential overlap, we can safely insert the rest of this level
      // (if the level is not 0) into the map without checking again because
      // the elements in the level are sorted and non-overlapping.
      auto lb = (found_end && l != 0) ?
        ranges.end() : ranges.lower_bound(&file->smallest);
      found_end = (lb == ranges.end());
      if (found_end || internal_comparator_->Compare(
            file->largest, (*lb).second->smallest) < 0) {
          ranges.emplace_hint(lb, &file->largest, file);
          size += file->fd.file_size;
      }
    }
  }
  return size;
}


2075
void Version::AddLiveFiles(std::vector<FileDescriptor>* live) {
2076
  for (int level = 0; level < storage_info_.num_levels(); level++) {
S
sdong 已提交
2077
    const std::vector<FileMetaData*>& files = storage_info_.files_[level];
2078
    for (const auto& file : files) {
2079
      live->push_back(file->fd);
2080 2081 2082 2083
    }
  }
}

Z
Zheng Shao 已提交
2084
std::string Version::DebugString(bool hex) const {
J
jorlow@chromium.org 已提交
2085
  std::string r;
S
sdong 已提交
2086
  for (int level = 0; level < storage_info_.num_levels_; level++) {
2087 2088 2089 2090 2091
    // E.g.,
    //   --- level 1 ---
    //   17:123['a' .. 'd']
    //   20:43['e' .. 'g']
    r.append("--- level ");
J
jorlow@chromium.org 已提交
2092
    AppendNumberTo(&r, level);
2093 2094
    r.append(" --- version# ");
    AppendNumberTo(&r, version_number_);
2095
    r.append(" ---\n");
S
sdong 已提交
2096
    const std::vector<FileMetaData*>& files = storage_info_.files_[level];
D
dgrogan@chromium.org 已提交
2097
    for (size_t i = 0; i < files.size(); i++) {
J
jorlow@chromium.org 已提交
2098
      r.push_back(' ');
2099
      AppendNumberTo(&r, files[i]->fd.GetNumber());
J
jorlow@chromium.org 已提交
2100
      r.push_back(':');
2101
      AppendNumberTo(&r, files[i]->fd.GetFileSize());
G
Gabor Cselle 已提交
2102
      r.append("[");
Z
Zheng Shao 已提交
2103
      r.append(files[i]->smallest.DebugString(hex));
G
Gabor Cselle 已提交
2104
      r.append(" .. ");
Z
Zheng Shao 已提交
2105
      r.append(files[i]->largest.DebugString(hex));
G
Gabor Cselle 已提交
2106
      r.append("]\n");
J
jorlow@chromium.org 已提交
2107 2108 2109 2110 2111
    }
  }
  return r;
}

2112 2113 2114 2115
// this is used to batch writes to the manifest file
struct VersionSet::ManifestWriter {
  Status status;
  bool done;
2116
  InstrumentedCondVar cv;
2117
  ColumnFamilyData* cfd;
2118
  const autovector<VersionEdit*>& edit_list;
A
Abhishek Kona 已提交
2119

2120
  explicit ManifestWriter(InstrumentedMutex* mu, ColumnFamilyData* _cfd,
2121 2122
                          const autovector<VersionEdit*>& e)
      : done(false), cv(mu), cfd(_cfd), edit_list(e) {}
2123 2124
};

2125 2126
VersionSet::VersionSet(const std::string& dbname,
                       const ImmutableDBOptions* db_options,
I
Igor Canadi 已提交
2127
                       const EnvOptions& storage_options, Cache* table_cache,
2128
                       WriteBufferManager* write_buffer_manager,
2129
                       WriteController* write_controller)
2130 2131 2132
    : column_family_set_(
          new ColumnFamilySet(dbname, db_options, storage_options, table_cache,
                              write_buffer_manager, write_controller)),
2133
      env_(db_options->env),
J
jorlow@chromium.org 已提交
2134
      dbname_(dbname),
2135
      db_options_(db_options),
J
jorlow@chromium.org 已提交
2136 2137
      next_file_number_(2),
      manifest_file_number_(0),  // Filled by Recover()
2138
      pending_manifest_file_number_(0),
2139 2140
      last_sequence_(0),
      prev_log_number_(0),
A
Abhishek Kona 已提交
2141
      current_version_number_(0),
2142
      manifest_file_size_(0),
I
Igor Canadi 已提交
2143
      env_options_(storage_options),
2144
      env_options_compactions_(env_options_) {}
J
jorlow@chromium.org 已提交
2145

2146 2147 2148 2149 2150
void CloseTables(void* ptr, size_t) {
  TableReader* table_reader = reinterpret_cast<TableReader*>(ptr);
  table_reader->Close();
}

J
jorlow@chromium.org 已提交
2151
VersionSet::~VersionSet() {
2152 2153
  // we need to delete column_family_set_ because its destructor depends on
  // VersionSet
2154 2155
  column_family_set_->get_table_cache()->ApplyToAllCacheEntries(&CloseTables,
                                                                false);
2156
  column_family_set_.reset();
2157 2158 2159 2160
  for (auto file : obsolete_files_) {
    delete file;
  }
  obsolete_files_.clear();
2161 2162
}

2163 2164
void VersionSet::AppendVersion(ColumnFamilyData* column_family_data,
                               Version* v) {
I
Igor Canadi 已提交
2165 2166
  // compute new compaction score
  v->storage_info()->ComputeCompactionScore(
Y
Yi Wu 已提交
2167
      *column_family_data->ioptions(),
2168
      *column_family_data->GetLatestMutableCFOptions());
I
Igor Canadi 已提交
2169

2170
  // Mark v finalized
S
sdong 已提交
2171
  v->storage_info_.SetFinalized();
2172

2173 2174
  // Make "v" current
  assert(v->refs_ == 0);
2175 2176 2177 2178 2179
  Version* current = column_family_data->current();
  assert(v != current);
  if (current != nullptr) {
    assert(current->refs_ > 0);
    current->Unref();
2180
  }
2181
  column_family_data->SetCurrent(v);
2182 2183 2184
  v->Ref();

  // Append to linked list
2185 2186
  v->prev_ = column_family_data->dummy_versions()->prev_;
  v->next_ = column_family_data->dummy_versions();
2187 2188 2189 2190
  v->prev_->next_ = v;
  v->next_->prev_ = v;
}

2191
Status VersionSet::LogAndApply(ColumnFamilyData* column_family_data,
2192
                               const MutableCFOptions& mutable_cf_options,
2193 2194 2195
                               const autovector<VersionEdit*>& edit_list,
                               InstrumentedMutex* mu, Directory* db_directory,
                               bool new_descriptor_log,
2196
                               const ColumnFamilyOptions* new_cf_options) {
2197
  mu->AssertHeld();
2198 2199 2200 2201 2202
  // num of edits
  auto num_edits = edit_list.size();
  if (num_edits == 0) {
    return Status::OK();
  } else if (num_edits > 1) {
2203
#ifndef NDEBUG
2204 2205 2206 2207
    // no group commits for column family add or drop
    for (auto& edit : edit_list) {
      assert(!edit->IsColumnFamilyManipulation());
    }
2208
#endif
2209
  }
2210

I
Igor Canadi 已提交
2211 2212 2213
  // 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) {
2214 2215
    assert(num_edits == 1);
    assert(edit_list[0]->is_column_family_add_);
2216
    assert(new_cf_options != nullptr);
2217
  }
2218

2219
  // queue our request
2220
  ManifestWriter w(mu, column_family_data, edit_list);
2221 2222 2223
  manifest_writers_.push_back(&w);
  while (!w.done && &w != manifest_writers_.front()) {
    w.cv.Wait();
2224
  }
2225 2226 2227
  if (w.done) {
    return w.status;
  }
I
Igor Canadi 已提交
2228 2229 2230 2231 2232 2233 2234 2235
  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();
    }
2236 2237
    // we steal this code to also inform about cf-drop
    return Status::ShutdownInProgress();
I
Igor Canadi 已提交
2238
  }
A
Abhishek Kona 已提交
2239

2240
  autovector<VersionEdit*> batch_edits;
2241
  Version* v = nullptr;
S
sdong 已提交
2242
  std::unique_ptr<BaseReferencedVersionBuilder> builder_guard(nullptr);
2243

2244 2245 2246
  // process all requests in the queue
  ManifestWriter* last_writer = &w;
  assert(!manifest_writers_.empty());
2247
  assert(manifest_writers_.front() == &w);
2248
  if (w.edit_list.front()->IsColumnFamilyManipulation()) {
2249
    // no group commits for column family add or drop
2250 2251
    LogAndApplyCFHelper(w.edit_list.front());
    batch_edits.push_back(w.edit_list.front());
2252 2253
  } else {
    v = new Version(column_family_data, this, current_version_number_++);
S
sdong 已提交
2254
    builder_guard.reset(new BaseReferencedVersionBuilder(column_family_data));
2255
    auto* builder = builder_guard->version_builder();
2256
    for (const auto& writer : manifest_writers_) {
2257
      if (writer->edit_list.front()->IsColumnFamilyManipulation() ||
2258 2259 2260 2261 2262 2263
          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;
2264 2265 2266 2267
      for (const auto& edit : writer->edit_list) {
        LogAndApplyHelper(column_family_data, builder, v, edit, mu);
        batch_edits.push_back(edit);
      }
2268
    }
S
sdong 已提交
2269
    builder->SaveTo(v->storage_info());
J
jorlow@chromium.org 已提交
2270 2271 2272 2273
  }

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

2277
  assert(pending_manifest_file_number_ == 0);
2278
  if (!descriptor_log_ ||
2279
      manifest_file_size_ > db_options_->max_manifest_file_size) {
2280
    pending_manifest_file_number_ = NewFileNumber();
2281
    batch_edits.back()->SetNextFile(next_file_number_.load());
A
Abhishek Kona 已提交
2282
    new_descriptor_log = true;
2283 2284
  } else {
    pending_manifest_file_number_ = manifest_file_number_;
A
Abhishek Kona 已提交
2285 2286
  }

I
Igor Canadi 已提交
2287
  if (new_descriptor_log) {
2288
    // if we're writing out new snapshot make sure to persist max column family
I
Igor Canadi 已提交
2289
    if (column_family_set_->GetMaxColumnFamily() > 0) {
2290 2291
      w.edit_list.front()->SetMaxColumnFamily(
          column_family_set_->GetMaxColumnFamily());
I
Igor Canadi 已提交
2292
    }
J
jorlow@chromium.org 已提交
2293 2294
  }

2295
  // Unlock during expensive operations. New writes cannot get here
2296
  // because &w is ensuring that all new writes get queued.
2297
  {
2298

2299
    mu->Unlock();
2300

S
sdong 已提交
2301
    TEST_SYNC_POINT("VersionSet::LogAndApply:WriteManifest");
2302
    if (!w.edit_list.front()->IsColumnFamilyManipulation() &&
2303
        db_options_->max_open_files == -1) {
2304 2305
      // unlimited table cache. Pre-load table handle now.
      // Need to do it out of the mutex.
2306
      builder_guard->version_builder()->LoadTableHandlers(
2307
          column_family_data->internal_stats(),
2308 2309
          column_family_data->ioptions()->optimize_filters_for_hits,
          true /* prefetch_index_and_filter_in_cache */);
2310 2311
    }

I
Igor Canadi 已提交
2312 2313
    // This is fine because everything inside of this block is serialized --
    // only one thread can be here at the same time
I
Igor Canadi 已提交
2314
    if (new_descriptor_log) {
2315
      // create manifest file
2316
      Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
2317
          "Creating manifest %" PRIu64 "\n", pending_manifest_file_number_);
I
Igor Canadi 已提交
2318
      unique_ptr<WritableFile> descriptor_file;
2319
      EnvOptions opt_env_opts = env_->OptimizeForManifestWrite(env_options_);
S
sdong 已提交
2320 2321
      s = NewWritableFile(
          env_, DescriptorFileName(dbname_, pending_manifest_file_number_),
2322
          &descriptor_file, opt_env_opts);
I
Igor Canadi 已提交
2323
      if (s.ok()) {
2324
        descriptor_file->SetPreallocationBlockSize(
2325
            db_options_->manifest_preallocation_size);
2326 2327 2328

        unique_ptr<WritableFileWriter> file_writer(
            new WritableFileWriter(std::move(descriptor_file), opt_env_opts));
2329 2330
        descriptor_log_.reset(
            new log::Writer(std::move(file_writer), 0, false));
I
Igor Canadi 已提交
2331 2332 2333 2334
        s = WriteSnapshot(descriptor_log_.get());
      }
    }

2335
    if (!w.edit_list.front()->IsColumnFamilyManipulation()) {
2336
      // This is cpu-heavy operations, which should be called outside mutex.
2337
      v->PrepareApply(mutable_cf_options, true);
2338
    }
2339 2340

    // Write new record to MANIFEST log
J
jorlow@chromium.org 已提交
2341
    if (s.ok()) {
I
Igor Canadi 已提交
2342 2343
      for (auto& e : batch_edits) {
        std::string record;
2344 2345 2346 2347 2348
        if (!e->EncodeTo(&record)) {
          s = Status::Corruption(
              "Unable to Encode VersionEdit:" + e->DebugString(true));
          break;
        }
S
sdong 已提交
2349 2350
        TEST_KILL_RANDOM("VersionSet::LogAndApply:BeforeAddRecord",
                         rocksdb_kill_odds * REDUCE_ODDS2);
2351 2352 2353 2354 2355
        s = descriptor_log_->AddRecord(record);
        if (!s.ok()) {
          break;
        }
      }
S
sdong 已提交
2356 2357
      if (s.ok()) {
        s = SyncManifest(env_, db_options_, descriptor_log_->file());
2358
      }
2359
      if (!s.ok()) {
2360 2361
        Log(InfoLogLevel::ERROR_LEVEL, db_options_->info_log,
            "MANIFEST write: %s\n", s.ToString().c_str());
2362
      }
J
jorlow@chromium.org 已提交
2363 2364
    }

2365 2366
    // If we just created a new descriptor file, install it by writing a
    // new CURRENT file that points to it.
2367
    if (s.ok() && new_descriptor_log) {
2368 2369 2370
      s = SetCurrentFile(
          env_, dbname_, pending_manifest_file_number_,
          db_options_->disable_data_sync ? nullptr : db_directory);
2371 2372
    }

2373 2374 2375 2376
    if (s.ok()) {
      // find offset in manifest file where this version is stored.
      new_manifest_file_size = descriptor_log_->file()->GetFileSize();
    }
A
Abhishek Kona 已提交
2377

2378
    if (w.edit_list.front()->is_column_family_drop_) {
2379
      TEST_SYNC_POINT("VersionSet::LogAndApply::ColumnFamilyDrop:0");
2380 2381 2382 2383
      TEST_SYNC_POINT("VersionSet::LogAndApply::ColumnFamilyDrop:1");
      TEST_SYNC_POINT("VersionSet::LogAndApply::ColumnFamilyDrop:2");
    }

2384
    LogFlush(db_options_->info_log);
2385
    TEST_SYNC_POINT("VersionSet::LogAndApply:WriteManifestDone");
2386
    mu->Lock();
J
jorlow@chromium.org 已提交
2387 2388
  }

2389 2390 2391 2392 2393 2394 2395
  // Append the old mainfest file to the obsolete_manifests_ list to be deleted
  // by PurgeObsoleteFiles later.
  if (s.ok() && new_descriptor_log) {
    obsolete_manifests_.emplace_back(
        DescriptorFileName("", manifest_file_number_));
  }

J
jorlow@chromium.org 已提交
2396 2397
  // Install the new version
  if (s.ok()) {
2398
    if (w.edit_list.front()->is_column_family_add_) {
2399 2400
      // no group commit on column family add
      assert(batch_edits.size() == 1);
2401
      assert(new_cf_options != nullptr);
2402 2403
      CreateColumnFamily(*new_cf_options, w.edit_list.front());
    } else if (w.edit_list.front()->is_column_family_drop_) {
2404
      assert(batch_edits.size() == 1);
I
Igor Canadi 已提交
2405
      column_family_data->SetDropped();
2406 2407 2408 2409
      if (column_family_data->Unref()) {
        delete column_family_data;
      }
    } else {
I
Igor Canadi 已提交
2410 2411 2412 2413 2414 2415 2416 2417
      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 已提交
2418
        assert(column_family_data->GetLogNumber() <= max_log_number_in_batch);
I
Igor Canadi 已提交
2419 2420
        column_family_data->SetLogNumber(max_log_number_in_batch);
      }
2421 2422 2423
      AppendVersion(column_family_data, v);
    }

2424
    manifest_file_number_ = pending_manifest_file_number_;
2425
    manifest_file_size_ = new_manifest_file_size;
2426
    prev_log_number_ = w.edit_list.front()->prev_log_number_;
J
jorlow@chromium.org 已提交
2427
  } else {
2428 2429 2430 2431
    std::string version_edits;
    for (auto& e : batch_edits) {
      version_edits = version_edits + "\n" + e->DebugString(true);
    }
2432
    Log(InfoLogLevel::ERROR_LEVEL, db_options_->info_log,
2433 2434 2435
        "[%s] Error in committing version edit to MANIFEST: %s",
        column_family_data ? column_family_data->GetName().c_str() : "<null>",
        version_edits.c_str());
J
jorlow@chromium.org 已提交
2436
    delete v;
2437
    if (new_descriptor_log) {
2438
      Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
2439 2440
        "Deleting manifest %" PRIu64 " current manifest %" PRIu64 "\n",
        manifest_file_number_, pending_manifest_file_number_);
2441
      descriptor_log_.reset();
2442 2443
      env_->DeleteFile(
          DescriptorFileName(dbname_, pending_manifest_file_number_));
J
jorlow@chromium.org 已提交
2444 2445
    }
  }
2446
  pending_manifest_file_number_ = 0;
J
jorlow@chromium.org 已提交
2447

2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462
  // 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 已提交
2463 2464 2465
  return s;
}

I
Igor Canadi 已提交
2466 2467
void VersionSet::LogAndApplyCFHelper(VersionEdit* edit) {
  assert(edit->IsColumnFamilyManipulation());
2468
  edit->SetNextFile(next_file_number_.load());
I
Igor Canadi 已提交
2469 2470 2471 2472 2473 2474 2475 2476
  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());
  }
}

S
sdong 已提交
2477 2478
void VersionSet::LogAndApplyHelper(ColumnFamilyData* cfd,
                                   VersionBuilder* builder, Version* v,
2479
                                   VersionEdit* edit, InstrumentedMutex* mu) {
2480
  mu->AssertHeld();
I
Igor Canadi 已提交
2481
  assert(!edit->IsColumnFamilyManipulation());
2482

2483 2484
  if (edit->has_log_number_) {
    assert(edit->log_number_ >= cfd->GetLogNumber());
2485
    assert(edit->log_number_ < next_file_number_.load());
I
Igor Canadi 已提交
2486
  }
2487

I
Igor Canadi 已提交
2488 2489 2490
  if (!edit->has_prev_log_number_) {
    edit->SetPrevLogNumber(prev_log_number_);
  }
2491
  edit->SetNextFile(next_file_number_.load());
I
Igor Canadi 已提交
2492 2493
  edit->SetLastSequence(last_sequence_);

2494
  builder->Apply(edit);
2495 2496
}

I
Igor Canadi 已提交
2497
Status VersionSet::Recover(
2498 2499
    const std::vector<ColumnFamilyDescriptor>& column_families,
    bool read_only) {
I
Igor Canadi 已提交
2500 2501 2502 2503 2504 2505 2506
  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 已提交
2507
  std::unordered_map<int, std::string> column_families_not_found;
J
jorlow@chromium.org 已提交
2508 2509

  // Read "CURRENT" file, which contains a pointer to the current manifest file
K
kailiu 已提交
2510 2511 2512 2513
  std::string manifest_filename;
  Status s = ReadFileToString(
      env_, CurrentFileName(dbname_), &manifest_filename
  );
J
jorlow@chromium.org 已提交
2514 2515 2516
  if (!s.ok()) {
    return s;
  }
K
kailiu 已提交
2517 2518
  if (manifest_filename.empty() ||
      manifest_filename.back() != '\n') {
J
jorlow@chromium.org 已提交
2519 2520
    return Status::Corruption("CURRENT file does not end with newline");
  }
K
kailiu 已提交
2521 2522
  // remove the trailing '\n'
  manifest_filename.resize(manifest_filename.size() - 1);
2523 2524 2525 2526 2527 2528
  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 已提交
2529

2530 2531
  Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
      "Recovering from manifest file: %s\n",
K
kailiu 已提交
2532
      manifest_filename.c_str());
H
heyongqiang 已提交
2533

K
kailiu 已提交
2534
  manifest_filename = dbname_ + "/" + manifest_filename;
2535 2536 2537 2538 2539 2540 2541 2542 2543 2544
  unique_ptr<SequentialFileReader> manifest_file_reader;
  {
    unique_ptr<SequentialFile> manifest_file;
    s = env_->NewSequentialFile(manifest_filename, &manifest_file,
                                env_options_);
    if (!s.ok()) {
      return s;
    }
    manifest_file_reader.reset(
        new SequentialFileReader(std::move(manifest_file)));
J
jorlow@chromium.org 已提交
2545
  }
I
Igor Canadi 已提交
2546 2547
  uint64_t current_manifest_file_size;
  s = env_->GetFileSize(manifest_filename, &current_manifest_file_size);
2548 2549 2550
  if (!s.ok()) {
    return s;
  }
J
jorlow@chromium.org 已提交
2551 2552

  bool have_log_number = false;
2553
  bool have_prev_log_number = false;
J
jorlow@chromium.org 已提交
2554 2555 2556
  bool have_next_file = false;
  bool have_last_sequence = false;
  uint64_t next_file = 0;
2557 2558
  uint64_t last_sequence = 0;
  uint64_t log_number = 0;
I
Igor Canadi 已提交
2559
  uint64_t previous_log_number = 0;
2560
  uint32_t max_column_family = 0;
S
sdong 已提交
2561
  std::unordered_map<uint32_t, BaseReferencedVersionBuilder*> builders;
J
jorlow@chromium.org 已提交
2562

2563
  // add default column family
2564
  auto default_cf_iter = cf_name_to_options.find(kDefaultColumnFamilyName);
I
Igor Canadi 已提交
2565
  if (default_cf_iter == cf_name_to_options.end()) {
I
Igor Canadi 已提交
2566
    return Status::InvalidArgument("Default column family not specified");
I
Igor Canadi 已提交
2567
  }
I
Igor Canadi 已提交
2568
  VersionEdit default_cf_edit;
2569
  default_cf_edit.AddColumnFamily(kDefaultColumnFamilyName);
I
Igor Canadi 已提交
2570 2571 2572
  default_cf_edit.SetColumnFamily(0);
  ColumnFamilyData* default_cfd =
      CreateColumnFamily(default_cf_iter->second, &default_cf_edit);
S
sdong 已提交
2573
  builders.insert({0, new BaseReferencedVersionBuilder(default_cfd)});
2574

J
jorlow@chromium.org 已提交
2575
  {
I
Igor Canadi 已提交
2576
    VersionSet::LogReporter reporter;
J
jorlow@chromium.org 已提交
2577
    reporter.status = &s;
2578 2579
    log::Reader reader(NULL, std::move(manifest_file_reader), &reporter,
                       true /*checksum*/, 0 /*initial_offset*/, 0);
J
jorlow@chromium.org 已提交
2580 2581 2582
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2583
      VersionEdit edit;
J
jorlow@chromium.org 已提交
2584
      s = edit.DecodeFrom(record);
2585 2586
      if (!s.ok()) {
        break;
J
jorlow@chromium.org 已提交
2587 2588
      }

I
Igor Canadi 已提交
2589 2590 2591 2592 2593
      // 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 已提交
2594 2595 2596
      bool cf_in_not_found =
          column_families_not_found.find(edit.column_family_) !=
          column_families_not_found.end();
I
Igor Canadi 已提交
2597 2598
      // in builders means that user supplied that column family
      // option AND that we encountered column family add record
I
Igor Canadi 已提交
2599 2600 2601 2602 2603 2604
      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));

2605 2606
      ColumnFamilyData* cfd = nullptr;

2607
      if (edit.is_column_family_add_) {
I
Igor Canadi 已提交
2608 2609 2610 2611 2612 2613 2614
        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 已提交
2615 2616
          column_families_not_found.insert(
              {edit.column_family_, edit.column_family_name_});
I
Igor Canadi 已提交
2617
        } else {
2618
          cfd = CreateColumnFamily(cf_options->second, &edit);
S
sdong 已提交
2619 2620
          builders.insert(
              {edit.column_family_, new BaseReferencedVersionBuilder(cfd)});
I
Igor Canadi 已提交
2621
        }
2622
      } else if (edit.is_column_family_drop_) {
I
Igor Canadi 已提交
2623 2624 2625 2626 2627
        if (cf_in_builders) {
          auto builder = builders.find(edit.column_family_);
          assert(builder != builders.end());
          delete builder->second;
          builders.erase(builder);
2628
          cfd = column_family_set_->GetColumnFamily(edit.column_family_);
2629 2630
          if (cfd->Unref()) {
            delete cfd;
2631
            cfd = nullptr;
2632 2633 2634 2635
          } else {
            // who else can have reference to cfd!?
            assert(false);
          }
I
Igor Canadi 已提交
2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649
        } 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;
        }

2650
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
I
Igor Canadi 已提交
2651 2652
        // this should never happen since cf_in_builders is true
        assert(cfd != nullptr);
2653
        if (edit.max_level_ >= cfd->current()->storage_info()->num_levels()) {
2654 2655 2656 2657 2658
          s = Status::InvalidArgument(
              "db has more levels than options.num_levels");
          break;
        }

2659 2660 2661 2662 2663
        // 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());
2664
        builder->second->version_builder()->Apply(&edit);
2665 2666 2667
      }

      if (cfd != nullptr) {
2668
        if (edit.has_log_number_) {
I
Igor Canadi 已提交
2669
          if (cfd->GetLogNumber() > edit.log_number_) {
2670
            Log(InfoLogLevel::WARN_LEVEL, db_options_->info_log,
2671 2672
                "MANIFEST corruption detected, but ignored - Log numbers in "
                "records NOT monotonically increasing");
2673 2674 2675
          } else {
            cfd->SetLogNumber(edit.log_number_);
            have_log_number = true;
I
Igor Canadi 已提交
2676
          }
2677
        }
I
Igor Canadi 已提交
2678 2679 2680 2681 2682 2683 2684
        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;
        }
2685 2686
      }

2687
      if (edit.has_prev_log_number_) {
I
Igor Canadi 已提交
2688
        previous_log_number = edit.prev_log_number_;
2689 2690 2691
        have_prev_log_number = true;
      }

J
jorlow@chromium.org 已提交
2692 2693 2694 2695 2696
      if (edit.has_next_file_number_) {
        next_file = edit.next_file_number_;
        have_next_file = true;
      }

2697 2698 2699 2700
      if (edit.has_max_column_family_) {
        max_column_family = edit.max_column_family_;
      }

J
jorlow@chromium.org 已提交
2701
      if (edit.has_last_sequence_) {
2702
        last_sequence = edit.last_sequence_;
J
jorlow@chromium.org 已提交
2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715
        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");
    }
2716 2717

    if (!have_prev_log_number) {
I
Igor Canadi 已提交
2718
      previous_log_number = 0;
2719
    }
2720

2721 2722
    column_family_set_->UpdateMaxColumnFamily(max_column_family);

2723 2724
    MarkFileNumberUsedDuringRecovery(previous_log_number);
    MarkFileNumberUsedDuringRecovery(log_number);
J
jorlow@chromium.org 已提交
2725 2726
  }

I
Igor Canadi 已提交
2727
  // there were some column families in the MANIFEST that weren't specified
2728
  // in the argument. This is OK in read_only mode
2729
  if (read_only == false && !column_families_not_found.empty()) {
2730
    std::string list_of_not_found;
I
Igor Canadi 已提交
2731 2732
    for (const auto& cf : column_families_not_found) {
      list_of_not_found += ", " + cf.second;
2733 2734
    }
    list_of_not_found = list_of_not_found.substr(2);
I
Igor Canadi 已提交
2735
    s = Status::InvalidArgument(
I
Igor Canadi 已提交
2736 2737
        "You have to open all column families. Column families not opened: " +
        list_of_not_found);
I
Igor Canadi 已提交
2738 2739
  }

J
jorlow@chromium.org 已提交
2740
  if (s.ok()) {
I
Igor Canadi 已提交
2741
    for (auto cfd : *column_family_set_) {
2742 2743 2744
      if (cfd->IsDropped()) {
        continue;
      }
2745 2746
      auto builders_iter = builders.find(cfd->GetID());
      assert(builders_iter != builders.end());
2747
      auto* builder = builders_iter->second->version_builder();
2748

2749
      if (db_options_->max_open_files == -1) {
2750 2751
        // unlimited table cache. Pre-load table handle now.
        // Need to do it out of the mutex.
2752 2753 2754
        builder->LoadTableHandlers(
            cfd->internal_stats(), db_options_->max_file_opening_threads,
            false /* prefetch_index_and_filter_in_cache */);
2755
      }
2756

2757
      Version* v = new Version(cfd, this, current_version_number_++);
S
sdong 已提交
2758
      builder->SaveTo(v->storage_info());
2759

2760
      // Install recovered version
2761 2762
      v->PrepareApply(*cfd->GetLatestMutableCFOptions(),
          !(db_options_->skip_stats_update_on_db_open));
I
Igor Canadi 已提交
2763
      AppendVersion(cfd, v);
2764
    }
2765

I
Igor Canadi 已提交
2766
    manifest_file_size_ = current_manifest_file_size;
2767
    next_file_number_.store(next_file + 1);
2768
    last_sequence_ = last_sequence;
I
Igor Canadi 已提交
2769
    prev_log_number_ = previous_log_number;
H
heyongqiang 已提交
2770

2771
    Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
2772
        "Recovered from manifest file:%s succeeded,"
K
Kai Liu 已提交
2773 2774
        "manifest_file_number is %lu, next_file_number is %lu, "
        "last_sequence is %lu, log_number is %lu,"
2775 2776
        "prev_log_number is %lu,"
        "max_column_family is %u\n",
2777
        manifest_filename.c_str(), (unsigned long)manifest_file_number_,
2778
        (unsigned long)next_file_number_.load(), (unsigned long)last_sequence_,
2779
        (unsigned long)log_number, (unsigned long)prev_log_number_,
2780
        column_family_set_->GetMaxColumnFamily());
2781 2782

    for (auto cfd : *column_family_set_) {
2783 2784 2785
      if (cfd->IsDropped()) {
        continue;
      }
2786
      Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
I
Igor Canadi 已提交
2787 2788
          "Column family [%s] (ID %u), log number is %" PRIu64 "\n",
          cfd->GetName().c_str(), cfd->GetID(), cfd->GetLogNumber());
2789
    }
J
jorlow@chromium.org 已提交
2790 2791
  }

Y
Yi Wu 已提交
2792
  for (auto& builder : builders) {
2793 2794 2795
    delete builder.second;
  }

J
jorlow@chromium.org 已提交
2796 2797 2798
  return s;
}

I
Igor Canadi 已提交
2799 2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814 2815
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;
2816 2817 2818

  unique_ptr<SequentialFileReader> file_reader;
  {
I
Igor Canadi 已提交
2819 2820 2821 2822 2823
  unique_ptr<SequentialFile> file;
  s = env->NewSequentialFile(dscname, &file, soptions);
  if (!s.ok()) {
    return s;
  }
2824 2825
  file_reader.reset(new SequentialFileReader(std::move(file)));
  }
I
Igor Canadi 已提交
2826 2827 2828

  std::map<uint32_t, std::string> column_family_names;
  // default column family is always implicitly there
2829
  column_family_names.insert({0, kDefaultColumnFamilyName});
I
Igor Canadi 已提交
2830 2831
  VersionSet::LogReporter reporter;
  reporter.status = &s;
2832 2833
  log::Reader reader(NULL, std::move(file_reader), &reporter, true /*checksum*/,
                     0 /*initial_offset*/, 0);
I
Igor Canadi 已提交
2834 2835 2836
  Slice record;
  std::string scratch;
  while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2837 2838 2839 2840 2841 2842
    VersionEdit edit;
    s = edit.DecodeFrom(record);
    if (!s.ok()) {
      break;
    }
    if (edit.is_column_family_add_) {
2843 2844 2845 2846 2847
      if (column_family_names.find(edit.column_family_) !=
          column_family_names.end()) {
        s = Status::Corruption("Manifest adding the same column family twice");
        break;
      }
2848 2849 2850
      column_family_names.insert(
          {edit.column_family_, edit.column_family_name_});
    } else if (edit.is_column_family_drop_) {
2851 2852 2853 2854 2855 2856
      if (column_family_names.find(edit.column_family_) ==
          column_family_names.end()) {
        s = Status::Corruption(
            "Manifest - dropping non-existing column family");
        break;
      }
2857 2858
      column_family_names.erase(edit.column_family_);
    }
I
Igor Canadi 已提交
2859 2860 2861 2862 2863 2864
  }

  column_families->clear();
  if (s.ok()) {
    for (const auto& iter : column_family_names) {
      column_families->push_back(iter.second);
2865
    }
I
Igor Canadi 已提交
2866 2867 2868 2869
  }

  return s;
}
2870

I
Igor Canadi 已提交
2871
#ifndef ROCKSDB_LITE
2872 2873
Status VersionSet::ReduceNumberOfLevels(const std::string& dbname,
                                        const Options* options,
2874
                                        const EnvOptions& env_options,
2875 2876 2877 2878 2879 2880
                                        int new_levels) {
  if (new_levels <= 1) {
    return Status::InvalidArgument(
        "Number of levels needs to be bigger than 1");
  }

2881
  ImmutableDBOptions db_options(*options);
I
Igor Canadi 已提交
2882
  ColumnFamilyOptions cf_options(*options);
2883 2884
  std::shared_ptr<Cache> tc(NewLRUCache(options->max_open_files - 10,
                                        options->table_cache_numshardbits));
S
sdong 已提交
2885
  WriteController wc(options->delayed_write_rate);
2886
  WriteBufferManager wb(options->db_write_buffer_size);
2887
  VersionSet versions(dbname, &db_options, env_options, tc.get(), &wb, &wc);
2888 2889
  Status status;

2890
  std::vector<ColumnFamilyDescriptor> dummy;
2891
  ColumnFamilyDescriptor dummy_descriptor(kDefaultColumnFamilyName,
I
Igor Canadi 已提交
2892
                                          ColumnFamilyOptions(*options));
I
Igor Canadi 已提交
2893
  dummy.push_back(dummy_descriptor);
2894
  status = versions.Recover(dummy);
2895 2896 2897 2898
  if (!status.ok()) {
    return status;
  }

2899
  Version* current_version =
2900
      versions.GetColumnFamilySet()->GetDefault()->current();
S
sdong 已提交
2901
  auto* vstorage = current_version->storage_info();
2902
  int current_levels = vstorage->num_levels();
2903 2904 2905 2906 2907 2908 2909 2910 2911 2912

  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++) {
S
sdong 已提交
2913
    int file_num = vstorage->NumLevelFiles(i);
2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929
    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);
      }
    }
  }

I
Igor Canadi 已提交
2930 2931 2932
  // 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
2933
  std::vector<FileMetaData*>* new_files_list =
I
Igor Canadi 已提交
2934
      new std::vector<FileMetaData*>[current_levels];
2935
  for (int i = 0; i < new_levels - 1; i++) {
S
sdong 已提交
2936
    new_files_list[i] = vstorage->LevelFiles(i);
2937 2938 2939
  }

  if (first_nonempty_level > 0) {
S
sdong 已提交
2940
    new_files_list[new_levels - 1] = vstorage->LevelFiles(first_nonempty_level);
2941 2942
  }

S
sdong 已提交
2943 2944 2945
  delete[] vstorage -> files_;
  vstorage->files_ = new_files_list;
  vstorage->num_levels_ = new_levels;
2946

Y
Yi Wu 已提交
2947
  MutableCFOptions mutable_cf_options(*options);
2948
  VersionEdit ve;
2949 2950
  InstrumentedMutex dummy_mutex;
  InstrumentedMutexLock l(&dummy_mutex);
2951 2952 2953
  return versions.LogAndApply(
      versions.GetColumnFamilySet()->GetDefault(),
      mutable_cf_options, &ve, &dummy_mutex, nullptr, true);
2954 2955
}

I
Igor Canadi 已提交
2956
Status VersionSet::DumpManifest(Options& options, std::string& dscname,
2957
                                bool verbose, bool hex, bool json) {
2958
  // Open the specified manifest file.
2959 2960 2961 2962 2963 2964 2965 2966 2967
  unique_ptr<SequentialFileReader> file_reader;
  Status s;
  {
    unique_ptr<SequentialFile> file;
    s = options.env->NewSequentialFile(dscname, &file, env_options_);
    if (!s.ok()) {
      return s;
    }
    file_reader.reset(new SequentialFileReader(std::move(file)));
2968 2969 2970 2971 2972 2973 2974
  }

  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;
I
Igor Canadi 已提交
2975
  uint64_t previous_log_number = 0;
2976
  int count = 0;
2977
  std::unordered_map<uint32_t, std::string> comparators;
S
sdong 已提交
2978
  std::unordered_map<uint32_t, BaseReferencedVersionBuilder*> builders;
2979 2980 2981

  // add default column family
  VersionEdit default_cf_edit;
2982
  default_cf_edit.AddColumnFamily(kDefaultColumnFamilyName);
2983 2984 2985
  default_cf_edit.SetColumnFamily(0);
  ColumnFamilyData* default_cfd =
      CreateColumnFamily(ColumnFamilyOptions(options), &default_cf_edit);
S
sdong 已提交
2986
  builders.insert({0, new BaseReferencedVersionBuilder(default_cfd)});
2987 2988

  {
I
Igor Canadi 已提交
2989
    VersionSet::LogReporter reporter;
2990
    reporter.status = &s;
2991 2992
    log::Reader reader(NULL, std::move(file_reader), &reporter,
                       true /*checksum*/, 0 /*initial_offset*/, 0);
2993 2994 2995
    Slice record;
    std::string scratch;
    while (reader.ReadRecord(&record, &scratch) && s.ok()) {
2996
      VersionEdit edit;
2997
      s = edit.DecodeFrom(record);
2998 2999
      if (!s.ok()) {
        break;
3000 3001
      }

3002
      // Write out each individual edit
3003 3004 3005 3006
      if (verbose && !json) {
        printf("%s\n", edit.DebugString(hex).c_str());
      } else if (json) {
        printf("%s\n", edit.DebugJSON(count, hex).c_str());
3007 3008 3009
      }
      count++;

3010 3011 3012 3013 3014
      bool cf_in_builders =
          builders.find(edit.column_family_) != builders.end();

      if (edit.has_comparator_) {
        comparators.insert({edit.column_family_, edit.comparator_});
3015 3016
      }

3017 3018
      ColumnFamilyData* cfd = nullptr;

3019 3020 3021 3022 3023 3024
      if (edit.is_column_family_add_) {
        if (cf_in_builders) {
          s = Status::Corruption(
              "Manifest adding the same column family twice");
          break;
        }
3025
        cfd = CreateColumnFamily(ColumnFamilyOptions(options), &edit);
S
sdong 已提交
3026 3027
        builders.insert(
            {edit.column_family_, new BaseReferencedVersionBuilder(cfd)});
3028 3029 3030 3031 3032 3033 3034 3035 3036 3037
      } 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_);
3038
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
3039 3040 3041
        assert(cfd != nullptr);
        cfd->Unref();
        delete cfd;
3042
        cfd = nullptr;
3043 3044 3045 3046 3047 3048 3049
      } else {
        if (!cf_in_builders) {
          s = Status::Corruption(
              "Manifest record referencing unknown column family");
          break;
        }

3050
        cfd = column_family_set_->GetColumnFamily(edit.column_family_);
3051 3052 3053 3054 3055 3056 3057 3058
        // 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());
3059
        builder->second->version_builder()->Apply(&edit);
3060 3061
      }

3062 3063 3064 3065
      if (cfd != nullptr && edit.has_log_number_) {
        cfd->SetLogNumber(edit.log_number_);
      }

3066
      if (edit.has_prev_log_number_) {
I
Igor Canadi 已提交
3067
        previous_log_number = edit.prev_log_number_;
3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079
        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;
      }
3080 3081 3082 3083

      if (edit.has_max_column_family_) {
        column_family_set_->UpdateMaxColumnFamily(edit.max_column_family_);
      }
3084 3085
    }
  }
3086
  file_reader.reset();
3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097

  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) {
I
Igor Canadi 已提交
3098
      previous_log_number = 0;
3099 3100 3101 3102
    }
  }

  if (s.ok()) {
3103
    for (auto cfd : *column_family_set_) {
3104 3105 3106
      if (cfd->IsDropped()) {
        continue;
      }
3107 3108
      auto builders_iter = builders.find(cfd->GetID());
      assert(builders_iter != builders.end());
3109
      auto builder = builders_iter->second->version_builder();
3110 3111

      Version* v = new Version(cfd, this, current_version_number_++);
S
sdong 已提交
3112
      builder->SaveTo(v->storage_info());
3113
      v->PrepareApply(*cfd->GetLatestMutableCFOptions(), false);
3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126

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

3128 3129 3130 3131 3132
    // Free builders
    for (auto& builder : builders) {
      delete builder.second;
    }

3133
    next_file_number_.store(next_file + 1);
3134
    last_sequence_ = last_sequence;
I
Igor Canadi 已提交
3135
    prev_log_number_ = previous_log_number;
3136

3137
    printf(
3138
        "next_file_number %lu last_sequence "
3139
        "%lu  prev_log_number %lu max_column_family %u\n",
3140
        (unsigned long)next_file_number_.load(), (unsigned long)last_sequence,
I
Igor Canadi 已提交
3141
        (unsigned long)previous_log_number,
3142
        column_family_set_->GetMaxColumnFamily());
3143
  }
3144

3145 3146
  return s;
}
I
Igor Canadi 已提交
3147
#endif  // ROCKSDB_LITE
3148

3149 3150 3151 3152 3153
void VersionSet::MarkFileNumberUsedDuringRecovery(uint64_t number) {
  // only called during recovery which is single threaded, so this works because
  // there can't be concurrent calls
  if (next_file_number_.load(std::memory_order_relaxed) <= number) {
    next_file_number_.store(number + 1, std::memory_order_relaxed);
3154 3155 3156
  }
}

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

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

I
Igor Canadi 已提交
3162 3163
  // 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 已提交
3164
  // (the same single thread), so we're safe to iterate.
I
Igor Canadi 已提交
3165
  for (auto cfd : *column_family_set_) {
3166 3167 3168
    if (cfd->IsDropped()) {
      continue;
    }
3169 3170 3171
    {
      // Store column family info
      VersionEdit edit;
3172
      if (cfd->GetID() != 0) {
3173 3174
        // default column family is always there,
        // no need to explicitly write it
3175 3176
        edit.AddColumnFamily(cfd->GetName());
        edit.SetColumnFamily(cfd->GetID());
I
Igor Canadi 已提交
3177 3178 3179 3180
      }
      edit.SetComparatorName(
          cfd->internal_comparator().user_comparator()->Name());
      std::string record;
3181 3182 3183 3184
      if (!edit.EncodeTo(&record)) {
        return Status::Corruption(
            "Unable to Encode VersionEdit:" + edit.DebugString(true));
      }
I
Igor Canadi 已提交
3185 3186 3187
      Status s = log->AddRecord(record);
      if (!s.ok()) {
        return s;
3188
      }
3189
    }
3190

3191 3192 3193
    {
      // Save files
      VersionEdit edit;
3194
      edit.SetColumnFamily(cfd->GetID());
3195

I
Igor Canadi 已提交
3196
      for (int level = 0; level < cfd->NumberLevels(); level++) {
S
sdong 已提交
3197 3198
        for (const auto& f :
             cfd->current()->storage_info()->LevelFiles(level)) {
3199 3200
          edit.AddFile(level, f->fd.GetNumber(), f->fd.GetPathId(),
                       f->fd.GetFileSize(), f->smallest, f->largest,
3201 3202
                       f->smallest_seqno, f->largest_seqno,
                       f->marked_for_compaction);
3203 3204
        }
      }
3205
      edit.SetLogNumber(cfd->GetLogNumber());
3206
      std::string record;
3207 3208 3209 3210
      if (!edit.EncodeTo(&record)) {
        return Status::Corruption(
            "Unable to Encode VersionEdit:" + edit.DebugString(true));
      }
3211 3212 3213
      Status s = log->AddRecord(record);
      if (!s.ok()) {
        return s;
3214
      }
3215 3216 3217
    }
  }

I
Igor Canadi 已提交
3218
  return Status::OK();
J
jorlow@chromium.org 已提交
3219 3220
}

3221 3222 3223 3224 3225 3226
// TODO(aekmekji): in CompactionJob::GenSubcompactionBoundaries(), this
// function is called repeatedly with consecutive pairs of slices. For example
// if the slice list is [a, b, c, d] this function is called with arguments
// (a,b) then (b,c) then (c,d). Knowing this, an optimization is possible where
// we avoid doing binary search for the keys b and c twice and instead somehow
// maintain state of where they first appear in the files.
3227
uint64_t VersionSet::ApproximateSize(Version* v, const Slice& start,
3228 3229
                                     const Slice& end, int start_level,
                                     int end_level) {
3230 3231
  // pre-condition
  assert(v->cfd_->internal_comparator().Compare(start, end) <= 0);
3232

3233
  uint64_t size = 0;
S
sdong 已提交
3234
  const auto* vstorage = v->storage_info();
3235 3236 3237
  end_level = end_level == -1
                  ? vstorage->num_non_empty_levels()
                  : std::min(end_level, vstorage->num_non_empty_levels());
3238

3239 3240 3241
  assert(start_level <= end_level);

  for (int level = start_level; level < end_level; level++) {
3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
    const LevelFilesBrief& files_brief = vstorage->LevelFilesBrief(level);
    if (!files_brief.num_files) {
      // empty level, skip exploration
      continue;
    }

    if (!level) {
      // level 0 data is sorted order, handle the use case explicitly
      size += ApproximateSizeLevel0(v, files_brief, start, end);
      continue;
    }

    assert(level > 0);
    assert(files_brief.num_files > 0);

    // identify the file position for starting key
I
Igor Canadi 已提交
3258 3259 3260
    const uint64_t idx_start = FindFileInRange(
        v->cfd_->internal_comparator(), files_brief, start,
        /*start=*/0, static_cast<uint32_t>(files_brief.num_files - 1));
3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280
    assert(idx_start < files_brief.num_files);

    // scan all files from the starting position until the ending position
    // inferred from the sorted order
    for (uint64_t i = idx_start; i < files_brief.num_files; i++) {
      uint64_t val;
      val = ApproximateSize(v, files_brief.files[i], end);
      if (!val) {
        // the files after this will not have the range
        break;
      }

      size += val;

      if (i == idx_start) {
        // subtract the bytes needed to be scanned to get to the starting
        // key
        val = ApproximateSize(v, files_brief.files[i], start);
        assert(size >= val);
        size -= val;
J
jorlow@chromium.org 已提交
3281 3282 3283
      }
    }
  }
3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319

  return size;
}

uint64_t VersionSet::ApproximateSizeLevel0(Version* v,
                                           const LevelFilesBrief& files_brief,
                                           const Slice& key_start,
                                           const Slice& key_end) {
  // level 0 files are not in sorted order, we need to iterate through
  // the list to compute the total bytes that require scanning
  uint64_t size = 0;
  for (size_t i = 0; i < files_brief.num_files; i++) {
    const uint64_t start = ApproximateSize(v, files_brief.files[i], key_start);
    const uint64_t end = ApproximateSize(v, files_brief.files[i], key_end);
    assert(end >= start);
    size += end - start;
  }
  return size;
}

uint64_t VersionSet::ApproximateSize(Version* v, const FdWithKeyRange& f,
                                     const Slice& key) {
  // pre-condition
  assert(v);

  uint64_t result = 0;
  if (v->cfd_->internal_comparator().Compare(f.largest_key, key) <= 0) {
    // Entire file is before "key", so just add the file size
    result = f.fd.GetFileSize();
  } else if (v->cfd_->internal_comparator().Compare(f.smallest_key, key) > 0) {
    // Entire file is after "key", so ignore
    result = 0;
  } else {
    // "key" falls in the range for this table.  Add the
    // approximate offset of "key" within the table.
    TableReader* table_reader_ptr;
S
sdong 已提交
3320
    InternalIterator* iter = v->cfd_->table_cache()->NewIterator(
3321
        ReadOptions(), env_options_, v->cfd_->internal_comparator(), f.fd,
3322
        nullptr /* range_del_agg */, &table_reader_ptr);
3323 3324 3325 3326 3327
    if (table_reader_ptr != nullptr) {
      result = table_reader_ptr->ApproximateOffsetOf(key);
    }
    delete iter;
  }
J
jorlow@chromium.org 已提交
3328 3329 3330
  return result;
}

3331
void VersionSet::AddLiveFiles(std::vector<FileDescriptor>* live_list) {
3332 3333
  // pre-calculate space requirement
  int64_t total_files = 0;
I
Igor Canadi 已提交
3334
  for (auto cfd : *column_family_set_) {
3335 3336
    Version* dummy_versions = cfd->dummy_versions();
    for (Version* v = dummy_versions->next_; v != dummy_versions;
I
Igor Canadi 已提交
3337
         v = v->next_) {
S
sdong 已提交
3338
      const auto* vstorage = v->storage_info();
3339
      for (int level = 0; level < vstorage->num_levels(); level++) {
S
sdong 已提交
3340
        total_files += vstorage->LevelFiles(level).size();
3341
      }
3342 3343 3344 3345
    }
  }

  // just one time extension to the right size
3346
  live_list->reserve(live_list->size() + static_cast<size_t>(total_files));
3347

I
Igor Canadi 已提交
3348
  for (auto cfd : *column_family_set_) {
3349 3350
    auto* current = cfd->current();
    bool found_current = false;
3351 3352
    Version* dummy_versions = cfd->dummy_versions();
    for (Version* v = dummy_versions->next_; v != dummy_versions;
I
Igor Canadi 已提交
3353
         v = v->next_) {
3354 3355 3356
      v->AddLiveFiles(live_list);
      if (v == current) {
        found_current = true;
J
jorlow@chromium.org 已提交
3357 3358
      }
    }
3359 3360 3361 3362 3363
    if (!found_current && current != nullptr) {
      // Should never happen unless it is a bug.
      assert(false);
      current->AddLiveFiles(live_list);
    }
J
jorlow@chromium.org 已提交
3364 3365 3366
  }
}

3367 3368
InternalIterator* VersionSet::MakeInputIterator(
    const Compaction* c, RangeDelAggregator* range_del_agg) {
L
Lei Jin 已提交
3369 3370 3371
  auto cfd = c->column_family_data();
  ReadOptions read_options;
  read_options.verify_checksums =
3372
    c->mutable_cf_options()->verify_checksums_in_compaction;
L
Lei Jin 已提交
3373
  read_options.fill_cache = false;
3374
  if (c->ShouldFormSubcompactions()) {
3375 3376
    read_options.total_order_seek = true;
  }
J
jorlow@chromium.org 已提交
3377 3378 3379 3380

  // 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
3381 3382 3383
  const size_t space = (c->level() == 0 ? c->input_levels(0)->num_files +
                                              c->num_input_levels() - 1
                                        : c->num_input_levels());
S
sdong 已提交
3384
  InternalIterator** list = new InternalIterator* [space];
3385 3386
  size_t num = 0;
  for (size_t which = 0; which < c->num_input_levels(); which++) {
F
Feng Zhu 已提交
3387
    if (c->input_levels(which)->num_files != 0) {
3388
      if (c->level(which) == 0) {
3389
        const LevelFilesBrief* flevel = c->input_levels(which);
F
Feng Zhu 已提交
3390
        for (size_t i = 0; i < flevel->num_files; i++) {
L
Lei Jin 已提交
3391
          list[num++] = cfd->table_cache()->NewIterator(
3392
              read_options, env_options_compactions_,
3393 3394 3395
              cfd->internal_comparator(), flevel->files[i].fd, range_del_agg,
              nullptr /* table_reader_ptr */,
              nullptr /* no per level latency histogram */,
3396
              true /* for_compaction */, nullptr /* arena */,
3397
              false /* skip_filters */, (int)which /* level */);
J
jorlow@chromium.org 已提交
3398 3399 3400
        }
      } else {
        // Create concatenating iterator for the files from this level
3401 3402 3403 3404 3405
        list[num++] = NewTwoLevelIterator(
            new LevelFileIteratorState(
                cfd->table_cache(), read_options, env_options_,
                cfd->internal_comparator(),
                nullptr /* no per level latency histogram */,
3406
                true /* for_compaction */, false /* prefix enabled */,
3407 3408
                false /* skip_filters */, (int)which /* level */,
                range_del_agg),
3409 3410
            new LevelFileNumIterator(cfd->internal_comparator(),
                                     c->input_levels(which)));
J
jorlow@chromium.org 已提交
3411 3412 3413 3414
      }
    }
  }
  assert(num <= space);
S
sdong 已提交
3415
  InternalIterator* result =
3416 3417
      NewMergingIterator(&c->column_family_data()->internal_comparator(), list,
                         static_cast<int>(num));
J
jorlow@chromium.org 已提交
3418 3419 3420 3421
  delete[] list;
  return result;
}

A
Abhishek Kona 已提交
3422
// verify that the files listed in this compaction are present
3423 3424
// in the current version
bool VersionSet::VerifyCompactionFileConsistency(Compaction* c) {
3425
#ifndef NDEBUG
I
Igor Canadi 已提交
3426
  Version* version = c->column_family_data()->current();
S
sdong 已提交
3427
  const VersionStorageInfo* vstorage = version->storage_info();
3428
  if (c->input_version() != version) {
3429 3430 3431
    Log(InfoLogLevel::INFO_LEVEL, db_options_->info_log,
        "[%s] compaction output being applied to a different base version from"
        " input version",
I
Igor Canadi 已提交
3432
        c->column_family_data()->GetName().c_str());
S
sdong 已提交
3433 3434 3435

    if (vstorage->compaction_style_ == kCompactionStyleLevel &&
        c->start_level() == 0 && c->num_input_levels() > 2U) {
3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446
      // We are doing a L0->base_level compaction. The assumption is if
      // base level is not L1, levels from L1 to base_level - 1 is empty.
      // This is ensured by having one compaction from L0 going on at the
      // same time in level-based compaction. So that during the time, no
      // compaction/flush can put files to those levels.
      for (int l = c->start_level() + 1; l < c->output_level(); l++) {
        if (vstorage->NumLevelFiles(l) != 0) {
          return false;
        }
      }
    }
3447 3448
  }

3449
  for (size_t input = 0; input < c->num_input_levels(); ++input) {
3450
    int level = c->level(input);
3451
    for (size_t i = 0; i < c->num_input_files(input); ++i) {
3452 3453
      uint64_t number = c->input(input, i)->fd.GetNumber();
      bool found = false;
3454
      for (size_t j = 0; j < vstorage->files_[level].size(); j++) {
3455 3456 3457 3458 3459
        FileMetaData* f = vstorage->files_[level][j];
        if (f->fd.GetNumber() == number) {
          found = true;
          break;
        }
3460
      }
3461 3462
      if (!found) {
        return false;  // input files non existent in current version
3463 3464 3465
      }
    }
  }
3466
#endif
3467 3468 3469
  return true;     // everything good
}

3470
Status VersionSet::GetMetadataForFile(uint64_t number, int* filelevel,
3471
                                      FileMetaData** meta,
3472 3473
                                      ColumnFamilyData** cfd) {
  for (auto cfd_iter : *column_family_set_) {
3474
    Version* version = cfd_iter->current();
S
sdong 已提交
3475
    const auto* vstorage = version->storage_info();
3476
    for (int level = 0; level < vstorage->num_levels(); level++) {
S
sdong 已提交
3477
      for (const auto& file : vstorage->LevelFiles(level)) {
3478
        if (file->fd.GetNumber() == number) {
3479
          *meta = file;
3480
          *filelevel = level;
3481
          *cfd = cfd_iter;
3482 3483
          return Status::OK();
        }
3484 3485 3486 3487 3488 3489
      }
    }
  }
  return Status::NotFound("File not present in any level");
}

3490
void VersionSet::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
I
Igor Canadi 已提交
3491
  for (auto cfd : *column_family_set_) {
3492 3493 3494
    if (cfd->IsDropped()) {
      continue;
    }
I
Igor Canadi 已提交
3495
    for (int level = 0; level < cfd->NumberLevels(); level++) {
S
sdong 已提交
3496 3497
      for (const auto& file :
           cfd->current()->storage_info()->LevelFiles(level)) {
3498
        LiveFileMetaData filemetadata;
3499
        filemetadata.column_family_name = cfd->GetName();
3500
        uint32_t path_id = file->fd.GetPathId();
3501 3502
        if (path_id < db_options_->db_paths.size()) {
          filemetadata.db_path = db_options_->db_paths[path_id].path;
3503
        } else {
3504 3505
          assert(!db_options_->db_paths.empty());
          filemetadata.db_path = db_options_->db_paths.back().path;
3506 3507
        }
        filemetadata.name = MakeTableFileName("", file->fd.GetNumber());
3508
        filemetadata.level = level;
3509
        filemetadata.size = file->fd.GetFileSize();
I
Igor Canadi 已提交
3510 3511 3512 3513
        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;
3514 3515
        metadata->push_back(filemetadata);
      }
3516 3517 3518 3519
    }
  }
}

I
Igor Canadi 已提交
3520
void VersionSet::GetObsoleteFiles(std::vector<FileMetaData*>* files,
3521
                                  std::vector<std::string>* manifest_filenames,
I
Igor Canadi 已提交
3522
                                  uint64_t min_pending_output) {
3523 3524
  assert(manifest_filenames->empty());
  obsolete_manifests_.swap(*manifest_filenames);
I
Igor Canadi 已提交
3525 3526 3527 3528 3529 3530 3531 3532 3533
  std::vector<FileMetaData*> pending_files;
  for (auto f : obsolete_files_) {
    if (f->fd.GetNumber() < min_pending_output) {
      files->push_back(f);
    } else {
      pending_files.push_back(f);
    }
  }
  obsolete_files_.swap(pending_files);
I
Igor Canadi 已提交
3534 3535
}

3536
ColumnFamilyData* VersionSet::CreateColumnFamily(
3537
    const ColumnFamilyOptions& cf_options, VersionEdit* edit) {
3538 3539
  assert(edit->is_column_family_add_);

3540
  Version* dummy_versions = new Version(nullptr, this);
3541 3542 3543
  // Ref() dummy version once so that later we can call Unref() to delete it
  // by avoiding calling "delete" explicitly (~Version is private)
  dummy_versions->Ref();
I
Igor Canadi 已提交
3544
  auto new_cfd = column_family_set_->CreateColumnFamily(
3545 3546
      edit->column_family_name_, edit->column_family_, dummy_versions,
      cf_options);
I
Igor Canadi 已提交
3547

3548 3549
  Version* v = new Version(new_cfd, this, current_version_number_++);

3550 3551 3552
  // Fill level target base information.
  v->storage_info()->CalculateBaseBytes(*new_cfd->ioptions(),
                                        *new_cfd->GetLatestMutableCFOptions());
3553
  AppendVersion(new_cfd, v);
3554 3555
  // GetLatestMutableCFOptions() is safe here without mutex since the
  // cfd is not available to client
A
agiardullo 已提交
3556 3557
  new_cfd->CreateNewMemtable(*new_cfd->GetLatestMutableCFOptions(),
                             LastSequence());
I
Igor Canadi 已提交
3558
  new_cfd->SetLogNumber(edit->log_number_);
3559 3560 3561
  return new_cfd;
}

3562 3563 3564 3565 3566 3567 3568 3569
uint64_t VersionSet::GetNumLiveVersions(Version* dummy_versions) {
  uint64_t count = 0;
  for (Version* v = dummy_versions->next_; v != dummy_versions; v = v->next_) {
    count++;
  }
  return count;
}

3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587
uint64_t VersionSet::GetTotalSstFilesSize(Version* dummy_versions) {
  std::unordered_set<uint64_t> unique_files;
  uint64_t total_files_size = 0;
  for (Version* v = dummy_versions->next_; v != dummy_versions; v = v->next_) {
    VersionStorageInfo* storage_info = v->storage_info();
    for (int level = 0; level < storage_info->num_levels_; level++) {
      for (const auto& file_meta : storage_info->LevelFiles(level)) {
        if (unique_files.find(file_meta->fd.packed_number_and_path_id) ==
            unique_files.end()) {
          unique_files.insert(file_meta->fd.packed_number_and_path_id);
          total_files_size += file_meta->fd.GetFileSize();
        }
      }
    }
  }
  return total_files_size;
}

3588
}  // namespace rocksdb