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

H
Haobo Xu 已提交
11
#include <atomic>
12
#include <deque>
13
#include <limits>
J
jorlow@chromium.org 已提交
14
#include <set>
T
Tomislav Novak 已提交
15
#include <utility>
16
#include <vector>
I
Igor Canadi 已提交
17
#include <string>
K
kailiu 已提交
18

J
jorlow@chromium.org 已提交
19 20 21
#include "db/dbformat.h"
#include "db/log_writer.h"
#include "db/snapshot.h"
22
#include "db/column_family.h"
23
#include "db/version_edit.h"
K
Kai Liu 已提交
24 25
#include "memtable_list.h"
#include "port/port.h"
26 27 28 29
#include "rocksdb/db.h"
#include "rocksdb/env.h"
#include "rocksdb/memtablerep.h"
#include "rocksdb/transaction_log.h"
30
#include "util/autovector.h"
31
#include "util/stop_watch.h"
32
#include "util/thread_local.h"
33
#include "util/scoped_arena_iterator.h"
I
Igor Canadi 已提交
34
#include "db/internal_stats.h"
35
#include "db/write_controller.h"
I
Igor Canadi 已提交
36
#include "db/flush_scheduler.h"
37

38
namespace rocksdb {
J
jorlow@chromium.org 已提交
39 40 41 42 43 44

class MemTable;
class TableCache;
class Version;
class VersionEdit;
class VersionSet;
D
Danny Guo 已提交
45
class CompactionFilterV2;
46
class Arena;
J
jorlow@chromium.org 已提交
47 48 49

class DBImpl : public DB {
 public:
I
Igor Canadi 已提交
50
  DBImpl(const DBOptions& options, const std::string& dbname);
J
jorlow@chromium.org 已提交
51 52 53
  virtual ~DBImpl();

  // Implementations of the DB interface
54 55
  using DB::Put;
  virtual Status Put(const WriteOptions& options,
56
                     ColumnFamilyHandle* column_family, const Slice& key,
57 58 59
                     const Slice& value);
  using DB::Merge;
  virtual Status Merge(const WriteOptions& options,
60 61
                       ColumnFamilyHandle* column_family, const Slice& key,
                       const Slice& value);
62 63
  using DB::Delete;
  virtual Status Delete(const WriteOptions& options,
64
                        ColumnFamilyHandle* column_family, const Slice& key);
65
  using DB::Write;
J
jorlow@chromium.org 已提交
66
  virtual Status Write(const WriteOptions& options, WriteBatch* updates);
67
  using DB::Get;
J
jorlow@chromium.org 已提交
68
  virtual Status Get(const ReadOptions& options,
69
                     ColumnFamilyHandle* column_family, const Slice& key,
J
jorlow@chromium.org 已提交
70
                     std::string* value);
71 72 73
  using DB::MultiGet;
  virtual std::vector<Status> MultiGet(
      const ReadOptions& options,
74
      const std::vector<ColumnFamilyHandle*>& column_family,
75
      const std::vector<Slice>& keys, std::vector<std::string>* values);
76

77
  virtual Status CreateColumnFamily(const ColumnFamilyOptions& options,
78
                                    const std::string& column_family,
79 80
                                    ColumnFamilyHandle** handle);
  virtual Status DropColumnFamily(ColumnFamilyHandle* column_family);
81

82 83 84 85
  // Returns false if key doesn't exist in the database and true if it may.
  // If value_found is not passed in as null, then return the value if found in
  // memory. On return, if value was found, then value_found will be set to true
  // , otherwise false.
86
  using DB::KeyMayExist;
87
  virtual bool KeyMayExist(const ReadOptions& options,
88 89
                           ColumnFamilyHandle* column_family, const Slice& key,
                           std::string* value, bool* value_found = nullptr);
90 91
  using DB::NewIterator;
  virtual Iterator* NewIterator(const ReadOptions& options,
92
                                ColumnFamilyHandle* column_family);
93 94
  virtual Status NewIterators(
      const ReadOptions& options,
I
Igor Canadi 已提交
95
      const std::vector<ColumnFamilyHandle*>& column_families,
96
      std::vector<Iterator*>* iterators);
J
jorlow@chromium.org 已提交
97 98
  virtual const Snapshot* GetSnapshot();
  virtual void ReleaseSnapshot(const Snapshot* snapshot);
99
  using DB::GetProperty;
100
  virtual bool GetProperty(ColumnFamilyHandle* column_family,
101
                           const Slice& property, std::string* value);
102 103 104
  using DB::GetIntProperty;
  virtual bool GetIntProperty(ColumnFamilyHandle* column_family,
                              const Slice& property, uint64_t* value) override;
105
  using DB::GetApproximateSizes;
106
  virtual void GetApproximateSizes(ColumnFamilyHandle* column_family,
107 108
                                   const Range* range, int n, uint64_t* sizes);
  using DB::CompactRange;
109
  virtual Status CompactRange(ColumnFamilyHandle* column_family,
110
                              const Slice* begin, const Slice* end,
111 112
                              bool reduce_level = false, int target_level = -1,
                              uint32_t target_path_id = 0);
113 114

