]>
git.proxmox.com Git - ceph.git/blob - ceph/src/rocksdb/db/options_file_test.cc
1 // Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
2 // This source code is licensed under the BSD-style license found in the
3 // LICENSE file in the root directory of this source tree. An additional grant
4 // of patent rights can be found in the PATENTS file in the same directory.
9 #include "db/db_impl.h"
10 #include "db/db_test_util.h"
11 #include "rocksdb/options.h"
12 #include "rocksdb/table.h"
13 #include "util/testharness.h"
16 class OptionsFileTest
: public testing::Test
{
18 OptionsFileTest() : dbname_(test::TmpDir() + "/options_file_test") {}
24 void UpdateOptionsFiles(DB
* db
,
25 std::unordered_set
<std::string
>* filename_history
,
26 int* options_files_count
) {
27 std::vector
<std::string
> filenames
;
28 db
->GetEnv()->GetChildren(db
->GetName(), &filenames
);
31 *options_files_count
= 0;
32 for (auto filename
: filenames
) {
33 if (ParseFileName(filename
, &number
, &type
) && type
== kOptionsFile
) {
34 filename_history
->insert(filename
);
35 (*options_files_count
)++;
40 // Verify whether the current Options Files are the latest ones.
41 void VerifyOptionsFileName(
42 DB
* db
, const std::unordered_set
<std::string
>& past_filenames
) {
43 std::vector
<std::string
> filenames
;
44 std::unordered_set
<std::string
> current_filenames
;
45 db
->GetEnv()->GetChildren(db
->GetName(), &filenames
);
48 for (auto filename
: filenames
) {
49 if (ParseFileName(filename
, &number
, &type
) && type
== kOptionsFile
) {
50 current_filenames
.insert(filename
);
53 for (auto past_filename
: past_filenames
) {
54 if (current_filenames
.find(past_filename
) != current_filenames
.end()) {
57 for (auto filename
: current_filenames
) {
58 ASSERT_GT(filename
, past_filename
);
64 TEST_F(OptionsFileTest
, NumberOfOptionsFiles
) {
65 const int kReopenCount
= 20;
67 opt
.create_if_missing
= true;
68 DestroyDB(dbname_
, opt
);
69 std::unordered_set
<std::string
> filename_history
;
71 for (int i
= 0; i
< kReopenCount
; ++i
) {
72 ASSERT_OK(DB::Open(opt
, dbname_
, &db
));
73 int num_options_files
= 0;
74 UpdateOptionsFiles(db
, &filename_history
, &num_options_files
);
75 ASSERT_GT(num_options_files
, 0);
76 ASSERT_LE(num_options_files
, 2);
77 // Make sure we always keep the latest option files.
78 VerifyOptionsFileName(db
, filename_history
);
83 TEST_F(OptionsFileTest
, OptionsFileName
) {
84 const uint64_t kOptionsFileNum
= 12345;
88 auto options_file_name
= OptionsFileName("", kOptionsFileNum
);
89 ASSERT_TRUE(ParseFileName(options_file_name
, &number
, &type
, nullptr));
90 ASSERT_EQ(type
, kOptionsFile
);
91 ASSERT_EQ(number
, kOptionsFileNum
);
93 const uint64_t kTempOptionsFileNum
= 54352;
94 auto temp_options_file_name
= TempOptionsFileName("", kTempOptionsFileNum
);
95 ASSERT_TRUE(ParseFileName(temp_options_file_name
, &number
, &type
, nullptr));
96 ASSERT_NE(temp_options_file_name
.find(kTempFileNameSuffix
),
98 ASSERT_EQ(type
, kTempFile
);
99 ASSERT_EQ(number
, kTempOptionsFileNum
);
101 } // namespace rocksdb
103 int main(int argc
, char** argv
) {
104 #if !(defined NDEBUG) || !defined(OS_WIN)
105 ::testing::InitGoogleTest(&argc
, argv
);
106 return RUN_ALL_TESTS();
109 #endif // !(defined NDEBUG) || !defined(OS_WIN)
115 int main(int argc
, char** argv
) {
116 printf("Skipped as Options file is not supported in RocksDBLite.\n");
119 #endif // !ROCKSDB_LITE