2020-05-29 20:24:19 +02:00
|
|
|
#!/usr/bin/env python3
|
2019-04-18 19:51:19 +02:00
|
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
2020-03-25 04:57:53 +01:00
|
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
|
2013-03-13 07:20:14 +01:00
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import time
|
2013-06-08 21:29:43 +02:00
|
|
|
import random
|
2020-06-13 04:24:11 +02:00
|
|
|
import re
|
2013-04-05 22:44:59 +02:00
|
|
|
import tempfile
|
2013-03-13 07:20:14 +01:00
|
|
|
import subprocess
|
2014-03-20 19:11:08 +01:00
|
|
|
import shutil
|
2015-10-19 20:43:14 +02:00
|
|
|
import argparse
|
2013-03-13 07:20:14 +01:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
# params overwrite priority:
|
|
|
|
# for default:
|
2018-05-09 22:32:03 +02:00
|
|
|
# default_params < {blackbox,whitebox}_default_params < args
|
2015-10-19 20:43:14 +02:00
|
|
|
# for simple:
|
2018-05-09 22:32:03 +02:00
|
|
|
# default_params < {blackbox,whitebox}_default_params <
|
|
|
|
# simple_default_params <
|
|
|
|
# {blackbox,whitebox}_simple_default_params < args
|
2019-08-23 01:30:30 +02:00
|
|
|
# for cf_consistency:
|
2018-10-30 22:01:38 +01:00
|
|
|
# default_params < {blackbox,whitebox}_default_params <
|
2019-08-23 01:30:30 +02:00
|
|
|
# cf_consistency_params < args
|
2019-12-12 00:59:45 +01:00
|
|
|
# for txn:
|
|
|
|
# default_params < {blackbox,whitebox}_default_params < txn_params < args
|
2022-03-17 03:00:04 +01:00
|
|
|
# for ts:
|
|
|
|
# default_params < {blackbox,whitebox}_default_params < ts_params < args
|
|
|
|
# for multiops_txn:
|
|
|
|
# default_params < {blackbox,whitebox}_default_params < multiops_txn_params < args
|
2013-03-13 07:20:14 +01:00
|
|
|
|
2018-04-30 21:23:45 +02:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
default_params = {
|
2017-10-21 00:17:03 +02:00
|
|
|
"acquire_snapshot_one_in": 10000,
|
2020-09-08 19:46:55 +02:00
|
|
|
"backup_max_size": 100 * 1024 * 1024,
|
|
|
|
# Consider larger number when backups considered more stable
|
|
|
|
"backup_one_in": 100000,
|
Integrity protection for live updates to WriteBatch (#7748)
Summary:
This PR adds the foundation classes for key-value integrity protection and the first use case: protecting live updates from the source buffers added to `WriteBatch` through the destination buffer in `MemTable`. The width of the protection info is not yet configurable -- only eight bytes per key is supported. This PR allows users to enable protection by constructing `WriteBatch` with `protection_bytes_per_key == 8`. It does not yet expose a way for users to get integrity protection via other write APIs (e.g., `Put()`, `Merge()`, `Delete()`, etc.).
The foundation classes (`ProtectionInfo.*`) embed the coverage info in their type, and provide `Protect.*()` and `Strip.*()` functions to navigate between types with different coverage. For making bytes per key configurable (for powers of two up to eight) in the future, these classes are templated on the unsigned integer type used to store the protection info. That integer contains the XOR'd result of hashes with independent seeds for all covered fields. For integer fields, the hash is computed on the raw unadjusted bytes, so the result is endian-dependent. The most significant bytes are truncated when the hash value (8 bytes) is wider than the protection integer.
When `WriteBatch` is constructed with `protection_bytes_per_key == 8`, we hold a `ProtectionInfoKVOTC` (i.e., one that covers key, value, optype aka `ValueType`, timestamp, and CF ID) for each entry added to the batch. The protection info is generated from the original buffers passed by the user, as well as the original metadata generated internally. When writing to memtable, each entry is transformed to a `ProtectionInfoKVOTS` (i.e., dropping coverage of CF ID and adding coverage of sequence number), since at that point we know the sequence number, and have already selected a memtable corresponding to a particular CF. This protection info is verified once the entry is encoded in the `MemTable` buffer.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/7748
Test Plan:
- an integration test to verify a wide variety of single-byte changes to the encoded `MemTable` buffer are caught
- add to stress/crash test to verify it works in variety of configs/operations without intentional corruption
- [deferred] unit tests for `ProtectionInfo.*` classes for edge cases like KV swap, `SliceParts` and `Slice` APIs are interchangeable, etc.
Reviewed By: pdillinger
Differential Revision: D25754492
Pulled By: ajkr
fbshipit-source-id: e481bac6c03c2ab268be41359730f1ceb9964866
2021-01-29 21:17:17 +01:00
|
|
|
"batch_protection_bytes_per_key": lambda: random.choice([0, 8]),
|
2015-10-19 20:43:14 +02:00
|
|
|
"block_size": 16384,
|
2019-12-10 17:38:23 +01:00
|
|
|
"bloom_bits": lambda: random.choice([random.randint(0,19),
|
|
|
|
random.lognormvariate(2.3, 1.3)]),
|
2019-04-30 18:46:40 +02:00
|
|
|
"cache_index_and_filter_blocks": lambda: random.randint(0, 1),
|
2022-02-15 22:55:22 +01:00
|
|
|
"cache_size": 8388608,
|
2022-04-06 19:33:00 +02:00
|
|
|
"reserve_table_reader_memory": lambda: random.choice([0, 1]),
|
2018-06-19 04:23:42 +02:00
|
|
|
"checkpoint_one_in": 1000000,
|
2019-12-17 00:24:26 +01:00
|
|
|
"compression_type": lambda: random.choice(
|
2022-01-31 22:19:55 +01:00
|
|
|
["none", "snappy", "zlib", "lz4", "lz4hc", "xpress", "zstd"]),
|
2019-12-21 01:13:19 +01:00
|
|
|
"bottommost_compression_type": lambda:
|
|
|
|
"disable" if random.randint(0, 1) == 0 else
|
|
|
|
random.choice(
|
2022-01-31 22:19:55 +01:00
|
|
|
["none", "snappy", "zlib", "lz4", "lz4hc", "xpress", "zstd"]),
|
Implement XXH3 block checksum type (#9069)
Summary:
XXH3 - latest hash function that is extremely fast on large
data, easily faster than crc32c on most any x86_64 hardware. In
integrating this hash function, I have handled the compression type byte
in a non-standard way to avoid using the streaming API (extra data
movement and active code size because of hash function complexity). This
approach got a thumbs-up from Yann Collet.
Existing functionality change:
* reject bad ChecksumType in options with InvalidArgument
This change split off from https://github.com/facebook/rocksdb/issues/9058 because context-aware checksum is
likely to be handled through different configuration than ChecksumType.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/9069
Test Plan:
tests updated, and substantially expanded. Unit tests now check
that we don't accidentally change the values generated by the checksum
algorithms ("schema test") and that we properly handle
invalid/unrecognized checksum types in options or in file footer.
DBTestBase::ChangeOptions (etc.) updated from two to one configuration
changing from default CRC32c ChecksumType. The point of this test code
is to detect possible interactions among features, and the likelihood of
some bad interaction being detected by including configurations other
than XXH3 and CRC32c--and then not detected by stress/crash test--is
extremely low.
Stress/crash test also updated (manual run long enough to see it accepts
new checksum type). db_bench also updated for microbenchmarking
checksums.
### Performance microbenchmark (PORTABLE=0 DEBUG_LEVEL=0, Broadwell processor)
./db_bench -benchmarks=crc32c,xxhash,xxhash64,xxh3,crc32c,xxhash,xxhash64,xxh3,crc32c,xxhash,xxhash64,xxh3
crc32c : 0.200 micros/op 5005220 ops/sec; 19551.6 MB/s (4096 per op)
xxhash : 0.807 micros/op 1238408 ops/sec; 4837.5 MB/s (4096 per op)
xxhash64 : 0.421 micros/op 2376514 ops/sec; 9283.3 MB/s (4096 per op)
xxh3 : 0.171 micros/op 5858391 ops/sec; 22884.3 MB/s (4096 per op)
crc32c : 0.206 micros/op 4859566 ops/sec; 18982.7 MB/s (4096 per op)
xxhash : 0.793 micros/op 1260850 ops/sec; 4925.2 MB/s (4096 per op)
xxhash64 : 0.410 micros/op 2439182 ops/sec; 9528.1 MB/s (4096 per op)
xxh3 : 0.161 micros/op 6202872 ops/sec; 24230.0 MB/s (4096 per op)
crc32c : 0.203 micros/op 4924686 ops/sec; 19237.1 MB/s (4096 per op)
xxhash : 0.839 micros/op 1192388 ops/sec; 4657.8 MB/s (4096 per op)
xxhash64 : 0.424 micros/op 2357391 ops/sec; 9208.6 MB/s (4096 per op)
xxh3 : 0.162 micros/op 6182678 ops/sec; 24151.1 MB/s (4096 per op)
As you can see, especially once warmed up, xxh3 is fastest.
### Performance macrobenchmark (PORTABLE=0 DEBUG_LEVEL=0, Broadwell processor)
Test
for I in `seq 1 50`; do for CHK in 0 1 2 3 4; do TEST_TMPDIR=/dev/shm/rocksdb$CHK ./db_bench -benchmarks=fillseq -memtablerep=vector -allow_concurrent_memtable_write=false -num=30000000 -checksum_type=$CHK 2>&1 | grep 'micros/op' | tee -a results-$CHK & done; wait; done
Results (ops/sec)
for FILE in results*; do echo -n "$FILE "; awk '{ s += $5; c++; } END { print 1.0 * s / c; }' < $FILE; done
results-0 252118 # kNoChecksum
results-1 251588 # kCRC32c
results-2 251863 # kxxHash
results-3 252016 # kxxHash64
results-4 252038 # kXXH3
Reviewed By: mrambacher
Differential Revision: D31905249
Pulled By: pdillinger
fbshipit-source-id: cb9b998ebe2523fc7c400eedf62124a78bf4b4d1
2021-10-29 07:13:47 +02:00
|
|
|
"checksum_type" : lambda: random.choice(["kCRC32c", "kxxHash", "kxxHash64", "kXXH3"]),
|
2018-08-07 00:23:03 +02:00
|
|
|
"compression_max_dict_bytes": lambda: 16384 * random.randint(0, 1),
|
|
|
|
"compression_zstd_max_train_bytes": lambda: 65536 * random.randint(0, 1),
|
2020-05-08 05:50:14 +02:00
|
|
|
# Disabled compression_parallel_threads as the feature is not stable
|
|
|
|
# lambda: random.choice([1] * 9 + [4])
|
|
|
|
"compression_parallel_threads": 1,
|
Limit buffering for collecting samples for compression dictionary (#7970)
Summary:
For dictionary compression, we need to collect some representative samples of the data to be compressed, which we use to either generate or train (when `CompressionOptions::zstd_max_train_bytes > 0`) a dictionary. Previously, the strategy was to buffer all the data blocks during flush, and up to the target file size during compaction. That strategy allowed us to randomly pick samples from as wide a range as possible that'd be guaranteed to land in a single output file.
However, some users try to make huge files in memory-constrained environments, where this strategy can cause OOM. This PR introduces an option, `CompressionOptions::max_dict_buffer_bytes`, that limits how much data blocks are buffered before we switch to unbuffered mode (which means creating the per-SST dictionary, writing out the buffered data, and compressing/writing new blocks as soon as they are built). It is not strict as we currently buffer more than just data blocks -- also keys are buffered. But it does make a step towards giving users predictable memory usage.
Related changes include:
- Changed sampling for dictionary compression to select unique data blocks when there is limited availability of data blocks
- Made use of `BlockBuilder::SwapAndReset()` to save an allocation+memcpy when buffering data blocks for building a dictionary
- Changed `ParseBoolean()` to accept an input containing characters after the boolean. This is necessary since, with this PR, a value for `CompressionOptions::enabled` is no longer necessarily the final component in the `CompressionOptions` string.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/7970
Test Plan:
- updated `CompressionOptions` unit tests to verify limit is respected (to the extent expected in the current implementation) in various scenarios of flush/compaction to bottommost/non-bottommost level
- looked at jemalloc heap profiles right before and after switching to unbuffered mode during flush/compaction. Verified memory usage in buffering is proportional to the limit set.
Reviewed By: pdillinger
Differential Revision: D26467994
Pulled By: ajkr
fbshipit-source-id: 3da4ef9fba59974e4ef40e40c01611002c861465
2021-02-19 23:06:59 +01:00
|
|
|
"compression_max_dict_buffer_bytes": lambda: (1 << random.randint(0, 40)) - 1,
|
2018-04-30 21:23:45 +02:00
|
|
|
"clear_column_family_one_in": 0,
|
2018-06-14 01:39:03 +02:00
|
|
|
"compact_files_one_in": 1000000,
|
|
|
|
"compact_range_one_in": 1000000,
|
2018-12-18 22:30:56 +01:00
|
|
|
"delpercent": 4,
|
|
|
|
"delrangepercent": 1,
|
2015-10-19 20:43:14 +02:00
|
|
|
"destroy_db_initially": 0,
|
2019-10-30 02:15:08 +01:00
|
|
|
"enable_pipelined_write": lambda: random.randint(0, 1),
|
2020-06-18 18:51:14 +02:00
|
|
|
"enable_compaction_filter": lambda: random.choice([0, 0, 0, 1]),
|
2021-09-28 23:12:23 +02:00
|
|
|
"expected_values_dir": lambda: setup_expected_values_dir(),
|
2021-05-05 21:53:42 +02:00
|
|
|
"fail_if_options_file_error": lambda: random.randint(0, 1),
|
2018-09-17 21:23:38 +02:00
|
|
|
"flush_one_in": 1000000,
|
2020-09-04 08:49:27 +02:00
|
|
|
"file_checksum_impl": lambda: random.choice(["none", "crc32c", "xxh64", "big"]),
|
2020-03-19 01:11:06 +01:00
|
|
|
"get_live_files_one_in": 1000000,
|
|
|
|
# Note: the following two are intentionally disabled as the corresponding
|
|
|
|
# APIs are not guaranteed to succeed.
|
|
|
|
"get_sorted_wal_files_one_in": 0,
|
|
|
|
"get_current_wal_file_one_in": 0,
|
2020-01-23 18:07:50 +01:00
|
|
|
# Temporarily disable hash index
|
2020-04-20 21:55:13 +02:00
|
|
|
"index_type": lambda: random.choice([0, 0, 0, 2, 2, 3]),
|
2020-06-18 18:51:14 +02:00
|
|
|
"iterpercent": 10,
|
2020-08-11 01:16:19 +02:00
|
|
|
"mark_for_compaction_one_file_in": lambda: 10 * random.randint(0, 1),
|
2015-10-19 20:43:14 +02:00
|
|
|
"max_background_compactions": 20,
|
|
|
|
"max_bytes_for_level_base": 10485760,
|
2022-01-31 22:19:55 +01:00
|
|
|
"max_key": 25000000,
|
2015-10-19 20:43:14 +02:00
|
|
|
"max_write_buffer_number": 3,
|
|
|
|
"mmap_read": lambda: random.randint(0, 1),
|
2021-10-12 01:22:10 +02:00
|
|
|
# Setting `nooverwritepercent > 0` is only possible because we do not vary
|
|
|
|
# the random seed, so the same keys are chosen by every run for disallowing
|
|
|
|
# overwrites.
|
2018-04-04 00:20:30 +02:00
|
|
|
"nooverwritepercent": 1,
|
2020-04-17 19:57:17 +02:00
|
|
|
"open_files": lambda : random.choice([-1, -1, 100, 500000]),
|
Minimize memory internal fragmentation for Bloom filters (#6427)
Summary:
New experimental option BBTO::optimize_filters_for_memory builds
filters that maximize their use of "usable size" from malloc_usable_size,
which is also used to compute block cache charges.
Rather than always "rounding up," we track state in the
BloomFilterPolicy object to mix essentially "rounding down" and
"rounding up" so that the average FP rate of all generated filters is
the same as without the option. (YMMV as heavily accessed filters might
be unluckily lower accuracy.)
Thus, the option near-minimizes what the block cache considers as
"memory used" for a given target Bloom filter false positive rate and
Bloom filter implementation. There are no forward or backward
compatibility issues with this change, though it only works on the
format_version=5 Bloom filter.
With Jemalloc, we see about 10% reduction in memory footprint (and block
cache charge) for Bloom filters, but 1-2% increase in storage footprint,
due to encoding efficiency losses (FP rate is non-linear with bits/key).
Why not weighted random round up/down rather than state tracking? By
only requiring malloc_usable_size, we don't actually know what the next
larger and next smaller usable sizes for the allocator are. We pick a
requested size, accept and use whatever usable size it has, and use the
difference to inform our next choice. This allows us to narrow in on the
right balance without tracking/predicting usable sizes.
Why not weight history of generated filter false positive rates by
number of keys? This could lead to excess skew in small filters after
generating a large filter.
Results from filter_bench with jemalloc (irrelevant details omitted):
(normal keys/filter, but high variance)
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=30000 -vary_key_count_ratio=0.9
Build avg ns/key: 29.6278
Number of filters: 5516
Total size (MB): 200.046
Reported total allocated memory (MB): 220.597
Reported internal fragmentation: 10.2732%
Bits/key stored: 10.0097
Average FP rate %: 0.965228
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=30000 -vary_key_count_ratio=0.9 -optimize_filters_for_memory
Build avg ns/key: 30.5104
Number of filters: 5464
Total size (MB): 200.015
Reported total allocated memory (MB): 200.322
Reported internal fragmentation: 0.153709%
Bits/key stored: 10.1011
Average FP rate %: 0.966313
(very few keys / filter, optimization not as effective due to ~59 byte
internal fragmentation in blocked Bloom filter representation)
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=1000 -vary_key_count_ratio=0.9
Build avg ns/key: 29.5649
Number of filters: 162950
Total size (MB): 200.001
Reported total allocated memory (MB): 224.624
Reported internal fragmentation: 12.3117%
Bits/key stored: 10.2951
Average FP rate %: 0.821534
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=1000 -vary_key_count_ratio=0.9 -optimize_filters_for_memory
Build avg ns/key: 31.8057
Number of filters: 159849
Total size (MB): 200
Reported total allocated memory (MB): 208.846
Reported internal fragmentation: 4.42297%
Bits/key stored: 10.4948
Average FP rate %: 0.811006
(high keys/filter)
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=1000000 -vary_key_count_ratio=0.9
Build avg ns/key: 29.7017
Number of filters: 164
Total size (MB): 200.352
Reported total allocated memory (MB): 221.5
Reported internal fragmentation: 10.5552%
Bits/key stored: 10.0003
Average FP rate %: 0.969358
$ ./filter_bench -quick -impl=2 -average_keys_per_filter=1000000 -vary_key_count_ratio=0.9 -optimize_filters_for_memory
Build avg ns/key: 30.7131
Number of filters: 160
Total size (MB): 200.928
Reported total allocated memory (MB): 200.938
Reported internal fragmentation: 0.00448054%
Bits/key stored: 10.1852
Average FP rate %: 0.963387
And from db_bench (block cache) with jemalloc:
$ ./db_bench -db=/dev/shm/dbbench.no_optimize -benchmarks=fillrandom -format_version=5 -value_size=90 -bloom_bits=10 -num=2000000 -threads=8 -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=false
$ ./db_bench -db=/dev/shm/dbbench -benchmarks=fillrandom -format_version=5 -value_size=90 -bloom_bits=10 -num=2000000 -threads=8 -optimize_filters_for_memory -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=false
$ (for FILE in /dev/shm/dbbench.no_optimize/*.sst; do ./sst_dump --file=$FILE --show_properties | grep 'filter block' ; done) | awk '{ t += $4; } END { print t; }'
17063835
$ (for FILE in /dev/shm/dbbench/*.sst; do ./sst_dump --file=$FILE --show_properties | grep 'filter block' ; done) | awk '{ t += $4; } END { print t; }'
17430747
$ #^ 2.1% additional filter storage
$ ./db_bench -db=/dev/shm/dbbench.no_optimize -use_existing_db -benchmarks=readrandom,stats -statistics -bloom_bits=10 -num=2000000 -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=false -duration=10 -cache_index_and_filter_blocks -cache_size=1000000000
rocksdb.block.cache.index.add COUNT : 33
rocksdb.block.cache.index.bytes.insert COUNT : 8440400
rocksdb.block.cache.filter.add COUNT : 33
rocksdb.block.cache.filter.bytes.insert COUNT : 21087528
rocksdb.bloom.filter.useful COUNT : 4963889
rocksdb.bloom.filter.full.positive COUNT : 1214081
rocksdb.bloom.filter.full.true.positive COUNT : 1161999
$ #^ 1.04 % observed FP rate
$ ./db_bench -db=/dev/shm/dbbench -use_existing_db -benchmarks=readrandom,stats -statistics -bloom_bits=10 -num=2000000 -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=false -optimize_filters_for_memory -duration=10 -cache_index_and_filter_blocks -cache_size=1000000000
rocksdb.block.cache.index.add COUNT : 33
rocksdb.block.cache.index.bytes.insert COUNT : 8448592
rocksdb.block.cache.filter.add COUNT : 33
rocksdb.block.cache.filter.bytes.insert COUNT : 18220328
rocksdb.bloom.filter.useful COUNT : 5360933
rocksdb.bloom.filter.full.positive COUNT : 1321315
rocksdb.bloom.filter.full.true.positive COUNT : 1262999
$ #^ 1.08 % observed FP rate, 13.6% less memory usage for filters
(Due to specific key density, this example tends to generate filters that are "worse than average" for internal fragmentation. "Better than average" cases can show little or no improvement.)
Pull Request resolved: https://github.com/facebook/rocksdb/pull/6427
Test Plan: unit test added, 'make check' with gcc, clang and valgrind
Reviewed By: siying
Differential Revision: D22124374
Pulled By: pdillinger
fbshipit-source-id: f3e3aa152f9043ddf4fae25799e76341d0d8714e
2020-06-22 22:30:57 +02:00
|
|
|
"optimize_filters_for_memory": lambda: random.randint(0, 1),
|
2019-10-14 19:33:31 +02:00
|
|
|
"partition_filters": lambda: random.randint(0, 1),
|
2020-10-11 23:52:49 +02:00
|
|
|
"partition_pinning": lambda: random.randint(0, 3),
|
2019-12-11 00:45:25 +01:00
|
|
|
"pause_background_one_in": 1000000,
|
2022-01-28 01:36:42 +01:00
|
|
|
"prefix_size" : lambda: random.choice([-1, 1, 5, 7, 8]),
|
2015-10-19 20:43:14 +02:00
|
|
|
"prefixpercent": 5,
|
|
|
|
"progress_reports": 0,
|
|
|
|
"readpercent": 45,
|
2019-03-15 19:58:37 +01:00
|
|
|
"recycle_log_file_num": lambda: random.randint(0, 1),
|
2017-10-21 00:17:03 +02:00
|
|
|
"snapshot_hold_ops": 100000,
|
2020-06-24 01:24:15 +02:00
|
|
|
"sst_file_manager_bytes_per_sec": lambda: random.choice([0, 104857600]),
|
|
|
|
"sst_file_manager_bytes_per_truncate": lambda: random.choice([0, 1048576]),
|
2019-12-15 00:17:05 +01:00
|
|
|
"long_running_snapshots": lambda: random.randint(0, 1),
|
2018-05-09 22:32:03 +02:00
|
|
|
"subcompactions": lambda: random.randint(1, 4),
|
2015-10-19 20:43:14 +02:00
|
|
|
"target_file_size_base": 2097152,
|
|
|
|
"target_file_size_multiplier": 2,
|
2021-12-16 18:04:07 +01:00
|
|
|
"test_batches_snapshots": lambda: random.randint(0, 1),
|
2020-10-11 23:52:49 +02:00
|
|
|
"top_level_index_pinning": lambda: random.randint(0, 3),
|
|
|
|
"unpartitioned_pinning": lambda: random.randint(0, 3),
|
2018-06-02 01:33:07 +02:00
|
|
|
"use_direct_reads": lambda: random.randint(0, 1),
|
|
|
|
"use_direct_io_for_flush_and_compaction": lambda: random.randint(0, 1),
|
2020-04-25 08:58:13 +02:00
|
|
|
"mock_direct_io": False,
|
2021-05-07 01:30:56 +02:00
|
|
|
"use_clock_cache": 0, # currently broken
|
2018-05-09 22:32:03 +02:00
|
|
|
"use_full_merge_v1": lambda: random.randint(0, 1),
|
|
|
|
"use_merge": lambda: random.randint(0, 1),
|
Add Bloom/Ribbon hybrid API support (#8679)
Summary:
This is essentially resurrection and fixing of the part of
https://github.com/facebook/rocksdb/issues/8198 that was reverted in https://github.com/facebook/rocksdb/issues/8212, using data added in https://github.com/facebook/rocksdb/issues/8246. Basically,
when configuring Ribbon filter, you can specify an LSM level before which
Bloom will be used instead of Ribbon. But Bloom is only considered for
Leveled and Universal compaction styles and file going into a known LSM
level. This way, SST file writer, FIFO compaction, etc. use Ribbon filter as
you would expect with NewRibbonFilterPolicy.
So that this can be controlled with a single int value and so that flushes
can be distinguished from intra-L0, we consider flush to go to level -1 for
the purposes of this option. (Explained in API comment.)
I also expect the most common and recommended Ribbon configuration to
use Bloom during flush, to minimize slowing down writes and because according
to my estimates, Ribbon only pays off if the structure lives in memory for
more than an hour. Thus, I have changed the default for NewRibbonFilterPolicy
to be this mild hybrid configuration. I don't really want to add something like
NewHybridFilterPolicy because at least the mild hybrid configuration (Bloom for
flush, Ribbon otherwise) should be considered a natural choice.
C APIs also updated, but because they don't support overloading,
rocksdb_filterpolicy_create_ribbon is kept pure ribbon for clarity and
rocksdb_filterpolicy_create_ribbon_hybrid must be called for a hybrid
configuration. While touching C API, I changed bits per key options from
int to double.
BuiltinFilterPolicy is needed so that LevelThresholdFilterPolicy doesn't inherit
unused fields from BloomFilterPolicy.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8679
Test Plan: new + updated tests, including crash test
Reviewed By: jay-zhuang
Differential Revision: D30445797
Pulled By: pdillinger
fbshipit-source-id: 6f5aeddfd6d79f7e55493b563c2d1d2d568892e1
2021-08-21 02:59:24 +02:00
|
|
|
# 999 -> use Bloom API
|
|
|
|
"ribbon_starting_level": lambda: random.choice([random.randint(-1, 10), 999]),
|
|
|
|
"use_block_based_filter": lambda: random.randint(0, 1),
|
2022-01-31 22:19:55 +01:00
|
|
|
"value_size_mult": 32,
|
2015-10-19 20:43:14 +02:00
|
|
|
"verify_checksum": 1,
|
|
|
|
"write_buffer_size": 4 * 1024 * 1024,
|
|
|
|
"writepercent": 35,
|
2019-11-27 22:17:28 +01:00
|
|
|
"format_version": lambda: random.choice([2, 3, 4, 5, 5]),
|
2018-09-19 21:08:42 +02:00
|
|
|
"index_block_restart_interval": lambda: random.choice(range(1, 16)),
|
2019-05-09 22:03:37 +02:00
|
|
|
"use_multiget" : lambda: random.randint(0, 1),
|
2019-08-27 00:00:04 +02:00
|
|
|
"periodic_compaction_seconds" :
|
|
|
|
lambda: random.choice([0, 0, 1, 2, 10, 100, 1000]),
|
|
|
|
"compaction_ttl" : lambda: random.choice([0, 0, 1, 2, 10, 100, 1000]),
|
2019-11-14 22:59:43 +01:00
|
|
|
# Test small max_manifest_file_size in a smaller chance, as most of the
|
|
|
|
# time we wnat manifest history to be preserved to help debug
|
|
|
|
"max_manifest_file_size" : lambda : random.choice(
|
2019-12-17 00:24:26 +01:00
|
|
|
[t * 16384 if t < 3 else 1024 * 1024 * 1024 for t in range(1, 30)]),
|
|
|
|
# Sync mode might make test runs slower so running it in a smaller chance
|
|
|
|
"sync" : lambda : random.choice(
|
2020-01-17 10:44:47 +01:00
|
|
|
[1 if t == 0 else 0 for t in range(0, 20)]),
|
2022-04-04 18:32:57 +02:00
|
|
|
# Disable compaction_readahead_size because the test is not passing.
|
2019-12-18 03:23:26 +01:00
|
|
|
#"compaction_readahead_size" : lambda : random.choice(
|
|
|
|
# [0, 0, 1024 * 1024]),
|
2019-12-17 00:24:26 +01:00
|
|
|
"db_write_buffer_size" : lambda: random.choice(
|
|
|
|
[0, 0, 0, 1024 * 1024, 8 * 1024 * 1024, 128 * 1024 * 1024]),
|
|
|
|
"avoid_unnecessary_blocking_io" : random.randint(0, 1),
|
|
|
|
"write_dbid_to_manifest" : random.randint(0, 1),
|
2022-02-15 22:55:22 +01:00
|
|
|
"avoid_flush_during_recovery" : lambda: random.choice(
|
2020-04-16 21:09:18 +02:00
|
|
|
[1 if t == 0 else 0 for t in range(0, 8)]),
|
2019-12-17 00:24:26 +01:00
|
|
|
"max_write_batch_group_size_bytes" : lambda: random.choice(
|
|
|
|
[16, 64, 1024 * 1024, 16 * 1024 * 1024]),
|
2019-12-30 19:13:50 +01:00
|
|
|
"level_compaction_dynamic_level_bytes" : True,
|
2019-12-20 17:46:52 +01:00
|
|
|
"verify_checksum_one_in": 1000000,
|
|
|
|
"verify_db_one_in": 100000,
|
2020-01-10 06:25:40 +01:00
|
|
|
"continuous_verification_interval" : 0,
|
|
|
|
"max_key_len": 3,
|
2020-04-11 02:18:56 +02:00
|
|
|
"key_len_percent_dist": "1,30,69",
|
2021-09-21 23:47:09 +02:00
|
|
|
"read_fault_one_in": lambda: random.choice([0, 32, 1000]),
|
2021-07-06 20:04:04 +02:00
|
|
|
"open_metadata_write_fault_one_in": lambda: random.choice([0, 0, 8]),
|
|
|
|
"open_write_fault_one_in": lambda: random.choice([0, 0, 16]),
|
|
|
|
"open_read_fault_one_in": lambda: random.choice([0, 0, 32]),
|
2020-07-14 21:10:56 +02:00
|
|
|
"sync_fault_injection": False,
|
|
|
|
"get_property_one_in": 1000000,
|
2020-10-09 17:30:44 +02:00
|
|
|
"paranoid_file_checks": lambda: random.choice([0, 1, 1, 1]),
|
2020-11-03 22:54:01 +01:00
|
|
|
"max_write_buffer_size_to_maintain": lambda: random.choice(
|
|
|
|
[0, 1024 * 1024, 2 * 1024 * 1024, 4 * 1024 * 1024, 8 * 1024 * 1024]),
|
Fix an error while running db_crashtest for non-user-ts tests (#8091)
Summary:
Fix the following error while running `make crash_test`
```
Traceback (most recent call last):
File "tools/db_crashtest.py", line 705, in <module>
main()
File "tools/db_crashtest.py", line 696, in main
blackbox_crash_main(args, unknown_args)
File "tools/db_crashtest.py", line 479, in blackbox_crash_main
+ list({'db': dbname}.items())), unknown_args)
File "tools/db_crashtest.py", line 414, in gen_cmd
finalzied_params = finalize_and_sanitize(params)
File "tools/db_crashtest.py", line 331, in finalize_and_sanitize
dest_params.get("user_timestamp_size") > 0):
TypeError: '>' not supported between instances of 'NoneType' and 'int'
```
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8091
Test Plan: make crash_test
Reviewed By: ltamasi
Differential Revision: D27268276
Pulled By: riversand963
fbshipit-source-id: ed2873b9587ecc51e24abc35ef2bd3d91fb1ed1b
2021-03-23 20:43:19 +01:00
|
|
|
"user_timestamp_size": 0,
|
2021-11-08 19:26:48 +01:00
|
|
|
"secondary_cache_fault_one_in" : lambda: random.choice([0, 0, 32]),
|
2021-12-08 21:43:09 +01:00
|
|
|
"prepopulate_block_cache" : lambda: random.choice([0, 1]),
|
2022-01-27 23:53:39 +01:00
|
|
|
"memtable_prefix_bloom_size_ratio": lambda: random.choice([0.001, 0.01, 0.1, 0.5]),
|
|
|
|
"memtable_whole_key_filtering": lambda: random.randint(0, 1),
|
Detect (new) Bloom/Ribbon Filter construction corruption (#9342)
Summary:
Note: rebase on and merge after https://github.com/facebook/rocksdb/pull/9349, https://github.com/facebook/rocksdb/pull/9345, (optional) https://github.com/facebook/rocksdb/pull/9393
**Context:**
(Quoted from pdillinger) Layers of information during new Bloom/Ribbon Filter construction in building block-based tables includes the following:
a) set of keys to add to filter
b) set of hashes to add to filter (64-bit hash applied to each key)
c) set of Bloom indices to set in filter, with duplicates
d) set of Bloom indices to set in filter, deduplicated
e) final filter and its checksum
This PR aims to detect corruption (e.g, unexpected hardware/software corruption on data structures residing in the memory for a long time) from b) to e) and leave a) as future works for application level.
- b)'s corruption is detected by verifying the xor checksum of the hash entries calculated as the entries accumulate before being added to the filter. (i.e, `XXPH3FilterBitsBuilder::MaybeVerifyHashEntriesChecksum()`)
- c) - e)'s corruption is detected by verifying the hash entries indeed exists in the constructed filter by re-querying these hash entries in the filter (i.e, `FilterBitsBuilder::MaybePostVerify()`) after computing the block checksum (except for PartitionFilter, which is done right after each `FilterBitsBuilder::Finish` for impl simplicity - see code comment for more). For this stage of detection, we assume hash entries are not corrupted after checking on b) since the time interval from b) to c) is relatively short IMO.
Option to enable this feature of detection is `BlockBasedTableOptions::detect_filter_construct_corruption` which is false by default.
**Summary:**
- Implemented new functions `XXPH3FilterBitsBuilder::MaybeVerifyHashEntriesChecksum()` and `FilterBitsBuilder::MaybePostVerify()`
- Ensured hash entries, final filter and banding and their [cache reservation ](https://github.com/facebook/rocksdb/issues/9073) are released properly despite corruption
- See [Filter.construction.artifacts.release.point.pdf ](https://github.com/facebook/rocksdb/files/7923487/Design.Filter.construction.artifacts.release.point.pdf) for high-level design
- Bundled and refactored hash entries's related artifact in XXPH3FilterBitsBuilder into `HashEntriesInfo` for better control on lifetime of these artifact during `SwapEntires`, `ResetEntries`
- Ensured RocksDB block-based table builder calls `FilterBitsBuilder::MaybePostVerify()` after constructing the filter by `FilterBitsBuilder::Finish()`
- When encountering such filter construction corruption, stop writing the filter content to files and mark such a block-based table building non-ok by storing the corruption status in the builder.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/9342
Test Plan:
- Added new unit test `DBFilterConstructionCorruptionTestWithParam.DetectCorruption`
- Included this new feature in `DBFilterConstructionReserveMemoryTestWithParam.ReserveMemory` as this feature heavily touch ReserveMemory's impl
- For fallback case, I run `./filter_bench -impl=3 -detect_filter_construct_corruption=true -reserve_table_builder_memory=true -strict_capacity_limit=true -quick -runs 10 | grep 'Build avg'` to make sure nothing break.
- Added to `filter_bench`: increased filter construction time by **30%**, mostly by `MaybePostVerify()`
- FastLocalBloom
- Before change: `./filter_bench -impl=2 -quick -runs 10 | grep 'Build avg'`: **28.86643s**
- After change:
- `./filter_bench -impl=2 -detect_filter_construct_corruption=false -quick -runs 10 | grep 'Build avg'` (expect a tiny increase due to MaybePostVerify is always called regardless): **27.6644s (-4% perf improvement might be due to now we don't drop bloom hash entry in `AddAllEntries` along iteration but in bulk later, same with the bypassing-MaybePostVerify case below)**
- `./filter_bench -impl=2 -detect_filter_construct_corruption=true -quick -runs 10 | grep 'Build avg'` (expect acceptable increase): **34.41159s (+20%)**
- `./filter_bench -impl=2 -detect_filter_construct_corruption=true -quick -runs 10 | grep 'Build avg'` (by-passing MaybePostVerify, expect minor increase): **27.13431s (-6%)**
- Standard128Ribbon
- Before change: `./filter_bench -impl=3 -quick -runs 10 | grep 'Build avg'`: **122.5384s**
- After change:
- `./filter_bench -impl=3 -detect_filter_construct_corruption=false -quick -runs 10 | grep 'Build avg'` (expect a tiny increase due to MaybePostVerify is always called regardless - verified by removing MaybePostVerify under this case and found only +-1ns difference): **124.3588s (+2%)**
- `./filter_bench -impl=3 -detect_filter_construct_corruption=true -quick -runs 10 | grep 'Build avg'`(expect acceptable increase): **159.4946s (+30%)**
- `./filter_bench -impl=3 -detect_filter_construct_corruption=true -quick -runs 10 | grep 'Build avg'`(by-passing MaybePostVerify, expect minor increase) : **125.258s (+2%)**
- Added to `db_stress`: `make crash_test`, `./db_stress --detect_filter_construct_corruption=true`
- Manually smoke-tested: manually corrupted the filter construction in some db level tests with basic PUT and background flush. As expected, the error did get returned to users in subsequent PUT and Flush status.
Reviewed By: pdillinger
Differential Revision: D33746928
Pulled By: hx235
fbshipit-source-id: cb056426be5a7debc1cd16f23bc250f36a08ca57
2022-02-02 02:41:20 +01:00
|
|
|
"detect_filter_construct_corruption": lambda: random.choice([0, 1]),
|
2022-03-30 22:52:37 +02:00
|
|
|
"adaptive_readahead": lambda: random.choice([0, 1]),
|
|
|
|
"async_io": lambda: random.choice([0, 1]),
|
2022-04-07 00:47:09 +02:00
|
|
|
"wal_compression": lambda: random.choice(["none", "zstd"]),
|
2015-10-19 20:43:14 +02:00
|
|
|
}
|
2013-03-13 07:20:14 +01:00
|
|
|
|
2018-06-04 06:28:41 +02:00
|
|
|
_TEST_DIR_ENV_VAR = 'TEST_TMPDIR'
|
2020-04-25 08:58:13 +02:00
|
|
|
_DEBUG_LEVEL_ENV_VAR = 'DEBUG_LEVEL'
|
|
|
|
|
2021-06-28 08:53:47 +02:00
|
|
|
stress_cmd = "./db_stress"
|
2020-04-25 08:58:13 +02:00
|
|
|
|
|
|
|
def is_release_mode():
|
|
|
|
return os.environ.get(_DEBUG_LEVEL_ENV_VAR) == "0"
|
2018-06-04 06:28:41 +02:00
|
|
|
|
2013-04-10 21:15:30 +02:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
def get_dbname(test_name):
|
2020-04-25 08:58:13 +02:00
|
|
|
test_dir_name = "rocksdb_crashtest_" + test_name
|
2018-06-04 06:28:41 +02:00
|
|
|
test_tmpdir = os.environ.get(_TEST_DIR_ENV_VAR)
|
2015-08-04 20:35:44 +02:00
|
|
|
if test_tmpdir is None or test_tmpdir == "":
|
2020-04-25 08:58:13 +02:00
|
|
|
dbname = tempfile.mkdtemp(prefix=test_dir_name)
|
2015-08-04 20:35:44 +02:00
|
|
|
else:
|
2020-04-25 08:58:13 +02:00
|
|
|
dbname = test_tmpdir + "/" + test_dir_name
|
2015-08-04 21:20:38 +02:00
|
|
|
shutil.rmtree(dbname, True)
|
2018-06-02 01:33:07 +02:00
|
|
|
os.mkdir(dbname)
|
2015-10-19 20:43:14 +02:00
|
|
|
return dbname
|
|
|
|
|
2021-09-28 23:12:23 +02:00
|
|
|
expected_values_dir = None
|
|
|
|
def setup_expected_values_dir():
|
|
|
|
global expected_values_dir
|
|
|
|
if expected_values_dir is not None:
|
|
|
|
return expected_values_dir
|
|
|
|
expected_dir_prefix = "rocksdb_crashtest_expected_"
|
2020-10-12 23:08:35 +02:00
|
|
|
test_tmpdir = os.environ.get(_TEST_DIR_ENV_VAR)
|
|
|
|
if test_tmpdir is None or test_tmpdir == "":
|
2021-09-28 23:12:23 +02:00
|
|
|
expected_values_dir = tempfile.mkdtemp(
|
|
|
|
prefix=expected_dir_prefix)
|
2020-10-12 23:08:35 +02:00
|
|
|
else:
|
2021-09-28 23:12:23 +02:00
|
|
|
# if tmpdir is specified, store the expected_values_dir under that dir
|
|
|
|
expected_values_dir = test_tmpdir + "/rocksdb_crashtest_expected"
|
|
|
|
if os.path.exists(expected_values_dir):
|
|
|
|
shutil.rmtree(expected_values_dir)
|
|
|
|
os.mkdir(expected_values_dir)
|
|
|
|
return expected_values_dir
|
2020-10-12 23:08:35 +02:00
|
|
|
|
2022-03-17 03:00:04 +01:00
|
|
|
multiops_txn_key_spaces_file = None
|
|
|
|
def setup_multiops_txn_key_spaces_file():
|
|
|
|
global multiops_txn_key_spaces_file
|
|
|
|
if multiops_txn_key_spaces_file is not None:
|
|
|
|
return multiops_txn_key_spaces_file
|
|
|
|
key_spaces_file_prefix = "rocksdb_crashtest_multiops_txn_key_spaces"
|
|
|
|
test_tmpdir = os.environ.get(_TEST_DIR_ENV_VAR)
|
|
|
|
if test_tmpdir is None or test_tmpdir == "":
|
|
|
|
multiops_txn_key_spaces_file = tempfile.mkstemp(
|
|
|
|
prefix=key_spaces_file_prefix)[1]
|
|
|
|
else:
|
2022-03-24 06:29:50 +01:00
|
|
|
if not os.path.exists(test_tmpdir):
|
|
|
|
os.mkdir(test_tmpdir)
|
2022-03-17 03:00:04 +01:00
|
|
|
multiops_txn_key_spaces_file = tempfile.mkstemp(
|
|
|
|
prefix=key_spaces_file_prefix, dir=test_tmpdir)[1]
|
|
|
|
return multiops_txn_key_spaces_file
|
|
|
|
|
2018-06-02 01:33:07 +02:00
|
|
|
|
|
|
|
def is_direct_io_supported(dbname):
|
|
|
|
with tempfile.NamedTemporaryFile(dir=dbname) as f:
|
|
|
|
try:
|
|
|
|
os.open(f.name, os.O_DIRECT)
|
2020-03-25 04:57:53 +01:00
|
|
|
except BaseException:
|
2018-06-02 01:33:07 +02:00
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
blackbox_default_params = {
|
2022-01-06 01:12:12 +01:00
|
|
|
"disable_wal": lambda: random.choice([0, 0, 0, 1]),
|
2015-10-19 20:43:14 +02:00
|
|
|
# total time for this script to test db_stress
|
|
|
|
"duration": 6000,
|
|
|
|
# time for one db_stress instance to run
|
|
|
|
"interval": 120,
|
|
|
|
# since we will be killing anyway, use large value for ops_per_thread
|
|
|
|
"ops_per_thread": 100000000,
|
2022-01-06 01:12:12 +01:00
|
|
|
"reopen": 0,
|
2015-10-19 20:43:14 +02:00
|
|
|
"set_options_one_in": 10000,
|
|
|
|
}
|
|
|
|
|
|
|
|
whitebox_default_params = {
|
2022-01-06 01:12:12 +01:00
|
|
|
# TODO: enable this once we figure out how to adjust kill odds for WAL-
|
|
|
|
# disabled runs, and either (1) separate full `db_stress` runs out of
|
|
|
|
# whitebox crash or (2) support verification at end of `db_stress` runs
|
|
|
|
# that ran with WAL disabled.
|
|
|
|
"disable_wal": 0,
|
2015-10-19 20:43:14 +02:00
|
|
|
"duration": 10000,
|
|
|
|
"log2_keys_per_lock": 10,
|
|
|
|
"ops_per_thread": 200000,
|
2016-07-22 20:46:13 +02:00
|
|
|
"random_kill_odd": 888887,
|
2022-01-06 01:12:12 +01:00
|
|
|
"reopen": 20,
|
2015-10-19 20:43:14 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
simple_default_params = {
|
2016-02-05 00:21:32 +01:00
|
|
|
"allow_concurrent_memtable_write": lambda: random.randint(0, 1),
|
2018-05-09 22:32:03 +02:00
|
|
|
"column_families": 1,
|
2022-03-12 00:47:30 +01:00
|
|
|
"experimental_mempurge_threshold": lambda: 10.0*random.random(),
|
2015-10-19 20:43:14 +02:00
|
|
|
"max_background_compactions": 1,
|
|
|
|
"max_bytes_for_level_base": 67108864,
|
|
|
|
"memtablerep": "skip_list",
|
|
|
|
"target_file_size_base": 16777216,
|
|
|
|
"target_file_size_multiplier": 1,
|
|
|
|
"test_batches_snapshots": 0,
|
|
|
|
"write_buffer_size": 32 * 1024 * 1024,
|
2019-12-17 00:24:26 +01:00
|
|
|
"level_compaction_dynamic_level_bytes": False,
|
2020-10-09 17:30:44 +02:00
|
|
|
"paranoid_file_checks": lambda: random.choice([0, 1, 1, 1]),
|
2015-10-19 20:43:14 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
blackbox_simple_default_params = {
|
|
|
|
"open_files": -1,
|
|
|
|
"set_options_one_in": 0,
|
|
|
|
}
|
|
|
|
|
2018-05-09 22:32:03 +02:00
|
|
|
whitebox_simple_default_params = {}
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2019-08-23 01:30:30 +02:00
|
|
|
cf_consistency_params = {
|
|
|
|
"disable_wal": lambda: random.randint(0, 1),
|
2018-10-30 22:01:38 +01:00
|
|
|
"reopen": 0,
|
2019-08-23 01:30:30 +02:00
|
|
|
"test_cf_consistency": 1,
|
2018-10-30 22:01:38 +01:00
|
|
|
# use small value for write_buffer_size so that RocksDB triggers flush
|
|
|
|
# more frequently
|
|
|
|
"write_buffer_size": 1024 * 1024,
|
2019-10-30 02:15:08 +01:00
|
|
|
"enable_pipelined_write": lambda: random.randint(0, 1),
|
2020-06-20 01:03:56 +02:00
|
|
|
# Snapshots are used heavily in this test mode, while they are incompatible
|
|
|
|
# with compaction filter.
|
|
|
|
"enable_compaction_filter": 0,
|
2018-10-30 22:01:38 +01:00
|
|
|
}
|
|
|
|
|
2019-12-12 00:59:45 +01:00
|
|
|
txn_params = {
|
|
|
|
"use_txn" : 1,
|
2019-12-12 19:34:52 +01:00
|
|
|
# Avoid lambda to set it once for the entire test
|
|
|
|
"txn_write_policy": random.randint(0, 2),
|
2019-12-13 19:23:01 +01:00
|
|
|
"unordered_write": random.randint(0, 1),
|
2022-01-06 01:12:12 +01:00
|
|
|
# TODO: there is such a thing as transactions with WAL disabled. We should
|
|
|
|
# cover that case.
|
2019-12-12 00:59:45 +01:00
|
|
|
"disable_wal": 0,
|
|
|
|
# OpenReadOnly after checkpoint is not currnetly compatible with WritePrepared txns
|
|
|
|
"checkpoint_one_in": 0,
|
|
|
|
# pipeline write is not currnetly compatible with WritePrepared txns
|
|
|
|
"enable_pipelined_write": 0,
|
|
|
|
}
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2020-06-13 04:24:11 +02:00
|
|
|
best_efforts_recovery_params = {
|
|
|
|
"best_efforts_recovery": True,
|
|
|
|
"skip_verifydb": True,
|
|
|
|
"verify_db_one_in": 0,
|
|
|
|
"continuous_verification_interval": 0,
|
|
|
|
}
|
|
|
|
|
2021-02-02 20:39:20 +01:00
|
|
|
blob_params = {
|
|
|
|
"allow_setting_blob_options_dynamically": 1,
|
|
|
|
# Enable blob files and GC with a 75% chance initially; note that they might still be
|
|
|
|
# enabled/disabled during the test via SetOptions
|
|
|
|
"enable_blob_files": lambda: random.choice([0] + [1] * 3),
|
2021-03-22 22:36:07 +01:00
|
|
|
"min_blob_size": lambda: random.choice([0, 8, 16]),
|
2021-02-02 20:39:20 +01:00
|
|
|
"blob_file_size": lambda: random.choice([1048576, 16777216, 268435456, 1073741824]),
|
|
|
|
"blob_compression_type": lambda: random.choice(["none", "snappy", "lz4", "zstd"]),
|
|
|
|
"enable_blob_garbage_collection": lambda: random.choice([0] + [1] * 3),
|
|
|
|
"blob_garbage_collection_age_cutoff": lambda: random.choice([0.0, 0.25, 0.5, 0.75, 1.0]),
|
Make it possible to force the garbage collection of the oldest blob files (#8994)
Summary:
The current BlobDB garbage collection logic works by relocating the valid
blobs from the oldest blob files as they are encountered during compaction,
and cleaning up blob files once they contain nothing but garbage. However,
with sufficiently skewed workloads, it is theoretically possible to end up in a
situation when few or no compactions get scheduled for the SST files that contain
references to the oldest blob files, which can lead to increased space amp due
to the lack of GC.
In order to efficiently handle such workloads, the patch adds a new BlobDB
configuration option called `blob_garbage_collection_force_threshold`,
which signals to BlobDB to schedule targeted compactions for the SST files
that keep alive the oldest batch of blob files if the overall ratio of garbage in
the given blob files meets the threshold *and* all the given blob files are
eligible for GC based on `blob_garbage_collection_age_cutoff`. (For example,
if the new option is set to 0.9, targeted compactions will get scheduled if the
sum of garbage bytes meets or exceeds 90% of the sum of total bytes in the
oldest blob files, assuming all affected blob files are below the age-based cutoff.)
The net result of these targeted compactions is that the valid blobs in the oldest
blob files are relocated and the oldest blob files themselves cleaned up (since
*all* SST files that rely on them get compacted away).
These targeted compactions are similar to periodic compactions in the sense
that they force certain SST files that otherwise would not get picked up to undergo
compaction and also in the sense that instead of merging files from multiple levels,
they target a single file. (Note: such compactions might still include neighboring files
from the same level due to the need of having a "clean cut" boundary but they never
include any files from any other level.)
This functionality is currently only supported with the leveled compaction style
and is inactive by default (since the default value is set to 1.0, i.e. 100%).
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8994
Test Plan: Ran `make check` and tested using `db_bench` and the stress/crash tests.
Reviewed By: riversand963
Differential Revision: D31489850
Pulled By: ltamasi
fbshipit-source-id: 44057d511726a0e2a03c5d9313d7511b3f0c4eab
2021-10-12 03:00:44 +02:00
|
|
|
"blob_garbage_collection_force_threshold": lambda: random.choice([0.5, 0.75, 1.0]),
|
2021-11-20 02:52:42 +01:00
|
|
|
"blob_compaction_readahead_size": lambda: random.choice([0, 1048576, 4194304]),
|
2021-02-02 20:39:20 +01:00
|
|
|
}
|
|
|
|
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
ts_params = {
|
|
|
|
"test_cf_consistency": 0,
|
|
|
|
"test_batches_snapshots": 0,
|
|
|
|
"user_timestamp_size": 8,
|
|
|
|
"use_merge": 0,
|
|
|
|
"use_full_merge_v1": 0,
|
|
|
|
"use_txn": 0,
|
|
|
|
"read_only": 0,
|
2022-02-04 01:17:30 +01:00
|
|
|
"backup_one_in": 0,
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
"secondary_catch_up_one_in": 0,
|
|
|
|
"continuous_verification_interval": 0,
|
|
|
|
"checkpoint_one_in": 0,
|
|
|
|
"enable_blob_files": 0,
|
|
|
|
"use_blob_db": 0,
|
|
|
|
"enable_compaction_filter": 0,
|
|
|
|
"ingest_external_file_one_in": 0,
|
2021-08-25 04:03:58 +02:00
|
|
|
"use_block_based_filter": 0,
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
}
|
|
|
|
|
2022-03-17 03:00:04 +01:00
|
|
|
multiops_txn_default_params = {
|
|
|
|
"test_cf_consistency": 0,
|
|
|
|
"test_batches_snapshots": 0,
|
|
|
|
"test_multi_ops_txns": 1,
|
|
|
|
"use_txn": 1,
|
|
|
|
"two_write_queues": lambda: random.choice([0, 1]),
|
|
|
|
# TODO: enable write-prepared
|
|
|
|
"disable_wal": 0,
|
|
|
|
"use_only_the_last_commit_time_batch_for_recovery": lambda: random.choice([0, 1]),
|
|
|
|
"clear_column_family_one_in": 0,
|
|
|
|
"column_families": 1,
|
|
|
|
"enable_pipelined_write": lambda: random.choice([0, 1]),
|
|
|
|
# This test already acquires snapshots in reads
|
|
|
|
"acquire_snapshot_one_in": 0,
|
|
|
|
"backup_one_in": 0,
|
|
|
|
"writepercent": 0,
|
|
|
|
"delpercent": 0,
|
|
|
|
"delrangepercent": 0,
|
|
|
|
"customopspercent": 80,
|
|
|
|
"readpercent": 5,
|
|
|
|
"iterpercent": 15,
|
|
|
|
"prefixpercent": 0,
|
|
|
|
"verify_db_one_in": 1000,
|
|
|
|
"continuous_verification_interval": 1000,
|
|
|
|
"delay_snapshot_read_one_in": 3,
|
|
|
|
# 65536 is the smallest possible value for write_buffer_size. Smaller
|
|
|
|
# values will be sanitized to 65536 during db open. SetOptions currently
|
|
|
|
# does not sanitize options, but very small write_buffer_size may cause
|
|
|
|
# assertion failure in
|
|
|
|
# https://github.com/facebook/rocksdb/blob/7.0.fb/db/memtable.cc#L117.
|
|
|
|
"write_buffer_size": 65536,
|
|
|
|
# flush more frequently to generate more files, thus trigger more
|
|
|
|
# compactions.
|
|
|
|
"flush_one_in": 1000,
|
|
|
|
"key_spaces_path": setup_multiops_txn_key_spaces_file(),
|
|
|
|
}
|
|
|
|
|
|
|
|
multiops_wc_txn_params = {
|
|
|
|
"txn_write_policy": 0,
|
|
|
|
# TODO re-enable pipelined write. Not well tested atm
|
|
|
|
"enable_pipelined_write": 0,
|
|
|
|
}
|
|
|
|
|
|
|
|
multiops_wp_txn_params = {
|
|
|
|
"txn_write_policy": 1,
|
|
|
|
"wp_snapshot_cache_bits": 1,
|
|
|
|
# try small wp_commit_cache_bits, e.g. 0 once we explore storing full
|
|
|
|
# commit sequence numbers in commit cache
|
|
|
|
"wp_commit_cache_bits": 10,
|
|
|
|
# pipeline write is not currnetly compatible with WritePrepared txns
|
|
|
|
"enable_pipelined_write": 0,
|
|
|
|
# OpenReadOnly after checkpoint is not currnetly compatible with WritePrepared txns
|
|
|
|
"checkpoint_one_in": 0,
|
|
|
|
}
|
|
|
|
|
2016-02-05 00:21:32 +01:00
|
|
|
def finalize_and_sanitize(src_params):
|
|
|
|
dest_params = dict([(k, v() if callable(v) else v)
|
|
|
|
for (k, v) in src_params.items()])
|
Limit buffering for collecting samples for compression dictionary (#7970)
Summary:
For dictionary compression, we need to collect some representative samples of the data to be compressed, which we use to either generate or train (when `CompressionOptions::zstd_max_train_bytes > 0`) a dictionary. Previously, the strategy was to buffer all the data blocks during flush, and up to the target file size during compaction. That strategy allowed us to randomly pick samples from as wide a range as possible that'd be guaranteed to land in a single output file.
However, some users try to make huge files in memory-constrained environments, where this strategy can cause OOM. This PR introduces an option, `CompressionOptions::max_dict_buffer_bytes`, that limits how much data blocks are buffered before we switch to unbuffered mode (which means creating the per-SST dictionary, writing out the buffered data, and compressing/writing new blocks as soon as they are built). It is not strict as we currently buffer more than just data blocks -- also keys are buffered. But it does make a step towards giving users predictable memory usage.
Related changes include:
- Changed sampling for dictionary compression to select unique data blocks when there is limited availability of data blocks
- Made use of `BlockBuilder::SwapAndReset()` to save an allocation+memcpy when buffering data blocks for building a dictionary
- Changed `ParseBoolean()` to accept an input containing characters after the boolean. This is necessary since, with this PR, a value for `CompressionOptions::enabled` is no longer necessarily the final component in the `CompressionOptions` string.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/7970
Test Plan:
- updated `CompressionOptions` unit tests to verify limit is respected (to the extent expected in the current implementation) in various scenarios of flush/compaction to bottommost/non-bottommost level
- looked at jemalloc heap profiles right before and after switching to unbuffered mode during flush/compaction. Verified memory usage in buffering is proportional to the limit set.
Reviewed By: pdillinger
Differential Revision: D26467994
Pulled By: ajkr
fbshipit-source-id: 3da4ef9fba59974e4ef40e40c01611002c861465
2021-02-19 23:06:59 +01:00
|
|
|
if dest_params.get("compression_max_dict_bytes") == 0:
|
|
|
|
dest_params["compression_zstd_max_train_bytes"] = 0
|
|
|
|
dest_params["compression_max_dict_buffer_bytes"] = 0
|
|
|
|
if dest_params.get("compression_type") != "zstd":
|
2018-08-07 00:23:03 +02:00
|
|
|
dest_params["compression_zstd_max_train_bytes"] = 0
|
2016-02-05 00:21:32 +01:00
|
|
|
if dest_params.get("allow_concurrent_memtable_write", 1) == 1:
|
2016-02-05 23:30:18 +01:00
|
|
|
dest_params["memtablerep"] = "skip_list"
|
2020-04-25 08:58:13 +02:00
|
|
|
if dest_params["mmap_read"] == 1:
|
2018-06-02 01:33:07 +02:00
|
|
|
dest_params["use_direct_io_for_flush_and_compaction"] = 0
|
|
|
|
dest_params["use_direct_reads"] = 0
|
2020-04-25 08:58:13 +02:00
|
|
|
if (dest_params["use_direct_io_for_flush_and_compaction"] == 1
|
|
|
|
or dest_params["use_direct_reads"] == 1) and \
|
|
|
|
not is_direct_io_supported(dest_params["db"]):
|
|
|
|
if is_release_mode():
|
2020-11-10 19:48:46 +01:00
|
|
|
print("{} does not support direct IO. Disabling use_direct_reads and "
|
|
|
|
"use_direct_io_for_flush_and_compaction.\n".format(
|
|
|
|
dest_params["db"]))
|
|
|
|
dest_params["use_direct_reads"] = 0
|
|
|
|
dest_params["use_direct_io_for_flush_and_compaction"] = 0
|
2020-04-25 08:58:13 +02:00
|
|
|
else:
|
|
|
|
dest_params["mock_direct_io"] = True
|
|
|
|
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
# DeleteRange is not currnetly compatible with Txns and timestamp
|
|
|
|
if (dest_params.get("test_batches_snapshots") == 1 or
|
|
|
|
dest_params.get("use_txn") == 1 or
|
|
|
|
dest_params.get("user_timestamp_size") > 0):
|
2018-12-18 22:30:56 +01:00
|
|
|
dest_params["delpercent"] += dest_params["delrangepercent"]
|
|
|
|
dest_params["delrangepercent"] = 0
|
2019-12-13 19:23:01 +01:00
|
|
|
# Only under WritePrepared txns, unordered_write would provide the same guarnatees as vanilla rocksdb
|
|
|
|
if dest_params.get("unordered_write", 0) == 1:
|
|
|
|
dest_params["txn_write_policy"] = 1
|
|
|
|
dest_params["allow_concurrent_memtable_write"] = 1
|
2019-08-23 01:30:30 +02:00
|
|
|
if dest_params.get("disable_wal", 0) == 1:
|
|
|
|
dest_params["atomic_flush"] = 1
|
2022-01-06 01:12:12 +01:00
|
|
|
# The `DbStressCompactionFilter` can apply memtable updates to SST
|
|
|
|
# files, which would be problematic without WAL since such updates are
|
|
|
|
# expected to be lost in crash recoveries.
|
|
|
|
dest_params["enable_compaction_filter"] = 0
|
2019-12-18 03:23:26 +01:00
|
|
|
dest_params["sync"] = 0
|
2020-12-17 20:51:04 +01:00
|
|
|
dest_params["write_fault_one_in"] = 0
|
2019-08-27 00:00:04 +02:00
|
|
|
if dest_params.get("open_files", 1) != -1:
|
|
|
|
# Compaction TTL and periodic compactions are only compatible
|
|
|
|
# with open_files = -1
|
|
|
|
dest_params["compaction_ttl"] = 0
|
|
|
|
dest_params["periodic_compaction_seconds"] = 0
|
2019-08-28 02:54:18 +02:00
|
|
|
if dest_params.get("compaction_style", 0) == 2:
|
|
|
|
# Disable compaction TTL in FIFO compaction, because right
|
|
|
|
# now assertion failures are triggered.
|
|
|
|
dest_params["compaction_ttl"] = 0
|
2019-11-01 01:26:46 +01:00
|
|
|
dest_params["periodic_compaction_seconds"] = 0
|
2019-09-11 23:11:38 +02:00
|
|
|
if dest_params["partition_filters"] == 1:
|
2019-10-14 19:33:31 +02:00
|
|
|
if dest_params["index_type"] != 2:
|
|
|
|
dest_params["partition_filters"] = 0
|
|
|
|
else:
|
|
|
|
dest_params["use_block_based_filter"] = 0
|
Add Bloom/Ribbon hybrid API support (#8679)
Summary:
This is essentially resurrection and fixing of the part of
https://github.com/facebook/rocksdb/issues/8198 that was reverted in https://github.com/facebook/rocksdb/issues/8212, using data added in https://github.com/facebook/rocksdb/issues/8246. Basically,
when configuring Ribbon filter, you can specify an LSM level before which
Bloom will be used instead of Ribbon. But Bloom is only considered for
Leveled and Universal compaction styles and file going into a known LSM
level. This way, SST file writer, FIFO compaction, etc. use Ribbon filter as
you would expect with NewRibbonFilterPolicy.
So that this can be controlled with a single int value and so that flushes
can be distinguished from intra-L0, we consider flush to go to level -1 for
the purposes of this option. (Explained in API comment.)
I also expect the most common and recommended Ribbon configuration to
use Bloom during flush, to minimize slowing down writes and because according
to my estimates, Ribbon only pays off if the structure lives in memory for
more than an hour. Thus, I have changed the default for NewRibbonFilterPolicy
to be this mild hybrid configuration. I don't really want to add something like
NewHybridFilterPolicy because at least the mild hybrid configuration (Bloom for
flush, Ribbon otherwise) should be considered a natural choice.
C APIs also updated, but because they don't support overloading,
rocksdb_filterpolicy_create_ribbon is kept pure ribbon for clarity and
rocksdb_filterpolicy_create_ribbon_hybrid must be called for a hybrid
configuration. While touching C API, I changed bits per key options from
int to double.
BuiltinFilterPolicy is needed so that LevelThresholdFilterPolicy doesn't inherit
unused fields from BloomFilterPolicy.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8679
Test Plan: new + updated tests, including crash test
Reviewed By: jay-zhuang
Differential Revision: D30445797
Pulled By: pdillinger
fbshipit-source-id: 6f5aeddfd6d79f7e55493b563c2d1d2d568892e1
2021-08-21 02:59:24 +02:00
|
|
|
if dest_params["ribbon_starting_level"] < 999:
|
|
|
|
dest_params["use_block_based_filter"] = 0
|
2019-10-30 19:34:38 +01:00
|
|
|
if dest_params.get("atomic_flush", 0) == 1:
|
|
|
|
# disable pipelined write when atomic flush is used.
|
|
|
|
dest_params["enable_pipelined_write"] = 0
|
2020-06-24 01:24:15 +02:00
|
|
|
if dest_params.get("sst_file_manager_bytes_per_sec", 0) == 0:
|
|
|
|
dest_params["sst_file_manager_bytes_per_truncate"] = 0
|
2020-06-18 18:51:14 +02:00
|
|
|
if dest_params.get("enable_compaction_filter", 0) == 1:
|
|
|
|
# Compaction filter is incompatible with snapshots. Need to avoid taking
|
|
|
|
# snapshots, as well as avoid operations that use snapshots for
|
|
|
|
# verification.
|
|
|
|
dest_params["acquire_snapshot_one_in"] = 0
|
|
|
|
dest_params["compact_range_one_in"] = 0
|
|
|
|
# Give the iterator ops away to reads.
|
|
|
|
dest_params["readpercent"] += dest_params.get("iterpercent", 10)
|
|
|
|
dest_params["iterpercent"] = 0
|
|
|
|
dest_params["test_batches_snapshots"] = 0
|
2022-01-28 08:10:26 +01:00
|
|
|
if dest_params.get("prefix_size") == -1:
|
|
|
|
dest_params["readpercent"] += dest_params.get("prefixpercent", 20)
|
|
|
|
dest_params["prefixpercent"] = 0
|
|
|
|
dest_params["test_batches_snapshots"] = 0
|
Integrity protection for live updates to WriteBatch (#7748)
Summary:
This PR adds the foundation classes for key-value integrity protection and the first use case: protecting live updates from the source buffers added to `WriteBatch` through the destination buffer in `MemTable`. The width of the protection info is not yet configurable -- only eight bytes per key is supported. This PR allows users to enable protection by constructing `WriteBatch` with `protection_bytes_per_key == 8`. It does not yet expose a way for users to get integrity protection via other write APIs (e.g., `Put()`, `Merge()`, `Delete()`, etc.).
The foundation classes (`ProtectionInfo.*`) embed the coverage info in their type, and provide `Protect.*()` and `Strip.*()` functions to navigate between types with different coverage. For making bytes per key configurable (for powers of two up to eight) in the future, these classes are templated on the unsigned integer type used to store the protection info. That integer contains the XOR'd result of hashes with independent seeds for all covered fields. For integer fields, the hash is computed on the raw unadjusted bytes, so the result is endian-dependent. The most significant bytes are truncated when the hash value (8 bytes) is wider than the protection integer.
When `WriteBatch` is constructed with `protection_bytes_per_key == 8`, we hold a `ProtectionInfoKVOTC` (i.e., one that covers key, value, optype aka `ValueType`, timestamp, and CF ID) for each entry added to the batch. The protection info is generated from the original buffers passed by the user, as well as the original metadata generated internally. When writing to memtable, each entry is transformed to a `ProtectionInfoKVOTS` (i.e., dropping coverage of CF ID and adding coverage of sequence number), since at that point we know the sequence number, and have already selected a memtable corresponding to a particular CF. This protection info is verified once the entry is encoded in the `MemTable` buffer.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/7748
Test Plan:
- an integration test to verify a wide variety of single-byte changes to the encoded `MemTable` buffer are caught
- add to stress/crash test to verify it works in variety of configs/operations without intentional corruption
- [deferred] unit tests for `ProtectionInfo.*` classes for edge cases like KV swap, `SliceParts` and `Slice` APIs are interchangeable, etc.
Reviewed By: pdillinger
Differential Revision: D25754492
Pulled By: ajkr
fbshipit-source-id: e481bac6c03c2ab268be41359730f1ceb9964866
2021-01-29 21:17:17 +01:00
|
|
|
if dest_params.get("test_batches_snapshots") == 0:
|
|
|
|
dest_params["batch_protection_bytes_per_key"] = 0
|
2022-01-27 23:53:39 +01:00
|
|
|
if (dest_params.get("prefix_size") == -1 and
|
|
|
|
dest_params.get("memtable_whole_key_filtering") == 0):
|
|
|
|
dest_params["memtable_prefix_bloom_size_ratio"] = 0
|
2022-03-17 03:00:04 +01:00
|
|
|
if dest_params.get("two_write_queues") == 1:
|
|
|
|
dest_params["enable_pipelined_write"] = 0
|
2016-02-05 00:21:32 +01:00
|
|
|
return dest_params
|
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
def gen_cmd_params(args):
|
|
|
|
params = {}
|
|
|
|
|
2018-05-09 22:32:03 +02:00
|
|
|
params.update(default_params)
|
|
|
|
if args.test_type == 'blackbox':
|
|
|
|
params.update(blackbox_default_params)
|
|
|
|
if args.test_type == 'whitebox':
|
|
|
|
params.update(whitebox_default_params)
|
2015-10-19 20:43:14 +02:00
|
|
|
if args.simple:
|
|
|
|
params.update(simple_default_params)
|
|
|
|
if args.test_type == 'blackbox':
|
|
|
|
params.update(blackbox_simple_default_params)
|
|
|
|
if args.test_type == 'whitebox':
|
|
|
|
params.update(whitebox_simple_default_params)
|
2019-08-23 01:30:30 +02:00
|
|
|
if args.cf_consistency:
|
|
|
|
params.update(cf_consistency_params)
|
2019-12-12 00:59:45 +01:00
|
|
|
if args.txn:
|
|
|
|
params.update(txn_params)
|
2020-06-13 04:24:11 +02:00
|
|
|
if args.test_best_efforts_recovery:
|
|
|
|
params.update(best_efforts_recovery_params)
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
if args.enable_ts:
|
|
|
|
params.update(ts_params)
|
2022-03-17 03:00:04 +01:00
|
|
|
if args.test_multiops_txn:
|
|
|
|
params.update(multiops_txn_default_params)
|
|
|
|
if args.write_policy == 'write_committed':
|
|
|
|
params.update(multiops_wc_txn_params)
|
|
|
|
elif args.write_policy == 'write_prepared':
|
|
|
|
params.update(multiops_wp_txn_params)
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2021-02-02 20:39:20 +01:00
|
|
|
# Best-effort recovery and BlobDB are currently incompatible. Test BE recovery
|
|
|
|
# if specified on the command line; otherwise, apply BlobDB related overrides
|
|
|
|
# with a 10% chance.
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
if (not args.test_best_efforts_recovery and
|
|
|
|
not args.enable_ts and
|
|
|
|
random.choice([0] * 9 + [1]) == 1):
|
2021-02-02 20:39:20 +01:00
|
|
|
params.update(blob_params)
|
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
for k, v in vars(args).items():
|
|
|
|
if v is not None:
|
|
|
|
params[k] = v
|
|
|
|
return params
|
|
|
|
|
|
|
|
|
2018-05-09 22:32:03 +02:00
|
|
|
def gen_cmd(params, unknown_params):
|
2019-12-21 01:13:19 +01:00
|
|
|
finalzied_params = finalize_and_sanitize(params)
|
2021-06-28 08:53:47 +02:00
|
|
|
cmd = [stress_cmd] + [
|
2016-02-05 00:21:32 +01:00
|
|
|
'--{0}={1}'.format(k, v)
|
2019-12-21 01:13:19 +01:00
|
|
|
for k, v in [(k, finalzied_params[k]) for k in sorted(finalzied_params)]
|
2016-07-22 20:46:13 +02:00
|
|
|
if k not in set(['test_type', 'simple', 'duration', 'interval',
|
2020-06-13 04:24:11 +02:00
|
|
|
'random_kill_odd', 'cf_consistency', 'txn',
|
2022-03-17 03:00:04 +01:00
|
|
|
'test_best_efforts_recovery', 'enable_ts',
|
|
|
|
'test_multiops_txn', 'write_policy', 'stress_cmd'])
|
2018-05-09 22:32:03 +02:00
|
|
|
and v is not None] + unknown_params
|
2015-10-19 20:43:14 +02:00
|
|
|
return cmd
|
|
|
|
|
|
|
|
|
2020-06-13 04:24:11 +02:00
|
|
|
# Inject inconsistency to db directory.
|
|
|
|
def inject_inconsistencies_to_db_dir(dir_path):
|
|
|
|
files = os.listdir(dir_path)
|
|
|
|
file_num_rgx = re.compile(r'(?P<number>[0-9]{6})')
|
|
|
|
largest_fnum = 0
|
|
|
|
for f in files:
|
|
|
|
m = file_num_rgx.search(f)
|
|
|
|
if m and not f.startswith('LOG'):
|
|
|
|
largest_fnum = max(largest_fnum, int(m.group('number')))
|
|
|
|
|
|
|
|
candidates = [
|
|
|
|
f for f in files if re.search(r'[0-9]+\.sst', f)
|
|
|
|
]
|
|
|
|
deleted = 0
|
|
|
|
corrupted = 0
|
|
|
|
for f in candidates:
|
|
|
|
rnd = random.randint(0, 99)
|
|
|
|
f_path = os.path.join(dir_path, f)
|
|
|
|
if rnd < 10:
|
|
|
|
os.unlink(f_path)
|
|
|
|
deleted = deleted + 1
|
|
|
|
elif 10 <= rnd and rnd < 30:
|
|
|
|
with open(f_path, "a") as fd:
|
|
|
|
fd.write('12345678')
|
|
|
|
corrupted = corrupted + 1
|
|
|
|
print('Removed %d table files' % deleted)
|
|
|
|
print('Corrupted %d table files' % corrupted)
|
|
|
|
|
|
|
|
# Add corrupted MANIFEST and SST
|
|
|
|
for num in range(largest_fnum + 1, largest_fnum + 10):
|
|
|
|
rnd = random.randint(0, 1)
|
|
|
|
fname = ("MANIFEST-%06d" % num) if rnd == 0 else ("%06d.sst" % num)
|
|
|
|
print('Write %s' % fname)
|
|
|
|
with open(os.path.join(dir_path, fname), "w") as fd:
|
|
|
|
fd.write("garbage")
|
|
|
|
|
2021-06-01 18:33:50 +02:00
|
|
|
def execute_cmd(cmd, timeout):
|
|
|
|
child = subprocess.Popen(cmd, stderr=subprocess.PIPE,
|
|
|
|
stdout=subprocess.PIPE)
|
|
|
|
print("Running db_stress with pid=%d: %s\n\n"
|
|
|
|
% (child.pid, ' '.join(cmd)))
|
|
|
|
|
|
|
|
try:
|
|
|
|
outs, errs = child.communicate(timeout=timeout)
|
|
|
|
hit_timeout = False
|
|
|
|
print("WARNING: db_stress ended before kill: exitcode=%d\n"
|
|
|
|
% child.returncode)
|
|
|
|
except subprocess.TimeoutExpired:
|
|
|
|
hit_timeout = True
|
|
|
|
child.kill()
|
|
|
|
print("KILLED %d\n" % child.pid)
|
|
|
|
outs, errs = child.communicate()
|
|
|
|
|
|
|
|
return hit_timeout, child.returncode, outs.decode('utf-8'), errs.decode('utf-8')
|
|
|
|
|
2020-06-13 04:24:11 +02:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
# This script runs and kills db_stress multiple times. It checks consistency
|
|
|
|
# in case of unsafe crashes in RocksDB.
|
2018-05-09 22:32:03 +02:00
|
|
|
def blackbox_crash_main(args, unknown_args):
|
2015-10-19 20:43:14 +02:00
|
|
|
cmd_params = gen_cmd_params(args)
|
2015-10-20 20:31:27 +02:00
|
|
|
dbname = get_dbname('blackbox')
|
2015-10-19 20:43:14 +02:00
|
|
|
exit_time = time.time() + cmd_params['duration']
|
|
|
|
|
|
|
|
print("Running blackbox-crash-test with \n"
|
|
|
|
+ "interval_between_crash=" + str(cmd_params['interval']) + "\n"
|
2018-05-09 22:32:03 +02:00
|
|
|
+ "total-duration=" + str(cmd_params['duration']) + "\n")
|
2014-03-20 19:11:08 +01:00
|
|
|
|
2013-03-13 07:20:14 +01:00
|
|
|
while time.time() < exit_time:
|
2018-04-30 21:23:45 +02:00
|
|
|
cmd = gen_cmd(dict(
|
2020-03-25 04:57:53 +01:00
|
|
|
list(cmd_params.items())
|
|
|
|
+ list({'db': dbname}.items())), unknown_args)
|
2013-08-21 02:37:49 +02:00
|
|
|
|
2021-06-01 18:33:50 +02:00
|
|
|
hit_timeout, retcode, outs, errs = execute_cmd(cmd, cmd_params['interval'])
|
|
|
|
|
|
|
|
if not hit_timeout:
|
2021-06-28 08:53:47 +02:00
|
|
|
print('Exit Before Killing')
|
2021-06-01 18:33:50 +02:00
|
|
|
print('stdout:')
|
|
|
|
print(outs)
|
|
|
|
print('stderr:')
|
|
|
|
print(errs)
|
|
|
|
sys.exit(2)
|
|
|
|
|
|
|
|
for line in errs.split('\n'):
|
|
|
|
if line != '' and not line.startswith('WARNING'):
|
2017-04-12 01:53:57 +02:00
|
|
|
print('stderr has error message:')
|
|
|
|
print('***' + line + '***')
|
2013-03-13 07:20:14 +01:00
|
|
|
|
2013-08-21 02:37:49 +02:00
|
|
|
time.sleep(1) # time to stabilize before the next run
|
2013-03-13 07:20:14 +01:00
|
|
|
|
2020-06-13 04:24:11 +02:00
|
|
|
if args.test_best_efforts_recovery:
|
|
|
|
inject_inconsistencies_to_db_dir(dbname)
|
|
|
|
|
|
|
|
time.sleep(1) # time to stabilize before the next run
|
|
|
|
|
2014-03-20 19:11:08 +01:00
|
|
|
# we need to clean up after ourselves -- only do this on test success
|
2015-10-20 20:31:27 +02:00
|
|
|
shutil.rmtree(dbname, True)
|
2014-03-20 19:11:08 +01:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
# This python script runs db_stress multiple times. Some runs with
|
|
|
|
# kill_random_test that causes rocksdb to crash at various points in code.
|
2018-05-09 22:32:03 +02:00
|
|
|
def whitebox_crash_main(args, unknown_args):
|
2015-10-19 20:43:14 +02:00
|
|
|
cmd_params = gen_cmd_params(args)
|
2015-10-20 20:31:27 +02:00
|
|
|
dbname = get_dbname('whitebox')
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
cur_time = time.time()
|
|
|
|
exit_time = cur_time + cmd_params['duration']
|
2020-03-25 04:57:53 +01:00
|
|
|
half_time = cur_time + cmd_params['duration'] // 2
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
print("Running whitebox-crash-test with \n"
|
2018-05-09 22:32:03 +02:00
|
|
|
+ "total-duration=" + str(cmd_params['duration']) + "\n")
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
total_check_mode = 4
|
|
|
|
check_mode = 0
|
2016-07-22 20:46:13 +02:00
|
|
|
kill_random_test = cmd_params['random_kill_odd']
|
2015-10-19 20:43:14 +02:00
|
|
|
kill_mode = 0
|
|
|
|
|
|
|
|
while time.time() < exit_time:
|
|
|
|
if check_mode == 0:
|
|
|
|
additional_opts = {
|
|
|
|
# use large ops per thread since we will kill it anyway
|
|
|
|
"ops_per_thread": 100 * cmd_params['ops_per_thread'],
|
|
|
|
}
|
2015-10-27 00:02:32 +01:00
|
|
|
# run with kill_random_test, with three modes.
|
|
|
|
# Mode 0 covers all kill points. Mode 1 covers less kill points but
|
|
|
|
# increases change of triggering them. Mode 2 covers even less
|
|
|
|
# frequent kill points and further increases triggering change.
|
2015-10-19 20:43:14 +02:00
|
|
|
if kill_mode == 0:
|
|
|
|
additional_opts.update({
|
|
|
|
"kill_random_test": kill_random_test,
|
|
|
|
})
|
|
|
|
elif kill_mode == 1:
|
2019-08-19 19:50:25 +02:00
|
|
|
if cmd_params.get('disable_wal', 0) == 1:
|
2020-03-25 04:57:53 +01:00
|
|
|
my_kill_odd = kill_random_test // 50 + 1
|
2019-08-19 19:50:25 +02:00
|
|
|
else:
|
2020-03-25 04:57:53 +01:00
|
|
|
my_kill_odd = kill_random_test // 10 + 1
|
2015-10-19 20:43:14 +02:00
|
|
|
additional_opts.update({
|
2019-08-19 19:50:25 +02:00
|
|
|
"kill_random_test": my_kill_odd,
|
2020-06-20 00:26:05 +02:00
|
|
|
"kill_exclude_prefixes": "WritableFileWriter::Append,"
|
2015-10-19 20:43:14 +02:00
|
|
|
+ "WritableFileWriter::WriteBuffered",
|
|
|
|
})
|
2015-10-27 00:02:32 +01:00
|
|
|
elif kill_mode == 2:
|
2016-07-22 20:46:13 +02:00
|
|
|
# TODO: May need to adjust random odds if kill_random_test
|
|
|
|
# is too small.
|
2015-10-27 00:02:32 +01:00
|
|
|
additional_opts.update({
|
2020-03-25 04:57:53 +01:00
|
|
|
"kill_random_test": (kill_random_test // 5000 + 1),
|
2020-06-20 00:26:05 +02:00
|
|
|
"kill_exclude_prefixes": "WritableFileWriter::Append,"
|
2015-10-27 00:02:32 +01:00
|
|
|
"WritableFileWriter::WriteBuffered,"
|
|
|
|
"PosixMmapFile::Allocate,WritableFileWriter::Flush",
|
|
|
|
})
|
|
|
|
# Run kill mode 0, 1 and 2 by turn.
|
|
|
|
kill_mode = (kill_mode + 1) % 3
|
2015-10-19 20:43:14 +02:00
|
|
|
elif check_mode == 1:
|
|
|
|
# normal run with universal compaction mode
|
|
|
|
additional_opts = {
|
|
|
|
"kill_random_test": None,
|
|
|
|
"ops_per_thread": cmd_params['ops_per_thread'],
|
|
|
|
"compaction_style": 1,
|
|
|
|
}
|
2020-05-07 03:06:04 +02:00
|
|
|
# Single level universal has a lot of special logic. Ensure we cover
|
|
|
|
# it sometimes.
|
|
|
|
if random.randint(0, 1) == 1:
|
|
|
|
additional_opts.update({
|
|
|
|
"num_levels": 1,
|
|
|
|
})
|
2015-10-19 20:43:14 +02:00
|
|
|
elif check_mode == 2:
|
|
|
|
# normal run with FIFO compaction mode
|
|
|
|
# ops_per_thread is divided by 5 because FIFO compaction
|
|
|
|
# style is quite a bit slower on reads with lot of files
|
|
|
|
additional_opts = {
|
|
|
|
"kill_random_test": None,
|
2020-03-25 04:57:53 +01:00
|
|
|
"ops_per_thread": cmd_params['ops_per_thread'] // 5,
|
2015-10-19 20:43:14 +02:00
|
|
|
"compaction_style": 2,
|
|
|
|
}
|
|
|
|
else:
|
|
|
|
# normal run
|
2018-06-26 21:35:26 +02:00
|
|
|
additional_opts = {
|
2015-10-19 20:43:14 +02:00
|
|
|
"kill_random_test": None,
|
|
|
|
"ops_per_thread": cmd_params['ops_per_thread'],
|
|
|
|
}
|
|
|
|
|
2020-03-25 04:57:53 +01:00
|
|
|
cmd = gen_cmd(dict(list(cmd_params.items())
|
|
|
|
+ list(additional_opts.items())
|
|
|
|
+ list({'db': dbname}.items())), unknown_args)
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2020-03-25 04:57:53 +01:00
|
|
|
print("Running:" + ' '.join(cmd) + "\n") # noqa: E999 T25377293 Grandfathered in
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2021-06-01 18:33:50 +02:00
|
|
|
# If the running time is 15 minutes over the run time, explicit kill and
|
|
|
|
# exit even if white box kill didn't hit. This is to guarantee run time
|
|
|
|
# limit, as if it runs as a job, running too long will create problems
|
|
|
|
# for job scheduling or execution.
|
|
|
|
# TODO detect a hanging condition. The job might run too long as RocksDB
|
|
|
|
# hits a hanging bug.
|
|
|
|
hit_timeout, retncode, stdoutdata, stderrdata = execute_cmd(
|
|
|
|
cmd, exit_time - time.time() + 900)
|
2015-10-19 20:43:14 +02:00
|
|
|
msg = ("check_mode={0}, kill option={1}, exitcode={2}\n".format(
|
|
|
|
check_mode, additional_opts['kill_random_test'], retncode))
|
2021-05-06 02:53:16 +02:00
|
|
|
|
2020-03-25 04:57:53 +01:00
|
|
|
print(msg)
|
|
|
|
print(stdoutdata)
|
2021-05-06 02:53:16 +02:00
|
|
|
print(stderrdata)
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2021-06-01 18:33:50 +02:00
|
|
|
if hit_timeout:
|
|
|
|
print("Killing the run for running too long")
|
|
|
|
break
|
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
expected = False
|
|
|
|
if additional_opts['kill_random_test'] is None and (retncode == 0):
|
|
|
|
# we expect zero retncode if no kill option
|
|
|
|
expected = True
|
2019-05-04 02:26:20 +02:00
|
|
|
elif additional_opts['kill_random_test'] is not None and retncode <= 0:
|
|
|
|
# When kill option is given, the test MIGHT kill itself.
|
|
|
|
# If it does, negative retncode is expected. Otherwise 0.
|
2015-10-19 20:43:14 +02:00
|
|
|
expected = True
|
|
|
|
|
|
|
|
if not expected:
|
2020-03-25 04:57:53 +01:00
|
|
|
print("TEST FAILED. See kill option and exit code above!!!\n")
|
2015-10-19 20:43:14 +02:00
|
|
|
sys.exit(1)
|
|
|
|
|
2021-05-06 02:53:16 +02:00
|
|
|
stderrdata = stderrdata.lower()
|
|
|
|
errorcount = (stderrdata.count('error') -
|
|
|
|
stderrdata.count('got errors 0 times'))
|
|
|
|
print("#times error occurred in output is " + str(errorcount) +
|
|
|
|
"\n")
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
if (errorcount > 0):
|
2020-03-25 04:57:53 +01:00
|
|
|
print("TEST FAILED. Output has 'error'!!!\n")
|
2015-10-19 20:43:14 +02:00
|
|
|
sys.exit(2)
|
2021-05-06 02:53:16 +02:00
|
|
|
if (stderrdata.find('fail') >= 0):
|
2020-03-25 04:57:53 +01:00
|
|
|
print("TEST FAILED. Output has 'fail'!!!\n")
|
2015-10-19 20:43:14 +02:00
|
|
|
sys.exit(2)
|
|
|
|
|
|
|
|
# First half of the duration, keep doing kill test. For the next half,
|
|
|
|
# try different modes.
|
|
|
|
if time.time() > half_time:
|
|
|
|
# we need to clean up after ourselves -- only do this on test
|
|
|
|
# success
|
2015-10-20 20:31:27 +02:00
|
|
|
shutil.rmtree(dbname, True)
|
2018-06-04 06:28:41 +02:00
|
|
|
os.mkdir(dbname)
|
2021-09-28 23:12:23 +02:00
|
|
|
cmd_params.pop('expected_values_dir', None)
|
2015-10-19 20:43:14 +02:00
|
|
|
check_mode = (check_mode + 1) % total_check_mode
|
|
|
|
|
|
|
|
time.sleep(1) # time to stabilize after a kill
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
2021-06-28 08:53:47 +02:00
|
|
|
global stress_cmd
|
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
parser = argparse.ArgumentParser(description="This script runs and kills \
|
|
|
|
db_stress multiple times")
|
|
|
|
parser.add_argument("test_type", choices=["blackbox", "whitebox"])
|
|
|
|
parser.add_argument("--simple", action="store_true")
|
2019-08-23 01:30:30 +02:00
|
|
|
parser.add_argument("--cf_consistency", action='store_true')
|
2019-12-12 00:59:45 +01:00
|
|
|
parser.add_argument("--txn", action='store_true')
|
2020-06-13 04:24:11 +02:00
|
|
|
parser.add_argument("--test_best_efforts_recovery", action='store_true')
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
parser.add_argument("--enable_ts", action='store_true')
|
2022-03-17 03:00:04 +01:00
|
|
|
parser.add_argument("--test_multiops_txn", action='store_true')
|
|
|
|
parser.add_argument("--write_policy", choices=["write_committed", "write_prepared"])
|
2021-06-28 08:53:47 +02:00
|
|
|
parser.add_argument("--stress_cmd")
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2020-03-25 04:57:53 +01:00
|
|
|
all_params = dict(list(default_params.items())
|
|
|
|
+ list(blackbox_default_params.items())
|
|
|
|
+ list(whitebox_default_params.items())
|
|
|
|
+ list(simple_default_params.items())
|
|
|
|
+ list(blackbox_simple_default_params.items())
|
2021-02-02 20:39:20 +01:00
|
|
|
+ list(whitebox_simple_default_params.items())
|
Add user-defined timestamps to db_stress (#8061)
Summary:
Add some basic test for user-defined timestamp to db_stress. Currently,
read with timestamp always tries to read using the current timestamp.
Due to the per-key timestamp-sequence ordering constraint, we only add timestamp-
related tests to the `NonBatchedOpsStressTest` since this test serializes accesses
to the same key and uses a file to cross-check data correctness.
The timestamp feature is not supported in a number of components, e.g. Merge, SingleDelete,
DeleteRange, CompactionFilter, Readonly instance, secondary instance, SST file ingestion, transaction,
etc. Therefore, db_stress should exit if user enables both timestamp and these features at the same
time. The (currently) incompatible features can be found in
`CheckAndSetOptionsForUserTimestamp`.
This PR also fixes a bug triggered when timestamp is enabled together with
`index_type=kBinarySearchWithFirstKey`. This bug fix will also be in another separate PR
with more unit tests coverage. Fixing it here because I do not want to exclude the index type
from crash test.
Pull Request resolved: https://github.com/facebook/rocksdb/pull/8061
Test Plan: make crash_test_with_ts
Reviewed By: jay-zhuang
Differential Revision: D27056282
Pulled By: riversand963
fbshipit-source-id: c3e00ad1023fdb9ebbdf9601ec18270c5e2925a9
2021-03-23 13:12:04 +01:00
|
|
|
+ list(blob_params.items())
|
2022-03-17 03:00:04 +01:00
|
|
|
+ list(ts_params.items())
|
|
|
|
+ list(multiops_txn_default_params.items())
|
|
|
|
+ list(multiops_wc_txn_params.items())
|
|
|
|
+ list(multiops_wp_txn_params.items()))
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
for k, v in all_params.items():
|
|
|
|
parser.add_argument("--" + k, type=type(v() if callable(v) else v))
|
2018-05-09 22:32:03 +02:00
|
|
|
# unknown_args are passed directly to db_stress
|
|
|
|
args, unknown_args = parser.parse_known_args()
|
2015-10-19 20:43:14 +02:00
|
|
|
|
2018-06-04 06:28:41 +02:00
|
|
|
test_tmpdir = os.environ.get(_TEST_DIR_ENV_VAR)
|
|
|
|
if test_tmpdir is not None and not os.path.isdir(test_tmpdir):
|
|
|
|
print('%s env var is set to a non-existent directory: %s' %
|
|
|
|
(_TEST_DIR_ENV_VAR, test_tmpdir))
|
|
|
|
sys.exit(1)
|
|
|
|
|
2021-06-28 08:53:47 +02:00
|
|
|
if args.stress_cmd:
|
|
|
|
stress_cmd = args.stress_cmd
|
2015-10-19 20:43:14 +02:00
|
|
|
if args.test_type == 'blackbox':
|
2018-05-09 22:32:03 +02:00
|
|
|
blackbox_crash_main(args, unknown_args)
|
2015-10-19 20:43:14 +02:00
|
|
|
if args.test_type == 'whitebox':
|
2018-05-09 22:32:03 +02:00
|
|
|
whitebox_crash_main(args, unknown_args)
|
2021-09-28 23:12:23 +02:00
|
|
|
# Only delete the `expected_values_dir` if test passes
|
|
|
|
if expected_values_dir is not None:
|
|
|
|
shutil.rmtree(expected_values_dir)
|
2022-03-17 03:00:04 +01:00
|
|
|
if multiops_txn_key_spaces_file is not None:
|
|
|
|
os.remove(multiops_txn_key_spaces_file)
|
2020-10-12 23:08:35 +02:00
|
|
|
|
2015-10-19 20:43:14 +02:00
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
main()
|