db_impl.cc 174.6 KB
Newer Older
1 2 3 4 5
//  Copyright (c) 2013, Facebook, Inc.  All rights reserved.
//  This source code is licensed under the BSD-style license found in the
//  LICENSE file in the root directory of this source tree. An additional grant
//  of patent rights can be found in the PATENTS file in the same directory.
//
J
jorlow@chromium.org 已提交
6 7 8 9 10 11
// 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 已提交
12
#ifndef __STDC_FORMAT_MACROS
I
Igor Canadi 已提交
13
#define __STDC_FORMAT_MACROS
L
liuhuahang 已提交
14 15
#endif

I
Igor Canadi 已提交
16
#include <inttypes.h>
17 18
#include <stdint.h>

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

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

94
namespace rocksdb {
J
jorlow@chromium.org 已提交
95

96
const std::string kDefaultColumnFamilyName("default");
97

I
Igor Canadi 已提交
98
void DumpRocksDBBuildVersion(Logger * log);
99

S
Stanislau Hlebik 已提交
100 101
struct DBImpl::WriteContext {
  autovector<SuperVersion*> superversions_to_free_;
102
  autovector<MemTable*> memtables_to_free_;
S
Stanislau Hlebik 已提交
103 104 105 106 107

  ~WriteContext() {
    for (auto& sv : superversions_to_free_) {
      delete sv;
    }
108 109 110
    for (auto& m : memtables_to_free_) {
      delete m;
    }
S
Stanislau Hlebik 已提交
111 112 113
  }
};

J
jorlow@chromium.org 已提交
114 115 116
Options SanitizeOptions(const std::string& dbname,
                        const InternalKeyComparator* icmp,
                        const Options& src) {
117
  auto db_options = SanitizeOptions(dbname, DBOptions(src));
118
  auto cf_options = SanitizeOptions(db_options, icmp, ColumnFamilyOptions(src));
119 120 121 122 123
  return Options(db_options, cf_options);
}

DBOptions SanitizeOptions(const std::string& dbname, const DBOptions& src) {
  DBOptions result = src;
L
Lei Jin 已提交
124

125 126
  // result.max_open_files means an "infinite" open files.
  if (result.max_open_files != -1) {
I
Igor Canadi 已提交
127 128 129 130 131
    int max_max_open_files = port::GetMaxOpenFiles();
    if (max_max_open_files == -1) {
      max_max_open_files = 1000000;
    }
    ClipToRange(&result.max_open_files, 20, max_max_open_files);
132
  }
133

134
  if (result.info_log == nullptr) {
K
Kai Liu 已提交
135 136
    Status s = CreateLoggerFromOptions(dbname, result.db_log_dir, src.env,
                                       result, &result.info_log);
J
jorlow@chromium.org 已提交
137 138
    if (!s.ok()) {
      // No place suitable for logging
139
      result.info_log = nullptr;
J
jorlow@chromium.org 已提交
140 141
    }
  }
142 143 144 145
  result.env->IncBackgroundThreadsIfNeeded(src.max_background_compactions,
                                           Env::Priority::LOW);
  result.env->IncBackgroundThreadsIfNeeded(src.max_background_flushes,
                                           Env::Priority::HIGH);
146

I
Igor Canadi 已提交
147
  if (result.rate_limiter.get() != nullptr) {
148 149 150 151 152
    if (result.bytes_per_sync == 0) {
      result.bytes_per_sync = 1024 * 1024;
    }
  }

153 154 155 156
  if (result.wal_dir.empty()) {
    // Use dbname as default
    result.wal_dir = dbname;
  }
157
  if (result.wal_dir.back() == '/') {
I
Igor Canadi 已提交
158
    result.wal_dir = result.wal_dir.substr(0, result.wal_dir.size() - 1);
159
  }
160

161
  if (result.db_paths.size() == 0) {
162
    result.db_paths.emplace_back(dbname, std::numeric_limits<uint64_t>::max());
163 164
  }

165 166 167 168
  if (result.compaction_readahead_size > 0) {
    result.new_table_reader_for_compaction_inputs = true;
  }

J
jorlow@chromium.org 已提交
169 170 171
  return result;
}

172 173
namespace {

174 175
Status SanitizeOptionsByTable(
    const DBOptions& db_opts,
176 177 178
    const std::vector<ColumnFamilyDescriptor>& column_families) {
  Status s;
  for (auto cf : column_families) {
179
    s = cf.options.table_factory->SanitizeOptions(db_opts, cf.options);
180 181 182 183 184 185 186
    if (!s.ok()) {
      return s;
    }
  }
  return Status::OK();
}

187
CompressionType GetCompressionFlush(const ImmutableCFOptions& ioptions) {
188 189 190 191 192 193
  // 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.

  bool can_compress;

194
  if (ioptions.compaction_style == kCompactionStyleUniversal) {
195
    can_compress =
196
        (ioptions.compaction_options_universal.compression_size_percent < 0);
197 198
  } else {
    // For leveled compress when min_level_to_compress == 0.
199 200
    can_compress = ioptions.compression_per_level.empty() ||
                   ioptions.compression_per_level[0] != kNoCompression;
201 202 203
  }

  if (can_compress) {
204
    return ioptions.compression;
205 206 207 208
  } else {
    return kNoCompression;
  }
}
I
Igor Canadi 已提交
209

210
void DumpSupportInfo(Logger* logger) {
I
Igor Canadi 已提交
211 212 213 214 215 216 217 218
  Log(InfoLogLevel::INFO_LEVEL, logger, "Compression algorithms supported:");
  Log(InfoLogLevel::INFO_LEVEL, logger, "\tSnappy supported: %d",
      Snappy_Supported());
  Log(InfoLogLevel::INFO_LEVEL, logger, "\tZlib supported: %d",
      Zlib_Supported());
  Log(InfoLogLevel::INFO_LEVEL, logger, "\tBzip supported: %d",
      BZip2_Supported());
  Log(InfoLogLevel::INFO_LEVEL, logger, "\tLZ4 supported: %d", LZ4_Supported());
219 220
  Log(InfoLogLevel::INFO_LEVEL, logger, "Fast CRC32 supported: %d",
      crc32c::IsFastCrc32Supported());
I
Igor Canadi 已提交
221 222
}

223
}  // namespace
224

I
Igor Canadi 已提交
225
DBImpl::DBImpl(const DBOptions& options, const std::string& dbname)
J
jorlow@chromium.org 已提交
226
    : env_(options.env),
H
heyongqiang 已提交
227
      dbname_(dbname),
228 229
      db_options_(SanitizeOptions(dbname, options)),
      stats_(db_options_.statistics.get()),
230
      db_lock_(nullptr),
I
Igor Canadi 已提交
231
      mutex_(stats_, env_, DB_MUTEX_WAIT_MICROS, options.use_adaptive_mutex),
I
Igor Canadi 已提交
232
      shutting_down_(false),
J
jorlow@chromium.org 已提交
233
      bg_cv_(&mutex_),
234
      logfile_number_(0),
235
      log_dir_synced_(false),
I
Igor Canadi 已提交
236
      log_empty_(true),
237
      default_cf_handle_(nullptr),
238
      log_sync_cv_(&mutex_),
I
Igor Canadi 已提交
239 240
      total_log_size_(0),
      max_total_in_memory_state_(0),
S
sdong 已提交
241
      is_snapshot_supported_(true),
242
      write_buffer_(options.db_write_buffer_size),
S
sdong 已提交
243 244
      write_controller_(options.delayed_write_rate),
      last_batch_group_size_(0),
245 246
      unscheduled_flushes_(0),
      unscheduled_compactions_(0),
247
      bg_compaction_scheduled_(0),
248
      num_running_compactions_(0),
249
      bg_manual_only_(0),
250
      bg_flush_scheduled_(0),
251
      num_running_flushes_(0),
252
      manual_compaction_(nullptr),
253
      disable_delete_obsolete_files_(0),
I
Igor Canadi 已提交
254 255 256
      delete_obsolete_files_next_run_(
          options.env->NowMicros() +
          db_options_.delete_obsolete_files_period_micros),
257
      last_stats_dump_time_microsec_(0),
258
      next_job_id_(1),
259
      flush_on_destroy_(false),
260
      env_options_(db_options_),
I
Igor Canadi 已提交
261 262 263
#ifndef ROCKSDB_LITE
      wal_manager_(db_options_, env_options_),
#endif  // ROCKSDB_LITE
I
Igor Canadi 已提交
264
      event_logger_(db_options_.info_log.get()),
265
      bg_work_paused_(0),
266
      refitting_level_(false),
267
      opened_successfully_(false) {
H
heyongqiang 已提交
268
  env_->GetAbsolutePath(dbname, &db_absolute_path_);
269

J
jorlow@chromium.org 已提交
270
  // Reserve ten files or so for other uses and give the rest to TableCache.
271
  // Give a large number for setting of "infinite" open files.
272 273
  const int table_cache_size = (db_options_.max_open_files == -1) ?
        4194304 : db_options_.max_open_files - 10;
274
  table_cache_ =
275
      NewLRUCache(table_cache_size, db_options_.table_cache_numshardbits);
276

277
  versions_.reset(new VersionSet(dbname_, &db_options_, env_options_,
278 279
                                 table_cache_.get(), &write_buffer_,
                                 &write_controller_));
I
Igor Canadi 已提交
280 281
  column_family_memtables_.reset(new ColumnFamilyMemTablesImpl(
      versions_->GetColumnFamilySet(), &flush_scheduler_));
282

I
Igor Canadi 已提交
283
  DumpRocksDBBuildVersion(db_options_.info_log.get());
284 285
  DumpDBFileSummary(db_options_, dbname_);
  db_options_.Dump(db_options_.info_log.get());
286
  DumpSupportInfo(db_options_.info_log.get());
J
jorlow@chromium.org 已提交
287 288
}

289
// Will lock the mutex_,  will wait for completion if wait is true
290
void DBImpl::CancelAllBackgroundWork(bool wait) {
291
  InstrumentedMutexLock l(&mutex_);
292
  shutting_down_.store(true, std::memory_order_release);
293
  bg_cv_.SignalAll();
294 295 296 297
  if (!wait) {
    return;
  }
  // Wait for background work to finish
298
  while (bg_compaction_scheduled_ || bg_flush_scheduled_) {
299 300
    bg_cv_.Wait();
  }
301 302
}

J
jorlow@chromium.org 已提交
303
DBImpl::~DBImpl() {
304
  mutex_.Lock();
305

306
  if (!shutting_down_.load(std::memory_order_acquire) && flush_on_destroy_) {
307
    for (auto cfd : *versions_->GetColumnFamilySet()) {
308
      if (!cfd->IsDropped() && !cfd->mem()->IsEmpty()) {
309 310 311 312
        cfd->Ref();
        mutex_.Unlock();
        FlushMemTable(cfd, FlushOptions());
        mutex_.Lock();
313
        cfd->Unref();
314 315
      }
    }
316
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();
317
  }
318 319 320
  mutex_.Unlock();
  // CancelAllBackgroundWork called with false means we just set the shutdown
  // marker. After this we do a variant of the waiting and unschedule work
321 322
  // (to consider: moving all the waiting into CancelAllBackgroundWork(true))
  CancelAllBackgroundWork(false);
323 324 325 326 327 328 329
  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
330
  while (bg_compaction_scheduled_ || bg_flush_scheduled_) {
331 332
    bg_cv_.Wait();
  }
333
  EraseThreadStatusDbInfo();
I
Igor Canadi 已提交
334 335
  flush_scheduler_.Clear();

336 337 338 339 340 341 342 343 344 345 346 347 348
  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 已提交
349 350 351 352 353
  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();
354 355
  }

I
Igor Canadi 已提交
356 357 358 359 360 361 362 363 364 365
  // 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_) {
366
    JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
367
    FindObsoleteFiles(&job_context, true);
368 369

    mutex_.Unlock();
I
Igor Canadi 已提交
370
    // manifest number starting from 2
I
Igor Canadi 已提交
371 372 373
    job_context.manifest_file_number = 1;
    if (job_context.HaveSomethingToDelete()) {
      PurgeObsoleteFiles(job_context);
374
    }
I
Igor Canadi 已提交
375
    job_context.Clean();
376
    mutex_.Lock();
377 378
  }

379 380 381
  for (auto l : logs_to_free_) {
    delete l;
  }
382 383 384
  for (auto& log : logs_) {
    log.ClearWriter();
  }
385
  logs_.clear();
386

387
  // versions need to be destroyed before table_cache since it can hold
388 389
  // references to table_cache.
  versions_.reset();
390
  mutex_.Unlock();
I
Igor Canadi 已提交
391 392 393
  if (db_lock_ != nullptr) {
    env_->UnlockFile(db_lock_);
  }
394

395
  LogFlush(db_options_.info_log);
J
jorlow@chromium.org 已提交
396 397 398
}

Status DBImpl::NewDB() {
399
  VersionEdit new_db;
400
  new_db.SetLogNumber(0);
J
jorlow@chromium.org 已提交
401 402 403
  new_db.SetNextFile(2);
  new_db.SetLastSequence(0);

404 405
  Status s;

406 407
  Log(InfoLogLevel::INFO_LEVEL,
      db_options_.info_log, "Creating manifest 1 \n");
J
jorlow@chromium.org 已提交
408 409
  const std::string manifest = DescriptorFileName(dbname_, 1);
  {
410 411
    unique_ptr<WritableFile> file;
    EnvOptions env_options = env_->OptimizeForManifestWrite(env_options_);
S
sdong 已提交
412
    s = NewWritableFile(env_, manifest, &file, env_options);
413 414 415 416 417 418 419
    if (!s.ok()) {
      return s;
    }
    file->SetPreallocationBlockSize(db_options_.manifest_preallocation_size);
    unique_ptr<WritableFileWriter> file_writer(
        new WritableFileWriter(std::move(file), env_options));
    log::Writer log(std::move(file_writer));
J
jorlow@chromium.org 已提交
420 421 422
    std::string record;
    new_db.EncodeTo(&record);
    s = log.AddRecord(record);
S
sdong 已提交
423 424 425
    if (s.ok()) {
      s = SyncManifest(env_, &db_options_, log.file());
    }
J
jorlow@chromium.org 已提交
426 427 428
  }
  if (s.ok()) {
    // Make "CURRENT" file that points to the new manifest file.
429
    s = SetCurrentFile(env_, dbname_, 1, directories_.GetDbDir());
J
jorlow@chromium.org 已提交
430 431 432 433 434 435 436
  } else {
    env_->DeleteFile(manifest);
  }
  return s;
}

void DBImpl::MaybeIgnoreError(Status* s) const {
437
  if (s->ok() || db_options_.paranoid_checks) {
J
jorlow@chromium.org 已提交
438 439
    // No change needed
  } else {
440 441
    Log(InfoLogLevel::WARN_LEVEL,
        db_options_.info_log, "Ignoring error %s", s->ToString().c_str());
J
jorlow@chromium.org 已提交
442 443 444 445
    *s = Status::OK();
  }
}

446
const Status DBImpl::CreateArchivalDirectory() {
447 448
  if (db_options_.WAL_ttl_seconds > 0 || db_options_.WAL_size_limit_MB > 0) {
    std::string archivalPath = ArchivalDirectory(db_options_.wal_dir);
449 450 451 452 453
    return env_->CreateDirIfMissing(archivalPath);
  }
  return Status::OK();
}

454
void DBImpl::PrintStatistics() {
455
  auto dbstats = db_options_.statistics.get();
456
  if (dbstats) {
Y
Yueh-Hsuan Chiang 已提交
457
    Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
M
Mark Callaghan 已提交
458
        "STATISTICS:\n %s",
459
        dbstats->ToString().c_str());
460 461 462
  }
}

463
void DBImpl::MaybeDumpStats() {
464
  if (db_options_.stats_dump_period_sec == 0) return;
H
Haobo Xu 已提交
465 466 467 468

  const uint64_t now_micros = env_->NowMicros();

  if (last_stats_dump_time_microsec_ +
469
      db_options_.stats_dump_period_sec * 1000000
H
Haobo Xu 已提交
470 471 472 473 474 475
      <= now_micros) {
    // 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;
476

477
#ifndef ROCKSDB_LITE
478 479 480
    bool tmp1 = false;
    bool tmp2 = false;
    DBPropertyType cf_property_type =
481
        GetPropertyType(DB::Properties::kCFStats, &tmp1, &tmp2);
482
    DBPropertyType db_property_type =
483
        GetPropertyType(DB::Properties::kDBStats, &tmp1, &tmp2);
H
Haobo Xu 已提交
484
    std::string stats;
485
    {
486
      InstrumentedMutexLock l(&mutex_);
487
      for (auto cfd : *versions_->GetColumnFamilySet()) {
488
        cfd->internal_stats()->GetStringProperty(cf_property_type,
489 490
                                                 DB::Properties::kCFStats,
                                                 &stats);
491
      }
492
      default_cf_internal_stats_->GetStringProperty(db_property_type,
493 494
                                                    DB::Properties::kDBStats,
                                                    &stats);
495
    }
Y
Yueh-Hsuan Chiang 已提交
496
    Log(InfoLogLevel::WARN_LEVEL,
497
        db_options_.info_log, "------- DUMPING STATS -------");
Y
Yueh-Hsuan Chiang 已提交
498
    Log(InfoLogLevel::WARN_LEVEL,
499
        db_options_.info_log, "%s", stats.c_str());
500
#endif  // !ROCKSDB_LITE
501

502
    PrintStatistics();
503 504 505
  }
}

I
Igor Canadi 已提交
506
// * Returns the list of live files in 'sst_live'
I
Igor Canadi 已提交
507
// If it's doing full scan:
I
Igor Canadi 已提交
508 509
// * Returns the list of all files in the filesystem in
// 'full_scan_candidate_files'.
I
Igor Canadi 已提交
510
// Otherwise, gets obsolete files from VersionSet.
I
Igor Canadi 已提交
511 512
// no_full_scan = true -- never do the full scan using GetChildren()
// force = false -- don't force the full scan, except every
513
//  db_options_.delete_obsolete_files_period_micros
I
Igor Canadi 已提交
514
// force = true -- force the full scan
I
Igor Canadi 已提交
515
void DBImpl::FindObsoleteFiles(JobContext* job_context, bool force,
I
Igor Canadi 已提交
516
                               bool no_full_scan) {
D
Dhruba Borthakur 已提交
517 518
  mutex_.AssertHeld();

519
  // if deletion is disabled, do nothing
520
  if (disable_delete_obsolete_files_ > 0) {
521 522 523
    return;
  }

524 525 526 527 528
  bool doing_the_full_scan = false;

  // logic for figurint out if we're doing the full scan
  if (no_full_scan) {
    doing_the_full_scan = false;
529
  } else if (force || db_options_.delete_obsolete_files_period_micros == 0) {
530 531 532
    doing_the_full_scan = true;
  } else {
    const uint64_t now_micros = env_->NowMicros();
I
Igor Canadi 已提交
533
    if (delete_obsolete_files_next_run_ < now_micros) {
534
      doing_the_full_scan = true;
I
Igor Canadi 已提交
535 536
      delete_obsolete_files_next_run_ =
          now_micros + db_options_.delete_obsolete_files_period_micros;
537 538 539
    }
  }

I
Igor Canadi 已提交
540 541
  // don't delete files that might be currently written to from compaction
  // threads
542 543 544
  // Since job_context->min_pending_output is set, until file scan finishes,
  // mutex_ cannot be released. Otherwise, we might see no min_pending_output
  // here but later find newer generated unfinalized files while scannint.
I
Igor Canadi 已提交
545 546 547 548 549 550 551
  if (!pending_outputs_.empty()) {
    job_context->min_pending_output = *pending_outputs_.begin();
  } else {
    // delete all of them
    job_context->min_pending_output = std::numeric_limits<uint64_t>::max();
  }

552 553
  // Get obsolete files.  This function will also update the list of
  // pending files in VersionSet().
I
Igor Canadi 已提交
554 555
  versions_->GetObsoleteFiles(&job_context->sst_delete_files,
                              job_context->min_pending_output);
I
Igor Canadi 已提交
556

I
Igor Canadi 已提交
557
  // store the current filenum, lognum, etc
558
  job_context->manifest_file_number = versions_->manifest_file_number();
I
Igor Canadi 已提交
559
  job_context->pending_manifest_file_number =
560
      versions_->pending_manifest_file_number();
I
Igor Canadi 已提交
561
  job_context->log_number = versions_->MinLogNumber();
562
  job_context->prev_log_number = versions_->prev_log_number();
I
Igor Canadi 已提交
563

I
Igor Canadi 已提交
564
  versions_->AddLiveFiles(&job_context->sst_live);
565
  if (doing_the_full_scan) {
I
Igor Canadi 已提交
566 567
    for (uint32_t path_id = 0; path_id < db_options_.db_paths.size();
         path_id++) {
568 569 570
      // set of all files in the directory. We'll exclude files that are still
      // alive in the subsequent processings.
      std::vector<std::string> files;
571
      env_->GetChildren(db_options_.db_paths[path_id].path,
572
                        &files);  // Ignore errors
573
      for (std::string file : files) {
I
Igor Canadi 已提交
574
        // TODO(icanadi) clean up this mess to avoid having one-off "/" prefixes
I
Igor Canadi 已提交
575 576
        job_context->full_scan_candidate_files.emplace_back("/" + file,
                                                            path_id);
577 578
      }
    }
579 580

    //Add log files in wal_dir
581
    if (db_options_.wal_dir != dbname_) {
582
      std::vector<std::string> log_files;
583
      env_->GetChildren(db_options_.wal_dir, &log_files);  // Ignore errors
584
      for (std::string log_file : log_files) {
I
Igor Canadi 已提交
585
        job_context->full_scan_candidate_files.emplace_back(log_file, 0);
586 587
      }
    }
588
    // Add info log files in db_log_dir
589
    if (!db_options_.db_log_dir.empty() && db_options_.db_log_dir != dbname_) {
590
      std::vector<std::string> info_log_files;
591 592
      // Ignore errors
      env_->GetChildren(db_options_.db_log_dir, &info_log_files);
593
      for (std::string log_file : info_log_files) {
I
Igor Canadi 已提交
594
        job_context->full_scan_candidate_files.emplace_back(log_file, 0);
595 596
      }
    }
597
  }
598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628

  if (!alive_log_files_.empty()) {
    uint64_t min_log_number = versions_->MinLogNumber();
    // find newly obsoleted log files
    while (alive_log_files_.begin()->number < min_log_number) {
      auto& earliest = *alive_log_files_.begin();
      job_context->log_delete_files.push_back(earliest.number);
      total_log_size_ -= earliest.size;
      alive_log_files_.pop_front();
      // Current log should always stay alive since it can't have
      // number < MinLogNumber().
      assert(alive_log_files_.size());
    }
    while (!logs_.empty() && logs_.front().number < min_log_number) {
      auto& log = logs_.front();
      if (log.getting_synced) {
        log_sync_cv_.Wait();
        // logs_ could have changed while we were waiting.
        continue;
      }
      logs_to_free_.push_back(log.ReleaseWriter());
      logs_.pop_front();
    }
    // Current log cannot be obsolete.
    assert(!logs_.empty());
  }

  // We're just cleaning up for DB::Write().
  assert(job_context->logs_to_free.empty());
  job_context->logs_to_free = logs_to_free_;
  logs_to_free_.clear();
629 630
}

631
namespace {
I
Igor Canadi 已提交
632 633
bool CompareCandidateFile(const JobContext::CandidateFileInfo& first,
                          const JobContext::CandidateFileInfo& second) {
634 635 636 637 638
  if (first.file_name > second.file_name) {
    return true;
  } else if (first.file_name < second.file_name) {
    return false;
  } else {
639
    return (first.path_id > second.path_id);
640 641 642 643
  }
}
};  // namespace

