Function shared_write_buffer_manager

src/rocksdb/tests/basic.rs:313–412

unsupported operation: can't call foreign function `rocksdb_create_default_env` on OS `linux`

()

#[mz_ore::test(tokio::test)]
#[cfg_attr(miri, ignore)] // unsupported operation: can't call foreign function `rocksdb_create_default_env` on OS `linux`
async fn shared_write_buffer_manager() -> Result<(), anyhow::Error> {
    // If the test aborts, this may not be cleaned up.
    let t = tempfile::tempdir()?;

    let write_buffer_memory_bytes = 10000;

    let shared_write_buffer_manager: SharedWriteBufferManager = SharedWriteBufferManager::default();
    let mut tuning_parameters: RocksDBTuningParameters = RocksDBTuningParameters {
        write_buffer_manager_memory_bytes: Some(write_buffer_memory_bytes),
        ..Default::default()
    };
    let mut rocksdb_config = RocksDBConfig::new(shared_write_buffer_manager.clone(), None);
    rocksdb_config.apply(tuning_parameters.clone());

    let instance1 = RocksDBInstance::<String, String>::new(
        t.path().join("1").as_path(),
        InstanceOptions::<bincode::DefaultOptions, String, StubMergeOperator<String>>::new(
            rocksdb::Env::new()?,
            2,
            None,
            bincode::DefaultOptions::new(),
        ),
        rocksdb_config.clone(),
        shared_metrics_for_tests()?,
        instance_metrics_for_tests()?,
    )?;

    // this is a no-op, but it won't return until instance1 has started
    instance1.manual_compaction().await?;

    {
        // Arc will be dropped by the end of this scope
        let buf = shared_write_buffer_manager.get().unwrap();
        assert!(buf.enabled());
        assert_eq!(write_buffer_memory_bytes, buf.get_buffer_size());
    }

    let updated_bytes = 20000;
    tuning_parameters.write_buffer_manager_memory_bytes = Some(updated_bytes);
    rocksdb_config.apply(tuning_parameters);

    let instance2 = RocksDBInstance::<String, String>::new(
        t.path().join("2").as_path(),
        InstanceOptions::<bincode::DefaultOptions, String, StubMergeOperator<String>>::new(
            rocksdb::Env::new()?,
            2,
            None,
            bincode::DefaultOptions::new(),
        ),
        rocksdb_config.clone(),
        shared_metrics_for_tests()?,
        instance_metrics_for_tests()?,
    )?;

    // this is a no-op, but it won't return until instance2 has started
    instance2.manual_compaction().await?;

    instance1.close().await?;