db_impl.cc 106.7 KB
Newer Older
1
//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
S
Siying Dong 已提交
2 3 4
//  This source code is licensed under both the GPLv2 (found in the
//  COPYING file in the root directory) and Apache 2.0 License
//  (found in the LICENSE.Apache file in the root directory).
5
//
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/db_impl.h"

L
liuhuahang 已提交
11
#ifndef __STDC_FORMAT_MACROS
I
Igor Canadi 已提交
12
#define __STDC_FORMAT_MACROS
L
liuhuahang 已提交
13
#endif
14
#include <stdint.h>
D
David Bernard 已提交
15
#ifdef OS_SOLARIS
D
David Bernard 已提交
16
#include <alloca.h>
D
David Bernard 已提交
17
#endif
18

J
jorlow@chromium.org 已提交
19
#include <algorithm>
20
#include <cstdio>
21
#include <map>
J
jorlow@chromium.org 已提交
22
#include <set>
23
#include <stdexcept>
24
#include <string>
25
#include <unordered_map>
26
#include <unordered_set>
T
Tomislav Novak 已提交
27
#include <utility>
28
#include <vector>
29

J
jorlow@chromium.org 已提交
30
#include "db/builder.h"
I
Igor Canadi 已提交
31
#include "db/compaction_job.h"
32
#include "db/db_info_dumper.h"
33
#include "db/db_iter.h"
K
kailiu 已提交
34
#include "db/dbformat.h"
35
#include "db/error_handler.h"
36
#include "db/event_helpers.h"
37
#include "db/external_sst_file_ingestion_job.h"
38 39
#include "db/flush_job.h"
#include "db/forward_iterator.h"
I
Igor Canadi 已提交
40
#include "db/job_context.h"
J
jorlow@chromium.org 已提交
41 42
#include "db/log_reader.h"
#include "db/log_writer.h"
43
#include "db/malloc_stats.h"
J
jorlow@chromium.org 已提交
44
#include "db/memtable.h"
K
kailiu 已提交
45
#include "db/memtable_list.h"
46
#include "db/merge_context.h"
47
#include "db/merge_helper.h"
A
Andrew Kryczka 已提交
48
#include "db/range_del_aggregator.h"
J
jorlow@chromium.org 已提交
49
#include "db/table_cache.h"
K
kailiu 已提交
50
#include "db/table_properties_collector.h"
51
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
52 53
#include "db/version_set.h"
#include "db/write_batch_internal.h"
A
agiardullo 已提交
54
#include "db/write_callback.h"
55 56
#include "memtable/hash_linklist_rep.h"
#include "memtable/hash_skiplist_rep.h"
57 58 59 60 61 62 63
#include "monitoring/iostats_context_imp.h"
#include "monitoring/perf_context_imp.h"
#include "monitoring/thread_status_updater.h"
#include "monitoring/thread_status_util.h"
#include "options/cf_options.h"
#include "options/options_helper.h"
#include "options/options_parser.h"
64
#include "port/port.h"
I
Igor Canadi 已提交
65
#include "rocksdb/cache.h"
66
#include "rocksdb/compaction_filter.h"
A
Aaron G 已提交
67
#include "rocksdb/convenience.h"
68 69 70 71 72
#include "rocksdb/db.h"
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
#include "rocksdb/statistics.h"
#include "rocksdb/status.h"
S
Siying Dong 已提交
73
#include "rocksdb/table.h"
74
#include "rocksdb/write_buffer_manager.h"
J
jorlow@chromium.org 已提交
75
#include "table/block.h"
76
#include "table/block_based_table_factory.h"
77
#include "table/merging_iterator.h"
K
kailiu 已提交
78
#include "table/table_builder.h"
J
jorlow@chromium.org 已提交
79
#include "table/two_level_iterator.h"
A
Aaron G 已提交
80
#include "tools/sst_dump_tool_imp.h"
81
#include "util/auto_roll_logger.h"
K
kailiu 已提交
82
#include "util/autovector.h"
83
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
84
#include "util/coding.h"
I
Igor Canadi 已提交
85
#include "util/compression.h"
86
#include "util/crc32c.h"
87
#include "util/file_reader_writer.h"
88
#include "util/file_util.h"
89
#include "util/filename.h"
H
Haobo Xu 已提交
90
#include "util/log_buffer.h"
91
#include "util/logging.h"
J
jorlow@chromium.org 已提交
92
#include "util/mutexlock.h"
93
#include "util/sst_file_manager_impl.h"
94
#include "util/stop_watch.h"
95
#include "util/string_util.h"
96
#include "util/sync_point.h"
J
jorlow@chromium.org 已提交
97

98
namespace rocksdb {
99
const std::string kDefaultColumnFamilyName("default");
100
void DumpRocksDBBuildVersion(Logger* log);
101

A
Aaron Gao 已提交
102 103 104
CompressionType GetCompressionFlush(
    const ImmutableCFOptions& ioptions,
    const MutableCFOptions& mutable_cf_options) {
105 106 107
  // Compressing memtable flushes might not help unless the sequential load
  // optimization is used for leveled compaction. Otherwise the CPU and
  // latency overhead is not offset by saving much space.
108
  if (ioptions.compaction_style == kCompactionStyleUniversal) {
109 110
    if (mutable_cf_options.compaction_options_universal
            .compression_size_percent < 0) {
111 112 113 114 115 116 117
      return mutable_cf_options.compression;
    } else {
      return kNoCompression;
    }
  } else if (!ioptions.compression_per_level.empty()) {
    // For leveled compress when min_level_to_compress != 0.
    return ioptions.compression_per_level[0];
118
  } else {
A
Aaron Gao 已提交
119
    return mutable_cf_options.compression;
120 121
  }
}
I
Igor Canadi 已提交
122

S
Siying Dong 已提交
123
namespace {
124
void DumpSupportInfo(Logger* logger) {
125
  ROCKS_LOG_HEADER(logger, "Compression algorithms supported:");
126 127 128 129 130 131 132
  for (auto& compression : OptionsHelper::compression_type_string_map) {
    if (compression.second != kNoCompression &&
        compression.second != kDisableCompressionOption) {
      ROCKS_LOG_HEADER(logger, "\t%s supported: %d", compression.first.c_str(),
                       CompressionTypeSupported(compression.second));
    }
  }
133 134
  ROCKS_LOG_HEADER(logger, "Fast CRC32 supported: %s",
                   crc32c::IsFastCrc32Supported().c_str());
I
Igor Canadi 已提交
135
}
136 137

int64_t kDefaultLowPriThrottledRate = 2 * 1024 * 1024;
138
}  // namespace
139

140
DBImpl::DBImpl(const DBOptions& options, const std::string& dbname,
141
               const bool seq_per_batch, const bool batch_per_txn)
J
jorlow@chromium.org 已提交
142
    : env_(options.env),
H
heyongqiang 已提交
143
      dbname_(dbname),
144
      own_info_log_(options.info_log == nullptr),
145 146 147
      initial_db_options_(SanitizeOptions(dbname, options)),
      immutable_db_options_(initial_db_options_),
      mutable_db_options_(initial_db_options_),
148
      stats_(immutable_db_options_.statistics.get()),
149
      db_lock_(nullptr),
150 151
      mutex_(stats_, env_, DB_MUTEX_WAIT_MICROS,
             immutable_db_options_.use_adaptive_mutex),
I
Igor Canadi 已提交
152
      shutting_down_(false),
J
jorlow@chromium.org 已提交
153
      bg_cv_(&mutex_),
154
      logfile_number_(0),
155
      log_dir_synced_(false),
I
Igor Canadi 已提交
156
      log_empty_(true),
157
      default_cf_handle_(nullptr),
158
      log_sync_cv_(&mutex_),
I
Igor Canadi 已提交
159 160
      total_log_size_(0),
      max_total_in_memory_state_(0),
S
sdong 已提交
161
      is_snapshot_supported_(true),
162
      write_buffer_manager_(immutable_db_options_.write_buffer_manager.get()),
163
      write_thread_(immutable_db_options_),
164
      nonmem_write_thread_(immutable_db_options_),
165
      write_controller_(mutable_db_options_.delayed_write_rate),
166 167 168 169 170
      // Use delayed_write_rate as a base line to determine the initial
      // low pri write rate limit. It may be adjusted later.
      low_pri_write_rate_limiter_(NewGenericRateLimiter(std::min(
          static_cast<int64_t>(mutable_db_options_.delayed_write_rate / 8),
          kDefaultLowPriThrottledRate))),
S
sdong 已提交
171
      last_batch_group_size_(0),
172 173
      unscheduled_flushes_(0),
      unscheduled_compactions_(0),
174
      bg_bottom_compaction_scheduled_(0),
175
      bg_compaction_scheduled_(0),
176
      num_running_compactions_(0),
177
      bg_flush_scheduled_(0),
178
      num_running_flushes_(0),
179
      bg_purge_scheduled_(0),
180
      disable_delete_obsolete_files_(0),
181
      pending_purge_obsolete_files_(0),
182
      delete_obsolete_files_last_run_(env_->NowMicros()),
183
      last_stats_dump_time_microsec_(0),
184
      next_job_id_(1),
185
      has_unpersisted_data_(false),
S
Siying Dong 已提交
186
      unable_to_release_oldest_log_(false),
187
      env_options_(BuildDBOptions(immutable_db_options_, mutable_db_options_)),
188
      env_options_for_compaction_(env_->OptimizeForCompactionTableWrite(
189
          env_options_, immutable_db_options_)),
190
      num_running_ingest_file_(0),
I
Igor Canadi 已提交
191
#ifndef ROCKSDB_LITE
192
      wal_manager_(immutable_db_options_, env_options_, seq_per_batch),
I
Igor Canadi 已提交
193
#endif  // ROCKSDB_LITE
194
      event_logger_(immutable_db_options_.info_log.get()),
195
      bg_work_paused_(0),
196
      bg_compaction_paused_(0),
197
      refitting_level_(false),
198
      opened_successfully_(false),
199
      two_write_queues_(options.two_write_queues),
200
      manual_wal_flush_(options.manual_wal_flush),
201
      seq_per_batch_(seq_per_batch),
202
      batch_per_txn_(batch_per_txn),
203 204 205 206 207 208 209 210 211 212 213
      // last_sequencee_ is always maintained by the main queue that also writes
      // to the memtable. When two_write_queues_ is disabled last seq in
      // memtable is the same as last seq published to the readers. When it is
      // enabled but seq_per_batch_ is disabled, last seq in memtable still
      // indicates last published seq since wal-only writes that go to the 2nd
      // queue do not consume a sequence number. Otherwise writes performed by
      // the 2nd queue could change what is visible to the readers. In this
      // cases, last_seq_same_as_publish_seq_==false, the 2nd queue maintains a
      // separate variable to indicate the last published sequence.
      last_seq_same_as_publish_seq_(
          !(seq_per_batch && options.two_write_queues)),
214 215 216 217
      // Since seq_per_batch_ is currently set only by WritePreparedTxn which
      // requires a custom gc for compaction, we use that to set use_custom_gc_
      // as well.
      use_custom_gc_(seq_per_batch),
218
      preserve_deletes_(options.preserve_deletes),
219 220
      closed_(false),
      error_handler_(immutable_db_options_, &mutex_) {
221 222 223
  // !batch_per_trx_ implies seq_per_batch_ because it is only unset for
  // WriteUnprepared, which should use seq_per_batch_.
  assert(batch_per_txn_ || seq_per_batch_);
H
heyongqiang 已提交
224
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
225

J
jorlow@chromium.org 已提交
226
  // Reserve ten files or so for other uses and give the rest to TableCache.
227
  // Give a large number for setting of "infinite" open files.
L
Leonidas Galanis 已提交
228
  const int table_cache_size = (mutable_db_options_.max_open_files == -1)
229
                                   ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
230
                                   : mutable_db_options_.max_open_files - 10;
231 232
  table_cache_ = NewLRUCache(table_cache_size,
                             immutable_db_options_.table_cache_numshardbits);
233

234
  versions_.reset(new VersionSet(dbname_, &immutable_db_options_, env_options_,
235
                                 table_cache_.get(), write_buffer_manager_,
236
                                 &write_controller_));
237 238
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
239

240 241 242 243 244
  DumpRocksDBBuildVersion(immutable_db_options_.info_log.get());
  DumpDBFileSummary(immutable_db_options_, dbname_);
  immutable_db_options_.Dump(immutable_db_options_.info_log.get());
  mutable_db_options_.Dump(immutable_db_options_.info_log.get());
  DumpSupportInfo(immutable_db_options_.info_log.get());
245 246 247 248 249

  // always open the DB with 0 here, which means if preserve_deletes_==true
  // we won't drop any deletion markers until SetPreserveDeletesSequenceNumber()
  // is called by client and this seqnum is advanced.
  preserve_deletes_seqnum_.store(0);
J
jorlow@chromium.org 已提交
250 251
}

252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
Status DBImpl::Resume() {
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Resuming DB");

  InstrumentedMutexLock db_mutex(&mutex_);

  if (!error_handler_.IsDBStopped() && !error_handler_.IsBGWorkStopped()) {
    // Nothing to do
    return Status::OK();
  }

  Status s = error_handler_.GetBGError();
  if (s.severity() > Status::Severity::kHardError) {
    ROCKS_LOG_INFO(immutable_db_options_.info_log,
        "DB resume requested but failed due to Fatal/Unrecoverable error");
    return s;
  }

  JobContext job_context(0);
  FindObsoleteFiles(&job_context, true);
  error_handler_.ClearBGError();
  mutex_.Unlock();

  job_context.manifest_file_number = 1;
  if (job_context.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(job_context);
  }
  job_context.Clean();

  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Successfully resumed DB");
  mutex_.Lock();
  MaybeScheduleFlushOrCompaction();

  // No need to check BGError again. If something happened, event listener would be
  // notified and the operation causing it would have failed
  return Status::OK();
}

289
// Will lock the mutex_,  will wait for completion if wait is true
290
void DBImpl::CancelAllBackgroundWork(bool wait) {
291
  InstrumentedMutexLock l(&mutex_);
292

293 294
  ROCKS_LOG_INFO(immutable_db_options_.info_log,
                 "Shutdown: canceling all background work");
295

296
  if (!shutting_down_.load(std::memory_order_acquire) &&
297
      has_unpersisted_data_.load(std::memory_order_relaxed) &&
Y
Yi Wu 已提交
298
      !mutable_db_options_.avoid_flush_during_shutdown) {
299
    for (auto cfd : *versions_->GetColumnFamilySet()) {
300
      if (!cfd->IsDropped() && cfd->initialized() && !cfd->mem()->IsEmpty()) {
301 302
        cfd->Ref();
        mutex_.Unlock();
303
        FlushMemTable(cfd, FlushOptions(), FlushReason::kShutDown);
304
        mutex_.Lock();
305
        cfd->Unref();
306 307
      }
    }
308
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
309
  }
310 311 312 313 314 315 316

  shutting_down_.store(true, std::memory_order_release);
  bg_cv_.SignalAll();
  if (!wait) {
    return;
  }
  // Wait for background work to finish
317 318
  while (bg_bottom_compaction_scheduled_ || bg_compaction_scheduled_ ||
         bg_flush_scheduled_) {
319 320 321 322
    bg_cv_.Wait();
  }
}

323
Status DBImpl::CloseHelper() {
324 325
  // CancelAllBackgroundWork called with false means we just set the shutdown
  // marker. After this we do a variant of the waiting and unschedule work
326 327
  // (to consider: moving all the waiting into CancelAllBackgroundWork(true))
  CancelAllBackgroundWork(false);
328 329
  int bottom_compactions_unscheduled =
      env_->UnSchedule(this, Env::Priority::BOTTOM);
330 331
  int compactions_unscheduled = env_->UnSchedule(this, Env::Priority::LOW);
  int flushes_unscheduled = env_->UnSchedule(this, Env::Priority::HIGH);
332
  Status ret;
333
  mutex_.Lock();
334
  bg_bottom_compaction_scheduled_ -= bottom_compactions_unscheduled;
335 336 337 338
  bg_compaction_scheduled_ -= compactions_unscheduled;
  bg_flush_scheduled_ -= flushes_unscheduled;

  // Wait for background work to finish
339
  while (bg_bottom_compaction_scheduled_ || bg_compaction_scheduled_ ||
340 341
         bg_flush_scheduled_ || bg_purge_scheduled_ ||
         pending_purge_obsolete_files_) {
342
    TEST_SYNC_POINT("DBImpl::~DBImpl:WaitJob");
343 344
    bg_cv_.Wait();
  }
345
  TEST_SYNC_POINT_CALLBACK("DBImpl::CloseHelper:PendingPurgeFinished",
346
                           &files_grabbed_for_purge_);
347
  EraseThreadStatusDbInfo();
I
Igor Canadi 已提交
348 349
  flush_scheduler_.Clear();

350 351 352 353 354 355 356 357 358 359 360 361 362
  while (!flush_queue_.empty()) {
    auto cfd = PopFirstFromFlushQueue();
    if (cfd->Unref()) {
      delete cfd;
    }
  }
  while (!compaction_queue_.empty()) {
    auto cfd = PopFirstFromCompactionQueue();
    if (cfd->Unref()) {
      delete cfd;
    }
  }

I
Igor Canadi 已提交
363 364 365 366 367
  if (default_cf_handle_ != nullptr) {
    // we need to delete handle outside of lock because it does its own locking
    mutex_.Unlock();
    delete default_cf_handle_;
    mutex_.Lock();
368 369
  }

I
Igor Canadi 已提交
370 371 372 373 374 375 376 377 378 379
  // Clean up obsolete files due to SuperVersion release.
  // (1) Need to delete to obsolete files before closing because RepairDB()
  // scans all existing files in the file system and builds manifest file.
  // Keeping obsolete files confuses the repair process.
  // (2) Need to check if we Open()/Recover() the DB successfully before
  // deleting because if VersionSet recover fails (may be due to corrupted
  // manifest file), it is not able to identify live files correctly. As a
  // result, all "live" files can get deleted by accident. However, corrupted
  // manifest is recoverable by RepairDB().
  if (opened_successfully_) {
380
    JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
381
    FindObsoleteFiles(&job_context, true);
382 383

    mutex_.Unlock();
I
Igor Canadi 已提交
384
    // manifest number starting from 2
I
Igor Canadi 已提交
385 386 387
    job_context.manifest_file_number = 1;
    if (job_context.HaveSomethingToDelete()) {
      PurgeObsoleteFiles(job_context);
388
    }
I
Igor Canadi 已提交
389
    job_context.Clean();
390
    mutex_.Lock();
391 392
  }

393 394 395
  for (auto l : logs_to_free_) {
    delete l;
  }
396
  for (auto& log : logs_) {
397 398 399
    uint64_t log_number = log.writer->get_log_number();
    Status s = log.ClearWriter();
    if (!s.ok()) {
400 401 402 403 404
      ROCKS_LOG_WARN(
          immutable_db_options_.info_log,
          "Unable to Sync WAL file %s with error -- %s",
          LogFileName(immutable_db_options_.wal_dir, log_number).c_str(),
          s.ToString().c_str());
405 406 407 408 409
      // Retain the first error
      if (ret.ok()) {
        ret = s;
      }
    }
410
  }
411
  logs_.clear();
412

413 414 415 416 417 418 419 420 421 422 423 424 425 426 427
  // Table cache may have table handles holding blocks from the block cache.
  // We need to release them before the block cache is destroyed. The block
  // cache may be destroyed inside versions_.reset(), when column family data
  // list is destroyed, so leaving handles in table cache after
  // versions_.reset() may cause issues.
  // Here we clean all unreferenced handles in table cache.
  // Now we assume all user queries have finished, so only version set itself
  // can possibly hold the blocks from block cache. After releasing unreferenced
  // handles here, only handles held by version set left and inside
  // versions_.reset(), we will release them. There, we need to make sure every
  // time a handle is released, we erase it from the cache too. By doing that,
  // we can guarantee that after versions_.reset(), table cache is empty
  // so the cache can be safely destroyed.
  table_cache_->EraseUnRefEntries();

I
Islam AbdelRahman 已提交
428 429 430 431
  for (auto& txn_entry : recovered_transactions_) {
    delete txn_entry.second;
  }

432
  // versions need to be destroyed before table_cache since it can hold
433 434
  // references to table_cache.
  versions_.reset();
435
  mutex_.Unlock();
I
Igor Canadi 已提交
436 437 438
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
439

440
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Shutdown complete");
441
  LogFlush(immutable_db_options_.info_log);
442 443

  if (immutable_db_options_.info_log && own_info_log_) {
444 445 446 447
    Status s = immutable_db_options_.info_log->Close();
    if (ret.ok()) {
      ret = s;
    }
448
  }
449
  return ret;
J
jorlow@chromium.org 已提交
450 451
}

452
Status DBImpl::CloseImpl() { return CloseHelper(); }
453 454 455 456 457 458 459

DBImpl::~DBImpl() {
  if (!closed_) {
    closed_ = true;
    CloseHelper();
  }
}
460

J
jorlow@chromium.org 已提交
461
void DBImpl::MaybeIgnoreError(Status* s) const {
462
  if (s->ok() || immutable_db_options_.paranoid_checks) {
J
jorlow@chromium.org 已提交
463 464
    // No change needed
  } else {
465 466
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "Ignoring error %s",
                   s->ToString().c_str());
J
jorlow@chromium.org 已提交
467 468 469 470
    *s = Status::OK();
  }
}

