2016-02-10 00:12:00 +01:00
|
|
|
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
2017-07-16 01:03:42 +02: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-10-29 01:52:32 +01:00
|
|
|
#pragma once
|
2015-09-02 22:58:22 +02:00
|
|
|
|
2014-10-29 01:52:32 +01:00
|
|
|
#include <algorithm>
|
2014-10-29 02:10:55 +01:00
|
|
|
#include <atomic>
|
2014-10-29 01:52:32 +01:00
|
|
|
#include <map>
|
2015-09-02 22:58:22 +02:00
|
|
|
#include <memory>
|
|
|
|
#include <set>
|
2014-10-29 01:52:32 +01:00
|
|
|
#include <string>
|
2015-09-02 22:58:22 +02:00
|
|
|
#include <utility>
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2020-02-11 00:42:46 +01:00
|
|
|
#include "db/version_edit.h"
|
2015-09-02 22:58:22 +02:00
|
|
|
#include "port/port.h"
|
2015-08-08 06:59:51 +02:00
|
|
|
#include "rocksdb/comparator.h"
|
Pass IOStatus to write path and set retryable IO Error as hard error in BG jobs (#6487)
Summary:
In the current code base, we use Status to get and store the returned status from the call. Specifically, for IO related functions, the current Status cannot reflect the IO Error details such as error scope, error retryable attribute, and others. With the implementation of https://github.com/facebook/rocksdb/issues/5761, we have the new Wrapper for IO, which returns IOStatus instead of Status. However, the IOStatus is purged at the lower level of write path and transferred to Status.
The first job of this PR is to pass the IOStatus to the write path (flush, WAL write, and Compaction). The second job is to identify the Retryable IO Error as HardError, and set the bg_error_ as HardError. In this case, the DB Instance becomes read only. User is informed of the Status and need to take actions to deal with it (e.g., call db->Resume()).
Pull Request resolved: https://github.com/facebook/rocksdb/pull/6487
Test Plan: Added the testing case to error_handler_fs_test. Pass make asan_check
Reviewed By: anand1976
Differential Revision: D20685017
Pulled By: zhichao-cao
fbshipit-source-id: ff85f042896243abcd6ef37877834e26f36b6eb0
2020-03-28 00:03:05 +01:00
|
|
|
#include "rocksdb/io_status.h"
|
2014-10-29 01:52:32 +01:00
|
|
|
#include "rocksdb/table.h"
|
2015-10-13 00:06:38 +02:00
|
|
|
#include "table/internal_iterator.h"
|
2014-10-29 01:52:32 +01:00
|
|
|
#include "table/table_builder.h"
|
2015-09-02 22:58:22 +02:00
|
|
|
#include "table/table_reader.h"
|
2019-05-30 20:21:38 +02:00
|
|
|
#include "test_util/testharness.h"
|
|
|
|
#include "test_util/testutil.h"
|
2019-05-31 02:39:43 +02:00
|
|
|
#include "util/kv_map.h"
|
|
|
|
#include "util/mutexlock.h"
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2020-02-20 21:07:53 +01:00
|
|
|
namespace ROCKSDB_NAMESPACE {
|
2014-11-14 20:35:48 +01:00
|
|
|
namespace mock {
|
2020-10-01 19:08:52 +02:00
|
|
|
using KVPair = std::pair<std::string, std::string>;
|
|
|
|
using KVVector = std::vector<KVPair>;
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2020-10-01 19:08:52 +02:00
|
|
|
KVVector MakeMockFile(std::initializer_list<KVPair> l = {});
|
2020-11-12 20:40:52 +01:00
|
|
|
void SortKVVector(KVVector* kv_vector,
|
|
|
|
const Comparator* ucmp = BytewiseComparator());
|
2014-10-29 01:52:32 +01:00
|
|
|
|
|
|
|
struct MockTableFileSystem {
|
|
|
|
port::Mutex mutex;
|
2020-10-01 19:08:52 +02:00
|
|
|
std::map<uint32_t, KVVector> files;
|
2014-10-29 01:52:32 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
class MockTableFactory : public TableFactory {
|
|
|
|
public:
|
2020-07-22 20:03:29 +02:00
|
|
|
enum MockCorruptionMode {
|
|
|
|
kCorruptNone,
|
|
|
|
kCorruptKey,
|
|
|
|
kCorruptValue,
|
2020-10-01 19:08:52 +02:00
|
|
|
kCorruptReorderKey,
|
2020-07-22 20:03:29 +02:00
|
|
|
};
|
|
|
|
|
2014-10-29 01:52:32 +01:00
|
|
|
MockTableFactory();
|
2021-08-19 19:09:30 +02:00
|
|
|
static const char* kClassName() { return "MockTable"; }
|
|
|
|
const char* Name() const override { return kClassName(); }
|
2020-06-29 23:51:57 +02:00
|
|
|
using TableFactory::NewTableReader;
|
2016-07-20 20:23:31 +02:00
|
|
|
Status NewTableReader(
|
2020-06-29 23:51:57 +02:00
|
|
|
const ReadOptions& ro, const TableReaderOptions& table_reader_options,
|
2018-11-09 20:17:34 +01:00
|
|
|
std::unique_ptr<RandomAccessFileReader>&& file, uint64_t file_size,
|
|
|
|
std::unique_ptr<TableReader>* table_reader,
|
2016-07-20 20:23:31 +02:00
|
|
|
bool prefetch_index_and_filter_in_cache = true) const override;
|
A new call back to TablePropertiesCollector to allow users know the entry is add, delete or merge
Summary:
Currently users have no idea a key is add, delete or merge from TablePropertiesCollector call back. Add a new function to add it.
Also refactor the codes so that
(1) make table property collector and internal table property collector two separate data structures with the later one now exposed
(2) table builders only receive internal table properties
Test Plan: Add cases in table_properties_collector_test to cover both of old and new ways of using TablePropertiesCollector.
Reviewers: yhchiang, igor.sugak, rven, igor
Reviewed By: rven, igor
Subscribers: meyering, yoshinorim, maykov, leveldb, dhruba
Differential Revision: https://reviews.facebook.net/D35373
2015-04-06 19:04:30 +02:00
|
|
|
TableBuilder* NewTableBuilder(
|
|
|
|
const TableBuilderOptions& table_builder_options,
|
2021-04-29 15:59:53 +02:00
|
|
|
WritableFileWriter* file) const override;
|
2014-10-29 01:52:32 +01:00
|
|
|
|
2014-11-14 20:35:48 +01:00
|
|
|
// This function will directly create mock table instead of going through
|
2015-09-02 22:58:22 +02:00
|
|
|
// MockTableBuilder. file_contents has to have a format of <internal_key,
|
|
|
|
// value>. Those key-value pairs will then be inserted into the mock table.
|
2014-11-14 20:35:48 +01:00
|
|
|
Status CreateMockTable(Env* env, const std::string& fname,
|
2020-10-01 19:08:52 +02:00
|
|
|
KVVector file_contents);
|
2014-11-14 20:35:48 +01:00
|
|
|
|
2020-09-15 01:59:00 +02:00
|
|
|
virtual std::string GetPrintableOptions() const override {
|
2014-10-29 01:52:32 +01:00
|
|
|
return std::string();
|
|
|
|
}
|
|
|
|
|
2020-07-22 20:03:29 +02:00
|
|
|
void SetCorruptionMode(MockCorruptionMode mode) { corrupt_mode_ = mode; }
|
2014-10-29 01:52:32 +01:00
|
|
|
// This function will assert that only a single file exists and that the
|
|
|
|
// contents are equal to file_contents
|
2020-10-01 19:08:52 +02:00
|
|
|
void AssertSingleFile(const KVVector& file_contents);
|
|
|
|
void AssertLatestFile(const KVVector& file_contents);
|
2014-10-29 01:52:32 +01:00
|
|
|
|
|
|
|
private:
|
2020-12-24 01:54:05 +01:00
|
|
|
Status GetAndWriteNextID(WritableFileWriter* file, uint32_t* id) const;
|
|
|
|
Status GetIDFromFile(RandomAccessFileReader* file, uint32_t* id) const;
|
2014-10-29 01:52:32 +01:00
|
|
|
|
|
|
|
mutable MockTableFileSystem file_system_;
|
|
|
|
mutable std::atomic<uint32_t> next_id_;
|
2020-07-22 20:03:29 +02:00
|
|
|
MockCorruptionMode corrupt_mode_;
|
2014-10-29 01:52:32 +01:00
|
|
|
};
|
|
|
|
|
2014-11-14 20:35:48 +01:00
|
|
|
} // namespace mock
|
2020-02-20 21:07:53 +01:00
|
|
|
} // namespace ROCKSDB_NAMESPACE
|