db_impl.cc 92.8 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

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

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

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

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

S
Siying Dong 已提交
126
namespace {
127
void DumpSupportInfo(Logger* logger) {
128 129 130 131 132 133 134 135
  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 已提交
136
}
137 138

int64_t kDefaultLowPriThrottledRate = 2 * 1024 * 1024;
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
      nonmem_write_thread_(immutable_db_options_),
164
      write_controller_(mutable_db_options_.delayed_write_rate),
165 166 167 168 169
      // Use delayed_write_rate as a base line to determine the initial
      // low pri write rate limit. It may be adjusted later.
      low_pri_write_rate_limiter_(NewGenericRateLimiter(std::min(
          static_cast<int64_t>(mutable_db_options_.delayed_write_rate / 8),
          kDefaultLowPriThrottledRate))),
S
sdong 已提交
170
      last_batch_group_size_(0),
171 172
      unscheduled_flushes_(0),
      unscheduled_compactions_(0),
173
      bg_compaction_scheduled_(0),
174
      num_running_compactions_(0),
175
      bg_flush_scheduled_(0),
176
      num_running_flushes_(0),
177
      bg_purge_scheduled_(0),
178
      disable_delete_obsolete_files_(0),
179
      delete_obsolete_files_last_run_(env_->NowMicros()),
180
      last_stats_dump_time_microsec_(0),
181
      next_job_id_(1),
182
      has_unpersisted_data_(false),
183
      unable_to_flush_oldest_log_(false),
184
      env_options_(BuildDBOptions(immutable_db_options_, mutable_db_options_)),
185
      num_running_ingest_file_(0),
I
Igor Canadi 已提交
186
#ifndef ROCKSDB_LITE
187
      wal_manager_(immutable_db_options_, env_options_),
I
Igor Canadi 已提交
188
#endif  // ROCKSDB_LITE
189
      event_logger_(immutable_db_options_.info_log.get()),
190
      bg_work_paused_(0),
191
      bg_compaction_paused_(0),
192
      refitting_level_(false),
193 194 195
      opened_successfully_(false),
      concurrent_prepare_(options.concurrent_prepare),
      manual_wal_flush_(options.manual_wal_flush) {
H
heyongqiang 已提交
196
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
197

J
jorlow@chromium.org 已提交
198
  // Reserve ten files or so for other uses and give the rest to TableCache.
199
  // Give a large number for setting of "infinite" open files.
L
Leonidas Galanis 已提交
200
  const int table_cache_size = (mutable_db_options_.max_open_files == -1)
201
                                   ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
202
                                   : mutable_db_options_.max_open_files - 10;
203 204
  table_cache_ = NewLRUCache(table_cache_size,
                             immutable_db_options_.table_cache_numshardbits);
205

206
  versions_.reset(new VersionSet(dbname_, &immutable_db_options_, env_options_,
207
                                 table_cache_.get(), write_buffer_manager_,
208
                                 &write_controller_));
209 210
  column_family_memtables_.reset(
      new ColumnFamilyMemTablesImpl(versions_->GetColumnFamilySet()));
211

212 213 214 215 216
  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 已提交
217 218
}

219
// Will lock the mutex_,  will wait for completion if wait is true
220
void DBImpl::CancelAllBackgroundWork(bool wait) {
221
  InstrumentedMutexLock l(&mutex_);
222

223 224
  ROCKS_LOG_INFO(immutable_db_options_.info_log,
                 "Shutdown: canceling all background work");
225

226
  if (!shutting_down_.load(std::memory_order_acquire) &&
227
      has_unpersisted_data_.load(std::memory_order_relaxed) &&
Y
Yi Wu 已提交
228
      !mutable_db_options_.avoid_flush_during_shutdown) {
229
    for (auto cfd : *versions_->GetColumnFamilySet()) {
230
      if (!cfd->IsDropped() && cfd->initialized() && !cfd->mem()->IsEmpty()) {
231 232 233 234
        cfd->Ref();
        mutex_.Unlock();
        FlushMemTable(cfd, FlushOptions());
        mutex_.Lock();
235
        cfd->Unref();
236 237
      }
    }
238
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
239
  }
240 241 242 243 244 245 246 247 248 249 250 251 252

  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() {
253 254
  // CancelAllBackgroundWork called with false means we just set the shutdown
  // marker. After this we do a variant of the waiting and unschedule work
255 256
  // (to consider: moving all the waiting into CancelAllBackgroundWork(true))
  CancelAllBackgroundWork(false);
257 258 259 260 261 262 263
  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
264 265 266
  while (bg_compaction_scheduled_ || bg_flush_scheduled_ ||
         bg_purge_scheduled_) {
    TEST_SYNC_POINT("DBImpl::~DBImpl:WaitJob");
267 268
    bg_cv_.Wait();
  }
269
  EraseThreadStatusDbInfo();
I
Igor Canadi 已提交
270 271
  flush_scheduler_.Clear();

272 273 274 275 276 277 278 279 280 281 282 283 284
  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 已提交
285 286 287 288 289
  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();
290 291
  }

I
Igor Canadi 已提交
292 293 294 295 296 297 298 299 300 301
  // 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_) {
302
    JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
303
    FindObsoleteFiles(&job_context, true);
304 305

    mutex_.Unlock();
I
Igor Canadi 已提交
306
    // manifest number starting from 2
I
Igor Canadi 已提交
307 308 309
    job_context.manifest_file_number = 1;
    if (job_context.HaveSomethingToDelete()) {
      PurgeObsoleteFiles(job_context);
310
    }
I
Igor Canadi 已提交
311
    job_context.Clean();
312
    mutex_.Lock();
313 314
  }

315 316 317
  for (auto l : logs_to_free_) {
    delete l;
  }
318 319 320
  for (auto& log : logs_) {
    log.ClearWriter();
  }
321
  logs_.clear();
322

323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
  // 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 已提交
338 339 340 341
  for (auto& txn_entry : recovered_transactions_) {
    delete txn_entry.second;
  }

342
  // versions need to be destroyed before table_cache since it can hold
343 344
  // references to table_cache.
  versions_.reset();
345
  mutex_.Unlock();
I
Igor Canadi 已提交
346 347 348
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
349

350
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "Shutdown complete");
351
  LogFlush(immutable_db_options_.info_log);