471
const Status DBImpl::CreateArchivalDirectory() {
472 473 474
  if (immutable_db_options_.wal_ttl_seconds > 0 ||
      immutable_db_options_.wal_size_limit_mb > 0) {
    std::string archivalPath = ArchivalDirectory(immutable_db_options_.wal_dir);
475 476 477 478 479
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

480
void DBImpl::PrintStatistics() {
481
  auto dbstats = immutable_db_options_.statistics.get();
482
  if (dbstats) {
483 484
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "STATISTICS:\n %s",
                   dbstats->ToString().c_str());
485 486 487
  }
}

488
void DBImpl::MaybeDumpStats() {
489 490 491 492 493
  mutex_.Lock();
  unsigned int stats_dump_period_sec =
      mutable_db_options_.stats_dump_period_sec;
  mutex_.Unlock();
  if (stats_dump_period_sec == 0) return;
H
Haobo Xu 已提交
494 495 496

  const uint64_t now_micros = env_->NowMicros();

497
  if (last_stats_dump_time_microsec_ + stats_dump_period_sec * 1000000 <=
498
      now_micros) {
H
Haobo Xu 已提交
499 500 501 502 503
    // Multiple threads could race in here simultaneously.
    // However, the last one will update last_stats_dump_time_microsec_
    // atomically. We could see more than one dump during one dump
    // period in rare cases.
    last_stats_dump_time_microsec_ = now_micros;
504

505
#ifndef ROCKSDB_LITE
506 507 508 509 510 511 512
    const DBPropertyInfo* cf_property_info =
        GetPropertyInfo(DB::Properties::kCFStats);
    assert(cf_property_info != nullptr);
    const DBPropertyInfo* db_property_info =
        GetPropertyInfo(DB::Properties::kDBStats);
    assert(db_property_info != nullptr);

H
Haobo Xu 已提交
513
    std::string stats;
514
    {
515
      InstrumentedMutexLock l(&mutex_);
516 517
      default_cf_internal_stats_->GetStringProperty(
          *db_property_info, DB::Properties::kDBStats, &stats);
518
      for (auto cfd : *versions_->GetColumnFamilySet()) {
519 520 521 522 523
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFStatsNoFileHistogram,
              &stats);
        }
524 525
      }
      for (auto cfd : *versions_->GetColumnFamilySet()) {
526 527 528 529
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFFileHistogram, &stats);
        }
530 531
      }
    }
532 533 534
    ROCKS_LOG_WARN(immutable_db_options_.info_log,
                   "------- DUMPING STATS -------");
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "%s", stats.c_str());
S
Siying Dong 已提交
535 536 537 538
    if (immutable_db_options_.dump_malloc_stats) {
      stats.clear();
      DumpMallocStats(&stats);
      if (!stats.empty()) {
539 540 541
        ROCKS_LOG_WARN(immutable_db_options_.info_log,
                       "------- Malloc STATS -------");
        ROCKS_LOG_WARN(immutable_db_options_.info_log, "%s", stats.c_str());
S
Siying Dong 已提交
542 543
      }
    }
544
#endif  // !ROCKSDB_LITE
545

546
    PrintStatistics();
547 548 549
  }
}

550 551 552 553 554 555 556 557 558 559
void DBImpl::ScheduleBgLogWriterClose(JobContext* job_context) {
  if (!job_context->logs_to_free.empty()) {
    for (auto l : job_context->logs_to_free) {
      AddToLogsToFreeQueue(l);
    }
    job_context->logs_to_free.clear();
    SchedulePurge();
  }
}

560 561 562 563 564 565 566 567 568 569
Directory* DBImpl::GetDataDir(ColumnFamilyData* cfd, size_t path_id) const {
  assert(cfd);
  Directory* ret_dir = cfd->GetDataDir(path_id);
  if (ret_dir == nullptr) {
    return directories_.GetDataDir(path_id);
  }
  return ret_dir;
}

Directory* DBImpl::Directories::GetDataDir(size_t path_id) const {
S
Siying Dong 已提交
570 571 572 573 574
  assert(path_id < data_dirs_.size());
  Directory* ret_dir = data_dirs_[path_id].get();
  if (ret_dir == nullptr) {
    // Should use db_dir_
    return db_dir_.get();
575
  }
S
Siying Dong 已提交
576
  return ret_dir;
577 578
}

579 580
Status DBImpl::SetOptions(
    ColumnFamilyHandle* column_family,
S
Siying Dong 已提交
581 582
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
583 584
  (void)column_family;
  (void)options_map;
S
Siying Dong 已提交
585 586 587 588 589 590 591 592
  return Status::NotSupported("Not supported in ROCKSDB LITE");
#else
  auto* cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
  if (options_map.empty()) {
    ROCKS_LOG_WARN(immutable_db_options_.info_log,
                   "SetOptions() on column family [%s], empty input",
                   cfd->GetName().c_str());
    return Status::InvalidArgument("empty input");
593 594
  }

S
Siying Dong 已提交
595 596 597 598
  MutableCFOptions new_options;
  Status s;
  Status persist_options_status;
  WriteThread::Writer w;
599
  SuperVersionContext sv_context(/* create_superversion */ true);
S
Siying Dong 已提交
600 601 602 603 604 605 606 607 608 609 610 611
  {
    InstrumentedMutexLock l(&mutex_);
    s = cfd->SetOptions(options_map);
    if (s.ok()) {
      new_options = *cfd->GetLatestMutableCFOptions();
      // Append new version to recompute compaction score.
      VersionEdit dummy_edit;
      versions_->LogAndApply(cfd, new_options, &dummy_edit, &mutex_,
                             directories_.GetDbDir());
      // Trigger possible flush/compactions. This has to be before we persist
      // options to file, otherwise there will be a deadlock with writer
      // thread.
612
      InstallSuperVersionAndScheduleWork(cfd, &sv_context, new_options);
613

Y
Yi Wu 已提交
614 615
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, true /*need_enter_write_thread*/);
616
      bg_cv_.SignalAll();
617 618
    }
  }
619
  sv_context.Clean();
620

621 622 623
  ROCKS_LOG_INFO(
      immutable_db_options_.info_log,
      "SetOptions() on column family [%s], inputs:", cfd->GetName().c_str());
S
Siying Dong 已提交
624 625 626
  for (const auto& o : options_map) {
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "%s: %s\n", o.first.c_str(),
                   o.second.c_str());
627
  }
S
Siying Dong 已提交
628 629 630 631 632
  if (s.ok()) {
    ROCKS_LOG_INFO(immutable_db_options_.info_log,
                   "[%s] SetOptions() succeeded", cfd->GetName().c_str());
    new_options.Dump(immutable_db_options_.info_log.get());
    if (!persist_options_status.ok()) {
Y
Yi Wu 已提交
633
      s = persist_options_status;
634
    }
635
  } else {
S
Siying Dong 已提交
636 637
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "[%s] SetOptions() failed",
                   cfd->GetName().c_str());
638
  }
S
Siying Dong 已提交
639 640 641
  LogFlush(immutable_db_options_.info_log);
  return s;
#endif  // ROCKSDB_LITE
642 643
}

S
Siying Dong 已提交
644 645 646
Status DBImpl::SetDBOptions(
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
647
  (void)options_map;
S
Siying Dong 已提交
648 649 650 651 652 653
  return Status::NotSupported("Not supported in ROCKSDB LITE");
#else
  if (options_map.empty()) {
    ROCKS_LOG_WARN(immutable_db_options_.info_log,
                   "SetDBOptions(), empty input.");
    return Status::InvalidArgument("empty input");
I
Igor Canadi 已提交
654
  }
655

S
Siying Dong 已提交
656 657 658
  MutableDBOptions new_options;
  Status s;
  Status persist_options_status;
659
  bool wal_changed = false;
S
Siying Dong 已提交
660 661 662 663 664 665 666 667 668 669 670 671 672
  WriteThread::Writer w;
  WriteContext write_context;
  {
    InstrumentedMutexLock l(&mutex_);
    s = GetMutableDBOptionsFromStrings(mutable_db_options_, options_map,
                                       &new_options);
    if (s.ok()) {
      if (new_options.max_background_compactions >
          mutable_db_options_.max_background_compactions) {
        env_->IncBackgroundThreadsIfNeeded(
            new_options.max_background_compactions, Env::Priority::LOW);
        MaybeScheduleFlushOrCompaction();
      }
673

674 675
      write_controller_.set_max_delayed_write_rate(
          new_options.delayed_write_rate);
L
Leonidas Galanis 已提交
676
      table_cache_.get()->SetCapacity(new_options.max_open_files == -1
677
                                          ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
678
                                          : new_options.max_open_files - 10);
679 680 681 682 683
      wal_changed = mutable_db_options_.wal_bytes_per_sync !=
                    new_options.wal_bytes_per_sync;
      if (new_options.bytes_per_sync == 0) {
        new_options.bytes_per_sync = 1024 * 1024;
      }
684
      mutable_db_options_ = new_options;
685 686
      env_options_for_compaction_ = EnvOptions(
          BuildDBOptions(immutable_db_options_, mutable_db_options_));
687
      env_options_for_compaction_ = env_->OptimizeForCompactionTableWrite(
688
          env_options_for_compaction_, immutable_db_options_);
689
      versions_->ChangeEnvOptions(mutable_db_options_);
690 691 692 693
      env_options_for_compaction_ = env_->OptimizeForCompactionTableRead(
          env_options_for_compaction_, immutable_db_options_);
      env_options_for_compaction_.compaction_readahead_size =
          mutable_db_options_.compaction_readahead_size;
694
      write_thread_.EnterUnbatched(&w, &mutex_);
695 696
      if (total_log_size_ > GetMaxTotalWalSize() || wal_changed) {
        Status purge_wal_status = SwitchWAL(&write_context);
697 698 699 700 701
        if (!purge_wal_status.ok()) {
          ROCKS_LOG_WARN(immutable_db_options_.info_log,
                         "Unable to purge WAL files in SetDBOptions() -- %s",
                         purge_wal_status.ToString().c_str());
        }
702
      }
Y
Yi Wu 已提交
703 704
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, false /*need_enter_write_thread*/);
705
      write_thread_.ExitUnbatched(&w);
706 707
    }
  }
708
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions(), inputs:");
709
  for (const auto& o : options_map) {
710 711
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "%s: %s\n", o.first.c_str(),
                   o.second.c_str());
712 713
  }
  if (s.ok()) {
714
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions() succeeded");
715 716 717 718 719 720 721
    new_options.Dump(immutable_db_options_.info_log.get());
    if (!persist_options_status.ok()) {
      if (immutable_db_options_.fail_if_options_file_error) {
        s = Status::IOError(
            "SetDBOptions() succeeded, but unable to persist options",
            persist_options_status.ToString());
      }
722 723 724
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "Unable to persist options in SetDBOptions() -- %s",
                     persist_options_status.ToString().c_str());
725 726
    }
  } else {
727
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "SetDBOptions failed");
L
Lei Jin 已提交
728
  }
729
  LogFlush(immutable_db_options_.info_log);
730
  return s;
I
Igor Canadi 已提交
731
#endif  // ROCKSDB_LITE
732 733
}

734
// return the same level if it cannot be moved
A
Andrew Kryczka 已提交
735 736 737
int DBImpl::FindMinimumEmptyLevelFitting(
    ColumnFamilyData* cfd, const MutableCFOptions& /*mutable_cf_options*/,
    int level) {
738
  mutex_.AssertHeld();
S
sdong 已提交
739
  const auto* vstorage = cfd->current()->storage_info();
740
  int minimum_level = level;
741
  for (int i = level - 1; i > 0; --i) {
742
    // stop if level i is not empty
S
sdong 已提交
743
    if (vstorage->NumLevelFiles(i) > 0) break;
744
    // stop if level i is too small (cannot fit the level files)
745
    if (vstorage->MaxBytesForLevel(i) < vstorage->NumLevelBytes(level)) {
746 747
      break;
    }
748 749 750 751 752 753

    minimum_level = i;
  }
  return minimum_level;
}

