1. 31 5月, 2023 3 次提交
    • P
      Better management of unreleased HISTORY (#11481) · 8848ec92
      Peter Dillinger 提交于
      Summary:
      See new NOTE in HISTORY.md and unreleased_history/README.txt
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11481
      
      Test Plan: some manual testing on my CentOS 8 system
      
      Reviewed By: jaykorean
      
      Differential Revision: D46233342
      
      Pulled By: pdillinger
      
      fbshipit-source-id: daf59cf3dc907f450b469090dcc481a30a7d7c0d
      8848ec92
    • C
      Fix flaky test: `DBCompactionTest.WaitForCompactShutdownWhileWaiting` (#11488) · e1c7209b
      Changyu Bi 提交于
      Summary:
      tsan complains with the following error message. This is likely due to DB object destroyed while WaitForCompact() is still running.
      ```
      [ RUN      ] DBCompactionTest.WaitForCompactShutdownWhileWaiting
      ==================
      WARNING: ThreadSanitizer: data race (pid=1128703)
        Atomic read of size 1 at 0x7b8c00000740 by thread T4:
          #0 pthread_cond_wait <null> (db_compaction_test+0x46970a)
          https://github.com/facebook/rocksdb/issues/1 rocksdb::port::CondVar::Wait() /root/project/port/port_posix.cc:119:23 (librocksdb.so.8.4+0x7c4c60)
          https://github.com/facebook/rocksdb/issues/2 rocksdb::InstrumentedCondVar::WaitInternal() /root/project/monitoring/instrumented_mutex.cc:69:9 (librocksdb.so.8.4+0x75f697)
          https://github.com/facebook/rocksdb/issues/3 rocksdb::InstrumentedCondVar::Wait() /root/project/monitoring/instrumented_mutex.cc:62:3 (librocksdb.so.8.4+0x75f697)
          https://github.com/facebook/rocksdb/issues/4 rocksdb::DBImpl::WaitForCompact(rocksdb::WaitForCompactOptions const&) /root/project/db/db_impl/db_impl_compaction_flush.cc:3978:14 (librocksdb.so.8.4+0x494174)
          https://github.com/facebook/rocksdb/issues/5 rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30::operator()() const /root/project/db/db_compaction_test.cc:3479:26 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/6 void std::__invoke_impl<void, rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30>(std::__invoke_other, rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30&&) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/invoke.h:61:14 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/7 std::__invoke_result<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30>::type std::__invoke<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30>(rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30&&) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/invoke.h:96:14 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/8 void std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30> >::_M_invoke<0ul>(std::_Index_tuple<0ul>) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:253:13 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/9 std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30> >::operator()() /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:260:11 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/10 std::thread::_State_impl<std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_30> > >::_M_run() /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:211:13 (db_compaction_test+0x5cdc90)
          https://github.com/facebook/rocksdb/issues/11 <null> <null> (libstdc++.so.6+0xda6b3)
      
        Previous write of size 1 at 0x7b8c00000740 by thread T5:
          #0 pthread_mutex_destroy <null> (db_compaction_test+0x46a4f8)
          https://github.com/facebook/rocksdb/issues/1 rocksdb::port::Mutex::~Mutex() /root/project/port/port_posix.cc:77:48 (librocksdb.so.8.4+0x7c480e)
          https://github.com/facebook/rocksdb/issues/2 rocksdb::InstrumentedMutex::~InstrumentedMutex() /root/project/./monitoring/instrumented_mutex.h:20:7 (librocksdb.so.8.4+0x41fda6)
          https://github.com/facebook/rocksdb/issues/3 rocksdb::DBImpl::~DBImpl() /root/project/db/db_impl/db_impl.cc:755:1 (librocksdb.so.8.4+0x41fda6)
          https://github.com/facebook/rocksdb/issues/4 rocksdb::DBImpl::~DBImpl() /root/project/db/db_impl/db_impl.cc:737:19 (librocksdb.so.8.4+0x4203d9)
          https://github.com/facebook/rocksdb/issues/5 rocksdb::DBTestBase::Close() /root/project/db/db_test_util.cc:670:3 (librocksdb_test_debug.so+0x57413)
          https://github.com/facebook/rocksdb/issues/6 rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31::operator()() const /root/project/db/db_compaction_test.cc:3485:49 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/7 void std::__invoke_impl<void, rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31>(std::__invoke_other, rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31&&) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/invoke.h:61:14 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/8 std::__invoke_result<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31>::type std::__invoke<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31>(rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31&&) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/invoke.h:96:14 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/9 void std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31> >::_M_invoke<0ul>(std::_Index_tuple<0ul>) /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:253:13 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/10 std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31> >::operator()() /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:260:11 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/11 std::thread::_State_impl<std::thread::_Invoker<std::tuple<rocksdb::DBCompactionTest_WaitForCompactShutdownWhileWaiting_Test::TestBody()::$_31> > >::_M_run() /usr/bin/../lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/std_thread.h:211:13 (db_compaction_test+0x5cdf03)
          https://github.com/facebook/rocksdb/issues/12 <null> <null> (libstdc++.so.6+0xda6b3)
      ```
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11488
      
      Test Plan:
      ```
      COMPILE_WITH_TSAN=1 CC=clang-13 CXX=clang++-13 ROCKSDB_DISABLE_ALIGNED_NEW=1 USE_CLANG=1 make V=1 -j32 db_compaction_test
      
      gtest-parallel --repeat=10000 ./db_compaction_test --gtest_filter="*WaitForCompactShutdownWhileWaiting*" -w200
      ```
      
      Reviewed By: jaykorean
      
      Differential Revision: D46293891
      
      Pulled By: cbi42
      
      fbshipit-source-id: 8ca259cb1e09a9e4f4095b2d084f2ba92b710b97
      e1c7209b
    • A
      Integrate CacheReservationManager with compressed secondary cache (#11449) · fcc358ba
      anand76 提交于
      Summary:
      This draft PR implements charging of reserved memory, for write buffers, table readers, and other purposes, proportionally to the block cache and the compressed secondary cache. The basic flow of memory reservation is maintained - clients use ```CacheReservationManager``` to request reservations, and ```CacheReservationManager``` inserts placeholder entries, i.e null value and non-zero charge, into the block cache. The ```CacheWithSecondaryAdapter``` wrapper uses its own instance of ```CacheReservationManager``` to keep track of reservations charged to the secondary cache, while the placeholder entries are inserted into the primary block cache. The design is as follows.
      
      When ```CacheWithSecondaryAdapter``` is constructed with the ```distribute_cache_res``` parameter set to true, it manages the entire memory budget across the primary and secondary cache. The secondary cache is assumed to be in memory, such as the ```CompressedSecondaryCache```. When a placeholder entry is inserted by a CacheReservationManager instance to reserve memory, the ```CacheWithSecondaryAdapter```ensures that the reservation is distributed proportionally across the primary/secondary caches.
      
      The primary block cache is initially sized to the sum of the primary cache budget + the secondary cache budget, as follows -
        |---------    Primary Cache Configured Capacity  -----------|
        |---Secondary Cache Budget----|----Primary Cache Budget-----|
      
      A ```ConcurrentCacheReservationManager``` member in the ```CacheWithSecondaryAdapter```, ```pri_cache_res_```, is used to help with tracking the distribution of memory reservations. Initially, it accounts for the entire secondary cache budget as a reservation against the primary cache. This shrinks the usable capacity of the primary cache to the budget that the user originally desired.
      
        |--Reservation for Sec Cache--|-Pri Cache Usable Capacity---|
      
      When a reservation placeholder is inserted into the adapter, it is inserted directly into the primary cache. This means the entire charge of the placeholder is counted against the primary cache. To compensate and count a portion of it against the secondary cache, the secondary cache ```Deflate()``` method is called to shrink it. Since the ```Deflate()``` causes the secondary actual usage to shrink, it is reflected here by releasing an equal amount from the ```pri_cache_res_``` reservation.
      
      For example, if the pri/sec ratio is 50/50, this would be the state after placeholder insertion -
      
        |-Reservation for Sec Cache-|-Pri Cache Usable Capacity-|-R-|
      
      Likewise, when the user inserted placeholder is released, the secondary cache ```Inflate()``` method is called to grow it, and the ```pri_cache_res_``` reservation is increased by an equal amount.
      
      Other alternatives -
      1. Another way of implementing this would have been to simply split the user reservation in ```CacheWithSecondaryAdapter``` into primary and secondary components. However, this would require allocating a structure to track the associated secondary cache reservation, which adds some complexity and overhead.
      2. Yet another option is to implement the splitting directly in ```CacheReservationManager```. However, there are multiple instances of ```CacheReservationManager``` in a DB instance, making it complicated to keep track of them.
      
      The PR contains the following changes -
      1. A new cache allocator, ```NewTieredVolatileCache()```, is defined for allocating a tiered primary block cache and compressed secondary cache. This internally allocates an instance of ```CacheWithSecondaryAdapter```.
      3. New interfaces, ```Deflate()``` and ```Inflate()```, are added to the ```SecondaryCache``` interface. The default implementaion returns ```NotSupported``` with overrides in ```CompressedSecondaryCache```.
      4. The ```CompressedSecondaryCache``` uses a ```ConcurrentCacheReservationManager``` instance to manage reservations done using ```Inflate()/Deflate()```.
      5. The ```CacheWithSecondaryAdapter``` optionally distributes memory reservations across the primary and secondary caches. The primary cache is sized to the total memory budget (primary + secondary), and the capacity allocated to secondary cache is "reserved" against the primary cache. For any subsequent reservations, the primary cache pre-reserved capacity is adjusted.
      
      Benchmarks -
      Baseline
      ```
      time ~/rocksdb_anand76/db_bench --db=/dev/shm/comp_cache_res/base --use_existing_db=true --benchmarks="readseq,readwhilewriting" --key_size=32 --value_size=1024 --num=20000000 --threads=32 --bloom_bits=10 --cache_size=30000000000 --use_compressed_secondary_cache=true --compressed_secondary_cache_size=5000000000 --duration=300 --cost_write_buffer_to_cache=true
      ```
      ```
      readseq      :       3.301 micros/op 9694317 ops/sec 66.018 seconds 640000000 operations; 9763.0 MB/s
      readwhilewriting :      22.921 micros/op 1396058 ops/sec 300.021 seconds 418846968 operations; 1405.9 MB/s (13068999 of 13068999 found)
      
      real    6m31.052s
      user    152m5.660s
      sys     26m18.738s
      ```
      With TieredVolatileCache
      ```
      time ~/rocksdb_anand76/db_bench --db=/dev/shm/comp_cache_res/base --use_existing_db=true --benchmarks="readseq,readwhilewriting" --key_size=32 --value_size=1024 --num=20000000 --threads=32 --bloom_bits=10 --cache_size=30000000000 --use_compressed_secondary_cache=true --compressed_secondary_cache_size=5000000000 --duration=300 --cost_write_buffer_to_cache=true --use_tiered_volatile_cache=true
      ```
      ```
      readseq      :       4.064 micros/op 7873915 ops/sec 81.281 seconds 640000000 operations; 7929.7 MB/s
      readwhilewriting :      20.944 micros/op 1527827 ops/sec 300.020 seconds 458378968 operations; 1538.6 MB/s (14296999 of 14296999 found)
      
      real    6m42.743s
      user    157m58.972s
      sys     33m16.671
      ```
      ```
      readseq      :       3.484 micros/op 9184967 ops/sec 69.679 seconds 640000000 operations; 9250.0 MB/s
      readwhilewriting :      21.261 micros/op 1505035 ops/sec 300.024 seconds 451545968 operations; 1515.7 MB/s (14101999 of 14101999 found)
      
      real    6m31.469s
      user    155m16.570s
      sys     27m47.834s
      ```
      
      ToDo -
      1. Add to db_stress
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11449
      
      Reviewed By: pdillinger
      
      Differential Revision: D46197388
      
      Pulled By: anand1976
      
      fbshipit-source-id: 42d16f0254df683db4929db20d06ff26030e90df
      fcc358ba
  2. 27 5月, 2023 2 次提交
    • A
      add WriteBatch::Release() (#11482) · 3e7fc881
      Andrew Kryczka 提交于
      Summary:
      Together with the existing constructor,
      `explicit WriteBatch(std::string&& rep)`, this enables transferring `WriteBatch` via its `std::string` representation. Associated info like KV checksums are dropped but the caller can use `WriteBatch::VerifyChecksum()` before taking ownership if needed.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11482
      
      Reviewed By: cbi42
      
      Differential Revision: D46233884
      
      Pulled By: ajkr
      
      fbshipit-source-id: 6bc64a6e75fb7bbf61d08c09520fc3705a7b44d8
      3e7fc881
    • S
      Tweak on IsTrivialMove() (#11467) · de1dd4ca
      Soli 提交于
      Summary:
      `output_level_` and `number_levels_` are not changing in iteration of `inputs_` files.
      
      Moving the check out of `for` loop could slightly improve performance.
      
      It is easier to review when ignore whitespace changes.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11467
      
      Reviewed By: cbi42
      
      Differential Revision: D46155962
      
      Pulled By: ajkr
      
      fbshipit-source-id: 45ec80b13152b3bed7305e6f707cb9b187d5f315
      de1dd4ca
  3. 26 5月, 2023 4 次提交
    • J
      Move WaitForCompect() change entry to Unreleased in History file (#11479) · 23f4e9ad
      Jay Huh 提交于
      Summary:
      Context:
      
      Because of the branch cut, History change made it to the previous release. Moving entry to Unreleased
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11479
      
      Test Plan: History change. Not needed.
      
      Reviewed By: pdillinger
      
      Differential Revision: D46226237
      
      Pulled By: jaykorean
      
      fbshipit-source-id: 33e7d84a05db254d227f05d76038fc6d225dbabf
      23f4e9ad
    • J
      Add WaitForCompact with WaitForCompactOptions to public API (#11436) · 81aeb159
      Jay Huh 提交于
      Summary:
      Context:
      
      This is the first PR for WaitForCompact() Implementation with WaitForCompactOptions. In this PR, we are introducing `Status WaitForCompact(const WaitForCompactOptions& wait_for_compact_options)` in the public API. This currently utilizes the existing internal `WaitForCompact()` implementation (with default abort_on_pause = false). `abort_on_pause` has been moved to `WaitForCompactOptions&`. In the later PRs, we will introduce the following two options in `WaitForCompactOptions`
      
      1. `bool flush = false` by default - If true, flush before waiting for compactions to finish. Must be set to true to ensure no immediate compactions (except perhaps periodic compactions) after closing and re-opening the DB.
      2. `bool close_db = false` by default - If true, will also close the DB upon compactions finishing.
      
      1. struct `WaitForCompactOptions` added to options.h and `abort_on_pause` in the internal API moved to the option struct.
      2. `Status WaitForCompact(const WaitForCompactOptions& wait_for_compact_options)` introduced in `db.h`
      3. Changed the internal WaitForCompact() to `WaitForCompact(const WaitForCompactOptions& wait_for_compact_options)` and checks for the `abort_on_pause` inside the option.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11436
      
      Test Plan:
      Following tests added
      - `DBCompactionTest::WaitForCompactWaitsOnCompactionToFinish`
      - `DBCompactionTest::WaitForCompactAbortOnPauseAborted`
      - `DBCompactionTest::WaitForCompactContinueAfterPauseNotAborted`
      - `DBCompactionTest::WaitForCompactShutdownWhileWaiting`
      - `TransactionTest::WaitForCompactAbortOnPause`
      
      NOTE: `TransactionTest::WaitForCompactAbortOnPause` was added to use `StackableDB` to ensure the wrapper function is in place.
      
      Reviewed By: pdillinger
      
      Differential Revision: D45799659
      
      Pulled By: jaykorean
      
      fbshipit-source-id: b5b58f95957f2ab47d1221dee32a61d6cdc4685b
      81aeb159
    • Y
      Add support to strip / pad timestamp when writing / reading a block (#11472) · d1ae7f6c
      Yu Zhang 提交于
      Summary:
      This patch adds support in `BlockBuilder` to strip user-defined timestamp from the `key` added via `Add(key, value)` and its equivalent APIs. The stripping logic is different when the key is either a user key or an internal key, so the `BlockBuilder` is created with a flag to indicate that. This patch also add support on the read path to APIs `NewIndexIterator`, `NewDataIterator` to support pad a min timestamp.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11472
      
      Test Plan:
      Three test modes are added to parameterize existing tests:
      UserDefinedTimestampTestMode::kNone -> UDT feature is not enabled
      UserDefinedTimestampTestMode::kNormal -> UDT feature enabled, write / read with min timestamp
      UserDefinedTimestampTestMode::kStripUserDefinedTimestamps -> UDT feature enabled, write / read with min timestamp, set `persist_user_defined_timestamps` where it applies to false.
      The tests read/write with min timestamp so that point read and range scan can correctly read values in all three test modes.
      
      `block_test` are parameterized to run with above three test modes and some additional parameteriazation
      
      ```
      make all check
      ./block_test --gtest_filter="P/BlockTest*"
      ./block_test --gtest_filter="P/IndexBlockTest*"
      ```
      
      Reviewed By: ltamasi
      
      Differential Revision: D46200539
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: 59f5d6b584639976b69c2943eba723bd47d9b3c0
      d1ae7f6c
    • H
      Fix StopWatch bug; Remove setting `record_read_stats` (#11474) · dcc6fc99
      Hui Xiao 提交于
      Summary:
      **Context/Summary:**
      - StopWatch enable stats even when `StatsLevel::kExceptTimers` is set. It's a harmless bug though since `reportTimeToHistogram()` will not report it anyway according to https://github.com/facebook/rocksdb/blob/main/include/rocksdb/statistics.h#L705
      -  https://github.com/facebook/rocksdb/pull/11288 should have removed logics of setting `record_read_stats = !for_compaction` as we don't differentiate `RandomAccessFileReader`'s stats behavior based on compaction or not (instead we now report stats of different IO activities including compaction to different stats). Fixing this should report more compaction related file read micros that aren't reported previously due to `for_compaction==true`
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11474
      
      Test Plan:
      - DB bench pre vs post fix with small max_open_files
      
      Setup command
      `./db_ bench  -db=/dev/shm/testdb/ -statistics=true -benchmarks=fillseq -key_size=32 -value_size=512 -num=5000 -write_buffer_size=655 -target_file_size_base=655 -disable_auto_compactions=true -compression_type=none -bloom_bits=3`
      
      Run command
      `./db_bench --open_files=1 -use_existing_db=true -db=/dev/shm/testdb2/ -statistics=true -benchmarks=compactall -key_size=32 -value_size=512 -num=5000 -write_buffer_size=655 -target_file_size_base=655 -disable_auto_compactions=true -compression_type=none -bloom_bits=3`
      
      Pre-fix
      ```
      rocksdb.sst.read.micros P50 : 2.056175 P95 : 4.647739 P99 : 8.948475 P100 : 25.000000 COUNT : 4451 SUM : 12827
      rocksdb.file.read.flush.micros P50 : 0.000000 P95 : 0.000000 P99 : 0.000000 P100 : 0.000000 COUNT : 0 SUM : 0
      rocksdb.file.read.compaction.micros P50 : 2.057397 P95 : 4.625253 P99 : 8.749474 P100 : 25.000000 COUNT : 4382 SUM : 12608
      rocksdb.file.read.db.open.micros P50 : 1.985294 P95 : 9.100000 P99 : 13.000000 P100 : 13.000000 COUNT : 69 SUM : 219
      ```
      
      Post-fix (with a higher `rocksdb.file.read.compaction.micros` count)
      ```
      rocksdb.sst.read.micros P50 : 1.858968 P95 : 3.653086 P99 : 5.968000 P100 : 21.000000 COUNT : 3548 SUM : 9119
      rocksdb.file.read.flush.micros P50 : 0.000000 P95 : 0.000000 P99 : 0.000000 P100 : 0.000000 COUNT : 0 SUM : 0
      rocksdb.file.read.compaction.micros P50 : 1.857027 P95 : 3.627614 P99 : 5.738621 P100 : 21.000000 COUNT : 3479 SUM : 8904
      rocksdb.file.read.db.open.micros P50 : 2.000000 P95 : 6.733333 P99 : 11.000000 P100 : 11.000000 COUNT : 69 SUM : 215
      ```
      - CI
      
      Reviewed By: ajkr
      
      Differential Revision: D46137221
      
      Pulled By: hx235
      
      fbshipit-source-id: e5b4ee7001af26f2ee0377bc6334f43b2a527388
      dcc6fc99
  4. 25 5月, 2023 3 次提交
    • P
      Document SyncPoint::LoadDependency (#11477) · e8710303
      Peter Dillinger 提交于
      Summary:
      It's easy to mix up the ordering when it's undocumented. For an example of the meaning of the order, see DBTest.ThreadStatusFlush.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11477
      
      Test Plan: comments only
      
      Reviewed By: jaykorean
      
      Differential Revision: D46166683
      
      Pulled By: pdillinger
      
      fbshipit-source-id: 33118ba7ef1b08eab7b077548fe2e70f2c309e3f
      e8710303
    • P
      Improve memory efficiency of many OptimisticTransactionDBs (#11439) · 17bc2774
      Peter Dillinger 提交于
      Summary:
      Currently it's easy to use a ton of memory with many small OptimisticTransactionDB instances, because each one by default allocates a million mutexes (40 bytes each on my compiler) for validating transactions. It even puts a lot of pressure on the allocator by allocating each one individually!
      
      In this change:
      * Create a new object and option that enables sharing these buckets of mutexes between instances. This is generally good for load balancing potential contention as various DBs become hotter or colder with txn writes. About the only cases where this sharing wouldn't make sense (e.g. each DB usually written by one thread) are cases that would be better off with OccValidationPolicy::kValidateSerial which doesn't use the buckets anyway.
      * Allocate the mutexes in a contiguous array, for efficiency
      * Add an option to ensure the mutexes are cache-aligned. In several other places we use cache-aligned mutexes but OptimisticTransactionDB historically does not. It should be a space-time trade-off the user can choose.
      * Provide some visibility into the memory used by the mutex buckets with an ApproximateMemoryUsage() function (also used in unit testing)
      * Share code with other users of "striped" mutexes, appropriate refactoring for customization & efficiency (e.g. using FastRange instead of modulus)
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11439
      
      Test Plan: unit tests added. Ran sized-up versions of stress test in unit test, including a before-and-after performance test showing no consistent difference. (NOTE: OptimisticTransactionDB not currently covered by db_stress!)
      
      Reviewed By: ltamasi
      
      Differential Revision: D45796393
      
      Pulled By: pdillinger
      
      fbshipit-source-id: ae2b3a26ad91ceeec15debcdc63ff48df6736a54
      17bc2774
    • A
      Implement missing compactrangeoptions from Java API (#10880) · 93e0715f
      Alan Paxton 提交于
      Summary:
      Add the following missing options to `src/main/java/org/rocksdb/CompactRangeOptions.java` and in `java/rocksjni/options.cc` in RocksJava.
      
      For the descriptions and API see the C++ file `include/rocksdb/options.h`, specifically the struct `CompactRangeOptions`
      
      * full_history_ts_low
      * canceled
      
      We changed the handle to return an object (of class `Java_org_rocksdb_CompactRangeOptions`) containing a `ROCKSDB_NAMESPACE::CompactRangeOptions` at (almost certainly) 0-offset, rather than a raw `ROCKSDB_NAMESPACE::CompactRangeOptions`.
      
      The `Java_org_rocksdb_CompactRangeOptions` contains as supplementary fields objects (std::string, std::atomic<bool>) which are passed as pointers to the `ROCKSDB_NAMESPACE::CompactRangeOptions` and which must therefore live for as long as the `ROCKSDB_NAMESPACE::CompactRangeOptions`. By placing them in a `Java_org_rocksdb_CompactRangeOptions` we achieve this.
      
      Because the field offset of the `ROCKSDB_NAMESPACE::CompactRangeOptions` member is (very probably) 0, casting the handle to ROCKSDB_NAMESPACE::CompactRangeOptions works (i.e. old methods didn’t have to be changed), but really that’s a minefield and the correct answer is to cast to the correct type (Java_org_rocksdb_CompactRangeOptions) and then use the ROCKSDB_NAMESPACE::CompactRangeOptions field in that. So the get/set methods for existing parameters have this change.
      
      Testing
      -------
      We added unit tests for getting and setting the newly implemented fields to `CompactRangeOptionsTest`
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/10880
      
      Reviewed By: ajkr
      
      Differential Revision: D41482476
      
      Pulled By: anand1976
      
      fbshipit-source-id: c70795e790436fb3544655920adf6fca62ed34e2
      93e0715f
  5. 24 5月, 2023 2 次提交
  6. 23 5月, 2023 6 次提交
    • Y
      Fix stress test failure caused by #11424 (#11470) · 68cc429b
      Yu Zhang 提交于
      Summary:
      The `ryw_expected_values` check only applies to when transaction is used.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11470
      
      Reviewed By: hx235
      
      Differential Revision: D46085614
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: 4757896c3a62975641adcf97db077a04a0f33030
      68cc429b
    • A
      Fix regression script for async_io benchmarks (#11462) · 53e0b2fe
      akankshamahajan 提交于
      Summary:
      Fix regression script for async_io benchmarks to report right ops/sec and latency
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11462
      
      Test Plan: Verified locally
      
      Reviewed By: anand1976
      
      Differential Revision: D46031147
      
      Pulled By: akankshamahajan15
      
      fbshipit-source-id: 33ba587e6569ab2f834381ac2538e61da6876405
      53e0b2fe
    • Y
      Add utils to use for handling user defined timestamp size record in WAL (#11451) · 11ebddb1
      Yu Zhang 提交于
      Summary:
      Add a util method `HandleWriteBatchTimestampSizeDifference` to handle a `WriteBatch` read from WAL log when user-defined timestamp size record is written and read. Two check modes are added: `kVerifyConsistency` that just verifies the recorded timestamp size are consistent with the running ones. This mode is to be used by `db_impl_secondary` for opening a DB as secondary instance. It will also be used by `db_impl_open` before the user comparator switch support is added to make a column switch between enabling/disable UDT feature. The other mode `kReconcileInconsistency` will be used by `db_impl_open` later when user comparator can be changed.
      
      Another change is to extract a method `CollectColumnFamilyIdsFromWriteBatch` in db_secondary_impl.h into its standalone util file so it can be shared.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11451
      
      Test Plan:
      ```
      make check
      ./udt_util_test
      ```
      
      Reviewed By: ltamasi
      
      Differential Revision: D45894386
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: b96790777f154cddab6d45d9ba2e5d20ebc6fe9d
      11ebddb1
    • Y
      Refactor WriteUnpreparedStressTest to be a unit test (#11424) · ffb5f1f4
      Yu Zhang 提交于
      Summary:
      This patch remove the "stress" aspect from the WriteUnpreparedStressTest and leave it to be a unit test for some correctness testing w.r.t. snapshot functionality. I added some read-your-write verification to the transaction test in db_stress.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11424
      
      Test Plan:
      `./write_unprepared_transaction_test`
      `./db_crashtest.py whitebox --txn`
      `./db_crashtest.py blackbox --txn`
      
      Reviewed By: hx235
      
      Differential Revision: D45551521
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: 20c3d510eb4255b08ddd7b6c85bdb4945436f6e8
      ffb5f1f4
    • F
      fix typo in detecting HAVE_AUXV_GETAUXVAL (#10913) · 5b945adf
      FishAndBird 提交于
      Summary:
      crc32 uses CPU heavily,  arm64 and ppc will benefited by crc32 accelerate.
      
      Only build via `cmake` affected
      
      - Arm64 Tested ok, crc32 acceralated, write 30GB data throughput promoted 30%.
      - ppc not tested.
      - x86_64 seems not affected.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/10913
      
      Reviewed By: pdillinger
      
      Differential Revision: D45959843
      
      Pulled By: ajkr
      
      fbshipit-source-id: 93c91f2702fec33cca69139a2544d7c5ebeac4c6
      5b945adf
    • A
      Repair/instate jemalloc build on M1 (#11257) · 6eb3770b
      Alan Paxton 提交于
      Summary:
      jemalloc was not building on M1 Macs. This makes it work.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11257
      
      Reviewed By: anand1976
      
      Differential Revision: D45959570
      
      Pulled By: ajkr
      
      fbshipit-source-id: 08c2b81b399f5003a2c159d037f9bcc5d0059556
      6eb3770b
  7. 20 5月, 2023 2 次提交
    • Y
      Update HISTORY.md/version.h/format compatiblity test for 8.3 release (#11464) · 509116c5
      Yu Zhang 提交于
      Summary: Pull Request resolved: https://github.com/facebook/rocksdb/pull/11464
      
      Reviewed By: akankshamahajan15
      
      Differential Revision: D46041333
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: 7d83cf9e611451fcc7f7e4a837681ed0d4271df4
      509116c5
    • P
      Much better stats for seeks and prefix filtering (#11460) · 39f5846e
      Peter Dillinger 提交于
      Summary:
      We want to know more about opportunities for better range filters, and the effectiveness of our own range filters. Currently the stats are very limited, essentially logging just hits and misses against prefix filters for range scans in BLOOM_FILTER_PREFIX_* without tracking the false positive rate. Perhaps confusingly, when prefix filters are used for point queries, the stats are currently going into the non-PREFIX tickers.
      
      This change does several things:
      * Introduce new stat tickers for seeks and related filtering, \*LEVEL_SEEK\*
        * Most importantly, allows us to see opportunities for range filtering. Specifically, we can count how many times a seek in an SST file accesses at least one data block, and how many times at least one value() is then accessed. If a data block was accessed but no value(), we can generally assume that the key(s) seen was(were) not of interest so could have been filtered with the right kind of filter, avoiding the data block access.
        * We can get the same level of detail when a filter (for now, prefix Bloom/ribbon) is used, or not. Specifically, we can infer a false positive rate for prefix filters (not available before) from the seek "false positive" rate: when a data block is accessed but no value() is called. (There can be other explanations for a seek false positive, but in typical iterator usage it would indicate a filter false positive.)
        * For efficiency, I wanted to avoid making additional calls to the prefix extractor (or key comparisons, etc.), which would be required if we wanted to more precisely detect filter false positives. I believe that instrumenting value() is the best balance of efficiency vs. accurately measuring what we are often interested in.
        * The stats are divided between last level and non-last levels, to help understand potential tiered storage use cases.
      * The old BLOOM_FILTER_PREFIX_* stats have a different meaning: no longer referring to iterators but to point queries using prefix filters. BLOOM_FILTER_PREFIX_TRUE_POSITIVE is added for computing the prefix false positive rate on point queries, which can be due to filter false positives as well as different keys with the same prefix.
      * Similarly, the non-PREFIX BLOOM_FILTER stats are now for whole key filtering only.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11460
      
      Test Plan:
      unit tests updated, including updating many to pop the stat value since last read to improve test
      readability and maintainability.
      
      Performance test shows a consistent small improvement with these changes, both with clang and with gcc. CPU profile indicates that RecordTick is using less CPU, and this makes sense at least for a high filter miss rate. Before, we were recording two ticks per filter miss in iterators (CHECKED & USEFUL) and now recording just one (FILTERED).
      
      Create DB with
      ```
      TEST_TMPDIR=/dev/shm ./db_bench -benchmarks=fillrandom -num=10000000 -disable_wal=1 -write_buffer_size=30000000 -bloom_bits=8 -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=0 -prefix_size=8
      ```
      And run simultaneous before&after with
      ```
      TEST_TMPDIR=/dev/shm ./db_bench -readonly -benchmarks=seekrandom[-X1000] -num=10000000 -bloom_bits=8 -compaction_style=2 -fifo_compaction_max_table_files_size_mb=10000 -fifo_compaction_allow_compaction=0 -prefix_size=8 -seek_nexts=1 -duration=20 -seed=43 -threads=8 -cache_size=1000000000 -statistics
      ```
      Before: seekrandom [AVG 275 runs] : 189680 (± 222) ops/sec;   18.4 (± 0.0) MB/sec
      After: seekrandom [AVG 275 runs] : 197110 (± 208) ops/sec;   19.1 (± 0.0) MB/sec
      
      Reviewed By: ajkr
      
      Differential Revision: D46029177
      
      Pulled By: pdillinger
      
      fbshipit-source-id: cdace79a2ea548d46c5900b068c5b7c3a02e5822
      39f5846e
  8. 19 5月, 2023 4 次提交
    • P
      Compatibility step for separating BlockCache and GeneralCache APIs (#11450) · 4067acab
      Peter Dillinger 提交于
      Summary:
      Add two type aliases for Cache: BlockCache and GeneralCache, and add LRUCacheOptions::MakeSharedGeneralCache(). This will ease upgrade to an intended future change to separate the cache API between block cache and other (general) uses, including row cache. Separating the APIs will make it easier to expose more details of block caching for customization. For example, it would be nice to pass the file unique ID and offset as the logical cache key instead of using a Slice, which could facilitate some file-specific customizations in block cache. This would also make it clear that HyperClockCache is not usable as a general cache, because it can only deal with fixed-size block cache keys.
      
      block_cache, row_cache, and blob_cache are the uses of Cache in the public API. blob_cache should be able to use BlockCache while row_cache is a GeneralCache user, as its keys are of arbitrary size.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11450
      
      Test Plan: see updated unit test.
      
      Reviewed By: anand1976
      
      Differential Revision: D45882067
      
      Pulled By: pdillinger
      
      fbshipit-source-id: ff5d9f0b644f87ae337a29a7728ce3ed07b2a4b2
      4067acab
    • M
      Support Clip DB to KeyRange (#11379) · 8d8eb0e7
      mayue.fight 提交于
      Summary:
      This PR is part of the request https://github.com/facebook/rocksdb/issues/11317.
      (Another part is https://github.com/facebook/rocksdb/pull/11378)
      
      ClipDB() will clip the entries in the CF according to the range [begin_key, end_key). All the entries outside this range will be completely deleted (including tombstones).
       This feature is mainly used to ensure that there is no overlapping Key when calling CreateColumnFamilyWithImports() to import multiple CFs.
      
      When Calling ClipDB [begin, end), there are the following steps
      
      1.  Quickly and directly delete files without overlap
       DeleteFilesInRanges(nullptr, begin) + DeleteFilesInRanges(end, nullptr)
      2. Delete the Key outside the range
      Delete[smallest_key, begin) + Delete[end, largest_key]
      3. Delete the tombstone through Manul Compact
      CompactRange(option, nullptr, nullptr)
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11379
      
      Reviewed By: ajkr
      
      Differential Revision: D45840358
      
      Pulled By: cbi42
      
      fbshipit-source-id: 54152e8a45fd8ede137f99787eb252f0b51440a4
      8d8eb0e7
    • H
      Improve comment of ExpectedValue in db stress (#11456) · 7263f51d
      Hui Xiao 提交于
      Summary:
      **Context/Summary:**
      https://github.com/facebook/rocksdb/pull/11424 made me realize there are a couple gaps in my `ExpectedValue` comments so I updated them, along with separating `ExpectedValue` into separate files so it's clearer that `ExpectedValue` can be used without updating `ExpectedState` (e.g, TestMultiGet() where we care about value base of expected value but not updating the ExpectedState).
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11456
      
      Test Plan: CI
      
      Reviewed By: jowlyzhang
      
      Differential Revision: D45965070
      
      Pulled By: hx235
      
      fbshipit-source-id: dcee690c13b00a3119757ea9d43b646f9644e1a9
      7263f51d
    • H
      Add `rocksdb.file.read.db.open.micros` (#11455) · 50046869
      Hui Xiao 提交于
      Summary:
      **Context/Summary:**
      `rocksdb.file.read.db.open.micros` is left out in https://github.com/facebook/rocksdb/pull/11288
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11455
      
      Test Plan:
      - db bench
      Setup: `./db_bench -db=/dev/shm/testdb/ -statistics=true -benchmarks="fillseq" -key_size=32 -value_size=512 -num=5000 -write_buffer_size=655 -target_file_size_base=655 -disable_auto_compactions=false -compression_type=none -bloom_bits=3`
      Run:
      `./db_bench --bloom_bits=3 --use_existing_db=1 --seed=1682546046158958 --partition_index_and_filters=1 --statistics=1 -db=/dev/shm/testdb/  -benchmarks=readrandom  -key_size=3200 -value_size=512 -num=0 -write_buffer_size=6550000 -disable_auto_compactions=false -target_file_size_base=6550000 -compression_type=none -file_checksum=1 -cache_size=1`
      
      ```
      rocksdb.sst.read.micros P50 : 3.979798 P95 : 9.738420 P99 : 19.566667 P100 : 39.000000 COUNT : 2360 SUM : 12148
      rocksdb.file.read.flush.micros P50 : 0.000000 P95 : 0.000000 P99 : 0.000000 P100 : 0.000000 COUNT : 0 SUM : 0
      rocksdb.file.read.compaction.micros P50 : 0.000000 P95 : 0.000000 P99 : 0.000000 P100 : 0.000000 COUNT : 0 SUM : 0
      rocksdb.file.read.db.open.micros P50 : 3.979798 P95 : 9.738420 P99 : 19.566667 P100 : 39.000000 COUNT : 2360 SUM : 12148
      ```
      
      Reviewed By: ajkr
      
      Differential Revision: D45951934
      
      Pulled By: hx235
      
      fbshipit-source-id: 6c88639dc1b10d98ecccc963ce32a8800495f55b
      50046869
  9. 18 5月, 2023 3 次提交
    • A
      Minimal RocksJava compliance with Java 8 language level (EB 1046) (#10951) · e110d713
      Alan Paxton 提交于
      Summary:
      Apply a small (and automatic) set of IntelliJ Java inspections/repairs to the Java interface to RocksDB Java and its tests.
      Partly enabled by the fact that we now (from RocksDB7) require java 8.
      
      Explicit <p> in empty lines in javadoc comments.
      
      Parameters and variables made final where possible.
      Anonymous subclasses converted lambdas.
      
      Some tests which previously used other assertion models were converted to assertj, e.g. (assertThat(actual).isEqualTo(expected)
      
      In a very few cases tests were found to be inoperative or broken, and were repaired. No problems with actual RocksDB behaviour were observed.
      
      This PR is intended to replace https://github.com/facebook/rocksdb/pull/9618 - that PR was not merged, and attempts to rebase it have yielded a questionable looking diff, so we choose to go back to square 1 here, and implement a conservative set of changes.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/10951
      
      Reviewed By: anand1976
      
      Differential Revision: D45057849
      
      Pulled By: ajkr
      
      fbshipit-source-id: e4ea46bfc80518ae86f37702b03ca9352bc11c3d
      e110d713
    • J
      Remove wait_unscheduled from waitForCompact internal API (#11443) · 586d78b3
      Jay Huh 提交于
      Summary:
      Context:
      
      In pull request https://github.com/facebook/rocksdb/issues/11436, we are introducing a new public API `waitForCompact(const WaitForCompactOptions& wait_for_compact_options)`. This API invokes the internal implementation `waitForCompact(bool wait_unscheduled=false)`. The unscheduled parameter indicates the compactions that are not yet scheduled but are required to process items in the queue.
      
      In certain cases, we are unable to wait for compactions, such as during a shutdown or when background jobs are paused. It is important to return the appropriate status in these scenarios. For all other cases, we should wait for all compaction and flush jobs, including the unscheduled ones. The primary purpose of this new API is to wait until the system has resolved its compaction debt. Currently, the usage of `wait_unscheduled` is limited to test code.
      
      This pull request eliminates the usage of wait_unscheduled. The internal `waitForCompact()` API now waits for unscheduled compactions unless the db is undergoing a shutdown. In the event of a shutdown, the API returns `Status::ShutdownInProgress()`.
      
      Additionally, a new parameter, `abort_on_pause`, has been introduced with a default value of `false`. This parameter addresses the possibility of waiting indefinitely for unscheduled jobs if `PauseBackgroundWork()` was called before `waitForCompact()` is invoked. By setting `abort_on_pause` to `true`, the API will immediately return `Status::Aborted`.
      
      Furthermore, all tests that previously called `waitForCompact(true)` have been fixed.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11443
      
      Test Plan:
      Existing tests that involve a shutdown in progress:
      
      - DBCompactionTest::CompactRangeShutdownWhileDelayed
      - DBTestWithParam::PreShutdownMultipleCompaction
      - DBTestWithParam::PreShutdownCompactionMiddle
      
      Reviewed By: pdillinger
      
      Differential Revision: D45923426
      
      Pulled By: jaykorean
      
      fbshipit-source-id: 7dc93fe6a6841a7d9d2d72866fa647090dba8eae
      586d78b3
    • P
      Change internal headers with duplicate names (#11408) · 206fdea3
      Peter Dillinger 提交于
      Summary:
      In IDE navigation I find it annoying that there are two statistics.h files (etc.) and often land on the wrong one. Here I migrate several headers to use the blah.h <- blah_impl.h <- blah.cc idiom. Although clang-format wants "blah.h" to be the top include for "blah.cc", I think overall this is an improvement.
      
      No public API changes.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11408
      
      Test Plan: existing tests
      
      Reviewed By: ltamasi
      
      Differential Revision: D45456696
      
      Pulled By: pdillinger
      
      fbshipit-source-id: 809d931253f3272c908cf5facf7e1d32fc507373
      206fdea3
  10. 16 5月, 2023 2 次提交
    • H
      Support parallel read and write/delete to same key in NonBatchedOpsStressTest (#11058) · 5fc57eec
      Hui Xiao 提交于
      Summary:
      **Context:**
      Current `NonBatchedOpsStressTest` does not allow multi-thread read (i.e, Get, Iterator) and write (i.e, Put, Merge) or delete to the same key. Every read or write/delete operation will acquire lock (`GetLocksForKeyRange`) on the target key to gain exclusive access to it. This does not align with RocksDB's nature of allowing multi-thread read and write/delete to the same key, that is concurrent threads can issue read/write/delete to RocksDB without external locking. Therefore this is a gap in our testing coverage.
      
      To close the gap, biggest challenge remains in verifying db value against expected state in presence of parallel read and write/delete. The challenge is due to read/write/delete to the db and read/write to expected state is not within one atomic operation. Therefore we may not know the exact expected state of a certain db read, as by the time we read the expected state for that db read, another write to expected state for another db write to the same key might have changed the expected state.
      
      **Summary:**
      Credited to ajkr's idea, we now solve this challenge by breaking the 32-bits expected value of a key into different parts that can be read and write to in parallel.
      
      Basically we divide the 32-bits expected value into `value_base` (corresponding to the previous whole 32 bits but now with some shrinking in the value base range we allow), `pending_write` (i.e, whether there is an ongoing concurrent write), `del_counter` (i.e, number of times a value has been deleted, analogous to value_base for write), `pending_delete` (similar to pending_write) and `deleted` (i.e whether a key is deleted).
      
      Also, we need to use incremental `value_base` instead of random value base as before because we want to control the range of value base a correct db read result can possibly be in presence of parallel read and write. In that way, we can verify the correctness of the read against expected state more easily. This is at the cost of reducing the randomness of the value generated in NonBatchedOpsStressTest we are willing to accept.
      
      (For detailed algorithm of how to use these parts to infer expected state of a key, see the PR)
      
      Misc: hide value_base detail from callers of ExpectedState by abstracting related logics into ExpectedValue class
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11058
      
      Test Plan:
      - Manual test of small number of keys (i.e, high chances of parallel read and write/delete to same key) with equally distributed read/write/deleted for 30 min
      ```
      python3 tools/db_crashtest.py --simple {blackbox|whitebox} --sync_fault_injection=1 --skip_verifydb=0 --continuous_verification_interval=1000 --clear_column_family_one_in=0 --max_key=10 --column_families=1 --threads=32 --readpercent=25 --writepercent=25 --nooverwritepercent=0 --iterpercent=25 --verify_iterator_with_expected_state_one_in=1 --num_iterations=5 --delpercent=15 --delrangepercent=10 --range_deletion_width=5 --use_merge={0|1} --use_put_entity_one_in=0 --use_txn=0 --verify_before_write=0 --user_timestamp_size=0 --compact_files_one_in=1000 --compact_range_one_in=1000 --flush_one_in=1000 --get_property_one_in=1000 --ingest_external_file_one_in=100 --backup_one_in=100 --checkpoint_one_in=100 --approximate_size_one_in=0 --acquire_snapshot_one_in=100 --use_multiget=0 --prefixpercent=0 --get_live_files_one_in=1000 --manual_wal_flush_one_in=1000 --pause_background_one_in=1000 --target_file_size_base=524288 --write_buffer_size=524288 --verify_checksum_one_in=1000 --verify_db_one_in=1000
      ```
      - Rehearsal stress test for normal parameter and aggressive parameter to see if such change can find what existing stress test can find (i.e, no regression in testing capability)
      - [Ongoing]Try to find new bugs with this change that are not found by current NonBatchedOpsStressTest with no parallel read and write/delete to same key
      
      Reviewed By: ajkr
      
      Differential Revision: D42257258
      
      Pulled By: hx235
      
      fbshipit-source-id: e6fdc18f1fad3753e5ac91731483a644d9b5b6eb
      5fc57eec
    • A
      Fix write stall stats dump format (#11445) · fb636f24
      Andrew Kryczka 提交于
      Summary:
      I noticed in https://github.com/facebook/rocksdb/issues/11426 there is a missing line break. Before this PR the output looked like
      
      ```
      $ ./db_bench -stats_per_interval=1 -stats_interval=100000
      ...
      Write Stall (count): cf-l0-file-count-limit-delays-with-ongoing-compaction: 0, cf-l0-file-count-limit-stops-with-ongoing-compaction: 0, l0-file-count-limit-delays: 0, l0-file-count-limit-stops: 0, memtable-limit-delays: 0, memtable-limit-stops: 0, pending-compaction-bytes-delays: 0, pending-compaction-bytes-stops: 0, total-delays: 0, total-stops: 0, Block cache LRUCache@0x7f8695831b50#2766536 capacity: 32.00 MB seed: 1155354975 usage: 0.09 KB table_size: 1024 occupancy: 1 collections: 1 last_copies: 0 last_secs: 9.3e-05 secs_since: 2
      ...
      Write Stall (count): write-buffer-manager-limit-stops: 0, num-running-compactions: 0
      ...
      ```
      
      After this PR it looks like
      
      ```
      $ ./db_bench -stats_per_interval=1 -stats_interval=100000
      ...
      Write Stall (count): cf-l0-file-count-limit-delays-with-ongoing-compaction: 0, cf-l0-file-count-limit-stops-with-ongoing-compaction: 0, l0-file-count-limit-delays: 0, l0-file-count-limit-stops: 0, memtable-limit-delays: 0, memtable-limit-stops: 0, pending-compaction-bytes-delays: 0, pending-compaction-bytes-stops: 0, total-delays: 0, total-stops: 0
      Block cache LRUCache@0x7f8e0d231b50#2736585 capacity: 32.00 MB seed: 920433955 usage: 0.09 KB table_size: 1024 occupancy: 1 collections: 1 last_copies: 1 last_secs: 6.5e-05 secs_since: 4
      ...
      Write Stall (count): write-buffer-manager-limit-stops: 0
      num-running-compactions: 0
      ...
      ```
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11445
      
      Reviewed By: hx235
      
      Differential Revision: D45844752
      
      Pulled By: ajkr
      
      fbshipit-source-id: 1c708cb05b6e270922ac2fa95f5d011f273347eb
      fb636f24
  11. 13 5月, 2023 2 次提交
    • A
      Delete temp OPTIONS file on failure to write it (#11423) · 2084cdf2
      anand76 提交于
      Summary:
      When the DB is opened, RocksDB creates a temp OPTIONS file, writes the current options to it, and renames it. In case of a failure, the temp file is left behind, and is not deleted by PurgeObsoleteFiles(). Fix this by explicitly deleting the temp file if writing to it or renaming it fails.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11423
      
      Test Plan: Add a unit test
      
      Reviewed By: akankshamahajan15
      
      Differential Revision: D45540454
      
      Pulled By: anand1976
      
      fbshipit-source-id: 47facdc30d8cc5667036312d04b21d3fc253c92e
      2084cdf2
    • A
      Add block checksum mismatch ticker stat (#11438) · 113f3250
      Andrew Kryczka 提交于
      Summary:
      Added a ticker stat, `BLOCK_CHECKSUM_MISMATCH_COUNT`, to count how many block checksum verifications detected a mismatch.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11438
      
      Test Plan: new unit test
      
      Reviewed By: pdillinger
      
      Differential Revision: D45788179
      
      Pulled By: ajkr
      
      fbshipit-source-id: e2b44eba7c23b3e110ebe69eaa78a710dec2590f
      113f3250
  12. 12 5月, 2023 2 次提交
    • Y
      Add support in log writer and reader for a user-defined timestamp size record (#11433) · 47235dda
      Yu Zhang 提交于
      Summary:
      This patch adds support to write and read a user-defined timestamp size record in log writer and log reader. It will be used by WAL logs to persist the user-defined timestamp format for subsequent WriteBatch records. Reading and writing UDT sizes for WAL logs are not included in this patch. It will be in a follow up.
      
      The syntax for the record is: at write time, one such record is added when log writer encountered any non-zero UDT size it hasn't recorded so far. At read time, all such records read up to a point are accumulated and applicable to all subsequent WriteBatch records.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11433
      
      Test Plan:
      ```
      make clean && make -j32 all
      ./log_test --gtest_filter="*WithTimestampSize*"
      ```
      
      Reviewed By: ltamasi
      
      Differential Revision: D45678708
      
      Pulled By: jowlyzhang
      
      fbshipit-source-id: b770c8f45bb7b9383b14aac9f22af781304fb41d
      47235dda
    • C
      Support compacting files to different temperatures in FIFO compaction (#11428) · 8827cd06
      Changyu Bi 提交于
      Summary:
      - Add a new option `CompactionOptionsFIFO::file_temperature_age_thresholds` that allows user to specify age thresholds for compacting files to different temperatures. File temperature can be used to store files in different storage media. The new options allows specifying multiple temperature-age pairs. The option uses struct for a temperature-age pair to use the existing parsing functionality to make the option dynamically settable.
      - Deprecate the old option `age_for_warm` that was added for a similar purpose.
      - Compaction score calculation logic is updated to check if a file needs to be compacted to change its temperature.
      - Some refactoring is done in `FIFOCompactionPicker::PickTemperatureChangeCompaction`.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11428
      
      Test Plan: adapted unit tests that were for `age_for_warm` to this new option.
      
      Reviewed By: ajkr
      
      Differential Revision: D45611412
      
      Pulled By: cbi42
      
      fbshipit-source-id: 2dc384841f61cc04abb9681e31aa2de0f0b06106
      8827cd06
  13. 11 5月, 2023 1 次提交
    • J
      Clean up rate limiter refill logic (#11425) · 7531cbda
      Jay Huh 提交于
      Summary:
      Context:
      
      This pull request update is in response to a comment made on https://github.com/facebook/rocksdb/pull/8596#discussion_r680264932. The current implementation of RefillBytesAndGrantRequestsLocked() drains all available_bytes, but the first request after the last wave of requesting/bytes granting is done is not being handled in the same way.
      
      This creates a scenario where if a request for a large amount of bytes is enqueued first, but there are not enough available_bytes to fulfill it, the request is put to sleep until the next refill time. Meanwhile, a later request for a smaller number of bytes comes in and is granted immediately. This behavior is not fair as the earlier request was made first.
      
      To address this issue, we have made changes to the code to exhaust the remaining available bytes from the request and queue the remaining. With this change, requests are granted in the order they are received, ensuring that earlier requests are not unfairly delayed by later, smaller requests. The specific scenario described above will no longer occur with this change. Also consolidated `granted` and `request_bytes` as part of the change since `granted` is equivalent to `request_bytes == 0`
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11425
      
      Test Plan: Added `AvailableByteSizeExhaustTest`
      
      Reviewed By: hx235
      
      Differential Revision: D45570711
      
      Pulled By: jaykorean
      
      fbshipit-source-id: a7117ed17bf4b8a7ae0f76124cb41902db1a2592
      7531cbda
  14. 10 5月, 2023 3 次提交
    • P
      Simplify detection of x86 CPU features (#11419) · 459969e9
      Peter Dillinger 提交于
      Summary:
      **Background** - runtime detection of certain x86 CPU features was added for optimizing CRC32c checksums, where performance is dramatically affected by the availability of certain CPU instructions and code using intrinsics for those instructions. And Java builds with native library try to be broadly compatible but performant.
      
      What has changed is that CRC32c is no longer the most efficient cheecksum on contemporary x86_64 hardware, nor the default checksum. XXH3 is generally faster and not as dramatically impacted by the availability of certain CPU instructions. For example, on my Skylake system using db_bench (similar on an older Skylake system without AVX512):
      
      PORTABLE=1 empty USE_SSE  : xxh3->8 GB/s   crc32c->0.8 GB/s  (no SSE4.2 nor AVX2 instructions)
      PORTABLE=1 USE_SSE=1      : xxh3->19 GB/s  crc32c->16 GB/s  (with SSE4.2 and AVX2)
      PORTABLE=0 USE_SSE ignored: xxh3->28 GB/s  crc32c->16 GB/s  (also some AVX512)
      
      Testing a ~10 year old system, with SSE4.2 but without AVX2, crc32c is a similar speed to the new systems but xxh3 is only about half that speed, also 8GB/s like the non-AVX2 compile above. Given that xxh3 has specific optimization for AVX2, I think we can infer that that crc32c is only fastest for that ~2008-2013 period when SSE4.2 was included but not AVX2. And given that xxh3 is only about 2x slower on these systems (not like >10x slower for unoptimized crc32c), I don't think we need to invest too much in optimally adapting to these old cases.
      
      x86 hardware that doesn't support fast CRC32c is now extremely rare, so requiring a custom build to support such hardware is fine IMHO.
      
      **This change** does two related things:
      * Remove runtime CPU detection for optimizing CRC32c on x86. Maintaining this code is non-zero work, and compiling special code that doesn't work on the configured target instruction set for code generation is always dubious. (On the one hand we have to ensure the CRC32c code uses SSE4.2 but on the other hand we have to ensure nothing else does.)
      * Detect CPU features in source code, not in build scripts. Although there are some hypothetical advantages to detectiong in build scripts (compiler generality), RocksDB supports at least three build systems: make, cmake, and buck. It's not practical to support feature detection on all three, and we have suffered from missed optimization opportunities by relying on missing or incomplete detection in cmake and buck. We also depend on some components like xxhash that do source code detection anyway.
      
      **In more detail:**
      * `HAVE_SSE42`, `HAVE_AVX2`, and `HAVE_PCLMUL` replaced by standard macros `__SSE4_2__`, `__AVX2__`, and `__PCLMUL__`.
      * MSVC does not provide high fidelity defines for SSE, PCLMUL, or POPCNT, but we can infer those from `__AVX__` or `__AVX2__` in a compatibility header. In rare cases of false negative or false positive feature detection, a build engineer should be able to set defines to work around the issue.
      * `__POPCNT__` is another standard define, but we happen to only need it on MSVC, where it is set by that compatibility header, or can be set by the build engineer.
      * `PORTABLE` can be set to a CPU type, e.g. "haswell", to compile for that CPU type.
      * `USE_SSE` is deprecated, now equivalent to PORTABLE=haswell, which roughly approximates its old behavior.
      
      Notably, this change should enable more builds to use the AVX2-optimized Bloom filter implementation.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11419
      
      Test Plan:
      existing tests, CI
      
      Manual performance tests after the change match the before above (none expected with make build).
      
      We also see AVX2 optimized Bloom filter code enabled when expected, by injecting a compiler error. (Performance difference is not big on my current CPU.)
      
      Reviewed By: ajkr
      
      Differential Revision: D45489041
      
      Pulled By: pdillinger
      
      fbshipit-source-id: 60ceb0dd2aa3b365c99ed08a8b2a087a9abb6a70
      459969e9
    • P
      Add hash_seed to Caches (#11391) · f4a02f2c
      Peter Dillinger 提交于
      Summary:
      See motivation and description in new ShardedCacheOptions::hash_seed option.
      
      Updated db_bench so that its seed param is used for the cache hash seed.
      Made its code more safe to ensure seed is set before use.
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11391
      
      Test Plan:
      unit tests added / updated
      
      **Performance** - no discernible difference seen running cache_bench repeatedly before & after. With lru_cache and hyper_clock_cache.
      
      Reviewed By: hx235
      
      Differential Revision: D45557797
      
      Pulled By: pdillinger
      
      fbshipit-source-id: 40bf4da6d66f9d41a8a0eb8e5cf4246a4aa07934
      f4a02f2c
    • A
      Fix build error: variable 'base_level' may be uninitialized (#11435) · 6ba4717f
      akankshamahajan 提交于
      Summary:
      Fix build error: variable 'base_level' may be uninitialized
      ```
       db_impl_compaction_flush.cc:1195:21: error: variable 'base_level' may be uninitialized when used here [-Werror,-Wconditional-uninitialized]
                  level = base_level;
      ```
                          ^~~~~~~~~~
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11435
      
      Test Plan: CircleCI jobs
      
      Reviewed By: cbi42
      
      Differential Revision: D45708176
      
      Pulled By: akankshamahajan15
      
      fbshipit-source-id: 851b1205b22b63d728495e5735fa91b0ad8e012b
      6ba4717f
  15. 09 5月, 2023 1 次提交
    • H
      Record and use the tail size to prefetch table tail (#11406) · 8f763bde
      Hui Xiao 提交于
      Summary:
      **Context:**
      We prefetch the tail part of a SST file (i.e, the blocks after data blocks till the end of the file) during each SST file open in hope to prefetch all the stuff at once ahead of time for later read e.g, footer, meta index, filter/index etc. The existing approach to estimate the tail size to prefetch is through `TailPrefetchStats` heuristics introduced in https://github.com/facebook/rocksdb/pull/4156, which has caused small reads in unlucky case (e.g,  small read into the tail buffer during table open in thread 1 under the same BlockBasedTableFactory object can make thread 2's tail prefetching use a small size that it shouldn't) and is hard to debug.  Therefore we decide to record the exact tail size and use it directly  to prefetch tail of the SST instead of relying heuristics.
      
      **Summary:**
      - Obtain and record in manifest the tail size in `BlockBasedTableBuilder::Finish()`
         - For backward compatibility, we fall back to TailPrefetchStats and last to simple heuristics that the tail size is a linear portion of the file size - see PR conversation for more.
      - Make`tail_start_offset` part of the table properties and deduct tail size to record in manifest for external files (e.g, file ingestion, import CF) and db repair (with no access to manifest).
      
      Pull Request resolved: https://github.com/facebook/rocksdb/pull/11406
      
      Test Plan:
      1. New UT
      2. db bench
      Note: db bench on /tmp/ where direct read is supported is too slow to finish and the default pinning setting in db bench is not helpful to profile # sst read of Get. Therefore I hacked the following to obtain the following comparison.
      ```
       diff --git a/table/block_based/block_based_table_reader.cc b/table/block_based/block_based_table_reader.cc
      index bd5669f0f..791484c1f 100644
       --- a/table/block_based/block_based_table_reader.cc
      +++ b/table/block_based/block_based_table_reader.cc
      @@ -838,7 +838,7 @@ Status BlockBasedTable::PrefetchTail(
                                  &tail_prefetch_size);
      
         // Try file system prefetch
      -  if (!file->use_direct_io() && !force_direct_prefetch) {
      +  if (false && !file->use_direct_io() && !force_direct_prefetch) {
           if (!file->Prefetch(prefetch_off, prefetch_len, ro.rate_limiter_priority)
                    .IsNotSupported()) {
             prefetch_buffer->reset(new FilePrefetchBuffer(
       diff --git a/tools/db_bench_tool.cc b/tools/db_bench_tool.cc
      index ea40f5fa0..39a0ac385 100644
       --- a/tools/db_bench_tool.cc
      +++ b/tools/db_bench_tool.cc
      @@ -4191,6 +4191,8 @@ class Benchmark {
                 std::shared_ptr<TableFactory>(NewCuckooTableFactory(table_options));
           } else {
             BlockBasedTableOptions block_based_options;
      +      block_based_options.metadata_cache_options.partition_pinning =
      +      PinningTier::kAll;
             block_based_options.checksum =
                 static_cast<ChecksumType>(FLAGS_checksum_type);
             if (FLAGS_use_hash_search) {
      ```
      Create DB
      ```
      ./db_bench --bloom_bits=3 --use_existing_db=1 --seed=1682546046158958 --partition_index_and_filters=1 --statistics=1 -db=/dev/shm/testdb/ -benchmarks=readrandom -key_size=3200 -value_size=512 -num=1000000 -write_buffer_size=6550000 -disable_auto_compactions=false -target_file_size_base=6550000 -compression_type=none
      ```
      ReadRandom
      ```
      ./db_bench --bloom_bits=3 --use_existing_db=1 --seed=1682546046158958 --partition_index_and_filters=1 --statistics=1 -db=/dev/shm/testdb/ -benchmarks=readrandom -key_size=3200 -value_size=512 -num=1000000 -write_buffer_size=6550000 -disable_auto_compactions=false -target_file_size_base=6550000 -compression_type=none
      ```
      (a) Existing (Use TailPrefetchStats for tail size + use seperate prefetch buffer in PartitionedFilter/IndexReader::CacheDependencies())
      ```
      rocksdb.table.open.prefetch.tail.hit COUNT : 3395
      rocksdb.sst.read.micros P50 : 5.655570 P95 : 9.931396 P99 : 14.845454 P100 : 585.000000 COUNT : 999905 SUM : 6590614
      ```
      
      (b) This PR (Record tail size + use the same tail buffer in PartitionedFilter/IndexReader::CacheDependencies())
      ```
      rocksdb.table.open.prefetch.tail.hit COUNT : 14257
      rocksdb.sst.read.micros P50 : 5.173347 P95 : 9.015017 P99 : 12.912610 P100 : 228.000000 COUNT : 998547 SUM : 5976540
      ```
      
      As we can see, we increase the prefetch tail hit count and decrease SST read count with this PR
      
      3. Test backward compatibility by stepping through reading with post-PR code on a db generated pre-PR.
      
      Reviewed By: pdillinger
      
      Differential Revision: D45413346
      
      Pulled By: hx235
      
      fbshipit-source-id: 7d5e36a60a72477218f79905168d688452a4c064
      8f763bde