db_impl.cc 107.5 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
  while (!flush_queue_.empty()) {
351 352 353 354 355 356
    const FlushRequest& flush_req = PopFirstFromFlushQueue();
    for (const auto& iter : flush_req) {
      ColumnFamilyData* cfd = iter.first;
      if (cfd->Unref()) {
        delete cfd;
      }
357 358 359 360 361 362 363 364 365
    }
  }
  while (!compaction_queue_.empty()) {
    auto cfd = PopFirstFromCompactionQueue();
    if (cfd->Unref()) {
      delete cfd;
    }
  }

I
Igor Canadi 已提交
366 367 368 369 370
  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();
371 372
  }

I
Igor Canadi 已提交
373 374 375 376 377 378 379 380 381 382
  // 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_) {
383
    JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
384
    FindObsoleteFiles(&job_context, true);
385 386

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

396 397 398
  for (auto l : logs_to_free_) {
    delete l;
  }
399
  for (auto& log : logs_) {
400 401 402
    uint64_t log_number = log.writer->get_log_number();
    Status s = log.ClearWriter();
    if (!s.ok()) {
403 404 405 406 407
      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());
408 409 410 411 412
      // Retain the first error
      if (ret.ok()) {
        ret = s;
      }
    }
413
  }
414
  logs_.clear();
415

416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
  // 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 已提交
431 432 433 434
  for (auto& txn_entry : recovered_transactions_) {
    delete txn_entry.second;
  }

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

443
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Shutdown complete");
444
  LogFlush(immutable_db_options_.info_log);
445 446

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

455
Status DBImpl::CloseImpl() { return CloseHelper(); }
456 457 458 459 460 461 462

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

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

474
const Status DBImpl::CreateArchivalDirectory() {
475 476 477
  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);
478 479 480 481 482
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

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

491
void DBImpl::MaybeDumpStats() {
492 493 494 495 496
  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 已提交
497 498 499

  const uint64_t now_micros = env_->NowMicros();

500
  if (last_stats_dump_time_microsec_ + stats_dump_period_sec * 1000000 <=
501
      now_micros) {
H
Haobo Xu 已提交
502 503 504 505 506
    // 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;
507

508
#ifndef ROCKSDB_LITE
509 510 511 512 513 514 515
    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 已提交
516
    std::string stats;
517
    {
518
      InstrumentedMutexLock l(&mutex_);
519 520
      default_cf_internal_stats_->GetStringProperty(
          *db_property_info, DB::Properties::kDBStats, &stats);
521
      for (auto cfd : *versions_->GetColumnFamilySet()) {
522 523 524 525 526
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFStatsNoFileHistogram,
              &stats);
        }
527 528
      }
      for (auto cfd : *versions_->GetColumnFamilySet()) {
529 530 531 532
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFFileHistogram, &stats);
        }
533 534
      }
    }
535 536 537
    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 已提交
538 539 540 541
    if (immutable_db_options_.dump_malloc_stats) {
      stats.clear();
      DumpMallocStats(&stats);
      if (!stats.empty()) {
542 543 544
        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 已提交
545 546
      }
    }
547
#endif  // !ROCKSDB_LITE
548

549
    PrintStatistics();
550 551 552
  }
}

553 554 555 556 557 558 559 560 561 562
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();
  }
}

563 564 565 566 567 568 569 570 571 572
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 已提交
573 574 575 576 577
  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();
578
  }
S
Siying Dong 已提交
579
  return ret_dir;
580 581
}

582 583
Status DBImpl::SetOptions(
    ColumnFamilyHandle* column_family,
S
Siying Dong 已提交
584 585
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
586 587
  (void)column_family;
  (void)options_map;
S
Siying Dong 已提交
588 589 590 591 592 593 594 595
  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");
596 597
  }

S
Siying Dong 已提交
598 599 600 601
  MutableCFOptions new_options;
  Status s;
  Status persist_options_status;
  WriteThread::Writer w;
602
  SuperVersionContext sv_context(/* create_superversion */ true);
