db_impl.cc 91.9 KB
Newer Older
1
//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
2 3 4
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
5 6
//  This source code is also licensed under the GPLv2 license found in the
//  COPYING file in the root directory of this source tree.
7
//
J
jorlow@chromium.org 已提交
8 9 10 11 12
// 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 已提交
13
#ifndef __STDC_FORMAT_MACROS
I
Igor Canadi 已提交
14
#define __STDC_FORMAT_MACROS
L
liuhuahang 已提交
15
#endif
I
Igor Canadi 已提交
16
#include <inttypes.h>
17
#include <stdint.h>
D
David Bernard 已提交
18
#ifdef OS_SOLARIS
D
David Bernard 已提交
19
#include <alloca.h>
D
David Bernard 已提交
20
#endif
21 22 23
#ifdef ROCKSDB_JEMALLOC
#include "jemalloc/jemalloc.h"
#endif
24

J
jorlow@chromium.org 已提交
25
#include <algorithm>
26 27
#include <climits>
#include <cstdio>
28
#include <map>
J
jorlow@chromium.org 已提交
29
#include <set>
30
#include <stdexcept>
31
#include <string>
32
#include <unordered_map>
33
#include <unordered_set>
T
Tomislav Novak 已提交
34
#include <utility>
35
#include <vector>
36

J
jorlow@chromium.org 已提交
37
#include "db/builder.h"
I
Igor Canadi 已提交
38
#include "db/compaction_job.h"
39
#include "db/db_info_dumper.h"
40
#include "db/db_iter.h"
K
kailiu 已提交
41
#include "db/dbformat.h"
42
#include "db/event_helpers.h"
43
#include "db/external_sst_file_ingestion_job.h"
44 45
#include "db/flush_job.h"
#include "db/forward_iterator.h"
I
Igor Canadi 已提交
46
#include "db/job_context.h"
J
jorlow@chromium.org 已提交
47 48
#include "db/log_reader.h"
#include "db/log_writer.h"
V
Venkatesh Radhakrishnan 已提交
49
#include "db/managed_iterator.h"
J
jorlow@chromium.org 已提交
50
#include "db/memtable.h"
K
kailiu 已提交
51
#include "db/memtable_list.h"
52
#include "db/merge_context.h"
53
#include "db/merge_helper.h"
A
Andrew Kryczka 已提交
54
#include "db/range_del_aggregator.h"
J
jorlow@chromium.org 已提交
55
#include "db/table_cache.h"
K
kailiu 已提交
56
#include "db/table_properties_collector.h"
57
#include "db/transaction_log_impl.h"
J
jorlow@chromium.org 已提交
58 59
#include "db/version_set.h"
#include "db/write_batch_internal.h"
A
agiardullo 已提交
60
#include "db/write_callback.h"
61 62
#include "memtable/hash_linklist_rep.h"
#include "memtable/hash_skiplist_rep.h"
63 64 65 66 67 68 69
#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"
70
#include "port/likely.h"
71
#include "port/port.h"
I
Igor Canadi 已提交
72
#include "rocksdb/cache.h"
73 74 75 76 77 78
#include "rocksdb/compaction_filter.h"
#include "rocksdb/db.h"
#include "rocksdb/env.h"
#include "rocksdb/merge_operator.h"
#include "rocksdb/statistics.h"
#include "rocksdb/status.h"
S
Siying Dong 已提交
79
#include "rocksdb/table.h"
80
#include "rocksdb/version.h"
81
#include "rocksdb/write_buffer_manager.h"
J
jorlow@chromium.org 已提交
82
#include "table/block.h"
83
#include "table/block_based_table_factory.h"
84
#include "table/merging_iterator.h"
K
kailiu 已提交
85
#include "table/table_builder.h"
J
jorlow@chromium.org 已提交
86
#include "table/two_level_iterator.h"
87
#include "util/auto_roll_logger.h"
K
kailiu 已提交
88
#include "util/autovector.h"
89
#include "util/build_version.h"
J
jorlow@chromium.org 已提交
90
#include "util/coding.h"
I
Igor Canadi 已提交
91
#include "util/compression.h"
92
#include "util/crc32c.h"
93
#include "util/file_reader_writer.h"
94
#include "util/file_util.h"
95
#include "util/filename.h"
H
Haobo Xu 已提交
96
#include "util/log_buffer.h"
97
#include "util/logging.h"
J
jorlow@chromium.org 已提交
98
#include "util/mutexlock.h"
99
#include "util/sst_file_manager_impl.h"
100
#include "util/stop_watch.h"
101
#include "util/string_util.h"
102
#include "util/sync_point.h"
J
jorlow@chromium.org 已提交
103

104
namespace rocksdb {
105
const std::string kDefaultColumnFamilyName("default");
I
Igor Canadi 已提交
106
void DumpRocksDBBuildVersion(Logger * log);
107

A
Aaron Gao 已提交
108 109 110
CompressionType GetCompressionFlush(
    const ImmutableCFOptions& ioptions,
    const MutableCFOptions& mutable_cf_options) {
111 112 113
  // 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.
114
  if (ioptions.compaction_style == kCompactionStyleUniversal) {
115 116 117 118 119 120 121 122
    if (ioptions.compaction_options_universal.compression_size_percent < 0) {
      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];
123
  } else {
A
Aaron Gao 已提交
124
    return mutable_cf_options.compression;
125 126
  }
}
I
Igor Canadi 已提交
127

S
Siying Dong 已提交
128
namespace {
129
void DumpSupportInfo(Logger* logger) {
130 131 132 133 134 135 136 137
  ROCKS_LOG_HEADER(logger, "Compression algorithms supported:");
  ROCKS_LOG_HEADER(logger, "\tSnappy supported: %d", Snappy_Supported());
  ROCKS_LOG_HEADER(logger, "\tZlib supported: %d", Zlib_Supported());
  ROCKS_LOG_HEADER(logger, "\tBzip supported: %d", BZip2_Supported());
  ROCKS_LOG_HEADER(logger, "\tLZ4 supported: %d", LZ4_Supported());
  ROCKS_LOG_HEADER(logger, "\tZSTD supported: %d", ZSTD_Supported());
  ROCKS_LOG_HEADER(logger, "Fast CRC32 supported: %d",
                   crc32c::IsFastCrc32Supported());
I
Igor Canadi 已提交
138
}
S
Siying Dong 已提交
139
} // namespace
140

I
Igor Canadi 已提交
141
DBImpl::DBImpl(const DBOptions& options, const std::string& dbname)
J
jorlow@chromium.org 已提交
142
    : env_(options.env),
H
heyongqiang 已提交
143
      dbname_(dbname),
144 145 146
      initial_db_options_(SanitizeOptions(dbname, options)),
      immutable_db_options_(initial_db_options_),
      mutable_db_options_(initial_db_options_),
147
      stats_(immutable_db_options_.statistics.get()),
148
      db_lock_(nullptr),
149 150
      mutex_(stats_, env_, DB_MUTEX_WAIT_MICROS,
             immutable_db_options_.use_adaptive_mutex),
I
Igor Canadi 已提交
151
      shutting_down_(false),
J
jorlow@chromium.org 已提交
152
      bg_cv_(&mutex_),
153
      logfile_number_(0),
154
      log_dir_synced_(false),
I
Igor Canadi 已提交
155
      log_empty_(true),
156
      default_cf_handle_(nullptr),
157
      log_sync_cv_(&mutex_),
I
Igor Canadi 已提交
158 159
      total_log_size_(0),
      max_total_in_memory_state_(0),
S
sdong 已提交
160
      is_snapshot_supported_(true),
161
      write_buffer_manager_(immutable_db_options_.write_buffer_manager.get()),
162
      write_thread_(immutable_db_options_),
163
      write_controller_(mutable_db_options_.delayed_write_rate),
S
sdong 已提交
164
      last_batch_group_size_(0),
165 166
      unscheduled_flushes_(0),
      unscheduled_compactions_(0),
167
      bg_compaction_scheduled_(0),
168
      num_running_compactions_(0),
169
      bg_flush_scheduled_(0),
170
      num_running_flushes_(0),
171
      bg_purge_scheduled_(0),
172
      disable_delete_obsolete_files_(0),
173
      delete_obsolete_files_last_run_(env_->NowMicros()),
174
      last_stats_dump_time_microsec_(0),
175
      next_job_id_(1),
176
      has_unpersisted_data_(false),
177
      unable_to_flush_oldest_log_(false),
178
      env_options_(BuildDBOptions(immutable_db_options_, mutable_db_options_)),
179
      num_running_ingest_file_(0),
I
Igor Canadi 已提交
180
#ifndef ROCKSDB_LITE
181
      wal_manager_(immutable_db_options_, env_options_),
I
Igor Canadi 已提交
182
#endif  // ROCKSDB_LITE
183
      event_logger_(immutable_db_options_.info_log.get()),
184
      bg_work_paused_(0),
185
      bg_compaction_paused_(0),
186
      refitting_level_(false),
187
      opened_successfully_(false) {
H
heyongqiang 已提交
188
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
189

J
jorlow@chromium.org 已提交
190
  // Reserve ten files or so for other uses and give the rest to TableCache.
191
  // Give a large number for setting of "infinite" open files.
L
Leonidas Galanis 已提交
192
  const int table_cache_size = (mutable_db_options_.max_open_files == -1)
193
                                   ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
194
                                   : mutable_db_options_.max_open_files - 10;
195 196
  table_cache_ = NewLRUCache(table_cache_size,
                             immutable_db_options_.table_cache_numshardbits);
197

198
  versions_.reset(new VersionSet(dbname_, &immutable_db_options_, env_options_,
199
                                 table_cache_.get(), write_buffer_manager_,
200
                                 &write_controller_));
201 202
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
203

204 205 206 207 208
  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());
J
jorlow@chromium.org 已提交
209 210
}