D
Dhruba Borthakur 已提交
644
// Diffs the files listed in filenames and those that do not
I
Igor Canadi 已提交
645
// belong to live files are posibly removed. Also, removes all the
646
// files in sst_delete_files and log_delete_files.
647
// It is not necessary to hold the mutex when invoking this method.
I
Igor Canadi 已提交
648
void DBImpl::PurgeObsoleteFiles(const JobContext& state) {
649 650
  // we'd better have sth to delete
  assert(state.HaveSomethingToDelete());
651

I
Igor Canadi 已提交
652 653 654 655
  // this checks if FindObsoleteFiles() was run before. If not, don't do
  // PurgeObsoleteFiles(). If FindObsoleteFiles() was run, we need to also
  // run PurgeObsoleteFiles(), even if disable_delete_obsolete_files_ is true
  if (state.manifest_file_number == 0) {
I
Igor Canadi 已提交
656 657
    return;
  }
658

659
  // Now, convert live list to an unordered map, WITHOUT mutex held;
660
  // set is slow.
661
  std::unordered_map<uint64_t, const FileDescriptor*> sst_live_map;
I
Igor Canadi 已提交
662
  for (const FileDescriptor& fd : state.sst_live) {
663 664
    sst_live_map[fd.GetNumber()] = &fd;
  }
I
Igor Canadi 已提交
665

I
Igor Canadi 已提交
666
  auto candidate_files = state.full_scan_candidate_files;
I
Igor Canadi 已提交
667 668 669
  candidate_files.reserve(candidate_files.size() +
                          state.sst_delete_files.size() +
                          state.log_delete_files.size());
K
kailiu 已提交
670 671
  // We may ignore the dbname when generating the file names.
  const char* kDumbDbName = "";
672
  for (auto file : state.sst_delete_files) {
673 674 675
    candidate_files.emplace_back(
        MakeTableFileName(kDumbDbName, file->fd.GetNumber()),
        file->fd.GetPathId());
676
    delete file;
I
Igor Canadi 已提交
677 678
  }

K
kailiu 已提交
679 680
  for (auto file_num : state.log_delete_files) {
    if (file_num > 0) {
681 682
      candidate_files.emplace_back(LogFileName(kDumbDbName, file_num).substr(1),
                                   0);
I
Igor Canadi 已提交
683 684
    }
  }
685

K
kailiu 已提交
686
  // dedup state.candidate_files so we don't try to delete the same
I
Igor Canadi 已提交
687
  // file twice
688
  sort(candidate_files.begin(), candidate_files.end(), CompareCandidateFile);
689 690
  candidate_files.erase(unique(candidate_files.begin(), candidate_files.end()),
                        candidate_files.end());
J
jorlow@chromium.org 已提交
691

692
  std::vector<std::string> old_info_log_files;
693
  InfoLogPrefix info_log_prefix(!db_options_.db_log_dir.empty(), dbname_);
694 695 696
  for (const auto& candidate_file : candidate_files) {
    std::string to_delete = candidate_file.file_name;
    uint32_t path_id = candidate_file.path_id;
K
kailiu 已提交
697 698 699
    uint64_t number;
    FileType type;
    // Ignore file if we cannot recognize it.
700
    if (!ParseFileName(to_delete, &number, info_log_prefix.prefix, &type)) {
K
kailiu 已提交
701 702
      continue;
    }
J
jorlow@chromium.org 已提交
703

K
kailiu 已提交
704 705 706 707 708 709 710 711
    bool keep = true;
    switch (type) {
      case kLogFile:
        keep = ((number >= state.log_number) ||
                (number == state.prev_log_number));
        break;
      case kDescriptorFile:
        // Keep my manifest file, and any newer incarnations'
712
        // (can happen during manifest roll)
K
kailiu 已提交
713 714 715
        keep = (number >= state.manifest_file_number);
        break;
      case kTableFile:
I
Igor Canadi 已提交
716 717 718 719
        // If the second condition is not there, this makes
        // DontDeletePendingOutputs fail
        keep = (sst_live_map.find(number) != sst_live_map.end()) ||
               number >= state.min_pending_output;
K
kailiu 已提交
720 721 722
        break;
      case kTempFile:
        // Any temp files that are currently being written to must
723 724 725 726
        // be recorded in pending_outputs_, which is inserted into "live".
        // Also, SetCurrentFile creates a temp file when writing out new
        // manifest, which is equal to state.pending_manifest_file_number. We
        // should not delete that file
727
        keep = (sst_live_map.find(number) != sst_live_map.end()) ||
728
               (number == state.pending_manifest_file_number);
K
kailiu 已提交
729 730 731 732
        break;
      case kInfoLogFile:
        keep = true;
        if (number != 0) {
733
          old_info_log_files.push_back(to_delete);
J
jorlow@chromium.org 已提交
734
        }
K
kailiu 已提交
735 736 737 738 739 740 741 742 743 744 745 746 747
        break;
      case kCurrentFile:
      case kDBLockFile:
      case kIdentityFile:
      case kMetaDatabase:
        keep = true;
        break;
    }

    if (keep) {
      continue;
    }

748
    std::string fname;
K
kailiu 已提交
749 750
    if (type == kTableFile) {
      // evict from cache
751
      TableCache::Evict(table_cache_.get(), number);
752
      fname = TableFileName(db_options_.db_paths, number, path_id);
753
    } else {
754 755
      fname = ((type == kLogFile) ?
          db_options_.wal_dir : dbname_) + "/" + to_delete;
K
kailiu 已提交
756
    }
757

758
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
759
    if (type == kLogFile && (db_options_.WAL_ttl_seconds > 0 ||
760
                              db_options_.WAL_size_limit_MB > 0)) {
I
Igor Canadi 已提交
761
      wal_manager_.ArchiveWALFile(fname, number);
762 763 764
      continue;
    }
#endif  // !ROCKSDB_LITE
I
Islam AbdelRahman 已提交
765
    Status file_deletion_status;
766 767
    if (type == kTableFile && path_id == 0) {
      file_deletion_status = DeleteOrMoveToTrash(&db_options_, fname);
I
Islam AbdelRahman 已提交
768 769 770
    } else {
      file_deletion_status = env_->DeleteFile(fname);
    }
771
    if (file_deletion_status.ok()) {
772
      Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
773
          "[JOB %d] Delete %s type=%d #%" PRIu64 " -- %s\n", state.job_id,
774 775
          fname.c_str(), type, number,
          file_deletion_status.ToString().c_str());
776 777 778 779 780 781
    } else if (env_->FileExists(fname).IsNotFound()) {
      Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
          "[JOB %d] Tried to delete a non-existing file %s type=%d #%" PRIu64
          " -- %s\n",
          state.job_id, fname.c_str(), type, number,
          file_deletion_status.ToString().c_str());
782 783 784 785 786 787 788
    } else {
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "[JOB %d] Failed to delete %s type=%d #%" PRIu64 " -- %s\n",
          state.job_id, fname.c_str(), type, number,
          file_deletion_status.ToString().c_str());
    }
    if (type == kTableFile) {
789 790 791 792
      EventHelpers::LogAndNotifyTableFileDeletion(
          &event_logger_, state.job_id, number, fname,
          file_deletion_status, GetName(),
          db_options_.listeners);
J
jorlow@chromium.org 已提交
793 794
    }
  }
H
heyongqiang 已提交
795

796
  // Delete old info log files.
797
  size_t old_info_log_file_count = old_info_log_files.size();
798
  if (old_info_log_file_count >= db_options_.keep_log_file_num) {
799
    std::sort(old_info_log_files.begin(), old_info_log_files.end());
800
    size_t end = old_info_log_file_count - db_options_.keep_log_file_num;
801
    for (unsigned int i = 0; i <= end; i++) {
802
      std::string& to_delete = old_info_log_files.at(i);
803 804
      std::string full_path_to_delete = (db_options_.db_log_dir.empty() ?
           dbname_ : db_options_.db_log_dir) + "/" + to_delete;
805
      Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
806
          "[JOB %d] Delete info log file %s\n", state.job_id,
807 808
          full_path_to_delete.c_str());
      Status s = env_->DeleteFile(full_path_to_delete);
809
      if (!s.ok()) {
810 811 812 813 814 815 816 817 818 819
        if (env_->FileExists(full_path_to_delete).IsNotFound()) {
          Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
              "[JOB %d] Tried to delete non-existing info log file %s FAILED "
              "-- %s\n",
              state.job_id, to_delete.c_str(), s.ToString().c_str());
        } else {
          Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
              "[JOB %d] Delete info log file %s FAILED -- %s\n", state.job_id,
              to_delete.c_str(), s.ToString().c_str());
        }
820
      }
H
heyongqiang 已提交
821 822
    }
  }
I
Igor Canadi 已提交
823 824 825
#ifndef ROCKSDB_LITE
  wal_manager_.PurgeObsoleteWALFiles();
#endif  // ROCKSDB_LITE
826
  LogFlush(db_options_.info_log);
D
Dhruba Borthakur 已提交
827 828 829 830
}

void DBImpl::DeleteObsoleteFiles() {
  mutex_.AssertHeld();
831
  JobContext job_context(next_job_id_.fetch_add(1));
I
Igor Canadi 已提交
832
  FindObsoleteFiles(&job_context, true);
833 834

  mutex_.Unlock();
I
Igor Canadi 已提交
835 836
  if (job_context.HaveSomethingToDelete()) {
    PurgeObsoleteFiles(job_context);
837
  }
I
Igor Canadi 已提交
838
  job_context.Clean();
839
  mutex_.Lock();
840 841
}

842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857
Status DBImpl::Directories::CreateAndNewDirectory(
    Env* env, const std::string& dirname,
    std::unique_ptr<Directory>* directory) const {
  // We call CreateDirIfMissing() as the directory may already exist (if we
  // are reopening a DB), when this happens we don't want creating the
  // directory to cause an error. However, we need to check if creating the
  // directory fails or else we may get an obscure message about the lock
  // file not existing. One real-world example of this occurring is if
  // env->CreateDirIfMissing() doesn't create intermediate directories, e.g.
  // when dbname_ is "dir/db" but when "dir" doesn't exist.
  Status s = env->CreateDirIfMissing(dirname);
  if (!s.ok()) {
    return s;
  }
  return env->NewDirectory(dirname, directory);
}
J
jorlow@chromium.org 已提交
858

859 860 861 862 863 864 865 866 867
Status DBImpl::Directories::SetDirectories(
    Env* env, const std::string& dbname, const std::string& wal_dir,
    const std::vector<DbPath>& data_paths) {
  Status s = CreateAndNewDirectory(env, dbname, &db_dir_);
  if (!s.ok()) {
    return s;
  }
  if (!wal_dir.empty() && dbname != wal_dir) {
    s = CreateAndNewDirectory(env, wal_dir, &wal_dir_);
868 869 870
    if (!s.ok()) {
      return s;
    }
871
  }
872

873 874 875 876 877 878 879 880
  data_dirs_.clear();
  for (auto& p : data_paths) {
    const std::string db_path = p.path;
    if (db_path == dbname) {
      data_dirs_.emplace_back(nullptr);
    } else {
      std::unique_ptr<Directory> path_directory;
      s = CreateAndNewDirectory(env, db_path, &path_directory);
881 882 883
      if (!s.ok()) {
        return s;
      }
884
      data_dirs_.emplace_back(path_directory.release());
885
    }
886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904
  }
  assert(data_dirs_.size() == data_paths.size());
  return Status::OK();
}

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

Status DBImpl::Recover(
    const std::vector<ColumnFamilyDescriptor>& column_families, bool read_only,
    bool error_if_log_file_exist) {
  mutex_.AssertHeld();
905

906 907 908 909 910
  bool is_new_db = false;
  assert(db_lock_ == nullptr);
  if (!read_only) {
    Status s = directories_.SetDirectories(env_, dbname_, db_options_.wal_dir,
                                           db_options_.db_paths);
911 912 913 914
    if (!s.ok()) {
      return s;
    }

915
    s = env_->LockFile(LockFileName(dbname_), &db_lock_);
916 917 918
    if (!s.ok()) {
      return s;
    }
J
jorlow@chromium.org 已提交
919

A
agiardullo 已提交
920 921
    s = env_->FileExists(CurrentFileName(dbname_));
    if (s.IsNotFound()) {
922
      if (db_options_.create_if_missing) {
923
        s = NewDB();
924
        is_new_db = true;
925 926 927 928 929 930
        if (!s.ok()) {
          return s;
        }
      } else {
        return Status::InvalidArgument(
            dbname_, "does not exist (create_if_missing is false)");
J
jorlow@chromium.org 已提交
931
      }
A
agiardullo 已提交
932
    } else if (s.ok()) {
933
      if (db_options_.error_if_exists) {
934 935 936
        return Status::InvalidArgument(
            dbname_, "exists (error_if_exists is true)");
      }
A
agiardullo 已提交
937 938 939 940
    } else {
      // Unexpected error reading file
      assert(s.IsIOError());
      return s;
J
jorlow@chromium.org 已提交
941
    }
M
Mayank Agarwal 已提交
942
    // Check for the IDENTITY file and create it if not there
A
agiardullo 已提交
943 944
    s = env_->FileExists(IdentityFileName(dbname_));
    if (s.IsNotFound()) {
M
Mayank Agarwal 已提交
945 946 947 948
      s = SetIdentityFile(env_, dbname_);
      if (!s.ok()) {
        return s;
      }
A
agiardullo 已提交
949 950 951
    } else if (!s.ok()) {
      assert(s.IsIOError());
      return s;
M
Mayank Agarwal 已提交
952
    }
J
jorlow@chromium.org 已提交
953 954
  }

955
  Status s = versions_->Recover(column_families, read_only);
956
  if (db_options_.paranoid_checks && s.ok()) {
I
Igor Canadi 已提交
957 958
    s = CheckConsistency();
  }
J
jorlow@chromium.org 已提交
959
  if (s.ok()) {
A
agiardullo 已提交
960
    SequenceNumber max_sequence(kMaxSequenceNumber);
961 962
    default_cf_handle_ = new ColumnFamilyHandleImpl(
        versions_->GetColumnFamilySet()->GetDefault(), this, &mutex_);
963
    default_cf_internal_stats_ = default_cf_handle_->cfd()->internal_stats();
964 965
    single_column_family_mode_ =
        versions_->GetColumnFamilySet()->NumberOfColumnFamilies() == 1;
966 967 968 969 970

    // Recover from all newer log files than the ones named in the
    // descriptor (new log files may have been added by the previous
    // incarnation without registering them in the descriptor).
    //
971
    // Note that prev_log_number() is no longer used, but we pay
972
    // attention to it in case we are recovering a database
973
    // produced by an older version of rocksdb.
974
    const uint64_t min_log = versions_->MinLogNumber();
975
    const uint64_t prev_log = versions_->prev_log_number();
976
    std::vector<std::string> filenames;
977
    s = env_->GetChildren(db_options_.wal_dir, &filenames);
978 979
    if (!s.ok()) {
      return s;
980
    }
K
kailiu 已提交
981

982 983
    std::vector<uint64_t> logs;
    for (size_t i = 0; i < filenames.size(); i++) {
K
kailiu 已提交
984 985
      uint64_t number;
      FileType type;
986 987 988 989 990 991 992 993 994
      if (ParseFileName(filenames[i], &number, &type) && type == kLogFile) {
        if (is_new_db) {
          return Status::Corruption(
              "While creating a new Db, wal_dir contains "
              "existing log file: ",
              filenames[i]);
        } else if ((number >= min_log) || (number == prev_log)) {
          logs.push_back(number);
        }
995
      }
J
jorlow@chromium.org 已提交
996
    }
997

H
heyongqiang 已提交
998 999 1000 1001 1002 1003
    if (logs.size() > 0 && error_if_log_file_exist) {
      return Status::Corruption(""
          "The db was opened in readonly mode with error_if_log_file_exist"
          "flag but a log file already exists");
    }

S
Stanislau Hlebik 已提交
1004 1005 1006 1007 1008 1009 1010
    if (!logs.empty()) {
      // Recover in the order in which the logs were generated
      std::sort(logs.begin(), logs.end());
      s = RecoverLogFiles(logs, &max_sequence, read_only);
      if (!s.ok()) {
        // Clear memtables if recovery failed
        for (auto cfd : *versions_->GetColumnFamilySet()) {
A
agiardullo 已提交
1011 1012
          cfd->CreateNewMemtable(*cfd->GetLatestMutableCFOptions(),
                                 kMaxSequenceNumber);
S
Stanislau Hlebik 已提交
1013 1014
        }
      }
1015
    }
L
Lei Jin 已提交
1016
    SetTickerCount(stats_, SEQUENCE_NUMBER, versions_->LastSequence());
J
jorlow@chromium.org 已提交
1017 1018
  }

L
Lei Jin 已提交
1019 1020
  // Initial value
  max_total_in_memory_state_ = 0;
I
Igor Canadi 已提交
1021
  for (auto cfd : *versions_->GetColumnFamilySet()) {
L
Lei Jin 已提交
1022 1023 1024
    auto* mutable_cf_options = cfd->GetLatestMutableCFOptions();
    max_total_in_memory_state_ += mutable_cf_options->write_buffer_size *
                                  mutable_cf_options->max_write_buffer_number;
I
Igor Canadi 已提交
1025 1026
  }

J
jorlow@chromium.org 已提交
1027 1028 1029
  return s;
}

S
Stanislau Hlebik 已提交
1030 1031 1032
// REQUIRES: log_numbers are sorted in ascending order
Status DBImpl::RecoverLogFiles(const std::vector<uint64_t>& log_numbers,
                               SequenceNumber* max_sequence, bool read_only) {
J
jorlow@chromium.org 已提交
1033 1034
  struct LogReporter : public log::Reader::Reporter {
    Env* env;
1035
    Logger* info_log;
J
jorlow@chromium.org 已提交
1036
    const char* fname;
1037
    Status* status;  // nullptr if db_options_.paranoid_checks==false
I
Igor Sugak 已提交
1038
    virtual void Corruption(size_t bytes, const Status& s) override {
1039 1040
      Log(InfoLogLevel::WARN_LEVEL,
          info_log, "%s%s: dropping %d bytes; %s",
1041
          (this->status == nullptr ? "(ignoring error) " : ""),
J
jorlow@chromium.org 已提交
1042
          fname, static_cast<int>(bytes), s.ToString().c_str());
K
krad 已提交
1043 1044 1045
      if (this->status != nullptr && this->status->ok()) {
        *this->status = s;
      }
J
jorlow@chromium.org 已提交
1046 1047 1048 1049
    }
  };

  mutex_.AssertHeld();
S
Stanislau Hlebik 已提交
1050
  Status status;
1051
  std::unordered_map<int, VersionEdit> version_edits;
1052
  // no need to refcount because iteration is under mutex
1053 1054
  for (auto cfd : *versions_->GetColumnFamilySet()) {
    VersionEdit edit;
1055 1056
    edit.SetColumnFamily(cfd->GetID());
    version_edits.insert({cfd->GetID(), edit});
1057
  }
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
  int job_id = next_job_id_.fetch_add(1);
  {
    auto stream = event_logger_.Log();
    stream << "job" << job_id << "event"
           << "recovery_started";
    stream << "log_files";
    stream.StartArray();
    for (auto log_number : log_numbers) {
      stream << log_number;
    }
    stream.EndArray();
  }
I
Igor Canadi 已提交
1070

K
krad 已提交
1071
  bool continue_replay_log = true;
S
Stanislau Hlebik 已提交
1072 1073 1074 1075
  for (auto log_number : log_numbers) {
    // The previous incarnation may not have written any MANIFEST
    // records after allocating this log number.  So we manually
    // update the file number allocation counter in VersionSet.
1076
    versions_->MarkFileNumberUsedDuringRecovery(log_number);
S
Stanislau Hlebik 已提交
1077 1078
    // Open the log file
    std::string fname = LogFileName(db_options_.wal_dir, log_number);
1079 1080 1081 1082
    unique_ptr<SequentialFileReader> file_reader;
    {
      unique_ptr<SequentialFile> file;
      status = env_->NewSequentialFile(fname, &file, env_options_);
S
Stanislau Hlebik 已提交
1083
      if (!status.ok()) {
1084 1085 1086 1087 1088 1089 1090 1091
        MaybeIgnoreError(&status);
        if (!status.ok()) {
          return status;
        } else {
          // Fail with one log file, but that's ok.
          // Try next one.
          continue;
        }
S
Stanislau Hlebik 已提交
1092
      }
1093
      file_reader.reset(new SequentialFileReader(std::move(file)));
J
jorlow@chromium.org 已提交
1094 1095
    }

S
Stanislau Hlebik 已提交
1096 1097 1098 1099 1100
    // Create the log reader.
    LogReporter reporter;
    reporter.env = env_;
    reporter.info_log = db_options_.info_log.get();
    reporter.fname = fname.c_str();
K
krad 已提交
1101 1102 1103 1104 1105 1106 1107
    if (!db_options_.paranoid_checks ||
        db_options_.wal_recovery_mode ==
            WALRecoveryMode::kSkipAnyCorruptedRecords) {
      reporter.status = nullptr;
    } else {
      reporter.status = &status;
    }
S
Stanislau Hlebik 已提交
1108 1109 1110 1111
    // We intentially make log::Reader do checksumming even if
    // paranoid_checks==false so that corruptions cause entire commits
    // to be skipped instead of propagating bad information (like overly
    // large sequence numbers).
1112
    log::Reader reader(std::move(file_reader), &reporter, true /*checksum*/,
S
Stanislau Hlebik 已提交
1113
                       0 /*initial_offset*/);
K
krad 已提交
1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "Recovering log #%" PRIu64 " mode %d skip-recovery %d", log_number,
        db_options_.wal_recovery_mode, !continue_replay_log);

    // Determine if we should tolerate incomplete records at the tail end of the
    // log
    bool report_eof_inconsistency;
    if (db_options_.wal_recovery_mode ==
        WALRecoveryMode::kAbsoluteConsistency) {
      // in clean shutdown we don't expect any error in the log files
      report_eof_inconsistency = true;
    } else {
      // for other modes ignore only incomplete records in the last log file
      // which is presumably due to write in progress during restart
      report_eof_inconsistency = false;

      // TODO krad: Evaluate if we need to move to a more strict mode where we
      // restrict the inconsistency to only the last log
    }
S
Stanislau Hlebik 已提交
1133 1134 1135 1136 1137

    // Read all the records and add to a memtable
    std::string scratch;
    Slice record;
    WriteBatch batch;
K
krad 已提交
1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150

    if (!continue_replay_log) {
      uint64_t bytes;
      if (env_->GetFileSize(fname, &bytes).ok()) {
        auto info_log = db_options_.info_log.get();
        Log(InfoLogLevel::WARN_LEVEL, info_log, "%s: dropping %d bytes",
            fname.c_str(), static_cast<int>(bytes));
      }
    }

    while (continue_replay_log &&
           reader.ReadRecord(&record, &scratch, report_eof_inconsistency) &&
           status.ok()) {
S
Stanislau Hlebik 已提交
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167
      if (record.size() < 12) {
        reporter.Corruption(record.size(),
                            Status::Corruption("log record too small"));
        continue;
      }
      WriteBatchInternal::SetContents(&batch, record);

      // If column family was not found, it might mean that the WAL write
      // batch references to the column family that was dropped after the
      // insert. We don't want to fail the whole write batch in that case --
      // we just ignore the update.
      // That's why we set ignore missing column families to true
      status = WriteBatchInternal::InsertInto(
          &batch, column_family_memtables_.get(), true, log_number);

      MaybeIgnoreError(&status);
      if (!status.ok()) {
1168 1169 1170 1171
        // We are treating this as a failure while reading since we read valid
        // blocks that do not form coherent data
        reporter.Corruption(record.size(), status);
        continue;
S
Stanislau Hlebik 已提交
1172
      }
1173

S
Stanislau Hlebik 已提交
1174 1175
      const SequenceNumber last_seq = WriteBatchInternal::Sequence(&batch) +
                                      WriteBatchInternal::Count(&batch) - 1;
A
agiardullo 已提交
1176
      if ((*max_sequence == kMaxSequenceNumber) || (last_seq > *max_sequence)) {
S
Stanislau Hlebik 已提交
1177 1178 1179 1180
        *max_sequence = last_seq;
      }

      if (!read_only) {
I
Igor Canadi 已提交
1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192
        // we can do this because this is called before client has access to the
        // DB and there is only a single thread operating on DB
        ColumnFamilyData* cfd;

        while ((cfd = flush_scheduler_.GetNextColumnFamily()) != nullptr) {
          cfd->Unref();
          // If this asserts, it means that InsertInto failed in
          // filtering updates to already-flushed column families
          assert(cfd->GetLogNumber() <= log_number);
          auto iter = version_edits.find(cfd->GetID());
          assert(iter != version_edits.end());
          VersionEdit* edit = &iter->second;
1193
          status = WriteLevel0TableForRecovery(job_id, cfd, cfd->mem(), edit);
I
Igor Canadi 已提交
1194 1195 1196 1197
          if (!status.ok()) {
            // Reflect errors immediately so that conditions like full
            // file-systems cause the DB::Open() to fail.
            return status;
1198
          }
A
agiardullo 已提交
1199 1200 1201

          cfd->CreateNewMemtable(*cfd->GetLatestMutableCFOptions(),
                                 *max_sequence);
1202
        }
J
jorlow@chromium.org 已提交
1203 1204 1205
      }
    }

1206
    if (!status.ok()) {
K
krad 已提交
1207
      if (db_options_.wal_recovery_mode ==
1208 1209 1210 1211
             WALRecoveryMode::kSkipAnyCorruptedRecords) {
        // We should ignore all errors unconditionally
        status = Status::OK();
      } else if (db_options_.wal_recovery_mode ==
K
krad 已提交
1212 1213 1214 1215 1216 1217 1218 1219
                 WALRecoveryMode::kPointInTimeRecovery) {
        // We should ignore the error but not continue replaying
        status = Status::OK();
        continue_replay_log = false;

        Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
            "Point in time recovered to log #%" PRIu64 " seq #%" PRIu64,
            log_number, *max_sequence);
1220 1221 1222 1223 1224
      } else {
        assert(db_options_.wal_recovery_mode ==
                  WALRecoveryMode::kTolerateCorruptedTailRecords
               || db_options_.wal_recovery_mode ==
                  WALRecoveryMode::kAbsoluteConsistency);
K
krad 已提交
1225 1226
        return status;
      }
1227 1228
    }

I
Igor Canadi 已提交
1229
    flush_scheduler_.Clear();
A
agiardullo 已提交
1230 1231
    if ((*max_sequence != kMaxSequenceNumber) &&
        (versions_->LastSequence() < *max_sequence)) {
S
Stanislau Hlebik 已提交
1232 1233
      versions_->SetLastSequence(*max_sequence);
    }
1234 1235
  }

1236
  if (!read_only) {
1237 1238
    // no need to refcount since client still doesn't have access
    // to the DB and can not drop column families while we iterate
S
Stanislau Hlebik 已提交
1239
    auto max_log_number = log_numbers.back();
1240
    for (auto cfd : *versions_->GetColumnFamilySet()) {
1241
      auto iter = version_edits.find(cfd->GetID());
1242 1243 1244
      assert(iter != version_edits.end());
      VersionEdit* edit = &iter->second;

S
Stanislau Hlebik 已提交
1245
      if (cfd->GetLogNumber() > max_log_number) {
1246
        // Column family cfd has already flushed the data
S
Stanislau Hlebik 已提交
1247
        // from all logs. Memtable has to be empty because
1248
        // we filter the updates based on log_number
1249
        // (in WriteBatch::InsertInto)
1250 1251 1252 1253 1254 1255 1256
        assert(cfd->mem()->GetFirstSequenceNumber() == 0);
        assert(edit->NumEntries() == 0);
        continue;
      }

      // flush the final memtable (if non-empty)
      if (cfd->mem()->GetFirstSequenceNumber() != 0) {
1257
        status = WriteLevel0TableForRecovery(job_id, cfd, cfd->mem(), edit);
S
Stanislau Hlebik 已提交
1258 1259 1260 1261
        if (!status.ok()) {
          // Recovery failed
          break;
        }
A
agiardullo 已提交
1262 1263 1264

        cfd->CreateNewMemtable(*cfd->GetLatestMutableCFOptions(),
                               *max_sequence);
1265
      }
J
jorlow@chromium.org 已提交
1266

1267
      // write MANIFEST with update
S
Stanislau Hlebik 已提交
1268
      // writing log_number in the manifest means that any log file
1269 1270
      // with number strongly less than (log_number + 1) is already
      // recovered and should be ignored on next reincarnation.
S
Stanislau Hlebik 已提交
1271 1272 1273
      // Since we already recovered max_log_number, we want all logs
      // with numbers `<= max_log_number` (includes this one) to be ignored
      edit->SetLogNumber(max_log_number + 1);
1274 1275 1276
      // we must mark the next log number as used, even though it's
      // not actually used. that is because VersionSet assumes
      // VersionSet::next_file_number_ always to be strictly greater than any
I
Igor Canadi 已提交
1277
      // log number
1278
      versions_->MarkFileNumberUsedDuringRecovery(max_log_number + 1);
1279 1280
      status = versions_->LogAndApply(
          cfd, *cfd->GetLatestMutableCFOptions(), edit, &mutex_);
1281
      if (!status.ok()) {
S
Stanislau Hlebik 已提交
1282 1283
        // Recovery failed
        break;
1284 1285
      }
    }
1286
  }
