2016-02-09 23:12:00 +00:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-15 23:03:42 +00:00
|
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
|
|
// (found in the LICENSE.Apache file in the root directory).
|
2014-11-14 19:38:26 +00:00
|
|
|
//
|
2019-05-30 03:44:08 +00:00
|
|
|
#include "file/file_util.h"
|
2015-08-19 22:02:17 +00:00
|
|
|
|
2014-11-14 19:38:26 +00:00
|
|
|
#include <string>
|
|
|
|
#include <algorithm>
|
2015-08-19 22:02:17 +00:00
|
|
|
|
2019-09-16 17:31:27 +00:00
|
|
|
#include "file/random_access_file_reader.h"
|
|
|
|
#include "file/sequence_file_reader.h"
|
2019-05-30 03:44:08 +00:00
|
|
|
#include "file/sst_file_manager_impl.h"
|
2019-09-16 17:31:27 +00:00
|
|
|
#include "file/writable_file_writer.h"
|
2014-11-14 19:38:26 +00:00
|
|
|
#include "rocksdb/env.h"
|
|
|
|
|
|
|
|
namespace rocksdb {
|
|
|
|
|
|
|
|
// Utility function to copy a file up to a specified length
|
|
|
|
Status CopyFile(Env* env, const std::string& source,
|
2016-12-29 02:38:20 +00:00
|
|
|
const std::string& destination, uint64_t size, bool use_fsync) {
|
2014-11-14 19:38:26 +00:00
|
|
|
const EnvOptions soptions;
|
|
|
|
Status s;
|
2018-11-09 19:17:34 +00:00
|
|
|
std::unique_ptr<SequentialFileReader> src_reader;
|
|
|
|
std::unique_ptr<WritableFileWriter> dest_writer;
|
Move rate_limiter, write buffering, most perf context instrumentation and most random kill out of Env
Summary: We want to keep Env a think layer for better portability. Less platform dependent codes should be moved out of Env. In this patch, I create a wrapper of file readers and writers, and put rate limiting, write buffering, as well as most perf context instrumentation and random kill out of Env. It will make it easier to maintain multiple Env in the future.
Test Plan: Run all existing unit tests.
Reviewers: anthony, kradhakrishnan, IslamAbdelRahman, yhchiang, igor
Reviewed By: igor
Subscribers: leveldb, dhruba
Differential Revision: https://reviews.facebook.net/D42321
2015-07-17 23:16:11 +00:00
|
|
|
|
|
|
|
{
|
2018-11-09 19:17:34 +00:00
|
|
|
std::unique_ptr<SequentialFile> srcfile;
|
2017-11-03 18:48:12 +00:00
|
|
|
s = env->NewSequentialFile(source, &srcfile, soptions);
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
2018-11-09 19:17:34 +00:00
|
|
|
std::unique_ptr<WritableFile> destfile;
|
2014-11-14 19:38:26 +00:00
|
|
|
s = env->NewWritableFile(destination, &destfile, soptions);
|
2017-11-03 18:48:12 +00:00
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
2014-11-14 19:38:26 +00:00
|
|
|
|
2017-11-03 18:48:12 +00:00
|
|
|
if (size == 0) {
|
|
|
|
// default argument means copy everything
|
2014-11-14 19:38:26 +00:00
|
|
|
s = env->GetFileSize(source, &size);
|
2017-11-03 18:48:12 +00:00
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
2014-11-14 19:38:26 +00:00
|
|
|
}
|
2018-06-21 15:34:24 +00:00
|
|
|
src_reader.reset(new SequentialFileReader(std::move(srcfile), source));
|
2018-08-23 17:04:10 +00:00
|
|
|
dest_writer.reset(
|
|
|
|
new WritableFileWriter(std::move(destfile), destination, soptions));
|
Move rate_limiter, write buffering, most perf context instrumentation and most random kill out of Env
Summary: We want to keep Env a think layer for better portability. Less platform dependent codes should be moved out of Env. In this patch, I create a wrapper of file readers and writers, and put rate limiting, write buffering, as well as most perf context instrumentation and random kill out of Env. It will make it easier to maintain multiple Env in the future.
Test Plan: Run all existing unit tests.
Reviewers: anthony, kradhakrishnan, IslamAbdelRahman, yhchiang, igor
Reviewed By: igor
Subscribers: leveldb, dhruba
Differential Revision: https://reviews.facebook.net/D42321
2015-07-17 23:16:11 +00:00
|
|
|
}
|
2014-11-14 19:38:26 +00:00
|
|
|
|
|
|
|
char buffer[4096];
|
|
|
|
Slice slice;
|
|
|
|
while (size > 0) {
|
2015-10-19 20:40:44 +00:00
|
|
|
size_t bytes_to_read = std::min(sizeof(buffer), static_cast<size_t>(size));
|
2017-11-03 18:48:12 +00:00
|
|
|
s = src_reader->Read(bytes_to_read, &slice, buffer);
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
2014-11-14 19:38:26 +00:00
|
|
|
}
|
2017-11-03 18:48:12 +00:00
|
|
|
if (slice.size() == 0) {
|
|
|
|
return Status::Corruption("file too small");
|
2014-11-14 19:38:26 +00:00
|
|
|
}
|
2017-11-03 18:48:12 +00:00
|
|
|
s = dest_writer->Append(slice);
|
2014-11-14 19:38:26 +00:00
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
size -= slice.size();
|
|
|
|
}
|
2018-04-19 21:00:52 +00:00
|
|
|
return dest_writer->Sync(use_fsync);
|
2014-11-14 19:38:26 +00:00
|
|
|
}
|
|
|
|
|
2016-03-17 17:07:21 +00:00
|
|
|
// Utility function to create a file with the provided contents
|
|
|
|
Status CreateFile(Env* env, const std::string& destination,
|
2018-08-28 19:35:17 +00:00
|
|
|
const std::string& contents, bool use_fsync) {
|
2016-03-17 17:07:21 +00:00
|
|
|
const EnvOptions soptions;
|
|
|
|
Status s;
|
2018-11-09 19:17:34 +00:00
|
|
|
std::unique_ptr<WritableFileWriter> dest_writer;
|
2016-03-17 17:07:21 +00:00
|
|
|
|
2018-11-09 19:17:34 +00:00
|
|
|
std::unique_ptr<WritableFile> destfile;
|
2016-03-17 17:07:21 +00:00
|
|
|
s = env->NewWritableFile(destination, &destfile, soptions);
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
2018-08-23 17:04:10 +00:00
|
|
|
dest_writer.reset(
|
|
|
|
new WritableFileWriter(std::move(destfile), destination, soptions));
|
2018-08-28 19:35:17 +00:00
|
|
|
s = dest_writer->Append(Slice(contents));
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
return dest_writer->Sync(use_fsync);
|
2016-03-17 17:07:21 +00:00
|
|
|
}
|
|
|
|
|
2019-01-29 22:27:30 +00:00
|
|
|
Status DeleteDBFile(const ImmutableDBOptions* db_options,
|
2019-07-07 04:04:22 +00:00
|
|
|
const std::string& fname, const std::string& dir_to_sync,
|
|
|
|
const bool force_bg, const bool force_fg) {
|
2016-12-22 01:35:00 +00:00
|
|
|
#ifndef ROCKSDB_LITE
|
2019-03-28 22:13:02 +00:00
|
|
|
SstFileManagerImpl* sfm =
|
2016-01-29 02:35:01 +00:00
|
|
|
static_cast<SstFileManagerImpl*>(db_options->sst_file_manager.get());
|
2019-07-07 04:04:22 +00:00
|
|
|
if (sfm && !force_fg) {
|
2019-01-29 22:27:30 +00:00
|
|
|
return sfm->ScheduleFileDeletion(fname, dir_to_sync, force_bg);
|
2015-08-19 22:02:17 +00:00
|
|
|
} else {
|
2016-01-29 02:35:01 +00:00
|
|
|
return db_options->env->DeleteFile(fname);
|
2015-08-19 22:02:17 +00:00
|
|
|
}
|
2016-12-22 01:35:00 +00:00
|
|
|
#else
|
2018-04-26 20:51:39 +00:00
|
|
|
(void)dir_to_sync;
|
2019-01-29 22:27:30 +00:00
|
|
|
(void)force_bg;
|
2019-07-07 04:04:22 +00:00
|
|
|
(void)force_fg;
|
2016-12-22 01:35:00 +00:00
|
|
|
// SstFileManager is not supported in ROCKSDB_LITE
|
2019-03-28 22:13:02 +00:00
|
|
|
// Delete file immediately
|
2016-12-22 01:35:00 +00:00
|
|
|
return db_options->env->DeleteFile(fname);
|
|
|
|
#endif
|
2015-08-19 22:02:17 +00:00
|
|
|
}
|
|
|
|
|
2019-07-07 04:04:22 +00:00
|
|
|
bool IsWalDirSameAsDBPath(const ImmutableDBOptions* db_options) {
|
|
|
|
bool same = false;
|
2019-07-09 18:01:12 +00:00
|
|
|
assert(!db_options->db_paths.empty());
|
2019-07-07 04:04:22 +00:00
|
|
|
Status s = db_options->env->AreFilesSame(db_options->wal_dir,
|
|
|
|
db_options->db_paths[0].path, &same);
|
|
|
|
if (s.IsNotSupported()) {
|
|
|
|
same = db_options->wal_dir == db_options->db_paths[0].path;
|
|
|
|
}
|
|
|
|
return same;
|
|
|
|
}
|
|
|
|
|
2014-11-14 19:38:26 +00:00
|
|
|
} // namespace rocksdb
|