J
jorlow@chromium.org 已提交
352 353 354
}

void DBImpl::MaybeIgnoreError(Status* s) const {
355
  if (s->ok() || immutable_db_options_.paranoid_checks) {
J
jorlow@chromium.org 已提交
356 357
    // No change needed
  } else {
358 359
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "Ignoring error %s",
                   s->ToString().c_str());
J
jorlow@chromium.org 已提交
360 361 362 363
    *s = Status::OK();
  }
}

364
const Status DBImpl::CreateArchivalDirectory() {
365 366 367
  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);
368 369 370 371 372
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

373
void DBImpl::PrintStatistics() {
374
  auto dbstats = immutable_db_options_.statistics.get();
375
  if (dbstats) {
376 377
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "STATISTICS:\n %s",
                   dbstats->ToString().c_str());
378 379 380
  }
}

381
void DBImpl::MaybeDumpStats() {
382 383 384 385 386
  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 已提交
387 388 389

  const uint64_t now_micros = env_->NowMicros();

390
  if (last_stats_dump_time_microsec_ + stats_dump_period_sec * 1000000 <=
391
      now_micros) {
H
Haobo Xu 已提交
392 393 394 395 396
    // 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;
397

398
#ifndef ROCKSDB_LITE
399 400 401 402 403 404 405
    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 已提交
406
    std::string stats;
407
    {
408
      InstrumentedMutexLock l(&mutex_);
409 410
      default_cf_internal_stats_->GetStringProperty(
          *db_property_info, DB::Properties::kDBStats, &stats);
411
      for (auto cfd : *versions_->GetColumnFamilySet()) {
412 413 414 415 416
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFStatsNoFileHistogram,
              &stats);
        }
417 418
      }
      for (auto cfd : *versions_->GetColumnFamilySet()) {
419 420 421 422
        if (cfd->initialized()) {
          cfd->internal_stats()->GetStringProperty(
              *cf_property_info, DB::Properties::kCFFileHistogram, &stats);
        }
423 424
      }
    }
425 426 427
    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 已提交
428 429 430 431
    if (immutable_db_options_.dump_malloc_stats) {
      stats.clear();
      DumpMallocStats(&stats);
      if (!stats.empty()) {
432 433 434
        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 已提交
435 436
      }
    }
437
#endif  // !ROCKSDB_LITE
438

439
    PrintStatistics();
440 441 442
  }
}

443 444 445 446 447 448 449 450 451 452
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 已提交
453 454 455 456 457 458
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();
459
  }
S
Siying Dong 已提交
460
  return ret_dir;
461 462
}

S
Siying Dong 已提交
463 464 465 466 467 468 469 470 471 472 473
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");
474 475
  }

S
Siying Dong 已提交
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494
  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;
495

Y
Yi Wu 已提交
496 497
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, true /*need_enter_write_thread*/);
498 499 500
    }
  }

S
Siying Dong 已提交
501 502 503 504 505 506
  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());
507
  }
S
Siying Dong 已提交
508 509 510 511 512
  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 已提交
513
      s = persist_options_status;
514
    }
515
  } else {
S
Siying Dong 已提交
516 517
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "[%s] SetOptions() failed",
                   cfd->GetName().c_str());
518
  }
S
Siying Dong 已提交
519 520 521
  LogFlush(immutable_db_options_.info_log);
  return s;
#endif  // ROCKSDB_LITE
522 523
}

S
Siying Dong 已提交
524 525 526 527 528 529 530 531 532
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 已提交
533
  }
534

S
Siying Dong 已提交
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
  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();
      }
551

552
      write_controller_.set_max_delayed_write_rate(new_options.delayed_write_rate);
L
Leonidas Galanis 已提交
553
      table_cache_.get()->SetCapacity(new_options.max_open_files == -1
554
                                          ? TableCache::kInfiniteCapacity
L
Leonidas Galanis 已提交
555
                                          : new_options.max_open_files - 10);
556

557 558
      mutable_db_options_ = new_options;

559
      write_thread_.EnterUnbatched(&w, &mutex_);
560
      if (total_log_size_ > GetMaxTotalWalSize()) {
561 562 563 564 565 566
        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());
        }