S
Siying Dong 已提交
603 604 605 606 607 608 609 610 611 612 613 614
  {
    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.
615
      InstallSuperVersionAndScheduleWork(cfd, &sv_context, new_options);
616

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

624 625 626
  ROCKS_LOG_INFO(
      immutable_db_options_.info_log,
      "SetOptions() on column family [%s], inputs:", cfd->GetName().c_str());
S
Siying Dong 已提交
627 628 629
  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());
630
  }
S
Siying Dong 已提交
631 632 633 634 635
  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 已提交
636
      s = persist_options_status;
637
    }
638
  } else {
S
Siying Dong 已提交
639 640
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "[%s] SetOptions() failed",
                   cfd->GetName().c_str());
641
  }
S
Siying Dong 已提交
642 643 644
  LogFlush(immutable_db_options_.info_log);
  return s;
#endif  // ROCKSDB_LITE
645 646
}

S
Siying Dong 已提交
647 648 649
Status DBImpl::SetDBOptions(
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
650
  (void)options_map;
S
Siying Dong 已提交
651 652 653 654 655 656
  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 已提交
657
  }
658

S
Siying Dong 已提交
659 660 661
  MutableDBOptions new_options;
  Status s;
  Status persist_options_status;
662
  bool wal_changed = false;
S
Siying Dong 已提交
663 664 665 666 667 668 669 670 671 672 673 674 675
  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();
      }
676

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

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

    minimum_level = i;
  }
  return minimum_level;
}

757
Status DBImpl::FlushWAL(bool sync) {
758
  if (manual_wal_flush_) {
759 760 761 762 763 764 765
    // 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());
766 767 768 769 770
      // 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;
771 772 773 774 775 776
    }
    if (!sync) {
      ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=false");
      return s;
    }
  }
777 778 779
  if (!sync) {
    return Status::OK();
  }
780 781 782 783 784
  // sync = true
  ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=true");
  return SyncWAL();
}

785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
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(
806 807 808 809
            "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());
810 811 812 813 814
      }
    }
    for (auto it = logs_.begin();
         it != logs_.end() && it->number <= current_log_number; ++it) {
      auto& log = *it;
S
Siying Dong 已提交
815 816 817 818
      assert(!log.getting_synced);
      log.getting_synced = true;
      logs_to_sync.push_back(log.writer);
    }
819

S
Siying Dong 已提交
820
    need_log_dir_sync = !log_dir_synced_;
821 822
  }

823
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:1");
S
Siying Dong 已提交
824 825 826 827 828 829 830 831 832 833
  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();
834
  }
835
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:2");
836

S
Siying Dong 已提交
837 838 839 840 841 842
  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");
843

S
Siying Dong 已提交
844
  return status;
845 846
}

847 848
void DBImpl::MarkLogsSynced(uint64_t up_to, bool synced_dir,
                            const Status& status) {
S
Siying Dong 已提交
849
  mutex_.AssertHeld();
850
  if (synced_dir && logfile_number_ == up_to && status.ok()) {
S
Siying Dong 已提交
851 852 853 854 855 856 857
    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());
858 859
      // To modify logs_ both mutex_ and log_write_mutex_ must be held
      InstrumentedMutexLock l(&log_write_mutex_);
S
Siying Dong 已提交
860 861 862 863 864 865 866 867 868
      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();
869 870
}

S
Siying Dong 已提交
871 872
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
  return versions_->LastSequence();
873 874
}

875 876
void DBImpl::SetLastPublishedSequence(SequenceNumber seq) {
  versions_->SetLastPublishedSequence(seq);
877 878
}

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

S
Siying Dong 已提交
888 889 890 891 892 893 894 895 896
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();
897
  }
898

S
Siying Dong 已提交
899 900 901 902 903 904 905
  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 已提交
906

S
Siying Dong 已提交
907 908 909
void DBImpl::SchedulePurge() {
  mutex_.AssertHeld();
  assert(opened_successfully_);
I
Islam AbdelRahman 已提交
910

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

916 917 918
void DBImpl::BackgroundCallPurge() {
  mutex_.Lock();

919 920 921 922 923 924 925
  // 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;
926
      auto dir_to_sync = purge_file->dir_to_sync;
927 928 929 930
      auto type = purge_file->type;
      auto number = purge_file->number;
      auto job_id = purge_file->job_id;
      purge_queue_.pop_front();
931

932
      mutex_.Unlock();
933
      DeleteObsoleteFileImpl(job_id, fname, dir_to_sync, type, number);
934 935 936 937 938 939 940 941 942
      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();
    }
943 944 945 946 947 948 949 950 951 952 953
  }
  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();
}

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

  DBImpl* db;
