db_impl.h 32.3 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>
I
Igor Canadi 已提交
14
#include <list>
15
#include <set>
I
Igor Canadi 已提交
16
#include <string>
17 18
#include <utility>
#include <vector>
K
kailiu 已提交
19

20
#include "db/column_family.h"
21
#include "db/compaction_job.h"
22
#include "db/dbformat.h"
23
#include "db/flush_job.h"
24 25
#include "db/flush_scheduler.h"
#include "db/internal_stats.h"
A
agiardullo 已提交
26 27
#include "db/log_writer.h"
#include "db/snapshot_impl.h"
28
#include "db/version_edit.h"
I
Igor Canadi 已提交
29
#include "db/wal_manager.h"
30 31
#include "db/write_controller.h"
#include "db/write_thread.h"
32
#include "db/writebuffer.h"
K
Kai Liu 已提交
33 34
#include "memtable_list.h"
#include "port/port.h"
35 36 37 38
#include "rocksdb/db.h"
#include "rocksdb/env.h"
#include "rocksdb/memtablerep.h"
#include "rocksdb/transaction_log.h"
39
#include "util/autovector.h"
I
Igor Canadi 已提交
40 41
#include "util/event_logger.h"
#include "util/hash.h"
42
#include "util/instrumented_mutex.h"
43 44 45
#include "util/scoped_arena_iterator.h"
#include "util/stop_watch.h"
#include "util/thread_local.h"
46

47
namespace rocksdb {
J
jorlow@chromium.org 已提交
48 49 50 51 52 53

class MemTable;
class TableCache;
class Version;
class VersionEdit;
class VersionSet;
D
Danny Guo 已提交
54
class CompactionFilterV2;
55
class Arena;
A
agiardullo 已提交
56
class WriteCallback;
I
Igor Canadi 已提交
57
struct JobContext;
58
struct ExternalSstFileInfo;
J
jorlow@chromium.org 已提交
59 60 61

class DBImpl : public DB {
 public:
I
Igor Canadi 已提交
62
  DBImpl(const DBOptions& options, const std::string& dbname);
J
jorlow@chromium.org 已提交
63 64 65
  virtual ~DBImpl();