567
      }
Y
Yi Wu 已提交
568 569
      persist_options_status = WriteOptionsFile(
          false /*need_mutex_lock*/, false /*need_enter_write_thread*/);
570
      write_thread_.ExitUnbatched(&w);
571 572
    }
  }
573
  ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions(), inputs:");
574
  for (const auto& o : options_map) {
575 576
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "%s: %s\n", o.first.c_str(),
                   o.second.c_str());
577 578
  }
  if (s.ok()) {
579
    ROCKS_LOG_INFO(immutable_db_options_.info_log, "SetDBOptions() succeeded");
580 581 582 583 584 585 586
    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());
      }
587 588 589
      ROCKS_LOG_WARN(immutable_db_options_.info_log,
                     "Unable to persist options in SetDBOptions() -- %s",
                     persist_options_status.ToString().c_str());
590 591
    }
  } else {
592
    ROCKS_LOG_WARN(immutable_db_options_.info_log, "SetDBOptions failed");
L
Lei Jin 已提交
593
  }
594
  LogFlush(immutable_db_options_.info_log);
595
  return s;
I
Igor Canadi 已提交
596
#endif  // ROCKSDB_LITE
597 598
}

599
// return the same level if it cannot be moved
600 601
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd,
    const MutableCFOptions& mutable_cf_options, int level) {
602
  mutex_.AssertHeld();
S
sdong 已提交
603
  const auto* vstorage = cfd->current()->storage_info();
604
  int minimum_level = level;
605
  for (int i = level - 1; i > 0; --i) {
606
    // stop if level i is not empty
S
sdong 已提交
607
    if (vstorage->NumLevelFiles(i) > 0) break;
608
    // stop if level i is too small (cannot fit the level files)
609
    if (vstorage->MaxBytesForLevel(i) < vstorage->NumLevelBytes(level)) {
610 611
      break;
    }
612 613 614 615 616 617

    minimum_level = i;
  }
  return minimum_level;
}

618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637
Status DBImpl::FlushWAL(bool sync) {
  {
    // We need to lock log_write_mutex_ since logs_ might change concurrently
    InstrumentedMutexLock wl(&log_write_mutex_);
    log::Writer* cur_log_writer = logs_.back().writer;
    auto s = cur_log_writer->WriteBuffer();
    if (!s.ok()) {
      ROCKS_LOG_ERROR(immutable_db_options_.info_log, "WAL flush error %s",
                      s.ToString().c_str());
    }
    if (!sync) {
      ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=false");
      return s;
    }
  }
  // sync = true
  ROCKS_LOG_DEBUG(immutable_db_options_.info_log, "FlushWAL sync=true");
  return SyncWAL();
}

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
  }

676
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:1");
S
Siying Dong 已提交
677 678 679 680 681 682 683 684 685 686
  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();
687
  }
688
  TEST_SYNC_POINT("DBWALTest::SyncWALNotWaitWrite:2");
689

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

S
Siying Dong 已提交
697
  return status;
698 699
}

S
Siying Dong 已提交
700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721
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();
722 723
}

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

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

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

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

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

756 757 758
void DBImpl::BackgroundCallPurge() {
  mutex_.Lock();

759 760 761 762 763 764 765 766 767 768 769 770
  // 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();
771

772 773 774 775 776 777 778 779 780 781 782 783 784
      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();
    }
785 786 787 788 789 790 791 792 793 794 795
  }
  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();
}

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

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

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

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

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

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

844 845
  delete state;
}
H
Hans Wennborg 已提交
846
}  // namespace
847

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

895 896 897 898
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

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

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

912 913 914
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

915 916 917 918 919 920
  // Acquire SuperVersion
  SuperVersion* sv = GetAndRefSuperVersion(cfd);

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

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

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

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

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

979 980
  {
    PERF_TIMER_GUARD(get_post_process_time);
981

982
    ReturnAndCleanupSuperVersion(cfd, sv);
983

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

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

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

1000
  SequenceNumber snapshot;
1001

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

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

1031 1032
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
1033

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

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

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

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

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

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

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

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
1109
  }
1110

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

1117
  return stat_list;
1118 1119
}

L
Lei Jin 已提交
1120
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
Y
Yi Wu 已提交
1121
                                  const std::string& column_family,
1122
                                  ColumnFamilyHandle** handle) {
Y
Yi Wu 已提交
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 1189 1190
  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 已提交
1191
  Status s;
1192
  Status persist_options_status;
I
Igor Canadi 已提交
1193
  *handle = nullptr;
1194 1195

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

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

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

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

1241 1242
      cfd->set_initialized();

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

  // this is outside the mutex
1255
  if (s.ok()) {
1256 1257
    NewThreadStatusCfInfo(
        reinterpret_cast<ColumnFamilyHandleImpl*>(*handle)->cfd());
1258
  }
1259
  return s;
1260 1261
}

1262
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
Y
Yi Wu 已提交
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 1290 1291 1292 1293
  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) {
1294 1295 1296
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
1297 1298
    return Status::InvalidArgument("Can't drop default column family");
  }
1299

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

I
Igor Canadi 已提交
1302 1303
  VersionEdit edit;
  edit.DropColumnFamily();