964
  InstrumentedMutex* mu;
965
  SuperVersion* super_version;
966
  bool background_purge;
967 968
};

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

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

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

    delete state->super_version;
I
Igor Canadi 已提交
986
    if (job_context.HaveSomethingToDelete()) {
987
      if (state->background_purge) {
988 989 990 991 992 993 994 995 996 997
        // 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);
      }
998
    }
I
Igor Canadi 已提交
999
    job_context.Clean();
I
Igor Canadi 已提交
1000
  }
T
Tomislav Novak 已提交
1001

1002 1003
  delete state;
}
H
Hans Wennborg 已提交
1004
}  // namespace
1005

A
Andrew Kryczka 已提交
1006 1007 1008 1009
InternalIterator* DBImpl::NewInternalIterator(
    const ReadOptions& read_options, ColumnFamilyData* cfd,
    SuperVersion* super_version, Arena* arena,
    RangeDelAggregator* range_del_agg) {
S
sdong 已提交
1010
  InternalIterator* internal_iter;
1011
  assert(arena != nullptr);
A
Andrew Kryczka 已提交
1012
  assert(range_del_agg != nullptr);
1013
  // Need to create internal iterator from the arena.
1014 1015 1016
  MergeIteratorBuilder merge_iter_builder(
      &cfd->internal_comparator(), arena,
      !read_options.total_order_seek &&
1017
          super_version->mutable_cf_options.prefix_extractor != nullptr);
1018 1019
  // Collect iterator for mutable mem
  merge_iter_builder.AddIterator(
L
Lei Jin 已提交
1020
      super_version->mem->NewIterator(read_options, arena));
1021
  std::unique_ptr<InternalIterator> range_del_iter;
A
Andrew Kryczka 已提交
1022 1023
  Status s;
  if (!read_options.ignore_range_deletions) {
1024 1025
    range_del_iter.reset(
        super_version->mem->NewRangeTombstoneIterator(read_options));
A
Andrew Kryczka 已提交
1026 1027
    s = range_del_agg->AddTombstones(std::move(range_del_iter));
  }
1028
  // Collect all needed child iterators for immutable memtables
A
Andrew Kryczka 已提交
1029 1030 1031 1032 1033 1034 1035
  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);
    }
  }
1036
  TEST_SYNC_POINT_CALLBACK("DBImpl::NewInternalIterator:StatusCallback", &s);
A
Andrew Kryczka 已提交
1037 1038
  if (s.ok()) {
    // Collect iterators for files in L0 - Ln
S
Sagar Vemuri 已提交
1039 1040 1041 1042
    if (read_options.read_tier != kMemtableTier) {
      super_version->current->AddIterators(read_options, env_options_,
                                           &merge_iter_builder, range_del_agg);
    }
A
Andrew Kryczka 已提交
1043 1044 1045 1046 1047 1048 1049
    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;
1050 1051
  } else {
    CleanupSuperVersion(super_version);
A
Andrew Kryczka 已提交
1052
  }
1053
  return NewErrorInternalIterator<Slice>(s, arena);
J
jorlow@chromium.org 已提交
1054 1055
}

1056 1057 1058 1059
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

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

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

1074 1075 1076
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

1077 1078 1079 1080 1081 1082 1083 1084 1085
  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);
    }
  }

1086 1087 1088 1089 1090 1091
  // Acquire SuperVersion
  SuperVersion* sv = GetAndRefSuperVersion(cfd);

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

1092
  SequenceNumber snapshot;