  using DB::NumberLevels;
115
  virtual int NumberLevels(ColumnFamilyHandle* column_family);
116
  using DB::MaxMemCompactionLevel;
117
  virtual int MaxMemCompactionLevel(ColumnFamilyHandle* column_family);
118
  using DB::Level0StopWriteTrigger;
119
  virtual int Level0StopWriteTrigger(ColumnFamilyHandle* column_family);
I
Igor Canadi 已提交
120
  virtual const std::string& GetName() const;
121
  virtual Env* GetEnv() const;
122
  using DB::GetOptions;
123
  virtual const Options& GetOptions(ColumnFamilyHandle* column_family) const;
124 125
  using DB::Flush;
  virtual Status Flush(const FlushOptions& options,
126
                       ColumnFamilyHandle* column_family);
I
Igor Canadi 已提交
127 128 129 130

  virtual SequenceNumber GetLatestSequenceNumber() const;

#ifndef ROCKSDB_LITE
131
  virtual Status DisableFileDeletions();
132
  virtual Status EnableFileDeletions(bool force);
133
  virtual int IsFileDeletionsEnabled() const;
I
Igor Canadi 已提交
134
  // All the returned filenames start with "/"
135
  virtual Status GetLiveFiles(std::vector<std::string>&,
136 137
                              uint64_t* manifest_file_size,
                              bool flush_memtable = true);
138
  virtual Status GetSortedWalFiles(VectorLogPtr& files);
I
Igor Canadi 已提交
139

140 141 142 143
  virtual Status GetUpdatesSince(
      SequenceNumber seq_number, unique_ptr<TransactionLogIterator>* iter,
      const TransactionLogIterator::ReadOptions&
          read_options = TransactionLogIterator::ReadOptions());
144 145
  virtual Status DeleteFile(std::string name);

146
  virtual void GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata);
I
Igor Canadi 已提交
147
#endif  // ROCKSDB_LITE
148

I
Igor Canadi 已提交
149 150 151 152
  // checks if all live files exist on file system and that their file sizes
  // match to our in-memory records
  virtual Status CheckConsistency();

153 154
  virtual Status GetDbIdentity(std::string& identity);

I
Igor Canadi 已提交
155
  Status RunManualCompaction(ColumnFamilyData* cfd, int input_level,
156 157
                             int output_level, uint32_t output_path_id,
                             const Slice* begin, const Slice* end);
158

159
#ifndef ROCKSDB_LITE
J
jorlow@chromium.org 已提交
160
  // Extra methods (for testing) that are not in the public DB interface
I
Igor Canadi 已提交
161
  // Implemented in db_impl_debug.cc
J
jorlow@chromium.org 已提交
162

163
  // Compact any files in the named level that overlap [*begin, *end]
164 165
  Status TEST_CompactRange(int level, const Slice* begin, const Slice* end,
                           ColumnFamilyHandle* column_family = nullptr);
J
jorlow@chromium.org 已提交
166

167
  // Force current memtable contents to be flushed.
168
  Status TEST_FlushMemTable(bool wait = true);
J
jorlow@chromium.org 已提交
169

170
  // Wait for memtable compaction
171
  Status TEST_WaitForFlushMemTable(ColumnFamilyHandle* column_family = nullptr);
172 173 174 175