  // Implementations of the DB interface
66 67
  using DB::Put;
  virtual Status Put(const WriteOptions& options,
68
                     ColumnFamilyHandle* column_family, const Slice& key,
I
Igor Sugak 已提交
69
                     const Slice& value) override;
70 71
  using DB::Merge;
  virtual Status Merge(const WriteOptions& options,
72
                       ColumnFamilyHandle* column_family, const Slice& key,
I
Igor Sugak 已提交
73
                       const Slice& value) override;
74 75
  using DB::Delete;
  virtual Status Delete(const WriteOptions& options,
I
Igor Sugak 已提交
76 77
                        ColumnFamilyHandle* column_family,
                        const Slice& key) override;
A
Andres Noetzli 已提交
78 79 80 81
  using DB::SingleDelete;
  virtual Status SingleDelete(const WriteOptions& options,
                              ColumnFamilyHandle* column_family,
                              const Slice& key) override;
82
  using DB::Write;
I
Igor Sugak 已提交
83 84
  virtual Status Write(const WriteOptions& options,
                       WriteBatch* updates) override;
A
agiardullo 已提交
85

86
  using DB::Get;
J
jorlow@chromium.org 已提交
87
  virtual Status Get(const ReadOptions& options,
88
                     ColumnFamilyHandle* column_family, const Slice& key,
I
Igor Sugak 已提交
89
                     std::string* value) override;
90 91 92
  using DB::MultiGet;
  virtual std::vector<Status> MultiGet(
      const ReadOptions& options,
93
      const std::vector<ColumnFamilyHandle*>& column_family,
I
Igor Sugak 已提交
94 95
      const std::vector<Slice>& keys,
      std::vector<std::string>* values) override;
96

97
  virtual Status CreateColumnFamily(const ColumnFamilyOptions& options,
98
                                    const std::string& column_family,
I
Igor Sugak 已提交
99 100
                                    ColumnFamilyHandle** handle) override;
  virtual Status DropColumnFamily(ColumnFamilyHandle* column_family) override;
101

102 103 104 105
  // 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.
106
  using DB::KeyMayExist;
107
  virtual bool KeyMayExist(const ReadOptions& options,
108
                           ColumnFamilyHandle* column_family, const Slice& key,
I
Igor Sugak 已提交
109 110
                           std::string* value,
                           bool* value_found = nullptr) override;
111 112
  using DB::NewIterator;
  virtual Iterator* NewIterator(const ReadOptions& options,
I
Igor Sugak 已提交
113
                                ColumnFamilyHandle* column_family) override;
114 115
  virtual Status NewIterators(
      const ReadOptions& options,
I
Igor Canadi 已提交
116
      const std::vector<ColumnFamilyHandle*>& column_families,
I
Igor Sugak 已提交
117 118 119
      std::vector<Iterator*>* iterators) override;
  virtual const Snapshot* GetSnapshot() override;
  virtual void ReleaseSnapshot(const Snapshot* snapshot) override;
120
  using DB::GetProperty;
121
  virtual bool GetProperty(ColumnFamilyHandle* column_family,
I
Igor Sugak 已提交
122
                           const Slice& property, std::string* value) override;
123 124 125
  using DB::GetIntProperty;
  virtual bool GetIntProperty(ColumnFamilyHandle* column_family,
                              const Slice& property, uint64_t* value) override;
126
  using DB::GetApproximateSizes;
127
  virtual void GetApproximateSizes(ColumnFamilyHandle* column_family,
128 129
                                   const Range* range, int n, uint64_t* sizes,
                                   bool include_memtable = false) override;
130
  using DB::CompactRange;
131 132 133
  virtual Status CompactRange(const CompactRangeOptions& options,
                              ColumnFamilyHandle* column_family,
                              const Slice* begin, const Slice* end) override;
134

135
  using DB::CompactFiles;
I
Igor Sugak 已提交
136 137 138 139 140
  virtual Status CompactFiles(const CompactionOptions& compact_options,
                              ColumnFamilyHandle* column_family,
                              const std::vector<std::string>& input_file_names,
                              const int output_level,
                              const int output_path_id = -1) override;
141

142
  using DB::SetOptions;
I
Igor Sugak 已提交
143 144 145
  Status SetOptions(
      ColumnFamilyHandle* column_family,
      const std::unordered_map<std::string, std::string>& options_map) override;
146

147
  using DB::NumberLevels;
I
Igor Sugak 已提交
148
  virtual int NumberLevels(ColumnFamilyHandle* column_family) override;
149
  using DB::MaxMemCompactionLevel;
I
Igor Sugak 已提交
150
  virtual int MaxMemCompactionLevel(ColumnFamilyHandle* column_family) override;
151
  using DB::Level0StopWriteTrigger;
I
Igor Sugak 已提交
152 153 154 155
  virtual int Level0StopWriteTrigger(
      ColumnFamilyHandle* column_family) override;
  virtual const std::string& GetName() const override;
  virtual Env* GetEnv() const override;
156
  using DB::GetOptions;
I
Igor Sugak 已提交
157 158
  virtual const Options& GetOptions(
      ColumnFamilyHandle* column_family) const override;
159 160
  using DB::GetDBOptions;
  virtual const DBOptions& GetDBOptions() const override;
161 162
  using DB::Flush;
  virtual Status Flush(const FlushOptions& options,
I
Igor Sugak 已提交
163
                       ColumnFamilyHandle* column_family) override;
164
  virtual Status SyncWAL() override;
I
Igor Canadi 已提交
165

I
Igor Sugak 已提交
166
  virtual SequenceNumber GetLatestSequenceNumber() const override;
I
Igor Canadi 已提交
167 168

#ifndef ROCKSDB_LITE
I
Igor Sugak 已提交
169 170
  virtual Status DisableFileDeletions() override;
  virtual Status EnableFileDeletions(bool force) override;
171
  virtual int IsFileDeletionsEnabled() const;
I
Igor Canadi 已提交
172
  // All the returned filenames start with "/"
173
  virtual Status GetLiveFiles(std::vector<std::string>&,
174
                              uint64_t* manifest_file_size,
I
Igor Sugak 已提交
175 176
                              bool flush_memtable = true) override;
  virtual Status GetSortedWalFiles(VectorLogPtr& files) override;
I
Igor Canadi 已提交
177

178 179 180
  virtual Status GetUpdatesSince(
      SequenceNumber seq_number, unique_ptr<TransactionLogIterator>* iter,
      const TransactionLogIterator::ReadOptions&
I
Igor Sugak 已提交
181 182
          read_options = TransactionLogIterator::ReadOptions()) override;
  virtual Status DeleteFile(std::string name) override;
183

I
Igor Sugak 已提交
184 185
  virtual void GetLiveFilesMetaData(
      std::vector<LiveFileMetaData>* metadata) override;
186 187 188 189 190 191 192 193 194