211
// Will lock the mutex_,  will wait for completion if wait is true
212
void DBImpl::CancelAllBackgroundWork(bool wait) {
213
  InstrumentedMutexLock l(&mutex_);
214

215 216
  ROCKS_LOG_INFO(immutable_db_options_.info_log,
                 "Shutdown: canceling all background work");
217

218
  if (!shutting_down_.load(std::memory_order_acquire) &&
219
      has_unpersisted_data_.load(std::memory_order_relaxed) &&
Y
Yi Wu 已提交
220
      !mutable_db_options_.avoid_flush_during_shutdown) {
221
    for (auto cfd : *versions_->GetColumnFamilySet()) {
222
      if (!cfd->IsDropped() && !cfd->mem()->IsEmpty()) {
223 224 225 226
        cfd->Ref();
        mutex_.Unlock();
        FlushMemTable(cfd, FlushOptions());
        mutex_.Lock();
227
        cfd->Unref();
228 229
      }
    }
230
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
231
  }
232 233 234 235 236 237 238 239 240 241 242 243 244

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

DBImpl::~DBImpl() {
245 246
  // CancelAllBackgroundWork called with false means we just set the shutdown
  // marker. After this we do a variant of the waiting and unschedule work
247 248
  // (to consider: moving all the waiting into CancelAllBackgroundWork(true))
  CancelAllBackgroundWork(false);
249 250 251 252 253 254 255
  int compactions_unscheduled = env_->UnSchedule(this, Env::Priority::LOW);
  int flushes_unscheduled = env_->UnSchedule(this, Env::Priority::HIGH);
  mutex_.Lock();
  bg_compaction_scheduled_ -= compactions_unscheduled;
  bg_flush_scheduled_ -= flushes_unscheduled;

  // Wait for background work to finish
256 257 258
  while (bg_compaction_scheduled_ || bg_flush_scheduled_ ||
         bg_purge_scheduled_) {
    TEST_SYNC_POINT("DBImpl::~DBImpl:WaitJob");
259 260
    bg_cv_.Wait();
  }
261
  EraseThreadStatusDbInfo();
I
Igor Canadi 已提交
262 263
  flush_scheduler_.Clear();

264 265 266 267 268 269 270 271 272 273 274 275 276
  while (!flush_queue_.empty()) {
    auto cfd = PopFirstFromFlushQueue();
    if (cfd->Unref()) {
      delete cfd;
    }
  }
  while (!compaction_queue_.empty()) {
    auto cfd = PopFirstFromCompactionQueue();
    if (cfd->Unref()) {
      delete cfd;
    }
  }

I
Igor Canadi 已提交
277 278 279 280 281
  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();
282 283
  }

I
Igor Canadi 已提交
284 285 286 287 288 289 290 291 292 293
  // 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_) {
294
    JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
295
    FindObsoleteFiles(&job_context, true);
296 297

    mutex_.Unlock();
I
Igor Canadi 已提交
298
    // manifest number starting from 2
I
Igor Canadi 已提交
299 300 301
    job_context.manifest_file_number = 1;
    if (job_context.HaveSomethingToDelete()) {
      PurgeObsoleteFiles(job_context);
302
    }
I
Igor Canadi 已提交
303
    job_context.Clean();
304
    mutex_.Lock();
305 306
  }

307 308 309
  for (auto l : logs_to_free_) {
    delete l;
  }
310 311 312
  for (auto& log : logs_) {
    log.ClearWriter();
  }
313
  logs_.clear();
314

315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
  // 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 已提交
330 331 332 333
  for (auto& txn_entry : recovered_transactions_) {
    delete txn_entry.second;
  }

334
  // versions need to be destroyed before table_cache since it can hold
335 336
  // references to table_cache.
  versions_.reset();
337
  mutex_.Unlock();
I
Igor Canadi 已提交
338 339 340
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
341

342
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Shutdown complete");
343
  LogFlush(immutable_db_options_.info_log);
J
jorlow@chromium.org 已提交
344 345 346
}

void DBImpl::MaybeIgnoreError(Status* s) const {
347
  if (s->ok() || immutable_db_options_.paranoid_checks) {
J
jorlow@chromium.org 已提交
348 349
    // No change needed
  } else {
350 351
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "Ignoring error %s",
                   s->ToString().c_str());
J
jorlow@chromium.org 已提交
352 353 354 355
    *s = Status::OK();
  }
}

356
const Status DBImpl::CreateArchivalDirectory() {
357 358 359
  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);
360 361 362 363 364
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

365
void DBImpl::PrintStatistics() {
366
  auto dbstats = immutable_db_options_.statistics.get();
367
  if (dbstats) {
368 369
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "STATISTICS:\n %s",
                   dbstats->ToString().c_str());
370 371 372
  }
}

373
#ifndef ROCKSDB_LITE
374 375 376 377 378 379 380 381 382 383 384 385 386 387
#ifdef ROCKSDB_JEMALLOC
typedef struct {
  char* cur;
  char* end;
} MallocStatus;

static void GetJemallocStatus(void* mstat_arg, const char* status) {
  MallocStatus* mstat = reinterpret_cast<MallocStatus*>(mstat_arg);
  size_t status_len = status ? strlen(status) : 0;
  size_t buf_size = (size_t)(mstat->end - mstat->cur);
  if (!status_len || status_len > buf_size) {
    return;
  }

388
  snprintf(mstat->cur, buf_size, "%s", status);
389 390 391 392 393 394 395
  mstat->cur += status_len;
}
#endif  // ROCKSDB_JEMALLOC

static void DumpMallocStats(std::string* stats) {
#ifdef ROCKSDB_JEMALLOC
  MallocStatus mstat;
396
  const unsigned int kMallocStatusLen = 1000000;
397
  std::unique_ptr<char[]> buf{new char[kMallocStatusLen + 1]};
398 399
  mstat.cur = buf.get();
  mstat.end = buf.get() + kMallocStatusLen;
T
Tamir Duberstein 已提交
400
  je_malloc_stats_print(GetJemallocStatus, &mstat, "");
401 402 403
  stats->append(buf.get());
#endif  // ROCKSDB_JEMALLOC
}
404
#endif  // !ROCKSDB_LITE
405

406
void DBImpl::MaybeDumpStats() {
407 408 409 410 411
  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 已提交
412 413 414

  const uint64_t now_micros = env_->NowMicros();

415
  if (last_stats_dump_time_microsec_ + stats_dump_period_sec * 1000000 <=
416
      now_micros) {
H
Haobo Xu 已提交
417 418 419 420 421
    // 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;
422

423
#ifndef ROCKSDB_LITE
424 425 426 427 428 429 430
    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 已提交
431
    std::string stats;
432
    {
433
      InstrumentedMutexLock l(&mutex_);
434 435
      default_cf_internal_stats_->GetStringProperty(
          *db_property_info, DB::Properties::kDBStats, &stats);
436
      for (auto cfd : *versions_->GetColumnFamilySet()) {
437
        cfd->internal_stats()->GetStringProperty(
438 439 440 441 442
            *cf_property_info, DB::Properties::kCFStatsNoFileHistogram, &stats);
      }
      for (auto cfd : *versions_->GetColumnFamilySet()) {
        cfd->internal_stats()->GetStringProperty(
            *cf_property_info, DB::Properties::kCFFileHistogram, &stats);
443 444
      }
    }
445 446 447
    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 已提交
448 449 450 451
    if (immutable_db_options_.dump_malloc_stats) {
      stats.clear();
      DumpMallocStats(&stats);
      if (!stats.empty()) {
452 453 454
        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 已提交
455 456
      }
    }
457
#endif  // !ROCKSDB_LITE
458

459
    PrintStatistics();
460 461 462
  }
}

463 464 465 466 467 468 469 470 471 472
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();
  }
}

S
Siying Dong 已提交
473 474 475 476 477 478
Directory* DBImpl::Directories::GetDataDir(size_t path_id) {
  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();
479
  }
S
Siying Dong 已提交
480
  return ret_dir;
481 482
}

S
Siying Dong 已提交
483 484 485 486 487 488 489 490 491 492 493
Status DBImpl::SetOptions(ColumnFamilyHandle* column_family,
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
  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");
494 495
  }

S
Siying Dong 已提交
496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
  MutableCFOptions new_options;
  Status s;
  Status persist_options_status;
  WriteThread::Writer w;
  {
    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.
      auto* old_sv =
          InstallSuperVersionAndScheduleWork(cfd, nullptr, new_options);
      delete old_sv;
515

Y
Yi Wu 已提交
516 517
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, true /*need_enter_write_thread*/);
518 519 520
    }
  }

S
Siying Dong 已提交
521 522 523 524 525 526
  ROCKS_LOG_INFO(immutable_db_options_.info_log,
                 "SetOptions() on column family [%s], inputs:",
                 cfd->GetName().c_str());
  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());
527
  }
S
Siying Dong 已提交
528 529 530 531 532
  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 已提交
533
      s = persist_options_status;
534
    }
535
  } else {
S
Siying Dong 已提交
536 537
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "[%s] SetOptions() failed",
                   cfd->GetName().c_str());
538
  }
S
Siying Dong 已提交
539 540 541
  LogFlush(immutable_db_options_.info_log);
  return s;
#endif  // ROCKSDB_LITE
542 543
}

S
Siying Dong 已提交
544 545 546 547 548 549 550 551 552
Status DBImpl::SetDBOptions(
    const std::unordered_map<std::string, std::string>& options_map) {
#ifdef ROCKSDB_LITE
  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 已提交
553
  }
554

S
Siying Dong 已提交
555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
  MutableDBOptions new_options;
  Status s;
  Status persist_options_status;
  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();
      }
571

572
      write_controller_.set_max_delayed_write_rate(new_options.delayed_write_rate);
L
Leonidas Galanis 已提交
573
      table_cache_.get()->SetCapacity(new_options.max_open_files == -1
574
                                          ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
575
                                          : new_options.max_open_files - 10);
576

577 578
      mutable_db_options_ = new_options;

579
      write_thread_.EnterUnbatched(&w, &mutex_);
580
      if (total_log_size_ > GetMaxTotalWalSize()) {
581 582 583 584 585 586
        Status purge_wal_status = HandleWALFull(&write_context);
        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());
        }
587
      }
Y
Yi Wu 已提交
588 589
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, false /*need_enter_write_thread*/);
590
      write_thread_.ExitUnbatched(&w);
591 592
    }
  }
593
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions(), inputs:");
594
  for (const auto& o : options_map) {
595 596
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "%s: %s\n", o.first.c_str(),
                   o.second.c_str());
597 598
  }
  if (s.ok()) {
599
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions() succeeded");
600 601 602 603 604 605 606
    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());
      }
607 608 609
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "Unable to persist options in SetDBOptions() -- %s",
                     persist_options_status.ToString().c_str());