1304 1305
  edit.SetColumnFamily(cfd->GetID());

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

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

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

1352
  return s;
1353 1354
}

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

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

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

I
Igor Canadi 已提交
1414
    auto snapshot =
1415 1416 1417
        read_options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(
                read_options.snapshot)->number_
I
Igor Canadi 已提交
1418
            : latest_snapshot;
T
Tomislav Novak 已提交
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 1452 1453 1454 1455
    // 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         | <----+
    // |       |   +-------------------+
    // |       |                       |
    // +-------+-----------------------+
    //
1456 1457
    // ArenaWrappedDBIter inlines an arena area where all the iterators in
    // the iterator tree are allocated in the order of being accessed when
1458 1459 1460 1461 1462
    // 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(
1463
        env_, read_options, *cfd->ioptions(), cfd->user_comparator(), snapshot,
S
Siying Dong 已提交
1464
        sv->mutable_cf_options.max_sequential_skip_in_iterations,
1465
        sv->version_number);
1466

S
Siying Dong 已提交
1467 1468 1469 1470
    InternalIterator* internal_iter =
        NewInternalIterator(read_options, cfd, sv, db_iter->GetArena(),
                            db_iter->GetRangeDelAggregator());
    db_iter->SetIterUnderDBIter(internal_iter);
1471

S
Siying Dong 已提交
1472
    return db_iter;
1473
  }
S
Siying Dong 已提交
1474 1475
  // To stop compiler from complaining
  return nullptr;
1476 1477
}

S
Siying Dong 已提交
1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496
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");
1497
    }
S
Siying Dong 已提交
1498 1499 1500 1501
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      auto iter = new ManagedIterator(this, read_options, cfd);
      iterators->push_back(iter);
1502
    }
S
Siying Dong 已提交
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513
#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(
1514
          env_, read_options, *cfd->ioptions(), cfd->user_comparator(), iter,
S
Siying Dong 已提交
1515 1516
          kMaxSequenceNumber,
          sv->mutable_cf_options.max_sequential_skip_in_iterations,
1517
          sv->version_number));
1518
    }
S
Siying Dong 已提交
1519 1520 1521
#endif
  } else {
    SequenceNumber latest_snapshot = versions_->LastSequence();
1522

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

S
Siying Dong 已提交
1528 1529 1530 1531 1532
      auto snapshot =
          read_options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(
                  read_options.snapshot)->number_
              : latest_snapshot;
1533

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

I
Igor Canadi 已提交
1546
  return Status::OK();
S
Stanislau Hlebik 已提交
1547 1548
}

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

S
Siying Dong 已提交
1551 1552 1553
#ifndef ROCKSDB_LITE
const Snapshot* DBImpl::GetSnapshotForWriteConflictBoundary() {
  return GetSnapshotImpl(true);
1554 1555 1556
}
#endif  // ROCKSDB_LITE

S
Siying Dong 已提交
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566
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 已提交
1567
  }
S
Siying Dong 已提交
1568 1569 1570
  return snapshots_.New(s, versions_->LastSequence(), unix_time,
                        is_write_conflict_boundary);
}
1571

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

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

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

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

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}
1610 1611

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

I
Igor Canadi 已提交
1635 1636 1637 1638
const std::string& DBImpl::GetName() const {
  return dbname_;
}

1639 1640 1641 1642
Env* DBImpl::GetEnv() const {
  return env_;
}

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

1650 1651 1652 1653
DBOptions DBImpl::GetDBOptions() const {
  InstrumentedMutexLock l(&mutex_);
  return BuildDBOptions(immutable_db_options_, mutable_db_options_);
}
1654

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

1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698
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;
}

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

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

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

1732 1733 1734
    if (!is_locked) {
      ReturnAndCleanupSuperVersion(cfd, sv);
    }
1735 1736 1737 1738 1739

    return ret;
  }
}

S
Siying Dong 已提交
1740 1741 1742 1743
#ifndef ROCKSDB_LITE
Status DBImpl::ResetStats() {
  InstrumentedMutexLock l(&mutex_);
  for (auto* cfd : *versions_->GetColumnFamilySet()) {
1744 1745 1746
    if (cfd->initialized()) {
      cfd->internal_stats()->Clear();
    }
S
Siying Dong 已提交
1747 1748 1749 1750 1751
  }
  return Status::OK();
}
#endif  // ROCKSDB_LITE

1752 1753
bool DBImpl::GetAggregatedIntProperty(const Slice& property,
                                      uint64_t* aggregated_value) {
1754 1755
  const DBPropertyInfo* property_info = GetPropertyInfo(property);
  if (property_info == nullptr || property_info->handle_int == nullptr) {
1756 1757 1758 1759 1760 1761 1762 1763 1764
    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()) {
1765 1766 1767
      if (!cfd->initialized()) {
        continue;
      }
1768
      if (GetIntPropertyInternal(cfd, *property_info, true, &value)) {
1769 1770 1771 1772 1773 1774 1775 1776 1777 1778
        sum += value;
      } else {
        return false;
      }
    }
  }
  *aggregated_value = sum;
  return true;
}

1779 1780
SuperVersion* DBImpl::GetAndRefSuperVersion(ColumnFamilyData* cfd) {
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
I
Igor Canadi 已提交
1781
  return cfd->GetThreadLocalSuperVersion(&mutex_);
1782 1783
}

