mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 20:43:57 +00:00
4b66d95344
Summary: The goal of this diff is to create a simple stress test with focus on catching: * bugs in compaction/flush processes, especially the ones that cause assertion errors * bugs in the code that deletes obsolete files There are two parts of the test: * write_stress, a binary that writes to the database * write_stress_runner.py, a script that invokes and kills write_stress Here are some interesting parts of write_stress: * Runs with very high concurrency of compactions and flushes (32 threads total) and tries to create a huge amount of small files * The keys written to the database are not uniformly distributed -- there is a 3-character prefix that mutates occasionally (in prefix mutator thread), in such a way that the first character mutates slower than second, which mutates slower than third character. That way, the compaction stress tests some interesting compaction features like trivial moves and bottommost level calculation * There is a thread that creates an iterator, holds it for couple of seconds and then iterates over all keys. This is supposed to test RocksDB's abilities to keep the files alive when there are references to them. * Some writes trigger WAL sync. This is stress testing our WAL sync code. * At the end of the run, we make sure that we didn't leak any of the sst files write_stress_runner.py changes the mode in which we run write_stress and also kills and restarts it. There are some interesting characteristics: * At the beginning we divide the full test runtime into smaller parts -- shorter runtimes (couple of seconds) and longer runtimes (100, 1000) seconds * The first time we run write_stress, we destroy the old DB. Every next time during the test, we use the same DB. * We can run in kill mode or clean-restart mode. Kill mode kills the write_stress violently. * We can run in mode where delete_obsolete_files_with_fullscan is true or false * We can run with low_open_files mode turned on or off. When it's turned on, we configure table cache to only hold a couple of files -- that way we need to reopen files every time we access them. Another goal was to create a stress test without a lot of parameters. So tools/write_stress_runner.py should only take one parameter -- runtime_sec and it should figure out everything else on its own. In a separate diff, I'll add this new test to our nightly legocastle runs. Test Plan: The goal of this test was to retroactively catch the following bugs: D33045, D48201, D46899, D42399. I failed to reproduce D48201, but all others have been caught! When i reverted https://reviews.facebook.net/D33045: ./write_stress --runtime_sec=200 --low_open_files_mode=true Iterator statuts not OK: IO error: /fast-rocksdb-tmp/rocksdb_test/write_stress/089166.sst: No such file or directory When i reverted https://reviews.facebook.net/D42399: python tools/write_stress_runner.py --runtime_sec=5000 Running write_stress, will kill after 5 seconds: ./write_stress --runtime_sec=-1 Running write_stress, will kill after 2 seconds: ./write_stress --runtime_sec=-1 --destroy_db=false --delete_obsolete_files_with_fullscan=true Running write_stress, will kill after 7 seconds: ./write_stress --runtime_sec=-1 --destroy_db=false Running write_stress, will kill after 5 seconds: ./write_stress --runtime_sec=-1 --destroy_db=false Running write_stress, will kill after 8 seconds: ./write_stress --runtime_sec=-1 --destroy_db=false --low_open_files_mode=true Write to DB failed: IO error: /fast-rocksdb-tmp/rocksdb_test/write_stress/019250.sst: No such file or directory ERROR: write_stress died with exitcode=-6 When i reverted https://reviews.facebook.net/D46899: python tools/write_stress_runner.py --runtime_sec=1000 runtime: 1000 Going to execute write stress for [3, 3, 100, 3, 2, 100, 1, 788] Running write_stress for 3 seconds: ./write_stress --runtime_sec=3 --low_open_files_mode=true Running write_stress for 3 seconds: ./write_stress --runtime_sec=3 --destroy_db=false --delete_obsolete_files_with_fullscan=true Running write_stress, will kill after 100 seconds: ./write_stress --runtime_sec=-1 --destroy_db=false --delete_obsolete_files_with_fullscan=true write_stress: db/db_impl.cc:2070: void rocksdb::DBImpl::MarkLogsSynced(uint64_t, bool, const rocksdb::Status&): Assertion `log.getting_synced' failed. ERROR: write_stress died with exitcode=-6 Reviewers: IslamAbdelRahman, yhchiang, rven, kradhakrishnan, sdong, anthony Reviewed By: anthony Subscribers: leveldb, dhruba Differential Revision: https://reviews.facebook.net/D49533
74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
#! /usr/bin/env python
|
|
import subprocess
|
|
import argparse
|
|
import random
|
|
import time
|
|
import sys
|
|
|
|
|
|
def generate_runtimes(total_runtime):
|
|
# combination of short runtimes and long runtimes, with heavier
|
|
# weight on short runtimes
|
|
possible_runtimes_sec = range(1, 10) + range(1, 20) + [100, 1000]
|
|
runtimes = []
|
|
while total_runtime > 0:
|
|
chosen = random.choice(possible_runtimes_sec)
|
|
chosen = min(chosen, total_runtime)
|
|
runtimes.append(chosen)
|
|
total_runtime -= chosen
|
|
return runtimes
|
|
|
|
|
|
def main(args):
|
|
runtimes = generate_runtimes(int(args.runtime_sec))
|
|
print "Going to execute write stress for " + str(runtimes)
|
|
first_time = True
|
|
|
|
for runtime in runtimes:
|
|
kill = random.choice([False, True])
|
|
|
|
cmd = './write_stress --runtime_sec=' + \
|
|
("-1" if kill else str(runtime))
|
|
|
|
if len(args.db) > 0:
|
|
cmd = cmd + ' --db=' + args.db
|
|
|
|
if first_time:
|
|
first_time = False
|
|
else:
|
|
# use current db
|
|
cmd = cmd + ' --destroy_db=false'
|
|
if random.choice([False, True]):
|
|
cmd = cmd + ' --delete_obsolete_files_with_fullscan=true'
|
|
if random.choice([False, True]):
|
|
cmd = cmd + ' --low_open_files_mode=true'
|
|
|
|
print("Running write_stress for %d seconds (%s): %s" %
|
|
(runtime, ("kill-mode" if kill else "clean-shutdown-mode"),
|
|
cmd))
|
|
|
|
child = subprocess.Popen([cmd], shell=True)
|
|
killtime = time.time() + runtime
|
|
while not kill or time.time() < killtime:
|
|
time.sleep(1)
|
|
if child.poll() is not None:
|
|
if child.returncode == 0:
|
|
break
|
|
else:
|
|
print("ERROR: write_stress died with exitcode=%d\n"
|
|
% child.returncode)
|
|
sys.exit(1)
|
|
if kill:
|
|
child.kill()
|
|
# breathe
|
|
time.sleep(3)
|
|
|
|
if __name__ == '__main__':
|
|
random.seed(time.time())
|
|
parser = argparse.ArgumentParser(description="This script runs and kills \
|
|
write_stress multiple times")
|
|
parser.add_argument("--runtime_sec", default='1000')
|
|
parser.add_argument("--db", default='')
|
|
args = parser.parse_args()
|
|
main(args)
|