  // Obtains the meta data of the specified column family of the DB.
  // Status::NotFound() will be returned if the current DB does not have
  // any column family match the specified name.
  // TODO(yhchiang): output parameter is placed in the end in this codebase.
  virtual void GetColumnFamilyMetaData(
      ColumnFamilyHandle* column_family,
      ColumnFamilyMetaData* metadata) override;

195 196 197 198
  // experimental API
  Status SuggestCompactRange(ColumnFamilyHandle* column_family,
                             const Slice* begin, const Slice* end);

199 200
  Status PromoteL0(ColumnFamilyHandle* column_family, int target_level);

A
agiardullo 已提交
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228
  // Similar to Write() but will call the callback once on the single write
  // thread to determine whether it is safe to perform the write.
  virtual Status WriteWithCallback(const WriteOptions& write_options,
                                   WriteBatch* my_batch,
                                   WriteCallback* callback);

  // Returns the sequence number that is guaranteed to be smaller than or equal
  // to the sequence number of any key that could be inserted into the current
  // memtables. It can then be assumed that any write with a larger(or equal)
  // sequence number will be present in this memtable or a later memtable.
  //
  // If the earliest sequence number could not be determined,
  // kMaxSequenceNumber will be returned.
  //
  // If include_history=true, will also search Memtables in MemTableList
  // History.
  SequenceNumber GetEarliestMemTableSequenceNumber(SuperVersion* sv,
                                                   bool include_history);

  // For a given key, check to see if there are any records for this key
  // in the memtables, including memtable history.

  // On success, *seq will contain the sequence number for the
  // latest such change or kMaxSequenceNumber if no records were present.
  // Returns OK on success, other status on error reading memtables.
  Status GetLatestSequenceForKeyFromMemtable(SuperVersion* sv, const Slice& key,
                                             SequenceNumber* seq);

229 230 231 232 233 234 235
  using DB::AddFile;
  virtual Status AddFile(ColumnFamilyHandle* column_family,
                         const ExternalSstFileInfo* file_info,
                         bool move_file) override;
  virtual Status AddFile(ColumnFamilyHandle* column_family,
                         const std::string& file_path, bool move_file) override;

I
Igor Canadi 已提交
236
#endif  // ROCKSDB_LITE
237

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

242
  virtual Status GetDbIdentity(std::string& identity) const override;
243

I
Igor Canadi 已提交
244
  Status RunManualCompaction(ColumnFamilyData* cfd, int input_level,
245
                             int output_level, uint32_t output_path_id,
246 247
                             const Slice* begin, const Slice* end,
                             bool disallow_trivial_move = false);
248

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

253
  // Compact any files in the named level that overlap [*begin, *end]
254
  Status TEST_CompactRange(int level, const Slice* begin, const Slice* end,
255 256
                           ColumnFamilyHandle* column_family = nullptr,
                           bool disallow_trivial_move = false);
J
jorlow@chromium.org 已提交
257

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

261
  // Wait for memtable compaction
262
  Status TEST_WaitForFlushMemTable(ColumnFamilyHandle* column_family = nullptr);
263 264 265 266