A
agiardullo 已提交
1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795
// 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);
}

1796 1797
void DBImpl::ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd,
                                          SuperVersion* sv) {
I
Igor Canadi 已提交
1798
  bool unref_sv = !cfd->ReturnThreadLocalSuperVersion(sv);
1799 1800 1801 1802 1803

  if (unref_sv) {
    // Release SuperVersion
    if (sv->Unref()) {
      {
1804
        InstrumentedMutexLock l(&mutex_);
1805 1806 1807 1808 1809 1810 1811
        sv->Cleanup();
      }
      delete sv;
      RecordTick(stats_, NUMBER_SUPERVERSION_CLEANUPS);
    }
    RecordTick(stats_, NUMBER_SUPERVERSION_RELEASES);
  }
1812 1813
}

A
agiardullo 已提交
1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837
// 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 已提交
1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851
// 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();
}

1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873
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);
}

1874
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
1875
                                 const Range* range, int n, uint64_t* sizes,
1876 1877 1878
                                 uint8_t include_flags) {
  assert(include_flags & DB::SizeApproximationFlags::INCLUDE_FILES ||
         include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES);
J
jorlow@chromium.org 已提交
1879
  Version* v;
1880 1881
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
1882 1883
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
  v = sv->current;
J
jorlow@chromium.org 已提交
1884 1885 1886

  for (int i = 0; i < n; i++) {
    // Convert user_key into a corresponding internal key.
1887 1888
    InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
V
Vitaliy Liptchinsky 已提交
1889
    sizes[i] = 0;
1890
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_FILES) {
V
Vitaliy Liptchinsky 已提交
1891
      sizes[i] += versions_->ApproximateSize(v, k1.Encode(), k2.Encode());
1892 1893
    }
    if (include_flags & DB::SizeApproximationFlags::INCLUDE_MEMTABLES) {
1894 1895
      sizes[i] += sv->mem->ApproximateStats(k1.Encode(), k2.Encode()).size;
      sizes[i] += sv->imm->ApproximateStats(k1.Encode(), k2.Encode()).size;
1896
    }
J
jorlow@chromium.org 已提交
1897 1898
  }

1899
  ReturnAndCleanupSuperVersion(cfd, sv);
J
jorlow@chromium.org 已提交
1900 1901
}

I
Igor Canadi 已提交
1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917
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 已提交
1918 1919 1920 1921 1922
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {

L
Lei Jin 已提交
1923
  RecordTick(stats_, GET_UPDATES_SINCE_CALLS);
I
Igor Canadi 已提交
1924 1925 1926
  if (seq > versions_->LastSequence()) {
    return Status::NotFound("Requested sequence not yet written in the db");
  }
I
Igor Canadi 已提交
1927
  return wal_manager_.GetUpdatesSince(seq, iter, read_options, versions_.get());
I
Igor Canadi 已提交
1928 1929
}

1930 1931 1932
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
1933 1934 1935
  WalFileType log_type;
  if (!ParseFileName(name, &number, &type, &log_type) ||
      (type != kTableFile && type != kLogFile)) {
1936 1937
    ROCKS_LOG_ERROR(immutable_db_options_.info_log, "DeleteFile %s failed.\n",
                    name.c_str());
1938 1939 1940
    return Status::InvalidArgument("Invalid file name");
  }

1941 1942 1943 1944
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
1945 1946 1947
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed - not archived log.\n",
                      name.c_str());
1948 1949
      return Status::NotSupported("Delete only supported for archived logs");
    }
1950 1951
    status =
        env_->DeleteFile(immutable_db_options_.wal_dir + "/" + name.c_str());
1952
    if (!status.ok()) {
1953 1954 1955
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "DeleteFile %s failed -- %s.\n", name.c_str(),
                      status.ToString().c_str());
1956 1957 1958 1959
    }
    return status;
  }

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

D
Dhruba Borthakur 已提交
1976
    // If the file is being compacted no need to delete.
1977
    if (metadata->being_compacted) {
1978 1979 1980
      ROCKS_LOG_INFO(immutable_db_options_.info_log,
                     "DeleteFile %s Skipped. File about to be compacted\n",
                     name.c_str());
I
Igor Canadi 已提交
1981
      job_context.Clean();
D
Dhruba Borthakur 已提交
1982
      return Status::OK();
1983 1984
    }

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

2019
  LogFlush(immutable_db_options_.info_log);
2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034
  // 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;
2035
  std::vector<FileMetaData*> deleted_files;
2036 2037 2038
  JobContext job_context(next_job_id_.fetch_add(1), true);
  {
    InstrumentedMutexLock l(&mutex_);
2039
    Version* input_version = cfd->current();
2040

2041 2042
    auto* vstorage = input_version->storage_info();
    for (int i = 1; i < cfd->NumberLevels(); i++) {
2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063
      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);
2064 2065 2066
      FileMetaData* level_file;
      for (uint32_t j = 0; j < level_files.size(); j++) {
        level_file = level_files[j];
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077
        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());
2078 2079
          deleted_files.push_back(level_file);
          level_file->being_compacted = true;
2080 2081 2082 2083
        }
      }
    }
    if (edit.GetDeletedFiles().empty()) {
2084
      job_context.Clean();
2085 2086
      return Status::OK();
    }