  // Wait for any compaction
  Status TEST_WaitForCompact();

J
jorlow@chromium.org 已提交
176 177 178
  // Return an internal iterator over the current state of the database.
  // The keys of this iterator are internal keys (see format.h).
  // The returned iterator should be deleted when no longer needed.
179 180
  Iterator* TEST_NewInternalIterator(
      Arena* arena, ColumnFamilyHandle* column_family = nullptr);
J
jorlow@chromium.org 已提交
181

182 183
  // Return the maximum overlapping data (in bytes) at next level for any
  // file at a level >= 1.
184 185
  int64_t TEST_MaxNextLevelOverlappingBytes(ColumnFamilyHandle* column_family =
                                                nullptr);
186

A
Abhishek Kona 已提交
187 188
  // Return the current manifest file no.
  uint64_t TEST_Current_Manifest_FileNo();
189 190 191 192

  // Trigger's a background call for testing.
  void TEST_PurgeObsoleteteWAL();

193
  // get total level0 file size. Only for testing.
194
  uint64_t TEST_GetLevel0TotalSize();
195

196 197 198 199 200
  void TEST_SetDefaultTimeToCheck(uint64_t default_interval_to_delete_obsolete_WAL)
  {
    default_interval_to_delete_obsolete_WAL_ = default_interval_to_delete_obsolete_WAL;
  }

201 202
  void TEST_GetFilesMetaData(ColumnFamilyHandle* column_family,
                             std::vector<std::vector<FileMetaData>>* metadata);
203

I
Igor Canadi 已提交
204 205 206 207
  Status TEST_ReadFirstRecord(const WalFileType type, const uint64_t number,
                              SequenceNumber* sequence);

  Status TEST_ReadFirstLine(const std::string& fname, SequenceNumber* sequence);
208 209 210 211 212 213 214 215 216 217 218

  void TEST_LockMutex();

  void TEST_UnlockMutex();

  // REQUIRES: mutex locked
  void* TEST_BeginWrite();

  // REQUIRES: mutex locked
  // pass the pointer that you got from TEST_BeginWrite()
  void TEST_EndWrite(void* w);
I
Igor Canadi 已提交
219 220
#endif  // NDEBUG

221 222 223 224 225 226 227 228 229 230 231
  // Structure to store information for candidate files to delete.
  struct CandidateFileInfo {
    std::string file_name;
    uint32_t path_id;
    CandidateFileInfo(std::string name, uint32_t path)
        : file_name(name), path_id(path) {}
    bool operator==(const CandidateFileInfo& other) const {
      return file_name == other.file_name && path_id == other.path_id;
    }
  };

I
Igor Canadi 已提交
232
  // needed for CleanupIteratorState
I
Igor Canadi 已提交
233 234
  struct DeletionState {
    inline bool HaveSomethingToDelete() const {
K
kailiu 已提交
235
      return  candidate_files.size() ||
I
Igor Canadi 已提交
236 237 238
        sst_delete_files.size() ||
        log_delete_files.size();
    }
239

I
Igor Canadi 已提交
240 241 242
    // a list of all files that we'll consider deleting
    // (every once in a while this is filled up with all files
    // in the DB directory)
243
    std::vector<CandidateFileInfo> candidate_files;
I
Igor Canadi 已提交
244 245

    // the list of all live sst files that cannot be deleted
246
    std::vector<FileDescriptor> sst_live;
I
Igor Canadi 已提交
247 248 249 250 251 252 253

    // a list of sst files that we need to delete
    std::vector<FileMetaData*> sst_delete_files;

    // a list of log files that we need to delete
    std::vector<uint64_t> log_delete_files;

254
    // a list of memtables to be free
K
Kai Liu 已提交
255
    autovector<MemTable*> memtables_to_free;
256

257
    autovector<SuperVersion*> superversions_to_free;
I
Igor Canadi 已提交
258

259
    SuperVersion* new_superversion;  // if nullptr no new superversion
I
Igor Canadi 已提交
260

I
Igor Canadi 已提交
261 262
    // the current manifest_file_number, log_number and prev_log_number
    // that corresponds to the set of files in 'live'.
263 264
    uint64_t manifest_file_number, pending_manifest_file_number, log_number,
        prev_log_number;
I
Igor Canadi 已提交
265

K
Kai Liu 已提交
266
    explicit DeletionState(bool create_superversion = false) {
I
Igor Canadi 已提交
267
      manifest_file_number = 0;
268
      pending_manifest_file_number = 0;
I
Igor Canadi 已提交
269 270
      log_number = 0;
      prev_log_number = 0;
271
      new_superversion = create_superversion ? new SuperVersion() : nullptr;
I
Igor Canadi 已提交
272 273 274 275 276 277 278
    }