L
Lei Jin 已提交
1093
  if (read_options.snapshot != nullptr) {
1094 1095 1096 1097 1098 1099 1100 1101
    // 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.
1102 1103
    snapshot =
        reinterpret_cast<const SnapshotImpl*>(read_options.snapshot)->number_;
1104 1105 1106
    if (callback) {
      snapshot = std::max(snapshot, callback->MaxUnpreparedSequenceNumber());
    }
1107
  } else {
1108 1109 1110 1111 1112 1113
    // 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.
1114 1115 1116 1117
    // 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.
1118 1119 1120
    snapshot = last_seq_same_as_publish_seq_
                   ? versions_->LastSequence()
                   : versions_->LastPublishedSequence();
J
jorlow@chromium.org 已提交
1121
  }
1122 1123
  TEST_SYNC_POINT("DBImpl::GetImpl:3");
  TEST_SYNC_POINT("DBImpl::GetImpl:4");
1124

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

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

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

1166 1167
  {
    PERF_TIMER_GUARD(get_post_process_time);
1168

1169
    ReturnAndCleanupSuperVersion(cfd, sv);
1170

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

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

1187
  SequenceNumber snapshot;
1188

1189
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
1190
    ColumnFamilyData* cfd;
1191 1192 1193 1194 1195
    SuperVersion* super_version;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
1196 1197 1198 1199 1200 1201
    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});
1202 1203 1204
    }
  }

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

1220 1221
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
1222

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

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

  // 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.
1235
  // merge_operands will contain the sequence of merges in the latter case.
1236
  size_t num_found = 0;
1237
  for (size_t i = 0; i < num_keys; ++i) {
1238
    merge_context.Clear();
1239
    Status& s = stat_list[i];
1240 1241 1242
    std::string* value = &(*values)[i];

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

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

  // Post processing (decrement reference counts and record statistics)
1281
  PERF_TIMER_GUARD(get_post_process_time);
1282 1283
  autovector<SuperVersion*> superversions_to_delete;

I
Igor Canadi 已提交
1284
  // TODO(icanadi) do we need lock here or just around Cleanup()?
1285 1286 1287 1288 1289 1290
  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);
1291 1292
    }
  }
1293 1294 1295 1296 1297 1298 1299
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
1300
  }
1301

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

1310
  return stat_list;
1311 1312
}

L
Lei Jin 已提交
1313
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
Y
Yi Wu 已提交
1314
                                  const std::string& column_family,
1315
                                  ColumnFamilyHandle** handle) {
Y
Yi Wu 已提交
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 1381 1382 1383
  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 已提交
1384
  Status s;
1385
  Status persist_options_status;
I
Igor Canadi 已提交
1386
  *handle = nullptr;
1387 1388

  s = CheckCompressionSupported(cf_options);
1389
  if (s.ok() && immutable_db_options_.allow_concurrent_memtable_write) {
1390 1391
    s = CheckConcurrentWritesSupported(cf_options);
  }
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
  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;
      }
    }
  }
1403 1404 1405 1406
  if (!s.ok()) {
    return s;
  }

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

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

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

1452 1453
      cfd->set_initialized();

Y
Yueh-Hsuan Chiang 已提交
1454
      *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
1455 1456 1457
      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 已提交
1458
    } else {
1459 1460 1461
      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 已提交
1462
    }
1463
  }  // InstrumentedMutexLock l(&mutex_)
Y
Yueh-Hsuan Chiang 已提交
1464

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

1474
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
Y
Yi Wu 已提交
1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505
  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) {
1506 1507 1508
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
1509 1510
    return Status::InvalidArgument("Can't drop default column family");
  }
1511

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

I
Igor Canadi 已提交
1514 1515
  VersionEdit edit;
  edit.DropColumnFamily();
1516 1517
  edit.SetColumnFamily(cfd->GetID());

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

    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()) {
1543
        if (!c->IsDropped() && !c->mem()->IsSnapshotSupported()) {
S
sdong 已提交
1544 1545 1546 1547 1548 1549
          new_is_snapshot_supported = false;
          break;
        }
      }
      is_snapshot_supported_ = new_is_snapshot_supported;
    }
1550
    bg_cv_.SignalAll();
1551
  }
1552