2087
    input_version->Ref();
2088 2089 2090 2091 2092 2093
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
                                    &edit, &mutex_, directories_.GetDbDir());
    if (status.ok()) {
      InstallSuperVersionAndScheduleWorkWrapper(
          cfd, &job_context, *cfd->GetLatestMutableCFOptions());
    }
2094 2095 2096 2097
    for (auto* deleted_file : deleted_files) {
      deleted_file->being_compacted = false;
    }
    input_version->Unref();
2098 2099 2100
    FindObsoleteFiles(&job_context, false);
  }  // lock released here

2101
  LogFlush(immutable_db_options_.info_log);
I
Igor Canadi 已提交
2102
  // remove files outside the db-lock
I
Igor Canadi 已提交
2103
  if (job_context.HaveSomethingToDelete()) {
2104
    // Call PurgeObsoleteFiles() without holding mutex.
I
Igor Canadi 已提交
2105
    PurgeObsoleteFiles(job_context);
2106
  }
I
Igor Canadi 已提交
2107
  job_context.Clean();
2108 2109 2110
  return status;
}

2111
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
2112
  InstrumentedMutexLock l(&mutex_);
2113
  versions_->GetLiveFilesMetaData(metadata);
2114
}
2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125

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 已提交
2126
#endif  // ROCKSDB_LITE
2127

I
Igor Canadi 已提交
2128 2129 2130 2131 2132 2133 2134
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

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

I
Igor Canadi 已提交
2138 2139
    uint64_t fsize = 0;
    Status s = env_->GetFileSize(file_path, &fsize);
D
dyniusz 已提交
2140 2141 2142 2143
    if (!s.ok() &&
        env_->GetFileSize(Rocks2LevelTableFileName(file_path), &fsize).ok()) {
      s = Status::OK();
    }
I
Igor Canadi 已提交
2144 2145 2146 2147
    if (!s.ok()) {
      corruption_messages +=
          "Can't access " + md.name + ": " + s.ToString() + "\n";
    } else if (fsize != md.size) {
2148
      corruption_messages += "Sst file size mismatch: " + file_path +
I
Igor Canadi 已提交
2149
                             ". Size recorded in manifest " +
2150 2151
                             ToString(md.size) + ", actual size " +
                             ToString(fsize) + "\n";
I
Igor Canadi 已提交
2152 2153 2154 2155 2156 2157 2158 2159 2160
    }
  }
  if (corruption_messages.size() == 0) {
    return Status::OK();
  } else {
    return Status::Corruption(corruption_messages);
  }
}

2161
Status DBImpl::GetDbIdentity(std::string& identity) const {
2162 2163
  std::string idfilename = IdentityFileName(dbname_);
  const EnvOptions soptions;
2164 2165 2166 2167 2168 2169 2170 2171 2172
  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)));
2173
  }
2174

2175 2176 2177 2178 2179
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
D
Dmitri Smirnov 已提交
2180
  char* buffer = reinterpret_cast<char*>(alloca(file_size));
2181
  Slice id;
2182
  s = id_file_reader->Read(static_cast<size_t>(file_size), &id, buffer);
2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193
  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;
}

2194
// Default implementation -- returns not supported status
L
Lei Jin 已提交
2195
Status DB::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
2196
                              const std::string& column_family_name,
2197
                              ColumnFamilyHandle** handle) {
2198
  return Status::NotSupported("");
2199
}
Y
Yi Wu 已提交
2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213

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("");
}

2214
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
2215
  return Status::NotSupported("");
2216
}
Y
Yi Wu 已提交
2217 2218 2219 2220 2221 2222

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

2223 2224 2225 2226
Status DB::DestroyColumnFamilyHandle(ColumnFamilyHandle* column_family) {
  delete column_family;
  return Status::OK();
}
2227

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

2230 2231 2232
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
2233
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
2234 2235
}

2236 2237 2238
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
2239
Status DestroyDB(const std::string& dbname, const Options& options) {
2240
  const ImmutableDBOptions soptions(SanitizeOptions(dbname, options));
2241
  Env* env = soptions.env;
J
jorlow@chromium.org 已提交
2242
  std::vector<std::string> filenames;
2243

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

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

2272 2273
    for (size_t path_id = 0; path_id < options.db_paths.size(); path_id++) {
      const auto& db_path = options.db_paths[path_id];
2274
      env->GetChildren(db_path.path, &filenames);
2275 2276 2277
      for (size_t i = 0; i < filenames.size(); i++) {
        if (ParseFileName(filenames[i], &number, &type) &&
            type == kTableFile) {  // Lock file will be deleted at end
2278
          std::string table_path = db_path.path + "/" + filenames[i];
2279
          Status del = DeleteSSTFile(&soptions, table_path,
2280
                                     static_cast<uint32_t>(path_id));
2281 2282 2283 2284 2285 2286 2287
          if (result.ok() && !del.ok()) {
            result = del;
          }
        }
      }
    }

I
Igor Canadi 已提交
2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305
    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;
2306
    env->GetChildren(archivedir, &archiveFiles);
2307 2308
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
2309 2310
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
2311
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
2312 2313 2314 2315 2316
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
2317

I
Islam AbdelRahman 已提交
2318
    // ignore case where no archival directory is present
2319
    env->DeleteDir(archivedir);
2320

J
jorlow@chromium.org 已提交
2321
    env->UnlockFile(lock);  // Ignore error since state is already gone
2322
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
2323
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
2324
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
2325 2326 2327 2328
  }
  return result;
}