  // Wait for any compaction
  Status TEST_WaitForCompact();

J
jorlow@chromium.org 已提交
267 268 269
  // 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.
270 271
  Iterator* TEST_NewInternalIterator(
      Arena* arena, ColumnFamilyHandle* column_family = nullptr);
J
jorlow@chromium.org 已提交
272

273 274
  // Return the maximum overlapping data (in bytes) at next level for any
  // file at a level >= 1.
275 276
  int64_t TEST_MaxNextLevelOverlappingBytes(ColumnFamilyHandle* column_family =
                                                nullptr);
277

A
Abhishek Kona 已提交
278 279
  // Return the current manifest file no.
  uint64_t TEST_Current_Manifest_FileNo();
280

281
  // get total level0 file size. Only for testing.
282
  uint64_t TEST_GetLevel0TotalSize();
283

284 285
  void TEST_GetFilesMetaData(ColumnFamilyHandle* column_family,
                             std::vector<std::vector<FileMetaData>>* metadata);
286

287 288 289 290 291 292 293 294 295 296
  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);
297

298
  uint64_t TEST_MaxTotalInMemoryState() const {
299 300
    return max_total_in_memory_state_;
  }
301

302 303
  size_t TEST_LogsToFreeSize();

304 305
  uint64_t TEST_LogfileNumber();

306
#endif  // ROCKSDB_LITE
I
Igor Canadi 已提交
307

I
Igor Canadi 已提交
308
  // Returns the list of live files in 'live' and the list
K
kailiu 已提交
309
  // of all files in the filesystem in 'candidate_files'.
I
Igor Canadi 已提交
310
  // If force == false and the last call was less than
311
  // db_options_.delete_obsolete_files_period_micros microseconds ago,
I
Igor Canadi 已提交
312 313
  // it will not fill up the job_context
  void FindObsoleteFiles(JobContext* job_context, bool force,
I
Igor Canadi 已提交
314 315 316 317 318 319
                         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.
I
Igor Canadi 已提交
320
  void PurgeObsoleteFiles(const JobContext& background_contet);
I
Igor Canadi 已提交
321

I
Igor Sugak 已提交
322
  ColumnFamilyHandle* DefaultColumnFamily() const override;
323

324 325
  const SnapshotList& snapshots() const { return snapshots_; }

326
  void CancelAllBackgroundWork(bool wait);
327

A
agiardullo 已提交
328 329 330 331 332 333 334 335 336 337 338
  // Find Super version and reference it. Based on options, it might return
  // the thread local cached one.
  // Call ReturnAndCleanupSuperVersion() when it is no longer needed.
  SuperVersion* GetAndRefSuperVersion(ColumnFamilyData* cfd);

  // Similar to the previous function but looks up based on a column family id.
  // nullptr will be returned if this column family no longer exists.
  // REQUIRED: this function should only be called on the write thread or if the
  // mutex is held.
  SuperVersion* GetAndRefSuperVersion(uint32_t column_family_id);

A
agiardullo 已提交
339 340 341
  // Same as above, should called without mutex held and not on write thread.
  SuperVersion* GetAndRefSuperVersionUnlocked(uint32_t column_family_id);

A
agiardullo 已提交
342 343 344 345 346 347 348 349 350 351
  // 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.
  void ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd, SuperVersion* sv);

  // Similar to the previous function but looks up based on a column family id.
  // nullptr will be returned if this column family no longer exists.
  // REQUIRED: this function should only be called on the write thread.
  void ReturnAndCleanupSuperVersion(uint32_t colun_family_id, SuperVersion* sv);

A
agiardullo 已提交
352 353 354 355
  // Same as above, should called without mutex held and not on write thread.
  void ReturnAndCleanupSuperVersionUnlocked(uint32_t colun_family_id,
                                            SuperVersion* sv);

A
agiardullo 已提交
356 357 358 359 360
  // REQUIRED: this function should only be called on the write thread or if the
  // mutex is held.  Return value only valid until next call to this function or
  // mutex is released.
  ColumnFamilyHandle* GetColumnFamilyHandle(uint32_t column_family_id);

A
agiardullo 已提交
361 362 363
  // Same as above, should called without mutex held and not on write thread.
  ColumnFamilyHandle* GetColumnFamilyHandleUnlocked(uint32_t column_family_id);

364
 protected:
H
heyongqiang 已提交
365 366
  Env* const env_;
  const std::string dbname_;
367
  unique_ptr<VersionSet> versions_;
368
  const DBOptions db_options_;
L
Lei Jin 已提交
369
  Statistics* stats_;
H
heyongqiang 已提交
370

371
  Iterator* NewInternalIterator(const ReadOptions&, ColumnFamilyData* cfd,
372
                                SuperVersion* super_version, Arena* arena);
373

374
  void NotifyOnFlushCompleted(ColumnFamilyData* cfd, FileMetaData* file_meta,
375 376
                              const MutableCFOptions& mutable_cf_options,
                              int job_id);
377

O
Ori Bernstein 已提交
378
  void NotifyOnCompactionCompleted(ColumnFamilyData* cfd,
379 380
                                   Compaction *c, const Status &st,
                                   const CompactionJobStats& job_stats,
381
                                   int job_id);
O
Ori Bernstein 已提交
382

Y
Yueh-Hsuan Chiang 已提交
383 384 385 386 387 388
  void NewThreadStatusCfInfo(ColumnFamilyData* cfd) const;

  void EraseThreadStatusCfInfo(ColumnFamilyData* cfd) const;