1553
  if (s.ok()) {
Y
Yueh-Hsuan Chiang 已提交
1554 1555 1556 1557
    // 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 已提交
1558
    assert(cfd->IsDropped());
1559 1560
    ROCKS_LOG_INFO(immutable_db_options_.info_log,
                   "Dropped column family with id %u\n", cfd->GetID());
1561
  } else {
1562 1563 1564
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Dropping column family with id %u FAILED -- %s\n",
                    cfd->GetID(), s.ToString().c_str());
1565 1566
  }

1567
  return s;
1568 1569
}

L
Lei Jin 已提交
1570
bool DBImpl::KeyMayExist(const ReadOptions& read_options,
1571 1572
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
M
Maysam Yabandeh 已提交
1573
  assert(value != nullptr);
1574
  if (value_found != nullptr) {
K
Kai Liu 已提交
1575 1576
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
1577
  }
L
Lei Jin 已提交
1578
  ReadOptions roptions = read_options;
1579
  roptions.read_tier = kBlockCacheTier;  // read from block cache only
M
Maysam Yabandeh 已提交
1580 1581 1582
  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 已提交
1583

1584
  // If block_cache is enabled and the index block of the table didn't
K
Kai Liu 已提交
1585 1586 1587
  // 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();
1588 1589
}

1590
Iterator* DBImpl::NewIterator(const ReadOptions& read_options,
1591
                              ColumnFamilyHandle* column_family) {
S
Siying Dong 已提交
1592 1593 1594 1595
  if (read_options.managed) {
    return NewErrorIterator(
        Status::NotSupported("Managed iterator is not supported anymore."));
  }
1596
  Iterator* result = nullptr;
1597 1598 1599 1600
  if (read_options.read_tier == kPersistedTier) {
    return NewErrorIterator(Status::NotSupported(
        "ReadTier::kPersistedData is not yet supported in iterators."));
  }
1601 1602 1603 1604 1605
  // 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()) {
1606 1607 1608 1609
    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."));
  }
1610 1611
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
Y
Yi Wu 已提交
1612
  ReadCallback* read_callback = nullptr;  // No read callback provided.
S
Siying Dong 已提交
1613
if (read_options.tailing) {
I
Igor Canadi 已提交
1614 1615
#ifdef ROCKSDB_LITE
    // not supported in lite version
1616 1617
    result = nullptr;

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

Y
Yi Wu 已提交
1639 1640 1641
ArenaWrappedDBIter* DBImpl::NewIteratorImpl(const ReadOptions& read_options,
                                            ColumnFamilyData* cfd,
                                            SequenceNumber snapshot,
Y
Yi Wu 已提交
1642
                                            ReadCallback* read_callback,
1643 1644
                                            bool allow_blob,
                                            bool allow_refresh) {
Y
Yi Wu 已提交
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 1687 1688 1689
  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(
1690
      env_, read_options, *cfd->ioptions(), sv->mutable_cf_options, snapshot,
Y
Yi Wu 已提交
1691
      sv->mutable_cf_options.max_sequential_skip_in_iterations,
1692 1693
      sv->version_number, read_callback, this, cfd, allow_blob,
      ((read_options.snapshot != nullptr) ? false : allow_refresh));
Y
Yi Wu 已提交
1694 1695 1696 1697 1698 1699 1700 1701 1702

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

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

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

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

1759
SnapshotImpl* DBImpl::GetSnapshotImpl(bool is_write_conflict_boundary) {
S
Siying Dong 已提交
1760 1761 1762 1763 1764 1765 1766 1767 1768
  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 已提交
1769
  }
1770
  auto snapshot_seq = last_seq_same_as_publish_seq_
1771
                          ? versions_->LastSequence()
1772
                          : versions_->LastPublishedSequence();
1773
  return snapshots_.New(s, snapshot_seq, unix_time, is_write_conflict_boundary);
S
Siying Dong 已提交
1774
}
1775

S
Siying Dong 已提交
1776 1777
void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  const SnapshotImpl* casted_s = reinterpret_cast<const SnapshotImpl*>(s);
S
Stanislau Hlebik 已提交
1778
  {
S
Siying Dong 已提交
1779 1780
    InstrumentedMutexLock l(&mutex_);
    snapshots_.Delete(casted_s);
1781 1782
    uint64_t oldest_snapshot;
    if (snapshots_.empty()) {
1783
      oldest_snapshot = last_seq_same_as_publish_seq_
1784
                            ? versions_->LastSequence()
1785
                            : versions_->LastPublishedSequence();
1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798
    } 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();
      }
    }