Y
Yi Wu 已提交
2329 2330
Status DBImpl::WriteOptionsFile(bool need_mutex_lock,
                                bool need_enter_write_thread) {
2331
#ifndef ROCKSDB_LITE
Y
Yi Wu 已提交
2332 2333 2334 2335 2336 2337 2338 2339 2340
  WriteThread::Writer w;
  if (need_mutex_lock) {
    mutex_.Lock();
  } else {
    mutex_.AssertHeld();
  }
  if (need_enter_write_thread) {
    write_thread_.EnterUnbatched(&w, &mutex_);
  }
2341 2342 2343

  std::vector<std::string> cf_names;
  std::vector<ColumnFamilyOptions> cf_opts;
2344 2345 2346 2347 2348

  // This part requires mutex to protect the column family options
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    if (cfd->IsDropped()) {
      continue;
2349
    }
2350
    cf_names.push_back(cfd->GetName());
2351
    cf_opts.push_back(cfd->GetLatestCFOptions());
2352 2353
  }

2354 2355
  // Unlock during expensive operations.  New writes cannot get here
  // because the single write thread ensures all new writes get queued.
2356 2357
  DBOptions db_options =
      BuildDBOptions(immutable_db_options_, mutable_db_options_);
2358 2359
  mutex_.Unlock();

2360 2361 2362
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:1");
  TEST_SYNC_POINT("DBImpl::WriteOptionsFile:2");

2363 2364
  std::string file_name =
      TempOptionsFileName(GetName(), versions_->NewFileNumber());
2365 2366
  Status s =
      PersistRocksDBOptions(db_options, cf_names, cf_opts, file_name, GetEnv());
2367 2368 2369 2370

  if (s.ok()) {
    s = RenameTempFileToOptionsFile(file_name);
  }
Y
Yi Wu 已提交
2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385
  // 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());
    }
  }
2386
#endif  // !ROCKSDB_LITE
Y
Yi Wu 已提交
2387
  return Status::OK();
2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401
}

#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()) {
2402 2403
      ROCKS_LOG_WARN(info_log, "Unable to delete options file %s",
                     iter->second.c_str());
2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433
    }
  }
}
}  // 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,
2434
                           immutable_db_options_.info_log, GetEnv());
2435 2436 2437 2438 2439 2440 2441 2442 2443
  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 已提交
2444 2445

  versions_->options_file_number_ = versions_->NewFileNumber();
2446
  std::string options_file_name =
W
Wanning Jiang 已提交
2447
      OptionsFileName(GetName(), versions_->options_file_number_);
2448 2449 2450 2451 2452 2453 2454 2455 2456 2457
  // 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 已提交
2458
#ifdef ROCKSDB_USING_THREAD_STATUS
2459

Y
Yueh-Hsuan Chiang 已提交
2460 2461
void DBImpl::NewThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
2462
  if (immutable_db_options_.enable_thread_tracking) {
2463 2464
    ThreadStatusUtil::NewColumnFamilyInfo(this, cfd, cfd->GetName(),
                                          cfd->ioptions()->env);
2465
  }
Y
Yueh-Hsuan Chiang 已提交
2466 2467 2468 2469
}

void DBImpl::EraseThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
2470
  if (immutable_db_options_.enable_thread_tracking) {
2471
    ThreadStatusUtil::EraseColumnFamilyInfo(cfd);
2472
  }
Y
Yueh-Hsuan Chiang 已提交
2473 2474 2475
}

void DBImpl::EraseThreadStatusDbInfo() const {
2476
  if (immutable_db_options_.enable_thread_tracking) {
2477
    ThreadStatusUtil::EraseDatabaseInfo(this);
2478
  }
Y
Yueh-Hsuan Chiang 已提交
2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493
}

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

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

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

2494 2495
//
// A global method that can dump out the build version
I
Igor Canadi 已提交
2496
void DumpRocksDBBuildVersion(Logger * log) {
I
Igor Canadi 已提交
2497
#if !defined(IOS_CROSS_COMPILE)
H
hyunwoo 已提交
2498
  // if we compile with Xcode, we don't run build_detect_version, so we don't
I
Igor Canadi 已提交
2499
  // generate util/build_version.cc
2500 2501 2502 2503
  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 已提交
2504
#endif
2505 2506
}