    ~DeletionState() {
      // free pending memtables
      for (auto m : memtables_to_free) {
        delete m;
      }
279 280 281 282
      // free superversions
      for (auto s : superversions_to_free) {
        delete s;
      }
I
Igor Canadi 已提交
283 284 285
      // if new_superversion was not used, it will be non-nullptr and needs
      // to be freed here
      delete new_superversion;
I
Igor Canadi 已提交
286 287 288 289
    }
  };

  // Returns the list of live files in 'live' and the list
K
kailiu 已提交
290
  // of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
291
  // If force == false and the last call was less than
292
  // db_options_.delete_obsolete_files_period_micros microseconds ago,
I
Igor Canadi 已提交
293 294 295 296 297 298 299 300 301 302 303
  // it will not fill up the deletion_state
  void FindObsoleteFiles(DeletionState& deletion_state,
                         bool force,
                         bool no_full_scan = false);

  // Diffs the files listed in filenames and those that do not
  // belong to live files are posibly removed. Also, removes all the
  // files in sst_delete_files and log_delete_files.
  // It is not necessary to hold the mutex when invoking this method.
  void PurgeObsoleteFiles(DeletionState& deletion_state);

304 305
  ColumnFamilyHandle* DefaultColumnFamily() const;

306
 protected:
H
heyongqiang 已提交
307 308
  Env* const env_;
  const std::string dbname_;
309
  unique_ptr<VersionSet> versions_;
310
  const DBOptions db_options_;
L
Lei Jin 已提交
311
  Statistics* stats_;
H
heyongqiang 已提交
312

313
  Iterator* NewInternalIterator(const ReadOptions&, ColumnFamilyData* cfd,
314
                                SuperVersion* super_version, Arena* arena);
315

J
jorlow@chromium.org 已提交
316 317
 private:
  friend class DB;
318
  friend class InternalStats;
I
Igor Canadi 已提交
319
#ifndef ROCKSDB_LITE
L
Lei Jin 已提交
320
  friend class ForwardIterator;
I
Igor Canadi 已提交
321
#endif
322
  friend struct SuperVersion;
323
  struct CompactionState;
324

S
Stanislau Hlebik 已提交
325
  struct WriteContext;
J
jorlow@chromium.org 已提交
326 327 328 329 330 331

  Status NewDB();

  // Recover the descriptor from persistent storage.  May do a significant
  // amount of work to recover recently logged updates.  Any changes to
  // be made to the descriptor are added to *edit.
332 333
  Status Recover(const std::vector<ColumnFamilyDescriptor>& column_families,
                 bool read_only = false, bool error_if_log_file_exist = false);
J
jorlow@chromium.org 已提交
334 335 336

  void MaybeIgnoreError(Status* s) const;

337 338
  const Status CreateArchivalDirectory();

J
jorlow@chromium.org 已提交
339 340 341
  // Delete any unneeded files and stale in-memory entries.
  void DeleteObsoleteFiles();

342
  // Flush the in-memory write buffer to storage.  Switches to a new
J
jorlow@chromium.org 已提交
343
  // log-file/memtable and writes a new descriptor iff successful.
344
  Status FlushMemTableToOutputFile(ColumnFamilyData* cfd, bool* madeProgress,
H
Haobo Xu 已提交
345 346
                                   DeletionState& deletion_state,
                                   LogBuffer* log_buffer);
J
jorlow@chromium.org 已提交
347

S
Stanislau Hlebik 已提交
348 349 350
  // REQUIRES: log_numbers are sorted in ascending order
  Status RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
                         SequenceNumber* max_sequence, bool read_only);
J
jorlow@chromium.org 已提交
351

352 353 354 355 356
  // The following two methods are used to flush a memtable to
  // storage. The first one is used atdatabase RecoveryTime (when the
  // database is opened) and is heavyweight because it holds the mutex
  // for the entire period. The second method WriteLevel0Table supports
  // concurrent flush memtables to storage.