I
Igor Canadi 已提交
1287

1288 1289 1290
  event_logger_.Log() << "job" << job_id << "event"
                      << "recovery_finished";

J
jorlow@chromium.org 已提交
1291 1292 1293
  return status;
}

1294 1295
Status DBImpl::WriteLevel0TableForRecovery(int job_id, ColumnFamilyData* cfd,
                                           MemTable* mem, VersionEdit* edit) {
J
jorlow@chromium.org 已提交
1296
  mutex_.AssertHeld();
1297
  const uint64_t start_micros = env_->NowMicros();
J
jorlow@chromium.org 已提交
1298
  FileMetaData meta;
1299
  meta.fd = FileDescriptor(versions_->NewFileNumber(), 0, 0);
I
Igor Canadi 已提交
1300 1301
  auto pending_outputs_inserted_elem =
      CaptureCurrentFileNumberInPendingOutputs();
1302 1303
  ReadOptions ro;
  ro.total_order_seek = true;
1304
  Arena arena;
1305
  Status s;
1306
  TableProperties table_properties;
1307
  {
1308
    ScopedArenaIterator iter(mem->NewIterator(ro, &arena));
1309 1310 1311
    Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
        "[%s] [WriteLevel0TableForRecovery]"
        " Level-0 table #%" PRIu64 ": started",
1312
        cfd->GetName().c_str(), meta.fd.GetNumber());
1313

1314 1315
    bool paranoid_file_checks =
        cfd->GetLatestMutableCFOptions()->paranoid_file_checks;
1316 1317
    {
      mutex_.Unlock();
1318
      TableFileCreationInfo info;
1319 1320
      s = BuildTable(
          dbname_, env_, *cfd->ioptions(), env_options_, cfd->table_cache(),
1321
          iter.get(), &meta, cfd->internal_comparator(),
1322 1323
          cfd->int_tbl_prop_collector_factories(), cfd->GetID(),
          snapshots_.GetAll(), GetCompressionFlush(*cfd->ioptions()),
1324 1325
          cfd->ioptions()->compression_opts, paranoid_file_checks,
          cfd->internal_stats(), Env::IO_HIGH, &info.table_properties);
1326
      LogFlush(db_options_.info_log);
1327 1328 1329 1330 1331 1332 1333 1334
      Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
          "[%s] [WriteLevel0TableForRecovery]"
          " Level-0 table #%" PRIu64 ": %" PRIu64 " bytes %s",
          cfd->GetName().c_str(), meta.fd.GetNumber(), meta.fd.GetFileSize(),
          s.ToString().c_str());

      // output to event logger
      if (s.ok()) {
1335 1336 1337 1338 1339 1340 1341 1342 1343
        info.db_name = dbname_;
        info.cf_name = cfd->GetName();
        info.file_path = TableFileName(db_options_.db_paths,
                                       meta.fd.GetNumber(),
                                       meta.fd.GetPathId());
        info.file_size = meta.fd.GetFileSize();
        info.job_id = job_id;
        EventHelpers::LogAndNotifyTableFileCreation(
            &event_logger_, db_options_.listeners, meta.fd, info);
1344
      }
1345 1346
      mutex_.Lock();
    }
1347
  }
I
Igor Canadi 已提交
1348
  ReleaseFileNumberFromPendingOutputs(pending_outputs_inserted_elem);
1349 1350 1351 1352

  // Note that if file_size is zero, the file has been deleted and
  // should not be added to the manifest.
  int level = 0;
1353
  if (s.ok() && meta.fd.GetFileSize() > 0) {
1354 1355
    edit->AddFile(level, meta.fd.GetNumber(), meta.fd.GetPathId(),
                  meta.fd.GetFileSize(), meta.smallest, meta.largest,
1356 1357
                  meta.smallest_seqno, meta.largest_seqno,
                  meta.marked_for_compaction);
1358 1359
  }

L
Lei Jin 已提交
1360
  InternalStats::CompactionStats stats(1);
1361
  stats.micros = env_->NowMicros() - start_micros;
1362
  stats.bytes_written = meta.fd.GetFileSize();
1363
  stats.num_output_files = 1;
1364
  cfd->internal_stats()->AddCompactionStats(level, stats);
1365 1366
  cfd->internal_stats()->AddCFStats(
      InternalStats::BYTES_FLUSHED, meta.fd.GetFileSize());
L
Lei Jin 已提交
1367
  RecordTick(stats_, COMPACT_WRITE_BYTES, meta.fd.GetFileSize());
J
jorlow@chromium.org 已提交
1368 1369 1370
  return s;
}

1371 1372
Status DBImpl::FlushMemTableToOutputFile(
    ColumnFamilyData* cfd, const MutableCFOptions& mutable_cf_options,
1373
    bool* made_progress, JobContext* job_context, LogBuffer* log_buffer) {
1374
  mutex_.AssertHeld();
1375
  assert(cfd->imm()->NumNotFlushed() != 0);
1376
  assert(cfd->imm()->IsFlushPending());
1377

I
Igor Canadi 已提交
1378 1379
  FlushJob flush_job(dbname_, cfd, db_options_, mutable_cf_options,
                     env_options_, versions_.get(), &mutex_, &shutting_down_,
I
Igor Canadi 已提交
1380
                     snapshots_.GetAll(), job_context, log_buffer,
1381
                     directories_.GetDbDir(), directories_.GetDataDir(0U),
I
Igor Canadi 已提交
1382 1383
                     GetCompressionFlush(*cfd->ioptions()), stats_,
                     &event_logger_);
1384

1385
  FileMetaData file_meta;
1386 1387 1388 1389 1390 1391 1392

  // Within flush_job.Run, rocksdb may call event listener to notify
  // file creation and deletion.
  //
  // Note that flush_job.Run will unlock and lock the db_mutex,
  // and EventListener callback will be called when the db_mutex
  // is unlocked by the current thread.
1393
  Status s = flush_job.Run(&file_meta);
J
jorlow@chromium.org 已提交
1394 1395

  if (s.ok()) {
I
Igor Canadi 已提交
1396 1397
    InstallSuperVersionAndScheduleWorkWrapper(cfd, job_context,
                                              mutable_cf_options);
1398 1399
    if (made_progress) {
      *made_progress = 1;
1400
    }
S
sdong 已提交
1401
    VersionStorageInfo::LevelSummaryStorage tmp;
I
Igor Canadi 已提交
1402
    LogToBuffer(log_buffer, "[%s] Level summary: %s\n", cfd->GetName().c_str(),
S
sdong 已提交
1403
                cfd->current()->storage_info()->LevelSummary(&tmp));
J
jorlow@chromium.org 已提交
1404
  }
1405

1406
  if (!s.ok() && !s.IsShutdownInProgress() && db_options_.paranoid_checks &&
1407 1408 1409 1410 1411
      bg_error_.ok()) {
    // if a bad error happened (not ShutdownInProgress) and paranoid_checks is
    // true, mark DB read-only
    bg_error_ = s;
  }
1412
  RecordFlushIOStats();
1413 1414 1415
#ifndef ROCKSDB_LITE
  if (s.ok()) {
    // may temporarily unlock and lock the mutex.
1416
    NotifyOnFlushCompleted(cfd, &file_meta, mutable_cf_options,
1417
                           job_context->job_id, flush_job.GetTableProperties());
1418 1419
  }
#endif  // ROCKSDB_LITE
J
jorlow@chromium.org 已提交
1420 1421 1422
  return s;
}

1423 1424 1425 1426
void DBImpl::NotifyOnFlushCompleted(ColumnFamilyData* cfd,
                                    FileMetaData* file_meta,
                                    const MutableCFOptions& mutable_cf_options,
                                    int job_id, TableProperties prop) {
1427
#ifndef ROCKSDB_LITE
1428
  if (db_options_.listeners.size() == 0U) {
1429 1430
    return;
  }
1431 1432 1433 1434
  mutex_.AssertHeld();
  if (shutting_down_.load(std::memory_order_acquire)) {
    return;
  }
1435
  bool triggered_writes_slowdown =
1436
      (cfd->current()->storage_info()->NumLevelFiles(0) >=
1437
       mutable_cf_options.level0_slowdown_writes_trigger);
1438
  bool triggered_writes_stop =
1439
      (cfd->current()->storage_info()->NumLevelFiles(0) >=
1440
       mutable_cf_options.level0_stop_writes_trigger);
1441 1442
  // release lock while notifying events
  mutex_.Unlock();
1443
  {
1444 1445 1446 1447 1448
    FlushJobInfo info;
    info.cf_name = cfd->GetName();
    // TODO(yhchiang): make db_paths dynamic in case flush does not
    //                 go to L0 in the future.
    info.file_path = MakeTableFileName(db_options_.db_paths[0].path,
1449
                                       file_meta->fd.GetNumber());
1450
    info.thread_id = env_->GetThreadID();
1451 1452 1453
    info.job_id = job_id;
    info.triggered_writes_slowdown = triggered_writes_slowdown;
    info.triggered_writes_stop = triggered_writes_stop;
1454 1455
    info.smallest_seqno = file_meta->smallest_seqno;
    info.largest_seqno = file_meta->largest_seqno;
1456
    info.table_properties = prop;
1457
    for (auto listener : db_options_.listeners) {
1458
      listener->OnFlushCompleted(this, info);
1459 1460
    }
  }
1461 1462 1463
  mutex_.Lock();
  // no need to signal bg_cv_ as it will be signaled at the end of the
  // flush process.
1464
#endif  // ROCKSDB_LITE
1465 1466
}

1467 1468 1469 1470
Status DBImpl::CompactRange(const CompactRangeOptions& options,
                            ColumnFamilyHandle* column_family,
                            const Slice* begin, const Slice* end) {
  if (options.target_path_id >= db_options_.db_paths.size()) {
1471 1472 1473
    return Status::InvalidArgument("Invalid target path ID");
  }

1474 1475
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
I
Igor Canadi 已提交
1476 1477

  Status s = FlushMemTable(cfd, FlushOptions());
L
Lei Jin 已提交
1478
  if (!s.ok()) {
1479
    LogFlush(db_options_.info_log);
L
Lei Jin 已提交
1480 1481 1482
    return s;
  }

I
Igor Canadi 已提交
1483
  int max_level_with_files = 0;
G
Gabor Cselle 已提交
1484
  {
1485
    InstrumentedMutexLock l(&mutex_);
I
Igor Canadi 已提交
1486
    Version* base = cfd->current();
1487 1488
    for (int level = 1; level < base->storage_info()->num_non_empty_levels();
         level++) {
S
sdong 已提交
1489
      if (base->storage_info()->OverlapInLevel(level, begin, end)) {
G
Gabor Cselle 已提交
1490 1491 1492 1493
        max_level_with_files = level;
      }
    }
  }
1494

1495
  int final_output_level = 0;
1496 1497 1498 1499
  if (cfd->ioptions()->compaction_style == kCompactionStyleUniversal &&
      cfd->NumberLevels() > 1) {
    // Always compact all files together.
    s = RunManualCompaction(cfd, ColumnFamilyData::kCompactAllLevels,
1500 1501
                            cfd->NumberLevels() - 1, options.target_path_id,
                            begin, end);
1502
    final_output_level = cfd->NumberLevels() - 1;
1503 1504
  } else {
    for (int level = 0; level <= max_level_with_files; level++) {
1505
      int output_level;
A
Andres Notzli 已提交
1506
      // in case the compaction is universal or if we're compacting the
1507 1508 1509 1510
      // bottom-most level, the output level will be the same as input one.
      // level 0 can never be the bottommost level (i.e. if all files are in
      // level 0, we will compact to level 1)
      if (cfd->ioptions()->compaction_style == kCompactionStyleUniversal ||
1511 1512 1513
          cfd->ioptions()->compaction_style == kCompactionStyleFIFO) {
        output_level = level;
      } else if (level == max_level_with_files && level > 0) {
1514 1515 1516 1517 1518 1519 1520
        if (options.bottommost_level_compaction ==
            BottommostLevelCompaction::kSkip) {
          // Skip bottommost level compaction
          continue;
        } else if (options.bottommost_level_compaction ==
                       BottommostLevelCompaction::kIfHaveCompactionFilter &&
                   cfd->ioptions()->compaction_filter == nullptr &&
I
Igor Canadi 已提交
1521
                   cfd->ioptions()->compaction_filter_factory == nullptr) {
A
Andres Notzli 已提交
1522 1523
          // Skip bottommost level compaction since we don't have a compaction
          // filter
1524 1525
          continue;
        }
1526
        output_level = level;
1527
      } else {
1528
        output_level = level + 1;
1529 1530 1531 1532 1533
        if (cfd->ioptions()->compaction_style == kCompactionStyleLevel &&
            cfd->ioptions()->level_compaction_dynamic_level_bytes &&
            level == 0) {
          output_level = ColumnFamilyData::kCompactToBaseLevel;
        }
1534
      }
1535 1536
      s = RunManualCompaction(cfd, level, output_level, options.target_path_id,
                              begin, end);
1537 1538 1539
      if (!s.ok()) {
        break;
      }
1540 1541 1542 1543 1544
      if (output_level == ColumnFamilyData::kCompactToBaseLevel) {
        final_output_level = cfd->NumberLevels() - 1;
      } else if (output_level > final_output_level) {
        final_output_level = output_level;
      }
1545 1546
      TEST_SYNC_POINT("DBImpl::RunManualCompaction()::1");
      TEST_SYNC_POINT("DBImpl::RunManualCompaction()::2");
1547
    }
G
Gabor Cselle 已提交
1548
  }
1549 1550 1551 1552
  if (!s.ok()) {
    LogFlush(db_options_.info_log);
    return s;
  }
1553

1554
  if (options.change_level) {
1555 1556 1557 1558 1559 1560 1561
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "[RefitLevel] waiting for background threads to stop");
    s = PauseBackgroundWork();
    if (s.ok()) {
      s = ReFitLevel(cfd, final_output_level, options.target_level);
    }
    ContinueBackgroundWork();
1562
  }
1563
  LogFlush(db_options_.info_log);
L
Lei Jin 已提交
1564

1565
  {
1566
    InstrumentedMutexLock l(&mutex_);
1567 1568 1569 1570 1571
    // an automatic compaction that has been scheduled might have been
    // preempted by the manual compactions. Need to schedule it back.
    MaybeScheduleFlushOrCompaction();
  }

L
Lei Jin 已提交
1572
  return s;
1573 1574
}

1575 1576 1577 1578 1579
Status DBImpl::CompactFiles(
    const CompactionOptions& compact_options,
    ColumnFamilyHandle* column_family,
    const std::vector<std::string>& input_file_names,
    const int output_level, const int output_path_id) {
I
Igor Canadi 已提交
1580 1581 1582 1583
#ifdef ROCKSDB_LITE
    // not supported in lite version
  return Status::NotSupported("Not supported in ROCKSDB LITE");
#else
1584 1585 1586 1587 1588 1589
  if (column_family == nullptr) {
    return Status::InvalidArgument("ColumnFamilyHandle must be non-null.");
  }

  auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
  assert(cfd);
1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614

  Status s;
  JobContext job_context(0, true);
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL,
                       db_options_.info_log.get());

  // Perform CompactFiles
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
  {
    InstrumentedMutexLock l(&mutex_);

    s = CompactFilesImpl(compact_options, cfd, sv->current,
                         input_file_names, output_level,
                         output_path_id, &job_context, &log_buffer);
  }
  ReturnAndCleanupSuperVersion(cfd, sv);

  // Find and delete obsolete files
  {
    InstrumentedMutexLock l(&mutex_);
    // If !s.ok(), this means that Compaction failed. In that case, we want
    // to delete all obsolete files we might have created and we force
    // FindObsoleteFiles(). This is because job_context does not
    // catch all created files if compaction failed.
    FindObsoleteFiles(&job_context, !s.ok());
1615
  }  // release the mutex
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625

  // delete unnecessary files if any, this is done outside the mutex
  if (job_context.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
    // Have to flush the info logs before bg_compaction_scheduled_--
    // because if bg_flush_scheduled_ becomes 0 and the lock is
    // released, the deconstructor of DB can kick in and destroy all the
    // states of DB so info_log might not be available after that point.
    // It also applies to access other states that DB owns.
    log_buffer.FlushBufferToLog();
    if (job_context.HaveSomethingToDelete()) {
1626
      // no mutex is locked here.  No need to Unlock() and Lock() here.
1627 1628 1629 1630 1631
      PurgeObsoleteFiles(job_context);
    }
    job_context.Clean();
  }

1632
  return s;
I
Igor Canadi 已提交
1633
#endif  // ROCKSDB_LITE
1634 1635
}

I
Igor Canadi 已提交
1636
#ifndef ROCKSDB_LITE
1637 1638 1639
Status DBImpl::CompactFilesImpl(
    const CompactionOptions& compact_options, ColumnFamilyData* cfd,
    Version* version, const std::vector<std::string>& input_file_names,
1640 1641
    const int output_level, int output_path_id, JobContext* job_context,
    LogBuffer* log_buffer) {
1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673
  mutex_.AssertHeld();

  if (shutting_down_.load(std::memory_order_acquire)) {
    return Status::ShutdownInProgress();
  }

  std::unordered_set<uint64_t> input_set;
  for (auto file_name : input_file_names) {
    input_set.insert(TableFileNameToNumber(file_name));
  }

  ColumnFamilyMetaData cf_meta;
  // TODO(yhchiang): can directly use version here if none of the
  // following functions call is pluggable to external developers.
  version->GetColumnFamilyMetaData(&cf_meta);

  if (output_path_id < 0) {
    if (db_options_.db_paths.size() == 1U) {
      output_path_id = 0;
    } else {
      return Status::NotSupported(
          "Automatic output path selection is not "
          "yet supported in CompactFiles()");
    }
  }

  Status s = cfd->compaction_picker()->SanitizeCompactionInputFiles(
      &input_set, cf_meta, output_level);
  if (!s.ok()) {
    return s;
  }

1674
  std::vector<CompactionInputFiles> input_files;
1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
  s = cfd->compaction_picker()->GetCompactionInputsFromFileNumbers(
      &input_files, &input_set, version->storage_info(), compact_options);
  if (!s.ok()) {
    return s;
  }

  for (auto inputs : input_files) {
    if (cfd->compaction_picker()->FilesInCompaction(inputs.files)) {
      return Status::Aborted(
          "Some of the necessary compaction input "
          "files are already being compacted");
    }
  }

  // At this point, CompactFiles will be run.
  bg_compaction_scheduled_++;

  unique_ptr<Compaction> c;
  assert(cfd->compaction_picker());
  c.reset(cfd->compaction_picker()->FormCompaction(
1695 1696
      compact_options, input_files, output_level, version->storage_info(),
      *cfd->GetLatestMutableCFOptions(), output_path_id));
1697 1698 1699
  assert(c);
  c->SetInputVersion(version);
  // deletion compaction currently not allowed in CompactFiles.
1700
  assert(!c->deletion_compaction());
1701

I
Igor Canadi 已提交
1702
  assert(is_snapshot_supported_ || snapshots_.empty());
1703
  CompactionJob compaction_job(
I
Igor Canadi 已提交
1704 1705
      job_context->job_id, c.get(), db_options_, env_options_, versions_.get(),
      &shutting_down_, log_buffer, directories_.GetDbDir(),
S
sdong 已提交
1706
      directories_.GetDataDir(c->output_path_id()), stats_, snapshots_.GetAll(),
1707
      table_cache_, &event_logger_,
1708 1709
      c->mutable_cf_options()->paranoid_file_checks,
      c->mutable_cf_options()->compaction_measure_io_stats, dbname_,
1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722
      nullptr);  // Here we pass a nullptr for CompactionJobStats because
                 // CompactFiles does not trigger OnCompactionCompleted(),
                 // which is the only place where CompactionJobStats is
                 // returned.  The idea of not triggering OnCompationCompleted()
                 // is that CompactFiles runs in the caller thread, so the user
                 // should always know when it completes.  As a result, it makes
                 // less sense to notify the users something they should already
                 // know.
                 //
                 // In the future, if we would like to add CompactionJobStats
                 // support for CompactFiles, we should have CompactFiles API
                 // pass a pointer of CompactionJobStats as the out-value
                 // instead of using EventListener.
1723 1724 1725
  compaction_job.Prepare();

  mutex_.Unlock();
1726
  compaction_job.Run();
1727
  mutex_.Lock();
1728

1729
  Status status = compaction_job.Install(*c->mutable_cf_options(), &mutex_);
1730
  if (status.ok()) {
I
Igor Canadi 已提交
1731 1732
    InstallSuperVersionAndScheduleWorkWrapper(
        c->column_family_data(), job_context, *c->mutable_cf_options());
1733 1734 1735 1736 1737 1738 1739 1740 1741
  }
  c->ReleaseCompactionFiles(s);
  c.reset();

  if (status.ok()) {
    // Done
  } else if (status.IsShutdownInProgress()) {
    // Ignore compaction errors found during shutting down
  } else {
1742 1743
    Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
        "[%s] [JOB %d] Compaction error: %s",
1744
        c->column_family_data()->GetName().c_str(), job_context->job_id,
1745 1746 1747 1748 1749 1750 1751
        status.ToString().c_str());
    if (db_options_.paranoid_checks && bg_error_.ok()) {
      bg_error_ = status;
    }
  }

  bg_compaction_scheduled_--;
1752 1753 1754
  if (bg_compaction_scheduled_ == 0) {
    bg_cv_.SignalAll();
  }
1755 1756 1757

  return status;
}
I
Igor Canadi 已提交
1758
#endif  // ROCKSDB_LITE
1759

1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778
Status DBImpl::PauseBackgroundWork() {
  InstrumentedMutexLock guard_lock(&mutex_);
  bg_work_paused_++;
  while (bg_compaction_scheduled_ > 0 || bg_flush_scheduled_ > 0) {
    bg_cv_.Wait();
  }
  return Status::OK();
}

Status DBImpl::ContinueBackgroundWork() {
  InstrumentedMutexLock guard_lock(&mutex_);
  assert(bg_work_paused_ > 0);
  bg_work_paused_--;
  if (bg_work_paused_ == 0) {
    MaybeScheduleFlushOrCompaction();
  }
  return Status::OK();
}

O
Ori Bernstein 已提交
1779
void DBImpl::NotifyOnCompactionCompleted(
1780 1781
    ColumnFamilyData* cfd, Compaction *c, const Status &st,
    const CompactionJobStats& compaction_job_stats,
1782
    const int job_id) {
O
Ori Bernstein 已提交
1783
#ifndef ROCKSDB_LITE
1784
  if (db_options_.listeners.size() == 0U) {
O
Ori Bernstein 已提交
1785 1786 1787 1788 1789 1790 1791 1792
    return;
  }
  mutex_.AssertHeld();
  if (shutting_down_.load(std::memory_order_acquire)) {
    return;
  }
  // release lock while notifying events
  mutex_.Unlock();
1793 1794 1795 1796
  {
    CompactionJobInfo info;
    info.cf_name = cfd->GetName();
    info.status = st;
1797
    info.thread_id = env_->GetThreadID();
1798 1799
    info.job_id = job_id;
    info.base_input_level = c->start_level();
1800
    info.output_level = c->output_level();
1801
    info.stats = compaction_job_stats;
1802
    info.table_properties = c->GetOutputTableProperties();
1803 1804
    for (size_t i = 0; i < c->num_input_levels(); ++i) {
      for (const auto fmd : *c->inputs(i)) {
1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815
        auto fn = TableFileName(db_options_.db_paths, fmd->fd.GetNumber(),
                                fmd->fd.GetPathId());
        info.input_files.push_back(fn);
        if (info.table_properties.count(fn) == 0) {
          std::shared_ptr<const TableProperties> tp;
          std::string fname;
          auto s = cfd->current()->GetTableProperties(&tp, fmd, &fname);
          if (s.ok()) {
            info.table_properties[fn] = tp;
          }
        }
1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827
      }
    }
    for (const auto newf : c->edit()->GetNewFiles()) {
      info.output_files.push_back(
          TableFileName(db_options_.db_paths,
                        newf.second.fd.GetNumber(),
                        newf.second.fd.GetPathId()));
    }
    for (auto listener : db_options_.listeners) {
      listener->OnCompactionCompleted(this, info);
    }
  }
O
Ori Bernstein 已提交
1828 1829 1830 1831 1832 1833
  mutex_.Lock();
  // no need to signal bg_cv_ as it will be signaled at the end of the
  // flush process.
#endif  // ROCKSDB_LITE
}

1834
Status DBImpl::SetOptions(ColumnFamilyHandle* column_family,
1835
    const std::unordered_map<std::string, std::string>& options_map) {
I
Igor Canadi 已提交
1836 1837 1838
#ifdef ROCKSDB_LITE
  return Status::NotSupported("Not supported in ROCKSDB LITE");
#else
L
Lei Jin 已提交
1839 1840
  auto* cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family)->cfd();
  if (options_map.empty()) {
1841 1842
    Log(InfoLogLevel::WARN_LEVEL,
        db_options_.info_log, "SetOptions() on column family [%s], empty input",
L
Lei Jin 已提交
1843
        cfd->GetName().c_str());
1844
    return Status::InvalidArgument("empty input");
L
Lei Jin 已提交
1845 1846 1847
  }

  MutableCFOptions new_options;
1848
  Status s;
L
Lei Jin 已提交
1849
  {
1850
    InstrumentedMutexLock l(&mutex_);
1851 1852
    s = cfd->SetOptions(options_map);
    if (s.ok()) {
L
Lei Jin 已提交
1853 1854 1855 1856
      new_options = *cfd->GetLatestMutableCFOptions();
    }
  }

1857 1858
  Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
      "SetOptions() on column family [%s], inputs:",
L
Lei Jin 已提交
1859 1860
      cfd->GetName().c_str());
  for (const auto& o : options_map) {
1861 1862
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "%s: %s\n", o.first.c_str(), o.second.c_str());
L
Lei Jin 已提交
1863
  }
1864
  if (s.ok()) {
1865 1866
    Log(InfoLogLevel::INFO_LEVEL,
        db_options_.info_log, "[%s] SetOptions succeeded",
L
Lei Jin 已提交
1867 1868 1869
        cfd->GetName().c_str());
    new_options.Dump(db_options_.info_log.get());
  } else {
1870 1871
    Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
        "[%s] SetOptions failed", cfd->GetName().c_str());
L
Lei Jin 已提交
1872
  }
S
sdong 已提交
1873
  LogFlush(db_options_.info_log);
1874
  return s;
I
Igor Canadi 已提交
1875
#endif  // ROCKSDB_LITE
1876 1877
}