610 611
    }
  } else {
612
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "SetDBOptions failed");
L
Lei Jin 已提交
613
  }
614
  LogFlush(immutable_db_options_.info_log);
615
  return s;
I
Igor Canadi 已提交
616
#endif  // ROCKSDB_LITE
617 618
}

619
// return the same level if it cannot be moved
620 621
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd,
    const MutableCFOptions& mutable_cf_options, int level) {
622
  mutex_.AssertHeld();
S
sdong 已提交
623
  const auto* vstorage = cfd->current()->storage_info();
624
  int minimum_level = level;
625
  for (int i = level - 1; i > 0; --i) {
626
    // stop if level i is not empty
S
sdong 已提交
627
    if (vstorage->NumLevelFiles(i) > 0) break;
628
    // stop if level i is too small (cannot fit the level files)
629
    if (vstorage->MaxBytesForLevel(i) < vstorage->NumLevelBytes(level)) {
630 631
      break;
    }
632 633 634 635 636 637

    minimum_level = i;
  }
  return minimum_level;
}

638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658
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(
659 660 661 662
            "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());
663 664 665 666 667
      }
    }
    for (auto it = logs_.begin();
         it != logs_.end() && it->number <= current_log_number; ++it) {
      auto& log = *it;
S
Siying Dong 已提交
668 669 670 671
      assert(!log.getting_synced);
      log.getting_synced = true;
      logs_to_sync.push_back(log.writer);
    }
672

S
Siying Dong 已提交
673
    need_log_dir_sync = !log_dir_synced_;
674 675
  }

S
Siying Dong 已提交
676 677 678 679 680 681 682 683 684 685
  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();
686 687
  }

S
Siying Dong 已提交
688 689 690 691 692 693
  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");
694

S
Siying Dong 已提交
695
  return status;
696 697
}

S
Siying Dong 已提交
698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
void DBImpl::MarkLogsSynced(
    uint64_t up_to, bool synced_dir, const Status& status) {
  mutex_.AssertHeld();
  if (synced_dir &&
      logfile_number_ == up_to &&
      status.ok()) {
    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());
      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();
720 721
}

S
Siying Dong 已提交
722 723
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
  return versions_->LastSequence();
724 725
}

S
Siying Dong 已提交
726 727 728 729 730 731 732 733 734
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();
735
  }
736

S
Siying Dong 已提交
737 738 739 740 741 742 743
  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 已提交
744

S
Siying Dong 已提交
745 746 747
void DBImpl::SchedulePurge() {
  mutex_.AssertHeld();
  assert(opened_successfully_);
I
Islam AbdelRahman 已提交
748

S
Siying Dong 已提交
749 750 751
  // Purge operations are put into High priority queue
  bg_purge_scheduled_++;
  env_->Schedule(&DBImpl::BGWorkPurge, this, Env::Priority::HIGH, nullptr);
I
Islam AbdelRahman 已提交
752 753
}

754 755 756
void DBImpl::BackgroundCallPurge() {
  mutex_.Lock();

757 758 759 760 761 762 763 764 765 766 767 768
  // 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;
      auto type = purge_file->type;
      auto number = purge_file->number;
      auto path_id = purge_file->path_id;
      auto job_id = purge_file->job_id;
      purge_queue_.pop_front();
769

770 771 772 773 774 775 776 777 778 779 780 781 782
      mutex_.Unlock();
      Status file_deletion_status;
      DeleteObsoleteFileImpl(file_deletion_status, job_id, fname, type, number,
                             path_id);
      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();
    }
783 784 785 786 787 788 789 790 791 792 793
  }
  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();
}

794 795
namespace {
struct IterState {
796
  IterState(DBImpl* _db, InstrumentedMutex* _mu, SuperVersion* _super_version,
797
            bool _background_purge)
798 799 800
      : db(_db),
        mu(_mu),
        super_version(_super_version),
801
        background_purge(_background_purge) {}
802 803

  DBImpl* db;
804
  InstrumentedMutex* mu;
805
  SuperVersion* super_version;
806
  bool background_purge;
807 808 809 810
};

static void CleanupIteratorState(void* arg1, void* arg2) {
  IterState* state = reinterpret_cast<IterState*>(arg1);
811

812
  if (state->super_version->Unref()) {
813 814 815
    // Job id == 0 means that this is not our background process, but rather
    // user thread
    JobContext job_context(0);
816

817 818
    state->mu->Lock();
    state->super_version->Cleanup();
I
Igor Canadi 已提交
819
    state->db->FindObsoleteFiles(&job_context, false, true);
820 821 822
    if (state->background_purge) {
      state->db->ScheduleBgLogWriterClose(&job_context);
    }
823 824 825
    state->mu->Unlock();

    delete state->super_version;
I
Igor Canadi 已提交
826
    if (job_context.HaveSomethingToDelete()) {
827
      if (state->background_purge) {
828 829 830 831 832 833 834 835 836 837
        // 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);
      }
838
    }
I
Igor Canadi 已提交
839
    job_context.Clean();
I
Igor Canadi 已提交
840
  }
T
Tomislav Novak 已提交
841

842 843
  delete state;
}
H
Hans Wennborg 已提交
844
}  // namespace
845

A
Andrew Kryczka 已提交
846 847 848 849
InternalIterator* DBImpl::NewInternalIterator(
    const ReadOptions& read_options, ColumnFamilyData* cfd,
    SuperVersion* super_version, Arena* arena,
    RangeDelAggregator* range_del_agg) {
S
sdong 已提交
850
  InternalIterator* internal_iter;
851
  assert(arena != nullptr);
A
Andrew Kryczka 已提交
852
  assert(range_del_agg != nullptr);
853
  // Need to create internal iterator from the arena.
854 855 856 857
  MergeIteratorBuilder merge_iter_builder(
      &cfd->internal_comparator(), arena,
      !read_options.total_order_seek &&
          cfd->ioptions()->prefix_extractor != nullptr);
858 859
  // Collect iterator for mutable mem
  merge_iter_builder.AddIterator(
L
Lei Jin 已提交
860
      super_version->mem->NewIterator(read_options, arena));
861
  std::unique_ptr<InternalIterator> range_del_iter;
A
Andrew Kryczka 已提交
862 863
  Status s;
  if (!read_options.ignore_range_deletions) {
864 865
    range_del_iter.reset(
        super_version->mem->NewRangeTombstoneIterator(read_options));
A
Andrew Kryczka 已提交
866 867
    s = range_del_agg->AddTombstones(std::move(range_del_iter));
  }
868
  // Collect all needed child iterators for immutable memtables
A
Andrew Kryczka 已提交
869 870 871 872 873 874 875 876 877
  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);
    }
  }
  if (s.ok()) {
    // Collect iterators for files in L0 - Ln
S
Sagar Vemuri 已提交
878 879 880 881
    if (read_options.read_tier != kMemtableTier) {
      super_version->current->AddIterators(read_options, env_options_,
                                           &merge_iter_builder, range_del_agg);
    }
A
Andrew Kryczka 已提交
882 883 884 885 886 887 888 889 890
    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;
  }
  return NewErrorInternalIterator(s);
J
jorlow@chromium.org 已提交
891 892
}

893 894 895 896
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

L
Lei Jin 已提交
897
Status DBImpl::Get(const ReadOptions& read_options,
S
Siying Dong 已提交
898 899 900
                   ColumnFamilyHandle* column_family, const Slice& key,
                   PinnableSlice* value) {
  return GetImpl(read_options, column_family, key, value);
L
Lei Jin 已提交
901 902 903
}

Status DBImpl::GetImpl(const ReadOptions& read_options,
904
                       ColumnFamilyHandle* column_family, const Slice& key,
M
Maysam Yabandeh 已提交
905 906
                       PinnableSlice* pinnable_val, bool* value_found) {
  assert(pinnable_val != nullptr);
L
Lei Jin 已提交
907
  StopWatch sw(env_, stats_, DB_GET);
908
  PERF_TIMER_GUARD(get_snapshot_time);
909

910 911 912
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

913 914 915 916 917 918
  // Acquire SuperVersion
  SuperVersion* sv = GetAndRefSuperVersion(cfd);

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

919
  SequenceNumber snapshot;
L
Lei Jin 已提交
920 921 922
  if (read_options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(
        read_options.snapshot)->number_;
923
  } else {
924 925 926 927 928 929 930 931 932 933
    // 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.
    // We shouldn't get snapshot before finding and referencing the
    // super versipon 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.
934
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
935
  }
936 937
  TEST_SYNC_POINT("DBImpl::GetImpl:3");
  TEST_SYNC_POINT("DBImpl::GetImpl:4");
938

939
  // Prepare to store a list of merge operations if merge occurs.
940
  MergeContext merge_context;
941
  RangeDelAggregator range_del_agg(cfd->internal_comparator(), snapshot);
942

943
  Status s;
944
  // First look in the memtable, then in the immutable memtable (if any).
945
  // s is both in/out. When in, s could either be OK or MergeInProgress.
946
  // merge_operands will contain the sequence of merges in the latter case.
947
  LookupKey lkey(key, snapshot);
L
Lei Jin 已提交
948
  PERF_TIMER_STOP(get_snapshot_time);
949

950 951
  bool skip_memtable = (read_options.read_tier == kPersistedTier &&
                        has_unpersisted_data_.load(std::memory_order_relaxed));
952 953
  bool done = false;
  if (!skip_memtable) {
M
Maysam Yabandeh 已提交
954 955
    if (sv->mem->Get(lkey, pinnable_val->GetSelf(), &s, &merge_context,
                     &range_del_agg, read_options)) {
956
      done = true;
M
Maysam Yabandeh 已提交
957
      pinnable_val->PinSelf();
958
      RecordTick(stats_, MEMTABLE_HIT);
A
Andrew Kryczka 已提交
959
    } else if ((s.ok() || s.IsMergeInProgress()) &&
M
Maysam Yabandeh 已提交
960 961
               sv->imm->Get(lkey, pinnable_val->GetSelf(), &s, &merge_context,
                            &range_del_agg, read_options)) {
962
      done = true;
M
Maysam Yabandeh 已提交
963
      pinnable_val->PinSelf();
964 965
      RecordTick(stats_, MEMTABLE_HIT);
    }
A
Andrew Kryczka 已提交
966 967 968
    if (!done && !s.ok() && !s.IsMergeInProgress()) {
      return s;
    }
969 970
  }
  if (!done) {
971
    PERF_TIMER_GUARD(get_from_output_files_time);
M
Maysam Yabandeh 已提交
972
    sv->current->Get(read_options, lkey, pinnable_val, &s, &merge_context,
A
Andrew Kryczka 已提交
973
                     &range_del_agg, value_found);
L
Lei Jin 已提交
974
    RecordTick(stats_, MEMTABLE_MISS);
975
  }
976

977 978
  {
    PERF_TIMER_GUARD(get_post_process_time);
979

980
    ReturnAndCleanupSuperVersion(cfd, sv);
981

982
    RecordTick(stats_, NUMBER_KEYS_READ);
M
Maysam Yabandeh 已提交
983 984 985
    size_t size = pinnable_val->size();
    RecordTick(stats_, BYTES_READ, size);
    MeasureTime(stats_, BYTES_PER_READ, size);
986
  }
987
  return s;
J
jorlow@chromium.org 已提交
988 989
}