357 358
  Status WriteLevel0TableForRecovery(ColumnFamilyData* cfd, MemTable* mem,
                                     VersionEdit* edit);
359
  Status WriteLevel0Table(ColumnFamilyData* cfd, autovector<MemTable*>& mems,
360
                          VersionEdit* edit, uint64_t* filenumber,
H
Haobo Xu 已提交
361
                          LogBuffer* log_buffer);
362 363 364 365 366 367 368 369 370 371 372 373 374 375
  // Information kept for every waiting writer
  struct Writer {
    Status status;
    WriteBatch* batch;
    bool sync;
    bool disableWAL;
    bool in_batch_group;
    bool done;
    uint64_t timeout_hint_us;
    port::CondVar cv;

    explicit Writer(port::Mutex* mu) : cv(mu) {}
  };

S
Stanislau Hlebik 已提交
376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
  // Before applying write operation (such as DBImpl::Write, DBImpl::Flush)
  // thread should grab the mutex_ and be the first on writers queue.
  // BeginWrite is used for it.
  // Be aware! Writer's job can be done by other thread (see DBImpl::Write
  // for examples), so check it via w.done before applying changes.
  //
  // Writer* w:                writer to be placed in the queue
  // uint64_t expiration_time: maximum time to be in the queue
  // See also: EndWrite
  Status BeginWrite(Writer* w, uint64_t expiration_time);

  // After doing write job, we need to remove already used writers from
  // writers_ queue and notify head of the queue about it.
  // EndWrite is used for this.
  //
  // Writer* w:           Writer, that was added by BeginWrite function
  // Writer* last_writer: Since we can join a few Writers (as DBImpl::Write
  //                      does)
  //                      we should pass last_writer as a parameter to
  //                      EndWrite
  //                      (if you don't touch other writers, just pass w)
  // Status status:       Status of write operation
  // See also: BeginWrite
  void EndWrite(Writer* w, Writer* last_writer, Status status);

401 402
  void DelayWrite(uint64_t expiration_time);

I
Igor Canadi 已提交
403
  Status ScheduleFlushes(WriteContext* context);
404

S
Stanislau Hlebik 已提交
405 406 407
  Status SetNewMemtableAndNewLogFile(ColumnFamilyData* cfd,
                                     WriteContext* context);

408 409
  void BuildBatchGroup(Writer** last_writer,
                       autovector<WriteBatch*>* write_batch_group);
J
jorlow@chromium.org 已提交
410

H
heyongqiang 已提交
411
  // Force current memtable contents to be flushed.
412
  Status FlushMemTable(ColumnFamilyData* cfd, const FlushOptions& options);
H
heyongqiang 已提交
413

414
  // Wait for memtable flushed
415
  Status WaitForFlushMemTable(ColumnFamilyData* cfd);
H
heyongqiang 已提交
416

417 418 419
  void RecordFlushIOStats();
  void RecordCompactionIOStats();

420
  void MaybeScheduleFlushOrCompaction();
421 422 423 424
  static void BGWorkCompaction(void* db);
  static void BGWorkFlush(void* db);
  void BackgroundCallCompaction();
  void BackgroundCallFlush();
425 426
  Status BackgroundCompaction(bool* madeProgress, DeletionState& deletion_state,
                              LogBuffer* log_buffer);
H
Haobo Xu 已提交
427 428
  Status BackgroundFlush(bool* madeProgress, DeletionState& deletion_state,
                         LogBuffer* log_buffer);
429
  void CleanupCompaction(CompactionState* compact, Status status);
I
Igor Canadi 已提交
430
  Status DoCompactionWork(CompactionState* compact,
H
Haobo Xu 已提交
431 432
                          DeletionState& deletion_state,
                          LogBuffer* log_buffer);
J
jorlow@chromium.org 已提交
433

I
Igor Canadi 已提交
434 435 436 437 438 439 440
  // This function is called as part of compaction. It enables Flush process to
  // preempt compaction, since it's higher prioirty
  // Returns: micros spent executing
  uint64_t CallFlushDuringCompaction(ColumnFamilyData* cfd,
                                     DeletionState& deletion_state,
                                     LogBuffer* log_buffer);