754
Status DBImpl::FlushWAL(bool sync) {
755
  if (manual_wal_flush_) {
756 757 758 759 760 761 762
    // We need to lock log_write_mutex_ since logs_ might change concurrently
    InstrumentedMutexLock wl(&log_write_mutex_);
    log::Writer* cur_log_writer = logs_.back().writer;
    auto s = cur_log_writer->WriteBuffer();
    if (!s.ok()) {
      ROCKS_LOG_ERROR(immutable_db_options_.info_log, "WAL flush error %s",
                      s.ToString().c_str());
763 764 765 766 767
      // In case there is a fs error we should set it globally to prevent the
      // future writes
      WriteStatusCheck(s);
      // whether sync or not, we should abort the rest of function upon error
      return s;
768 769 770 771 772 773
    }
    if (!sync) {
      ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=false");
      return s;
    }
  }
774 775 776
  if (!sync) {
    return Status::OK();
  }
777 778 779 780 781
  // sync = true
  ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=true");
  return SyncWAL();
}

782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802
Status DBImpl::SyncWAL() {
  autovector<log::Writer*, 1> logs_to_sync;
  bool need_log_dir_sync;
  uint64_t current_log_number;

  {
    InstrumentedMutexLock l(&mutex_);
    assert(!logs_.empty());

    // This SyncWAL() call only cares about logs up to this number.
    current_log_number = logfile_number_;

    while (logs_.front().number <= current_log_number &&
           logs_.front().getting_synced) {
      log_sync_cv_.Wait();
    }
    // First check that logs are safe to sync in background.
    for (auto it = logs_.begin();
         it != logs_.end() && it->number <= current_log_number; ++it) {
      if (!it->writer->file()->writable_file()->IsSyncThreadSafe()) {
        return Status::NotSupported(
803 804 805 806
            "SyncWAL() is not supported for this implementation of WAL file",
            immutable_db_options_.allow_mmap_writes
                ? "try setting Options::allow_mmap_writes to false"
                : Slice());
807 808 809 810 811
      }
    }
    for (auto it = logs_.begin();
         it != logs_.end() && it->number <= current_log_number; ++it) {
      auto& log = *it;
S
Siying Dong 已提交
812 813 814 815
      assert(!log.getting_synced);
      log.getting_synced = true;
      logs_to_sync.push_back(log.writer);
    }
816

S
Siying Dong 已提交
817
    need_log_dir_sync = !log_dir_synced_;
818 819
  }

820
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:1");
S
Siying Dong 已提交
821 822 823 824 825 826 827 828 829 830
  RecordTick(stats_, WAL_FILE_SYNCED);
  Status status;
  for (log::Writer* log : logs_to_sync) {
    status = log->file()->SyncWithoutFlush(immutable_db_options_.use_fsync);
    if (!status.ok()) {
      break;
    }
  }
  if (status.ok() && need_log_dir_sync) {
    status = directories_.GetWalDir()->Fsync();
831
  }
832
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:2");
833

S
Siying Dong 已提交
834 835 836 837 838 839
  TEST_SYNC_POINT("DBImpl::SyncWAL:BeforeMarkLogsSynced:1");
  {
    InstrumentedMutexLock l(&mutex_);
    MarkLogsSynced(current_log_number, need_log_dir_sync, status);
  }
  TEST_SYNC_POINT("DBImpl::SyncWAL:BeforeMarkLogsSynced:2");
840

S
Siying Dong 已提交
841
  return status;
842 843
}

844 845
void DBImpl::MarkLogsSynced(uint64_t up_to, bool synced_dir,
                            const Status& status) {
S
Siying Dong 已提交
846
  mutex_.AssertHeld();
847
  if (synced_dir && logfile_number_ == up_to && status.ok()) {
S
Siying Dong 已提交
848 849 850 851 852 853 854
    log_dir_synced_ = true;
  }
  for (auto it = logs_.begin(); it != logs_.end() && it->number <= up_to;) {
    auto& log = *it;
    assert(log.getting_synced);
    if (status.ok() && logs_.size() > 1) {
      logs_to_free_.push_back(log.ReleaseWriter());
855 856
      // To modify logs_ both mutex_ and log_write_mutex_ must be held
      InstrumentedMutexLock l(&log_write_mutex_);
S
Siying Dong 已提交
857 858 859 860 861 862 863 864 865
      it = logs_.erase(it);
    } else {
      log.getting_synced = false;
      ++it;
    }
  }
  assert(!status.ok() || logs_.empty() || logs_[0].number > up_to ||
         (logs_.size() == 1 && !logs_[0].getting_synced));
  log_sync_cv_.SignalAll();
866 867
}

S
Siying Dong 已提交
868 869
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
  return versions_->LastSequence();
870 871
}

872 873
void DBImpl::SetLastPublishedSequence(SequenceNumber seq) {
  versions_->SetLastPublishedSequence(seq);
874 875
}

876 877 878 879 880 881 882 883 884
bool DBImpl::SetPreserveDeletesSequenceNumber(SequenceNumber seqnum) {
  if (seqnum > preserve_deletes_seqnum_.load()) {
    preserve_deletes_seqnum_.store(seqnum);
    return true;
  } else {
    return false;
  }
}

S
Siying Dong 已提交
885 886 887 888 889 890 891 892 893
InternalIterator* DBImpl::NewInternalIterator(
    Arena* arena, RangeDelAggregator* range_del_agg,
    ColumnFamilyHandle* column_family) {
  ColumnFamilyData* cfd;
  if (column_family == nullptr) {
    cfd = default_cf_handle_->cfd();
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    cfd = cfh->cfd();
894
  }
895

S
Siying Dong 已提交
896 897 898 899 900 901 902
  mutex_.Lock();
  SuperVersion* super_version = cfd->GetSuperVersion()->Ref();
  mutex_.Unlock();
  ReadOptions roptions;
  return NewInternalIterator(roptions, cfd, super_version, arena,
                             range_del_agg);
}
I
Islam AbdelRahman 已提交
903

S
Siying Dong 已提交
904 905 906
void DBImpl::SchedulePurge() {
  mutex_.AssertHeld();
  assert(opened_successfully_);
I
Islam AbdelRahman 已提交
907

S
Siying Dong 已提交
908 909 910
  // Purge operations are put into High priority queue
  bg_purge_scheduled_++;
  env_->Schedule(&DBImpl::BGWorkPurge, this, Env::Priority::HIGH, nullptr);
I
Islam AbdelRahman 已提交
911 912
}

913 914 915
void DBImpl::BackgroundCallPurge() {
  mutex_.Lock();

916 917 918 919 920 921 922
  // We use one single loop to clear both queues so that after existing the loop
  // both queues are empty. This is stricter than what is needed, but can make
  // it easier for us to reason the correctness.
  while (!purge_queue_.empty() || !logs_to_free_queue_.empty()) {
    if (!purge_queue_.empty()) {
      auto purge_file = purge_queue_.begin();
      auto fname = purge_file->fname;
923
      auto dir_to_sync = purge_file->dir_to_sync;
924 925 926 927
      auto type = purge_file->type;
      auto number = purge_file->number;
      auto job_id = purge_file->job_id;
      purge_queue_.pop_front();
928

929
      mutex_.Unlock();
930
      DeleteObsoleteFileImpl(job_id, fname, dir_to_sync, type, number);
931 932 933 934 935 936 937 938 939
      mutex_.Lock();
    } else {
      assert(!logs_to_free_queue_.empty());
      log::Writer* log_writer = *(logs_to_free_queue_.begin());
      logs_to_free_queue_.pop_front();
      mutex_.Unlock();
      delete log_writer;
      mutex_.Lock();
    }
940 941 942 943 944 945 946 947 948 949 950
  }
  bg_purge_scheduled_--;

  bg_cv_.SignalAll();
  // IMPORTANT:there should be no code after calling SignalAll. This call may
  // signal the DB destructor that it's OK to proceed with destruction. In
  // that case, all DB variables will be dealloacated and referencing them
  // will cause trouble.
  mutex_.Unlock();
}

951 952
namespace {
struct IterState {
953
  IterState(DBImpl* _db, InstrumentedMutex* _mu, SuperVersion* _super_version,
954
            bool _background_purge)
955 956 957
      : db(_db),
        mu(_mu),
        super_version(_super_version),
958
        background_purge(_background_purge) {}
959 960

  DBImpl* db;
961
  InstrumentedMutex* mu;
962
  SuperVersion* super_version;
963
  bool background_purge;
964 965
};

A
Andrew Kryczka 已提交
966
static void CleanupIteratorState(void* arg1, void* /*arg2*/) {
967
  IterState* state = reinterpret_cast<IterState*>(arg1);
968

969
  if (state->super_version->Unref()) {
970 971 972
    // Job id == 0 means that this is not our background process, but rather
    // user thread
    JobContext job_context(0);
973

974 975
    state->mu->Lock();
    state->super_version->Cleanup();
I
Igor Canadi 已提交
976
    state->db->FindObsoleteFiles(&job_context, false, true);
977 978 979
    if (state->background_purge) {
      state->db->ScheduleBgLogWriterClose(&job_context);
    }
980 981 982
    state->mu->Unlock();

    delete state->super_version;
I
Igor Canadi 已提交
983
    if (job_context.HaveSomethingToDelete()) {
984
      if (state->background_purge) {
985 986 987 988 989 990 991 992 993 994
        // PurgeObsoleteFiles here does not delete files. Instead, it adds the
        // files to be deleted to a job queue, and deletes it in a separate
        // background thread.
        state->db->PurgeObsoleteFiles(job_context, true /* schedule only */);
        state->mu->Lock();
        state->db->SchedulePurge();
        state->mu->Unlock();
      } else {
        state->db->PurgeObsoleteFiles(job_context);
      }
995
    }
I
Igor Canadi 已提交
996
    job_context.Clean();
I
Igor Canadi 已提交
997
  }
T
Tomislav Novak 已提交
998

999 1000
  delete state;
}
H
Hans Wennborg 已提交
1001
}  // namespace
1002

A
Andrew Kryczka 已提交
1003 1004 1005 1006
InternalIterator* DBImpl::NewInternalIterator(
    const ReadOptions& read_options, ColumnFamilyData* cfd,
    SuperVersion* super_version, Arena* arena,
    RangeDelAggregator* range_del_agg) {
S
sdong 已提交
1007
  InternalIterator* internal_iter;
1008
  assert(arena != nullptr);
A
Andrew Kryczka 已提交
1009
  assert(range_del_agg != nullptr);
1010
  // Need to create internal iterator from the arena.
1011 1012 1013
  MergeIteratorBuilder merge_iter_builder(
      &cfd->internal_comparator(), arena,
      !read_options.total_order_seek &&
1014
          super_version->mutable_cf_options.prefix_extractor != nullptr);
1015 1016
  // Collect iterator for mutable mem
  merge_iter_builder.AddIterator(
L
Lei Jin 已提交
1017
      super_version->mem->NewIterator(read_options, arena));
1018
  std::unique_ptr<InternalIterator> range_del_iter;
A
Andrew Kryczka 已提交
1019 1020
  Status s;
  if (!read_options.ignore_range_deletions) {
1021 1022
    range_del_iter.reset(
        super_version->mem->NewRangeTombstoneIterator(read_options));
A
Andrew Kryczka 已提交
1023 1024
    s = range_del_agg->AddTombstones(std::move(range_del_iter));
  }
1025
  // Collect all needed child iterators for immutable memtables
A
Andrew Kryczka 已提交
1026 1027 1028 1029 1030 1031 1032
  if (s.ok()) {
    super_version->imm->AddIterators(read_options, &merge_iter_builder);
    if (!read_options.ignore_range_deletions) {
      s = super_version->imm->AddRangeTombstoneIterators(read_options, arena,
                                                         range_del_agg);
    }
  }
1033
  TEST_SYNC_POINT_CALLBACK("DBImpl::NewInternalIterator:StatusCallback", &s);
A
Andrew Kryczka 已提交
1034 1035
  if (s.ok()) {
    // Collect iterators for files in L0 - Ln
S
Sagar Vemuri 已提交
1036 1037 1038 1039
    if (read_options.read_tier != kMemtableTier) {
      super_version->current->AddIterators(read_options, env_options_,
                                           &merge_iter_builder, range_del_agg);
    }
A
Andrew Kryczka 已提交
1040 1041 1042 1043 1044 1045 1046
    internal_iter = merge_iter_builder.Finish();
    IterState* cleanup =
        new IterState(this, &mutex_, super_version,
                      read_options.background_purge_on_iterator_cleanup);
    internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);

    return internal_iter;
1047 1048
  } else {
    CleanupSuperVersion(super_version);
A
Andrew Kryczka 已提交
1049
  }
1050
  return NewErrorInternalIterator<Slice>(s, arena);
J
jorlow@chromium.org 已提交
1051 1052
}

1053 1054 1055 1056
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

L
Lei Jin 已提交
1057
Status DBImpl::Get(const ReadOptions& read_options,
S
Siying Dong 已提交
1058 1059 1060
                   ColumnFamilyHandle* column_family, const Slice& key,
                   PinnableSlice* value) {
  return GetImpl(read_options, column_family, key, value);
L
Lei Jin 已提交
1061 1062 1063
}

Status DBImpl::GetImpl(const ReadOptions& read_options,
1064
                       ColumnFamilyHandle* column_family, const Slice& key,
1065
                       PinnableSlice* pinnable_val, bool* value_found,
Y
Yi Wu 已提交
1066
                       ReadCallback* callback, bool* is_blob_index) {
M
Maysam Yabandeh 已提交
1067
  assert(pinnable_val != nullptr);
L
Lei Jin 已提交
1068
  StopWatch sw(env_, stats_, DB_GET);
1069
  PERF_TIMER_GUARD(get_snapshot_time);
1070

1071 1072 1073
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

1074 1075 1076 1077 1078 1079 1080 1081 1082
  if (tracer_) {
    // TODO: This mutex should be removed later, to improve performance when
    // tracing is enabled.
    InstrumentedMutexLock lock(&trace_mutex_);
    if (tracer_) {
      tracer_->Get(column_family, key);
    }
  }

1083 1084 1085 1086 1087 1088
  // Acquire SuperVersion
  SuperVersion* sv = GetAndRefSuperVersion(cfd);

  TEST_SYNC_POINT("DBImpl::GetImpl:1");
  TEST_SYNC_POINT("DBImpl::GetImpl:2");

1089
  SequenceNumber snapshot;
L
Lei Jin 已提交
1090
  if (read_options.snapshot != nullptr) {
1091 1092 1093 1094 1095 1096 1097 1098
    // Note: In WritePrepared txns this is not necessary but not harmful
    // either.  Because prep_seq > snapshot => commit_seq > snapshot so if
    // a snapshot is specified we should be fine with skipping seq numbers
    // that are greater than that.
    //
    // In WriteUnprepared, we cannot set snapshot in the lookup key because we
    // may skip uncommitted data that should be visible to the transaction for
    // reading own writes.
1099 1100
    snapshot =
        reinterpret_cast<const SnapshotImpl*>(read_options.snapshot)->number_;
1101 1102 1103
    if (callback) {
      snapshot = std::max(snapshot, callback->MaxUnpreparedSequenceNumber());
    }
1104
  } else {
1105 1106 1107 1108 1109 1110
    // Since we get and reference the super version before getting
    // the snapshot number, without a mutex protection, it is possible
    // that a memtable switch happened in the middle and not all the
    // data for this snapshot is available. But it will contain all
    // the data available in the super version we have, which is also
    // a valid snapshot to read from.
1111 1112 1113 1114
    // We shouldn't get snapshot before finding and referencing the super
    // version because a flush happening in between may compact away data for
    // the snapshot, but the snapshot is earlier than the data overwriting it,
    // so users may see wrong results.
1115 1116 1117
    snapshot = last_seq_same_as_publish_seq_
                   ? versions_->LastSequence()
                   : versions_->LastPublishedSequence();
J
jorlow@chromium.org 已提交
1118
  }
1119 1120
  TEST_SYNC_POINT("DBImpl::GetImpl:3");
  TEST_SYNC_POINT("DBImpl::GetImpl:4");
1121

1122
  // Prepare to store a list of merge operations if merge occurs.
1123
  MergeContext merge_context;
1124
  RangeDelAggregator range_del_agg(cfd->internal_comparator(), snapshot);
1125

1126
  Status s;
1127
  // First look in the memtable, then in the immutable memtable (if any).
1128
  // s is both in/out. When in, s could either be OK or MergeInProgress.
1129
  // merge_operands will contain the sequence of merges in the latter case.
1130
  LookupKey lkey(key, snapshot);
L
Lei Jin 已提交
1131
  PERF_TIMER_STOP(get_snapshot_time);
1132

1133 1134
  bool skip_memtable = (read_options.read_tier == kPersistedTier &&
                        has_unpersisted_data_.load(std::memory_order_relaxed));
1135 1136
  bool done = false;
  if (!skip_memtable) {
M
Maysam Yabandeh 已提交
1137
    if (sv->mem->Get(lkey, pinnable_val->GetSelf(), &s, &merge_context,
Y
Yi Wu 已提交
1138
                     &range_del_agg, read_options, callback, is_blob_index)) {
1139
      done = true;
M
Maysam Yabandeh 已提交
1140
      pinnable_val->PinSelf();
1141
      RecordTick(stats_, MEMTABLE_HIT);
A
Andrew Kryczka 已提交
1142
    } else if ((s.ok() || s.IsMergeInProgress()) &&
M
Maysam Yabandeh 已提交
1143
               sv->imm->Get(lkey, pinnable_val->GetSelf(), &s, &merge_context,
Y
Yi Wu 已提交
1144 1145
                            &range_del_agg, read_options, callback,
                            is_blob_index)) {
1146
      done = true;
M
Maysam Yabandeh 已提交
1147
      pinnable_val->PinSelf();
1148 1149
      RecordTick(stats_, MEMTABLE_HIT);
    }
A
Andrew Kryczka 已提交
1150
    if (!done && !s.ok() && !s.IsMergeInProgress()) {
1151
      ReturnAndCleanupSuperVersion(cfd, sv);
A
Andrew Kryczka 已提交
1152 1153
      return s;
    }
1154 1155
  }
  if (!done) {
1156
    PERF_TIMER_GUARD(get_from_output_files_time);
M
Maysam Yabandeh 已提交
1157
    sv->current->Get(read_options, lkey, pinnable_val, &s, &merge_context,
Y
Yi Wu 已提交
1158 1159
                     &range_del_agg, value_found, nullptr, nullptr, callback,
                     is_blob_index);
L
Lei Jin 已提交
1160
    RecordTick(stats_, MEMTABLE_MISS);
1161
  }
1162

1163 1164
  {
    PERF_TIMER_GUARD(get_post_process_time);
1165

1166
    ReturnAndCleanupSuperVersion(cfd, sv);
1167

1168
    RecordTick(stats_, NUMBER_KEYS_READ);
M
Maysam Yabandeh 已提交
1169 1170 1171
    size_t size = pinnable_val->size();
    RecordTick(stats_, BYTES_READ, size);
    MeasureTime(stats_, BYTES_PER_READ, size);
1172
    PERF_COUNTER_ADD(get_read_bytes, size);
1173
  }
1174
  return s;
J
jorlow@chromium.org 已提交
1175 1176
}