A
agiardullo 已提交
2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523
#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
2524 2525 2526
Status DBImpl::GetLatestSequenceForKey(SuperVersion* sv, const Slice& key,
                                       bool cache_only, SequenceNumber* seq,
                                       bool* found_record_for_key) {
A
agiardullo 已提交
2527 2528
  Status s;
  MergeContext merge_context;
2529 2530
  RangeDelAggregator range_del_agg(sv->mem->GetInternalKeyComparator(),
                                   kMaxSequenceNumber);
A
agiardullo 已提交
2531

A
Andrew Kryczka 已提交
2532
  ReadOptions read_options;
A
agiardullo 已提交
2533 2534 2535 2536
  SequenceNumber current_seq = versions_->LastSequence();
  LookupKey lkey(key, current_seq);

  *seq = kMaxSequenceNumber;
2537 2538
  *found_record_for_key = false;

A
agiardullo 已提交
2539
  // Check if there is a record for this key in the latest memtable
M
Maysam Yabandeh 已提交
2540
  sv->mem->Get(lkey, nullptr, &s, &merge_context, &range_del_agg, seq,
A
Andrew Kryczka 已提交
2541
               read_options);
A
agiardullo 已提交
2542 2543 2544

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2545 2546 2547
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTable::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2548 2549 2550 2551 2552 2553

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check immutable memtables
2554
    *found_record_for_key = true;
A
agiardullo 已提交
2555 2556 2557 2558
    return Status::OK();
  }

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

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2564 2565 2566
    ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                    "Unexpected status returned from MemTableList::Get: %s\n",
                    s.ToString().c_str());
A
agiardullo 已提交
2567 2568 2569 2570 2571 2572

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check memtable history
2573
    *found_record_for_key = true;
A
agiardullo 已提交
2574 2575 2576 2577
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
A
Andrew Kryczka 已提交
2578 2579
  sv->imm->GetFromHistory(lkey, nullptr, &s, &merge_context, &range_del_agg,
                          seq, read_options);
A
agiardullo 已提交
2580 2581 2582

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
2583 2584
    ROCKS_LOG_ERROR(
        immutable_db_options_.info_log,
A
agiardullo 已提交
2585 2586 2587 2588 2589 2590
        "Unexpected status returned from MemTableList::GetFromHistory: %s\n",
        s.ToString().c_str());

    return s;
  }

2591 2592 2593 2594 2595 2596 2597 2598 2599 2600
  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 已提交
2601
    sv->current->Get(read_options, lkey, nullptr, &s, &merge_context,
A
Andrew Kryczka 已提交
2602 2603
                     &range_del_agg, nullptr /* value_found */,
                     found_record_for_key, seq);
2604 2605 2606

    if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
      // unexpected error reading SST files
2607 2608 2609
      ROCKS_LOG_ERROR(immutable_db_options_.info_log,
                      "Unexpected status returned from Version::Get: %s\n",
                      s.ToString().c_str());
2610 2611 2612 2613 2614

      return s;
    }
  }

A
agiardullo 已提交
2615 2616
  return Status::OK();
}
2617 2618 2619 2620 2621 2622 2623 2624 2625

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();

2626 2627 2628 2629 2630 2631 2632 2633 2634
  // 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");
    }
  }

2635 2636 2637 2638 2639 2640 2641
  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_);
2642 2643 2644 2645 2646 2647
    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
2648 2649 2650 2651 2652 2653 2654 2655
    pending_output_elem = CaptureCurrentFileNumberInPendingOutputs();
  }

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

2656
  TEST_SYNC_POINT("DBImpl::AddFile:Start");
2657 2658 2659 2660 2661
  {
    // Lock db mutex
    InstrumentedMutexLock l(&mutex_);
    TEST_SYNC_POINT("DBImpl::AddFile:MutexLock");

2662
    // Stop writes to the DB by entering both write threads
2663 2664
    WriteThread::Writer w;
    write_thread_.EnterUnbatched(&w, &mutex_);
2665 2666 2667 2668
    WriteThread::Writer nonmem_w;
    if (concurrent_prepare_) {
      nonmem_write_thread_.EnterUnbatched(&nonmem_w, &mutex_);
    }
2669

2670 2671
    num_running_ingest_file_++;

2672 2673 2674 2675 2676 2677
    // We cannot ingest a file into a dropped CF
    if (cfd->IsDropped()) {
      status = Status::InvalidArgument(
          "Cannot ingest an external file into a dropped CF");
    }

2678
    // Figure out if we need to flush the memtable first
2679 2680 2681
    if (status.ok()) {
      bool need_flush = false;
      status = ingestion_job.NeedsFlush(&need_flush);
2682 2683
      TEST_SYNC_POINT_CALLBACK("DBImpl::IngestExternalFile:NeedFlush",
                               &need_flush);
2684 2685 2686 2687 2688
      if (status.ok() && need_flush) {
        mutex_.Unlock();
        status = FlushMemTable(cfd, FlushOptions(), true /* writes_stopped */);
        mutex_.Lock();
      }
2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708
    }

    // 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
2709 2710 2711
    if (concurrent_prepare_) {
      nonmem_write_thread_.ExitUnbatched(&nonmem_w);
    }
2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732
    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);

2733 2734 2735 2736
  if (status.ok()) {
    NotifyOnExternalFileIngested(cfd, ingestion_job);
  }

2737 2738 2739
  return status;
}

2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761
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
}

2762 2763 2764 2765 2766 2767 2768
void DBImpl::WaitForIngestFile() {
  mutex_.AssertHeld();
  while (num_running_ingest_file_ > 0) {
    bg_cv_.Wait();
  }
}

A
agiardullo 已提交
2769 2770
#endif  // ROCKSDB_LITE

2771
}  // namespace rocksdb