990
std::vector<Status> DBImpl::MultiGet(
L
Lei Jin 已提交
991
    const ReadOptions& read_options,
992
    const std::vector<ColumnFamilyHandle*>& column_family,
993
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
994

L
Lei Jin 已提交
995
  StopWatch sw(env_, stats_, DB_MULTIGET);
996
  PERF_TIMER_GUARD(get_snapshot_time);
K
Kai Liu 已提交
997

998
  SequenceNumber snapshot;
999

1000
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
1001
    ColumnFamilyData* cfd;
1002 1003 1004 1005 1006
    SuperVersion* super_version;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
1007 1008 1009 1010 1011 1012
    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});
1013 1014 1015
    }
  }

1016
  mutex_.Lock();
L
Lei Jin 已提交
1017 1018 1019
  if (read_options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(
        read_options.snapshot)->number_;
1020 1021 1022
  } else {
    snapshot = versions_->LastSequence();
  }
1023
  for (auto mgd_iter : multiget_cf_data) {
1024 1025
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
1026
  }
1027
  mutex_.Unlock();
1028

1029 1030
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
1031

1032
  // Note: this always resizes the values array
1033 1034 1035
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
1036 1037

  // Keep track of bytes that we read for statistics-recording later
1038
  uint64_t bytes_read = 0;
L
Lei Jin 已提交
1039
  PERF_TIMER_STOP(get_snapshot_time);
1040 1041 1042 1043

  // 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.
1044
  // merge_operands will contain the sequence of merges in the latter case.
1045
  for (size_t i = 0; i < num_keys; ++i) {
1046
    merge_context.Clear();
1047
    Status& s = stat_list[i];
1048 1049 1050
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
1051
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
A
Andrew Kryczka 已提交
1052
    RangeDelAggregator range_del_agg(cfh->cfd()->internal_comparator(),
1053
                                     snapshot);
1054
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
1055 1056 1057
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
1058
    bool skip_memtable =
1059 1060
        (read_options.read_tier == kPersistedTier &&
         has_unpersisted_data_.load(std::memory_order_relaxed));
1061 1062
    bool done = false;
    if (!skip_memtable) {
M
Maysam Yabandeh 已提交
1063
      if (super_version->mem->Get(lkey, value, &s, &merge_context,
A
Andrew Kryczka 已提交
1064
                                  &range_del_agg, read_options)) {
1065 1066
        done = true;
        // TODO(?): RecordTick(stats_, MEMTABLE_HIT)?
M
Maysam Yabandeh 已提交
1067
      } else if (super_version->imm->Get(lkey, value, &s, &merge_context,
A
Andrew Kryczka 已提交
1068
                                         &range_del_agg, read_options)) {
1069 1070 1071 1072 1073
        done = true;
        // TODO(?): RecordTick(stats_, MEMTABLE_HIT)?
      }
    }
    if (!done) {
M
Maysam Yabandeh 已提交
1074
      PinnableSlice pinnable_val;
1075
      PERF_TIMER_GUARD(get_from_output_files_time);
M
Maysam Yabandeh 已提交
1076 1077 1078
      super_version->current->Get(read_options, lkey, &pinnable_val, &s,
                                  &merge_context, &range_del_agg);
      value->assign(pinnable_val.data(), pinnable_val.size());
1079
      // TODO(?): RecordTick(stats_, MEMTABLE_MISS)?
1080 1081 1082
    }

    if (s.ok()) {
M
Maysam Yabandeh 已提交
1083
      bytes_read += value->size();
1084 1085 1086 1087
    }
  }

  // Post processing (decrement reference counts and record statistics)
1088
  PERF_TIMER_GUARD(get_post_process_time);
1089 1090
  autovector<SuperVersion*> superversions_to_delete;

I
Igor Canadi 已提交
1091
  // TODO(icanadi) do we need lock here or just around Cleanup()?
1092 1093 1094 1095 1096 1097
  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);
1098 1099
    }
  }
1100 1101 1102 1103 1104 1105 1106
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
1107
  }
1108

L
Lei Jin 已提交
1109 1110 1111
  RecordTick(stats_, NUMBER_MULTIGET_CALLS);
  RecordTick(stats_, NUMBER_MULTIGET_KEYS_READ, num_keys);
  RecordTick(stats_, NUMBER_MULTIGET_BYTES_READ, bytes_read);
1112
  MeasureTime(stats_, BYTES_PER_MULTIGET, bytes_read);
L
Lei Jin 已提交
1113
  PERF_TIMER_STOP(get_post_process_time);
1114

1115
  return stat_list;
1116 1117
}

L
Lei Jin 已提交
1118
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
Y
Yi Wu 已提交
1119
                                  const std::string& column_family,
1120
                                  ColumnFamilyHandle** handle) {
Y
Yi Wu 已提交
1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188
  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 已提交
1189
  Status s;
1190
  Status persist_options_status;
I
Igor Canadi 已提交
1191
  *handle = nullptr;
1192 1193

  s = CheckCompressionSupported(cf_options);
1194
  if (s.ok() && immutable_db_options_.allow_concurrent_memtable_write) {
1195 1196
    s = CheckConcurrentWritesSupported(cf_options);
  }
1197 1198 1199 1200
  if (!s.ok()) {
    return s;
  }

Y
Yueh-Hsuan Chiang 已提交
1201
  {
1202
    InstrumentedMutexLock l(&mutex_);
I
Igor Canadi 已提交
1203

Y
Yueh-Hsuan Chiang 已提交
1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216
    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 已提交
1217
    {  // write thread
1218 1219
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
I
Igor Canadi 已提交
1220 1221
      // LogAndApply will both write the creation in MANIFEST and create
      // ColumnFamilyData object
1222 1223 1224
      s = versions_->LogAndApply(nullptr, MutableCFOptions(cf_options), &edit,
                                 &mutex_, directories_.GetDbDir(), false,
                                 &cf_options);
1225
      write_thread_.ExitUnbatched(&w);
I
Igor Canadi 已提交
1226
    }
Y
Yueh-Hsuan Chiang 已提交
1227 1228 1229 1230 1231
    if (s.ok()) {
      single_column_family_mode_ = false;
      auto* cfd =
          versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
      assert(cfd != nullptr);
I
Igor Canadi 已提交
1232
      delete InstallSuperVersionAndScheduleWork(
Y
Yueh-Hsuan Chiang 已提交
1233
          cfd, nullptr, *cfd->GetLatestMutableCFOptions());
S
sdong 已提交
1234 1235 1236 1237 1238

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

Y
Yueh-Hsuan Chiang 已提交
1239
      *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
1240 1241 1242
      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 已提交
1243
    } else {
1244 1245 1246
      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 已提交
1247
    }
1248
  }  // InstrumentedMutexLock l(&mutex_)
Y
Yueh-Hsuan Chiang 已提交
1249 1250

  // this is outside the mutex
1251
  if (s.ok()) {
1252 1253
    NewThreadStatusCfInfo(
        reinterpret_cast<ColumnFamilyHandleImpl*>(*handle)->cfd());
1254
  }
1255
  return s;
1256 1257
}

1258
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
Y
Yi Wu 已提交
1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289
  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) {
1290 1291 1292
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
1293 1294
    return Status::InvalidArgument("Can't drop default column family");
  }
1295

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

I
Igor Canadi 已提交
1298 1299
  VersionEdit edit;
  edit.DropColumnFamily();
1300 1301
  edit.SetColumnFamily(cfd->GetID());

1302
  Status s;
1303
  {
1304
    InstrumentedMutexLock l(&mutex_);
1305 1306 1307 1308
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
1309
      // we drop column family from a single write thread
1310 1311
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
1312 1313
      s = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
                                 &edit, &mutex_);
1314
      write_thread_.ExitUnbatched(&w);
1315
    }
S
sdong 已提交
1316 1317 1318 1319 1320 1321

    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()) {
1322
        if (!c->IsDropped() && !c->mem()->IsSnapshotSupported()) {
S
sdong 已提交
1323 1324 1325 1326 1327 1328
          new_is_snapshot_supported = false;
          break;
        }
      }
      is_snapshot_supported_ = new_is_snapshot_supported;
    }
1329
  }
1330

1331
  if (s.ok()) {
Y
Yueh-Hsuan Chiang 已提交
1332 1333 1334 1335
    // 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 已提交
1336
    assert(cfd->IsDropped());
L
Lei Jin 已提交
1337 1338 1339
    auto* mutable_cf_options = cfd->GetLatestMutableCFOptions();
    max_total_in_memory_state_ -= mutable_cf_options->write_buffer_size *
                                  mutable_cf_options->max_write_buffer_number;
1340 1341
    ROCKS_LOG_INFO(immutable_db_options_.info_log,
                   "Dropped column family with id %u\n", cfd->GetID());
1342
  } else {
1343 1344 1345
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Dropping column family with id %u FAILED -- %s\n",
                    cfd->GetID(), s.ToString().c_str());
1346 1347
  }

1348
  return s;
1349 1350
}