1177
std::vector<Status> DBImpl::MultiGet(
L
Lei Jin 已提交
1178
    const ReadOptions& read_options,
1179
    const std::vector<ColumnFamilyHandle*>& column_family,
1180
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
L
Lei Jin 已提交
1181
  StopWatch sw(env_, stats_, DB_MULTIGET);
1182
  PERF_TIMER_GUARD(get_snapshot_time);
K
Kai Liu 已提交
1183

1184
  SequenceNumber snapshot;
1185

1186
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
1187
    ColumnFamilyData* cfd;
1188 1189 1190 1191 1192
    SuperVersion* super_version;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
1193 1194 1195 1196 1197 1198
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(cf);
    auto cfd = cfh->cfd();
    if (multiget_cf_data.find(cfd->GetID()) == multiget_cf_data.end()) {
      auto mgcfd = new MultiGetColumnFamilyData();
      mgcfd->cfd = cfd;
      multiget_cf_data.insert({cfd->GetID(), mgcfd});
1199 1200 1201
    }
  }

1202
  mutex_.Lock();
L
Lei Jin 已提交
1203
  if (read_options.snapshot != nullptr) {
1204 1205
    snapshot =
        reinterpret_cast<const SnapshotImpl*>(read_options.snapshot)->number_;
1206
  } else {
1207 1208 1209
    snapshot = last_seq_same_as_publish_seq_
                   ? versions_->LastSequence()
                   : versions_->LastPublishedSequence();
1210
  }
1211
  for (auto mgd_iter : multiget_cf_data) {
1212 1213
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
1214
  }
1215
  mutex_.Unlock();
1216

1217 1218
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
1219

1220
  // Note: this always resizes the values array
1221 1222 1223
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
1224 1225

  // Keep track of bytes that we read for statistics-recording later
1226
  uint64_t bytes_read = 0;
L
Lei Jin 已提交
1227
  PERF_TIMER_STOP(get_snapshot_time);
1228 1229 1230 1231

  // For each of the given keys, apply the entire "get" process as follows:
  // First look in the memtable, then in the immutable memtable (if any).
  // s is both in/out. When in, s could either be OK or MergeInProgress.
1232
  // merge_operands will contain the sequence of merges in the latter case.
1233
  size_t num_found = 0;
1234
  for (size_t i = 0; i < num_keys; ++i) {
1235
    merge_context.Clear();
1236
    Status& s = stat_list[i];
1237 1238 1239
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
1240
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
A
Andrew Kryczka 已提交
1241
    RangeDelAggregator range_del_agg(cfh->cfd()->internal_comparator(),
1242
                                     snapshot);
1243
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
1244 1245 1246
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
1247
    bool skip_memtable =
1248 1249
        (read_options.read_tier == kPersistedTier &&
         has_unpersisted_data_.load(std::memory_order_relaxed));
1250 1251
    bool done = false;
    if (!skip_memtable) {
M
Maysam Yabandeh 已提交
1252
      if (super_version->mem->Get(lkey, value, &s, &merge_context,
A
Andrew Kryczka 已提交
1253
                                  &range_del_agg, read_options)) {
1254
        done = true;
1255
        RecordTick(stats_, MEMTABLE_HIT);
M
Maysam Yabandeh 已提交
1256
      } else if (super_version->imm->Get(lkey, value, &s, &merge_context,
A
Andrew Kryczka 已提交
1257
                                         &range_del_agg, read_options)) {
1258
        done = true;
1259
        RecordTick(stats_, MEMTABLE_HIT);
1260 1261 1262
      }
    }
    if (!done) {
M
Maysam Yabandeh 已提交
1263
      PinnableSlice pinnable_val;
1264
      PERF_TIMER_GUARD(get_from_output_files_time);
M
Maysam Yabandeh 已提交
1265 1266 1267
      super_version->current->Get(read_options, lkey, &pinnable_val, &s,
                                  &merge_context, &range_del_agg);
      value->assign(pinnable_val.data(), pinnable_val.size());
1268
      RecordTick(stats_, MEMTABLE_MISS);
1269 1270 1271
    }

    if (s.ok()) {
M
Maysam Yabandeh 已提交
1272
      bytes_read += value->size();
1273
      num_found++;
1274 1275 1276 1277
    }
  }

  // Post processing (decrement reference counts and record statistics)
1278
  PERF_TIMER_GUARD(get_post_process_time);
1279 1280
  autovector<SuperVersion*> superversions_to_delete;

I
Igor Canadi 已提交
1281
  // TODO(icanadi) do we need lock here or just around Cleanup()?
1282 1283 1284 1285 1286 1287
  mutex_.Lock();
  for (auto mgd_iter : multiget_cf_data) {
    auto mgd = mgd_iter.second;
    if (mgd->super_version->Unref()) {
      mgd->super_version->Cleanup();
      superversions_to_delete.push_back(mgd->super_version);
1288 1289
    }
  }
1290 1291 1292 1293 1294 1295 1296
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
1297
  }
1298

L
Lei Jin 已提交
1299 1300
  RecordTick(stats_, NUMBER_MULTIGET_CALLS);
  RecordTick(stats_, NUMBER_MULTIGET_KEYS_READ, num_keys);
1301
  RecordTick(stats_, NUMBER_MULTIGET_KEYS_FOUND, num_found);
L
Lei Jin 已提交
1302
  RecordTick(stats_, NUMBER_MULTIGET_BYTES_READ, bytes_read);
1303
  MeasureTime(stats_, BYTES_PER_MULTIGET, bytes_read);
1304
  PERF_COUNTER_ADD(multiget_read_bytes, bytes_read);
L
Lei Jin 已提交
1305
  PERF_TIMER_STOP(get_post_process_time);
1306

1307
  return stat_list;
1308 1309
}

L
Lei Jin 已提交
1310
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
Y
Yi Wu 已提交
1311
                                  const std::string& column_family,
1312
                                  ColumnFamilyHandle** handle) {
Y
Yi Wu 已提交
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
  assert(handle != nullptr);
  Status s = CreateColumnFamilyImpl(cf_options, column_family, handle);
  if (s.ok()) {
    s = WriteOptionsFile(true /*need_mutex_lock*/,
                         true /*need_enter_write_thread*/);
  }
  return s;
}

Status DBImpl::CreateColumnFamilies(
    const ColumnFamilyOptions& cf_options,
    const std::vector<std::string>& column_family_names,
    std::vector<ColumnFamilyHandle*>* handles) {
  assert(handles != nullptr);
  handles->clear();
  size_t num_cf = column_family_names.size();
  Status s;
  bool success_once = false;
  for (size_t i = 0; i < num_cf; i++) {
    ColumnFamilyHandle* handle;
    s = CreateColumnFamilyImpl(cf_options, column_family_names[i], &handle);
    if (!s.ok()) {
      break;
    }
    handles->push_back(handle);
    success_once = true;
  }
  if (success_once) {
    Status persist_options_status = WriteOptionsFile(
        true /*need_mutex_lock*/, true /*need_enter_write_thread*/);
    if (s.ok() && !persist_options_status.ok()) {
      s = persist_options_status;
    }
  }
  return s;
}

Status DBImpl::CreateColumnFamilies(
    const std::vector<ColumnFamilyDescriptor>& column_families,
    std::vector<ColumnFamilyHandle*>* handles) {
  assert(handles != nullptr);
  handles->clear();
  size_t num_cf = column_families.size();
  Status s;
  bool success_once = false;
  for (size_t i = 0; i < num_cf; i++) {
    ColumnFamilyHandle* handle;
    s = CreateColumnFamilyImpl(column_families[i].options,
                               column_families[i].name, &handle);
    if (!s.ok()) {
      break;
    }
    handles->push_back(handle);
    success_once = true;
  }
  if (success_once) {
    Status persist_options_status = WriteOptionsFile(
        true /*need_mutex_lock*/, true /*need_enter_write_thread*/);
    if (s.ok() && !persist_options_status.ok()) {
      s = persist_options_status;
    }
  }
  return s;
}

Status DBImpl::CreateColumnFamilyImpl(const ColumnFamilyOptions& cf_options,
                                      const std::string& column_family_name,
                                      ColumnFamilyHandle** handle) {
Y
Yueh-Hsuan Chiang 已提交
1381
  Status s;
1382
  Status persist_options_status;
I
Igor Canadi 已提交
1383
  *handle = nullptr;
1384 1385

  s = CheckCompressionSupported(cf_options);
1386
  if (s.ok() && immutable_db_options_.allow_concurrent_memtable_write) {
1387 1388
    s = CheckConcurrentWritesSupported(cf_options);
  }
1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399
  if (s.ok()) {
    s = CheckCFPathsSupported(initial_db_options_, cf_options);
  }
  if (s.ok()) {
    for (auto& cf_path : cf_options.cf_paths) {
      s = env_->CreateDirIfMissing(cf_path.path);
      if (!s.ok()) {
        break;
      }
    }
  }
1400 1401 1402 1403
  if (!s.ok()) {
    return s;
  }

1404
  SuperVersionContext sv_context(/* create_superversion */ true);
Y
Yueh-Hsuan Chiang 已提交
1405
  {
1406
    InstrumentedMutexLock l(&mutex_);
I
Igor Canadi 已提交
1407

Y
Yueh-Hsuan Chiang 已提交
1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420
    if (versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name) !=
        nullptr) {
      return Status::InvalidArgument("Column family already exists");
    }
    VersionEdit edit;
    edit.AddColumnFamily(column_family_name);
    uint32_t new_id = versions_->GetColumnFamilySet()->GetNextColumnFamilyID();
    edit.SetColumnFamily(new_id);
    edit.SetLogNumber(logfile_number_);
    edit.SetComparatorName(cf_options.comparator->Name());

    // LogAndApply will both write the creation in MANIFEST and create
    // ColumnFamilyData object
I
Igor Canadi 已提交
1421
    {  // write thread
1422 1423
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
I
Igor Canadi 已提交
1424 1425
      // LogAndApply will both write the creation in MANIFEST and create
      // ColumnFamilyData object
1426 1427 1428
      s = versions_->LogAndApply(nullptr, MutableCFOptions(cf_options), &edit,
                                 &mutex_, directories_.GetDbDir(), false,
                                 &cf_options);
1429
      write_thread_.ExitUnbatched(&w);
I
Igor Canadi 已提交
1430
    }
1431 1432 1433 1434 1435 1436
    if (s.ok()) {
      auto* cfd =
          versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
      assert(cfd != nullptr);
      s = cfd->AddDirectories();
    }
Y
Yueh-Hsuan Chiang 已提交
1437 1438 1439 1440 1441
    if (s.ok()) {
      single_column_family_mode_ = false;
      auto* cfd =
          versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
      assert(cfd != nullptr);
1442 1443
      InstallSuperVersionAndScheduleWork(cfd, &sv_context,
                                         *cfd->GetLatestMutableCFOptions());
S
sdong 已提交
1444 1445 1446 1447 1448

      if (!cfd->mem()->IsSnapshotSupported()) {
        is_snapshot_supported_ = false;
      }

1449 1450
      cfd->set_initialized();

Y
Yueh-Hsuan Chiang 已提交
1451
      *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
1452 1453 1454
      ROCKS_LOG_INFO(immutable_db_options_.info_log,
                     "Created column family [%s] (ID %u)",
                     column_family_name.c_str(), (unsigned)cfd->GetID());
Y
Yueh-Hsuan Chiang 已提交
1455
    } else {
1456 1457 1458
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "Creating column family [%s] FAILED -- %s",
                      column_family_name.c_str(), s.ToString().c_str());
Y
Yueh-Hsuan Chiang 已提交
1459
    }
1460
  }  // InstrumentedMutexLock l(&mutex_)
Y
Yueh-Hsuan Chiang 已提交
1461

1462
  sv_context.Clean();
Y
Yueh-Hsuan Chiang 已提交
1463
  // this is outside the mutex
1464
  if (s.ok()) {
1465 1466
    NewThreadStatusCfInfo(
        reinterpret_cast<ColumnFamilyHandleImpl*>(*handle)->cfd());
1467
  }
1468
  return s;
1469 1470
}

1471
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
Y
Yi Wu 已提交
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
  assert(column_family != nullptr);
  Status s = DropColumnFamilyImpl(column_family);
  if (s.ok()) {
    s = WriteOptionsFile(true /*need_mutex_lock*/,
                         true /*need_enter_write_thread*/);
  }
  return s;
}

Status DBImpl::DropColumnFamilies(
    const std::vector<ColumnFamilyHandle*>& column_families) {
  Status s;
  bool success_once = false;
  for (auto* handle : column_families) {
    s = DropColumnFamilyImpl(handle);
    if (!s.ok()) {
      break;
    }
    success_once = true;
  }
  if (success_once) {
    Status persist_options_status = WriteOptionsFile(
        true /*need_mutex_lock*/, true /*need_enter_write_thread*/);
    if (s.ok() && !persist_options_status.ok()) {
      s = persist_options_status;
    }
  }
  return s;
}

Status DBImpl::DropColumnFamilyImpl(ColumnFamilyHandle* column_family) {
1503 1504 1505
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
1506 1507
    return Status::InvalidArgument("Can't drop default column family");
  }
1508

S
sdong 已提交
1509 1510
  bool cf_support_snapshot = cfd->mem()->IsSnapshotSupported();

I
Igor Canadi 已提交
1511 1512
  VersionEdit edit;
  edit.DropColumnFamily();
1513 1514
  edit.SetColumnFamily(cfd->GetID());

1515
  Status s;
1516
  {
1517
    InstrumentedMutexLock l(&mutex_);
1518 1519 1520 1521
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
1522
      // we drop column family from a single write thread
1523 1524
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
1525 1526
      s = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(), &edit,
                                 &mutex_);
1527
      write_thread_.ExitUnbatched(&w);
1528
    }
