2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2014-11-14 20:38:26 +01:00
|
|
|
// 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.
|
|
|
|
//
|
2015-08-20 00:02:17 +02:00
|
|
|
#include "util/file_util.h"
|
|
|
|
|
2014-11-14 20:38:26 +01:00
|
|
|
#include <string>
|
|
|
|
#include <algorithm>
|
2015-08-20 00:02:17 +02:00
|
|
|
|
2014-11-14 20:38:26 +01:00
|
|
|
#include "rocksdb/env.h"
|
2016-01-29 03:35:01 +01:00
|
|
|
#include "util/sst_file_manager_impl.h"
|
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-18 01:16:11 +02:00
|
|
|
#include "util/file_reader_writer.h"
|
2014-11-14 20:38:26 +01:00
|
|
|
|
|
|
|
namespace rocksdb {
|
|
|
|
|
|
|
|
// Utility function to copy a file up to a specified length
|
|
|
|
Status CopyFile(Env* env, const std::string& source,
|
|
|
|
const std::string& destination, uint64_t size) {
|
|
|
|
const EnvOptions soptions;
|
|
|
|
Status s;
|
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-18 01:16:11 +02:00
|
|
|
unique_ptr<SequentialFileReader> src_reader;
|
|
|
|
unique_ptr<WritableFileWriter> dest_writer;
|
|
|
|
|
|
|
|
{
|
|
|
|
unique_ptr<SequentialFile> srcfile;
|
2014-11-14 20:38:26 +01:00
|
|
|
s = env->NewSequentialFile(source, &srcfile, soptions);
|
|
|
|
unique_ptr<WritableFile> destfile;
|
|
|
|
if (s.ok()) {
|
|
|
|
s = env->NewWritableFile(destination, &destfile, soptions);
|
|
|
|
} else {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (size == 0) {
|
|
|
|
// default argument means copy everything
|
|
|
|
if (s.ok()) {
|
|
|
|
s = env->GetFileSize(source, &size);
|
|
|
|
} else {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
}
|
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-18 01:16:11 +02:00
|
|
|
src_reader.reset(new SequentialFileReader(std::move(srcfile)));
|
|
|
|
dest_writer.reset(new WritableFileWriter(std::move(destfile), soptions));
|
|
|
|
}
|
2014-11-14 20:38:26 +01:00
|
|
|
|
|
|
|
char buffer[4096];
|
|
|
|
Slice slice;
|
|
|
|
while (size > 0) {
|
2015-10-19 22:40:44 +02:00
|
|
|
size_t bytes_to_read = std::min(sizeof(buffer), static_cast<size_t>(size));
|
2014-11-14 20:38:26 +01:00
|
|
|
if (s.ok()) {
|
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-18 01:16:11 +02:00
|
|
|
s = src_reader->Read(bytes_to_read, &slice, buffer);
|
2014-11-14 20:38:26 +01:00
|
|
|
}
|
|
|
|
if (s.ok()) {
|
|
|
|
if (slice.size() == 0) {
|
|
|
|
return Status::Corruption("file too small");
|
|
|
|
}
|
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-18 01:16:11 +02:00
|
|
|
s = dest_writer->Append(slice);
|
2014-11-14 20:38:26 +01:00
|
|
|
}
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
size -= slice.size();
|
|
|
|
}
|
|
|
|
return Status::OK();
|
|
|
|
}
|
|
|
|
|
2016-03-17 18:07:21 +01:00
|
|
|
// Utility function to create a file with the provided contents
|
|
|
|
Status CreateFile(Env* env, const std::string& destination,
|
|
|
|
const std::string& contents) {
|
|
|
|
const EnvOptions soptions;
|
|
|
|
Status s;
|
|
|
|
unique_ptr<WritableFileWriter> dest_writer;
|
|
|
|
|
|
|
|
unique_ptr<WritableFile> destfile;
|
|
|
|
s = env->NewWritableFile(destination, &destfile, soptions);
|
|
|
|
if (!s.ok()) {
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
dest_writer.reset(new WritableFileWriter(std::move(destfile), soptions));
|
|
|
|
return dest_writer->Append(Slice(contents));
|
|
|
|
}
|
|
|
|
|
2016-09-24 01:34:04 +02:00
|
|
|
Status DeleteSSTFile(const ImmutableDBOptions* db_options,
|
|
|
|
const std::string& fname, uint32_t path_id) {
|
2016-01-29 03:35:01 +01:00
|
|
|
// TODO(tec): support sst_file_manager for multiple path_ids
|
2016-12-22 02:35:00 +01:00
|
|
|
#ifndef ROCKSDB_LITE
|
2016-01-29 03:35:01 +01:00
|
|
|
auto sfm =
|
|
|
|
static_cast<SstFileManagerImpl*>(db_options->sst_file_manager.get());
|
|
|
|
if (sfm && path_id == 0) {
|
|
|
|
return sfm->ScheduleFileDeletion(fname);
|
2015-08-20 00:02:17 +02:00
|
|
|
} else {
|
2016-01-29 03:35:01 +01:00
|
|
|
return db_options->env->DeleteFile(fname);
|
2015-08-20 00:02:17 +02:00
|
|
|
}
|
2016-12-22 02:35:00 +01:00
|
|
|
#else
|
|
|
|
// SstFileManager is not supported in ROCKSDB_LITE
|
|
|
|
return db_options->env->DeleteFile(fname);
|
|
|
|
#endif
|
2015-08-20 00:02:17 +02:00
|
|
|
}
|
|
|
|
|
2014-11-14 20:38:26 +01:00
|
|
|
} // namespace rocksdb
|