D
Danny Guo 已提交
441 442 443
  // Call compaction filter if is_compaction_v2 is not true. Then iterate
  // through input and compact the kv-pairs
  Status ProcessKeyValueCompaction(
444
    bool is_snapshot_supported,
D
Danny Guo 已提交
445 446 447 448 449 450 451 452 453 454 455 456 457 458 459
    SequenceNumber visible_at_tip,
    SequenceNumber earliest_snapshot,
    SequenceNumber latest_snapshot,
    DeletionState& deletion_state,
    bool bottommost_level,
    int64_t& imm_micros,
    Iterator* input,
    CompactionState* compact,
    bool is_compaction_v2,
    LogBuffer* log_buffer);

  // Call compaction_filter_v2->Filter() on kv-pairs in compact
  void CallCompactionFilterV2(CompactionState* compact,
    CompactionFilterV2* compaction_filter_v2);

J
jorlow@chromium.org 已提交
460 461
  Status OpenCompactionOutputFile(CompactionState* compact);
  Status FinishCompactionOutputFile(CompactionState* compact, Iterator* input);
462 463
  Status InstallCompactionResults(CompactionState* compact,
                                  LogBuffer* log_buffer);
464 465
  void AllocateCompactionOutputFileNumbers(CompactionState* compact);
  void ReleaseCompactionUnusedFileNumbers(CompactionState* compact);
A
Abhishek Kona 已提交
466

I
Igor Canadi 已提交
467 468 469 470 471 472
#ifdef ROCKSDB_LITE
  void PurgeObsoleteWALFiles() {
    // this function is used for archiving WAL files. we don't need this in
    // ROCKSDB_LITE
  }
#else
473
  void PurgeObsoleteWALFiles();
474

475 476 477
  Status GetSortedWalsOfType(const std::string& path,
                             VectorLogPtr& log_files,
                             WalFileType type);
478

479 480 481 482 483
  // Requires: all_logs should be sorted with earliest log file first
  // Retains all log files in all_logs which contain updates with seq no.
  // Greater Than or Equal to the requested SequenceNumber.
  Status RetainProbableWalFiles(VectorLogPtr& all_logs,
                                const SequenceNumber target);
484

485
  Status ReadFirstRecord(const WalFileType type, const uint64_t number,
I
Igor Canadi 已提交
486
                         SequenceNumber* sequence);
487

I
Igor Canadi 已提交
488
  Status ReadFirstLine(const std::string& fname, SequenceNumber* sequence);
I
Igor Canadi 已提交
489
#endif  // ROCKSDB_LITE
490

491 492
  void PrintStatistics();

493
  // dump rocksdb.stats to LOG
494 495
  void MaybeDumpStats();

496 497 498 499 500 501 502
  // Return true if the current db supports snapshot.  If the current
  // DB does not support snapshot, then calling GetSnapshot() will always
  // return nullptr.
  //
  // @see GetSnapshot()
  virtual bool IsSnapshotSupported() const;

503 504
  // Return the minimum empty level that could hold the total data in the
  // input level. Return the input level, if such level could not be found.
I
Igor Canadi 已提交
505
  int FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd, int level);
506

507 508 509
  // Move the files in the input level to the target level.
  // If target_level < 0, automatically calculate the minimum level that could
  // hold the data set.
I
Igor Canadi 已提交
510
  Status ReFitLevel(ColumnFamilyData* cfd, int level, int target_level = -1);
511

J
jorlow@chromium.org 已提交
512
  // table_cache_ provides its own synchronization
I
Igor Canadi 已提交
513
  std::shared_ptr<Cache> table_cache_;
J
jorlow@chromium.org 已提交
514

515
  // Lock over the persistent DB state.  Non-nullptr iff successfully acquired.
J
jorlow@chromium.org 已提交
516 517 518 519 520
  FileLock* db_lock_;

  // State below is protected by mutex_
  port::Mutex mutex_;
  port::AtomicPointer shutting_down_;
521 522 523 524 525 526 527 528 529
  // This condition variable is signaled on these conditions:
  // * whenever bg_compaction_scheduled_ goes down to 0
  // * if bg_manual_only_ > 0, whenever a compaction finishes, even if it hasn't
  // made any progress
  // * whenever a compaction made any progress
  // * whenever bg_flush_scheduled_ value decreases (i.e. whenever a flush is
  // done, even if it didn't make any progress)
  // * whenever there is an error in background flush or compaction
  port::CondVar bg_cv_;