1878
// return the same level if it cannot be moved
1879 1880
int DBImpl::FindMinimumEmptyLevelFitting(ColumnFamilyData* cfd,
    const MutableCFOptions& mutable_cf_options, int level) {
1881
  mutex_.AssertHeld();
S
sdong 已提交
1882
  const auto* vstorage = cfd->current()->storage_info();
1883
  int minimum_level = level;
1884
  for (int i = level - 1; i > 0; --i) {
1885
    // stop if level i is not empty
S
sdong 已提交
1886
    if (vstorage->NumLevelFiles(i) > 0) break;
1887
    // stop if level i is too small (cannot fit the level files)
1888
    if (vstorage->MaxBytesForLevel(i) < vstorage->NumLevelBytes(level)) {
1889 1890
      break;
    }
1891 1892 1893 1894 1895 1896

    minimum_level = i;
  }
  return minimum_level;
}

1897 1898
// REQUIREMENT: block all background work by calling PauseBackgroundWork()
// before calling this function
I
Igor Canadi 已提交
1899 1900
Status DBImpl::ReFitLevel(ColumnFamilyData* cfd, int level, int target_level) {
  assert(level < cfd->NumberLevels());
1901 1902 1903
  if (target_level >= cfd->NumberLevels()) {
    return Status::InvalidArgument("Target level exceeds number of levels");
  }
1904

1905 1906 1907 1908
  std::unique_ptr<SuperVersion> superversion_to_free;
  std::unique_ptr<SuperVersion> new_superversion(new SuperVersion());

  Status status;
I
Igor Canadi 已提交
1909

I
Islam AbdelRahman 已提交
1910
  InstrumentedMutexLock guard_lock(&mutex_);
1911 1912 1913

  // only allow one thread refitting
  if (refitting_level_) {
1914 1915
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "[ReFitLevel] another thread is refitting");
L
Lei Jin 已提交
1916
    return Status::NotSupported("another thread is refitting");
1917 1918 1919
  }
  refitting_level_ = true;

1920
  const MutableCFOptions mutable_cf_options = *cfd->GetLatestMutableCFOptions();
1921
  // move to a smaller level
1922 1923
  int to_level = target_level;
  if (target_level < 0) {
1924
    to_level = FindMinimumEmptyLevelFitting(cfd, mutable_cf_options, level);
1925
  }
1926

1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941
  auto* vstorage = cfd->current()->storage_info();
  if (to_level > level) {
    if (level == 0) {
      return Status::NotSupported(
          "Cannot change from level 0 to other levels.");
    }
    // Check levels are empty for a trivial move
    for (int l = level + 1; l <= to_level; l++) {
      if (vstorage->NumLevelFiles(l) > 0) {
        return Status::NotSupported(
            "Levels between source and target are not empty for a move.");
      }
    }
  }
  if (to_level != level) {
1942
    Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
1943 1944
        "[%s] Before refitting:\n%s", cfd->GetName().c_str(),
        cfd->current()->DebugString().data());
1945

1946
    VersionEdit edit;
I
Igor Canadi 已提交
1947
    edit.SetColumnFamily(cfd->GetID());
1948
    for (const auto& f : vstorage->LevelFiles(level)) {
1949
      edit.DeleteFile(level, f->fd.GetNumber());
1950 1951
      edit.AddFile(to_level, f->fd.GetNumber(), f->fd.GetPathId(),
                   f->fd.GetFileSize(), f->smallest, f->largest,
1952 1953
                   f->smallest_seqno, f->largest_seqno,
                   f->marked_for_compaction);
1954
    }
1955
    Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
1956 1957
        "[%s] Apply version edit:\n%s", cfd->GetName().c_str(),
        edit.DebugString().data());
1958

1959 1960
    status = versions_->LogAndApply(cfd, mutable_cf_options, &edit, &mutex_,
                                    directories_.GetDbDir());
1961 1962
    superversion_to_free.reset(InstallSuperVersionAndScheduleWork(
        cfd, new_superversion.release(), mutable_cf_options));
1963

1964 1965
    Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
        "[%s] LogAndApply: %s\n", cfd->GetName().c_str(),
I
Igor Canadi 已提交
1966
        status.ToString().data());
1967 1968

    if (status.ok()) {
1969
      Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
1970 1971
          "[%s] After refitting:\n%s", cfd->GetName().c_str(),
          cfd->current()->DebugString().data());
1972 1973 1974 1975
    }
  }

  refitting_level_ = false;
I
Igor Canadi 已提交
1976

L
Lei Jin 已提交
1977
  return status;
G
Gabor Cselle 已提交
1978 1979
}

1980 1981 1982
int DBImpl::NumberLevels(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  return cfh->cfd()->NumberLevels();
1983 1984
}

1985
int DBImpl::MaxMemCompactionLevel(ColumnFamilyHandle* column_family) {
1986
  return 0;
1987 1988
}

1989 1990
int DBImpl::Level0StopWriteTrigger(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
1991
  InstrumentedMutexLock l(&mutex_);
L
Lei Jin 已提交
1992 1993
  return cfh->cfd()->GetSuperVersion()->
      mutable_cf_options.level0_stop_writes_trigger;
1994 1995
}

L
Lei Jin 已提交
1996
Status DBImpl::Flush(const FlushOptions& flush_options,
1997 1998
                     ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
L
Lei Jin 已提交
1999
  return FlushMemTable(cfh->cfd(), flush_options);
H
heyongqiang 已提交
2000 2001
}

2002 2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033
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(
          "SyncWAL() is not supported for this implementation of WAL file",
          db_options_.allow_mmap_writes
            ? "try setting Options::allow_mmap_writes to false"
            : Slice());
      }
    }
    for (auto it = logs_.begin();
         it != logs_.end() && it->number <= current_log_number; ++it) {
      auto& log = *it;
      assert(!log.getting_synced);
      log.getting_synced = true;
2034
      logs_to_sync.push_back(log.writer);
2035 2036 2037 2038 2039
    }

    need_log_dir_sync = !log_dir_synced_;
  }

2040
  RecordTick(stats_, WAL_FILE_SYNCED);
2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071
  Status status;
  for (log::Writer* log : logs_to_sync) {
    status = log->file()->SyncWithoutFlush(db_options_.use_fsync);
    if (!status.ok()) {
      break;
    }
  }
  if (status.ok() && need_log_dir_sync) {
    status = directories_.GetWalDir()->Fsync();
  }

  {
    InstrumentedMutexLock l(&mutex_);
    MarkLogsSynced(current_log_number, need_log_dir_sync, status);
  }

  return status;
}

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) {
2072
      logs_to_free_.push_back(log.ReleaseWriter());
A
Andres Noetzli 已提交
2073
      it = logs_.erase(it);
2074 2075 2076 2077 2078
    } else {
      log.getting_synced = false;
      ++it;
    }
  }
A
Andres Noetzli 已提交
2079
  assert(logs_.empty() || (logs_.size() == 1 && !logs_[0].getting_synced));
2080 2081 2082
  log_sync_cv_.SignalAll();
}

2083
SequenceNumber DBImpl::GetLatestSequenceNumber() const {
2084 2085 2086
  return versions_->LastSequence();
}

I
Igor Canadi 已提交
2087
Status DBImpl::RunManualCompaction(ColumnFamilyData* cfd, int input_level,
2088
                                   int output_level, uint32_t output_path_id,
2089 2090
                                   const Slice* begin, const Slice* end,
                                   bool disallow_trivial_move) {
2091 2092
  assert(input_level == ColumnFamilyData::kCompactAllLevels ||
         input_level >= 0);
2093

G
Gabor Cselle 已提交
2094 2095
  InternalKey begin_storage, end_storage;

H
hans@chromium.org 已提交
2096
  ManualCompaction manual;
I
Igor Canadi 已提交
2097
  manual.cfd = cfd;
2098 2099
  manual.input_level = input_level;
  manual.output_level = output_level;
2100
  manual.output_path_id = output_path_id;
G
Gabor Cselle 已提交
2101
  manual.done = false;
2102
  manual.in_progress = false;
2103
  manual.disallow_trivial_move = disallow_trivial_move;
2104 2105 2106
  // For universal compaction, we enforce every manual compaction to compact
  // all files.
  if (begin == nullptr ||
2107 2108
      cfd->ioptions()->compaction_style == kCompactionStyleUniversal ||
      cfd->ioptions()->compaction_style == kCompactionStyleFIFO) {
2109
    manual.begin = nullptr;
G
Gabor Cselle 已提交
2110
  } else {
2111
    begin_storage.SetMaxPossibleForUserKey(*begin);
G
Gabor Cselle 已提交
2112 2113
    manual.begin = &begin_storage;
  }
2114
  if (end == nullptr ||
2115 2116
      cfd->ioptions()->compaction_style == kCompactionStyleUniversal ||
      cfd->ioptions()->compaction_style == kCompactionStyleFIFO) {
2117
    manual.end = nullptr;
G
Gabor Cselle 已提交
2118
  } else {
2119
    end_storage.SetMinPossibleForUserKey(*end);
G
Gabor Cselle 已提交
2120 2121 2122
    manual.end = &end_storage;
  }

2123
  InstrumentedMutexLock l(&mutex_);
2124

2125 2126 2127 2128 2129 2130
  // When a manual compaction arrives, temporarily disable scheduling of
  // non-manual compactions and wait until the number of scheduled compaction
  // jobs drops to zero. This is needed to ensure that this manual compaction
  // can compact any range of keys/files.
  //
  // bg_manual_only_ is non-zero when at least one thread is inside
2131
  // RunManualCompaction(), i.e. during that time no other compaction will
2132 2133 2134
  // get scheduled (see MaybeScheduleFlushOrCompaction).
  //
  // Note that the following loop doesn't stop more that one thread calling
2135
  // RunManualCompaction() from getting to the second while loop below.
2136 2137 2138 2139 2140
  // However, only one of them will actually schedule compaction, while
  // others will wait on a condition variable until it completes.

  ++bg_manual_only_;
  while (bg_compaction_scheduled_ > 0) {
2141
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
I
Igor Canadi 已提交
2142 2143 2144
        "[%s] Manual compaction waiting for all other scheduled background "
        "compactions to finish",
        cfd->GetName().c_str());
2145 2146
    bg_cv_.Wait();
  }
2147

2148 2149
  Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
      "[%s] Manual compaction starting",
I
Igor Canadi 已提交
2150
      cfd->GetName().c_str());
2151

2152 2153 2154 2155
  // We don't check bg_error_ here, because if we get the error in compaction,
  // the compaction will set manual.status to bg_error_ and set manual.done to
  // true.
  while (!manual.done) {
2156 2157 2158
    assert(bg_manual_only_ > 0);
    if (manual_compaction_ != nullptr) {
      // Running either this or some other manual compaction
G
Gabor Cselle 已提交
2159
      bg_cv_.Wait();
2160 2161
    } else {
      manual_compaction_ = &manual;
2162
      bg_compaction_scheduled_++;
2163
      env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW, this);
G
Gabor Cselle 已提交
2164
    }
H
hans@chromium.org 已提交
2165
  }
2166

2167 2168 2169
  assert(!manual.in_progress);
  assert(bg_manual_only_ > 0);
  --bg_manual_only_;
L
Lei Jin 已提交
2170
  return manual.status;
J
jorlow@chromium.org 已提交
2171 2172
}

2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189
InternalIterator* DBImpl::NewInternalIterator(
    Arena* arena, 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();
  }

  mutex_.Lock();
  SuperVersion* super_version = cfd->GetSuperVersion()->Ref();
  mutex_.Unlock();
  ReadOptions roptions;
  return NewInternalIterator(roptions, cfd, super_version, arena);
}

2190
Status DBImpl::FlushMemTable(ColumnFamilyData* cfd,
L
Lei Jin 已提交
2191
                             const FlushOptions& flush_options) {
S
Stanislau Hlebik 已提交
2192 2193 2194
  Status s;
  {
    WriteContext context;
2195
    InstrumentedMutexLock guard_lock(&mutex_);
2196

2197
    if (cfd->imm()->NumNotFlushed() == 0 && cfd->mem()->IsEmpty()) {
2198 2199 2200 2201
      // Nothing to flush
      return Status::OK();
    }

2202 2203
    WriteThread::Writer w;
    write_thread_.EnterUnbatched(&w, &mutex_);
S
Stanislau Hlebik 已提交
2204

I
Igor Canadi 已提交
2205
    // SwitchMemtable() will release and reacquire mutex
S
Stanislau Hlebik 已提交
2206
    // during execution
I
Igor Canadi 已提交
2207
    s = SwitchMemtable(cfd, &context);
2208
    write_thread_.ExitUnbatched(&w);
2209

S
Stanislau Hlebik 已提交
2210 2211
    cfd->imm()->FlushRequested();

2212 2213 2214
    // schedule flush
    SchedulePendingFlush(cfd);
    MaybeScheduleFlushOrCompaction();
S
Stanislau Hlebik 已提交
2215
  }
S
Stanislau Hlebik 已提交
2216

L
Lei Jin 已提交
2217
  if (s.ok() && flush_options.wait) {
2218
    // Wait until the compaction completes
2219
    s = WaitForFlushMemTable(cfd);
2220 2221
  }
  return s;
J
jorlow@chromium.org 已提交
2222 2223
}

2224
Status DBImpl::WaitForFlushMemTable(ColumnFamilyData* cfd) {
2225 2226
  Status s;
  // Wait until the compaction completes
2227
  InstrumentedMutexLock l(&mutex_);
2228
  while (cfd->imm()->NumNotFlushed() > 0 && bg_error_.ok()) {
2229 2230 2231
    if (shutting_down_.load(std::memory_order_acquire)) {
      return Status::ShutdownInProgress();
    }
2232 2233
    bg_cv_.Wait();
  }
2234
  if (!bg_error_.ok()) {
2235 2236 2237
    s = bg_error_;
  }
  return s;
H
heyongqiang 已提交
2238 2239
}

2240
void DBImpl::MaybeScheduleFlushOrCompaction() {
J
jorlow@chromium.org 已提交
2241
  mutex_.AssertHeld();
2242 2243 2244 2245
  if (!opened_successfully_) {
    // Compaction may introduce data race to DB open
    return;
  }
2246 2247
  if (bg_work_paused_ > 0) {
    // we paused the background work
2248
    return;
I
Igor Canadi 已提交
2249
  } else if (shutting_down_.load(std::memory_order_acquire)) {
J
jorlow@chromium.org 已提交
2250
    // DB is being deleted; no more background compactions
2251 2252
    return;
  }
2253

2254 2255 2256 2257
  while (unscheduled_flushes_ > 0 &&
         bg_flush_scheduled_ < db_options_.max_background_flushes) {
    unscheduled_flushes_--;
    bg_flush_scheduled_++;
2258
    env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::HIGH, this);
2259 2260
  }

2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272
  // special case -- if max_background_flushes == 0, then schedule flush on a
  // compaction thread
  if (db_options_.max_background_flushes == 0) {
    while (unscheduled_flushes_ > 0 &&
           bg_flush_scheduled_ + bg_compaction_scheduled_ <
               db_options_.max_background_compactions) {
      unscheduled_flushes_--;
      bg_flush_scheduled_++;
      env_->Schedule(&DBImpl::BGWorkFlush, this, Env::Priority::LOW, this);
    }
  }

2273 2274 2275 2276 2277 2278
  if (bg_manual_only_) {
    // only manual compactions are allowed to run. don't schedule automatic
    // compactions
    return;
  }

2279 2280 2281 2282
  while (bg_compaction_scheduled_ < db_options_.max_background_compactions &&
         unscheduled_compactions_ > 0) {
    bg_compaction_scheduled_++;
    unscheduled_compactions_--;
2283
    env_->Schedule(&DBImpl::BGWorkCompaction, this, Env::Priority::LOW, this);
2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329
  }
}

void DBImpl::AddToCompactionQueue(ColumnFamilyData* cfd) {
  assert(!cfd->pending_compaction());
  cfd->Ref();
  compaction_queue_.push_back(cfd);
  cfd->set_pending_compaction(true);
}

ColumnFamilyData* DBImpl::PopFirstFromCompactionQueue() {
  assert(!compaction_queue_.empty());
  auto cfd = *compaction_queue_.begin();
  compaction_queue_.pop_front();
  assert(cfd->pending_compaction());
  cfd->set_pending_compaction(false);
  return cfd;
}

void DBImpl::AddToFlushQueue(ColumnFamilyData* cfd) {
  assert(!cfd->pending_flush());
  cfd->Ref();
  flush_queue_.push_back(cfd);
  cfd->set_pending_flush(true);
}

ColumnFamilyData* DBImpl::PopFirstFromFlushQueue() {
  assert(!flush_queue_.empty());
  auto cfd = *flush_queue_.begin();
  flush_queue_.pop_front();
  assert(cfd->pending_flush());
  cfd->set_pending_flush(false);
  return cfd;
}

void DBImpl::SchedulePendingFlush(ColumnFamilyData* cfd) {
  if (!cfd->pending_flush() && cfd->imm()->IsFlushPending()) {
    AddToFlushQueue(cfd);
    ++unscheduled_flushes_;
  }
}

void DBImpl::SchedulePendingCompaction(ColumnFamilyData* cfd) {
  if (!cfd->pending_compaction() && cfd->NeedsCompaction()) {
    AddToCompactionQueue(cfd);
    ++unscheduled_compactions_;
2330 2331 2332
  }
}

2333
void DBImpl::RecordFlushIOStats() {
I
Igor Canadi 已提交
2334
  RecordTick(stats_, FLUSH_WRITE_BYTES, IOSTATS(bytes_written));
2335 2336 2337
  IOSTATS_RESET(bytes_written);
}

2338
void DBImpl::BGWorkFlush(void* db) {
2339
  IOSTATS_SET_THREAD_POOL_ID(Env::Priority::HIGH);
2340
  TEST_SYNC_POINT("DBImpl::BGWorkFlush");
2341
  reinterpret_cast<DBImpl*>(db)->BackgroundCallFlush();
2342
  TEST_SYNC_POINT("DBImpl::BGWorkFlush:done");
2343 2344 2345
}

void DBImpl::BGWorkCompaction(void* db) {
2346
  IOSTATS_SET_THREAD_POOL_ID(Env::Priority::LOW);
2347
  TEST_SYNC_POINT("DBImpl::BGWorkCompaction");
2348 2349 2350
  reinterpret_cast<DBImpl*>(db)->BackgroundCallCompaction();
}

2351
Status DBImpl::BackgroundFlush(bool* made_progress, JobContext* job_context,
H
Haobo Xu 已提交
2352
                               LogBuffer* log_buffer) {
2353
  mutex_.AssertHeld();
2354

2355 2356 2357 2358 2359 2360 2361
  Status status = bg_error_;
  if (status.ok() && shutting_down_.load(std::memory_order_acquire)) {
    status = Status::ShutdownInProgress();
  }

  if (!status.ok()) {
    return status;
2362 2363
  }

2364 2365 2366
  ColumnFamilyData* cfd = nullptr;
  while (!flush_queue_.empty()) {
    // This cfd is already referenced
I
Igor Canadi 已提交
2367
    auto first_cfd = PopFirstFromFlushQueue();
2368

I
Igor Canadi 已提交
2369
    if (first_cfd->IsDropped() || !first_cfd->imm()->IsFlushPending()) {
2370
      // can't flush this CF, try next one
I
Igor Canadi 已提交
2371 2372
      if (first_cfd->Unref()) {
        delete first_cfd;
2373 2374
      }
      continue;
2375
    }
2376 2377

    // found a flush!
I
Igor Canadi 已提交
2378
    cfd = first_cfd;
2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391
    break;
  }

  if (cfd != nullptr) {
    const MutableCFOptions mutable_cf_options =
        *cfd->GetLatestMutableCFOptions();
    LogToBuffer(
        log_buffer,
        "Calling FlushMemTableToOutputFile with column "
        "family [%s], flush slots available %d, compaction slots available %d",
        cfd->GetName().c_str(),
        db_options_.max_background_flushes - bg_flush_scheduled_,
        db_options_.max_background_compactions - bg_compaction_scheduled_);
2392
    status = FlushMemTableToOutputFile(cfd, mutable_cf_options, made_progress,
2393 2394 2395
                                       job_context, log_buffer);
    if (cfd->Unref()) {
      delete cfd;
2396
    }
J
jorlow@chromium.org 已提交
2397
  }
2398
  return status;
J
jorlow@chromium.org 已提交
2399 2400
}

2401
void DBImpl::BackgroundCallFlush() {
2402
  bool made_progress = false;
2403
  JobContext job_context(next_job_id_.fetch_add(1), true);
2404 2405
  assert(bg_flush_scheduled_);

2406
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, db_options_.info_log.get());
H
Haobo Xu 已提交
2407
  {
2408
    InstrumentedMutexLock l(&mutex_);
2409
    num_running_flushes_++;
H
Haobo Xu 已提交
2410

I
Igor Canadi 已提交
2411 2412 2413
    auto pending_outputs_inserted_elem =
        CaptureCurrentFileNumberInPendingOutputs();

2414
    Status s = BackgroundFlush(&made_progress, &job_context, &log_buffer);
2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431
    if (!s.ok() && !s.IsShutdownInProgress()) {
      // Wait a little bit before retrying background flush in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed flushes for the duration of
      // the problem.
      uint64_t error_cnt =
        default_cf_internal_stats_->BumpAndGetBackgroundErrorCount();
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      mutex_.Unlock();
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "Waiting after background flush error: %s"
          "Accumulated background error counts: %" PRIu64,
          s.ToString().c_str(), error_cnt);
      log_buffer.FlushBufferToLog();
      LogFlush(db_options_.info_log);
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
H
Haobo Xu 已提交
2432 2433
    }

I
Igor Canadi 已提交
2434 2435
    ReleaseFileNumberFromPendingOutputs(pending_outputs_inserted_elem);

2436 2437 2438
    // If flush failed, we want to delete all temporary files that we might have
    // created. Thus, we force full scan in FindObsoleteFiles()
    FindObsoleteFiles(&job_context, !s.ok() && !s.IsShutdownInProgress());
H
Haobo Xu 已提交
2439
    // delete unnecessary files if any, this is done outside the mutex
I
Igor Canadi 已提交
2440
    if (job_context.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
2441
      mutex_.Unlock();
2442 2443 2444 2445 2446
      // Have to flush the info logs before bg_flush_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
2447
      log_buffer.FlushBufferToLog();
I
Igor Canadi 已提交
2448 2449
      if (job_context.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(job_context);
2450
      }
I
Igor Canadi 已提交
2451
      job_context.Clean();
2452 2453
      mutex_.Lock();
    }
I
Igor Canadi 已提交
2454

2455 2456
    assert(num_running_flushes_ > 0);
    num_running_flushes_--;
H
Haobo Xu 已提交
2457
    bg_flush_scheduled_--;
2458 2459
    // See if there's more work to be done
    MaybeScheduleFlushOrCompaction();
I
Igor Canadi 已提交
2460
    RecordFlushIOStats();
H
Haobo Xu 已提交
2461
    bg_cv_.SignalAll();
I
Igor Canadi 已提交
2462 2463 2464 2465
    // 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.
2466
  }
J
jorlow@chromium.org 已提交
2467 2468
}

2469
void DBImpl::BackgroundCallCompaction() {
2470
  bool made_progress = false;
2471
  JobContext job_context(next_job_id_.fetch_add(1), true);
H
Haobo Xu 已提交
2472 2473

  MaybeDumpStats();
2474
  LogBuffer log_buffer(InfoLogLevel::INFO_LEVEL, db_options_.info_log.get());
2475
  {
2476
    InstrumentedMutexLock l(&mutex_);
2477
    num_running_compactions_++;
I
Igor Canadi 已提交
2478 2479 2480 2481

    auto pending_outputs_inserted_elem =
        CaptureCurrentFileNumberInPendingOutputs();

2482
    assert(bg_compaction_scheduled_);
2483
    Status s = BackgroundCompaction(&made_progress, &job_context, &log_buffer);
2484 2485 2486 2487 2488 2489
    if (!s.ok() && !s.IsShutdownInProgress()) {
      // Wait a little bit before retrying background compaction in
      // case this is an environmental problem and we do not want to
      // chew up resources for failed compactions for the duration of
      // the problem.
      uint64_t error_cnt =
2490
          default_cf_internal_stats_->BumpAndGetBackgroundErrorCount();
2491 2492 2493 2494 2495 2496 2497 2498 2499 2500
      bg_cv_.SignalAll();  // In case a waiter can proceed despite the error
      mutex_.Unlock();
      log_buffer.FlushBufferToLog();
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "Waiting after background compaction error: %s, "
          "Accumulated background error counts: %" PRIu64,
          s.ToString().c_str(), error_cnt);
      LogFlush(db_options_.info_log);
      env_->SleepForMicroseconds(1000000);
      mutex_.Lock();
2501
    }
H
Haobo Xu 已提交
2502

I
Igor Canadi 已提交
2503 2504
    ReleaseFileNumberFromPendingOutputs(pending_outputs_inserted_elem);

2505 2506 2507 2508
    // If compaction failed, we want to delete all temporary files that we might
    // have created (they might not be all recorded in job_context in case of a
    // failure). Thus, we force full scan in FindObsoleteFiles()
    FindObsoleteFiles(&job_context, !s.ok() && !s.IsShutdownInProgress());
2509 2510

    // delete unnecessary files if any, this is done outside the mutex
I
Igor Canadi 已提交
2511
    if (job_context.HaveSomethingToDelete() || !log_buffer.IsEmpty()) {
2512
      mutex_.Unlock();
2513 2514 2515 2516 2517
      // Have to flush the info logs before bg_compaction_scheduled_--
      // because if bg_flush_scheduled_ becomes 0 and the lock is
      // released, the deconstructor of DB can kick in and destroy all the
      // states of DB so info_log might not be available after that point.
      // It also applies to access other states that DB owns.
H
Haobo Xu 已提交
2518
      log_buffer.FlushBufferToLog();
I
Igor Canadi 已提交
2519 2520
      if (job_context.HaveSomethingToDelete()) {
        PurgeObsoleteFiles(job_context);
2521
      }
I
Igor Canadi 已提交
2522
      job_context.Clean();
2523 2524
      mutex_.Lock();
    }
D
Dhruba Borthakur 已提交
2525

2526 2527
    assert(num_running_compactions_ > 0);
    num_running_compactions_--;
2528
    bg_compaction_scheduled_--;
J
jorlow@chromium.org 已提交
2529

2530 2531
    versions_->GetColumnFamilySet()->FreeDeadColumnFamilies();

2532 2533
    // See if there's more work to be done
    MaybeScheduleFlushOrCompaction();
2534
    if (made_progress || bg_compaction_scheduled_ == 0 || bg_manual_only_ > 0) {
2535
      // signal if
2536
      // * made_progress -- need to wakeup DelayWrite
2537 2538 2539 2540 2541 2542
      // * bg_compaction_scheduled_ == 0 -- need to wakeup ~DBImpl
      // * bg_manual_only_ > 0 -- need to wakeup RunManualCompaction
      // If none of this is true, there is no need to signal since nobody is
      // waiting for it
      bg_cv_.SignalAll();
    }
I
Igor Canadi 已提交
2543 2544 2545 2546
    // 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.
2547
  }