1799
  }
S
Siying Dong 已提交
1800
  delete casted_s;
1801 1802
}

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

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

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}
1824 1825

Status DBImpl::GetPropertiesOfTablesInRange(ColumnFamilyHandle* column_family,
1826
                                            const Range* range, std::size_t n,
1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846
                                            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 已提交
1847
#endif  // ROCKSDB_LITE
1848

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

1851
Env* DBImpl::GetEnv() const { return env_; }
1852

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

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

1865
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
1866
                         const Slice& property, std::string* value) {
1867
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
1868
  value->clear();
1869
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
1870 1871 1872
  if (property_info == nullptr) {
    return false;
  } else if (property_info->handle_int) {
1873
    uint64_t int_value;
1874 1875
    bool ret_value =
        GetIntPropertyInternal(cfd, *property_info, false, &int_value);
1876
    if (ret_value) {
1877
      *value = ToString(int_value);
1878 1879
    }
    return ret_value;
1880
  } else if (property_info->handle_string) {
1881
    InstrumentedMutexLock l(&mutex_);
1882
    return cfd->internal_stats()->GetStringProperty(*property_info, property,
1883
                                                    value);
1884 1885 1886 1887 1888 1889 1890
  } 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;
1891
  }
1892 1893 1894 1895
  // 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 已提交
1896 1897
}

1898 1899
bool DBImpl::GetMapProperty(ColumnFamilyHandle* column_family,
                            const Slice& property,
1900
                            std::map<std::string, std::string>* value) {
1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915
  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;
}

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

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

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

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

    return ret;
  }
}

1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
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 已提交
1967 1968 1969 1970
#ifndef ROCKSDB_LITE
Status DBImpl::ResetStats() {
  InstrumentedMutexLock l(&mutex_);
  for (auto* cfd : *versions_->GetColumnFamilySet()) {
1971 1972 1973
    if (cfd->initialized()) {
      cfd->internal_stats()->Clear();
    }
S
Siying Dong 已提交
1974 1975 1976 1977 1978
  }
  return Status::OK();
}
#endif  // ROCKSDB_LITE

1979 1980
bool DBImpl::GetAggregatedIntProperty(const Slice& property,
                                      uint64_t* aggregated_value) {
1981 1982
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1983 1984 1985 1986 1987 1988 1989 1990 1991
    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()) {
1992 1993 1994
      if (!cfd->initialized()) {
        continue;
      }
1995
      if (GetIntPropertyInternal(cfd, *property_info, true, &value)) {
1996 1997 1998 1999 2000 2001 2002 2003 2004 2005
        sum += value;
      } else {
        return false;
      }
    }
  }
  *aggregated_value = sum;
  return true;
}

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

A
agiardullo 已提交
2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022
// 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);
}

2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
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);
}

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

A
agiardullo 已提交
2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066
// 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 已提交
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080
// 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();
}

2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102
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);
}

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

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

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

I
Igor Canadi 已提交
2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146
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 已提交
2147 2148 2149 2150
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {
D
Dmitri Smirnov 已提交
2151

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

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

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

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

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

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

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

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

2271
    auto* vstorage = input_version->storage_info();
2272 2273 2274 2275 2276
    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)) {
2277
          continue;
2278
        }
2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293
        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;
        }

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

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

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

2352 2353
void DBImpl::GetColumnFamilyMetaData(ColumnFamilyHandle* column_family,
                                     ColumnFamilyMetaData* cf_meta) {
2354 2355 2356 2357 2358 2359 2360
  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 已提交
2361
#endif  // ROCKSDB_LITE
2362

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

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

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

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

2411 2412 2413 2414 2415
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
2416
  char* buffer = reinterpret_cast<char*>(alloca(static_cast<size_t>(file_size)));
2417
  Slice id;
2418
  s = id_file_reader->Read(static_cast<size_t>(file_size), &id, buffer);
2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429
  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;
}

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

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

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

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

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

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