L
Lei Jin 已提交
1351
bool DBImpl::KeyMayExist(const ReadOptions& read_options,
1352 1353
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
M
Maysam Yabandeh 已提交
1354
  assert(value != nullptr);
1355
  if (value_found != nullptr) {
K
Kai Liu 已提交
1356 1357
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
1358
  }
L
Lei Jin 已提交
1359
  ReadOptions roptions = read_options;
1360
  roptions.read_tier = kBlockCacheTier; // read from block cache only
M
Maysam Yabandeh 已提交
1361 1362 1363
  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 已提交
1364

1365
  // If block_cache is enabled and the index block of the table didn't
K
Kai Liu 已提交
1366 1367 1368
  // 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();
1369 1370
}

1371
Iterator* DBImpl::NewIterator(const ReadOptions& read_options,
1372
                              ColumnFamilyHandle* column_family) {
1373 1374 1375 1376
  if (read_options.read_tier == kPersistedTier) {
    return NewErrorIterator(Status::NotSupported(
        "ReadTier::kPersistedData is not yet supported in iterators."));
  }
1377 1378
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
V
Venkatesh Radhakrishnan 已提交
1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393
  if (read_options.managed) {
#ifdef ROCKSDB_LITE
    // not supported in lite version
    return NewErrorIterator(Status::InvalidArgument(
        "Managed Iterators not supported in RocksDBLite."));
#else
    if ((read_options.tailing) || (read_options.snapshot != nullptr) ||
        (is_snapshot_supported_)) {
      return new ManagedIterator(this, read_options, cfd);
    }
    // Managed iter not supported
    return NewErrorIterator(Status::InvalidArgument(
        "Managed Iterators not supported without snapshots."));
#endif
  } else if (read_options.tailing) {
I
Igor Canadi 已提交
1394 1395 1396 1397
#ifdef ROCKSDB_LITE
    // not supported in lite version
    return nullptr;
#else
1398 1399
    SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
    auto iter = new ForwardIterator(this, read_options, cfd, sv);
1400
    return NewDBIterator(
1401
        env_, read_options, *cfd->ioptions(), cfd->user_comparator(), iter,
1402 1403
        kMaxSequenceNumber,
        sv->mutable_cf_options.max_sequential_skip_in_iterations,
1404
        sv->version_number);
I
Igor Canadi 已提交
1405
#endif
T
Tomislav Novak 已提交
1406
  } else {
1407
    SequenceNumber latest_snapshot = versions_->LastSequence();
1408
    SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
1409

I
Igor Canadi 已提交
1410
    auto snapshot =
1411 1412 1413
        read_options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(
                read_options.snapshot)->number_
I
Igor Canadi 已提交
1414
            : latest_snapshot;
T
Tomislav Novak 已提交
1415

1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
    // 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         | <----+
    // |       |   +-------------------+
    // |       |                       |
    // +-------+-----------------------+
    //
1452 1453
    // ArenaWrappedDBIter inlines an arena area where all the iterators in
    // the iterator tree are allocated in the order of being accessed when
1454 1455 1456 1457 1458
    // 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(
1459
        env_, read_options, *cfd->ioptions(), cfd->user_comparator(), snapshot,
S
Siying Dong 已提交
1460
        sv->mutable_cf_options.max_sequential_skip_in_iterations,
1461
        sv->version_number);
1462

S
Siying Dong 已提交
1463 1464 1465 1466
    InternalIterator* internal_iter =
        NewInternalIterator(read_options, cfd, sv, db_iter->GetArena(),
                            db_iter->GetRangeDelAggregator());
    db_iter->SetIterUnderDBIter(internal_iter);
1467

S
Siying Dong 已提交
1468
    return db_iter;
1469
  }
S
Siying Dong 已提交
1470 1471
  // To stop compiler from complaining
  return nullptr;
1472 1473
}

S
Siying Dong 已提交
1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492
Status DBImpl::NewIterators(
    const ReadOptions& read_options,
    const std::vector<ColumnFamilyHandle*>& column_families,
    std::vector<Iterator*>* iterators) {
  if (read_options.read_tier == kPersistedTier) {
    return Status::NotSupported(
        "ReadTier::kPersistedData is not yet supported in iterators.");
  }
  iterators->clear();
  iterators->reserve(column_families.size());
  if (read_options.managed) {
#ifdef ROCKSDB_LITE
    return Status::InvalidArgument(
        "Managed interator not supported in RocksDB lite");
#else
    if ((!read_options.tailing) && (read_options.snapshot == nullptr) &&
        (!is_snapshot_supported_)) {
      return Status::InvalidArgument(
          "Managed interator not supported without snapshots");
1493
    }
S
Siying Dong 已提交
1494 1495 1496 1497
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      auto iter = new ManagedIterator(this, read_options, cfd);
      iterators->push_back(iter);
1498
    }
S
Siying Dong 已提交
1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
#endif
  } else if (read_options.tailing) {
#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(
1510
          env_, read_options, *cfd->ioptions(), cfd->user_comparator(), iter,
S
Siying Dong 已提交
1511 1512
          kMaxSequenceNumber,
          sv->mutable_cf_options.max_sequential_skip_in_iterations,
1513
          sv->version_number));
1514
    }
S
Siying Dong 已提交
1515 1516 1517
#endif
  } else {
    SequenceNumber latest_snapshot = versions_->LastSequence();
1518

S
Siying Dong 已提交
1519 1520 1521 1522
    for (size_t i = 0; i < column_families.size(); ++i) {
      auto* cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(
          column_families[i])->cfd();
      SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
1523

S
Siying Dong 已提交
1524 1525 1526 1527 1528
      auto snapshot =
          read_options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(
                  read_options.snapshot)->number_
              : latest_snapshot;
1529

S
Siying Dong 已提交
1530
      ArenaWrappedDBIter* db_iter = NewArenaWrappedDbIterator(
1531 1532 1533
          env_, read_options, *cfd->ioptions(), cfd->user_comparator(),
          snapshot, sv->mutable_cf_options.max_sequential_skip_in_iterations,
          sv->version_number);
S
Siying Dong 已提交
1534 1535 1536 1537 1538
      InternalIterator* internal_iter =
          NewInternalIterator(read_options, cfd, sv, db_iter->GetArena(),
                              db_iter->GetRangeDelAggregator());
      db_iter->SetIterUnderDBIter(internal_iter);
      iterators->push_back(db_iter);
1539 1540
    }
  }
1541

I
Igor Canadi 已提交
1542
  return Status::OK();
S
Stanislau Hlebik 已提交
1543 1544
}

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

S
Siying Dong 已提交
1547 1548 1549
#ifndef ROCKSDB_LITE
const Snapshot* DBImpl::GetSnapshotForWriteConflictBoundary() {
  return GetSnapshotImpl(true);
1550 1551 1552
}
#endif  // ROCKSDB_LITE

S
Siying Dong 已提交
1553 1554 1555 1556 1557 1558 1559 1560 1561 1562
const Snapshot* DBImpl::GetSnapshotImpl(bool is_write_conflict_boundary) {
  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 已提交
1563
  }
S
Siying Dong 已提交
1564 1565 1566
  return snapshots_.New(s, versions_->LastSequence(), unix_time,
                        is_write_conflict_boundary);
}
1567

S
Siying Dong 已提交
1568 1569
void DBImpl::ReleaseSnapshot(const Snapshot* s) {
  const SnapshotImpl* casted_s = reinterpret_cast<const SnapshotImpl*>(s);
S
Stanislau Hlebik 已提交
1570
  {
S
Siying Dong 已提交
1571 1572
    InstrumentedMutexLock l(&mutex_);
    snapshots_.Delete(casted_s);
1573
  }
S
Siying Dong 已提交
1574
  delete casted_s;
1575 1576
}

A
Anirban Rahut 已提交
1577 1578 1579 1580 1581 1582 1583 1584
bool DBImpl::HasActiveSnapshotLaterThanSN(SequenceNumber sn) {
  InstrumentedMutexLock l(&mutex_);
  if (snapshots_.empty()) {
    return false;
  }
  return (snapshots_.newest()->GetSequenceNumber() > sn);
}

I
Igor Canadi 已提交
1585
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
1586 1587 1588 1589 1590
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

1591 1592
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
1593
  auto version = cfd->current();
1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}
1606 1607

Status DBImpl::GetPropertiesOfTablesInRange(ColumnFamilyHandle* column_family,
1608
                                            const Range* range, std::size_t n,
1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628
                                            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 已提交
1629
#endif  // ROCKSDB_LITE
1630

I
Igor Canadi 已提交
1631 1632 1633 1634
const std::string& DBImpl::GetName() const {
  return dbname_;
}

1635 1636 1637 1638
Env* DBImpl::GetEnv() const {
  return env_;
}

1639 1640
Options DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  InstrumentedMutexLock l(&mutex_);
1641
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
1642 1643
  return Options(BuildDBOptions(immutable_db_options_, mutable_db_options_),
                 cfh->cfd()->GetLatestCFOptions());
I
Igor Canadi 已提交
1644 1645
}

1646 1647 1648 1649
DBOptions DBImpl::GetDBOptions() const {
  InstrumentedMutexLock l(&mutex_);
  return BuildDBOptions(immutable_db_options_, mutable_db_options_);
}
1650

1651
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
1652
                         const Slice& property, std::string* value) {
1653
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
1654
  value->clear();
1655
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
1656 1657 1658
  if (property_info == nullptr) {
    return false;
  } else if (property_info->handle_int) {
1659
    uint64_t int_value;
1660 1661
    bool ret_value =
        GetIntPropertyInternal(cfd, *property_info, false, &int_value);
1662
    if (ret_value) {
1663
      *value = ToString(int_value);
1664 1665
    }
    return ret_value;
1666
  } else if (property_info->handle_string) {
1667
    InstrumentedMutexLock l(&mutex_);
1668
    return cfd->internal_stats()->GetStringProperty(*property_info, property,
1669 1670
                                                    value);
  }
1671 1672 1673 1674
  // 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 已提交
1675 1676
}

1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
bool DBImpl::GetMapProperty(ColumnFamilyHandle* column_family,
                            const Slice& property,
                            std::map<std::string, double>* value) {
  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;
}

1695 1696
bool DBImpl::GetIntProperty(ColumnFamilyHandle* column_family,
                            const Slice& property, uint64_t* value) {
1697 1698
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1699 1700
    return false;
  }
1701
  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
1702
  return GetIntPropertyInternal(cfd, *property_info, false, value);
1703 1704
}