  void EraseThreadStatusDbInfo() const;

A
agiardullo 已提交
389 390 391
  Status WriteImpl(const WriteOptions& options, WriteBatch* updates,
                   WriteCallback* callback);

J
jorlow@chromium.org 已提交
392 393
 private:
  friend class DB;
394
  friend class InternalStats;
I
Igor Canadi 已提交
395
#ifndef ROCKSDB_LITE
L
Lei Jin 已提交
396
  friend class ForwardIterator;
I
Igor Canadi 已提交
397
#endif
398
  friend struct SuperVersion;
L
Lei Jin 已提交
399
  friend class CompactedDBImpl;
A
agiardullo 已提交
400 401 402
#ifndef NDEBUG
  friend class XFTransactionWriteHandler;
#endif
403
  struct CompactionState;
404

S
Stanislau Hlebik 已提交
405
  struct WriteContext;
J
jorlow@chromium.org 已提交
406 407 408 409 410 411

  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.
412 413
  Status Recover(const std::vector<ColumnFamilyDescriptor>& column_families,
                 bool read_only = false, bool error_if_log_file_exist = false);
J
jorlow@chromium.org 已提交
414 415 416

  void MaybeIgnoreError(Status* s) const;

417 418
  const Status CreateArchivalDirectory();

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

I
Igor Canadi 已提交
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
  // Background process needs to call
  //     auto x = CaptureCurrentFileNumberInPendingOutputs()
  //     <do something>
  //     ReleaseFileNumberFromPendingOutputs(x)
  // This will protect any temporary files created while <do something> is
  // executing from being deleted.
  // -----------
  // This function will capture current file number and append it to
  // pending_outputs_. This will prevent any background process to delete any
  // file created after this point.
  std::list<uint64_t>::iterator CaptureCurrentFileNumberInPendingOutputs();
  // This function should be called with the result of
  // CaptureCurrentFileNumberInPendingOutputs(). It then marks that any file
  // created between the calls CaptureCurrentFileNumberInPendingOutputs() and
  // ReleaseFileNumberFromPendingOutputs() can now be deleted (if it's not live
  // and blocked by any other pending_outputs_ calls)
  void ReleaseFileNumberFromPendingOutputs(std::list<uint64_t>::iterator v);

440
  // Flush the in-memory write buffer to storage.  Switches to a new
J
jorlow@chromium.org 已提交
441
  // log-file/memtable and writes a new descriptor iff successful.
I
Igor Canadi 已提交
442 443 444 445
  Status FlushMemTableToOutputFile(ColumnFamilyData* cfd,
                                   const MutableCFOptions& mutable_cf_options,
                                   bool* madeProgress, JobContext* job_context,
                                   LogBuffer* log_buffer);
J
jorlow@chromium.org 已提交
446

S
Stanislau Hlebik 已提交
447 448 449
  // 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 已提交
450

451 452 453 454 455
  // 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.
456 457
  Status WriteLevel0TableForRecovery(int job_id, ColumnFamilyData* cfd,
                                     MemTable* mem, VersionEdit* edit);
S
sdong 已提交
458 459 460

  // num_bytes: for slowdown case, delay time is calculated based on
  //            `num_bytes` going through.
461
  Status DelayWrite(uint64_t num_bytes);
462

I
Igor Canadi 已提交
463
  Status ScheduleFlushes(WriteContext* context);
464

I
Igor Canadi 已提交
465
  Status SwitchMemtable(ColumnFamilyData* cfd, WriteContext* context);
S
Stanislau Hlebik 已提交
466

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

470
  // Wait for memtable flushed
471
  Status WaitForFlushMemTable(ColumnFamilyData* cfd);
H
heyongqiang 已提交
472

473 474 475
  void RecordFlushIOStats();
  void RecordCompactionIOStats();

I
Igor Canadi 已提交
476
#ifndef ROCKSDB_LITE
477 478 479
  Status CompactFilesImpl(
      const CompactionOptions& compact_options, ColumnFamilyData* cfd,
      Version* version, const std::vector<std::string>& input_file_names,
480 481
      const int output_level, int output_path_id, JobContext* job_context,
      LogBuffer* log_buffer);
I
Igor Canadi 已提交
482
#endif  // ROCKSDB_LITE
483 484 485