530
  uint64_t logfile_number_;
531
  unique_ptr<log::Writer> log_;
I
Igor Canadi 已提交
532
  bool log_empty_;
533
  ColumnFamilyHandleImpl* default_cf_handle_;
534
  InternalStats* default_cf_internal_stats_;
535
  unique_ptr<ColumnFamilyMemTablesImpl> column_family_memtables_;
I
Igor Canadi 已提交
536 537 538 539 540 541 542 543 544 545 546 547 548
  struct LogFileNumberSize {
    explicit LogFileNumberSize(uint64_t _number)
        : number(_number), size(0), getting_flushed(false) {}
    void AddSize(uint64_t new_size) { size += new_size; }
    uint64_t number;
    uint64_t size;
    bool getting_flushed;
  };
  std::deque<LogFileNumberSize> alive_log_files_;
  uint64_t total_log_size_;
  // only used for dynamically adjusting max_total_wal_size. it is a sum of
  // [write_buffer_size * max_write_buffer_number] over all column families
  uint64_t max_total_in_memory_state_;
549 550 551
  // If true, we have only one (default) column family. We use this to optimize
  // some code-paths
  bool single_column_family_mode_;
I
Igor Canadi 已提交
552

553 554
  std::unique_ptr<Directory> db_directory_;

555 556
  // Queue of writers.
  std::deque<Writer*> writers_;
557
  WriteBatch tmp_batch_;
558

559
  WriteController write_controller_;
I
Igor Canadi 已提交
560
  FlushScheduler flush_scheduler_;
561

J
jorlow@chromium.org 已提交
562 563
  SnapshotList snapshots_;

I
Igor Canadi 已提交
564 565 566 567
  // cache for ReadFirstRecord() calls
  std::unordered_map<uint64_t, SequenceNumber> read_first_record_cache_;
  port::Mutex read_first_record_cache_mutex_;

J
jorlow@chromium.org 已提交
568 569
  // Set of table files to protect from deletion because they are
  // part of ongoing compactions.
570 571
  // map from pending file number ID to their path IDs.
  FileNumToPathIdMap pending_outputs_;
J
jorlow@chromium.org 已提交
572

573 574 575 576
  // At least one compaction or flush job is pending but not yet scheduled
  // because of the max background thread limit.
  bool bg_schedule_needed_;

577
  // count how many background compactions are running or have been scheduled
578
  int bg_compaction_scheduled_;
J
jorlow@chromium.org 已提交
579

580 581 582 583 584
  // If non-zero, MaybeScheduleFlushOrCompaction() will only schedule manual
  // compactions (if manual_compaction_ is not null). This mechanism enables
  // manual compactions to wait until all other compactions are finished.
  int bg_manual_only_;

585 586 587
  // number of background memtable flush jobs, submitted to the HIGH pool
  int bg_flush_scheduled_;

H
hans@chromium.org 已提交
588 589
  // Information for a manual compaction
  struct ManualCompaction {
I
Igor Canadi 已提交
590
    ColumnFamilyData* cfd;
591 592
    int input_level;
    int output_level;
593
    uint32_t output_path_id;
G
Gabor Cselle 已提交
594
    bool done;
L
Lei Jin 已提交
595
    Status status;
596
    bool in_progress;           // compaction request being processed?
597 598
    const InternalKey* begin;   // nullptr means beginning of key range
    const InternalKey* end;     // nullptr means end of key range
G
Gabor Cselle 已提交
599
    InternalKey tmp_storage;    // Used to keep track of compaction progress
H
hans@chromium.org 已提交
600 601
  };
  ManualCompaction* manual_compaction_;
J
jorlow@chromium.org 已提交
602 603 604 605

  // Have we encountered a background error in paranoid mode?
  Status bg_error_;

606
  // shall we disable deletion of obsolete files
607 608 609 610 611 612
  // if 0 the deletion is enabled.
  // if non-zero, files will not be getting deleted
  // This enables two different threads to call
  // EnableFileDeletions() and DisableFileDeletions()
  // without any synchronization
  int disable_delete_obsolete_files_;
613

614 615 616
  // last time when DeleteObsoleteFiles was invoked
  uint64_t delete_obsolete_files_last_run_;

617 618 619
  // last time when PurgeObsoleteWALFiles ran.
  uint64_t purge_wal_files_last_run_;