Y
Yi Wu 已提交
1529 1530 1531 1532 1533
    if (s.ok()) {
      auto* mutable_cf_options = cfd->GetLatestMutableCFOptions();
      max_total_in_memory_state_ -= mutable_cf_options->write_buffer_size *
                                    mutable_cf_options->max_write_buffer_number;
    }
S
sdong 已提交
1534 1535 1536 1537 1538 1539

    if (!cf_support_snapshot) {
      // Dropped Column Family doesn't support snapshot. Need to recalculate
      // is_snapshot_supported_.
      bool new_is_snapshot_supported = true;
      for (auto c : *versions_->GetColumnFamilySet()) {
1540
        if (!c->IsDropped() && !c->mem()->IsSnapshotSupported()) {
S
sdong 已提交
1541 1542 1543 1544 1545 1546
          new_is_snapshot_supported = false;
          break;
        }
      }
      is_snapshot_supported_ = new_is_snapshot_supported;
    }
1547
    bg_cv_.SignalAll();
1548
  }
1549

1550
  if (s.ok()) {
Y
Yueh-Hsuan Chiang 已提交
1551 1552 1553 1554
    // Note that here we erase the associated cf_info of the to-be-dropped
    // cfd before its ref-count goes to zero to avoid having to erase cf_info
    // later inside db_mutex.
    EraseThreadStatusCfInfo(cfd);
I
Igor Canadi 已提交
1555
    assert(cfd->IsDropped());
1556 1557
    ROCKS_LOG_INFO(immutable_db_options_.info_log,
                   "Dropped column family with id %u\n", cfd->GetID());
1558
  } else {
1559 1560 1561
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Dropping column family with id %u FAILED -- %s\n",
                    cfd->GetID(), s.ToString().c_str());
1562 1563
  }

1564
  return s;
1565 1566
}

L
Lei Jin 已提交
1567
bool DBImpl::KeyMayExist(const ReadOptions& read_options,
1568 1569
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
M
Maysam Yabandeh 已提交
1570
  assert(value != nullptr);
1571
  if (value_found != nullptr) {
K
Kai Liu 已提交
1572 1573
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
1574
  }
L
Lei Jin 已提交
1575
  ReadOptions roptions = read_options;
1576
  roptions.read_tier = kBlockCacheTier;  // read from block cache only
M
Maysam Yabandeh 已提交
1577 1578 1579
  PinnableSlice pinnable_val;
  auto s = GetImpl(roptions, column_family, key, &pinnable_val, value_found);
  value->assign(pinnable_val.data(), pinnable_val.size());
K
Kai Liu 已提交
1580

1581
  // If block_cache is enabled and the index block of the table didn't
K
Kai Liu 已提交
1582 1583 1584
  // not present in block_cache, the return value will be Status::Incomplete.
  // In this case, key may still exist in the table.
  return s.ok() || s.IsIncomplete();
1585 1586
}

1587
Iterator* DBImpl::NewIterator(const ReadOptions& read_options,
1588
                              ColumnFamilyHandle* column_family) {
S
Siying Dong 已提交
1589 1590 1591 1592
  if (read_options.managed) {
    return NewErrorIterator(
        Status::NotSupported("Managed iterator is not supported anymore."));
  }
1593
  Iterator* result = nullptr;
1594 1595 1596 1597
  if (read_options.read_tier == kPersistedTier) {
    return NewErrorIterator(Status::NotSupported(
        "ReadTier::kPersistedData is not yet supported in iterators."));
  }
1598 1599 1600 1601 1602
  // if iterator wants internal keys, we can only proceed if
  // we can guarantee the deletes haven't been processed yet
  if (immutable_db_options_.preserve_deletes &&
      read_options.iter_start_seqnum > 0 &&
      read_options.iter_start_seqnum < preserve_deletes_seqnum_.load()) {
1603 1604 1605 1606
    return NewErrorIterator(Status::InvalidArgument(
        "Iterator requested internal keys which are too old and are not"
        " guaranteed to be preserved, try larger iter_start_seqnum opt."));
  }
1607 1608
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
Y
Yi Wu 已提交
1609
  ReadCallback* read_callback = nullptr;  // No read callback provided.
S
Siying Dong 已提交
1610
if (read_options.tailing) {
I
Igor Canadi 已提交
1611 1612
#ifdef ROCKSDB_LITE
    // not supported in lite version
1613 1614
    result = nullptr;

I
Igor Canadi 已提交
1615
#else
1616 1617
    SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
    auto iter = new ForwardIterator(this, read_options, cfd, sv);
1618 1619 1620 1621 1622
    result = NewDBIterator(
        env_, read_options, *cfd->ioptions(), sv->mutable_cf_options,
        cfd->user_comparator(), iter, kMaxSequenceNumber,
        sv->mutable_cf_options.max_sequential_skip_in_iterations,
        read_callback);
I
Igor Canadi 已提交
1623
#endif
T
Tomislav Novak 已提交
1624
  } else {
1625
    // Note: no need to consider the special case of
1626
    // last_seq_same_as_publish_seq_==false since NewIterator is overridden in
1627
    // WritePreparedTxnDB
Y
Yi Wu 已提交
1628 1629 1630
    auto snapshot = read_options.snapshot != nullptr
                        ? read_options.snapshot->GetSequenceNumber()
                        : versions_->LastSequence();
1631
    result = NewIteratorImpl(read_options, cfd, snapshot, read_callback);
1632
  }
1633
  return result;
1634 1635
}

Y
Yi Wu 已提交
1636 1637 1638
ArenaWrappedDBIter* DBImpl::NewIteratorImpl(const ReadOptions& read_options,
                                            ColumnFamilyData* cfd,
                                            SequenceNumber snapshot,
Y
Yi Wu 已提交
1639
                                            ReadCallback* read_callback,
1640 1641
                                            bool allow_blob,
                                            bool allow_refresh) {
Y
Yi Wu 已提交
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686
  SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);

  // Try to generate a DB iterator tree in continuous memory area to be
  // cache friendly. Here is an example of result:
  // +-------------------------------+
  // |                               |
  // | ArenaWrappedDBIter            |
  // |  +                            |
  // |  +---> Inner Iterator   ------------+
  // |  |                            |     |
  // |  |    +-- -- -- -- -- -- -- --+     |
  // |  +--- | Arena                 |     |
  // |       |                       |     |
  // |          Allocated Memory:    |     |
  // |       |   +-------------------+     |
  // |       |   | DBIter            | <---+
  // |           |  +                |
  // |       |   |  +-> iter_  ------------+
  // |       |   |                   |     |
  // |       |   +-------------------+     |
  // |       |   | MergingIterator   | <---+
  // |           |  +                |
  // |       |   |  +->child iter1  ------------+
  // |       |   |  |                |          |
  // |           |  +->child iter2  ----------+ |
  // |       |   |  |                |        | |
  // |       |   |  +->child iter3  --------+ | |
  // |           |                   |      | | |
  // |       |   +-------------------+      | | |
  // |       |   | Iterator1         | <--------+
  // |       |   +-------------------+      | |
  // |       |   | Iterator2         | <------+
  // |       |   +-------------------+      |
  // |       |   | Iterator3         | <----+
  // |       |   +-------------------+
  // |       |                       |
  // +-------+-----------------------+
  //
  // ArenaWrappedDBIter inlines an arena area where all the iterators in
  // the iterator tree are allocated in the order of being accessed when
  // querying.
  // Laying out the iterators in the order of being accessed makes it more
  // likely that any iterator pointer is close to the iterator it points to so
  // that they are likely to be in the same cache line and/or page.
  ArenaWrappedDBIter* db_iter = NewArenaWrappedDbIterator(
1687
      env_, read_options, *cfd->ioptions(), sv->mutable_cf_options, snapshot,
Y
Yi Wu 已提交
1688
      sv->mutable_cf_options.max_sequential_skip_in_iterations,
Y
Yi Wu 已提交
1689
      sv->version_number, read_callback,
1690 1691
      ((read_options.snapshot != nullptr) ? nullptr : this), cfd, allow_blob,
      allow_refresh);
Y
Yi Wu 已提交
1692 1693 1694 1695 1696 1697 1698 1699 1700

  InternalIterator* internal_iter =
      NewInternalIterator(read_options, cfd, sv, db_iter->GetArena(),
                          db_iter->GetRangeDelAggregator());
  db_iter->SetIterUnderDBIter(internal_iter);

  return db_iter;
}

S
Siying Dong 已提交
1701 1702 1703 1704
Status DBImpl::NewIterators(
    const ReadOptions& read_options,
    const std::vector<ColumnFamilyHandle*>& column_families,
    std::vector<Iterator*>* iterators) {
S
Siying Dong 已提交
1705 1706 1707
  if (read_options.managed) {
    return Status::NotSupported("Managed iterator is not supported anymore.");
  }
S
Siying Dong 已提交
1708 1709 1710 1711
  if (read_options.read_tier == kPersistedTier) {
    return Status::NotSupported(
        "ReadTier::kPersistedData is not yet supported in iterators.");
  }
Y
Yi Wu 已提交
1712
  ReadCallback* read_callback = nullptr;  // No read callback provided.
S
Siying Dong 已提交
1713 1714
  iterators->clear();
  iterators->reserve(column_families.size());
S
Siying Dong 已提交
1715
  if (read_options.tailing) {
S
Siying Dong 已提交
1716 1717 1718 1719 1720 1721 1722 1723 1724
#ifdef ROCKSDB_LITE
    return Status::InvalidArgument(
        "Tailing interator not supported in RocksDB lite");
#else
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
      auto iter = new ForwardIterator(this, read_options, cfd, sv);
      iterators->push_back(NewDBIterator(
1725 1726
          env_, read_options, *cfd->ioptions(), sv->mutable_cf_options,
          cfd->user_comparator(), iter, kMaxSequenceNumber,
Y
Yi Wu 已提交
1727 1728
          sv->mutable_cf_options.max_sequential_skip_in_iterations,
          read_callback));
1729
    }
S
Siying Dong 已提交
1730 1731
#endif
  } else {
1732
    // Note: no need to consider the special case of
1733
    // last_seq_same_as_publish_seq_==false since NewIterators is overridden in
1734
    // WritePreparedTxnDB
Y
Yi Wu 已提交
1735 1736 1737
    auto snapshot = read_options.snapshot != nullptr
                        ? read_options.snapshot->GetSequenceNumber()
                        : versions_->LastSequence();
S
Siying Dong 已提交
1738
    for (size_t i = 0; i < column_families.size(); ++i) {
1739 1740
      auto* cfd =
          reinterpret_cast<ColumnFamilyHandleImpl*>(column_families[i])->cfd();
Y
Yi Wu 已提交
1741 1742
      iterators->push_back(
          NewIteratorImpl(read_options, cfd, snapshot, read_callback));
1743 1744
    }
  }
1745

I
Igor Canadi 已提交
1746
  return Status::OK();
S
Stanislau Hlebik 已提交
1747 1748
}

S
Siying Dong 已提交
1749
const Snapshot* DBImpl::GetSnapshot() { return GetSnapshotImpl(false); }
1750

S
Siying Dong 已提交
1751 1752 1753
#ifndef ROCKSDB_LITE
const Snapshot* DBImpl::GetSnapshotForWriteConflictBoundary() {
  return GetSnapshotImpl(true);
1754 1755 1756
}
#endif  // ROCKSDB_LITE

1757
SnapshotImpl* DBImpl::GetSnapshotImpl(bool is_write_conflict_boundary) {
S
Siying Dong 已提交
1758 1759 1760 1761 1762 1763 1764 1765 1766
  int64_t unix_time = 0;
  env_->GetCurrentTime(&unix_time);  // Ignore error
  SnapshotImpl* s = new SnapshotImpl;

  InstrumentedMutexLock l(&mutex_);
  // returns null if the underlying memtable does not support snapshot.
  if (!is_snapshot_supported_) {
    delete s;
    return nullptr;
S
Sage Weil 已提交
1767
  }
1768
  auto snapshot_seq = last_seq_same_as_publish_seq_
1769
                          ? versions_->LastSequence()
1770
                          : versions_->LastPublishedSequence();
1771
  return snapshots_.New(s, snapshot_seq, unix_time, is_write_conflict_boundary);
S
Siying Dong 已提交
1772
}
1773

S
Siying Dong 已提交
1774 1775
void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  const SnapshotImpl* casted_s = reinterpret_cast<const SnapshotImpl*>(s);
S
Stanislau Hlebik 已提交
1776
  {
S
Siying Dong 已提交
1777 1778
    InstrumentedMutexLock l(&mutex_);
    snapshots_.Delete(casted_s);
1779 1780
    uint64_t oldest_snapshot;
    if (snapshots_.empty()) {
1781
      oldest_snapshot = last_seq_same_as_publish_seq_
1782
                            ? versions_->LastSequence()
1783
                            : versions_->LastPublishedSequence();
1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796
    } else {
      oldest_snapshot = snapshots_.oldest()->number_;
    }
    for (auto* cfd : *versions_->GetColumnFamilySet()) {
      cfd->current()->storage_info()->UpdateOldestSnapshot(oldest_snapshot);
      if (!cfd->current()
               ->storage_info()
               ->BottommostFilesMarkedForCompaction()
               .empty()) {
        SchedulePendingCompaction(cfd);
        MaybeScheduleFlushOrCompaction();
      }
    }
1797
  }
S
Siying Dong 已提交
1798
  delete casted_s;
1799 1800
}

I
Igor Canadi 已提交
1801
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
1802 1803 1804 1805 1806
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

1807 1808
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
1809
  auto version = cfd->current();
1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

  // Decrement the ref count
  mutex_.Lock();
  version->Unref();
  mutex_.Unlock();

  return s;
}
1822 1823

Status DBImpl::GetPropertiesOfTablesInRange(ColumnFamilyHandle* column_family,
1824
                                            const Range* range, std::size_t n,
1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844
                                            TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

  // Increment the ref count
  mutex_.Lock();
  auto version = cfd->current();
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfTablesInRange(range, n, props);

  // Decrement the ref count
  mutex_.Lock();
  version->Unref();
  mutex_.Unlock();

  return s;
}

I
Igor Canadi 已提交
1845
#endif  // ROCKSDB_LITE
1846

1847
const std::string& DBImpl::GetName() const { return dbname_; }
I
Igor Canadi 已提交
1848

1849
Env* DBImpl::GetEnv() const { return env_; }
1850

1851 1852
Options DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  InstrumentedMutexLock l(&mutex_);
1853
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
1854 1855
  return Options(BuildDBOptions(immutable_db_options_, mutable_db_options_),
                 cfh->cfd()->GetLatestCFOptions());
I
Igor Canadi 已提交
1856 1857
}

1858 1859 1860 1861
DBOptions DBImpl::GetDBOptions() const {
  InstrumentedMutexLock l(&mutex_);
  return BuildDBOptions(immutable_db_options_, mutable_db_options_);
}
1862

1863
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
1864
                         const Slice& property, std::string* value) {
1865
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
1866
  value->clear();
1867
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
1868 1869 1870
  if (property_info == nullptr) {
    return false;
  } else if (property_info->handle_int) {
1871
    uint64_t int_value;
1872 1873
    bool ret_value =
        GetIntPropertyInternal(cfd, *property_info, false, &int_value);
1874
    if (ret_value) {
1875
      *value = ToString(int_value);
1876 1877
    }
    return ret_value;
1878
  } else if (property_info->handle_string) {
1879
    InstrumentedMutexLock l(&mutex_);
1880
    return cfd->internal_stats()->GetStringProperty(*property_info, property,
1881
                                                    value);
1882 1883 1884 1885 1886 1887 1888
  } else if (property_info->handle_string_dbimpl) {
    std::string tmp_value;
    bool ret_value = (this->*(property_info->handle_string_dbimpl))(&tmp_value);
    if (ret_value) {
      *value = tmp_value;
    }
    return ret_value;
1889
  }
1890 1891 1892 1893
  // Shouldn't reach here since exactly one of handle_string and handle_int
  // should be non-nullptr.
  assert(false);
  return false;
J
jorlow@chromium.org 已提交
1894 1895
}

1896 1897
bool DBImpl::GetMapProperty(ColumnFamilyHandle* column_family,
                            const Slice& property,
1898
                            std::map<std::string, std::string>* value) {
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  value->clear();
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
  if (property_info == nullptr) {
    return false;
  } else if (property_info->handle_map) {
    InstrumentedMutexLock l(&mutex_);
    return cfd->internal_stats()->GetMapProperty(*property_info, property,
                                                 value);
  }
  // If we reach this point it means that handle_map is not provided for the
  // requested property
  return false;
}

