ceph/src/rocksdb/utilities/blob_db/blob_db_impl_filesnapshot.cc

   1 //  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
   2 //  This source code is licensed under both the GPLv2 (found in the
   3 //  COPYING file in the root directory) and Apache 2.0 License
   4 //  (found in the LICENSE.Apache file in the root directory).
   5
   6 #ifndef ROCKSDB_LITE
   7
   8 #include "file/filename.h"
   9 #include "logging/logging.h"
  10 #include "util/cast_util.h"
  11 #include "util/mutexlock.h"
  12 #include "utilities/blob_db/blob_db_impl.h"
  13
  14 // BlobDBImpl methods to get snapshot of files, e.g. for replication.
  15
  16 namespace ROCKSDB_NAMESPACE {
  17 namespace blob_db {
  18
  19 Status BlobDBImpl::DisableFileDeletions() {
  20   // Disable base DB file deletions.
  21   Status s = db_impl_->DisableFileDeletions();
  22   if (!s.ok()) {
  23     return s;
  24   }
  25
  26   int count = 0;
  27   {
  28     // Hold delete_file_mutex_ to make sure no DeleteObsoleteFiles job
  29     // is running.
  30     MutexLock l(&delete_file_mutex_);
  31     count = ++disable_file_deletions_;
  32   }
  33
  34   ROCKS_LOG_INFO(db_options_.info_log,
  35                  "Disabled blob file deletions. count: %d", count);
  36   return Status::OK();
  37 }
  38
  39 Status BlobDBImpl::EnableFileDeletions(bool force) {
  40   // Enable base DB file deletions.
  41   Status s = db_impl_->EnableFileDeletions(force);
  42   if (!s.ok()) {
  43     return s;
  44   }
  45
  46   int count = 0;
  47   {
  48     MutexLock l(&delete_file_mutex_);
  49     if (force) {
  50       disable_file_deletions_ = 0;
  51     } else if (disable_file_deletions_ > 0) {
  52       count = --disable_file_deletions_;
  53     }
  54     assert(count >= 0);
  55   }
  56
  57   ROCKS_LOG_INFO(db_options_.info_log, "Enabled blob file deletions. count: %d",
  58                  count);
  59   // Consider trigger DeleteobsoleteFiles once after re-enabled, if we are to
  60   // make DeleteobsoleteFiles re-run interval configuration.
  61   return Status::OK();
  62 }
  63
  64 Status BlobDBImpl::GetLiveFiles(std::vector<std::string>& ret,
  65                                 uint64_t* manifest_file_size,
  66                                 bool flush_memtable) {
  67   if (!bdb_options_.path_relative) {
  68     return Status::NotSupported(
  69         "Not able to get relative blob file path from absolute blob_dir.");
  70   }
  71   // Hold a lock in the beginning to avoid updates to base DB during the call
  72   ReadLock rl(&mutex_);
  73   Status s = db_->GetLiveFiles(ret, manifest_file_size, flush_memtable);
  74   if (!s.ok()) {
  75     return s;
  76   }
  77   ret.reserve(ret.size() + blob_files_.size());
  78   for (auto bfile_pair : blob_files_) {
  79     auto blob_file = bfile_pair.second;
  80     // Path should be relative to db_name, but begin with slash.
  81     ret.emplace_back(
  82         BlobFileName("", bdb_options_.blob_dir, blob_file->BlobFileNumber()));
  83   }
  84   return Status::OK();
  85 }
  86
  87 void BlobDBImpl::GetLiveFilesMetaData(std::vector<LiveFileMetaData>* metadata) {
  88   // Path should be relative to db_name.
  89   assert(bdb_options_.path_relative);
  90   // Hold a lock in the beginning to avoid updates to base DB during the call
  91   ReadLock rl(&mutex_);
  92   db_->GetLiveFilesMetaData(metadata);
  93   for (auto bfile_pair : blob_files_) {
  94     auto blob_file = bfile_pair.second;
  95     LiveFileMetaData filemetadata;
  96     filemetadata.size = blob_file->GetFileSize();
  97     const uint64_t file_number = blob_file->BlobFileNumber();
  98     // Path should be relative to db_name, but begin with slash.
  99     filemetadata.name = BlobFileName("", bdb_options_.blob_dir, file_number);
 100     filemetadata.file_number = file_number;
 101     if (blob_file->HasTTL()) {
 102       filemetadata.oldest_ancester_time = blob_file->GetExpirationRange().first;
 103     }
 104     auto cfh =
 105         static_cast_with_check<ColumnFamilyHandleImpl>(DefaultColumnFamily());
 106     filemetadata.column_family_name = cfh->GetName();
 107     metadata->emplace_back(filemetadata);
 108   }
 109 }
 110
 111 }  // namespace blob_db
 112 }  // namespace ROCKSDB_NAMESPACE
 113 #endif  // !ROCKSDB_LITE