  ColumnFamilyData* GetColumnFamilyDataByName(const std::string& cf_name);

486
  void MaybeScheduleFlushOrCompaction();
487 488
  void SchedulePendingFlush(ColumnFamilyData* cfd);
  void SchedulePendingCompaction(ColumnFamilyData* cfd);
489 490 491 492
  static void BGWorkCompaction(void* db);
  static void BGWorkFlush(void* db);
  void BackgroundCallCompaction();
  void BackgroundCallFlush();
I
Igor Canadi 已提交
493
  Status BackgroundCompaction(bool* madeProgress, JobContext* job_context,
494
                              LogBuffer* log_buffer);
I
Igor Canadi 已提交
495
  Status BackgroundFlush(bool* madeProgress, JobContext* job_context,
H
Haobo Xu 已提交
496
                         LogBuffer* log_buffer);
J
jorlow@chromium.org 已提交
497

498 499
  void PrintStatistics();

500
  // dump rocksdb.stats to LOG
501 502
  void MaybeDumpStats();

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.
505 506
  int FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd,
      const MutableCFOptions& mutable_cf_options, int level);
507

508 509 510
  // 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 已提交
511
  Status ReFitLevel(ColumnFamilyData* cfd, int level, int target_level = -1);
512

513 514 515 516 517 518
  // helper functions for adding and removing from flush & compaction queues
  void AddToCompactionQueue(ColumnFamilyData* cfd);
  ColumnFamilyData* PopFirstFromCompactionQueue();
  void AddToFlushQueue(ColumnFamilyData* cfd);
  ColumnFamilyData* PopFirstFromFlushQueue();

519 520 521
  // helper function to call after some of the logs_ were synced
  void MarkLogsSynced(uint64_t up_to, bool synced_dir, const Status& status);

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

525
  // Lock over the persistent DB state.  Non-nullptr iff successfully acquired.
J
jorlow@chromium.org 已提交
526 527 528
  FileLock* db_lock_;

  // State below is protected by mutex_
529
  InstrumentedMutex mutex_;
I
Igor Canadi 已提交
530
  std::atomic<bool> shutting_down_;
531 532 533 534 535 536 537 538
  // 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
539
  InstrumentedCondVar bg_cv_;
540
  uint64_t logfile_number_;
541
  bool log_dir_synced_;
I
Igor Canadi 已提交
542
  bool log_empty_;
543
  ColumnFamilyHandleImpl* default_cf_handle_;
544
  InternalStats* default_cf_internal_stats_;
545
  unique_ptr<ColumnFamilyMemTablesImpl> column_family_memtables_;
I
Igor Canadi 已提交
546 547
  struct LogFileNumberSize {
    explicit LogFileNumberSize(uint64_t _number)
548
        : number(_number) {}
I
Igor Canadi 已提交
549 550
    void AddSize(uint64_t new_size) { size += new_size; }
    uint64_t number;
551 552 553 554
    uint64_t size = 0;
    bool getting_flushed = false;
  };
  struct LogWriterNumber {
555 556 557 558 559 560 561 562 563 564 565 566 567 568
    // pass ownership of _writer
    LogWriterNumber(uint64_t _number, log::Writer* _writer)
        : number(_number), writer(_writer) {}

    log::Writer* ReleaseWriter() {
      auto* w = writer;
      writer = nullptr;
      return w;
    }
    void ClearWriter() {
      delete writer;
      writer = nullptr;
    }

569
    uint64_t number;
570 571 572
    // Visual Studio doesn't support deque's member to be noncopyable because
    // of a unique_ptr as a member.
    log::Writer* writer;  // own
573 574
    // true for some prefix of logs_
    bool getting_synced = false;
I
Igor Canadi 已提交
575 576
  };
  std::deque<LogFileNumberSize> alive_log_files_;
577 578 579 580 581 582 583 584 585 586
  // Log files that aren't fully synced, and the current log file.
  // Synchronization:
  //  - push_back() is done from write thread with locked mutex_,
  //  - pop_front() is done from any thread with locked mutex_,
  //  - back() and items with getting_synced=true are not popped,
  //  - it follows that write thread with unlocked mutex_ can safely access
  //    back() and items with getting_synced=true.
  std::deque<LogWriterNumber> logs_;
  // Signaled when getting_synced becomes false for some of the logs_.
  InstrumentedCondVar log_sync_cv_;
I
Igor Canadi 已提交
587 588 589 590
  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_;
591 592 593
  // If true, we have only one (default) column family. We use this to optimize
  // some code-paths
  bool single_column_family_mode_;
594 595 596
  // If this is non-empty, we need to delete these log files in background
  // threads. Protected by db mutex.
  autovector<log::Writer*> logs_to_free_;