1914 1915
bool DBImpl::GetIntProperty(ColumnFamilyHandle* column_family,
                            const Slice& property, uint64_t* value) {
1916 1917
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1918 1919
    return false;
  }
1920
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
1921
  return GetIntPropertyInternal(cfd, *property_info, false, value);
1922 1923
}

1924
bool DBImpl::GetIntPropertyInternal(ColumnFamilyData* cfd,
1925 1926 1927 1928
                                    const DBPropertyInfo& property_info,
                                    bool is_locked, uint64_t* value) {
  assert(property_info.handle_int != nullptr);
  if (!property_info.need_out_of_mutex) {
1929 1930
    if (is_locked) {
      mutex_.AssertHeld();
1931
      return cfd->internal_stats()->GetIntProperty(property_info, value, this);
1932 1933
    } else {
      InstrumentedMutexLock l(&mutex_);
1934
      return cfd->internal_stats()->GetIntProperty(property_info, value, this);
1935
    }
1936
  } else {
1937 1938 1939 1940 1941 1942
    SuperVersion* sv = nullptr;
    if (!is_locked) {
      sv = GetAndRefSuperVersion(cfd);
    } else {
      sv = cfd->GetSuperVersion();
    }
1943 1944

    bool ret = cfd->internal_stats()->GetIntPropertyOutOfMutex(
1945
        property_info, sv->current, value);
1946

1947 1948 1949
    if (!is_locked) {
      ReturnAndCleanupSuperVersion(cfd, sv);
    }
1950 1951 1952 1953 1954

    return ret;
  }
}

1955 1956 1957 1958 1959 1960 1961 1962 1963 1964
bool DBImpl::GetPropertyHandleOptionsStatistics(std::string* value) {
  assert(value != nullptr);
  Statistics* statistics = immutable_db_options_.statistics.get();
  if (!statistics) {
    return false;
  }
  *value = statistics->ToString();
  return true;
}

S
Siying Dong 已提交
1965 1966 1967 1968
#ifndef ROCKSDB_LITE
Status DBImpl::ResetStats() {
  InstrumentedMutexLock l(&mutex_);
  for (auto* cfd : *versions_->GetColumnFamilySet()) {
1969 1970 1971
    if (cfd->initialized()) {
      cfd->internal_stats()->Clear();
    }
S
Siying Dong 已提交
1972 1973 1974 1975 1976
  }
  return Status::OK();
}
#endif  // ROCKSDB_LITE

1977 1978
bool DBImpl::GetAggregatedIntProperty(const Slice& property,
                                      uint64_t* aggregated_value) {
1979 1980
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1981 1982 1983 1984 1985 1986 1987 1988 1989
    return false;
  }

  uint64_t sum = 0;
  {
    // Needs mutex to protect the list of column families.
    InstrumentedMutexLock l(&mutex_);
    uint64_t value;
    for (auto* cfd : *versions_->GetColumnFamilySet()) {
1990 1991 1992
      if (!cfd->initialized()) {
        continue;
      }
1993
      if (GetIntPropertyInternal(cfd, *property_info, true, &value)) {
1994 1995 1996 1997 1998 1999 2000 2001 2002 2003
        sum += value;
      } else {
        return false;
      }
    }
  }
  *aggregated_value = sum;
  return true;
}

2004 2005
SuperVersion* DBImpl::GetAndRefSuperVersion(ColumnFamilyData* cfd) {
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
I
Igor Canadi 已提交
2006
  return cfd->GetThreadLocalSuperVersion(&mutex_);
2007 2008
}

A
agiardullo 已提交
2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020
// REQUIRED: this function should only be called on the write thread or if the
// mutex is held.
SuperVersion* DBImpl::GetAndRefSuperVersion(uint32_t column_family_id) {
  auto column_family_set = versions_->GetColumnFamilySet();
  auto cfd = column_family_set->GetColumnFamily(column_family_id);
  if (!cfd) {
    return nullptr;
  }

  return GetAndRefSuperVersion(cfd);
}

2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033
void DBImpl::CleanupSuperVersion(SuperVersion* sv) {
  // Release SuperVersion
  if (sv->Unref()) {
    {
      InstrumentedMutexLock l(&mutex_);
      sv->Cleanup();
    }
    delete sv;
    RecordTick(stats_, NUMBER_SUPERVERSION_CLEANUPS);
  }
  RecordTick(stats_, NUMBER_SUPERVERSION_RELEASES);
}

2034 2035
void DBImpl::ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd,
                                          SuperVersion* sv) {
2036 2037
  if (!cfd->ReturnThreadLocalSuperVersion(sv)) {
    CleanupSuperVersion(sv);
2038
  }
2039 2040
}

A
agiardullo 已提交
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064
// REQUIRED: this function should only be called on the write thread.
void DBImpl::ReturnAndCleanupSuperVersion(uint32_t column_family_id,
                                          SuperVersion* sv) {
  auto column_family_set = versions_->GetColumnFamilySet();
  auto cfd = column_family_set->GetColumnFamily(column_family_id);

  // If SuperVersion is held, and we successfully fetched a cfd using
  // GetAndRefSuperVersion(), it must still exist.
  assert(cfd != nullptr);
  ReturnAndCleanupSuperVersion(cfd, sv);
}

// REQUIRED: this function should only be called on the write thread or if the
// mutex is held.
ColumnFamilyHandle* DBImpl::GetColumnFamilyHandle(uint32_t column_family_id) {
  ColumnFamilyMemTables* cf_memtables = column_family_memtables_.get();

  if (!cf_memtables->Seek(column_family_id)) {
    return nullptr;
  }

  return cf_memtables->GetColumnFamilyHandle();
}

A
Anirban Rahut 已提交
2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078
// REQUIRED: mutex is NOT held.
ColumnFamilyHandle* DBImpl::GetColumnFamilyHandleUnlocked(
    uint32_t column_family_id) {
  ColumnFamilyMemTables* cf_memtables = column_family_memtables_.get();

  InstrumentedMutexLock l(&mutex_);

  if (!cf_memtables->Seek(column_family_id)) {
    return nullptr;
  }

  return cf_memtables->GetColumnFamilyHandle();
}

2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100
void DBImpl::GetApproximateMemTableStats(ColumnFamilyHandle* column_family,
                                         const Range& range,
                                         uint64_t* const count,
                                         uint64_t* const size) {
  ColumnFamilyHandleImpl* cfh =
      reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  ColumnFamilyData* cfd = cfh->cfd();
  SuperVersion* sv = GetAndRefSuperVersion(cfd);

  // Convert user_key into a corresponding internal key.
  InternalKey k1(range.start, kMaxSequenceNumber, kValueTypeForSeek);
  InternalKey k2(range.limit, kMaxSequenceNumber, kValueTypeForSeek);
  MemTable::MemTableStats memStats =
      sv->mem->ApproximateStats(k1.Encode(), k2.Encode());
  MemTable::MemTableStats immStats =
      sv->imm->ApproximateStats(k1.Encode(), k2.Encode());
  *count = memStats.count + immStats.count;
  *size = memStats.size + immStats.size;

  ReturnAndCleanupSuperVersion(cfd, sv);
}

2101
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
2102
                                 const Range* range, int n, uint64_t* sizes,
2103 2104 2105
                                 uint8_t include_flags) {
  assert(include_flags & DB::SizeApproximationFlags::INCLUDE_FILES ||
         include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES);
J
jorlow@chromium.org 已提交
2106
  Version* v;
2107 2108
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
2109 2110
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
  v = sv->current;
J
jorlow@chromium.org 已提交
2111 2112 2113

  for (int i = 0; i < n; i++) {
    // Convert user_key into a corresponding internal key.
2114 2115
    InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
V
Vitaliy Liptchinsky 已提交
2116
    sizes[i] = 0;
2117
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_FILES) {
V
Vitaliy Liptchinsky 已提交
2118
      sizes[i] += versions_->ApproximateSize(v, k1.Encode(), k2.Encode());
2119 2120
    }
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES) {
2121 2122
      sizes[i] += sv->mem->ApproximateStats(k1.Encode(), k2.Encode()).size;
      sizes[i] += sv->imm->ApproximateStats(k1.Encode(), k2.Encode()).size;
2123
    }
J
jorlow@chromium.org 已提交
2124 2125
  }

2126
  ReturnAndCleanupSuperVersion(cfd, sv);
J
jorlow@chromium.org 已提交
2127 2128
}

I
Igor Canadi 已提交
2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144
std::list<uint64_t>::iterator
DBImpl::CaptureCurrentFileNumberInPendingOutputs() {
  // We need to remember the iterator of our insert, because after the
  // background job is done, we need to remove that element from
  // pending_outputs_.
  pending_outputs_.push_back(versions_->current_next_file_number());
  auto pending_outputs_inserted_elem = pending_outputs_.end();
  --pending_outputs_inserted_elem;
  return pending_outputs_inserted_elem;
}

void DBImpl::ReleaseFileNumberFromPendingOutputs(
    std::list<uint64_t>::iterator v) {
  pending_outputs_.erase(v);
}

I
Igor Canadi 已提交
2145 2146 2147 2148
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {
D
Dmitri Smirnov 已提交
2149

L
Lei Jin 已提交
2150
  RecordTick(stats_, GET_UPDATES_SINCE_CALLS);
I
Igor Canadi 已提交
2151 2152 2153
  if (seq > versions_->LastSequence()) {
    return Status::NotFound("Requested sequence not yet written in the db");
  }
I
Igor Canadi 已提交
2154
  return wal_manager_.GetUpdatesSince(seq, iter, read_options, versions_.get());
I
Igor Canadi 已提交
2155 2156
}

2157 2158 2159
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
2160 2161 2162
  WalFileType log_type;
  if (!ParseFileName(name, &number, &type, &log_type) ||
      (type != kTableFile && type != kLogFile)) {
2163 2164
    ROCKS_LOG_ERROR(immutable_db_options_.info_log, "DeleteFile %s failed.\n",
                    name.c_str());
2165 2166 2167
    return Status::InvalidArgument("Invalid file name");
  }

2168 2169 2170 2171
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
2172 2173 2174
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed - not archived log.\n",
                      name.c_str());
2175 2176
      return Status::NotSupported("Delete only supported for archived logs");
    }
C
Changli Gao 已提交
2177
    status = wal_manager_.DeleteFile(name, number);
2178
    if (!status.ok()) {
2179 2180 2181
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed -- %s.\n", name.c_str(),
                      status.ToString().c_str());
2182 2183 2184 2185
    }
    return status;
  }

2186
  int level;
I
Igor Canadi 已提交
2187
  FileMetaData* metadata;
2188
  ColumnFamilyData* cfd;
2189
  VersionEdit edit;
2190
  JobContext job_context(next_job_id_.fetch_add(1), true);
D
Dhruba Borthakur 已提交
2191
  {
2192
    InstrumentedMutexLock l(&mutex_);
2193
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
2194
    if (!status.ok()) {
2195 2196
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "DeleteFile %s failed. File not found\n", name.c_str());
I
Igor Canadi 已提交
2197
      job_context.Clean();
D
Dhruba Borthakur 已提交
2198 2199
      return Status::InvalidArgument("File not found");
    }
2200
    assert(level < cfd->NumberLevels());
2201

D
Dhruba Borthakur 已提交
2202
    // If the file is being compacted no need to delete.
2203
    if (metadata->being_compacted) {
2204 2205 2206
      ROCKS_LOG_INFO(immutable_db_options_.info_log,
                     "DeleteFile %s Skipped. File about to be compacted\n",
                     name.c_str());
I
Igor Canadi 已提交
2207
      job_context.Clean();
D
Dhruba Borthakur 已提交
2208
      return Status::OK();
2209 2210
    }

D
Dhruba Borthakur 已提交
2211 2212 2213
    // Only the files in the last level can be deleted externally.
    // This is to make sure that any deletion tombstones are not
    // lost. Check that the level passed is the last level.
S
sdong 已提交
2214
    auto* vstoreage = cfd->current()->storage_info();
I
Igor Canadi 已提交
2215
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
S
sdong 已提交
2216
      if (vstoreage->NumLevelFiles(i) != 0) {
2217 2218 2219
        ROCKS_LOG_WARN(immutable_db_options_.info_log,
                       "DeleteFile %s FAILED. File not in last level\n",
                       name.c_str());
I
Igor Canadi 已提交
2220
        job_context.Clean();
D
Dhruba Borthakur 已提交
2221 2222 2223
        return Status::InvalidArgument("File not in last level");
      }
    }
2224
    // if level == 0, it has to be the oldest file
S
sdong 已提交
2225 2226
    if (level == 0 &&
        vstoreage->LevelFiles(0).back()->fd.GetNumber() != number) {
2227 2228 2229 2230
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "DeleteFile %s failed ---"
                     " target file in level 0 must be the oldest.",
                     name.c_str());
I
Igor Canadi 已提交
2231
      job_context.Clean();
2232 2233 2234
      return Status::InvalidArgument("File in level 0, but not oldest");
    }
    edit.SetColumnFamily(cfd->GetID());
D
Dhruba Borthakur 已提交
2235
    edit.DeleteFile(level, number);
2236
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
2237
                                    &edit, &mutex_, directories_.GetDbDir());
I
Igor Canadi 已提交
2238
    if (status.ok()) {
Y
Yanqin Jin 已提交
2239 2240 2241
      InstallSuperVersionAndScheduleWork(
          cfd, &job_context.superversion_contexts[0],
          *cfd->GetLatestMutableCFOptions(), FlushReason::kDeleteFiles);
I
Igor Canadi 已提交
2242
    }
I
Igor Canadi 已提交
2243 2244
    FindObsoleteFiles(&job_context, false);
  }  // lock released here
2245

2246
  LogFlush(immutable_db_options_.info_log);
2247 2248 2249 2250 2251 2252 2253 2254 2255
  // remove files outside the db-lock
  if (job_context.HaveSomethingToDelete()) {
    // Call PurgeObsoleteFiles() without holding mutex.
    PurgeObsoleteFiles(job_context);
  }
  job_context.Clean();
  return status;
}

2256 2257 2258
Status DBImpl::DeleteFilesInRanges(ColumnFamilyHandle* column_family,
                                   const RangePtr* ranges, size_t n,
                                   bool include_end) {
2259 2260 2261 2262
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  ColumnFamilyData* cfd = cfh->cfd();
  VersionEdit edit;
2263
  std::set<FileMetaData*> deleted_files;
2264 2265 2266
  JobContext job_context(next_job_id_.fetch_add(1), true);
  {
    InstrumentedMutexLock l(&mutex_);
2267
    Version* input_version = cfd->current();
2268

2269
    auto* vstorage = input_version->storage_info();
2270 2271 2272 2273 2274
    for (size_t r = 0; r < n; r++) {
      auto begin = ranges[r].start, end = ranges[r].limit;
      for (int i = 1; i < cfd->NumberLevels(); i++) {
        if (vstorage->LevelFiles(i).empty() ||
            !vstorage->OverlapInLevel(i, begin, end)) {
2275
          continue;
2276
        }
2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291
        std::vector<FileMetaData*> level_files;
        InternalKey begin_storage, end_storage, *begin_key, *end_key;
        if (begin == nullptr) {
          begin_key = nullptr;
        } else {
          begin_storage.SetMinPossibleForUserKey(*begin);
          begin_key = &begin_storage;
        }
        if (end == nullptr) {
          end_key = nullptr;
        } else {
          end_storage.SetMaxPossibleForUserKey(*end);
          end_key = &end_storage;
        }

2292 2293 2294
        vstorage->GetCleanInputsWithinInterval(
            i, begin_key, end_key, &level_files, -1 /* hint_index */,
            nullptr /* file_index */);
2295 2296 2297 2298 2299 2300 2301 2302 2303 2304
        FileMetaData* level_file;
        for (uint32_t j = 0; j < level_files.size(); j++) {
          level_file = level_files[j];
          if (level_file->being_compacted) {
            continue;
          }
          if (deleted_files.find(level_file) != deleted_files.end()) {
            continue;
          }
          if (!include_end && end != nullptr &&
2305 2306
              cfd->user_comparator()->Compare(level_file->largest.user_key(),
                                              *end) == 0) {
2307 2308 2309 2310 2311 2312 2313
            continue;
          }
          edit.SetColumnFamily(cfd->GetID());
          edit.DeleteFile(i, level_file->fd.GetNumber());
          deleted_files.insert(level_file);
          level_file->being_compacted = true;
        }
2314 2315 2316
      }
    }
    if (edit.GetDeletedFiles().empty()) {
2317
      job_context.Clean();
2318 2319
      return Status::OK();
    }
2320
    input_version->Ref();
2321 2322 2323
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
                                    &edit, &mutex_, directories_.GetDbDir());
    if (status.ok()) {
Y
Yanqin Jin 已提交
2324 2325 2326
      InstallSuperVersionAndScheduleWork(
          cfd, &job_context.superversion_contexts[0],
          *cfd->GetLatestMutableCFOptions(), FlushReason::kDeleteFiles);
2327
    }
2328 2329 2330 2331
    for (auto* deleted_file : deleted_files) {
      deleted_file->being_compacted = false;
    }
    input_version->Unref();
2332 2333 2334
    FindObsoleteFiles(&job_context, false);
  }  // lock released here