2464 2465 2466 2467 2468 2469 2470 2471 2472
DB::~DB() {}

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

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

2480
Snapshot::~Snapshot() {}
2481

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

D
Dmitri Smirnov 已提交
2488 2489 2490
  // 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 已提交
2491 2492
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2493

J
jorlow@chromium.org 已提交
2494
  FileLock* lock;
2495 2496
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2497 2498 2499
  if (result.ok()) {
    uint64_t number;
    FileType type;
2500
    InfoLogPrefix info_log_prefix(!soptions.db_log_dir.empty(), dbname);
D
Dmitri Smirnov 已提交
2501 2502 2503
    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 已提交
2504
        Status del;
D
Dmitri Smirnov 已提交
2505
        std::string path_to_delete = dbname + "/" + fname;
K
Kosie van der Merwe 已提交
2506
        if (type == kMetaDatabase) {
2507 2508
          del = DestroyDB(path_to_delete, options);
        } else if (type == kTableFile) {
2509
          del = DeleteSSTFile(&soptions, path_to_delete, dbname);
K
Kosie van der Merwe 已提交
2510
        } else {
2511
          del = env->DeleteFile(path_to_delete);
K
Kosie van der Merwe 已提交
2512
        }
J
jorlow@chromium.org 已提交
2513 2514 2515 2516 2517
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2518

2519 2520
    std::vector<std::string> paths;

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

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

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

D
Dmitri Smirnov 已提交
2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573
    // 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 已提交
2574 2575
        }
      }
D
Dmitri Smirnov 已提交
2576
      env->DeleteDir(archivedir);
I
Igor Canadi 已提交
2577 2578
    }

D
Dmitri Smirnov 已提交
2579 2580 2581 2582 2583 2584 2585 2586
    // 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;
          }
2587 2588
        }
      }
D
Dmitri Smirnov 已提交
2589
      env->DeleteDir(soptions.wal_dir);
2590
    }
2591

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

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

  std::vector<std::string> cf_names;
  std::vector<ColumnFamilyOptions> cf_opts;
2614 2615 2616 2617 2618

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

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

2630 2631 2632
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:1");
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:2");

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

  if (s.ok()) {
    s = RenameTempFileToOptionsFile(file_name);
  }
Y
Yi Wu 已提交
2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655
  // 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());
    }
  }
2656 2657 2658
#else
  (void)need_mutex_lock;
  (void)need_enter_write_thread;
2659
#endif  // !ROCKSDB_LITE
Y
Yi Wu 已提交
2660
  return Status::OK();
2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674
}

#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()) {
2675 2676
      ROCKS_LOG_WARN(info_log, "Unable to delete options file %s",
                     iter->second.c_str());
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 2705 2706
    }
  }
}
}  // 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,
2707
                           immutable_db_options_.info_log, GetEnv());
2708 2709 2710 2711 2712 2713 2714 2715 2716
  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 已提交
2717 2718

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

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

D
Daniel Black 已提交
2734
#ifdef ROCKSDB_USING_THREAD_STATUS
2735

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

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

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

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

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

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

2763 2764
//
// A global method that can dump out the build version
2765
void DumpRocksDBBuildVersion(Logger* log) {
I
Igor Canadi 已提交
2766
#if !defined(IOS_CROSS_COMPILE)
H
hyunwoo 已提交
2767
  // if we compile with Xcode, we don't run build_detect_version, so we don't
I
Igor Canadi 已提交
2768
  // generate util/build_version.cc
2769 2770 2771 2772
  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 已提交
2773
#endif
2774 2775
}

A
agiardullo 已提交
2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792
#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
2793 2794
Status DBImpl::GetLatestSequenceForKey(SuperVersion* sv, const Slice& key,
                                       bool cache_only, SequenceNumber* seq,
2795 2796
                                       bool* found_record_for_key,
                                       bool* is_blob_index) {
A
agiardullo 已提交
2797 2798
  Status s;
  MergeContext merge_context;
2799 2800
  RangeDelAggregator range_del_agg(sv->mem->GetInternalKeyComparator(),
                                   kMaxSequenceNumber);
A
agiardullo 已提交
2801

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

  *seq = kMaxSequenceNumber;
2807 2808
  *found_record_for_key = false;

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

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

    return s;
  }

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

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

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

    return s;
  }

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

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

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

    return s;
  }