I
Igor Canadi 已提交
597

S
sdong 已提交
598 599
  bool is_snapshot_supported_;

600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
  // Class to maintain directories for all database paths other than main one.
  class Directories {
   public:
    Status SetDirectories(Env* env, const std::string& dbname,
                          const std::string& wal_dir,
                          const std::vector<DbPath>& data_paths);

    Directory* GetDataDir(size_t path_id);

    Directory* GetWalDir() {
      if (wal_dir_) {
        return wal_dir_.get();
      }
      return db_dir_.get();
    }

    Directory* GetDbDir() { return db_dir_.get(); }

   private:
    std::unique_ptr<Directory> db_dir_;
    std::vector<std::unique_ptr<Directory>> data_dirs_;
    std::unique_ptr<Directory> wal_dir_;

    Status CreateAndNewDirectory(Env* env, const std::string& dirname,
                                 std::unique_ptr<Directory>* directory) const;
  };

  Directories directories_;
628

629 630
  WriteBuffer write_buffer_;

I
Igor Canadi 已提交
631 632
  WriteThread write_thread_;

633
  WriteBatch tmp_batch_;
634

635
  WriteController write_controller_;
S
sdong 已提交
636 637 638 639 640

  // Size of the last batch group. In slowdown mode, next write needs to
  // sleep if it uses up the quota.
  uint64_t last_batch_group_size_;

I
Igor Canadi 已提交
641
  FlushScheduler flush_scheduler_;
642

J
jorlow@chromium.org 已提交
643 644
  SnapshotList snapshots_;

I
Igor Canadi 已提交
645 646 647 648 649 650 651 652 653 654
  // For each background job, pending_outputs_ keeps the current file number at
  // the time that background job started.
  // FindObsoleteFiles()/PurgeObsoleteFiles() never deletes any file that has
  // number bigger than any of the file number in pending_outputs_. Since file
  // numbers grow monotonically, this also means that pending_outputs_ is always
  // sorted. After a background job is done executing, its file number is
  // deleted from pending_outputs_, which allows PurgeObsoleteFiles() to clean
  // it up.
  // State is protected with db mutex.
  std::list<uint64_t> pending_outputs_;
J
jorlow@chromium.org 已提交
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 681
  // flush_queue_ and compaction_queue_ hold column families that we need to
  // flush and compact, respectively.
  // A column family is inserted into flush_queue_ when it satisfies condition
  // cfd->imm()->IsFlushPending()
  // A column family is inserted into compaction_queue_ when it satisfied
  // condition cfd->NeedsCompaction()
  // Column families in this list are all Ref()-erenced
  // TODO(icanadi) Provide some kind of ReferencedColumnFamily class that will
  // do RAII on ColumnFamilyData
  // Column families are in this queue when they need to be flushed or
  // compacted. Consumers of these queues are flush and compaction threads. When
  // column family is put on this queue, we increase unscheduled_flushes_ and
  // unscheduled_compactions_. When these variables are bigger than zero, that
  // means we need to schedule background threads for compaction and thread.
  // Once the background threads are scheduled, we decrease unscheduled_flushes_
  // and unscheduled_compactions_. That way we keep track of number of
  // compaction and flush threads we need to schedule. This scheduling is done
  // in MaybeScheduleFlushOrCompaction()
  // invariant(column family present in flush_queue_ <==>
  // ColumnFamilyData::pending_flush_ == true)
  std::deque<ColumnFamilyData*> flush_queue_;
  // invariant(column family present in compaction_queue_ <==>
  // ColumnFamilyData::pending_compaction_ == true)
  std::deque<ColumnFamilyData*> compaction_queue_;
  int unscheduled_flushes_;
  int unscheduled_compactions_;
682

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

686 687 688 689 690
  // 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_;

691 692 693
  // number of background memtable flush jobs, submitted to the HIGH pool
  int bg_flush_scheduled_;

H
hans@chromium.org 已提交
694 695
  // Information for a manual compaction
  struct ManualCompaction {
I
Igor Canadi 已提交
696
    ColumnFamilyData* cfd;
697 698
    int input_level;
    int output_level;
699
    uint32_t output_path_id;
G
Gabor Cselle 已提交
700
    bool done;
L
Lei Jin 已提交
701
    Status status;
702 703 704 705 706
    bool in_progress;             // compaction request being processed?
    const InternalKey* begin;     // nullptr means beginning of key range
    const InternalKey* end;       // nullptr means end of key range
    InternalKey tmp_storage;      // Used to keep track of compaction progress
    bool disallow_trivial_move;   // Force actual compaction to run
H
hans@chromium.org 已提交
707 708
  };
  ManualCompaction* manual_compaction_;