J
jorlow@chromium.org 已提交
2548 2549
}

2550 2551
Status DBImpl::BackgroundCompaction(bool* made_progress,
                                    JobContext* job_context,
2552
                                    LogBuffer* log_buffer) {
2553
  *made_progress = false;
J
jorlow@chromium.org 已提交
2554
  mutex_.AssertHeld();
2555

2556 2557
  bool is_manual = (manual_compaction_ != nullptr) &&
                   (manual_compaction_->in_progress == false);
2558 2559
  bool trivial_move_disallowed = is_manual &&
                                 manual_compaction_->disallow_trivial_move;
2560

2561
  CompactionJobStats compaction_job_stats;
2562 2563 2564 2565 2566 2567
  Status status = bg_error_;
  if (status.ok() && shutting_down_.load(std::memory_order_acquire)) {
    status = Status::ShutdownInProgress();
  }

  if (!status.ok()) {
2568
    if (is_manual) {
2569
      manual_compaction_->status = status;
2570 2571 2572 2573
      manual_compaction_->done = true;
      manual_compaction_->in_progress = false;
      manual_compaction_ = nullptr;
    }
2574
    return status;
2575 2576
  }

2577 2578 2579
  if (is_manual) {
    // another thread cannot pick up the same work
    manual_compaction_->in_progress = true;
2580 2581 2582 2583
  } else if (manual_compaction_ != nullptr) {
    // there should be no automatic compactions running when manual compaction
    // is running
    return Status::OK();
2584 2585 2586
  }

  unique_ptr<Compaction> c;
2587 2588
  InternalKey manual_end_storage;
  InternalKey* manual_end = &manual_end_storage;
H
hans@chromium.org 已提交
2589
  if (is_manual) {
G
Gabor Cselle 已提交
2590
    ManualCompaction* m = manual_compaction_;
2591
    assert(m->in_progress);
2592 2593 2594
    c.reset(m->cfd->CompactRange(
          *m->cfd->GetLatestMutableCFOptions(), m->input_level, m->output_level,
          m->output_path_id, m->begin, m->end, &manual_end));
2595
    if (!c) {
2596
      m->done = true;
2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611 2612
      LogToBuffer(log_buffer,
                  "[%s] Manual compaction from level-%d from %s .. "
                  "%s; nothing to do\n",
                  m->cfd->GetName().c_str(), m->input_level,
                  (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
                  (m->end ? m->end->DebugString().c_str() : "(end)"));
    } else {
      LogToBuffer(log_buffer,
                  "[%s] Manual compaction from level-%d to level-%d from %s .. "
                  "%s; will stop at %s\n",
                  m->cfd->GetName().c_str(), m->input_level, c->output_level(),
                  (m->begin ? m->begin->DebugString().c_str() : "(begin)"),
                  (m->end ? m->end->DebugString().c_str() : "(end)"),
                  ((m->done || manual_end == nullptr)
                       ? "(end)"
                       : manual_end->DebugString().c_str()));
G
Gabor Cselle 已提交
2613
    }
2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645
  } else if (!compaction_queue_.empty()) {
    // cfd is referenced here
    auto cfd = PopFirstFromCompactionQueue();
    // We unreference here because the following code will take a Ref() on
    // this cfd if it is going to use it (Compaction class holds a
    // reference).
    // This will all happen under a mutex so we don't have to be afraid of
    // somebody else deleting it.
    if (cfd->Unref()) {
      delete cfd;
      // This was the last reference of the column family, so no need to
      // compact.
      return Status::OK();
    }

    // Pick up latest mutable CF Options and use it throughout the
    // compaction job
    // Compaction makes a copy of the latest MutableCFOptions. It should be used
    // throughout the compaction procedure to make sure consistency. It will
    // eventually be installed into SuperVersion
    auto* mutable_cf_options = cfd->GetLatestMutableCFOptions();
    if (!mutable_cf_options->disable_auto_compactions && !cfd->IsDropped()) {
      // NOTE: try to avoid unnecessary copy of MutableCFOptions if
      // compaction is not necessary. Need to make sure mutex is held
      // until we make a copy in the following code
      c.reset(cfd->PickCompaction(*mutable_cf_options, log_buffer));
      if (c != nullptr) {
        // update statistics
        MeasureTime(stats_, NUM_FILES_IN_SINGLE_COMPACTION,
                    c->inputs(0)->size());
        // There are three things that can change compaction score:
        // 1) When flush or compaction finish. This case is covered by
I
Igor Canadi 已提交
2646
        // InstallSuperVersionAndScheduleWork
2647
        // 2) When MutableCFOptions changes. This case is also covered by
I
Igor Canadi 已提交
2648 2649
        // InstallSuperVersionAndScheduleWork, because this is when the new
        // options take effect.
2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660
        // 3) When we Pick a new compaction, we "remove" those files being
        // compacted from the calculation, which then influences compaction
        // score. Here we check if we need the new compaction even without the
        // files that are currently being compacted. If we need another
        // compaction, we might be able to execute it in parallel, so we add it
        // to the queue and schedule a new thread.
        if (cfd->NeedsCompaction()) {
          // Yes, we need more compactions!
          AddToCompactionQueue(cfd);
          ++unscheduled_compactions_;
          MaybeScheduleFlushOrCompaction();
I
Igor Canadi 已提交
2661
        }
I
Igor Canadi 已提交
2662 2663
      }
    }
J
jorlow@chromium.org 已提交
2664 2665
  }

2666
  if (!c) {
H
hans@chromium.org 已提交
2667
    // Nothing to do
2668
    LogToBuffer(log_buffer, "Compaction nothing to do");
2669
  } else if (c->deletion_compaction()) {
I
Igor Canadi 已提交
2670 2671 2672 2673
    // TODO(icanadi) Do we want to honor snapshots here? i.e. not delete old
    // file if there is alive snapshot pointing to it
    assert(c->num_input_files(1) == 0);
    assert(c->level() == 0);
2674
    assert(c->column_family_data()->ioptions()->compaction_style ==
I
Igor Canadi 已提交
2675
           kCompactionStyleFIFO);
2676 2677 2678

    compaction_job_stats.num_input_files = c->num_input_files(0);

I
Igor Canadi 已提交
2679
    for (const auto& f : *c->inputs(0)) {
2680
      c->edit()->DeleteFile(c->level(), f->fd.GetNumber());
I
Igor Canadi 已提交
2681
    }
2682 2683 2684
    status = versions_->LogAndApply(c->column_family_data(),
                                    *c->mutable_cf_options(), c->edit(),
                                    &mutex_, directories_.GetDbDir());
I
Igor Canadi 已提交
2685 2686
    InstallSuperVersionAndScheduleWorkWrapper(
        c->column_family_data(), job_context, *c->mutable_cf_options());
I
Igor Canadi 已提交
2687 2688 2689
    LogToBuffer(log_buffer, "[%s] Deleted %d files\n",
                c->column_family_data()->GetName().c_str(),
                c->num_input_files(0));
2690
    *made_progress = true;
2691
  } else if (!trivial_move_disallowed && c->IsTrivialMove()) {
2692
    TEST_SYNC_POINT("DBImpl::BackgroundCompaction:TrivialMove");
2693 2694 2695 2696 2697
    // Instrument for event update
    // TODO(yhchiang): add op details for showing trivial-move.
    ThreadStatusUtil::SetColumnFamily(c->column_family_data());
    ThreadStatusUtil::SetThreadOperation(ThreadStatus::OP_COMPACTION);

2698 2699
    compaction_job_stats.num_input_files = c->num_input_files(0);

2700 2701 2702
    // Move files to next level
    int32_t moved_files = 0;
    int64_t moved_bytes = 0;
2703
    for (unsigned int l = 0; l < c->num_input_levels(); l++) {
2704
      if (c->level(l) == c->output_level()) {
2705 2706 2707 2708
        continue;
      }
      for (size_t i = 0; i < c->num_input_files(l); i++) {
        FileMetaData* f = c->input(l, i);
2709
        c->edit()->DeleteFile(c->level(l), f->fd.GetNumber());
2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721
        c->edit()->AddFile(c->output_level(), f->fd.GetNumber(),
                           f->fd.GetPathId(), f->fd.GetFileSize(), f->smallest,
                           f->largest, f->smallest_seqno, f->largest_seqno,
                           f->marked_for_compaction);

        LogToBuffer(log_buffer,
                    "[%s] Moving #%" PRIu64 " to level-%d %" PRIu64 " bytes\n",
                    c->column_family_data()->GetName().c_str(),
                    f->fd.GetNumber(), c->output_level(), f->fd.GetFileSize());
        ++moved_files;
        moved_bytes += f->fd.GetFileSize();
      }
2722
    }
2723

2724
    status = versions_->LogAndApply(c->column_family_data(),
2725 2726
                                    *c->mutable_cf_options(), c->edit(),
                                    &mutex_, directories_.GetDbDir());
2727
    // Use latest MutableCFOptions
I
Igor Canadi 已提交
2728 2729
    InstallSuperVersionAndScheduleWorkWrapper(
        c->column_family_data(), job_context, *c->mutable_cf_options());
2730

S
sdong 已提交
2731
    VersionStorageInfo::LevelSummaryStorage tmp;
I
Islam AbdelRahman 已提交
2732
    c->column_family_data()->internal_stats()->IncBytesMoved(c->output_level(),
2733
                                                             moved_bytes);
2734 2735 2736 2737
    {
      event_logger_.LogToBuffer(log_buffer)
          << "job" << job_context->job_id << "event"
          << "trivial_move"
I
Islam AbdelRahman 已提交
2738
          << "destination_level" << c->output_level() << "files" << moved_files
2739
          << "total_files_size" << moved_bytes;
2740
    }
2741 2742
    LogToBuffer(
        log_buffer,
2743
        "[%s] Moved #%d files to level-%d %" PRIu64 " bytes %s: %s\n",
I
Islam AbdelRahman 已提交
2744 2745
        c->column_family_data()->GetName().c_str(), moved_files,
        c->output_level(), moved_bytes, status.ToString().c_str(),
2746
        c->column_family_data()->current()->storage_info()->LevelSummary(&tmp));
2747
    *made_progress = true;
2748 2749 2750

    // Clear Instrument
    ThreadStatusUtil::ResetThreadStatus();
J
jorlow@chromium.org 已提交
2751
  } else {
2752 2753 2754
    int output_level  __attribute__((unused)) = c->output_level();
    TEST_SYNC_POINT_CALLBACK("DBImpl::BackgroundCompaction:NonTrivial",
                             &output_level);
I
Igor Canadi 已提交
2755
    assert(is_snapshot_supported_ || snapshots_.empty());
2756
    CompactionJob compaction_job(
I
Igor Canadi 已提交
2757 2758
        job_context->job_id, c.get(), db_options_, env_options_,
        versions_.get(), &shutting_down_, log_buffer, directories_.GetDbDir(),
2759
        directories_.GetDataDir(c->output_path_id()), stats_,
2760
        snapshots_.GetAll(), table_cache_, &event_logger_,
2761 2762
        c->mutable_cf_options()->paranoid_file_checks,
        c->mutable_cf_options()->compaction_measure_io_stats, dbname_,
S
sdong 已提交
2763
        &compaction_job_stats);
I
Igor Canadi 已提交
2764
    compaction_job.Prepare();
2765

I
Igor Canadi 已提交
2766
    mutex_.Unlock();
2767
    compaction_job.Run();
2768
    TEST_SYNC_POINT("DBImpl::BackgroundCompaction:NonTrivial:AfterRun");
I
Igor Canadi 已提交
2769
    mutex_.Lock();
2770

2771
    status = compaction_job.Install(*c->mutable_cf_options(), &mutex_);
I
Igor Canadi 已提交
2772
    if (status.ok()) {
I
Igor Canadi 已提交
2773 2774
      InstallSuperVersionAndScheduleWorkWrapper(
          c->column_family_data(), job_context, *c->mutable_cf_options());
I
Igor Canadi 已提交
2775
    }
2776
    *made_progress = true;
O
Ori Bernstein 已提交
2777 2778
  }
  if (c != nullptr) {
2779 2780 2781
    NotifyOnCompactionCompleted(
        c->column_family_data(), c.get(), status,
        compaction_job_stats, job_context->job_id);
I
Igor Canadi 已提交
2782
    c->ReleaseCompactionFiles(status);
2783
    *made_progress = true;
J
jorlow@chromium.org 已提交
2784
  }
2785
  // this will unref its input_version and column_family_data
2786
  c.reset();
J
jorlow@chromium.org 已提交
2787 2788 2789

  if (status.ok()) {
    // Done
F
Feng Zhu 已提交
2790
  } else if (status.IsShutdownInProgress()) {
J
jorlow@chromium.org 已提交
2791 2792
    // Ignore compaction errors found during shutting down
  } else {
2793
    Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log, "Compaction error: %s",
2794
        status.ToString().c_str());
2795
    if (db_options_.paranoid_checks && bg_error_.ok()) {
J
jorlow@chromium.org 已提交
2796 2797 2798
      bg_error_ = status;
    }
  }
H
hans@chromium.org 已提交
2799 2800

  if (is_manual) {
G
Gabor Cselle 已提交
2801
    ManualCompaction* m = manual_compaction_;
2802
    if (!status.ok()) {
L
Lei Jin 已提交
2803
      m->status = status;
2804 2805
      m->done = true;
    }
2806 2807 2808 2809 2810 2811 2812 2813 2814
    // For universal compaction:
    //   Because universal compaction always happens at level 0, so one
    //   compaction will pick up all overlapped files. No files will be
    //   filtered out due to size limit and left for a successive compaction.
    //   So we can safely conclude the current compaction.
    //
    //   Also note that, if we don't stop here, then the current compaction
    //   writes a new file back to level 0, which will be used in successive
    //   compaction. Hence the manual compaction will never finish.
2815 2816 2817 2818 2819
    //
    // Stop the compaction if manual_end points to nullptr -- this means
    // that we compacted the whole range. manual_end should always point
    // to nullptr in case of universal compaction
    if (manual_end == nullptr) {
2820 2821
      m->done = true;
    }
G
Gabor Cselle 已提交
2822 2823 2824
    if (!m->done) {
      // We only compacted part of the requested range.  Update *m
      // to the range that is left to be compacted.
I
Igor Canadi 已提交
2825
      // Universal and FIFO compactions should always compact the whole range
S
sdong 已提交
2826 2827 2828
      assert(m->cfd->ioptions()->compaction_style !=
                 kCompactionStyleUniversal ||
             m->cfd->ioptions()->num_levels > 1);
2829
      assert(m->cfd->ioptions()->compaction_style != kCompactionStyleFIFO);
2830
      m->tmp_storage = *manual_end;
G
Gabor Cselle 已提交
2831 2832
      m->begin = &m->tmp_storage;
    }
2833
    m->in_progress = false; // not being processed anymore
2834
    manual_compaction_ = nullptr;
H
hans@chromium.org 已提交
2835
  }
2836
  return status;
J
jorlow@chromium.org 已提交
2837 2838
}

2839 2840
namespace {
struct IterState {
2841
  IterState(DBImpl* _db, InstrumentedMutex* _mu, SuperVersion* _super_version)
I
Igor Canadi 已提交
2842
      : db(_db), mu(_mu), super_version(_super_version) {}
2843 2844

  DBImpl* db;
2845
  InstrumentedMutex* mu;
2846
  SuperVersion* super_version;
2847 2848 2849 2850
};

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

2852
  if (state->super_version->Unref()) {
2853 2854 2855
    // Job id == 0 means that this is not our background process, but rather
    // user thread
    JobContext job_context(0);
2856

2857 2858
    state->mu->Lock();
    state->super_version->Cleanup();
I
Igor Canadi 已提交
2859
    state->db->FindObsoleteFiles(&job_context, false, true);
2860 2861 2862
    state->mu->Unlock();

    delete state->super_version;
I
Igor Canadi 已提交
2863 2864
    if (job_context.HaveSomethingToDelete()) {
      state->db->PurgeObsoleteFiles(job_context);
2865
    }
I
Igor Canadi 已提交
2866
    job_context.Clean();
I
Igor Canadi 已提交
2867
  }
T
Tomislav Novak 已提交
2868

2869 2870
  delete state;
}
H
Hans Wennborg 已提交
2871
}  // namespace
2872

S
sdong 已提交
2873 2874 2875 2876 2877
InternalIterator* DBImpl::NewInternalIterator(const ReadOptions& read_options,
                                              ColumnFamilyData* cfd,
                                              SuperVersion* super_version,
                                              Arena* arena) {
  InternalIterator* internal_iter;
2878 2879 2880 2881 2882
  assert(arena != nullptr);
  // Need to create internal iterator from the arena.
  MergeIteratorBuilder merge_iter_builder(&cfd->internal_comparator(), arena);
  // Collect iterator for mutable mem
  merge_iter_builder.AddIterator(
L
Lei Jin 已提交
2883
      super_version->mem->NewIterator(read_options, arena));
2884
  // Collect all needed child iterators for immutable memtables
L
Lei Jin 已提交
2885
  super_version->imm->AddIterators(read_options, &merge_iter_builder);
2886
  // Collect iterators for files in L0 - Ln
L
Lei Jin 已提交
2887
  super_version->current->AddIterators(read_options, env_options_,
2888 2889
                                       &merge_iter_builder);
  internal_iter = merge_iter_builder.Finish();
2890
  IterState* cleanup = new IterState(this, &mutex_, super_version);
2891
  internal_iter->RegisterCleanup(CleanupIteratorState, cleanup, nullptr);
J
jorlow@chromium.org 已提交
2892 2893 2894 2895

  return internal_iter;
}

2896 2897 2898 2899
ColumnFamilyHandle* DBImpl::DefaultColumnFamily() const {
  return default_cf_handle_;
}

L
Lei Jin 已提交
2900
Status DBImpl::Get(const ReadOptions& read_options,
2901
                   ColumnFamilyHandle* column_family, const Slice& key,
J
jorlow@chromium.org 已提交
2902
                   std::string* value) {
L
Lei Jin 已提交
2903
  return GetImpl(read_options, column_family, key, value);
2904 2905
}

I
Igor Canadi 已提交
2906 2907
// JobContext gets created and destructed outside of the lock --
// we
I
Igor Canadi 已提交
2908 2909
// use this convinently to:
// * malloc one SuperVersion() outside of the lock -- new_superversion
2910
// * delete SuperVersion()s outside of the lock -- superversions_to_free
I
Igor Canadi 已提交
2911
//
I
Igor Canadi 已提交
2912 2913 2914
// However, if InstallSuperVersionAndScheduleWork() gets called twice with the
// same job_context, we can't reuse the SuperVersion() that got
// malloced because
I
Igor Canadi 已提交
2915 2916 2917
// first call already used it. In that rare case, we take a hit and create a
// new SuperVersion() inside of the mutex. We do similar thing
// for superversion_to_free
I
Igor Canadi 已提交
2918
void DBImpl::InstallSuperVersionAndScheduleWorkWrapper(
I
Igor Canadi 已提交
2919
    ColumnFamilyData* cfd, JobContext* job_context,
2920
    const MutableCFOptions& mutable_cf_options) {
2921
  mutex_.AssertHeld();
I
Igor Canadi 已提交
2922
  SuperVersion* old_superversion = InstallSuperVersionAndScheduleWork(
I
Igor Canadi 已提交
2923 2924 2925
      cfd, job_context->new_superversion, mutable_cf_options);
  job_context->new_superversion = nullptr;
  job_context->superversions_to_free.push_back(old_superversion);
I
Igor Canadi 已提交
2926 2927
}

I
Igor Canadi 已提交
2928
SuperVersion* DBImpl::InstallSuperVersionAndScheduleWork(
L
Lei Jin 已提交
2929
    ColumnFamilyData* cfd, SuperVersion* new_sv,
I
Igor Canadi 已提交
2930
    const MutableCFOptions& mutable_cf_options) {
L
Lei Jin 已提交
2931
  mutex_.AssertHeld();
2932 2933 2934 2935 2936 2937 2938 2939 2940

  // Update max_total_in_memory_state_
  size_t old_memtable_size = 0;
  auto* old_sv = cfd->GetSuperVersion();
  if (old_sv) {
    old_memtable_size = old_sv->mutable_cf_options.write_buffer_size *
                        old_sv->mutable_cf_options.max_write_buffer_number;
  }

L
Lei Jin 已提交
2941 2942 2943
  auto* old = cfd->InstallSuperVersion(
      new_sv ? new_sv : new SuperVersion(), &mutex_, mutable_cf_options);

2944
  // Whenever we install new SuperVersion, we might need to issue new flushes or
I
Igor Canadi 已提交
2945 2946 2947 2948
  // compactions.
  SchedulePendingFlush(cfd);
  SchedulePendingCompaction(cfd);
  MaybeScheduleFlushOrCompaction();
L
Lei Jin 已提交
2949 2950 2951 2952 2953 2954 2955 2956 2957 2958

  // Update max_total_in_memory_state_
  max_total_in_memory_state_ =
      max_total_in_memory_state_ - old_memtable_size +
      mutable_cf_options.write_buffer_size *
      mutable_cf_options.max_write_buffer_number;
  return old;
}

Status DBImpl::GetImpl(const ReadOptions& read_options,
2959 2960
                       ColumnFamilyHandle* column_family, const Slice& key,
                       std::string* value, bool* value_found) {
L
Lei Jin 已提交
2961
  StopWatch sw(env_, stats_, DB_GET);
2962
  PERF_TIMER_GUARD(get_snapshot_time);
2963

2964 2965 2966
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

2967
  SequenceNumber snapshot;
L
Lei Jin 已提交
2968 2969 2970
  if (read_options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(
        read_options.snapshot)->number_;
2971 2972
  } else {
    snapshot = versions_->LastSequence();
J
jorlow@chromium.org 已提交
2973
  }
2974
  // Acquire SuperVersion
2975
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
2976
  // Prepare to store a list of merge operations if merge occurs.
2977
  MergeContext merge_context;
2978

2979
  Status s;
2980
  // First look in the memtable, then in the immutable memtable (if any).
2981
  // s is both in/out. When in, s could either be OK or MergeInProgress.
2982
  // merge_operands will contain the sequence of merges in the latter case.
2983
  LookupKey lkey(key, snapshot);
L
Lei Jin 已提交
2984
  PERF_TIMER_STOP(get_snapshot_time);
2985

2986
  if (sv->mem->Get(lkey, value, &s, &merge_context)) {
2987
    // Done
L
Lei Jin 已提交
2988
    RecordTick(stats_, MEMTABLE_HIT);
2989
  } else if (sv->imm->Get(lkey, value, &s, &merge_context)) {
2990
    // Done
L
Lei Jin 已提交
2991
    RecordTick(stats_, MEMTABLE_HIT);
2992
  } else {
2993
    PERF_TIMER_GUARD(get_from_output_files_time);
L
Lei Jin 已提交
2994 2995
    sv->current->Get(read_options, lkey, value, &s, &merge_context,
                     value_found);
L
Lei Jin 已提交
2996
    RecordTick(stats_, MEMTABLE_MISS);
2997
  }
2998

2999 3000
  {
    PERF_TIMER_GUARD(get_post_process_time);
3001

3002
    ReturnAndCleanupSuperVersion(cfd, sv);
3003

3004 3005 3006
    RecordTick(stats_, NUMBER_KEYS_READ);
    RecordTick(stats_, BYTES_READ, value->size());
  }
3007
  return s;
J
jorlow@chromium.org 已提交
3008 3009
}

3010
std::vector<Status> DBImpl::MultiGet(
L
Lei Jin 已提交
3011
    const ReadOptions& read_options,
3012
    const std::vector<ColumnFamilyHandle*>& column_family,
3013
    const std::vector<Slice>& keys, std::vector<std::string>* values) {
3014

L
Lei Jin 已提交
3015
  StopWatch sw(env_, stats_, DB_MULTIGET);
3016
  PERF_TIMER_GUARD(get_snapshot_time);
K
Kai Liu 已提交
3017

3018
  SequenceNumber snapshot;
3019

3020
  struct MultiGetColumnFamilyData {
I
Igor Canadi 已提交
3021
    ColumnFamilyData* cfd;
3022 3023 3024 3025 3026
    SuperVersion* super_version;
  };
  std::unordered_map<uint32_t, MultiGetColumnFamilyData*> multiget_cf_data;
  // fill up and allocate outside of mutex
  for (auto cf : column_family) {
3027 3028 3029 3030 3031 3032
    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});
3033 3034 3035
    }
  }

3036
  mutex_.Lock();
L
Lei Jin 已提交
3037 3038 3039
  if (read_options.snapshot != nullptr) {
    snapshot = reinterpret_cast<const SnapshotImpl*>(
        read_options.snapshot)->number_;
3040 3041 3042
  } else {
    snapshot = versions_->LastSequence();
  }
3043
  for (auto mgd_iter : multiget_cf_data) {
3044 3045
    mgd_iter.second->super_version =
        mgd_iter.second->cfd->GetSuperVersion()->Ref();
3046
  }
3047
  mutex_.Unlock();
3048

3049 3050
  // Contain a list of merge operations if merge occurs.
  MergeContext merge_context;
3051

3052
  // Note: this always resizes the values array
3053 3054 3055
  size_t num_keys = keys.size();
  std::vector<Status> stat_list(num_keys);
  values->resize(num_keys);
3056 3057

  // Keep track of bytes that we read for statistics-recording later
3058
  uint64_t bytes_read = 0;
L
Lei Jin 已提交
3059
  PERF_TIMER_STOP(get_snapshot_time);
3060 3061 3062 3063

  // 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.
3064
  // merge_operands will contain the sequence of merges in the latter case.
3065
  for (size_t i = 0; i < num_keys; ++i) {
3066
    merge_context.Clear();
3067
    Status& s = stat_list[i];
3068 3069 3070
    std::string* value = &(*values)[i];

    LookupKey lkey(keys[i], snapshot);
3071 3072
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family[i]);
    auto mgd_iter = multiget_cf_data.find(cfh->cfd()->GetID());