1705
bool DBImpl::GetIntPropertyInternal(ColumnFamilyData* cfd,
1706 1707 1708 1709
                                    const DBPropertyInfo& property_info,
                                    bool is_locked, uint64_t* value) {
  assert(property_info.handle_int != nullptr);
  if (!property_info.need_out_of_mutex) {
1710 1711
    if (is_locked) {
      mutex_.AssertHeld();
1712
      return cfd->internal_stats()->GetIntProperty(property_info, value, this);
1713 1714
    } else {
      InstrumentedMutexLock l(&mutex_);
1715
      return cfd->internal_stats()->GetIntProperty(property_info, value, this);
1716
    }
1717
  } else {
1718 1719 1720 1721 1722 1723
    SuperVersion* sv = nullptr;
    if (!is_locked) {
      sv = GetAndRefSuperVersion(cfd);
    } else {
      sv = cfd->GetSuperVersion();
    }
1724 1725

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

1728 1729 1730
    if (!is_locked) {
      ReturnAndCleanupSuperVersion(cfd, sv);
    }
1731 1732 1733 1734 1735

    return ret;
  }
}

S
Siying Dong 已提交
1736 1737 1738 1739 1740 1741 1742 1743 1744 1745
#ifndef ROCKSDB_LITE
Status DBImpl::ResetStats() {
  InstrumentedMutexLock l(&mutex_);
  for (auto* cfd : *versions_->GetColumnFamilySet()) {
    cfd->internal_stats()->Clear();
  }
  return Status::OK();
}
#endif  // ROCKSDB_LITE

1746 1747
bool DBImpl::GetAggregatedIntProperty(const Slice& property,
                                      uint64_t* aggregated_value) {
1748 1749
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1750 1751 1752 1753 1754 1755 1756 1757 1758
    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()) {
1759
      if (GetIntPropertyInternal(cfd, *property_info, true, &value)) {
1760 1761 1762 1763 1764 1765 1766 1767 1768 1769
        sum += value;
      } else {
        return false;
      }
    }
  }
  *aggregated_value = sum;
  return true;
}

1770 1771
SuperVersion* DBImpl::GetAndRefSuperVersion(ColumnFamilyData* cfd) {
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
I
Igor Canadi 已提交
1772
  return cfd->GetThreadLocalSuperVersion(&mutex_);
1773 1774
}

A
agiardullo 已提交
1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786
// 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);
}

1787 1788
void DBImpl::ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd,
                                          SuperVersion* sv) {
I
Igor Canadi 已提交
1789
  bool unref_sv = !cfd->ReturnThreadLocalSuperVersion(sv);
1790 1791 1792 1793 1794

  if (unref_sv) {
    // Release SuperVersion
    if (sv->Unref()) {
      {
1795
        InstrumentedMutexLock l(&mutex_);
1796 1797 1798 1799 1800 1801 1802
        sv->Cleanup();
      }
      delete sv;
      RecordTick(stats_, NUMBER_SUPERVERSION_CLEANUPS);
    }
    RecordTick(stats_, NUMBER_SUPERVERSION_RELEASES);
  }
1803 1804
}

A
agiardullo 已提交
1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828
// 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 已提交
1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842
// 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();
}

1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864
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);
}

1865
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
1866
                                 const Range* range, int n, uint64_t* sizes,
1867 1868 1869
                                 uint8_t include_flags) {
  assert(include_flags & DB::SizeApproximationFlags::INCLUDE_FILES ||
         include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES);
J
jorlow@chromium.org 已提交
1870
  Version* v;
1871 1872
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
1873 1874
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
  v = sv->current;
J
jorlow@chromium.org 已提交
1875 1876 1877

  for (int i = 0; i < n; i++) {
    // Convert user_key into a corresponding internal key.
1878 1879
    InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
V
Vitaliy Liptchinsky 已提交
1880
    sizes[i] = 0;
1881
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_FILES) {
V
Vitaliy Liptchinsky 已提交
1882
      sizes[i] += versions_->ApproximateSize(v, k1.Encode(), k2.Encode());
1883 1884
    }
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES) {
1885 1886
      sizes[i] += sv->mem->ApproximateStats(k1.Encode(), k2.Encode()).size;
      sizes[i] += sv->imm->ApproximateStats(k1.Encode(), k2.Encode()).size;
1887
    }
J
jorlow@chromium.org 已提交
1888 1889
  }

1890
  ReturnAndCleanupSuperVersion(cfd, sv);
J
jorlow@chromium.org 已提交
1891 1892
}

I
Igor Canadi 已提交
1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908
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 已提交
1909 1910 1911 1912 1913
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {

L
Lei Jin 已提交
1914
  RecordTick(stats_, GET_UPDATES_SINCE_CALLS);
I
Igor Canadi 已提交
1915 1916 1917
  if (seq > versions_->LastSequence()) {
    return Status::NotFound("Requested sequence not yet written in the db");
  }
I
Igor Canadi 已提交
1918
  return wal_manager_.GetUpdatesSince(seq, iter, read_options, versions_.get());
I
Igor Canadi 已提交
1919 1920
}

1921 1922 1923
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
1924 1925 1926
  WalFileType log_type;
  if (!ParseFileName(name, &number, &type, &log_type) ||
      (type != kTableFile && type != kLogFile)) {
1927 1928
    ROCKS_LOG_ERROR(immutable_db_options_.info_log, "DeleteFile %s failed.\n",
                    name.c_str());
1929 1930 1931
    return Status::InvalidArgument("Invalid file name");
  }

1932 1933 1934 1935
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
1936 1937 1938
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed - not archived log.\n",
                      name.c_str());
1939 1940
      return Status::NotSupported("Delete only supported for archived logs");
    }
1941 1942
    status =
        env_->DeleteFile(immutable_db_options_.wal_dir + "/" + name.c_str());
1943
    if (!status.ok()) {
1944 1945 1946
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed -- %s.\n", name.c_str(),
                      status.ToString().c_str());
1947 1948 1949 1950
    }
    return status;
  }

1951
  int level;
I
Igor Canadi 已提交
1952
  FileMetaData* metadata;
1953
  ColumnFamilyData* cfd;
1954
  VersionEdit edit;
1955
  JobContext job_context(next_job_id_.fetch_add(1), true);
D
Dhruba Borthakur 已提交
1956
  {
1957
    InstrumentedMutexLock l(&mutex_);
1958
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
1959
    if (!status.ok()) {
1960 1961
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "DeleteFile %s failed. File not found\n", name.c_str());
I
Igor Canadi 已提交
1962
      job_context.Clean();
D
Dhruba Borthakur 已提交
1963 1964
      return Status::InvalidArgument("File not found");
    }
1965
    assert(level < cfd->NumberLevels());
1966

D
Dhruba Borthakur 已提交
1967
    // If the file is being compacted no need to delete.
1968
    if (metadata->being_compacted) {
1969 1970 1971
      ROCKS_LOG_INFO(immutable_db_options_.info_log,
                     "DeleteFile %s Skipped. File about to be compacted\n",
                     name.c_str());
I
Igor Canadi 已提交
1972
      job_context.Clean();
D
Dhruba Borthakur 已提交
1973
      return Status::OK();
1974 1975
    }

D
Dhruba Borthakur 已提交
1976 1977 1978
    // 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 已提交
1979
    auto* vstoreage = cfd->current()->storage_info();
I
Igor Canadi 已提交
1980
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
S
sdong 已提交
1981
      if (vstoreage->NumLevelFiles(i) != 0) {
1982 1983 1984
        ROCKS_LOG_WARN(immutable_db_options_.info_log,
                       "DeleteFile %s FAILED. File not in last level\n",
                       name.c_str());
I
Igor Canadi 已提交
1985
        job_context.Clean();
D
Dhruba Borthakur 已提交
1986 1987 1988
        return Status::InvalidArgument("File not in last level");
      }
    }
1989
    // if level == 0, it has to be the oldest file
S
sdong 已提交
1990 1991
    if (level == 0 &&
        vstoreage->LevelFiles(0).back()->fd.GetNumber() != number) {
1992 1993 1994 1995
      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 已提交
1996
      job_context.Clean();
1997 1998 1999
      return Status::InvalidArgument("File in level 0, but not oldest");
    }
    edit.SetColumnFamily(cfd->GetID());
D
Dhruba Borthakur 已提交
2000
    edit.DeleteFile(level, number);
2001
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
2002
                                    &edit, &mutex_, directories_.GetDbDir());
I
Igor Canadi 已提交
2003
    if (status.ok()) {
I
Igor Canadi 已提交
2004 2005
      InstallSuperVersionAndScheduleWorkWrapper(
          cfd, &job_context, *cfd->GetLatestMutableCFOptions());
I
Igor Canadi 已提交
2006
    }
I
Igor Canadi 已提交
2007 2008
    FindObsoleteFiles(&job_context, false);
  }  // lock released here
2009

2010
  LogFlush(immutable_db_options_.info_log);
2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025
  // remove files outside the db-lock
  if (job_context.HaveSomethingToDelete()) {
    // Call PurgeObsoleteFiles() without holding mutex.
    PurgeObsoleteFiles(job_context);
  }
  job_context.Clean();
  return status;
}