2335
  LogFlush(immutable_db_options_.info_log);
I
Igor Canadi 已提交
2336
  // remove files outside the db-lock
I
Igor Canadi 已提交
2337
  if (job_context.HaveSomethingToDelete()) {
2338
    // Call PurgeObsoleteFiles() without holding mutex.
I
Igor Canadi 已提交
2339
    PurgeObsoleteFiles(job_context);
2340
  }
I
Igor Canadi 已提交
2341
  job_context.Clean();
2342 2343 2344
  return status;
}

2345
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
2346
  InstrumentedMutexLock l(&mutex_);
2347
  versions_->GetLiveFilesMetaData(metadata);
2348
}
2349

2350 2351
void DBImpl::GetColumnFamilyMetaData(ColumnFamilyHandle* column_family,
                                     ColumnFamilyMetaData* cf_meta) {
2352 2353 2354 2355 2356 2357 2358
  assert(column_family);
  auto* cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
  auto* sv = GetAndRefSuperVersion(cfd);
  sv->current->GetColumnFamilyMetaData(cf_meta);
  ReturnAndCleanupSuperVersion(cfd, sv);
}

I
Igor Canadi 已提交
2359
#endif  // ROCKSDB_LITE
2360

I
Igor Canadi 已提交
2361 2362 2363 2364 2365 2366 2367
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

  std::string corruption_messages;
  for (const auto& md : metadata) {
2368 2369
    // md.name has a leading "/".
    std::string file_path = md.db_path + md.name;
2370

I
Igor Canadi 已提交
2371 2372
    uint64_t fsize = 0;
    Status s = env_->GetFileSize(file_path, &fsize);
D
dyniusz 已提交
2373 2374 2375 2376
    if (!s.ok() &&
        env_->GetFileSize(Rocks2LevelTableFileName(file_path), &fsize).ok()) {
      s = Status::OK();
    }
I
Igor Canadi 已提交
2377 2378 2379 2380
    if (!s.ok()) {
      corruption_messages +=
          "Can't access " + md.name + ": " + s.ToString() + "\n";
    } else if (fsize != md.size) {
2381
      corruption_messages += "Sst file size mismatch: " + file_path +
I
Igor Canadi 已提交
2382
                             ". Size recorded in manifest " +
2383 2384
                             ToString(md.size) + ", actual size " +
                             ToString(fsize) + "\n";
I
Igor Canadi 已提交
2385 2386 2387 2388 2389 2390 2391 2392 2393
    }
  }
  if (corruption_messages.size() == 0) {
    return Status::OK();
  } else {
    return Status::Corruption(corruption_messages);
  }
}

2394
Status DBImpl::GetDbIdentity(std::string& identity) const {
2395 2396
  std::string idfilename = IdentityFileName(dbname_);
  const EnvOptions soptions;
2397 2398 2399 2400 2401 2402 2403 2404
  unique_ptr<SequentialFileReader> id_file_reader;
  Status s;
  {
    unique_ptr<SequentialFile> idfile;
    s = env_->NewSequentialFile(idfilename, &idfile, soptions);
    if (!s.ok()) {
      return s;
    }
2405 2406
    id_file_reader.reset(
        new SequentialFileReader(std::move(idfile), idfilename));
2407
  }
2408

2409 2410 2411 2412 2413
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
D
Dmitri Smirnov 已提交
2414
  char* buffer = reinterpret_cast<char*>(alloca(file_size));
2415
  Slice id;
2416
  s = id_file_reader->Read(static_cast<size_t>(file_size), &id, buffer);
2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427
  if (!s.ok()) {
    return s;
  }
  identity.assign(id.ToString());
  // If last character is '\n' remove it from identity
  if (identity.size() > 0 && identity.back() == '\n') {
    identity.pop_back();
  }
  return s;
}

2428
// Default implementation -- returns not supported status
A
Andrew Kryczka 已提交
2429 2430 2431
Status DB::CreateColumnFamily(const ColumnFamilyOptions& /*cf_options*/,
                              const std::string& /*column_family_name*/,
                              ColumnFamilyHandle** /*handle*/) {
2432
  return Status::NotSupported("");
2433
}
Y
Yi Wu 已提交
2434 2435

Status DB::CreateColumnFamilies(
A
Andrew Kryczka 已提交
2436 2437 2438
    const ColumnFamilyOptions& /*cf_options*/,
    const std::vector<std::string>& /*column_family_names*/,
    std::vector<ColumnFamilyHandle*>* /*handles*/) {
Y
Yi Wu 已提交
2439 2440 2441 2442
  return Status::NotSupported("");
}

Status DB::CreateColumnFamilies(
A
Andrew Kryczka 已提交
2443 2444
    const std::vector<ColumnFamilyDescriptor>& /*column_families*/,
    std::vector<ColumnFamilyHandle*>* /*handles*/) {
Y
Yi Wu 已提交
2445 2446 2447
  return Status::NotSupported("");
}

A
Andrew Kryczka 已提交
2448
Status DB::DropColumnFamily(ColumnFamilyHandle* /*column_family*/) {
2449
  return Status::NotSupported("");
2450
}
Y
Yi Wu 已提交
2451 2452

Status DB::DropColumnFamilies(
A
Andrew Kryczka 已提交
2453
    const std::vector<ColumnFamilyHandle*>& /*column_families*/) {
Y
Yi Wu 已提交
2454 2455 2456
  return Status::NotSupported("");
}

2457 2458 2459 2460
Status DB::DestroyColumnFamilyHandle(ColumnFamilyHandle* column_family) {
  delete column_family;
  return Status::OK();
}
2461

2462 2463 2464 2465 2466 2467 2468 2469 2470
DB::~DB() {}

Status DBImpl::Close() {
  if (!closed_) {
    closed_ = true;
    return CloseImpl();
  }
  return Status::OK();
}
J
jorlow@chromium.org 已提交
2471

2472 2473 2474
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
2475
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
2476 2477
}

2478
Snapshot::~Snapshot() {}
2479

2480 2481
Status DestroyDB(const std::string& dbname, const Options& options,
                 const std::vector<ColumnFamilyDescriptor>& column_families) {
D
Dmitri Smirnov 已提交
2482
  ImmutableDBOptions soptions(SanitizeOptions(dbname, options));
2483
  Env* env = soptions.env;
J
jorlow@chromium.org 已提交
2484
  std::vector<std::string> filenames;
2485

D
Dmitri Smirnov 已提交
2486 2487 2488
  // Reset the logger because it holds a handle to the
  // log file and prevents cleanup and directory removal
  soptions.info_log.reset();
J
jorlow@chromium.org 已提交
2489 2490
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2491

J
jorlow@chromium.org 已提交
2492
  FileLock* lock;
2493 2494
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2495 2496 2497
  if (result.ok()) {
    uint64_t number;
    FileType type;
2498
    InfoLogPrefix info_log_prefix(!soptions.db_log_dir.empty(), dbname);
D
Dmitri Smirnov 已提交
2499 2500 2501
    for (const auto& fname : filenames) {
      if (ParseFileName(fname, &number, info_log_prefix.prefix, &type) &&
        type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
2502
        Status del;
D
Dmitri Smirnov 已提交
2503
        std::string path_to_delete = dbname + "/" + fname;
K
Kosie van der Merwe 已提交
2504
        if (type == kMetaDatabase) {
2505 2506
          del = DestroyDB(path_to_delete, options);
        } else if (type == kTableFile) {
2507
          del = DeleteSSTFile(&soptions, path_to_delete, dbname);
K
Kosie van der Merwe 已提交
2508
        } else {
2509
          del = env->DeleteFile(path_to_delete);
K
Kosie van der Merwe 已提交
2510
        }
J
jorlow@chromium.org 已提交
2511 2512 2513 2514 2515
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2516

2517 2518
    std::vector<std::string> paths;

D
Dmitri Smirnov 已提交
2519 2520
    for (const auto& path : options.db_paths) {
      paths.emplace_back(path.path);
2521
    }
D
Dmitri Smirnov 已提交
2522 2523 2524
    for (const auto& cf : column_families) {
      for (const auto& path : cf.options.cf_paths) {
        paths.emplace_back(path.path);
2525 2526 2527 2528 2529 2530 2531 2532 2533 2534
      }
    }

    // Remove duplicate paths.
    // Note that we compare only the actual paths but not path ids.
    // This reason is that same path can appear at different path_ids
    // for different column families.
    std::sort(paths.begin(), paths.end());
    paths.erase(std::unique(paths.begin(), paths.end()), paths.end());

D
Dmitri Smirnov 已提交
2535 2536 2537 2538
    for (const auto& path : paths) {
      if (env->GetChildren(path, &filenames).ok()) {
        for (const auto& fname : filenames) {
          if (ParseFileName(fname, &number, &type) &&
2539
            type == kTableFile) {  // Lock file will be deleted at end
D
Dmitri Smirnov 已提交
2540 2541 2542 2543 2544
            std::string table_path = path + "/" + fname;
            Status del = DeleteSSTFile(&soptions, table_path, dbname);
            if (result.ok() && !del.ok()) {
              result = del;
            }
2545 2546
          }
        }
D
Dmitri Smirnov 已提交
2547
        env->DeleteDir(path);
2548 2549 2550
      }
    }

I
Igor Canadi 已提交
2551 2552
    std::vector<std::string> walDirFiles;
    std::string archivedir = ArchivalDirectory(dbname);
D
Dmitri Smirnov 已提交
2553
    bool wal_dir_exists = false;
I
Igor Canadi 已提交
2554
    if (dbname != soptions.wal_dir) {
D
Dmitri Smirnov 已提交
2555
      wal_dir_exists = env->GetChildren(soptions.wal_dir, &walDirFiles).ok();
I
Igor Canadi 已提交
2556 2557 2558
      archivedir = ArchivalDirectory(soptions.wal_dir);
    }

D
Dmitri Smirnov 已提交
2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571
    // Archive dir may be inside wal dir or dbname and should be
    // processed and removed before those otherwise we have issues
    // removing them
    std::vector<std::string> archiveFiles;
    if (env->GetChildren(archivedir, &archiveFiles).ok()) {
      // Delete archival files.
      for (const auto& file : archiveFiles) {
        if (ParseFileName(file, &number, &type) &&
          type == kLogFile) {
          Status del = env->DeleteFile(archivedir + "/" + file);
          if (result.ok() && !del.ok()) {
            result = del;
          }
I
Igor Canadi 已提交
2572 2573
        }
      }
D
Dmitri Smirnov 已提交
2574
      env->DeleteDir(archivedir);
I
Igor Canadi 已提交
2575 2576
    }

D
Dmitri Smirnov 已提交
2577 2578 2579 2580 2581 2582 2583 2584
    // Delete log files in the WAL dir
    if (wal_dir_exists) {
      for (const auto& file : walDirFiles) {
        if (ParseFileName(file, &number, &type) && type == kLogFile) {
          Status del = env->DeleteFile(LogFileName(soptions.wal_dir, number));
          if (result.ok() && !del.ok()) {
            result = del;
          }
2585 2586
        }
      }
D
Dmitri Smirnov 已提交
2587
      env->DeleteDir(soptions.wal_dir);
2588
    }
2589

J
jorlow@chromium.org 已提交
2590
    env->UnlockFile(lock);  // Ignore error since state is already gone
2591
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2592 2593 2594 2595 2596
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
  }
  return result;
}

Y
Yi Wu 已提交
2597 2598
Status DBImpl::WriteOptionsFile(bool need_mutex_lock,
                                bool need_enter_write_thread) {
2599
#ifndef ROCKSDB_LITE
Y
Yi Wu 已提交
2600 2601 2602 2603 2604 2605 2606 2607 2608
  WriteThread::Writer w;
  if (need_mutex_lock) {
    mutex_.Lock();
  } else {
    mutex_.AssertHeld();
  }
  if (need_enter_write_thread) {
    write_thread_.EnterUnbatched(&w, &mutex_);
  }
2609 2610 2611

  std::vector<std::string> cf_names;
  std::vector<ColumnFamilyOptions> cf_opts;
2612 2613 2614 2615 2616

  // This part requires mutex to protect the column family options
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    if (cfd->IsDropped()) {
      continue;
2617
    }
2618
    cf_names.push_back(cfd->GetName());
2619
    cf_opts.push_back(cfd->GetLatestCFOptions());
2620 2621
  }

2622 2623
  // Unlock during expensive operations.  New writes cannot get here
  // because the single write thread ensures all new writes get queued.
2624 2625
  DBOptions db_options =
      BuildDBOptions(immutable_db_options_, mutable_db_options_);
2626 2627
  mutex_.Unlock();

2628 2629 2630
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:1");
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:2");

2631 2632
  std::string file_name =
      TempOptionsFileName(GetName(), versions_->NewFileNumber());
2633 2634
  Status s =
      PersistRocksDBOptions(db_options, cf_names, cf_opts, file_name, GetEnv());
2635 2636 2637 2638

  if (s.ok()) {
    s = RenameTempFileToOptionsFile(file_name);
  }
Y
Yi Wu 已提交
2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653
  // restore lock
  if (!need_mutex_lock) {
    mutex_.Lock();
  }
  if (need_enter_write_thread) {
    write_thread_.ExitUnbatched(&w);
  }
  if (!s.ok()) {
    ROCKS_LOG_WARN(immutable_db_options_.info_log,
                   "Unnable to persist options -- %s", s.ToString().c_str());
    if (immutable_db_options_.fail_if_options_file_error) {
      return Status::IOError("Unable to persist options.",
                             s.ToString().c_str());
    }
  }
2654 2655 2656
#else
  (void)need_mutex_lock;
  (void)need_enter_write_thread;
2657
#endif  // !ROCKSDB_LITE
Y
Yi Wu 已提交
2658
  return Status::OK();
2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672
}

#ifndef ROCKSDB_LITE
namespace {
void DeleteOptionsFilesHelper(const std::map<uint64_t, std::string>& filenames,
                              const size_t num_files_to_keep,
                              const std::shared_ptr<Logger>& info_log,
                              Env* env) {
  if (filenames.size() <= num_files_to_keep) {
    return;
  }
  for (auto iter = std::next(filenames.begin(), num_files_to_keep);
       iter != filenames.end(); ++iter) {
    if (!env->DeleteFile(iter->second).ok()) {
2673 2674
      ROCKS_LOG_WARN(info_log, "Unable to delete options file %s",
                     iter->second.c_str());
2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704
    }
  }
}
}  // namespace
#endif  // !ROCKSDB_LITE

Status DBImpl::DeleteObsoleteOptionsFiles() {
#ifndef ROCKSDB_LITE
  std::vector<std::string> filenames;
  // use ordered map to store keep the filenames sorted from the newest
  // to the oldest.
  std::map<uint64_t, std::string> options_filenames;
  Status s;
  s = GetEnv()->GetChildren(GetName(), &filenames);
  if (!s.ok()) {
    return s;
  }
  for (auto& filename : filenames) {
    uint64_t file_number;
    FileType type;
    if (ParseFileName(filename, &file_number, &type) && type == kOptionsFile) {
      options_filenames.insert(
          {std::numeric_limits<uint64_t>::max() - file_number,
           GetName() + "/" + filename});
    }
  }

  // Keeps the latest 2 Options file
  const size_t kNumOptionsFilesKept = 2;
  DeleteOptionsFilesHelper(options_filenames, kNumOptionsFilesKept,
2705
                           immutable_db_options_.info_log, GetEnv());
2706 2707 2708 2709 2710 2711 2712 2713 2714
  return Status::OK();
#else
  return Status::OK();
#endif  // !ROCKSDB_LITE
}

Status DBImpl::RenameTempFileToOptionsFile(const std::string& file_name) {
#ifndef ROCKSDB_LITE
  Status s;
W
Wanning Jiang 已提交
2715 2716

  versions_->options_file_number_ = versions_->NewFileNumber();
2717
  std::string options_file_name =
W
Wanning Jiang 已提交
2718
      OptionsFileName(GetName(), versions_->options_file_number_);
2719 2720 2721
  // Retry if the file name happen to conflict with an existing one.
  s = GetEnv()->RenameFile(file_name, options_file_name);

2722 2723 2724
  if (0 == disable_delete_obsolete_files_) {
    DeleteObsoleteOptionsFiles();
  }
2725 2726
  return s;
#else
2727
  (void)file_name;
2728 2729 2730 2731
  return Status::OK();
#endif  // !ROCKSDB_LITE
}

D
Daniel Black 已提交
2732
#ifdef ROCKSDB_USING_THREAD_STATUS
2733

2734
void DBImpl::NewThreadStatusCfInfo(ColumnFamilyData* cfd) const {
2735
  if (immutable_db_options_.enable_thread_tracking) {
2736 2737
    ThreadStatusUtil::NewColumnFamilyInfo(this, cfd, cfd->GetName(),
                                          cfd->ioptions()->env);
2738
  }
Y
Yueh-Hsuan Chiang 已提交
2739 2740
}

2741
void DBImpl::EraseThreadStatusCfInfo(ColumnFamilyData* cfd) const {
2742
  if (immutable_db_options_.enable_thread_tracking) {
2743
    ThreadStatusUtil::EraseColumnFamilyInfo(cfd);
2744
  }
Y
Yueh-Hsuan Chiang 已提交
2745 2746 2747
}

void DBImpl::EraseThreadStatusDbInfo() const {
2748
  if (immutable_db_options_.enable_thread_tracking) {
2749
    ThreadStatusUtil::EraseDatabaseInfo(this);
2750
  }
Y
Yueh-Hsuan Chiang 已提交
2751 2752 2753
}

#else
2754
void DBImpl::NewThreadStatusCfInfo(ColumnFamilyData* /*cfd*/) const {}
Y
Yueh-Hsuan Chiang 已提交
2755

2756
void DBImpl::EraseThreadStatusCfInfo(ColumnFamilyData* /*cfd*/) const {}
Y
Yueh-Hsuan Chiang 已提交
2757

2758
void DBImpl::EraseThreadStatusDbInfo() const {}
Y
Yueh-Hsuan Chiang 已提交
2759 2760
#endif  // ROCKSDB_USING_THREAD_STATUS

2761 2762
//
// A global method that can dump out the build version
2763
void DumpRocksDBBuildVersion(Logger* log) {
I
Igor Canadi 已提交
2764
#if !defined(IOS_CROSS_COMPILE)
H
hyunwoo 已提交
2765
  // if we compile with Xcode, we don't run build_detect_version, so we don't
I
Igor Canadi 已提交
2766
  // generate util/build_version.cc
2767 2768 2769 2770
  ROCKS_LOG_HEADER(log, "RocksDB version: %d.%d.%d\n", ROCKSDB_MAJOR,
                   ROCKSDB_MINOR, ROCKSDB_PATCH);
  ROCKS_LOG_HEADER(log, "Git sha %s", rocksdb_build_git_sha);
  ROCKS_LOG_HEADER(log, "Compile date %s", rocksdb_build_compile_date);
I
Igor Canadi 已提交
2771
#endif
2772 2773
}

A
agiardullo 已提交
2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790
#ifndef ROCKSDB_LITE
SequenceNumber DBImpl::GetEarliestMemTableSequenceNumber(SuperVersion* sv,
                                                         bool include_history) {
  // Find the earliest sequence number that we know we can rely on reading
  // from the memtable without needing to check sst files.
  SequenceNumber earliest_seq =
      sv->imm->GetEarliestSequenceNumber(include_history);
  if (earliest_seq == kMaxSequenceNumber) {
    earliest_seq = sv->mem->GetEarliestSequenceNumber();
  }
  assert(sv->mem->GetEarliestSequenceNumber() >= earliest_seq);

  return earliest_seq;
}
#endif  // ROCKSDB_LITE

#ifndef ROCKSDB_LITE
2791 2792
Status DBImpl::GetLatestSequenceForKey(SuperVersion* sv, const Slice& key,
                                       bool cache_only, SequenceNumber* seq,
2793 2794
                                       bool* found_record_for_key,
                                       bool* is_blob_index) {
A
agiardullo 已提交
2795 2796
  Status s;
  MergeContext merge_context;
2797 2798
  RangeDelAggregator range_del_agg(sv->mem->GetInternalKeyComparator(),
                                   kMaxSequenceNumber);
A
agiardullo 已提交
2799

A
Andrew Kryczka 已提交
2800
  ReadOptions read_options;
A
agiardullo 已提交
2801 2802 2803 2804
  SequenceNumber current_seq = versions_->LastSequence();
  LookupKey lkey(key, current_seq);

  *seq = kMaxSequenceNumber;
2805 2806
  *found_record_for_key = false;

A
agiardullo 已提交
2807
  // Check if there is a record for this key in the latest memtable
M
Maysam Yabandeh 已提交
2808
  sv->mem->Get(lkey, nullptr, &s, &merge_context, &range_del_agg, seq,
2809
               read_options, nullptr /*read_callback*/, is_blob_index);
A
agiardullo 已提交
2810 2811 2812

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2813 2814 2815
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTable::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2816 2817 2818 2819 2820 2821

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check immutable memtables
2822
    *found_record_for_key = true;
A
agiardullo 已提交
2823 2824 2825 2826
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
M
Maysam Yabandeh 已提交
2827
  sv->imm->Get(lkey, nullptr, &s, &merge_context, &range_del_agg, seq,
2828
               read_options, nullptr /*read_callback*/, is_blob_index);
A
agiardullo 已提交
2829 2830 2831

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2832 2833 2834
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTableList::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2835 2836 2837 2838 2839 2840

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check memtable history
2841
    *found_record_for_key = true;
A
agiardullo 已提交
2842 2843 2844 2845
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
A
Andrew Kryczka 已提交
2846
  sv->imm->GetFromHistory(lkey, nullptr, &s, &merge_context, &range_del_agg,
2847
                          seq, read_options, is_blob_index);
A
agiardullo 已提交
2848 2849 2850

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2851 2852
    ROCKS_LOG_ERROR(
        immutable_db_options_.info_log,
A
agiardullo 已提交
2853 2854 2855 2856 2857 2858
        "Unexpected status returned from MemTableList::GetFromHistory: %s\n",
        s.ToString().c_str());

    return s;
  }

2859 2860 2861 2862 2863 2864 2865 2866 2867 2868
  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check SST files
    *found_record_for_key = true;
    return Status::OK();
  }

  // TODO(agiardullo): possible optimization: consider checking cached
  // SST files if cache_only=true?
  if (!cache_only) {
    // Check tables
R
Reid Horuff 已提交
2869
    sv->current->Get(read_options, lkey, nullptr, &s, &merge_context,
A
Andrew Kryczka 已提交
2870
                     &range_del_agg, nullptr /* value_found */,
2871 2872
                     found_record_for_key, seq, nullptr /*read_callback*/,
                     is_blob_index);
2873 2874 2875

    if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
      // unexpected error reading SST files
2876 2877 2878
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "Unexpected status returned from Version::Get: %s\n",
                      s.ToString().c_str());