3073 3074 3075
    assert(mgd_iter != multiget_cf_data.end());
    auto mgd = mgd_iter->second;
    auto super_version = mgd->super_version;
3076
    if (super_version->mem->Get(lkey, value, &s, &merge_context)) {
3077
      // Done
3078
    } else if (super_version->imm->Get(lkey, value, &s, &merge_context)) {
3079 3080
      // Done
    } else {
3081
      PERF_TIMER_GUARD(get_from_output_files_time);
L
Lei Jin 已提交
3082 3083
      super_version->current->Get(read_options, lkey, value, &s,
                                  &merge_context);
3084 3085 3086
    }

    if (s.ok()) {
3087
      bytes_read += value->size();
3088 3089 3090 3091
    }
  }

  // Post processing (decrement reference counts and record statistics)
3092
  PERF_TIMER_GUARD(get_post_process_time);
3093 3094
  autovector<SuperVersion*> superversions_to_delete;

I
Igor Canadi 已提交
3095
  // TODO(icanadi) do we need lock here or just around Cleanup()?
3096 3097 3098 3099 3100 3101
  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);
3102 3103
    }
  }
3104 3105 3106 3107 3108 3109 3110
  mutex_.Unlock();

  for (auto td : superversions_to_delete) {
    delete td;
  }
  for (auto mgd : multiget_cf_data) {
    delete mgd.second;
3111
  }
3112

L
Lei Jin 已提交
3113 3114 3115
  RecordTick(stats_, NUMBER_MULTIGET_CALLS);
  RecordTick(stats_, NUMBER_MULTIGET_KEYS_READ, num_keys);
  RecordTick(stats_, NUMBER_MULTIGET_BYTES_READ, bytes_read);
L
Lei Jin 已提交
3116
  PERF_TIMER_STOP(get_post_process_time);
3117

3118
  return stat_list;
3119 3120
}

3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175
#ifndef ROCKSDB_LITE
Status DBImpl::AddFile(ColumnFamilyHandle* column_family,
                       const std::string& file_path, bool move_file) {
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  ColumnFamilyData* cfd = cfh->cfd();

  ExternalSstFileInfo file_info;
  file_info.file_path = file_path;
  status = env_->GetFileSize(file_path, &file_info.file_size);
  if (!status.ok()) {
    return status;
  }

  // Access the file using TableReader to extract
  // version, number of entries, smallest user key, largest user key
  std::unique_ptr<RandomAccessFile> sst_file;
  status = env_->NewRandomAccessFile(file_path, &sst_file, env_options_);
  if (!status.ok()) {
    return status;
  }
  std::unique_ptr<RandomAccessFileReader> sst_file_reader;
  sst_file_reader.reset(new RandomAccessFileReader(std::move(sst_file)));

  std::unique_ptr<TableReader> table_reader;
  status = cfd->ioptions()->table_factory->NewTableReader(
      TableReaderOptions(*cfd->ioptions(), env_options_,
                         cfd->internal_comparator()),
      std::move(sst_file_reader), file_info.file_size, &table_reader);
  if (!status.ok()) {
    return status;
  }

  // Get the external sst file version from table properties
  const UserCollectedProperties& user_collected_properties =
      table_reader->GetTableProperties()->user_collected_properties;
  UserCollectedProperties::const_iterator external_sst_file_version_iter =
      user_collected_properties.find(ExternalSstFilePropertyNames::kVersion);
  if (external_sst_file_version_iter == user_collected_properties.end()) {
    return Status::InvalidArgument("Generated table version not found");
  }

  file_info.version =
      DecodeFixed32(external_sst_file_version_iter->second.c_str());
  if (file_info.version == 1) {
    // version 1 imply that all sequence numbers in table equal 0
    file_info.sequence_number = 0;
  } else {
    return Status::InvalidArgument("Generated table version is not supported");
  }

  // Get number of entries in table
  file_info.num_entries = table_reader->GetTableProperties()->num_entries;

  ParsedInternalKey key;
S
sdong 已提交
3176 3177
  std::unique_ptr<InternalIterator> iter(
      table_reader->NewIterator(ReadOptions()));
3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336

  // Get first (smallest) key from file
  iter->SeekToFirst();
  if (!ParseInternalKey(iter->key(), &key)) {
    return Status::Corruption("Generated table have corrupted keys");
  }
  if (key.sequence != 0) {
    return Status::Corruption("Generated table have non zero sequence number");
  }
  file_info.smallest_key = key.user_key.ToString();

  // Get last (largest) key from file
  iter->SeekToLast();
  if (!ParseInternalKey(iter->key(), &key)) {
    return Status::Corruption("Generated table have corrupted keys");
  }
  if (key.sequence != 0) {
    return Status::Corruption("Generated table have non zero sequence number");
  }
  file_info.largest_key = key.user_key.ToString();

  return AddFile(column_family, &file_info, move_file);
}

Status DBImpl::AddFile(ColumnFamilyHandle* column_family,
                       const ExternalSstFileInfo* file_info, bool move_file) {
  Status status;
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  ColumnFamilyData* cfd = cfh->cfd();

  if (cfd->NumberLevels() <= 1) {
    return Status::NotSupported(
        "AddFile requires a database with at least 2 levels");
  }
  if (file_info->version != 1) {
    return Status::InvalidArgument("Generated table version is not supported");
  }
  // version 1 imply that file have only Put Operations with Sequence Number = 0

  FileMetaData meta;
  meta.smallest =
      InternalKey(file_info->smallest_key, file_info->sequence_number,
                  ValueType::kTypeValue);
  meta.largest = InternalKey(file_info->largest_key, file_info->sequence_number,
                             ValueType::kTypeValue);
  if (!meta.smallest.Valid() || !meta.largest.Valid()) {
    return Status::Corruption("Generated table have corrupted keys");
  }
  meta.smallest_seqno = file_info->sequence_number;
  meta.largest_seqno = file_info->sequence_number;
  if (meta.smallest_seqno != 0 || meta.largest_seqno != 0) {
    return Status::InvalidArgument(
        "Non zero sequence numbers are not supported");
  }
  // Generate a location for the new table
  meta.fd = FileDescriptor(versions_->NewFileNumber(), 0, file_info->file_size);
  std::string db_fname = TableFileName(
      db_options_.db_paths, meta.fd.GetNumber(), meta.fd.GetPathId());

  if (move_file) {
    status = env_->LinkFile(file_info->file_path, db_fname);
    if (status.IsNotSupported()) {
      // Original file is on a different FS, use copy instead of hard linking
      status = CopyFile(env_, file_info->file_path, db_fname, 0);
    }
  } else {
    status = CopyFile(env_, file_info->file_path, db_fname, 0);
  }
  if (!status.ok()) {
    return status;
  }

  {
    InstrumentedMutexLock l(&mutex_);
    const MutableCFOptions mutable_cf_options =
        *cfd->GetLatestMutableCFOptions();

    WriteThread::Writer w;
    write_thread_.EnterUnbatched(&w, &mutex_);

    // Make sure memtables are empty
    if (!cfd->mem()->IsEmpty() || cfd->imm()->NumNotFlushed() > 0) {
      // Cannot add the file since the keys in memtable
      // will hide the keys in file
      status = Status::NotSupported("Memtable is not empty");
    }

    // Make sure last sequence number is 0, if there are existing files then
    // they should have sequence number = 0
    if (status.ok() && versions_->LastSequence() > 0) {
      status = Status::NotSupported("Last Sequence number is not zero");
    }

    auto* vstorage = cfd->current()->storage_info();
    if (status.ok()) {
      // Make sure that the key range in the file we will add does not overlap
      // with previously added files
      Slice smallest_user_key = meta.smallest.user_key();
      Slice largest_user_key = meta.largest.user_key();
      for (int level = 0; level < vstorage->num_non_empty_levels(); level++) {
        if (vstorage->OverlapInLevel(level, &smallest_user_key,
                                     &largest_user_key)) {
          status = Status::NotSupported("Cannot add overlapping files");
          break;
        }
      }
    }

    if (status.ok()) {
      // We add the file to the last level
      int target_level = cfd->NumberLevels() - 1;
      if (cfd->ioptions()->level_compaction_dynamic_level_bytes == false) {
        // If we are using dynamic level compaction we add the file to
        // last level with files
        target_level = vstorage->num_non_empty_levels() - 1;
        if (target_level <= 0) {
          target_level = 1;
        }
      }
      VersionEdit edit;
      edit.SetColumnFamily(cfd->GetID());
      edit.AddFile(target_level, meta.fd.GetNumber(), meta.fd.GetPathId(),
                   meta.fd.GetFileSize(), meta.smallest, meta.largest,
                   meta.smallest_seqno, meta.largest_seqno,
                   meta.marked_for_compaction);

      status = versions_->LogAndApply(cfd, mutable_cf_options, &edit, &mutex_,
                                      directories_.GetDbDir());
    }
    write_thread_.ExitUnbatched(&w);

    if (status.ok()) {
      delete InstallSuperVersionAndScheduleWork(cfd, nullptr,
                                                mutable_cf_options);
    }
  }

  if (!status.ok()) {
    // We failed to add the file to the database
    Status s = env_->DeleteFile(db_fname);
    if (!s.ok()) {
      Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
          "AddFile() clean up for file %s failed : %s", db_fname.c_str(),
          s.ToString().c_str());
    }
  } else if (status.ok() && move_file) {
    // The file was moved and added successfully, remove original file link
    Status s = env_->DeleteFile(file_info->file_path);
    if (!s.ok()) {
      Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
          "%s was added to DB successfully but failed to remove original file "
          "link : %s",
          file_info->file_path.c_str(), s.ToString().c_str());
    }
  }
  return status;
}
#endif  // ROCKSDB_LITE

L
Lei Jin 已提交
3337
Status DBImpl::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
3338
                                  const std::string& column_family_name,
3339
                                  ColumnFamilyHandle** handle) {
Y
Yueh-Hsuan Chiang 已提交
3340
  Status s;
I
Igor Canadi 已提交
3341
  *handle = nullptr;
3342 3343 3344 3345 3346 3347

  s = CheckCompressionSupported(cf_options);
  if (!s.ok()) {
    return s;
  }

Y
Yueh-Hsuan Chiang 已提交
3348
  {
3349
    InstrumentedMutexLock l(&mutex_);
I
Igor Canadi 已提交
3350

Y
Yueh-Hsuan Chiang 已提交
3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364
    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
    Options opt(db_options_, cf_options);
I
Igor Canadi 已提交
3365
    {  // write thread
3366 3367
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
I
Igor Canadi 已提交
3368 3369 3370 3371
      // LogAndApply will both write the creation in MANIFEST and create
      // ColumnFamilyData object
      s = versions_->LogAndApply(
          nullptr, MutableCFOptions(opt, ImmutableCFOptions(opt)), &edit,
3372
          &mutex_, directories_.GetDbDir(), false, &cf_options);
3373
      write_thread_.ExitUnbatched(&w);
I
Igor Canadi 已提交
3374
    }
Y
Yueh-Hsuan Chiang 已提交
3375 3376 3377 3378 3379
    if (s.ok()) {
      single_column_family_mode_ = false;
      auto* cfd =
          versions_->GetColumnFamilySet()->GetColumnFamily(column_family_name);
      assert(cfd != nullptr);
I
Igor Canadi 已提交
3380
      delete InstallSuperVersionAndScheduleWork(
Y
Yueh-Hsuan Chiang 已提交
3381
          cfd, nullptr, *cfd->GetLatestMutableCFOptions());
S
sdong 已提交
3382 3383 3384 3385 3386

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

Y
Yueh-Hsuan Chiang 已提交
3387 3388 3389 3390 3391 3392 3393 3394 3395
      *handle = new ColumnFamilyHandleImpl(cfd, this, &mutex_);
      Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
          "Created column family [%s] (ID %u)",
          column_family_name.c_str(), (unsigned)cfd->GetID());
    } else {
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "Creating column family [%s] FAILED -- %s",
          column_family_name.c_str(), s.ToString().c_str());
    }
3396
  }  // InstrumentedMutexLock l(&mutex_)
Y
Yueh-Hsuan Chiang 已提交
3397 3398

  // this is outside the mutex
3399
  if (s.ok()) {
Y
Yueh-Hsuan Chiang 已提交
3400 3401
    NewThreadStatusCfInfo(
        reinterpret_cast<ColumnFamilyHandleImpl*>(*handle)->cfd());
3402
  }
3403
  return s;
3404 3405
}

3406 3407 3408 3409
Status DBImpl::DropColumnFamily(ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
  if (cfd->GetID() == 0) {
3410 3411
    return Status::InvalidArgument("Can't drop default column family");
  }
3412

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

I
Igor Canadi 已提交
3415 3416
  VersionEdit edit;
  edit.DropColumnFamily();
3417 3418
  edit.SetColumnFamily(cfd->GetID());

3419
  Status s;
3420
  {
3421
    InstrumentedMutexLock l(&mutex_);
3422 3423 3424 3425
    if (cfd->IsDropped()) {
      s = Status::InvalidArgument("Column family already dropped!\n");
    }
    if (s.ok()) {
3426
      // we drop column family from a single write thread
3427 3428
      WriteThread::Writer w;
      write_thread_.EnterUnbatched(&w, &mutex_);
3429 3430
      s = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
                                 &edit, &mutex_);
3431
      write_thread_.ExitUnbatched(&w);
3432
    }
S
sdong 已提交
3433 3434 3435 3436 3437 3438

    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()) {
3439
        if (!c->IsDropped() && !c->mem()->IsSnapshotSupported()) {
S
sdong 已提交
3440 3441 3442 3443 3444 3445
          new_is_snapshot_supported = false;
          break;
        }
      }
      is_snapshot_supported_ = new_is_snapshot_supported;
    }
3446
  }
3447

3448
  if (s.ok()) {
Y
Yueh-Hsuan Chiang 已提交
3449 3450 3451 3452
    // 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 已提交
3453
    assert(cfd->IsDropped());
L
Lei Jin 已提交
3454 3455 3456
    auto* mutable_cf_options = cfd->GetLatestMutableCFOptions();
    max_total_in_memory_state_ -= mutable_cf_options->write_buffer_size *
                                  mutable_cf_options->max_write_buffer_number;
3457 3458
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "Dropped column family with id %u\n",
3459
        cfd->GetID());
3460
  } else {
3461
    Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
3462
        "Dropping column family with id %u FAILED -- %s\n",
3463 3464 3465
        cfd->GetID(), s.ToString().c_str());
  }

3466
  return s;
3467 3468
}

L
Lei Jin 已提交
3469
bool DBImpl::KeyMayExist(const ReadOptions& read_options,
3470 3471
                         ColumnFamilyHandle* column_family, const Slice& key,
                         std::string* value, bool* value_found) {
3472
  if (value_found != nullptr) {
K
Kai Liu 已提交
3473 3474
    // falsify later if key-may-exist but can't fetch value
    *value_found = true;
3475
  }
L
Lei Jin 已提交
3476
  ReadOptions roptions = read_options;
3477
  roptions.read_tier = kBlockCacheTier; // read from block cache only
3478
  auto s = GetImpl(roptions, column_family, key, value, value_found);
K
Kai Liu 已提交
3479

3480
  // If block_cache is enabled and the index block of the table didn't
K
Kai Liu 已提交
3481 3482 3483
  // 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();
3484 3485
}

3486
Iterator* DBImpl::NewIterator(const ReadOptions& read_options,
3487 3488 3489
                              ColumnFamilyHandle* column_family) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
T
Tomislav Novak 已提交
3490

V
Venkatesh Radhakrishnan 已提交
3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508
  XFUNC_TEST("", "managed_new", managed_new1, xf_manage_new,
             reinterpret_cast<DBImpl*>(this),
             const_cast<ReadOptions*>(&read_options), is_snapshot_supported_);
  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 已提交
3509 3510 3511 3512
#ifdef ROCKSDB_LITE
    // not supported in lite version
    return nullptr;
#else
3513 3514
    SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
    auto iter = new ForwardIterator(this, read_options, cfd, sv);
3515
    return NewDBIterator(env_, *cfd->ioptions(), cfd->user_comparator(), iter,
3516 3517 3518
        kMaxSequenceNumber,
        sv->mutable_cf_options.max_sequential_skip_in_iterations,
        read_options.iterate_upper_bound);
I
Igor Canadi 已提交
3519
#endif
T
Tomislav Novak 已提交
3520
  } else {
3521
    SequenceNumber latest_snapshot = versions_->LastSequence();
3522
    SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
3523

I
Igor Canadi 已提交
3524
    auto snapshot =
3525 3526 3527
        read_options.snapshot != nullptr
            ? reinterpret_cast<const SnapshotImpl*>(
                read_options.snapshot)->number_
I
Igor Canadi 已提交
3528
            : latest_snapshot;
T
Tomislav Novak 已提交
3529

3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565
    // 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         | <----+
    // |       |   +-------------------+
    // |       |                       |
    // +-------+-----------------------+
    //
3566 3567
    // ArenaWrappedDBIter inlines an arena area where all the iterators in
    // the iterator tree are allocated in the order of being accessed when
3568 3569 3570 3571 3572
    // 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(
3573
        env_, *cfd->ioptions(), cfd->user_comparator(),
3574
        snapshot, sv->mutable_cf_options.max_sequential_skip_in_iterations,
3575
        read_options.iterate_upper_bound);
3576

S
sdong 已提交
3577
    InternalIterator* internal_iter =
3578
        NewInternalIterator(read_options, cfd, sv, db_iter->GetArena());
3579 3580 3581 3582
    db_iter->SetIterUnderDBIter(internal_iter);

    return db_iter;
  }
3583 3584
  // To stop compiler from complaining
  return nullptr;
J
jorlow@chromium.org 已提交
3585 3586
}

3587
Status DBImpl::NewIterators(
3588
    const ReadOptions& read_options,
I
Igor Canadi 已提交
3589
    const std::vector<ColumnFamilyHandle*>& column_families,
3590
    std::vector<Iterator*>* iterators) {
I
Igor Canadi 已提交
3591 3592
  iterators->clear();
  iterators->reserve(column_families.size());
V
Venkatesh Radhakrishnan 已提交
3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612
  XFUNC_TEST("", "managed_new", managed_new1, xf_manage_new,
             reinterpret_cast<DBImpl*>(this),
             const_cast<ReadOptions*>(&read_options), is_snapshot_supported_);
  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");
    }
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
      auto iter = new ManagedIterator(this, read_options, cfd);
      iterators->push_back(iter);
    }
#endif
  } else if (read_options.tailing) {
I
Igor Canadi 已提交
3613 3614 3615 3616
#ifdef ROCKSDB_LITE
    return Status::InvalidArgument(
        "Tailing interator not supported in RocksDB lite");
#else
I
Igor Canadi 已提交
3617 3618
    for (auto cfh : column_families) {
      auto cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(cfh)->cfd();
3619 3620
      SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
      auto iter = new ForwardIterator(this, read_options, cfd, sv);
L
Lei Jin 已提交
3621
      iterators->push_back(
3622
          NewDBIterator(env_, *cfd->ioptions(), cfd->user_comparator(), iter,
3623 3624
              kMaxSequenceNumber,
              sv->mutable_cf_options.max_sequential_skip_in_iterations));
I
Igor Canadi 已提交
3625
    }
I
Igor Canadi 已提交
3626
#endif
I
Igor Canadi 已提交
3627
  } else {
3628 3629
    SequenceNumber latest_snapshot = versions_->LastSequence();

I
Igor Canadi 已提交
3630
    for (size_t i = 0; i < column_families.size(); ++i) {
3631 3632 3633
      auto* cfd = reinterpret_cast<ColumnFamilyHandleImpl*>(
          column_families[i])->cfd();
      SuperVersion* sv = cfd->GetReferencedSuperVersion(&mutex_);
I
Igor Canadi 已提交
3634 3635

      auto snapshot =
3636 3637 3638
          read_options.snapshot != nullptr
              ? reinterpret_cast<const SnapshotImpl*>(
                  read_options.snapshot)->number_
I
Igor Canadi 已提交
3639 3640
              : latest_snapshot;

3641
      ArenaWrappedDBIter* db_iter = NewArenaWrappedDbIterator(
3642
          env_, *cfd->ioptions(), cfd->user_comparator(), snapshot,
3643
          sv->mutable_cf_options.max_sequential_skip_in_iterations);
S
sdong 已提交
3644 3645
      InternalIterator* internal_iter =
          NewInternalIterator(read_options, cfd, sv, db_iter->GetArena());
3646 3647
      db_iter->SetIterUnderDBIter(internal_iter);
      iterators->push_back(db_iter);
I
Igor Canadi 已提交
3648 3649 3650 3651
    }
  }

  return Status::OK();
3652 3653
}

J
jorlow@chromium.org 已提交
3654
const Snapshot* DBImpl::GetSnapshot() {
3655 3656
  int64_t unix_time = 0;
  env_->GetCurrentTime(&unix_time);  // Ignore error
3657
  SnapshotImpl* s = new SnapshotImpl;
3658

3659
  InstrumentedMutexLock l(&mutex_);
3660
  // returns null if the underlying memtable does not support snapshot.
3661 3662 3663 3664 3665
  if (!is_snapshot_supported_) {
    delete s;
    return nullptr;
  }
  return snapshots_.New(s, versions_->LastSequence(), unix_time);
J
jorlow@chromium.org 已提交
3666 3667 3668
}

void DBImpl::ReleaseSnapshot(const Snapshot* s) {
3669 3670 3671 3672 3673 3674
  const SnapshotImpl* casted_s = reinterpret_cast<const SnapshotImpl*>(s);
  {
    InstrumentedMutexLock l(&mutex_);
    snapshots_.Delete(casted_s);
  }
  delete casted_s;
J
jorlow@chromium.org 已提交
3675 3676 3677
}

// Convenience methods
3678 3679
Status DBImpl::Put(const WriteOptions& o, ColumnFamilyHandle* column_family,
                   const Slice& key, const Slice& val) {
3680
  return DB::Put(o, column_family, key, val);
J
jorlow@chromium.org 已提交
3681 3682
}

3683 3684 3685
Status DBImpl::Merge(const WriteOptions& o, ColumnFamilyHandle* column_family,
                     const Slice& key, const Slice& val) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
3686
  if (!cfh->cfd()->ioptions()->merge_operator) {
3687 3688
    return Status::NotSupported("Provide a merge_operator when opening DB");
  } else {
3689
    return DB::Merge(o, column_family, key, val);
3690 3691 3692
  }
}

L
Lei Jin 已提交
3693
Status DBImpl::Delete(const WriteOptions& write_options,
3694
                      ColumnFamilyHandle* column_family, const Slice& key) {
L
Lei Jin 已提交
3695
  return DB::Delete(write_options, column_family, key);
J
jorlow@chromium.org 已提交
3696 3697
}

A
Andres Noetzli 已提交
3698 3699 3700 3701 3702 3703
Status DBImpl::SingleDelete(const WriteOptions& write_options,
                            ColumnFamilyHandle* column_family,
                            const Slice& key) {
  return DB::SingleDelete(write_options, column_family, key);
}

L
Lei Jin 已提交
3704
Status DBImpl::Write(const WriteOptions& write_options, WriteBatch* my_batch) {
A
agiardullo 已提交
3705 3706 3707
  return WriteImpl(write_options, my_batch, nullptr);
}

A
agiardullo 已提交
3708
#ifndef ROCKSDB_LITE
A
agiardullo 已提交
3709 3710 3711 3712 3713
Status DBImpl::WriteWithCallback(const WriteOptions& write_options,
                                 WriteBatch* my_batch,
                                 WriteCallback* callback) {
  return WriteImpl(write_options, my_batch, callback);
}
A
agiardullo 已提交
3714
#endif  // ROCKSDB_LITE
A
agiardullo 已提交
3715 3716 3717