Status DBImpl::DeleteFilesInRange(ColumnFamilyHandle* column_family,
                                  const Slice* begin, const Slice* end) {
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  ColumnFamilyData* cfd = cfh->cfd();
  VersionEdit edit;
2026
  std::vector<FileMetaData*> deleted_files;
2027 2028 2029
  JobContext job_context(next_job_id_.fetch_add(1), true);
  {
    InstrumentedMutexLock l(&mutex_);
2030
    Version* input_version = cfd->current();
2031

2032 2033
    auto* vstorage = input_version->storage_info();
    for (int i = 1; i < cfd->NumberLevels(); i++) {
2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054
      if (vstorage->LevelFiles(i).empty() ||
          !vstorage->OverlapInLevel(i, begin, end)) {
        continue;
      }
      std::vector<FileMetaData*> level_files;
      InternalKey begin_storage, end_storage, *begin_key, *end_key;
      if (begin == nullptr) {
        begin_key = nullptr;
      } else {
        begin_storage.SetMaxPossibleForUserKey(*begin);
        begin_key = &begin_storage;
      }
      if (end == nullptr) {
        end_key = nullptr;
      } else {
        end_storage.SetMinPossibleForUserKey(*end);
        end_key = &end_storage;
      }

      vstorage->GetOverlappingInputs(i, begin_key, end_key, &level_files, -1,
                                     nullptr, false);
2055 2056 2057
      FileMetaData* level_file;
      for (uint32_t j = 0; j < level_files.size(); j++) {
        level_file = level_files[j];
2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068
        if (((begin == nullptr) ||
             (cfd->internal_comparator().user_comparator()->Compare(
                  level_file->smallest.user_key(), *begin) >= 0)) &&
            ((end == nullptr) ||
             (cfd->internal_comparator().user_comparator()->Compare(
                  level_file->largest.user_key(), *end) <= 0))) {
          if (level_file->being_compacted) {
            continue;
          }
          edit.SetColumnFamily(cfd->GetID());
          edit.DeleteFile(i, level_file->fd.GetNumber());
2069 2070
          deleted_files.push_back(level_file);
          level_file->being_compacted = true;
2071 2072 2073 2074
        }
      }
    }
    if (edit.GetDeletedFiles().empty()) {
2075
      job_context.Clean();
2076 2077
      return Status::OK();
    }
2078
    input_version->Ref();
2079 2080 2081 2082 2083 2084
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
                                    &edit, &mutex_, directories_.GetDbDir());
    if (status.ok()) {
      InstallSuperVersionAndScheduleWorkWrapper(
          cfd, &job_context, *cfd->GetLatestMutableCFOptions());
    }
2085 2086 2087 2088
    for (auto* deleted_file : deleted_files) {
      deleted_file->being_compacted = false;
    }
    input_version->Unref();
2089 2090 2091
    FindObsoleteFiles(&job_context, false);
  }  // lock released here

2092
  LogFlush(immutable_db_options_.info_log);
I
Igor Canadi 已提交
2093
  // remove files outside the db-lock
I
Igor Canadi 已提交
2094
  if (job_context.HaveSomethingToDelete()) {
2095
    // Call PurgeObsoleteFiles() without holding mutex.
I
Igor Canadi 已提交
2096
    PurgeObsoleteFiles(job_context);
2097
  }
I
Igor Canadi 已提交
2098
  job_context.Clean();
2099 2100 2101
  return status;
}

2102
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
2103
  InstrumentedMutexLock l(&mutex_);
2104
  versions_->GetLiveFilesMetaData(metadata);
2105
}
2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116

void DBImpl::GetColumnFamilyMetaData(
    ColumnFamilyHandle* column_family,
    ColumnFamilyMetaData* cf_meta) {
  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 已提交
2117
#endif  // ROCKSDB_LITE
2118

I
Igor Canadi 已提交
2119 2120 2121 2122 2123 2124 2125
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

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

I
Igor Canadi 已提交
2129 2130
    uint64_t fsize = 0;
    Status s = env_->GetFileSize(file_path, &fsize);
D
dyniusz 已提交
2131 2132 2133 2134
    if (!s.ok() &&
        env_->GetFileSize(Rocks2LevelTableFileName(file_path), &fsize).ok()) {
      s = Status::OK();
    }
I
Igor Canadi 已提交
2135 2136 2137 2138
    if (!s.ok()) {
      corruption_messages +=
          "Can't access " + md.name + ": " + s.ToString() + "\n";
    } else if (fsize != md.size) {
2139
      corruption_messages += "Sst file size mismatch: " + file_path +
I
Igor Canadi 已提交
2140
                             ". Size recorded in manifest " +
2141 2142
                             ToString(md.size) + ", actual size " +
                             ToString(fsize) + "\n";
I
Igor Canadi 已提交
2143 2144 2145 2146 2147 2148 2149 2150 2151
    }
  }
  if (corruption_messages.size() == 0) {
    return Status::OK();
  } else {
    return Status::Corruption(corruption_messages);
  }
}

2152
Status DBImpl::GetDbIdentity(std::string& identity) const {
2153 2154
  std::string idfilename = IdentityFileName(dbname_);
  const EnvOptions soptions;
2155 2156 2157 2158 2159 2160 2161 2162 2163
  unique_ptr<SequentialFileReader> id_file_reader;
  Status s;
  {
    unique_ptr<SequentialFile> idfile;
    s = env_->NewSequentialFile(idfilename, &idfile, soptions);
    if (!s.ok()) {
      return s;
    }
    id_file_reader.reset(new SequentialFileReader(std::move(idfile)));
2164
  }
2165

2166 2167 2168 2169 2170
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
D
Dmitri Smirnov 已提交
2171
  char* buffer = reinterpret_cast<char*>(alloca(file_size));
2172
  Slice id;
2173
  s = id_file_reader->Read(static_cast<size_t>(file_size), &id, buffer);
2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184
  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;
}

2185
// Default implementation -- returns not supported status
L
Lei Jin 已提交
2186
Status DB::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
2187
                              const std::string& column_family_name,
2188
                              ColumnFamilyHandle** handle) {
2189
  return Status::NotSupported("");
2190
}
Y
Yi Wu 已提交
2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204

Status DB::CreateColumnFamilies(
    const ColumnFamilyOptions& cf_options,
    const std::vector<std::string>& column_family_names,
    std::vector<ColumnFamilyHandle*>* handles) {
  return Status::NotSupported("");
}

Status DB::CreateColumnFamilies(
    const std::vector<ColumnFamilyDescriptor>& column_families,
    std::vector<ColumnFamilyHandle*>* handles) {
  return Status::NotSupported("");
}

2205
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
2206
  return Status::NotSupported("");
2207
}
Y
Yi Wu 已提交
2208 2209 2210 2211 2212 2213

Status DB::DropColumnFamilies(
    const std::vector<ColumnFamilyHandle*>& column_families) {
  return Status::NotSupported("");
}

2214 2215 2216 2217
Status DB::DestroyColumnFamilyHandle(ColumnFamilyHandle* column_family) {
  delete column_family;
  return Status::OK();
}
2218

J
jorlow@chromium.org 已提交
2219 2220
DB::~DB() { }

2221 2222 2223
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
2224
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
2225 2226
}

2227 2228 2229
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2230
Status DestroyDB(const std::string& dbname, const Options& options) {
2231
  const ImmutableDBOptions soptions(SanitizeOptions(dbname, options));
2232
  Env* env = soptions.env;
J
jorlow@chromium.org 已提交
2233
  std::vector<std::string> filenames;
2234

J
jorlow@chromium.org 已提交
2235 2236
  // Ignore error in case directory does not exist
  env->GetChildren(dbname, &filenames);
2237

J
jorlow@chromium.org 已提交
2238
  FileLock* lock;
2239 2240
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
2241 2242 2243
  if (result.ok()) {
    uint64_t number;
    FileType type;
2244
    InfoLogPrefix info_log_prefix(!soptions.db_log_dir.empty(), dbname);
D
dgrogan@chromium.org 已提交
2245
    for (size_t i = 0; i < filenames.size(); i++) {
2246
      if (ParseFileName(filenames[i], &number, info_log_prefix.prefix, &type) &&
2247
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
2248
        Status del;
2249
        std::string path_to_delete = dbname + "/" + filenames[i];
K
Kosie van der Merwe 已提交
2250
        if (type == kMetaDatabase) {
2251 2252
          del = DestroyDB(path_to_delete, options);
        } else if (type == kTableFile) {
2253
          del = DeleteSSTFile(&soptions, path_to_delete, 0);
K
Kosie van der Merwe 已提交
2254
        } else {
2255
          del = env->DeleteFile(path_to_delete);
K
Kosie van der Merwe 已提交
2256
        }
J
jorlow@chromium.org 已提交
2257 2258 2259 2260 2261
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2262

2263 2264
    for (size_t path_id = 0; path_id < options.db_paths.size(); path_id++) {
      const auto& db_path = options.db_paths[path_id];
2265
      env->GetChildren(db_path.path, &filenames);
2266 2267 2268
      for (size_t i = 0; i < filenames.size(); i++) {
        if (ParseFileName(filenames[i], &number, &type) &&
            type == kTableFile) {  // Lock file will be deleted at end
2269
          std::string table_path = db_path.path + "/" + filenames[i];
2270
          Status del = DeleteSSTFile(&soptions, table_path,
2271
                                     static_cast<uint32_t>(path_id));
2272 2273 2274 2275 2276 2277 2278
          if (result.ok() && !del.ok()) {
            result = del;
          }
        }
      }
    }

I
Igor Canadi 已提交
2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296
    std::vector<std::string> walDirFiles;
    std::string archivedir = ArchivalDirectory(dbname);
    if (dbname != soptions.wal_dir) {
      env->GetChildren(soptions.wal_dir, &walDirFiles);
      archivedir = ArchivalDirectory(soptions.wal_dir);
    }

    // Delete log files in the WAL dir
    for (const auto& file : walDirFiles) {
      if (ParseFileName(file, &number, &type) && type == kLogFile) {
        Status del = env->DeleteFile(soptions.wal_dir + "/" + file);
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }

    std::vector<std::string> archiveFiles;
2297
    env->GetChildren(archivedir, &archiveFiles);
2298 2299
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
2300 2301
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
2302
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
2303 2304 2305 2306 2307
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2308

I
Islam AbdelRahman 已提交
2309
    // ignore case where no archival directory is present
2310
    env->DeleteDir(archivedir);
2311

J
jorlow@chromium.org 已提交
2312
    env->UnlockFile(lock);  // Ignore error since state is already gone
2313
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2314
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
2315
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
2316 2317 2318 2319
  }
  return result;
}

Y
Yi Wu 已提交
2320 2321
Status DBImpl::WriteOptionsFile(bool need_mutex_lock,
                                bool need_enter_write_thread) {
2322
#ifndef ROCKSDB_LITE
Y
Yi Wu 已提交
2323 2324 2325 2326 2327 2328 2329 2330 2331
  WriteThread::Writer w;
  if (need_mutex_lock) {
    mutex_.Lock();
  } else {
    mutex_.AssertHeld();
  }
  if (need_enter_write_thread) {
    write_thread_.EnterUnbatched(&w, &mutex_);
  }
2332 2333 2334

  std::vector<std::string> cf_names;
  std::vector<ColumnFamilyOptions> cf_opts;
2335 2336 2337 2338 2339

  // This part requires mutex to protect the column family options
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    if (cfd->IsDropped()) {
      continue;
2340
    }
2341
    cf_names.push_back(cfd->GetName());
2342
    cf_opts.push_back(cfd->GetLatestCFOptions());
2343 2344
  }

2345 2346
  // Unlock during expensive operations.  New writes cannot get here
  // because the single write thread ensures all new writes get queued.
2347 2348
  DBOptions db_options =
      BuildDBOptions(immutable_db_options_, mutable_db_options_);
2349 2350 2351 2352
  mutex_.Unlock();

  std::string file_name =
      TempOptionsFileName(GetName(), versions_->NewFileNumber());
2353 2354
  Status s =
      PersistRocksDBOptions(db_options, cf_names, cf_opts, file_name, GetEnv());
2355 2356 2357 2358

  if (s.ok()) {
    s = RenameTempFileToOptionsFile(file_name);
  }
Y
Yi Wu 已提交
2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373
  // 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());
    }
  }
2374
#endif  // !ROCKSDB_LITE
Y
Yi Wu 已提交
2375
  return Status::OK();
2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389
}