2879 2880 2881 2882 2883

      return s;
    }
  }

A
agiardullo 已提交
2884 2885
  return Status::OK();
}
2886 2887 2888 2889 2890

Status DBImpl::IngestExternalFile(
    ColumnFamilyHandle* column_family,
    const std::vector<std::string>& external_files,
    const IngestExternalFileOptions& ingestion_options) {
2891 2892 2893 2894
  if (external_files.empty()) {
    return Status::InvalidArgument("external_files is empty");
  }

2895 2896 2897 2898
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

2899 2900 2901 2902 2903
  // Ingest should immediately fail if ingest_behind is requested,
  // but the DB doesn't support it.
  if (ingestion_options.ingest_behind) {
    if (!immutable_db_options_.allow_ingest_behind) {
      return Status::InvalidArgument(
2904
          "Can't ingest_behind file in DB with allow_ingest_behind=false");
2905 2906 2907
    }
  }

2908 2909 2910 2911
  ExternalSstFileIngestionJob ingestion_job(env_, versions_.get(), cfd,
                                            immutable_db_options_, env_options_,
                                            &snapshots_, ingestion_options);

2912 2913 2914
  SuperVersionContext dummy_sv_ctx(/* create_superversion */ true);
  VersionEdit dummy_edit;
  uint64_t next_file_number = 0;
2915 2916 2917
  std::list<uint64_t>::iterator pending_output_elem;
  {
    InstrumentedMutexLock l(&mutex_);
2918
    if (error_handler_.IsDBStopped()) {
2919
      // Don't ingest files when there is a bg_error
2920
      return error_handler_.GetBGError();
2921 2922 2923
    }

    // Make sure that bg cleanup wont delete the files that we are ingesting
2924
    pending_output_elem = CaptureCurrentFileNumberInPendingOutputs();
2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939

    // If crash happen after a hard link established, Recover function may
    // reuse the file number that has already assigned to the internal file,
    // and this will overwrite the external file. To protect the external
    // file, we have to make sure the file number will never being reused.
    next_file_number = versions_->FetchAddFileNumber(external_files.size());
    auto cf_options = cfd->GetLatestMutableCFOptions();
    status = versions_->LogAndApply(cfd, *cf_options, &dummy_edit, &mutex_,
                                    directories_.GetDbDir());
    if (status.ok()) {
      InstallSuperVersionAndScheduleWork(cfd, &dummy_sv_ctx, *cf_options);
    }
  }
  dummy_sv_ctx.Clean();
  if (!status.ok()) {
2940 2941
    InstrumentedMutexLock l(&mutex_);
    ReleaseFileNumberFromPendingOutputs(pending_output_elem);
2942
    return status;
2943 2944
  }

2945
  SuperVersion* super_version = cfd->GetReferencedSuperVersion(&mutex_);
2946 2947
  status =
      ingestion_job.Prepare(external_files, next_file_number, super_version);
2948
  CleanupSuperVersion(super_version);
2949
  if (!status.ok()) {
2950
    InstrumentedMutexLock l(&mutex_);
2951
    ReleaseFileNumberFromPendingOutputs(pending_output_elem);
2952 2953 2954
    return status;
  }

2955
  SuperVersionContext sv_context(/* create_superversion */ true);
2956
  TEST_SYNC_POINT("DBImpl::AddFile:Start");
2957 2958 2959 2960 2961
  {
    // Lock db mutex
    InstrumentedMutexLock l(&mutex_);
    TEST_SYNC_POINT("DBImpl::AddFile:MutexLock");

2962
    // Stop writes to the DB by entering both write threads
2963 2964
    WriteThread::Writer w;
    write_thread_.EnterUnbatched(&w, &mutex_);
2965
    WriteThread::Writer nonmem_w;
2966
    if (two_write_queues_) {
2967 2968
      nonmem_write_thread_.EnterUnbatched(&nonmem_w, &mutex_);
    }
2969

2970 2971
    num_running_ingest_file_++;

2972 2973 2974 2975 2976 2977
    // We cannot ingest a file into a dropped CF
    if (cfd->IsDropped()) {
      status = Status::InvalidArgument(
          "Cannot ingest an external file into a dropped CF");
    }

2978
    // Figure out if we need to flush the memtable first
2979 2980
    if (status.ok()) {
      bool need_flush = false;
2981
      status = ingestion_job.NeedsFlush(&need_flush, cfd->GetSuperVersion());
2982 2983
      TEST_SYNC_POINT_CALLBACK("DBImpl::IngestExternalFile:NeedFlush",
                               &need_flush);
2984 2985
      if (status.ok() && need_flush) {
        mutex_.Unlock();
2986 2987 2988
        status = FlushMemTable(cfd, FlushOptions(),
                               FlushReason::kExternalFileIngestion,
                               true /* writes_stopped */);
2989 2990
        mutex_.Lock();
      }
2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005
    }

    // Run the ingestion job
    if (status.ok()) {
      status = ingestion_job.Run();
    }

    // Install job edit [Mutex will be unlocked here]
    auto mutable_cf_options = cfd->GetLatestMutableCFOptions();
    if (status.ok()) {
      status =
          versions_->LogAndApply(cfd, *mutable_cf_options, ingestion_job.edit(),
                                 &mutex_, directories_.GetDbDir());
    }
    if (status.ok()) {
3006
      InstallSuperVersionAndScheduleWork(cfd, &sv_context, *mutable_cf_options,
Z
Zhongyi Xie 已提交
3007
                                         FlushReason::kExternalFileIngestion);
3008 3009 3010
    }

    // Resume writes to the DB
3011
    if (two_write_queues_) {
3012 3013
      nonmem_write_thread_.ExitUnbatched(&nonmem_w);
    }
3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032
    write_thread_.ExitUnbatched(&w);

    // Update stats
    if (status.ok()) {
      ingestion_job.UpdateStats();
    }

    ReleaseFileNumberFromPendingOutputs(pending_output_elem);

    num_running_ingest_file_--;
    if (num_running_ingest_file_ == 0) {
      bg_cv_.SignalAll();
    }

    TEST_SYNC_POINT("DBImpl::AddFile:MutexUnlock");
  }
  // mutex_ is unlocked here

  // Cleanup
3033
  sv_context.Clean();
3034 3035
  ingestion_job.Cleanup(status);

3036 3037 3038 3039
  if (status.ok()) {
    NotifyOnExternalFileIngested(cfd, ingestion_job);
  }

3040 3041 3042
  return status;
}

A
Aaron G 已提交
3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062
Status DBImpl::VerifyChecksum() {
  Status s;
  Options options;
  EnvOptions env_options;
  std::vector<ColumnFamilyData*> cfd_list;
  {
    InstrumentedMutexLock l(&mutex_);
    for (auto cfd : *versions_->GetColumnFamilySet()) {
      if (!cfd->IsDropped() && cfd->initialized()) {
        cfd->Ref();
        cfd_list.push_back(cfd);
      }
    }
  }
  std::vector<SuperVersion*> sv_list;
  for (auto cfd : cfd_list) {
    sv_list.push_back(cfd->GetReferencedSuperVersion(&mutex_));
  }
  for (auto& sv : sv_list) {
    VersionStorageInfo* vstorage = sv->current->storage_info();
3063
    ColumnFamilyData* cfd = sv->current->cfd();
A
Aaron G 已提交
3064 3065 3066 3067
    for (int i = 0; i < vstorage->num_non_empty_levels() && s.ok(); i++) {
      for (size_t j = 0; j < vstorage->LevelFilesBrief(i).num_files && s.ok();
           j++) {
        const auto& fd = vstorage->LevelFilesBrief(i).files[j].fd;
3068
        std::string fname = TableFileName(cfd->ioptions()->cf_paths,
A
Aaron G 已提交
3069
                                          fd.GetNumber(), fd.GetPathId());
3070
        s = rocksdb::VerifySstFileChecksum(options, env_options, fname);
A
Aaron G 已提交
3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085
      }
    }
    if (!s.ok()) {
      break;
    }
  }
  {
    InstrumentedMutexLock l(&mutex_);
    for (auto sv : sv_list) {
      if (sv && sv->Unref()) {
        sv->Cleanup();
        delete sv;
      }
    }
    for (auto cfd : cfd_list) {
3086
      cfd->Unref();
A
Aaron G 已提交
3087 3088 3089 3090 3091
    }
  }
  return s;
}

3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110
void DBImpl::NotifyOnExternalFileIngested(
    ColumnFamilyData* cfd, const ExternalSstFileIngestionJob& ingestion_job) {
  if (immutable_db_options_.listeners.empty()) {
    return;
  }

  for (const IngestedFileInfo& f : ingestion_job.files_to_ingest()) {
    ExternalFileIngestionInfo info;
    info.cf_name = cfd->GetName();
    info.external_file_path = f.external_file_path;
    info.internal_file_path = f.internal_file_path;
    info.global_seqno = f.assigned_seqno;
    info.table_properties = f.table_properties;
    for (auto listener : immutable_db_options_.listeners) {
      listener->OnExternalFileIngested(this, info);
    }
  }
}

3111 3112 3113 3114 3115 3116 3117
void DBImpl::WaitForIngestFile() {
  mutex_.AssertHeld();
  while (num_running_ingest_file_ > 0) {
    bg_cv_.Wait();
  }
}

3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131
Status DBImpl::StartTrace(const TraceOptions& /* options */,
                          std::unique_ptr<TraceWriter>&& trace_writer) {
  InstrumentedMutexLock lock(&trace_mutex_);
  tracer_.reset(new Tracer(env_, std::move(trace_writer)));
  return Status::OK();
}

Status DBImpl::EndTrace() {
  InstrumentedMutexLock lock(&trace_mutex_);
  Status s = tracer_->Close();
  tracer_.reset();
  return s;
}

A
agiardullo 已提交
3132
#endif  // ROCKSDB_LITE
3133
}  // namespace rocksdb