Status DBImpl::WriteImpl(const WriteOptions& write_options,
                         WriteBatch* my_batch, WriteCallback* callback) {
S
Stanislau Hlebik 已提交
3718 3719 3720
  if (my_batch == nullptr) {
    return Status::Corruption("Batch is nullptr!");
  }
3721 3722 3723
  if (write_options.timeout_hint_us != 0) {
    return Status::InvalidArgument("timeout_hint_us is deprecated");
  }
A
agiardullo 已提交
3724 3725

  Status status;
A
agiardullo 已提交
3726 3727
  bool callback_failed = false;

A
agiardullo 已提交
3728 3729 3730 3731 3732 3733 3734 3735 3736 3737
  bool xfunc_attempted_write = false;
  XFUNC_TEST("transaction", "transaction_xftest_write_impl",
             xf_transaction_write1, xf_transaction_write, write_options,
             db_options_, my_batch, callback, this, &status,
             &xfunc_attempted_write);
  if (xfunc_attempted_write) {
    // Test already did the write
    return status;
  }

3738
  PERF_TIMER_GUARD(write_pre_and_post_process_time);
3739
  WriteThread::Writer w;
S
Stanislau Hlebik 已提交
3740
  w.batch = my_batch;
L
Lei Jin 已提交
3741 3742
  w.sync = write_options.sync;
  w.disableWAL = write_options.disableWAL;
S
Stanislau Hlebik 已提交
3743 3744
  w.in_batch_group = false;
  w.done = false;
A
agiardullo 已提交
3745
  w.has_callback = (callback != nullptr) ? true : false;
S
Stanislau Hlebik 已提交
3746

L
Lei Jin 已提交
3747
  if (!write_options.disableWAL) {
S
Stanislau Hlebik 已提交
3748 3749 3750
    RecordTick(stats_, WRITE_WITH_WAL);
  }

3751 3752
  StopWatch write_sw(env_, db_options_.statistics.get(), DB_WRITE);

3753 3754 3755 3756 3757 3758 3759 3760
  write_thread_.JoinBatchGroup(&w);
  if (w.done) {
    // write was done by someone else, no need to grab mutex
    RecordTick(stats_, WRITE_DONE_BY_OTHER);
    return w.status;
  }
  // else we are the leader of the write batch group

S
Stanislau Hlebik 已提交
3761 3762
  WriteContext context;
  mutex_.Lock();
I
Igor Canadi 已提交
3763 3764 3765 3766 3767

  if (!write_options.disableWAL) {
    default_cf_internal_stats_->AddDBStats(InternalStats::WRITE_WITH_WAL, 1);
  }

S
Stanislau Hlebik 已提交
3768 3769
  RecordTick(stats_, WRITE_DONE_BY_SELF);
  default_cf_internal_stats_->AddDBStats(InternalStats::WRITE_DONE_BY_SELF, 1);
3770

3771 3772 3773 3774
  // Once reaches this point, the current writer "w" will try to do its write
  // job.  It may also pick up some of the remaining writers in the "writers_"
  // when it finds suitable, and finish them in the same write batch.
  // This is how a write job could be done by the other writer.
3775 3776 3777
  assert(!single_column_family_mode_ ||
         versions_->GetColumnFamilySet()->NumberOfColumnFamilies() == 1);

3778
  uint64_t max_total_wal_size = (db_options_.max_total_wal_size == 0)
3779
                                    ? 4 * max_total_in_memory_state_
3780
                                    : db_options_.max_total_wal_size;
3781
  if (UNLIKELY(!single_column_family_mode_) &&
3782
      alive_log_files_.begin()->getting_flushed == false &&
I
Igor Canadi 已提交
3783
      total_log_size_ > max_total_wal_size) {
I
Igor Canadi 已提交
3784
    uint64_t flush_column_family_if_log_file = alive_log_files_.begin()->number;
I
Igor Canadi 已提交
3785
    alive_log_files_.begin()->getting_flushed = true;
3786
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
3787 3788 3789
        "Flushing all column families with data in WAL number %" PRIu64
        ". Total log size is %" PRIu64 " while max_total_wal_size is %" PRIu64,
        flush_column_family_if_log_file, total_log_size_, max_total_wal_size);
I
Igor Canadi 已提交
3790 3791
    // no need to refcount because drop is happening in write thread, so can't
    // happen while we're in the write thread
3792
    for (auto cfd : *versions_->GetColumnFamilySet()) {
3793 3794 3795
      if (cfd->IsDropped()) {
        continue;
      }
I
Igor Canadi 已提交
3796
      if (cfd->GetLogNumber() <= flush_column_family_if_log_file) {
I
Igor Canadi 已提交
3797
        status = SwitchMemtable(cfd, &context);
I
Igor Canadi 已提交
3798 3799 3800
        if (!status.ok()) {
          break;
        }
S
Stanislau Hlebik 已提交
3801
        cfd->imm()->FlushRequested();
3802
        SchedulePendingFlush(cfd);
3803
      }
3804
    }
I
Igor Canadi 已提交
3805
    MaybeScheduleFlushOrCompaction();
3806 3807 3808 3809 3810 3811 3812 3813
  } else if (UNLIKELY(write_buffer_.ShouldFlush())) {
    Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
        "Flushing all column families. Write buffer is using %" PRIu64
        " bytes out of a total of %" PRIu64 ".",
        write_buffer_.memory_usage(), write_buffer_.buffer_size());
    // no need to refcount because drop is happening in write thread, so can't
    // happen while we're in the write thread
    for (auto cfd : *versions_->GetColumnFamilySet()) {
3814 3815 3816
      if (cfd->IsDropped()) {
        continue;
      }
3817
      if (!cfd->mem()->IsEmpty()) {
I
Igor Canadi 已提交
3818
        status = SwitchMemtable(cfd, &context);
3819 3820 3821 3822
        if (!status.ok()) {
          break;
        }
        cfd->imm()->FlushRequested();
3823
        SchedulePendingFlush(cfd);
3824 3825 3826
      }
    }
    MaybeScheduleFlushOrCompaction();
I
Igor Canadi 已提交
3827 3828 3829 3830 3831 3832 3833 3834 3835 3836
  }

  if (UNLIKELY(status.ok() && !bg_error_.ok())) {
    status = bg_error_;
  }

  if (UNLIKELY(status.ok() && !flush_scheduler_.Empty())) {
    status = ScheduleFlushes(&context);
  }

S
sdong 已提交
3837 3838
  if (UNLIKELY(status.ok()) &&
      (write_controller_.IsStopped() || write_controller_.NeedsDelay())) {
3839 3840
    PERF_TIMER_STOP(write_pre_and_post_process_time);
    PERF_TIMER_GUARD(write_delay_time);
S
sdong 已提交
3841 3842 3843 3844
    // We don't know size of curent batch so that we always use the size
    // for previous one. It might create a fairness issue that expiration
    // might happen for smaller writes but larger writes can go through.
    // Can optimize it if it is an issue.
3845
    status = DelayWrite(last_batch_group_size_);
3846
    PERF_TIMER_START(write_pre_and_post_process_time);
I
Igor Canadi 已提交
3847 3848
  }

D
dgrogan@chromium.org 已提交
3849
  uint64_t last_sequence = versions_->LastSequence();
I
Igor Canadi 已提交
3850
  WriteThread::Writer* last_writer = &w;
A
agiardullo 已提交
3851
  autovector<WriteBatch*> write_batch_group;
3852 3853
  bool need_log_sync = !write_options.disableWAL && write_options.sync;
  bool need_log_dir_sync = need_log_sync && !log_dir_synced_;
A
agiardullo 已提交
3854

S
Stanislau Hlebik 已提交
3855
  if (status.ok()) {
3856 3857
    last_batch_group_size_ = write_thread_.EnterAsBatchGroupLeader(
        &w, &last_writer, &write_batch_group);
3858

3859
    if (need_log_sync) {
3860 3861 3862 3863 3864 3865 3866 3867 3868
      while (logs_.front().getting_synced) {
        log_sync_cv_.Wait();
      }
      for (auto& log : logs_) {
        assert(!log.getting_synced);
        log.getting_synced = true;
      }
    }

3869 3870 3871
    // Add to log and apply to memtable.  We can release the lock
    // during this phase since &w is currently responsible for logging
    // and protects against concurrent loggers and concurrent writes
3872
    // into memtables
A
agiardullo 已提交
3873 3874 3875 3876 3877 3878 3879

    mutex_.Unlock();

    if (callback != nullptr) {
      // If this write has a validation callback, check to see if this write
      // is able to be written.  Must be called on the write thread.
      status = callback->Callback(this);
A
agiardullo 已提交
3880
      callback_failed = true;
A
agiardullo 已提交
3881 3882 3883 3884 3885 3886 3887 3888
    }
  } else {
    mutex_.Unlock();
  }

  // At this point the mutex is unlocked

  if (status.ok()) {
3889 3890 3891 3892 3893 3894 3895 3896 3897 3898
      WriteBatch* updates = nullptr;
      if (write_batch_group.size() == 1) {
        updates = write_batch_group[0];
      } else {
        updates = &tmp_batch_;
        for (size_t i = 0; i < write_batch_group.size(); ++i) {
          WriteBatchInternal::Append(updates, write_batch_group[i]);
        }
      }

3899 3900 3901 3902
      const SequenceNumber current_sequence = last_sequence + 1;
      WriteBatchInternal::SetSequence(updates, current_sequence);
      int my_batch_count = WriteBatchInternal::Count(updates);
      last_sequence += my_batch_count;
3903
      const uint64_t batch_size = WriteBatchInternal::ByteSize(updates);
3904
      // Record statistics
L
Lei Jin 已提交
3905
      RecordTick(stats_, NUMBER_KEYS_WRITTEN, my_batch_count);
S
sdong 已提交
3906
      RecordTick(stats_, BYTES_WRITTEN, batch_size);
L
Lei Jin 已提交
3907
      if (write_options.disableWAL) {
3908
        flush_on_destroy_ = true;
3909
      }
L
Lei Jin 已提交
3910
      PERF_TIMER_STOP(write_pre_and_post_process_time);
3911

3912
      uint64_t log_size = 0;
L
Lei Jin 已提交
3913
      if (!write_options.disableWAL) {
3914
        PERF_TIMER_GUARD(write_wal_time);
3915
        Slice log_entry = WriteBatchInternal::Contents(updates);
3916
        status = logs_.back().writer->AddRecord(log_entry);
I
Igor Canadi 已提交
3917 3918
        total_log_size_ += log_entry.size();
        alive_log_files_.back().AddSize(log_entry.size());
3919
        log_empty_ = false;
3920
        log_size = log_entry.size();
L
Lei Jin 已提交
3921
        RecordTick(stats_, WAL_FILE_BYTES, log_size);
3922
        if (status.ok() && need_log_sync) {
I
Igor Canadi 已提交
3923
          RecordTick(stats_, WAL_FILE_SYNCED);
3924
          StopWatch sw(env_, stats_, WAL_FILE_SYNC_MICROS);
3925 3926 3927 3928 3929 3930 3931 3932
          // It's safe to access logs_ with unlocked mutex_ here because:
          //  - we've set getting_synced=true for all logs,
          //    so other threads won't pop from logs_ while we're here,
          //  - only writer thread can push to logs_, and we're in
          //    writer thread, so no one will push to logs_,
          //  - as long as other threads don't modify it, it's safe to read
          //    from std::deque from multiple threads concurrently.
          for (auto& log : logs_) {
3933
            status = log.writer->file()->Sync(db_options_.use_fsync);
3934 3935 3936 3937
            if (!status.ok()) {
              break;
            }
          }
3938
          if (status.ok() && need_log_dir_sync) {
3939 3940 3941 3942 3943
            // We only sync WAL directory the first time WAL syncing is
            // requested, so that in case users never turn on WAL sync,
            // we can avoid the disk I/O in the write code path.
            status = directories_.GetWalDir()->Fsync();
          }
H
heyongqiang 已提交
3944
        }
3945 3946
      }
      if (status.ok()) {
3947
        PERF_TIMER_GUARD(write_memtable_time);
Y
Yueh-Hsuan Chiang 已提交
3948

3949
        status = WriteBatchInternal::InsertInto(
I
Igor Canadi 已提交
3950
            updates, column_family_memtables_.get(),
L
Lei Jin 已提交
3951
            write_options.ignore_missing_column_families, 0, this, false);
Y
Yueh-Hsuan Chiang 已提交
3952 3953 3954 3955 3956 3957 3958 3959
        // A non-OK status here indicates iteration failure (either in-memory
        // writebatch corruption (very bad), or the client specified invalid
        // column family).  This will later on trigger bg_error_.
        //
        // Note that existing logic was not sound. Any partial failure writing
        // into the memtable would result in a state that some write ops might
        // have succeeded in memtable but Status reports error for all writes.

L
Lei Jin 已提交
3960
        SetTickerCount(stats_, SEQUENCE_NUMBER, last_sequence);
3961
      }
L
Lei Jin 已提交
3962
      PERF_TIMER_START(write_pre_and_post_process_time);
3963 3964 3965
      if (updates == &tmp_batch_) {
        tmp_batch_.Clear();
      }
3966
      mutex_.Lock();
A
agiardullo 已提交
3967

3968 3969 3970
      // internal stats
      default_cf_internal_stats_->AddDBStats(
          InternalStats::BYTES_WRITTEN, batch_size);
S
sdong 已提交
3971 3972
      default_cf_internal_stats_->AddDBStats(InternalStats::NUMBER_KEYS_WRITTEN,
                                             my_batch_count);
L
Lei Jin 已提交
3973
      if (!write_options.disableWAL) {
3974 3975 3976 3977
        if (write_options.sync) {
          default_cf_internal_stats_->AddDBStats(InternalStats::WAL_FILE_SYNCED,
                                                 1);
        }
3978 3979 3980
        default_cf_internal_stats_->AddDBStats(
            InternalStats::WAL_FILE_BYTES, log_size);
      }
3981
      if (status.ok()) {
3982
        versions_->SetLastSequence(last_sequence);
3983
      }
A
agiardullo 已提交
3984 3985 3986
  } else {
    // Operation failed.  Make sure sure mutex is held for cleanup code below.
    mutex_.Lock();
3987
  }
A
agiardullo 已提交
3988

A
agiardullo 已提交
3989
  if (db_options_.paranoid_checks && !status.ok() && !callback_failed &&
3990
      !status.IsBusy() && bg_error_.ok()) {
I
Igor Canadi 已提交
3991 3992
    bg_error_ = status; // stop compaction & fail any further writes
  }
3993

A
agiardullo 已提交
3994
  mutex_.AssertHeld();
3995

3996 3997
  if (need_log_sync) {
    MarkLogsSynced(logfile_number_, need_log_dir_sync, status);
3998 3999
  }

4000 4001 4002 4003 4004 4005 4006 4007 4008
  uint64_t writes_for_other = write_batch_group.size() - 1;
  if (writes_for_other > 0) {
    default_cf_internal_stats_->AddDBStats(InternalStats::WRITE_DONE_BY_OTHER,
                                           writes_for_other);
    if (!write_options.disableWAL) {
      default_cf_internal_stats_->AddDBStats(InternalStats::WRITE_WITH_WAL,
                                             writes_for_other);
    }
  }
4009

I
Igor Canadi 已提交
4010
  mutex_.Unlock();
4011

4012 4013
  write_thread_.ExitAsBatchGroupLeader(&w, last_writer, status);

J
jorlow@chromium.org 已提交
4014 4015 4016
  return status;
}

4017 4018
// REQUIRES: mutex_ is held
// REQUIRES: this thread is currently at the front of the writer queue
4019
Status DBImpl::DelayWrite(uint64_t num_bytes) {
S
sdong 已提交
4020 4021 4022 4023
  uint64_t time_delayed = 0;
  bool delayed = false;
  {
    StopWatch sw(env_, stats_, WRITE_STALL, &time_delayed);
S
sdong 已提交
4024 4025
    auto delay = write_controller_.GetDelay(env_, num_bytes);
    if (delay > 0) {
S
sdong 已提交
4026 4027
      mutex_.Unlock();
      delayed = true;
S
sdong 已提交
4028
      TEST_SYNC_POINT("DBImpl::DelayWrite:Sleep");
4029 4030
      // hopefully we don't have to sleep more than 2 billion microseconds
      env_->SleepForMicroseconds(static_cast<int>(delay));
S
sdong 已提交
4031 4032
      mutex_.Lock();
    }
4033

4034
    while (bg_error_.ok() && write_controller_.IsStopped()) {
S
sdong 已提交
4035
      delayed = true;
4036 4037
      TEST_SYNC_POINT("DBImpl::DelayWrite:Wait");
      bg_cv_.Wait();
4038 4039
    }
  }
S
sdong 已提交
4040 4041 4042 4043 4044
  if (delayed) {
    default_cf_internal_stats_->AddDBStats(InternalStats::WRITE_STALL_MICROS,
                                           time_delayed);
    RecordTick(stats_, STALL_MICROS, time_delayed);
  }
I
Igor Canadi 已提交
4045 4046

  return bg_error_;
4047 4048
}

I
Igor Canadi 已提交
4049 4050 4051
Status DBImpl::ScheduleFlushes(WriteContext* context) {
  ColumnFamilyData* cfd;
  while ((cfd = flush_scheduler_.GetNextColumnFamily()) != nullptr) {
I
Igor Canadi 已提交
4052
    auto status = SwitchMemtable(cfd, context);
I
Igor Canadi 已提交
4053 4054 4055 4056 4057
    if (cfd->Unref()) {
      delete cfd;
    }
    if (!status.ok()) {
      return status;
S
Stanislau Hlebik 已提交
4058 4059
    }
  }
I
Igor Canadi 已提交
4060
  return Status::OK();
S
Stanislau Hlebik 已提交
4061 4062 4063 4064
}

// REQUIRES: mutex_ is held
// REQUIRES: this thread is currently at the front of the writer queue
I
Igor Canadi 已提交
4065
Status DBImpl::SwitchMemtable(ColumnFamilyData* cfd, WriteContext* context) {
S
Stanislau Hlebik 已提交
4066 4067 4068 4069 4070 4071 4072
  mutex_.AssertHeld();
  unique_ptr<WritableFile> lfile;
  log::Writer* new_log = nullptr;
  MemTable* new_mem = nullptr;

  // Attempt to switch to a new memtable and trigger flush of old.
  // Do this without holding the dbmutex lock.
4073
  assert(versions_->prev_log_number() == 0);
S
Stanislau Hlebik 已提交
4074 4075 4076 4077
  bool creating_new_log = !log_empty_;
  uint64_t new_log_number =
      creating_new_log ? versions_->NewFileNumber() : logfile_number_;
  SuperVersion* new_superversion = nullptr;
4078
  const MutableCFOptions mutable_cf_options = *cfd->GetLatestMutableCFOptions();
S
Stanislau Hlebik 已提交
4079 4080 4081 4082
  mutex_.Unlock();
  Status s;
  {
    if (creating_new_log) {
4083 4084
      EnvOptions opt_env_opt =
          env_->OptimizeForLogWrite(env_options_, db_options_);
S
sdong 已提交
4085 4086 4087
      s = NewWritableFile(env_,
                          LogFileName(db_options_.wal_dir, new_log_number),
                          &lfile, opt_env_opt);
S
Stanislau Hlebik 已提交
4088 4089 4090
      if (s.ok()) {
        // Our final size should be less than write_buffer_size
        // (compression, etc) but err on the side of caution.
L
Lei Jin 已提交
4091 4092
        lfile->SetPreallocationBlockSize(
            1.1 * mutable_cf_options.write_buffer_size);
4093 4094 4095
        unique_ptr<WritableFileWriter> file_writer(
            new WritableFileWriter(std::move(lfile), opt_env_opt));
        new_log = new log::Writer(std::move(file_writer));
I
Igor Canadi 已提交
4096
      }
S
Stanislau Hlebik 已提交
4097 4098 4099
    }

    if (s.ok()) {
A
agiardullo 已提交
4100 4101
      SequenceNumber seq = versions_->LastSequence();
      new_mem = cfd->ConstructNewMemtable(mutable_cf_options, seq);
S
Stanislau Hlebik 已提交
4102 4103 4104
      new_superversion = new SuperVersion();
    }
  }
4105
  Log(InfoLogLevel::DEBUG_LEVEL, db_options_.info_log,
I
Igor Canadi 已提交
4106 4107
      "[%s] New memtable created with log file: #%" PRIu64 "\n",
      cfd->GetName().c_str(), new_log_number);
S
Stanislau Hlebik 已提交
4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119
  mutex_.Lock();
  if (!s.ok()) {
    // how do we fail if we're not creating new log?
    assert(creating_new_log);
    assert(!new_mem);
    assert(!new_log);
    return s;
  }
  if (creating_new_log) {
    logfile_number_ = new_log_number;
    assert(new_log != nullptr);
    log_empty_ = true;
4120
    log_dir_synced_ = false;
4121
    logs_.emplace_back(logfile_number_, new_log);
S
Stanislau Hlebik 已提交
4122
    alive_log_files_.push_back(LogFileNumberSize(logfile_number_));
I
Igor Canadi 已提交
4123
    for (auto loop_cfd : *versions_->GetColumnFamilySet()) {
S
Stanislau Hlebik 已提交
4124 4125 4126 4127
      // all this is just optimization to delete logs that
      // are no longer needed -- if CF is empty, that means it
      // doesn't need that particular log to stay alive, so we just
      // advance the log number. no need to persist this in the manifest
I
Igor Canadi 已提交
4128
      if (loop_cfd->mem()->GetFirstSequenceNumber() == 0 &&
4129
          loop_cfd->imm()->NumNotFlushed() == 0) {
I
Igor Canadi 已提交
4130
        loop_cfd->SetLogNumber(logfile_number_);
4131
      }
4132 4133
    }
  }
S
Stanislau Hlebik 已提交
4134
  cfd->mem()->SetNextLogNumber(logfile_number_);
4135
  cfd->imm()->Add(cfd->mem(), &context->memtables_to_free_);
S
Stanislau Hlebik 已提交
4136 4137
  new_mem->Ref();
  cfd->SetMemtable(new_mem);
I
Igor Canadi 已提交
4138 4139
  context->superversions_to_free_.push_back(InstallSuperVersionAndScheduleWork(
      cfd, new_superversion, mutable_cf_options));
4140 4141 4142
  return s;
}

I
Igor Canadi 已提交
4143
#ifndef ROCKSDB_LITE
I
Igor Canadi 已提交
4144 4145 4146 4147 4148
Status DBImpl::GetPropertiesOfAllTables(ColumnFamilyHandle* column_family,
                                        TablePropertiesCollection* props) {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();

4149 4150
  // Increment the ref count
  mutex_.Lock();
I
Igor Canadi 已提交
4151
  auto version = cfd->current();
4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163
  version->Ref();
  mutex_.Unlock();

  auto s = version->GetPropertiesOfAllTables(props);

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

  return s;
}
4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182 4183 4184 4185 4186

Status DBImpl::GetPropertiesOfTablesInRange(ColumnFamilyHandle* column_family,
                                            const Range* range, int n,
                                            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 已提交
4187
#endif  // ROCKSDB_LITE
4188

I
Igor Canadi 已提交
4189 4190 4191 4192
const std::string& DBImpl::GetName() const {
  return dbname_;
}

4193 4194 4195 4196
Env* DBImpl::GetEnv() const {
  return env_;
}

4197 4198
const Options& DBImpl::GetOptions(ColumnFamilyHandle* column_family) const {
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
I
Igor Canadi 已提交
4199
  return *cfh->cfd()->options();
I
Igor Canadi 已提交
4200 4201
}

4202 4203
const DBOptions& DBImpl::GetDBOptions() const { return db_options_; }

4204
bool DBImpl::GetProperty(ColumnFamilyHandle* column_family,
4205
                         const Slice& property, std::string* value) {
4206 4207
  bool is_int_property = false;
  bool need_out_of_mutex = false;
4208 4209 4210
  DBPropertyType property_type =
      GetPropertyType(property, &is_int_property, &need_out_of_mutex);

4211
  value->clear();
4212 4213 4214 4215 4216
  if (is_int_property) {
    uint64_t int_value;
    bool ret_value = GetIntPropertyInternal(column_family, property_type,
                                            need_out_of_mutex, &int_value);
    if (ret_value) {
4217
      *value = ToString(int_value);
4218 4219 4220 4221 4222
    }
    return ret_value;
  } else {
    auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
    auto cfd = cfh->cfd();
4223
    InstrumentedMutexLock l(&mutex_);
4224 4225 4226
    return cfd->internal_stats()->GetStringProperty(property_type, property,
                                                    value);
  }
J
jorlow@chromium.org 已提交
4227 4228
}

4229 4230
bool DBImpl::GetIntProperty(ColumnFamilyHandle* column_family,
                            const Slice& property, uint64_t* value) {
4231 4232
  bool is_int_property = false;
  bool need_out_of_mutex = false;
4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244
  DBPropertyType property_type =
      GetPropertyType(property, &is_int_property, &need_out_of_mutex);
  if (!is_int_property) {
    return false;
  }
  return GetIntPropertyInternal(column_family, property_type, need_out_of_mutex,
                                value);
}

bool DBImpl::GetIntPropertyInternal(ColumnFamilyHandle* column_family,
                                    DBPropertyType property_type,
                                    bool need_out_of_mutex, uint64_t* value) {
4245 4246
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
4247 4248

  if (!need_out_of_mutex) {
4249
    InstrumentedMutexLock l(&mutex_);
4250
    return cfd->internal_stats()->GetIntProperty(property_type, value, this);
4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264
  } else {
    SuperVersion* sv = GetAndRefSuperVersion(cfd);

    bool ret = cfd->internal_stats()->GetIntPropertyOutOfMutex(
        property_type, sv->current, value);

    ReturnAndCleanupSuperVersion(cfd, sv);

    return ret;
  }
}

SuperVersion* DBImpl::GetAndRefSuperVersion(ColumnFamilyData* cfd) {
  // TODO(ljin): consider using GetReferencedSuperVersion() directly
I
Igor Canadi 已提交
4265
  return cfd->GetThreadLocalSuperVersion(&mutex_);
4266 4267
}

A
agiardullo 已提交
4268 4269 4270 4271 4272 4273 4274 4275 4276 4277 4278 4279
// 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);
}

A
agiardullo 已提交
4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295
// REQUIRED:  mutex is NOT held
SuperVersion* DBImpl::GetAndRefSuperVersionUnlocked(uint32_t column_family_id) {
  ColumnFamilyData* cfd;
  {
    InstrumentedMutexLock l(&mutex_);
    auto column_family_set = versions_->GetColumnFamilySet();
    cfd = column_family_set->GetColumnFamily(column_family_id);
  }

  if (!cfd) {
    return nullptr;
  }

  return GetAndRefSuperVersion(cfd);
}

4296 4297
void DBImpl::ReturnAndCleanupSuperVersion(ColumnFamilyData* cfd,
                                          SuperVersion* sv) {
I
Igor Canadi 已提交
4298
  bool unref_sv = !cfd->ReturnThreadLocalSuperVersion(sv);
4299 4300 4301 4302 4303

  if (unref_sv) {
    // Release SuperVersion
    if (sv->Unref()) {
      {
4304
        InstrumentedMutexLock l(&mutex_);
4305 4306 4307 4308 4309 4310 4311
        sv->Cleanup();
      }
      delete sv;
      RecordTick(stats_, NUMBER_SUPERVERSION_CLEANUPS);
    }
    RecordTick(stats_, NUMBER_SUPERVERSION_RELEASES);
  }
4312 4313
}

A
agiardullo 已提交
4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325
// 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);
}

A
agiardullo 已提交
4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341
// REQUIRED: Mutex should NOT be held.
void DBImpl::ReturnAndCleanupSuperVersionUnlocked(uint32_t column_family_id,
                                                  SuperVersion* sv) {
  ColumnFamilyData* cfd;
  {
    InstrumentedMutexLock l(&mutex_);
    auto column_family_set = versions_->GetColumnFamilySet();
    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);
}

A
agiardullo 已提交
4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353
// 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
agiardullo 已提交
4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367
// 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();
}

4368
void DBImpl::GetApproximateSizes(ColumnFamilyHandle* column_family,
4369 4370
                                 const Range* range, int n, uint64_t* sizes,
                                 bool include_memtable) {
J
jorlow@chromium.org 已提交
4371
  Version* v;
4372 4373
  auto cfh = reinterpret_cast<ColumnFamilyHandleImpl*>(column_family);
  auto cfd = cfh->cfd();
4374 4375
  SuperVersion* sv = GetAndRefSuperVersion(cfd);
  v = sv->current;
J
jorlow@chromium.org 已提交
4376 4377 4378

  for (int i = 0; i < n; i++) {
    // Convert user_key into a corresponding internal key.
4379 4380 4381
    InternalKey k1(range[i].start, kMaxSequenceNumber, kValueTypeForSeek);
    InternalKey k2(range[i].limit, kMaxSequenceNumber, kValueTypeForSeek);
    sizes[i] = versions_->ApproximateSize(v, k1.Encode(), k2.Encode());
4382 4383 4384 4385
    if (include_memtable) {
      sizes[i] += sv->mem->ApproximateSize(k1.Encode(), k2.Encode());
      sizes[i] += sv->imm->ApproximateSize(k1.Encode(), k2.Encode());
    }
J
jorlow@chromium.org 已提交
4386 4387
  }

4388
  ReturnAndCleanupSuperVersion(cfd, sv);
J
jorlow@chromium.org 已提交
4389 4390
}