J
jorlow@chromium.org 已提交
709 710 711 712

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

713
  // shall we disable deletion of obsolete files
714 715 716 717 718 719
  // 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_;
720

I
Igor Canadi 已提交
721 722
  // next time when we should run DeleteObsoleteFiles with full scan
  uint64_t delete_obsolete_files_next_run_;
723

724
  // last time stats were dumped to LOG
H
Haobo Xu 已提交
725
  std::atomic<uint64_t> last_stats_dump_time_microsec_;
726

727 728 729 730
  // Each flush or compaction gets its own job id. this counter makes sure
  // they're unique
  std::atomic<int> next_job_id_;

731 732
  bool flush_on_destroy_; // Used when disableWAL is true.

H
heyongqiang 已提交
733
  static const int KEEP_LOG_FILE_NUM = 1000;
D
Dmitri Smirnov 已提交
734
  // MSVC version 1800 still does not have constexpr for ::max()
735
  static const uint64_t kNoTimeOut = port::kMaxUint64;
D
Dmitri Smirnov 已提交
736

H
heyongqiang 已提交
737
  std::string db_absolute_path_;
H
heyongqiang 已提交
738

739
  // The options to access storage files
L
Lei Jin 已提交
740
  const EnvOptions env_options_;
741

I
Igor Canadi 已提交
742 743 744 745
#ifndef ROCKSDB_LITE
  WalManager wal_manager_;
#endif  // ROCKSDB_LITE

I
Igor Canadi 已提交
746 747 748
  // Unified interface for logging events
  EventLogger event_logger_;

749 750 751 752 753 754
  // A value of true temporarily disables scheduling of background work
  bool bg_work_gate_closed_;

  // Guard against multiple concurrent refitting
  bool refitting_level_;

755 756 757
  // Indicate DB was opened successfully
  bool opened_successfully_;

J
jorlow@chromium.org 已提交
758 759 760 761
  // No copying allowed
  DBImpl(const DBImpl&);
  void operator=(const DBImpl&);

762 763 764 765 766 767
  // 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);
768

I
Igor Canadi 已提交
769
  // Background threads call this function, which is just a wrapper around
I
Igor Canadi 已提交
770 771 772
  // the InstallSuperVersion() function. Background threads carry
  // job_context which can have new_superversion already
  // allocated.
I
Igor Canadi 已提交
773
  void InstallSuperVersionAndScheduleWorkWrapper(
I
Igor Canadi 已提交
774 775
      ColumnFamilyData* cfd, JobContext* job_context,
      const MutableCFOptions& mutable_cf_options);
L
Lei Jin 已提交
776

777 778 779
  // All ColumnFamily state changes go through this function. Here we analyze
  // the new state and we schedule background work if we detect that the new
  // state needs flush or compaction.
I
Igor Canadi 已提交
780 781 782
  SuperVersion* InstallSuperVersionAndScheduleWork(
      ColumnFamilyData* cfd, SuperVersion* new_sv,
      const MutableCFOptions& mutable_cf_options);
I
Igor Canadi 已提交
783

I
Igor Canadi 已提交
784
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
785 786 787
  using DB::GetPropertiesOfAllTables;
  virtual Status GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                          TablePropertiesCollection* props)
788
      override;
I
Igor Canadi 已提交
789
#endif  // ROCKSDB_LITE
790

791 792
  // Function that Get and KeyMayExist call with no_io true or false
  // Note: 'value_found' from KeyMayExist propagates here
793 794 795
  Status GetImpl(const ReadOptions& options, ColumnFamilyHandle* column_family,
                 const Slice& key, std::string* value,
                 bool* value_found = nullptr);
796 797 798 799

  bool GetIntPropertyInternal(ColumnFamilyHandle* column_family,
                              DBPropertyType property_type,
                              bool need_out_of_mutex, uint64_t* value);
J
jorlow@chromium.org 已提交
800 801 802 803 804 805 806
};

// 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);
807
extern DBOptions SanitizeOptions(const std::string& db, const DBOptions& src);
S
Siying Dong 已提交
808

M
miguelportilla 已提交
809 810 811 812 813 814 815
// 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;
}

816
}  // namespace rocksdb