#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()) {
2390 2391
      ROCKS_LOG_WARN(info_log, "Unable to delete options file %s",
                     iter->second.c_str());
2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421
    }
  }
}
}  // 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,
2422
                           immutable_db_options_.info_log, GetEnv());
2423 2424 2425 2426 2427 2428 2429 2430 2431
  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 已提交
2432 2433

  versions_->options_file_number_ = versions_->NewFileNumber();
2434
  std::string options_file_name =
W
Wanning Jiang 已提交
2435
      OptionsFileName(GetName(), versions_->options_file_number_);
2436 2437 2438 2439 2440 2441 2442 2443 2444 2445
  // Retry if the file name happen to conflict with an existing one.
  s = GetEnv()->RenameFile(file_name, options_file_name);

  DeleteObsoleteOptionsFiles();
  return s;
#else
  return Status::OK();
#endif  // !ROCKSDB_LITE
}

D
Daniel Black 已提交
2446
#ifdef ROCKSDB_USING_THREAD_STATUS
2447

Y
Yueh-Hsuan Chiang 已提交
2448 2449
void DBImpl::NewThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
2450
  if (immutable_db_options_.enable_thread_tracking) {
2451 2452
    ThreadStatusUtil::NewColumnFamilyInfo(this, cfd, cfd->GetName(),
                                          cfd->ioptions()->env);
2453
  }
Y
Yueh-Hsuan Chiang 已提交
2454 2455 2456 2457
}

void DBImpl::EraseThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
2458
  if (immutable_db_options_.enable_thread_tracking) {
2459
    ThreadStatusUtil::EraseColumnFamilyInfo(cfd);
2460
  }
Y
Yueh-Hsuan Chiang 已提交
2461 2462 2463
}

void DBImpl::EraseThreadStatusDbInfo() const {
2464
  if (immutable_db_options_.enable_thread_tracking) {
2465
    ThreadStatusUtil::EraseDatabaseInfo(this);
2466
  }
Y
Yueh-Hsuan Chiang 已提交
2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481
}

#else
void DBImpl::NewThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
}

void DBImpl::EraseThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
}

void DBImpl::EraseThreadStatusDbInfo() const {
}
#endif  // ROCKSDB_USING_THREAD_STATUS

2482 2483
//
// A global method that can dump out the build version
I
Igor Canadi 已提交
2484
void DumpRocksDBBuildVersion(Logger * log) {
I
Igor Canadi 已提交
2485
#if !defined(IOS_CROSS_COMPILE)
I
Igor Canadi 已提交
2486 2487
  // if we compile with Xcode, we don't run build_detect_vesion, so we don't
  // generate util/build_version.cc
2488 2489 2490 2491
  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 已提交
2492
#endif
2493 2494
}

A
agiardullo 已提交
2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511
#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
2512 2513 2514
Status DBImpl::GetLatestSequenceForKey(SuperVersion* sv, const Slice& key,
                                       bool cache_only, SequenceNumber* seq,
                                       bool* found_record_for_key) {
A
agiardullo 已提交
2515 2516
  Status s;
  MergeContext merge_context;
2517 2518
  RangeDelAggregator range_del_agg(sv->mem->GetInternalKeyComparator(),
                                   kMaxSequenceNumber);
A
agiardullo 已提交
2519

A
Andrew Kryczka 已提交
2520
  ReadOptions read_options;
A
agiardullo 已提交
2521 2522 2523 2524
  SequenceNumber current_seq = versions_->LastSequence();
  LookupKey lkey(key, current_seq);

  *seq = kMaxSequenceNumber;
2525 2526
  *found_record_for_key = false;

A
agiardullo 已提交
2527
  // Check if there is a record for this key in the latest memtable
M
Maysam Yabandeh 已提交
2528
  sv->mem->Get(lkey, nullptr, &s, &merge_context, &range_del_agg, seq,
A
Andrew Kryczka 已提交
2529
               read_options);
A
agiardullo 已提交
2530 2531 2532

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2533 2534 2535
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTable::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2536 2537 2538 2539 2540 2541

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check immutable memtables
2542
    *found_record_for_key = true;
A
agiardullo 已提交
2543 2544 2545 2546
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
M
Maysam Yabandeh 已提交
2547
  sv->imm->Get(lkey, nullptr, &s, &merge_context, &range_del_agg, seq,
A
Andrew Kryczka 已提交
2548
               read_options);
A
agiardullo 已提交
2549 2550 2551

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2552 2553 2554
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTableList::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2555 2556 2557 2558 2559 2560

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check memtable history
2561
    *found_record_for_key = true;
A
agiardullo 已提交
2562 2563 2564 2565
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
A
Andrew Kryczka 已提交
2566 2567
  sv->imm->GetFromHistory(lkey, nullptr, &s, &merge_context, &range_del_agg,
                          seq, read_options);
A
agiardullo 已提交
2568 2569 2570

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2571 2572
    ROCKS_LOG_ERROR(
        immutable_db_options_.info_log,
A
agiardullo 已提交
2573 2574 2575 2576 2577 2578
        "Unexpected status returned from MemTableList::GetFromHistory: %s\n",
        s.ToString().c_str());

    return s;
  }

2579 2580 2581 2582 2583 2584 2585 2586 2587 2588
  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 已提交
2589
    sv->current->Get(read_options, lkey, nullptr, &s, &merge_context,
A
Andrew Kryczka 已提交
2590 2591
                     &range_del_agg, nullptr /* value_found */,
                     found_record_for_key, seq);
2592 2593 2594

    if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
      // unexpected error reading SST files
2595 2596 2597
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "Unexpected status returned from Version::Get: %s\n",
                      s.ToString().c_str());
2598 2599 2600 2601 2602

      return s;
    }
  }

A
agiardullo 已提交
2603 2604
  return Status::OK();
}
2605 2606 2607 2608 2609 2610 2611 2612 2613

Status DBImpl::IngestExternalFile(
    ColumnFamilyHandle* column_family,
    const std::vector<std::string>& external_files,
    const IngestExternalFileOptions& ingestion_options) {
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

2614 2615 2616 2617 2618 2619 2620 2621 2622
  // 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(
        "Can't ingest_behind file in DB with allow_ingest_behind=false");
    }
  }

2623 2624 2625 2626 2627 2628 2629
  ExternalSstFileIngestionJob ingestion_job(env_, versions_.get(), cfd,
                                            immutable_db_options_, env_options_,
                                            &snapshots_, ingestion_options);

  std::list<uint64_t>::iterator pending_output_elem;
  {
    InstrumentedMutexLock l(&mutex_);
2630 2631 2632 2633 2634 2635
    if (!bg_error_.ok()) {
      // Don't ingest files when there is a bg_error
      return bg_error_;
    }

    // Make sure that bg cleanup wont delete the files that we are ingesting
2636 2637 2638 2639 2640 2641 2642 2643
    pending_output_elem = CaptureCurrentFileNumberInPendingOutputs();
  }

  status = ingestion_job.Prepare(external_files);
  if (!status.ok()) {
    return status;
  }

2644
  TEST_SYNC_POINT("DBImpl::AddFile:Start");
2645 2646 2647 2648 2649 2650 2651 2652 2653
  {
    // Lock db mutex
    InstrumentedMutexLock l(&mutex_);
    TEST_SYNC_POINT("DBImpl::AddFile:MutexLock");

    // Stop writes to the DB
    WriteThread::Writer w;
    write_thread_.EnterUnbatched(&w, &mutex_);

2654 2655
    num_running_ingest_file_++;

2656 2657 2658 2659 2660 2661
    // We cannot ingest a file into a dropped CF
    if (cfd->IsDropped()) {
      status = Status::InvalidArgument(
          "Cannot ingest an external file into a dropped CF");
    }

2662
    // Figure out if we need to flush the memtable first
2663 2664 2665
    if (status.ok()) {
      bool need_flush = false;
      status = ingestion_job.NeedsFlush(&need_flush);
2666 2667
      TEST_SYNC_POINT_CALLBACK("DBImpl::IngestExternalFile:NeedFlush",
                               &need_flush);
2668 2669 2670 2671 2672
      if (status.ok() && need_flush) {
        mutex_.Unlock();
        status = FlushMemTable(cfd, FlushOptions(), true /* writes_stopped */);
        mutex_.Lock();
      }
2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713
    }

    // 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()) {
      delete InstallSuperVersionAndScheduleWork(cfd, nullptr,
                                                *mutable_cf_options);
    }

    // Resume writes to the DB
    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
  ingestion_job.Cleanup(status);

2714 2715 2716 2717
  if (status.ok()) {
    NotifyOnExternalFileIngested(cfd, ingestion_job);
  }

2718 2719 2720
  return status;
}

2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742
void DBImpl::NotifyOnExternalFileIngested(
    ColumnFamilyData* cfd, const ExternalSstFileIngestionJob& ingestion_job) {
#ifndef ROCKSDB_LITE
  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);
    }
  }

#endif
}

2743 2744 2745 2746 2747 2748 2749
void DBImpl::WaitForIngestFile() {
  mutex_.AssertHeld();
  while (num_running_ingest_file_ > 0) {
    bg_cv_.Wait();
  }
}

A
agiardullo 已提交
2750 2751
#endif  // ROCKSDB_LITE

2752
}  // namespace rocksdb