620
  // last time stats were dumped to LOG
H
Haobo Xu 已提交
621
  std::atomic<uint64_t> last_stats_dump_time_microsec_;
622

623 624 625 626
  // obsolete files will be deleted every this seconds if ttl deletion is
  // enabled and archive size_limit is disabled.
  uint64_t default_interval_to_delete_obsolete_WAL_;

627 628
  bool flush_on_destroy_; // Used when disableWAL is true.

H
heyongqiang 已提交
629
  static const int KEEP_LOG_FILE_NUM = 1000;
630
  static const uint64_t kNoTimeOut = std::numeric_limits<uint64_t>::max();
H
heyongqiang 已提交
631
  std::string db_absolute_path_;
H
heyongqiang 已提交
632

633
  // The options to access storage files
L
Lei Jin 已提交
634
  const EnvOptions env_options_;
635

636 637 638 639 640 641
  // A value of true temporarily disables scheduling of background work
  bool bg_work_gate_closed_;

  // Guard against multiple concurrent refitting
  bool refitting_level_;

642 643 644
  // Indicate DB was opened successfully
  bool opened_successfully_;

J
jorlow@chromium.org 已提交
645 646 647 648
  // No copying allowed
  DBImpl(const DBImpl&);
  void operator=(const DBImpl&);

649 650 651 652 653 654
  // Return the earliest snapshot where seqno is visible.
  // Store the snapshot right before that, if any, in prev_snapshot
  inline SequenceNumber findEarliestVisibleSnapshot(
    SequenceNumber in,
    std::vector<SequenceNumber>& snapshots,
    SequenceNumber* prev_snapshot);
655

I
Igor Canadi 已提交
656
  // Background threads call this function, which is just a wrapper around
657
  // the cfd->InstallSuperVersion() function. Background threads carry
I
Igor Canadi 已提交
658
  // deletion_state which can have new_superversion already allocated.
659 660
  void InstallSuperVersion(ColumnFamilyData* cfd,
                           DeletionState& deletion_state);
I
Igor Canadi 已提交
661

662 663 664 665 666 667 668 669 670 671
  // Find Super version and reference it. Based on options, it might return
  // the thread local cached one.
  inline SuperVersion* GetAndRefSuperVersion(ColumnFamilyData* cfd);

  // Un-reference the super version and return it to thread local cache if
  // needed. If it is the last reference of the super version. Clean it up
  // after un-referencing it.
  inline void ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd,
                                           SuperVersion* sv);

I
Igor Canadi 已提交
672
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
673 674 675
  using DB::GetPropertiesOfAllTables;
  virtual Status GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                          TablePropertiesCollection* props)
676
      override;
I
Igor Canadi 已提交
677
#endif  // ROCKSDB_LITE
678

679 680
  // Function that Get and KeyMayExist call with no_io true or false
  // Note: 'value_found' from KeyMayExist propagates here
681 682 683
  Status GetImpl(const ReadOptions& options, ColumnFamilyHandle* column_family,
                 const Slice& key, std::string* value,
                 bool* value_found = nullptr);
684 685 686 687

  bool GetIntPropertyInternal(ColumnFamilyHandle* column_family,
                              DBPropertyType property_type,
                              bool need_out_of_mutex, uint64_t* value);
J
jorlow@chromium.org 已提交
688 689 690 691 692 693 694
};

// Sanitize db options.  The caller should delete result.info_log if
// it is not equal to src.info_log.
extern Options SanitizeOptions(const std::string& db,
                               const InternalKeyComparator* icmp,
                               const Options& src);
695
extern DBOptions SanitizeOptions(const std::string& db, const DBOptions& src);
S
Siying Dong 已提交
696

M
miguelportilla 已提交
697 698 699 700 701 702 703
// Fix user-supplied options to be reasonable
template <class T, class V>
static void ClipToRange(T* ptr, V minvalue, V maxvalue) {
  if (static_cast<V>(*ptr) > maxvalue) *ptr = maxvalue;
  if (static_cast<V>(*ptr) < minvalue) *ptr = minvalue;
}

F
Feng Zhu 已提交
704 705 706 707
// Dump db file summary, implemented in util/
extern void DumpDBFileSummary(const DBOptions& options,
                              const std::string& dbname);

708
}  // namespace rocksdb