I
Igor Canadi 已提交
4391 4392 4393 4394 4395 4396 4397 4398 4399 4400 4401 4402 4403 4404 4405 4406
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 已提交
4407 4408 4409 4410 4411
#ifndef ROCKSDB_LITE
Status DBImpl::GetUpdatesSince(
    SequenceNumber seq, unique_ptr<TransactionLogIterator>* iter,
    const TransactionLogIterator::ReadOptions& read_options) {

L
Lei Jin 已提交
4412
  RecordTick(stats_, GET_UPDATES_SINCE_CALLS);
I
Igor Canadi 已提交
4413 4414 4415
  if (seq > versions_->LastSequence()) {
    return Status::NotFound("Requested sequence not yet written in the db");
  }
I
Igor Canadi 已提交
4416
  return wal_manager_.GetUpdatesSince(seq, iter, read_options, versions_.get());
I
Igor Canadi 已提交
4417 4418
}

4419 4420 4421
Status DBImpl::DeleteFile(std::string name) {
  uint64_t number;
  FileType type;
4422 4423 4424
  WalFileType log_type;
  if (!ParseFileName(name, &number, &type, &log_type) ||
      (type != kTableFile && type != kLogFile)) {
4425 4426
    Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
        "DeleteFile %s failed.\n", name.c_str());
4427 4428 4429
    return Status::InvalidArgument("Invalid file name");
  }

4430 4431 4432 4433
  Status status;
  if (type == kLogFile) {
    // Only allow deleting archived log files
    if (log_type != kArchivedLogFile) {
4434 4435
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "DeleteFile %s failed - not archived log.\n",
4436
          name.c_str());
4437 4438
      return Status::NotSupported("Delete only supported for archived logs");
    }
4439
    status = env_->DeleteFile(db_options_.wal_dir + "/" + name.c_str());
4440
    if (!status.ok()) {
4441 4442
      Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
          "DeleteFile %s failed -- %s.\n",
4443
          name.c_str(), status.ToString().c_str());
4444 4445 4446 4447
    }
    return status;
  }

4448
  int level;
I
Igor Canadi 已提交
4449
  FileMetaData* metadata;
4450
  ColumnFamilyData* cfd;
4451
  VersionEdit edit;
4452
  JobContext job_context(next_job_id_.fetch_add(1), true);
D
Dhruba Borthakur 已提交
4453
  {
4454
    InstrumentedMutexLock l(&mutex_);
4455
    status = versions_->GetMetadataForFile(number, &level, &metadata, &cfd);
D
Dhruba Borthakur 已提交
4456
    if (!status.ok()) {
4457 4458
      Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
          "DeleteFile %s failed. File not found\n", name.c_str());
I
Igor Canadi 已提交
4459
      job_context.Clean();
D
Dhruba Borthakur 已提交
4460 4461
      return Status::InvalidArgument("File not found");
    }
4462
    assert(level < cfd->NumberLevels());
4463

D
Dhruba Borthakur 已提交
4464
    // If the file is being compacted no need to delete.
4465
    if (metadata->being_compacted) {
4466
      Log(InfoLogLevel::INFO_LEVEL, db_options_.info_log,
4467
          "DeleteFile %s Skipped. File about to be compacted\n", name.c_str());
I
Igor Canadi 已提交
4468
      job_context.Clean();
D
Dhruba Borthakur 已提交
4469
      return Status::OK();
4470 4471
    }

D
Dhruba Borthakur 已提交
4472 4473 4474
    // 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 已提交
4475
    auto* vstoreage = cfd->current()->storage_info();
I
Igor Canadi 已提交
4476
    for (int i = level + 1; i < cfd->NumberLevels(); i++) {
S
sdong 已提交
4477
      if (vstoreage->NumLevelFiles(i) != 0) {
4478
        Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
4479
            "DeleteFile %s FAILED. File not in last level\n", name.c_str());
I
Igor Canadi 已提交
4480
        job_context.Clean();
D
Dhruba Borthakur 已提交
4481 4482 4483
        return Status::InvalidArgument("File not in last level");
      }
    }
4484
    // if level == 0, it has to be the oldest file
S
sdong 已提交
4485 4486
    if (level == 0 &&
        vstoreage->LevelFiles(0).back()->fd.GetNumber() != number) {
4487 4488
      Log(InfoLogLevel::WARN_LEVEL, db_options_.info_log,
          "DeleteFile %s failed ---"
L
Lei Jin 已提交
4489
          " target file in level 0 must be the oldest.", name.c_str());
I
Igor Canadi 已提交
4490
      job_context.Clean();
4491 4492 4493
      return Status::InvalidArgument("File in level 0, but not oldest");
    }
    edit.SetColumnFamily(cfd->GetID());
D
Dhruba Borthakur 已提交
4494
    edit.DeleteFile(level, number);
4495
    status = versions_->LogAndApply(cfd, *cfd->GetLatestMutableCFOptions(),
4496
                                    &edit, &mutex_, directories_.GetDbDir());
I
Igor Canadi 已提交
4497
    if (status.ok()) {
I
Igor Canadi 已提交
4498 4499
      InstallSuperVersionAndScheduleWorkWrapper(
          cfd, &job_context, *cfd->GetLatestMutableCFOptions());
I
Igor Canadi 已提交
4500
    }
I
Igor Canadi 已提交
4501 4502
    FindObsoleteFiles(&job_context, false);
  }  // lock released here
4503

4504
  LogFlush(db_options_.info_log);
I
Igor Canadi 已提交
4505
  // remove files outside the db-lock
I
Igor Canadi 已提交
4506
  if (job_context.HaveSomethingToDelete()) {
4507
    // Call PurgeObsoleteFiles() without holding mutex.
I
Igor Canadi 已提交
4508
    PurgeObsoleteFiles(job_context);
4509
  }
I
Igor Canadi 已提交
4510
  job_context.Clean();
4511 4512 4513
  return status;
}

4514
void DBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
4515
  InstrumentedMutexLock l(&mutex_);
4516
  versions_->GetLiveFilesMetaData(metadata);
4517
}
4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528

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 已提交
4529
#endif  // ROCKSDB_LITE
4530

I
Igor Canadi 已提交
4531 4532 4533 4534 4535 4536 4537
Status DBImpl::CheckConsistency() {
  mutex_.AssertHeld();
  std::vector<LiveFileMetaData> metadata;
  versions_->GetLiveFilesMetaData(&metadata);

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

I
Igor Canadi 已提交
4541 4542
    uint64_t fsize = 0;
    Status s = env_->GetFileSize(file_path, &fsize);
D
dyniusz 已提交
4543 4544 4545 4546
    if (!s.ok() &&
        env_->GetFileSize(Rocks2LevelTableFileName(file_path), &fsize).ok()) {
      s = Status::OK();
    }
I
Igor Canadi 已提交
4547 4548 4549 4550
    if (!s.ok()) {
      corruption_messages +=
          "Can't access " + md.name + ": " + s.ToString() + "\n";
    } else if (fsize != md.size) {
4551
      corruption_messages += "Sst file size mismatch: " + file_path +
I
Igor Canadi 已提交
4552
                             ". Size recorded in manifest " +
4553 4554
                             ToString(md.size) + ", actual size " +
                             ToString(fsize) + "\n";
I
Igor Canadi 已提交
4555 4556 4557 4558 4559 4560 4561 4562 4563
    }
  }
  if (corruption_messages.size() == 0) {
    return Status::OK();
  } else {
    return Status::Corruption(corruption_messages);
  }
}

4564
Status DBImpl::GetDbIdentity(std::string& identity) const {
4565 4566
  std::string idfilename = IdentityFileName(dbname_);
  const EnvOptions soptions;
4567 4568 4569 4570 4571 4572 4573 4574 4575
  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)));
4576
  }
4577

4578 4579 4580 4581 4582
  uint64_t file_size;
  s = env_->GetFileSize(idfilename, &file_size);
  if (!s.ok()) {
    return s;
  }
D
Dmitri Smirnov 已提交
4583
  char* buffer = reinterpret_cast<char*>(alloca(file_size));
4584
  Slice id;
4585
  s = id_file_reader->Read(static_cast<size_t>(file_size), &id, buffer);
4586 4587 4588 4589 4590 4591 4592 4593 4594 4595 4596
  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;
}

J
jorlow@chromium.org 已提交
4597 4598
// Default implementations of convenience methods that subclasses of DB
// can call if they wish
4599
Status DB::Put(const WriteOptions& opt, ColumnFamilyHandle* column_family,
4600
               const Slice& key, const Slice& value) {
4601 4602 4603 4604
  // Pre-allocate size of write batch conservatively.
  // 8 bytes are taken by header, 4 bytes for count, 1 byte for type,
  // and we allocate 11 extra bytes for key length, as well as value length.
  WriteBatch batch(key.size() + value.size() + 24);
4605
  batch.Put(column_family, key, value);
J
jorlow@chromium.org 已提交
4606 4607 4608
  return Write(opt, &batch);
}

4609 4610
Status DB::Delete(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                  const Slice& key) {
J
jorlow@chromium.org 已提交
4611
  WriteBatch batch;
4612
  batch.Delete(column_family, key);
J
jorlow@chromium.org 已提交
4613 4614 4615
  return Write(opt, &batch);
}

A
Andres Noetzli 已提交
4616 4617 4618 4619 4620 4621 4622
Status DB::SingleDelete(const WriteOptions& opt,
                        ColumnFamilyHandle* column_family, const Slice& key) {
  WriteBatch batch;
  batch.SingleDelete(column_family, key);
  return Write(opt, &batch);
}

4623 4624
Status DB::Merge(const WriteOptions& opt, ColumnFamilyHandle* column_family,
                 const Slice& key, const Slice& value) {
4625
  WriteBatch batch;
4626
  batch.Merge(column_family, key, value);
4627 4628 4629
  return Write(opt, &batch);
}

4630
// Default implementation -- returns not supported status
L
Lei Jin 已提交
4631
Status DB::CreateColumnFamily(const ColumnFamilyOptions& cf_options,
4632
                              const std::string& column_family_name,
4633
                              ColumnFamilyHandle** handle) {
4634
  return Status::NotSupported("");
4635
}
4636
Status DB::DropColumnFamily(ColumnFamilyHandle* column_family) {
4637
  return Status::NotSupported("");
4638 4639
}

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

J
Jim Paton 已提交
4642
Status DB::Open(const Options& options, const std::string& dbname, DB** dbptr) {
4643 4644 4645 4646
  DBOptions db_options(options);
  ColumnFamilyOptions cf_options(options);
  std::vector<ColumnFamilyDescriptor> column_families;
  column_families.push_back(
4647
      ColumnFamilyDescriptor(kDefaultColumnFamilyName, cf_options));
4648
  std::vector<ColumnFamilyHandle*> handles;
4649
  Status s = DB::Open(db_options, dbname, column_families, &handles, dbptr);
4650 4651 4652 4653 4654 4655 4656
  if (s.ok()) {
    assert(handles.size() == 1);
    // i can delete the handle since DBImpl is always holding a reference to
    // default column family
    delete handles[0];
  }
  return s;
4657 4658
}

4659 4660
Status DB::Open(const DBOptions& db_options, const std::string& dbname,
                const std::vector<ColumnFamilyDescriptor>& column_families,
4661
                std::vector<ColumnFamilyHandle*>* handles, DB** dbptr) {
4662
  Status s = SanitizeOptionsByTable(db_options, column_families);
4663 4664 4665
  if (!s.ok()) {
    return s;
  }
4666

4667 4668 4669 4670 4671 4672
  for (auto& cfd : column_families) {
    s = CheckCompressionSupported(cfd.options);
    if (!s.ok()) {
      return s;
    }
    if (db_options.db_paths.size() > 1) {
4673 4674
      if ((cfd.options.compaction_style != kCompactionStyleUniversal) &&
          (cfd.options.compaction_style != kCompactionStyleLevel)) {
4675 4676
        return Status::NotSupported(
            "More than one DB paths are only supported in "
4677
            "universal and level compaction styles. ");
4678 4679
      }
    }
4680
  }
4681

4682 4683 4684
  if (db_options.db_paths.size() > 4) {
    return Status::NotSupported(
        "More than four DB paths are not supported yet. ");
4685 4686
  }

4687
  *dbptr = nullptr;
4688
  handles->clear();
J
jorlow@chromium.org 已提交
4689

I
Igor Canadi 已提交
4690 4691 4692 4693
  size_t max_write_buffer_size = 0;
  for (auto cf : column_families) {
    max_write_buffer_size =
        std::max(max_write_buffer_size, cf.options.write_buffer_size);
4694
  }
4695

I
Igor Canadi 已提交
4696
  DBImpl* impl = new DBImpl(db_options, dbname);
4697
  s = impl->env_->CreateDirIfMissing(impl->db_options_.wal_dir);
4698
  if (s.ok()) {
4699
    for (auto db_path : impl->db_options_.db_paths) {
4700
      s = impl->env_->CreateDirIfMissing(db_path.path);
4701 4702 4703 4704 4705 4706
      if (!s.ok()) {
        break;
      }
    }
  }

4707 4708 4709 4710 4711 4712
  if (!s.ok()) {
    delete impl;
    return s;
  }

  s = impl->CreateArchivalDirectory();
4713 4714 4715 4716
  if (!s.ok()) {
    delete impl;
    return s;
  }
J
jorlow@chromium.org 已提交
4717
  impl->mutex_.Lock();
4718 4719
  // Handles create_if_missing, error_if_exists
  s = impl->Recover(column_families);
J
jorlow@chromium.org 已提交
4720
  if (s.ok()) {
4721
    uint64_t new_log_number = impl->versions_->NewFileNumber();
4722
    unique_ptr<WritableFile> lfile;
4723
    EnvOptions soptions(db_options);
4724 4725
    EnvOptions opt_env_options =
        impl->db_options_.env->OptimizeForLogWrite(soptions, impl->db_options_);
S
sdong 已提交
4726 4727 4728
    s = NewWritableFile(impl->db_options_.env,
                        LogFileName(impl->db_options_.wal_dir, new_log_number),
                        &lfile, opt_env_options);
J
jorlow@chromium.org 已提交
4729
    if (s.ok()) {
4730
      lfile->SetPreallocationBlockSize(1.1 * max_write_buffer_size);
4731
      impl->logfile_number_ = new_log_number;
4732 4733
      unique_ptr<WritableFileWriter> file_writer(
          new WritableFileWriter(std::move(lfile), opt_env_options));
4734 4735
      impl->logs_.emplace_back(new_log_number,
                               new log::Writer(std::move(file_writer)));
I
Igor Canadi 已提交
4736

4737 4738
      // set column family handles
      for (auto cf : column_families) {
I
Igor Canadi 已提交
4739 4740
        auto cfd =
            impl->versions_->GetColumnFamilySet()->GetColumnFamily(cf.name);
I
Igor Canadi 已提交
4741 4742 4743
        if (cfd != nullptr) {
          handles->push_back(
              new ColumnFamilyHandleImpl(cfd, impl, &impl->mutex_));
4744
          impl->NewThreadStatusCfInfo(cfd);
I
Igor Canadi 已提交
4745 4746 4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760
        } else {
          if (db_options.create_missing_column_families) {
            // missing column family, create it
            ColumnFamilyHandle* handle;
            impl->mutex_.Unlock();
            s = impl->CreateColumnFamily(cf.options, cf.name, &handle);
            impl->mutex_.Lock();
            if (s.ok()) {
              handles->push_back(handle);
            } else {
              break;
            }
          } else {
            s = Status::InvalidArgument("Column family not found: ", cf.name);
            break;
          }
I
Igor Canadi 已提交
4761
        }
4762
      }
I
Igor Canadi 已提交
4763 4764
    }
    if (s.ok()) {
4765
      for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
I
Igor Canadi 已提交
4766
        delete impl->InstallSuperVersionAndScheduleWork(
L
Lei Jin 已提交
4767
            cfd, nullptr, *cfd->GetLatestMutableCFOptions());
4768
      }
I
Igor Canadi 已提交
4769 4770
      impl->alive_log_files_.push_back(
          DBImpl::LogFileNumberSize(impl->logfile_number_));
J
jorlow@chromium.org 已提交
4771
      impl->DeleteObsoleteFiles();
4772
      s = impl->directories_.GetDbDir()->Fsync();
J
jorlow@chromium.org 已提交
4773 4774
    }
  }
4775

I
Igor Canadi 已提交
4776 4777
  if (s.ok()) {
    for (auto cfd : *impl->versions_->GetColumnFamilySet()) {
S
sdong 已提交
4778
      if (cfd->ioptions()->compaction_style == kCompactionStyleFIFO) {
S
sdong 已提交
4779
        auto* vstorage = cfd->current()->storage_info();
4780
        for (int i = 1; i < vstorage->num_levels(); ++i) {
S
sdong 已提交
4781
          int num_files = vstorage->NumLevelFiles(i);
I
Igor Canadi 已提交
4782
          if (num_files > 0) {
I
Igor Canadi 已提交
4783 4784
            s = Status::InvalidArgument(
                "Not all files are at level 0. Cannot "
S
sdong 已提交
4785
                "open with FIFO compaction style.");
I
Igor Canadi 已提交
4786 4787 4788 4789
            break;
          }
        }
      }
S
sdong 已提交
4790 4791 4792
      if (!cfd->mem()->IsSnapshotSupported()) {
        impl->is_snapshot_supported_ = false;
      }
4793
      if (cfd->ioptions()->merge_operator != nullptr &&
4794 4795 4796 4797 4798
          !cfd->mem()->IsMergeOperatorSupported()) {
        s = Status::InvalidArgument(
            "The memtable of column family %s does not support merge operator "
            "its options.merge_operator is non-null", cfd->GetName().c_str());
      }
I
Igor Canadi 已提交
4799
      if (!s.ok()) {
4800 4801 4802 4803
        break;
      }
    }
  }
4804 4805 4806 4807 4808
  TEST_SYNC_POINT("DBImpl::Open:Opened");
  if (s.ok()) {
    impl->opened_successfully_ = true;
    impl->MaybeScheduleFlushOrCompaction();
  }
4809 4810
  impl->mutex_.Unlock();

J
jorlow@chromium.org 已提交
4811
  if (s.ok()) {
S
sdong 已提交
4812 4813
    Log(InfoLogLevel::INFO_LEVEL, impl->db_options_.info_log, "DB pointer %p",
        impl);
4814 4815
    LogFlush(impl->db_options_.info_log);

J
jorlow@chromium.org 已提交
4816 4817
    *dbptr = impl;
  } else {
Y
Yueh-Hsuan Chiang 已提交
4818
    for (auto* h : *handles) {
4819 4820
      delete h;
    }
4821
    handles->clear();
J
jorlow@chromium.org 已提交
4822 4823 4824 4825 4826
    delete impl;
  }
  return s;
}

4827 4828 4829
Status DB::ListColumnFamilies(const DBOptions& db_options,
                              const std::string& name,
                              std::vector<std::string>* column_families) {
I
Igor Canadi 已提交
4830
  return VersionSet::ListColumnFamilies(column_families, name, db_options.env);
4831 4832
}

4833 4834 4835
Snapshot::~Snapshot() {
}

J
jorlow@chromium.org 已提交
4836
Status DestroyDB(const std::string& dbname, const Options& options) {
4837
  const InternalKeyComparator comparator(options.comparator);
4838
  const Options& soptions(SanitizeOptions(dbname, &comparator, options));
4839
  Env* env = soptions.env;
J
jorlow@chromium.org 已提交
4840
  std::vector<std::string> filenames;
4841

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

J
jorlow@chromium.org 已提交
4845
  FileLock* lock;
4846 4847
  const std::string lockname = LockFileName(dbname);
  Status result = env->LockFile(lockname, &lock);
J
jorlow@chromium.org 已提交
4848 4849 4850
  if (result.ok()) {
    uint64_t number;
    FileType type;
4851
    InfoLogPrefix info_log_prefix(!options.db_log_dir.empty(), dbname);
D
dgrogan@chromium.org 已提交
4852
    for (size_t i = 0; i < filenames.size(); i++) {
4853
      if (ParseFileName(filenames[i], &number, info_log_prefix.prefix, &type) &&
4854
          type != kDBLockFile) {  // Lock file will be deleted at end
K
Kosie van der Merwe 已提交
4855
        Status del;
4856
        std::string path_to_delete = dbname + "/" + filenames[i];
K
Kosie van der Merwe 已提交
4857
        if (type == kMetaDatabase) {
4858 4859 4860
          del = DestroyDB(path_to_delete, options);
        } else if (type == kTableFile) {
          del = DeleteOrMoveToTrash(&options, path_to_delete);
K
Kosie van der Merwe 已提交
4861
        } else {
4862
          del = env->DeleteFile(path_to_delete);
K
Kosie van der Merwe 已提交
4863
        }
J
jorlow@chromium.org 已提交
4864 4865 4866 4867 4868
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4869

4870 4871
    for (size_t path_id = 0; path_id < options.db_paths.size(); path_id++) {
      const auto& db_path = options.db_paths[path_id];
4872
      env->GetChildren(db_path.path, &filenames);
4873 4874 4875
      for (size_t i = 0; i < filenames.size(); i++) {
        if (ParseFileName(filenames[i], &number, &type) &&
            type == kTableFile) {  // Lock file will be deleted at end
4876 4877 4878 4879 4880 4881 4882
          Status del;
          std::string table_path = db_path.path + "/" + filenames[i];
          if (path_id == 0) {
            del = DeleteOrMoveToTrash(&options, table_path);
          } else {
            del = env->DeleteFile(table_path);
          }
4883 4884 4885 4886 4887 4888 4889
          if (result.ok() && !del.ok()) {
            result = del;
          }
        }
      }
    }

I
Igor Canadi 已提交
4890 4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907
    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;
4908
    env->GetChildren(archivedir, &archiveFiles);
4909 4910
    // Delete archival files.
    for (size_t i = 0; i < archiveFiles.size(); ++i) {
4911 4912
      if (ParseFileName(archiveFiles[i], &number, &type) &&
          type == kLogFile) {
4913
        Status del = env->DeleteFile(archivedir + "/" + archiveFiles[i]);
4914 4915 4916 4917 4918
        if (result.ok() && !del.ok()) {
          result = del;
        }
      }
    }
4919
    // ignore case where no archival directory is present.
4920
    env->DeleteDir(archivedir);
4921

J
jorlow@chromium.org 已提交
4922
    env->UnlockFile(lock);  // Ignore error since state is already gone
4923
    env->DeleteFile(lockname);
J
jorlow@chromium.org 已提交
4924
    env->DeleteDir(dbname);  // Ignore error in case dir contains other files
4925
    env->DeleteDir(soptions.wal_dir);
J
jorlow@chromium.org 已提交
4926 4927 4928 4929
  }
  return result;
}

Y
Yueh-Hsuan Chiang 已提交
4930
#if ROCKSDB_USING_THREAD_STATUS
4931

Y
Yueh-Hsuan Chiang 已提交
4932 4933
void DBImpl::NewThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
4934
  if (db_options_.enable_thread_tracking) {
4935
    ThreadStatusUtil::NewColumnFamilyInfo(this, cfd);
4936
  }
Y
Yueh-Hsuan Chiang 已提交
4937 4938 4939 4940
}

void DBImpl::EraseThreadStatusCfInfo(
    ColumnFamilyData* cfd) const {
4941
  if (db_options_.enable_thread_tracking) {
4942
    ThreadStatusUtil::EraseColumnFamilyInfo(cfd);
4943
  }
Y
Yueh-Hsuan Chiang 已提交
4944 4945 4946
}

void DBImpl::EraseThreadStatusDbInfo() const {
4947
  if (db_options_.enable_thread_tracking) {
4948
    ThreadStatusUtil::EraseDatabaseInfo(this);
4949
  }
Y
Yueh-Hsuan Chiang 已提交
4950 4951 4952 4953 4954 4955 4956 4957 4958 4959 4960 4961 4962 4963 4964
}

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

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

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

4965 4966
//
// A global method that can dump out the build version
I
Igor Canadi 已提交
4967
void DumpRocksDBBuildVersion(Logger * log) {
I
Igor Canadi 已提交
4968
#if !defined(IOS_CROSS_COMPILE)
I
Igor Canadi 已提交
4969 4970
  // if we compile with Xcode, we don't run build_detect_vesion, so we don't
  // generate util/build_version.cc
4971 4972 4973 4974
  Header(log, "RocksDB version: %d.%d.%d\n", ROCKSDB_MAJOR, ROCKSDB_MINOR,
         ROCKSDB_PATCH);
  Header(log, "Git sha %s", rocksdb_build_git_sha);
  Header(log, "Compile date %s", rocksdb_build_compile_date);
I
Igor Canadi 已提交
4975
#endif
4976 4977
}

A
agiardullo 已提交
4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990 4991 4992 4993 4994 4995 4996 4997 4998 4999 5000 5001 5002 5003 5004 5005 5006 5007 5008 5009 5010 5011 5012 5013 5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034 5035 5036 5037 5038 5039 5040 5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052 5053 5054 5055 5056
#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
Status DBImpl::GetLatestSequenceForKeyFromMemtable(SuperVersion* sv,
                                                   const Slice& key,
                                                   SequenceNumber* seq) {
  Status s;
  std::string value;
  MergeContext merge_context;

  SequenceNumber current_seq = versions_->LastSequence();
  LookupKey lkey(key, current_seq);

  *seq = kMaxSequenceNumber;

  // Check if there is a record for this key in the latest memtable
  sv->mem->Get(lkey, &value, &s, &merge_context, seq);

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
    Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
        "Unexpected status returned from MemTable::Get: %s\n",
        s.ToString().c_str());

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check immutable memtables
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
  sv->imm->Get(lkey, &value, &s, &merge_context, seq);

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
    Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
        "Unexpected status returned from MemTableList::Get: %s\n",
        s.ToString().c_str());

    return s;
  }

  if (*seq != kMaxSequenceNumber) {
    // Found a sequence number, no need to check memtable history
    return Status::OK();
  }

  // Check if there is a record for this key in the immutable memtables
  sv->imm->GetFromHistory(lkey, &value, &s, &merge_context, seq);

  if (!(s.ok() || s.IsNotFound() || s.IsMergeInProgress())) {
    // unexpected error reading memtable.
    Log(InfoLogLevel::ERROR_LEVEL, db_options_.info_log,
        "Unexpected status returned from MemTableList::GetFromHistory: %s\n",
        s.ToString().c_str());

    return s;
  }

  return Status::OK();
}
#endif  // ROCKSDB_LITE

5057
}  // namespace rocksdb