2861 2862 2863 2864 2865 2866 2867 2868 2869 2870
  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 已提交
2871
    sv->current->Get(read_options, lkey, nullptr, &s, &merge_context,
A
Andrew Kryczka 已提交
2872
                     &range_del_agg, nullptr /* value_found */,
2873 2874
                     found_record_for_key, seq, nullptr /*read_callback*/,
                     is_blob_index);
2875 2876 2877

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

      return s;
    }
  }

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

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

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

2901 2902 2903 2904 2905
  // 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(
2906
          "Can't ingest_behind file in DB with allow_ingest_behind=false");
2907 2908 2909
    }
  }

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

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

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

    // 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()) {
2942 2943
    InstrumentedMutexLock l(&mutex_);
    ReleaseFileNumberFromPendingOutputs(pending_output_elem);
2944
    return status;
2945 2946
  }

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

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

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

2972 2973
    num_running_ingest_file_++;

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

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

    // 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()) {
3008
      InstallSuperVersionAndScheduleWork(cfd, &sv_context, *mutable_cf_options,
Z
Zhongyi Xie 已提交
3009
                                         FlushReason::kExternalFileIngestion);
3010 3011 3012
    }

    // Resume writes to the DB
3013
    if (two_write_queues_) {
3014 3015
      nonmem_write_thread_.ExitUnbatched(&nonmem_w);
    }
3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034
    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
3035
  sv_context.Clean();
3036 3037
  ingestion_job.Cleanup(status);

3038 3039 3040 3041
  if (status.ok()) {
    NotifyOnExternalFileIngested(cfd, ingestion_job);
  }

3042 3043 3044
  return status;
}

A
Aaron G 已提交
3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062
Status DBImpl::VerifyChecksum() {
  Status s;
  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();
3064 3065 3066 3067 3068 3069
    Options opts;
    {
      InstrumentedMutexLock l(&mutex_);
      opts = Options(BuildDBOptions(immutable_db_options_,
         mutable_db_options_), cfd->GetLatestCFOptions());
    }
A
Aaron G 已提交
3070 3071 3072 3073
    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;
3074
        std::string fname = TableFileName(cfd->ioptions()->cf_paths,
A
Aaron G 已提交
3075
                                          fd.GetNumber(), fd.GetPathId());
3076
        s = rocksdb::VerifySstFileChecksum(opts, env_options_, fname);
A
Aaron G 已提交
3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091
      }
    }
    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) {
3092
      cfd->Unref();
A
Aaron G 已提交
3093 3094 3095 3096 3097
    }
  }
  return s;
}

3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116
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);
    }
  }
}

3117 3118 3119 3120 3121 3122 3123
void DBImpl::WaitForIngestFile() {
  mutex_.AssertHeld();
  while (num_running_ingest_file_ > 0) {
    bg_cv_.Wait();
  }
}

3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137
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;
}

3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148
Status DBImpl::TraceIteratorSeek(const uint32_t& cf_id, const Slice& key) {
  Status s;
  if (tracer_) {
    InstrumentedMutexLock lock(&trace_mutex_);
    if (tracer_) {
      s = tracer_->IteratorSeek(cf_id, key);
    }
  }
  return s;
}

3149 3150
Status DBImpl::TraceIteratorSeekForPrev(const uint32_t& cf_id,
                                        const Slice& key) {
3151 3152 3153 3154 3155 3156 3157 3158 3159 3160
  Status s;
  if (tracer_) {
    InstrumentedMutexLock lock(&trace_mutex_);
    if (tracer_) {
      s = tracer_->IteratorSeekForPrev(cf_id, key);
    }
  }
  return s;
}

A
agiardullo 已提交
3161
#endif  // ROCKSDB_LITE
3162

3163
}  // namespace rocksdb